Architectural: Add Wala Framework Setup

pom.xml

@@ -159,6 +159,11 @@
             <artifactId>byte-buddy-agent</artifactId>
             <version>1.14.19</version>
         </dependency>
+        <dependency>
+            <groupId>com.ibm.wala</groupId>
+            <artifactId>com.ibm.wala.core</artifactId>
+            <version>1.6.7</version>
+        </dependency>
     </dependencies>
     <build>
         <plugins>
@@ -591,7 +596,9 @@
                         <version>3.3.0</version>
                         <configuration>
                             <forkCount>1</forkCount>
-                            <argLine>-javaagent:/home/sarps/IdeaProjects/Ares2/target/ares-2.0.0-SNAPSHOT-agent.jar -Xbootclasspath/a:${user.home}/.m2/repository/org/aspectj/aspectjrt/${aspectj.version}/aspectjrt-${aspectj.version}.jar</argLine>
+                            <argLine>-javaagent:/home/sarps/IdeaProjects/Ares2/target/ares-2.0.0-SNAPSHOT-agent.jar
+                                -Xbootclasspath/a:${user.home}/.m2/repository/org/aspectj/aspectjrt/${aspectj.version}/aspectjrt-${aspectj.version}.jar
+                            </argLine>
                         </configuration>
                     </plugin>
                 </plugins>

src/main/java/de/tum/cit/ase/ares/api/architecture/java/wala/JavaWalaSecurityTestCase.java

@@ -0,0 +1,92 @@
+package de.tum.cit.ase.ares.api.architecture.java.wala;
+
+import com.ibm.wala.classLoader.IClass;
+import com.ibm.wala.classLoader.IMethod;
+import com.ibm.wala.classLoader.Language;
+import com.ibm.wala.core.util.config.AnalysisScopeReader;
+import com.ibm.wala.ipa.callgraph.*;
+import com.ibm.wala.ipa.callgraph.impl.DefaultEntrypoint;
+import com.ibm.wala.ipa.callgraph.impl.Util;
+import com.ibm.wala.ipa.callgraph.propagation.InstanceKey;
+import com.ibm.wala.ipa.cha.ClassHierarchy;
+import com.ibm.wala.ipa.cha.ClassHierarchyException;
+import com.ibm.wala.ipa.cha.ClassHierarchyFactory;
+import com.ibm.wala.types.ClassLoaderReference;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.BufferedWriter;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+public class JavaWalaSecurityTestCase {
+
+    private static final Logger log = LoggerFactory.getLogger(JavaWalaSecurityTestCase.class);
+
+    public static CallGraph buildCallGraph() {
+        try {
+            // Create an AnalysisScope for the Java 21 JRT modules
+            AnalysisScope.createJavaAnalysisScope();
+            // get entire classpath of the project
+            String classpath = System.getProperty("java.class.path");
+            AnalysisScope scope = AnalysisScopeReader.instance.makeJavaBinaryAnalysisScope(
+                    classpath,
+                    null);
+
+            // Build the class hierarchy
+            ClassHierarchy chaWholeApplication = ClassHierarchyFactory.make(scope);
+
+            // Create a list to store entry points
+            List<Entrypoint> customEntryPoints = new ArrayList<>();
+
+            // Create AnalysisOptions for call graph
+            AnalysisOptions options = new AnalysisOptions(scope, customEntryPoints);
+
+
+            // Create call graph builder (n-CFA, context-sensitive, etc.)
+            CallGraphBuilder<InstanceKey> builder = Util.makeZeroCFABuilder(Language.JAVA, options, new AnalysisCacheImpl(), chaWholeApplication);
+
+            long startTime = System.currentTimeMillis();
+            // Generate the call graph
+            CallGraph callGraph = builder.makeCallGraph(options, null);
+            long endTime = System.currentTimeMillis();
+
+            log.info("Building call graph took {} ms", endTime - startTime);
+
+            List<CGNode> violatingMethods = ReachabilityChecker.isReachable(callGraph, callGraph.getEntrypointNodes().iterator(), node -> node.getMethod().getSignature().startsWith("sun.nio.fs.UnixFileSystemProvider") || node.getMethod().getSignature().startsWith("java.security.AccessController") || node.getMethod().getSignature().startsWith("java.lang.CLassLoader.findResource"));
+            log.info("Violating methods: {}", violatingMethods);
+            return callGraph;
+        } catch (IOException  | CallGraphBuilderCancelException | ClassHierarchyException e) {
+            throw new SecurityException("Error building call graph", e); //$NON-NLS-1$
+        }
+    }
+
+    // Write the CallGraph to a DOT file
+    public static void writeCallGraphToDot(CallGraph callGraph, String filePath) throws IOException {
+        try (BufferedWriter writer = new BufferedWriter(new FileWriter(filePath))) {
+            writer.write("digraph G {\n");
+
+            // Iterate through the nodes in the call graph
+            for (CGNode node : callGraph) {
+                // Write the node representation
+                writer.write("    \"" + node.getMethod().getSignature() + "\";\n");
+
+                // Iterate through the edges (call sites) from the current node
+                for (Iterator<CGNode> it = callGraph.getSuccNodes(node); it.hasNext(); ) {
+                    CGNode successor = it.next();
+                    // Write the edge representation
+                    writer.write("    \"" + node.getMethod().getSignature() + "\" -> \"" + successor.getMethod().getSignature() + "\";\n");
+                }
+            }
+            writer.write("}\n");
+        }
+    }
+
+    public static void main(String[] args) throws IOException {
+        CallGraph callGraph = buildCallGraph();
+        writeCallGraphToDot(callGraph, "callgraph.dot");
+    }
+}

src/main/java/de/tum/cit/ase/ares/api/architecture/java/wala/ReachabilityChecker.java

@@ -0,0 +1,56 @@
+package de.tum.cit.ase.ares.api.architecture.java.wala;
+
+import com.ibm.wala.classLoader.IClass;
+import com.ibm.wala.classLoader.IMethod;
+import com.ibm.wala.ipa.callgraph.CGNode;
+import com.ibm.wala.ipa.callgraph.CallGraph;
+import com.ibm.wala.ipa.callgraph.impl.DefaultEntrypoint;
+import com.ibm.wala.ipa.cha.ClassHierarchy;
+import com.ibm.wala.types.ClassLoaderReference;
+import com.ibm.wala.util.graph.traverse.DFSPathFinder;
+
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+import java.util.function.Predicate;
+
+/**
+ * A utility class to check reachability in a call graph.
+ */
+class ReachabilityChecker {
+
+    private ReachabilityChecker() {
+        // Utility class
+    }
+
+    /**
+     * Checks if a target node is reachable from a set of start nodes in a call graph.
+     * @param callGraph The call graph to search in.
+     * @param startNodes The start nodes to search from.
+     * @param targetNodeFilter A filter to determine if a node is a target node.
+     * @return A list of nodes that are reachable from the start nodes and match the target node filter.
+     */
+    public static List<CGNode> isReachable(CallGraph callGraph, Iterator<CGNode> startNodes, Predicate<CGNode> targetNodeFilter) {
+        // TODO: Implement our own pathfinder instead of the default
+        return new DFSPathFinder<>(callGraph, startNodes, targetNodeFilter).find();
+    }
+
+    // TODO: This is currently not working as expected!!!
+    public static List<DefaultEntrypoint> getEntryPointsFromStudentSubmission(CallGraph callGraph, Iterator<CGNode> startNodes, ClassHierarchy cha, ClassHierarchy chaClassPath) {
+        // Iterate through all classes in the application classloader
+        List<DefaultEntrypoint> customEntryPoints = new ArrayList<>();
+            for (IClass klass : cha) {
+                if (klass.getClassLoader().getReference().equals(ClassLoaderReference.Application)) {
+                    // Iterate through all declared methods in each class
+                    for (IMethod method : klass.getDeclaredMethods()) {
+                        // Exclude the 'main' methods from being entry points
+                        if (!method.getName().toString().equals("main")) {
+                            customEntryPoints.add(new DefaultEntrypoint(method.getReference(), chaClassPath));
+                        }
+                    }
+                }
+            }
+        return customEntryPoints;
+    }
+}
+