1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
package lombok.eclipse.agent;
import java.io.File;
import java.io.IOException;
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.lang.instrument.Instrumentation;
import java.lang.instrument.UnmodifiableClassException;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.net.URI;
import java.security.ProtectionDomain;
import java.util.jar.JarFile;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class EclipseParserPatcher {
private EclipseParserPatcher() {}
private static class Patcher implements ClassFileTransformer {
@Override public byte[] transform(ClassLoader loader, String className,
Class<?> classBeingRedefined,
ProtectionDomain protectionDomain, byte[] classfileBuffer)
throws IllegalClassFormatException {
if ( ECLIPSE_PARSER_CLASS_NAME.equals(className) ) {
try {
return runTransform("lombok.eclipse.agent.EclipseParserTransformer", classfileBuffer);
} catch ( Throwable t ) {
System.err.println("Wasn't able to patch eclipse's Parser class:");
t.printStackTrace();
}
}
if ( ECLIPSE_CUD_CLASS_NAME.equals(className) ) {
try {
return runTransform("lombok.eclipse.agent.EclipseCUDTransformer", classfileBuffer);
} catch ( Throwable t ) {
System.err.println("Wasn't able to patch eclipse's CompilationUnitDeclaration class:");
t.printStackTrace();
}
}
return null;
}
}
private static byte[] runTransform(String className, byte[] classfileBuffer) throws Exception {
Class<?> transformerClass = Class.forName(className);
Constructor<?> constructor = transformerClass.getDeclaredConstructor();
constructor.setAccessible(true);
Object instance = constructor.newInstance();
Method m = transformerClass.getDeclaredMethod("transform", byte[].class);
m.setAccessible(true);
return (byte[])m.invoke(instance, classfileBuffer);
}
static final String ECLIPSE_CUD_CLASS_NAME = "org/eclipse/jdt/internal/compiler/ast/CompilationUnitDeclaration";
static final String ECLIPSE_PARSER_CLASS_NAME = "org/eclipse/jdt/internal/compiler/parser/Parser";
public static void agentmain(String agentArgs, Instrumentation instrumentation) throws Exception {
registerPatcher(instrumentation, true);
addLombokToSearchPaths(instrumentation);
}
private static void addLombokToSearchPaths(Instrumentation instrumentation) throws Exception {
String path = findPathOfOurClassloader();
instrumentation.appendToSystemClassLoaderSearch(new JarFile(path + "/lombok.jar"));
instrumentation.appendToBootstrapClassLoaderSearch(new JarFile(path + "/lombok.eclipse.agent.jar"));
}
private static String findPathOfOurClassloader() throws Exception {
ClassLoader loader = EclipseParserPatcher.class.getClassLoader();
if ( loader == null ) loader = ClassLoader.getSystemClassLoader();
URI uri = loader.getResource(EclipseParserPatcher.class.getName().replace('.', '/') + ".class").toURI();
Pattern p = Pattern.compile("^jar:file:([^\\!]+)\\!.*\\.class$");
Matcher m = p.matcher(uri.toString());
if ( !m.matches() ) return ".";
return new File(m.group(1)).getParent();
}
public static void premain(String agentArgs, Instrumentation instrumentation) throws Exception {
registerPatcher(instrumentation, false);
addLombokToSearchPaths(instrumentation);
}
private static void registerPatcher(Instrumentation instrumentation, boolean transformExisting) throws IOException {
instrumentation.addTransformer(new Patcher(), true);
if ( transformExisting ) for ( Class<?> c : instrumentation.getAllLoadedClasses() ) {
if ( c.getName().equals(ECLIPSE_PARSER_CLASS_NAME) ) {
try {
instrumentation.retransformClasses(c);
} catch ( UnmodifiableClassException ex ) {
throw new UnsupportedOperationException(
"The eclipse parser class is already loaded and cannot be modified. " +
"You'll have to restart eclipse in order to use Lombok in eclipse.");
}
}
}
}
}
|
