|
| 1 | +/** |
| 2 | + * ***************************************************************************** |
| 3 | + * Copyright (c) 2025 Fraunhofer IEM, Paderborn, Germany. This program and the |
| 4 | + * accompanying materials are made available under the terms of the Eclipse |
| 5 | + * Public License 2.0 which is available at http://www.eclipse.org/legal/epl-2.0. |
| 6 | + * |
| 7 | + * <p>SPDX-License-Identifier: EPL-2.0 |
| 8 | + * |
| 9 | + * <p>Contributors: Johannes Spaeth - initial API and implementation |
| 10 | + * ***************************************************************************** |
| 11 | + */ |
1 | 12 | package boomerang.scope.opal
|
2 | 13 |
|
3 |
| -import boomerang.scope.{CallGraph, InvokeExpr} |
| 14 | +import boomerang.scope.CallGraph |
4 | 15 | import boomerang.scope.CallGraph.Edge
|
5 |
| -import boomerang.scope.opal.tac.{OpalFunctionInvokeExpr, OpalMethod, OpalMethodInvokeExpr, OpalPhantomMethod, OpalStatement} |
| 16 | +import boomerang.scope.InvokeExpr |
| 17 | +import boomerang.scope.opal.tac.OpalFunctionInvokeExpr |
| 18 | +import boomerang.scope.opal.tac.OpalMethod |
| 19 | +import boomerang.scope.opal.tac.OpalMethodInvokeExpr |
| 20 | +import boomerang.scope.opal.tac.OpalPhantomMethod |
| 21 | +import boomerang.scope.opal.tac.OpalStatement |
6 | 22 | import boomerang.scope.opal.transformation.TacBodyBuilder
|
| 23 | +import org.opalj.br.DefinedMethod |
| 24 | +import org.opalj.br.Method |
| 25 | +import org.opalj.br.MultipleDefinedMethods |
| 26 | +import org.opalj.br.VirtualDeclaredMethod |
7 | 27 | import org.opalj.br.analyses.Project
|
8 |
| -import org.opalj.br.{DefinedMethod, Method, MultipleDefinedMethods, VirtualDeclaredMethod} |
9 |
| -import org.opalj.tac.{NonVirtualFunctionCall, StaticFunctionCall, VirtualFunctionCall} |
| 28 | +import org.opalj.tac.NonVirtualFunctionCall |
| 29 | +import org.opalj.tac.StaticFunctionCall |
| 30 | +import org.opalj.tac.VirtualFunctionCall |
10 | 31 |
|
11 |
| -class OpalCallGraph(project: Project[_], callGraph: org.opalj.tac.cg.CallGraph, entryPoints: Set[Method]) extends CallGraph { |
| 32 | +class OpalCallGraph( |
| 33 | + project: Project[_], |
| 34 | + callGraph: org.opalj.tac.cg.CallGraph, |
| 35 | + entryPoints: Set[Method] |
| 36 | +) extends CallGraph { |
12 | 37 |
|
13 |
| - // TODO Deal with <clinit> |
14 |
| - callGraph.reachableMethods().foreach(method => { |
15 |
| - method.method match { |
16 |
| - case definedMethod: DefinedMethod => |
17 |
| - if (definedMethod.definedMethod.body.isDefined) { |
18 |
| - addEdgesFromMethod(definedMethod) |
19 |
| - } |
20 |
| - // TODO Should this case be considered? |
21 |
| - // case definedMethods: MultipleDefinedMethods => |
22 |
| - // definedMethods.foreachDefinedMethod(m => addEdgesFromMethod(m)) |
23 |
| - case _ => |
24 |
| - } |
25 |
| - }) |
| 38 | + // TODO Deal with <clinit> |
| 39 | + callGraph |
| 40 | + .reachableMethods() |
| 41 | + .foreach(method => { |
| 42 | + method.method match { |
| 43 | + case definedMethod: DefinedMethod => |
| 44 | + if (definedMethod.definedMethod.body.isDefined) { |
| 45 | + addEdgesFromMethod(definedMethod) |
| 46 | + } |
| 47 | + // TODO Should this case be considered? |
| 48 | + // case definedMethods: MultipleDefinedMethods => |
| 49 | + // definedMethods.foreachDefinedMethod(m => addEdgesFromMethod(m)) |
| 50 | + case _ => |
| 51 | + } |
| 52 | + }) |
26 | 53 |
|
27 |
| - private def addEdgesFromMethod(method: DefinedMethod): Unit = { |
28 |
| - val tacCode = TacBodyBuilder(project, method.definedMethod) |
| 54 | + private def addEdgesFromMethod(method: DefinedMethod): Unit = { |
| 55 | + val tacCode = TacBodyBuilder(project, method.definedMethod) |
29 | 56 |
|
30 |
| - tacCode.statements.foreach(stmt => { |
31 |
| - val srcStatement = new OpalStatement(stmt, OpalMethod(method.definedMethod, tacCode)) |
| 57 | + tacCode.statements.foreach(stmt => { |
| 58 | + val srcStatement = |
| 59 | + new OpalStatement(stmt, OpalMethod(method.definedMethod, tacCode)) |
32 | 60 |
|
33 |
| - if (srcStatement.containsInvokeExpr()) { |
34 |
| - // Due to inlining variables, the PC's of statements and invoke expressions may differ |
35 |
| - val invokeExprPc = getPcForInvokeExpr(srcStatement.getInvokeExpr) |
36 |
| - val callees = callGraph.directCalleesOf(method, invokeExprPc) |
| 61 | + if (srcStatement.containsInvokeExpr()) { |
| 62 | + // Due to inlining variables, the PC's of statements and invoke expressions may differ |
| 63 | + val invokeExprPc = getPcForInvokeExpr(srcStatement.getInvokeExpr) |
| 64 | + val callees = callGraph.directCalleesOf(method, invokeExprPc) |
37 | 65 |
|
38 |
| - callees.foreach(callee => { |
39 |
| - callee.method match { |
40 |
| - case definedMethod: DefinedMethod => |
41 |
| - val method = definedMethod.definedMethod |
| 66 | + callees.foreach(callee => { |
| 67 | + callee.method match { |
| 68 | + case definedMethod: DefinedMethod => |
| 69 | + val method = definedMethod.definedMethod |
42 | 70 |
|
43 |
| - if (method.body.isDefined) { |
44 |
| - val targetMethod = OpalMethod(method) |
| 71 | + if (method.body.isDefined) { |
| 72 | + val targetMethod = OpalMethod(method) |
45 | 73 |
|
46 |
| - addEdge(new Edge(srcStatement, targetMethod)) |
47 |
| - } else { |
48 |
| - val targetMethod = OpalPhantomMethod(definedMethod.declaringClassType, definedMethod.name, definedMethod.descriptor, method.isStatic) |
| 74 | + addEdge(new Edge(srcStatement, targetMethod)) |
| 75 | + } else { |
| 76 | + val targetMethod = OpalPhantomMethod( |
| 77 | + definedMethod.declaringClassType, |
| 78 | + definedMethod.name, |
| 79 | + definedMethod.descriptor, |
| 80 | + method.isStatic |
| 81 | + ) |
49 | 82 |
|
50 |
| - addEdge(new Edge(srcStatement, targetMethod)) |
51 |
| - } |
52 |
| - case virtualMethod: VirtualDeclaredMethod => |
53 |
| - val targetMethod = OpalPhantomMethod(virtualMethod.declaringClassType, virtualMethod.name, virtualMethod.descriptor, srcStatement.getInvokeExpr.isStaticInvokeExpr) |
| 83 | + addEdge(new Edge(srcStatement, targetMethod)) |
| 84 | + } |
| 85 | + case virtualMethod: VirtualDeclaredMethod => |
| 86 | + val targetMethod = OpalPhantomMethod( |
| 87 | + virtualMethod.declaringClassType, |
| 88 | + virtualMethod.name, |
| 89 | + virtualMethod.descriptor, |
| 90 | + srcStatement.getInvokeExpr.isStaticInvokeExpr |
| 91 | + ) |
54 | 92 |
|
55 |
| - addEdge(new Edge(srcStatement, targetMethod)) |
56 |
| - case definedMethods: MultipleDefinedMethods => |
57 |
| - definedMethods.foreachDefinedMethod(method => { |
58 |
| - val targetMethod = OpalMethod(method) |
| 93 | + addEdge(new Edge(srcStatement, targetMethod)) |
| 94 | + case definedMethods: MultipleDefinedMethods => |
| 95 | + definedMethods.foreachDefinedMethod(method => { |
| 96 | + val targetMethod = OpalMethod(method) |
59 | 97 |
|
60 |
| - addEdge(new Edge(srcStatement, targetMethod)) |
61 |
| - }) |
62 |
| - } |
| 98 | + addEdge(new Edge(srcStatement, targetMethod)) |
| 99 | + }) |
| 100 | + } |
| 101 | + }) |
| 102 | + } |
63 | 103 | })
|
64 |
| - } |
65 |
| - }) |
66 |
| - } |
67 |
| - |
68 |
| - private def getPcForInvokeExpr(invokeExpr: InvokeExpr): Int = { |
69 |
| - invokeExpr match { |
70 |
| - case methodInvokeExpr: OpalMethodInvokeExpr => methodInvokeExpr.delegate.pc |
71 |
| - case functionInvokeExpr: OpalFunctionInvokeExpr => |
72 |
| - functionInvokeExpr.delegate match { |
73 |
| - case call: NonVirtualFunctionCall[_] => call.pc |
74 |
| - case call: VirtualFunctionCall[_] => call.pc |
75 |
| - case call: StaticFunctionCall[_] => call.pc |
76 |
| - case _ => throw new RuntimeException("Unknown function call: " + functionInvokeExpr) |
77 |
| - } |
78 |
| - case _ => throw new RuntimeException("Unknown invoke expression: " + invokeExpr) |
79 | 104 | }
|
80 |
| - } |
81 | 105 |
|
82 |
| - // Explicitly add static initializers (<clinit>) as they are called only implicitly |
83 |
| - callGraph.reachableMethods().foreach(method => { |
84 |
| - method.method match { |
85 |
| - case definedMethod: DefinedMethod if definedMethod.definedMethod.isStaticInitializer => |
86 |
| - if (definedMethod.definedMethod.body.isDefined) { |
87 |
| - addEntryPoint(OpalMethod(definedMethod.definedMethod)) |
| 106 | + private def getPcForInvokeExpr(invokeExpr: InvokeExpr): Int = { |
| 107 | + invokeExpr match { |
| 108 | + case methodInvokeExpr: OpalMethodInvokeExpr => |
| 109 | + methodInvokeExpr.delegate.pc |
| 110 | + case functionInvokeExpr: OpalFunctionInvokeExpr => |
| 111 | + functionInvokeExpr.delegate match { |
| 112 | + case call: NonVirtualFunctionCall[_] => call.pc |
| 113 | + case call: VirtualFunctionCall[_] => call.pc |
| 114 | + case call: StaticFunctionCall[_] => call.pc |
| 115 | + case _ => |
| 116 | + throw new RuntimeException( |
| 117 | + "Unknown function call: " + functionInvokeExpr |
| 118 | + ) |
| 119 | + } |
| 120 | + case _ => |
| 121 | + throw new RuntimeException("Unknown invoke expression: " + invokeExpr) |
88 | 122 | }
|
89 |
| - case _ => |
90 | 123 | }
|
91 |
| - }) |
92 | 124 |
|
93 |
| - entryPoints.foreach(entryPoint => { |
94 |
| - if (entryPoint.body.isDefined) { |
95 |
| - addEntryPoint(OpalMethod(entryPoint)) |
96 |
| - } |
97 |
| - }) |
| 125 | + // Explicitly add static initializers (<clinit>) as they are called only implicitly |
| 126 | + callGraph |
| 127 | + .reachableMethods() |
| 128 | + .foreach(method => { |
| 129 | + method.method match { |
| 130 | + case definedMethod: DefinedMethod |
| 131 | + if definedMethod.definedMethod.isStaticInitializer => |
| 132 | + if (definedMethod.definedMethod.body.isDefined) { |
| 133 | + addEntryPoint(OpalMethod(definedMethod.definedMethod)) |
| 134 | + } |
| 135 | + case _ => |
| 136 | + } |
| 137 | + }) |
| 138 | + |
| 139 | + entryPoints.foreach(entryPoint => { |
| 140 | + if (entryPoint.body.isDefined) { |
| 141 | + addEntryPoint(OpalMethod(entryPoint)) |
| 142 | + } |
| 143 | + }) |
98 | 144 | }
|
0 commit comments