BAEL-7881: Introduction to SootUp

libraries-5/pom.xml

@@ -132,6 +132,26 @@
             <artifactId>lanterna</artifactId>
             <version>${lanterna.version}</version>
         </dependency>
+        <dependency>
+            <groupId>org.soot-oss</groupId>
+            <artifactId>sootup.core</artifactId>
+            <version>${sootup.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.soot-oss</groupId>
+            <artifactId>sootup.java.core</artifactId>
+            <version>${sootup.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.soot-oss</groupId>
+            <artifactId>sootup.java.sourcecode</artifactId>
+            <version>${sootup.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.soot-oss</groupId>
+            <artifactId>sootup.java.bytecode</artifactId>
+            <version>${sootup.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-test</artifactId>
@@ -149,6 +169,7 @@
         <armeria.version>1.29.2</armeria.version>
         <yauaa.version>7.28.1</yauaa.version>
         <yavi.version>0.14.1</yavi.version>
+        <sootup.version>1.3.0</sootup.version>
     </properties>
 
 </project>

libraries-5/src/test/java/com/baeldung/sootup/AnalyzeUnitTest.java

@@ -0,0 +1,57 @@
+package com.baeldung.sootup;
+
+import org.junit.jupiter.api.Test;
+import sootup.core.inputlocation.AnalysisInputLocation;
+import sootup.java.bytecode.inputlocation.JavaClassPathAnalysisInputLocation;
+import sootup.java.bytecode.inputlocation.JrtFileSystemAnalysisInputLocation;
+import sootup.java.bytecode.inputlocation.OTFCompileAnalysisInputLocation;
+import sootup.java.core.views.JavaView;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+public class AnalyzeUnitTest {
+    @Test
+    void whenAnalyzingTheJVM_thenWeCanListClasses() {
+        AnalysisInputLocation inputLocation = new JrtFileSystemAnalysisInputLocation();
+
+        JavaView view = new JavaView(inputLocation);
+
+        assertTrue(view.getClasses().size() > 0);
+    }
+
+    @Test
+    void whenAnalyzingThisTestClass_thenWeCanListClasses() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/AnalyzeUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        assertEquals(1, view.getClasses().size());
+    }
+
+    @Test
+    void whenAnalyzingAString_thenWeCanListClasses() throws IOException {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/AnalyzeUnitTest.java");
+        String javaContents = Files.readString(javaFile);
+
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation("AnalyzeUnitTest.java", javaContents);
+
+        JavaView view = new JavaView(inputLocation);
+
+        assertEquals(1, view.getClasses().size());
+    }
+
+    @Test
+    void whenAnalyzingCompiledByteCode_thenWeCanListClasses() {
+        AnalysisInputLocation inputLocation = new JavaClassPathAnalysisInputLocation("target/classes");
+
+        JavaView view = new JavaView(inputLocation);
+
+        assertTrue(view.getClasses().size() > 0);
+    }
+}

libraries-5/src/test/java/com/baeldung/sootup/ClassUnitTest.java

@@ -0,0 +1,72 @@
+package com.baeldung.sootup;
+
+import org.junit.jupiter.api.Test;
+import sootup.core.IdentifierFactory;
+import sootup.core.inputlocation.AnalysisInputLocation;
+import sootup.core.model.SootClass;
+import sootup.core.types.ClassType;
+import sootup.java.bytecode.inputlocation.OTFCompileAnalysisInputLocation;
+import sootup.java.core.JavaSootClass;
+import sootup.java.core.views.JavaView;
+
+import java.nio.file.Path;
+import java.util.Optional;
+import java.util.Set;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+public class ClassUnitTest {
+    @Test
+    void whenAnalyzingThisTestClass_thenWeCanGetASingleClass() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/ClassUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.ClassUnitTest");
+
+        Optional<JavaSootClass> sootClass = view.getClass(javaClass);
+        assertTrue(sootClass.isPresent());
+
+        JavaSootClass classUnitTest = sootClass.get();
+        assertTrue(classUnitTest.isPublic());
+        assertTrue(classUnitTest.isConcrete());
+        assertFalse(classUnitTest.isFinal());
+        assertFalse(classUnitTest.isEnum());
+    }
+
+    @Test
+    void whenAnalyzingThisTestClass_thenWeCanGetTheSuperclass() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/ClassUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.ClassUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Optional<? extends ClassType> superclass = sootClass.getSuperclass();
+
+        assertTrue(superclass.isPresent());
+        assertEquals("java.lang.Object", superclass.get().getFullyQualifiedName());
+    }
+
+    @Test
+    void whenAnalyzingThisTestClass_thenWeCanGetTheInterfaces() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/ClassUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.ClassUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Set<? extends ClassType> interfaces = sootClass.getInterfaces();
+
+        assertTrue(interfaces.isEmpty());
+    }
+
+}

libraries-5/src/test/java/com/baeldung/sootup/FieldUnitTest.java

@@ -0,0 +1,54 @@
+package com.baeldung.sootup;
+
+import org.junit.jupiter.api.Test;
+import sootup.core.IdentifierFactory;
+import sootup.core.inputlocation.AnalysisInputLocation;
+import sootup.core.model.SootClass;
+import sootup.core.model.SootField;
+import sootup.core.types.ClassType;
+import sootup.java.bytecode.inputlocation.OTFCompileAnalysisInputLocation;
+import sootup.java.core.views.JavaView;
+
+import java.nio.file.Path;
+import java.util.Optional;
+import java.util.Set;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+public class FieldUnitTest {
+    private String aField;
+
+    @Test
+    void whenAnalyzingThisClass_thenWeCanAccessFields() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/FieldUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.FieldUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Set<? extends SootField> fields = sootClass.getFields();
+        assertEquals(1, fields.size());
+    }
+
+    @Test
+    void whenAnalyzingThisClass_thenWeCanAccessASingleField() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/FieldUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.FieldUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Optional<? extends SootField> field = sootClass.getField("aField");
+        assertTrue(field.isPresent());
+
+        SootField sootField = field.get();
+        assertTrue(sootField.isPrivate());
+        assertFalse(sootField.isStatic());
+    }
+}

libraries-5/src/test/java/com/baeldung/sootup/MethodBodyUnitTest.java

@@ -0,0 +1,84 @@
+package com.baeldung.sootup;
+
+import org.junit.jupiter.api.Test;
+import sootup.core.IdentifierFactory;
+import sootup.core.graph.StmtGraph;
+import sootup.core.inputlocation.AnalysisInputLocation;
+import sootup.core.jimple.common.stmt.Stmt;
+import sootup.core.model.Body;
+import sootup.core.model.SootClass;
+import sootup.core.model.SootMethod;
+import sootup.core.types.ClassType;
+import sootup.java.bytecode.inputlocation.OTFCompileAnalysisInputLocation;
+import sootup.java.core.views.JavaView;
+
+import java.nio.file.Path;
+import java.util.List;
+import java.util.Optional;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+public class MethodBodyUnitTest {
+    @Test
+    void whenAnalyzingAMethod_thenWeCanAccessTheLocals() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/MethodBodyUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.MethodBodyUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Optional<? extends SootMethod> method = sootClass.getMethod("someMethod",
+                List.of(
+                        identifierFactory.getClassType("java.lang.String")
+                ));
+        assertTrue(method.isPresent());
+
+        SootMethod sootMethod = method.get();
+
+        Body methodBody = sootMethod.getBody();
+        assertTrue(methodBody.getLocalCount() > 0);
+        var thisLocal = methodBody.getLocals()
+                .stream()
+                .filter(local -> local.getName().equals("this"))
+                .findFirst();
+        assertTrue(thisLocal.isPresent());
+        assertEquals(javaClass, thisLocal.get().getType());
+    }
+
+    @Test
+    void whenAnalyzingAMethod_thenWeCanAccessTheCallGraph() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/MethodBodyUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.MethodBodyUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Optional<? extends SootMethod> method = sootClass.getMethod("someMethod",
+                List.of(
+                        identifierFactory.getClassType("java.lang.String")
+                ));
+        assertTrue(method.isPresent());
+
+        SootMethod sootMethod = method.get();
+
+        Body methodBody = sootMethod.getBody();
+        StmtGraph<?> stmtGraph = methodBody.getStmtGraph();
+        List<Stmt> stmts = stmtGraph.getStmts();
+        for (Stmt stmt : stmts) {
+            System.out.println(stmt);
+        }
+    }
+
+
+    private void someMethod(String name) {
+        var capitald = name.toUpperCase();
+        System.out.println("Hello, " + capitald);
+    }
+}

libraries-5/src/test/java/com/baeldung/sootup/MethodUnitTest.java

@@ -0,0 +1,75 @@
+package com.baeldung.sootup;
+
+import org.junit.jupiter.api.Test;
+import sootup.core.IdentifierFactory;
+import sootup.core.inputlocation.AnalysisInputLocation;
+import sootup.core.model.SootClass;
+import sootup.core.model.SootMethod;
+import sootup.core.types.ClassType;
+import sootup.java.bytecode.inputlocation.OTFCompileAnalysisInputLocation;
+import sootup.java.core.views.JavaView;
+
+import java.nio.file.Path;
+import java.util.List;
+import java.util.Optional;
+import java.util.Set;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+public class MethodUnitTest {
+
+    @Test
+    void whenAnalyzingThisClass_thenWeCanAccessMethods() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/MethodUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.MethodUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Set<? extends SootMethod> methods = sootClass.getMethods();
+        assertTrue(methods.size() > 0);
+    }
+
+    @Test
+    void whenAnalyzingThisClass_thenWeCanAccessASingleMethod() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/MethodUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.MethodUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Optional<? extends SootMethod> method = sootClass.getMethod("someMethod",
+            List.of(
+                identifierFactory.getClassType("java.lang.String")
+            ));
+        assertTrue(method.isPresent());
+
+        SootMethod sootMethod = method.get();
+        assertTrue(sootMethod.isPrivate());
+        assertTrue(sootMethod.isConcrete());
+    }
+
+    @Test
+    void whenAnalyzingThisClass_thenWeCanListMethodsByName() {
+        Path javaFile = Path.of("src/test/java/com/baeldung/sootup/MethodUnitTest.java");
+        AnalysisInputLocation inputLocation = new OTFCompileAnalysisInputLocation(javaFile);
+
+        JavaView view = new JavaView(inputLocation);
+
+        IdentifierFactory identifierFactory = view.getIdentifierFactory();
+        ClassType javaClass = identifierFactory.getClassType("com.baeldung.sootup.MethodUnitTest");
+
+        SootClass sootClass = view.getClassOrThrow(javaClass);
+        Set<? extends SootMethod> method = sootClass.getMethodsByName("someMethod");
+        assertEquals(1, method.size());
+    }
+
+    private void someMethod(String name) {}
+}