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
105
106
107
108
109
110
111
112
113
|
// Note that "-processor org/checkerframework/checker.nullness.NullnessChecker"
// is added to the invocation of the compiler!
// TODO: add a @Processor method-annotation to parameterize
/**
* This class has auxiliary methods to compile a class and return its classfile. It is used by
* defaultPersists/Driver and inheritDeclAnnoPersist/Driver.
*/
import com.sun.tools.classfile.ClassFile;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.lang.reflect.Method;
import java.util.StringJoiner;
public class PersistUtil {
public static String testClassOf(Method m) {
TestClass tc = m.getAnnotation(TestClass.class);
if (tc != null) {
return tc.value();
} else {
return "Test";
}
}
public static ClassFile compileAndReturn(String fullFile, String testClass) throws Exception {
File source = writeTestFile(fullFile);
File clazzFile = compileTestFile(source, testClass);
return ClassFile.read(clazzFile);
}
public static File writeTestFile(String fullFile) throws IOException {
File f = new File("Test.java");
PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(f)));
out.println(fullFile);
out.close();
return f;
}
public static File compileTestFile(File f, String testClass) {
int rc =
com.sun.tools.javac.Main.compile(
new String[] {
"-source",
"1.8",
"-g",
"-processor",
"org.checkerframework.checker.nullness.NullnessChecker",
f.getPath()
});
if (rc != 0) {
throw new Error("compilation failed. rc=" + rc);
}
String path;
if (f.getParent() != null) {
path = f.getParent();
} else {
path = "";
}
return new File(path + testClass + ".class");
}
public static String wrap(String compact) {
StringJoiner sj = new StringJoiner(System.lineSeparator());
// Automatically import java.util
sj.add("");
sj.add("import java.util.*;");
sj.add("import java.lang.annotation.*;");
// And the Nullness qualifiers
sj.add("import org.checkerframework.framework.qual.DefaultQualifier;");
sj.add("import org.checkerframework.checker.nullness.qual.*;");
sj.add("import org.checkerframework.dataflow.qual.*;");
sj.add("");
boolean isSnippet =
!(compact.startsWith("class") || compact.contains(" class"))
&& !compact.contains("interface")
&& !compact.contains("enum");
if (isSnippet) {
sj.add("class Test {");
}
sj.add(compact);
if (isSnippet) {
sj.add("}");
sj.add("");
}
return sj.toString();
}
}
/**
* The name of the class that should be analyzed. Should only need to be provided when analyzing
* inner classes.
*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
@interface TestClass {
String value() default "Test";
}
|
