IDEA-285172 - [decompiler] - StrongConnectivityHelper refactoring
[idea/community.git] / plugins / java-decompiler / engine / src / org / jetbrains / java / decompiler / modules / renamer / ConverterHelper.java
1 // Copyright 2000-2017 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
2 package org.jetbrains.java.decompiler.modules.renamer;
3
4 import org.jetbrains.java.decompiler.code.CodeConstants;
5 import org.jetbrains.java.decompiler.main.extern.IIdentifierRenamer;
6
7 import java.util.Arrays;
8 import java.util.HashSet;
9 import java.util.Locale;
10 import java.util.Set;
11
12 public class ConverterHelper implements IIdentifierRenamer {
13   private static final Set<String> KEYWORDS = new HashSet<>(Arrays.asList(
14     "abstract", "do", "if", "package", "synchronized", "boolean", "double", "implements", "private", "this", "break", "else", "import",
15     "protected", "throw", "byte", "extends", "instanceof", "public", "throws", "case", "false", "int", "return", "transient", "catch",
16     "final", "interface", "short", "true", "char", "finally", "long", "static", "try", "class", "float", "native", "strictfp", "void",
17     "const", "for", "new", "super", "volatile", "continue", "goto", "null", "switch", "while", "default", "assert", "enum"));
18   private static final Set<String> RESERVED_WINDOWS_NAMESPACE = new HashSet<>(Arrays.asList(
19     "con", "prn", "aux", "nul",
20     "com1", "com2", "com3", "com4", "com5", "com6", "com7", "com8", "com9",
21     "lpt1", "lpt2", "lpt3", "lpt4", "lpt5", "lpt6", "lpt7", "lpt8", "lpt9"));
22
23   private int classCounter = 0;
24   private int fieldCounter = 0;
25   private int methodCounter = 0;
26   private final Set<String> setNonStandardClassNames = new HashSet<>();
27
28   @Override
29   public boolean toBeRenamed(Type elementType, String className, String element, String descriptor) {
30     String value = elementType == Type.ELEMENT_CLASS ? className : element;
31     return value == null ||
32            value.length() <= 2 ||
33            !isValidIdentifier(elementType == Type.ELEMENT_METHOD, value) ||
34            KEYWORDS.contains(value) ||
35            elementType == Type.ELEMENT_CLASS && (
36              RESERVED_WINDOWS_NAMESPACE.contains(value.toLowerCase(Locale.ENGLISH)) ||
37              value.length() > 255 - ".class".length());
38   }
39
40   /**
41    * Return {@code true} if, and only if identifier passed is compliant to JLS9 section 3.8 AND DOES NOT CONTAINS so-called "ignorable" characters.
42    * Ignorable characters are removed by javac silently during compilation and thus may appear only in specially crafted obfuscated classes.
43    * For more information about "ignorable" characters see <a href="https://bugs.openjdk.java.net/browse/JDK-7144981">JDK-7144981</a>.
44    *
45    * @param identifier Identifier to be checked
46    * @return {@code true} in case {@code identifier} passed can be used as an identifier; {@code false} otherwise.
47    */
48   private static boolean isValidIdentifier(boolean isMethod, String identifier) {
49
50     assert identifier != null : "Null identifier passed to the isValidIdentifier() method.";
51     assert !identifier.isEmpty() : "Empty identifier passed to the isValidIdentifier() method.";
52
53     if (isMethod && (identifier.equals(CodeConstants.INIT_NAME) || identifier.equals(CodeConstants.CLINIT_NAME))) {
54       return true;
55     }
56
57     if (!Character.isJavaIdentifierStart(identifier.charAt(0))) {
58       return false;
59     }
60
61     char[] chars = identifier.toCharArray();
62
63     for(int i = 1; i < chars.length; i++) {
64       char ch = chars[i];
65
66       if ((!Character.isJavaIdentifierPart(ch)) || Character.isIdentifierIgnorable(ch)) {
67         return false;
68       }
69     }
70
71     return true;
72
73   }
74
75   // TODO: consider possible conflicts with not renamed classes, fields and methods!
76   // We should get all relevant information here.
77   @Override
78   public String getNextClassName(String fullName, String shortName) {
79     if (shortName == null) {
80       return "class_" + (classCounter++);
81     }
82
83     int index = 0;
84     while (index < shortName.length() && Character.isDigit(shortName.charAt(index))) {
85       index++;
86     }
87
88     if (index == 0 || index == shortName.length()) {
89       return "class_" + (classCounter++);
90     }
91     else {
92       String name = shortName.substring(index);
93       if (setNonStandardClassNames.contains(name)) {
94         return "Inner" + name + "_" + (classCounter++);
95       }
96       else {
97         setNonStandardClassNames.add(name);
98         return "Inner" + name;
99       }
100     }
101   }
102
103   @Override
104   public String getNextFieldName(String className, String field, String descriptor) {
105     return "field_" + (fieldCounter++);
106   }
107
108   @Override
109   public String getNextMethodName(String className, String method, String descriptor) {
110     return "method_" + (methodCounter++);
111   }
112
113   // *****************************************************************************
114   // static methods
115   // *****************************************************************************
116
117   public static String getSimpleClassName(String fullName) {
118     return fullName.substring(fullName.lastIndexOf('/') + 1);
119   }
120
121   public static String replaceSimpleClassName(String fullName, String newName) {
122     return fullName.substring(0, fullName.lastIndexOf('/') + 1) + newName;
123   }
124 }