forked from soot-oss/soot
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathEntryPointsTest.java
More file actions
86 lines (80 loc) · 2.78 KB
/
EntryPointsTest.java
File metadata and controls
86 lines (80 loc) · 2.78 KB
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
/*-
* #%L
* Soot - a J*va Optimization Framework
* %%
* Copyright (C) 1997 - 2014 Raja Vallee-Rai and others
* %%
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation, either version 2.1 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Lesser Public License for more details.
*
* You should have received a copy of the GNU General Lesser Public
* License along with this program. If not, see
* <http://www.gnu.org/licenses/lgpl-2.1.html>.
* #L%
*/
package soot;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import org.junit.Test;
import soot.jimple.toolkits.callgraph.CallGraph;
import soot.jimple.toolkits.callgraph.Edge;
import soot.options.Options;
/**
* Tests the EntryPoints class
*
* @author Linghui Luo
*/
public class EntryPointsTest {
@Test
public void testClinitOf() {
Path cp = Paths.get("src", "test", "resources", "Clinit", "bin");
G.reset();
Options.v().set_prepend_classpath(true);
Options.v().set_process_dir(Collections.singletonList(cp.toFile().getAbsolutePath()));
Options.v().set_src_prec(Options.src_prec_class);
Options.v().set_allow_phantom_refs(true);
Options.v().set_ignore_resolving_levels(true);
Options.v().setPhaseOption("cg.spark", "on");
Options.v().setPhaseOption("cg.spark", "string-constants:true");
Options.v().set_whole_program(true);
Scene.v().loadNecessaryClasses();
SootMethod mainMethod = Scene.v().getMainMethod();
Scene.v().setEntryPoints(Collections.singletonList(mainMethod));
PackManager.v().getPack("cg").apply();
CallGraph cg = Scene.v().getCallGraph();
boolean found = false;
for (Edge edge : cg) {
if (edge.getSrc().method().getSignature().equals("<soot.Main: void main(java.lang.String[])>")) {
if (edge.getTgt().method().getSignature().equals("<soot.A: void <clinit>()>")) { // A1 is used in main
found = true;
break;
}
}
}
assertTrue(found);
SootClass a1 = Scene.v().getSootClassUnsafe("soot.A1");
SootClass a = Scene.v().getSootClassUnsafe("soot.A");
assertTrue(a1 != null);
List<String> clinits1 = new ArrayList<>();
EntryPoints.v().clinitsOf(a1).forEach(e -> {
clinits1.add(e.toString());
});
List<String> clinits = new ArrayList<>();
EntryPoints.v().clinitsOf(a).forEach(e -> {
clinits.add(e.toString());
});
assertEquals(clinits1, clinits);
}
}