arkanalyzer 1.0.86 → 1.0.87

package/lib/callgraph/algorithm/AbstractAnalysis.js

@@ -38,13 +38,16 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AbstractAnalysis = void 0;
+const Expr_1 = require("../../core/base/Expr");
+const Stmt_1 = require("../../core/base/Stmt");
 const Type_1 = require("../../core/base/Type");
 const logger_1 = __importStar(require("../../utils/logger"));
+const IntWorkList_1 = require("../../utils/IntWorkList");
 const PtsDS_1 = require("../pointerAnalysis/PtsDS");
 const logger = logger_1.default.getLogger(logger_1.LOG_MODULE_TYPE.ARKANALYZER, 'CG');
 class AbstractAnalysis {
     constructor(s, cg) {
-        this.workList = [];
+        this.workList = new IntWorkList_1.IntWorkList();
         this.classHierarchyCache = new Map();
         this.scene = s;
         this.cg = cg;
@@ -84,8 +87,8 @@ class AbstractAnalysis {
     }
     start(displayGeneratedMethod) {
         this.init();
-        while (this.workList.length !== 0) {
-            const method = this.workList.shift();
+        while (!this.workList.isEmpty()) {
+            const method = this.workList.pop();
             const cgNode = this.cg.getNode(method);
             if (this.processedMethod.contains(method) || cgNode.isSdkMethod()) {
                 continue;
@@ -94,9 +97,7 @@ class AbstractAnalysis {
             this.preProcessMethod(method).forEach((cs) => {
                 this.workList.push(cs.calleeFuncID);
             });
-            this.processMethod(method).forEach((cs) => {
-                this.processCallSite(method, cs, displayGeneratedMethod);
-            });
+            this.processMethod(method, displayGeneratedMethod, false);
         }
     }
     projectStart(displayGeneratedMethod) {
@@ -107,9 +108,7 @@ class AbstractAnalysis {
                 continue;
             }
             this.preProcessMethod(cgNode.getID());
-            this.processMethod(cgNode.getID()).forEach((cs) => {
-                this.processCallSite(cgNode.getID(), cs, displayGeneratedMethod, true);
-            });
+            this.processMethod(cgNode.getID(), displayGeneratedMethod, true);
         }
         this.cgBuilder.setEntries();
     }
@@ -136,27 +135,28 @@ class AbstractAnalysis {
             this.workList.push(entryFunc);
         });
     }
-    processMethod(methodID) {
+    processMethod(methodID, displayGeneratedMethod, isProject = false) {
         let cgNode = this.cg.getNode(methodID);
         let arkMethod = this.scene.getMethod(cgNode.getMethod(), true);
-        let calleeMethods = [];
         if (!arkMethod) {
             throw new Error('can not find method');
         }
         const cfg = arkMethod.getCfg();
         if (!cfg) {
-            return [];
+            return;
         }
-        cfg.getStmts().forEach(stmt => {
-            if (stmt.containsInvokeExpr()) {
+        cfg.getBlocks().forEach(block => {
+            block.getStmts().forEach(stmt => {
+                if (!(stmt instanceof Stmt_1.ArkInvokeStmt || (stmt instanceof Stmt_1.ArkAssignStmt && stmt.getRightOp() instanceof Expr_1.AbstractInvokeExpr))) {
+                    return;
+                }
                 this.resolveCall(cgNode.getID(), stmt).forEach(callSite => {
-                    calleeMethods.push(callSite);
                     this.cg.addStmtToCallSiteMap(stmt, callSite);
                     this.cg.addMethodToCallSiteMap(callSite.calleeFuncID, callSite);
+                    this.processCallSite(methodID, callSite, displayGeneratedMethod, isProject);
                 });
-            }
+            });
         });
-        return calleeMethods;
     }
     getParamAnonymousMethod(invokeExpr) {
         let paramMethod = [];

package/lib/callgraph/algorithm/RapidTypeAnalysis.d.ts

@@ -4,11 +4,16 @@ import { ClassSignature } from '../../core/model/ArkSignature';
 import { NodeID } from '../../core/graph/BaseExplicitGraph';
 import { CallGraph, CallSite, FuncID } from '../model/CallGraph';
 import { AbstractAnalysis } from './AbstractAnalysis';
+import { CallGraphBuilder } from '../model/builder/CallGraphBuilder';
 export declare class RapidTypeAnalysis extends AbstractAnalysis {
     private instancedClasses;
     private ignoredCalls;
-    constructor(scene: Scene, cg: CallGraph);
+    private enableThisPrune;
+    constructor(scene: Scene, cg: CallGraph, cb: CallGraphBuilder, enableThisPrune?: boolean);
+    protected init(): void;
     resolveCall(callerMethod: NodeID, invokeStmt: Stmt): CallSite[];
+    private tryResolveAggressiveThisPruneCall;
+    private resolveConstructorClassFromNew;
     protected preProcessMethod(funcID: FuncID): CallSite[];
     private collectInstancedClassesInMethod;
     addIgnoredCalls(arkClass: ClassSignature, callerID: FuncID, calleeID: FuncID, invokeStmt: Stmt): void;

package/lib/callgraph/algorithm/RapidTypeAnalysis.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RapidTypeAnalysis.d.ts","sourceRoot":"","sources":["../../../src/callgraph/algorithm/RapidTypeAnalysis.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAE5C,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAMtD,qBAAa,iBAAkB,SAAQ,gBAAgB;IAEnD,OAAO,CAAC,gBAAgB,CAAkC;IAE1D,OAAO,CAAC,YAAY,CAA2F;gBAEnG,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS;IAIhC,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,GAAG,QAAQ,EAAE;IA4DtE,SAAS,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ,EAAE;IAqBtD,OAAO,CAAC,+BAA+B;IA4BhC,eAAe,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI;CAK/G"}
+{"version":3,"file":"RapidTypeAnalysis.d.ts","sourceRoot":"","sources":["../../../src/callgraph/algorithm/RapidTypeAnalysis.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAiB,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAGtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AAOrE,qBAAa,iBAAkB,SAAQ,gBAAgB;IAEnD,OAAO,CAAC,gBAAgB,CAAkC;IAE1D,OAAO,CAAC,YAAY,CAA2F;IAC/G,OAAO,CAAC,eAAe,CAAU;gBAErB,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,gBAAgB,EAAE,eAAe,GAAE,OAAe;IAM/F,SAAS,CAAC,IAAI,IAAI,IAAI;IAYf,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,GAAG,QAAQ,EAAE;IA4FtE,OAAO,CAAC,iCAAiC;IAkCzC,OAAO,CAAC,8BAA8B;IAqCtC,SAAS,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ,EAAE;IA2BtD,OAAO,CAAC,+BAA+B;IA+BhC,eAAe,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI;CAK/G"}

package/lib/callgraph/algorithm/RapidTypeAnalysis.js

@@ -39,16 +39,33 @@ var __importStar = (this && this.__importStar) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.RapidTypeAnalysis = void 0;
 const Expr_1 = require("../../core/base/Expr");
+const Stmt_1 = require("../../core/base/Stmt");
 const AbstractAnalysis_1 = require("./AbstractAnalysis");
 const logger_1 = __importStar(require("../../utils/logger"));
+const TSConst_1 = require("../../core/common/TSConst");
+const Local_1 = require("../../core/base/Local");
+const Const_1 = require("../../core/common/Const");
 const logger = logger_1.default.getLogger(logger_1.LOG_MODULE_TYPE.ARKANALYZER, 'RTA');
 class RapidTypeAnalysis extends AbstractAnalysis_1.AbstractAnalysis {
-    constructor(scene, cg) {
+    constructor(scene, cg, cb, enableThisPrune = false) {
         super(scene, cg);
         // TODO: signature duplicated check
         this.instancedClasses = new Set();
         // TODO: Set duplicated check
         this.ignoredCalls = new Map();
+        this.cgBuilder = cb;
+        this.enableThisPrune = enableThisPrune;
+    }
+    init() {
+        this.scene.getClasses().filter(c => c.getName() === Const_1.DEFAULT_ARK_CLASS_NAME).forEach(c => {
+            var _a;
+            const dfltMethod = (_a = c.getMethodWithName(Const_1.DEFAULT_ARK_METHOD_NAME)) === null || _a === void 0 ? void 0 : _a.getSignature();
+            if (dfltMethod) {
+                const funcID = this.cg.getCallGraphNodeByMethod(dfltMethod).getID();
+                this.workList.push(funcID);
+            }
+        });
+        super.init();
     }
     resolveCall(callerMethod, invokeStmt) {
         let invokeExpr = invokeStmt.getInvokeExpr();
@@ -70,6 +87,31 @@ class RapidTypeAnalysis extends AbstractAnalysis_1.AbstractAnalysis {
         }
         else {
             let declareClass = calleeMethod.getDeclaringArkClass();
+            const methodName = calleeMethod.getName();
+            // Aggressive heuristic: when enabled and the call is `this.foo()`,
+            // only keep the implementation of `foo` in the current declaring class,
+            // or the first superclass that defines it (e.g. B.m() calling this.foo() → A.foo() when B extends A).
+            const prunedCallSite = this.tryResolveAggressiveThisPruneCall(invokeStmt, methodName, callerMethod);
+            if (prunedCallSite) {
+                resolveResult.push(prunedCallSite);
+                return resolveResult;
+            }
+            const directCallSite = this.resolveConstructorClassFromNew(methodName, invokeStmt, callerMethod);
+            if (directCallSite) {
+                resolveResult.push(directCallSite);
+                return resolveResult;
+            }
+            // Private methods cannot be overridden; only the declaring class has an implementation.
+            if (calleeMethod.isPrivate()) {
+                const calleeNode = this.cg.getCallGraphNodeByMethod(calleeMethod.getSignature());
+                if (this.instancedClasses.has(declareClass.getSignature())) {
+                    resolveResult.push(this.cg.getCallSiteManager().newCallSite(invokeStmt, undefined, calleeNode.getID(), callerMethod));
+                }
+                else {
+                    this.addIgnoredCalls(declareClass.getSignature(), callerMethod, calleeNode.getID(), invokeStmt);
+                }
+                return resolveResult;
+            }
             // TODO: super class method should be placed at the end
             this.getClassHierarchy(declareClass).forEach((arkClass) => {
                 let possibleCalleeMethod = arkClass.getMethodWithName(calleeMethod.getName());
@@ -82,26 +124,84 @@ class RapidTypeAnalysis extends AbstractAnalysis_1.AbstractAnalysis {
                     return;
                 }
                 let calleeNode = this.cg.getCallGraphNodeByMethod(possibleCalleeMethod.getSignature());
-                if (!this.instancedClasses.has(arkClass.getSignature())) {
-                    this.addIgnoredCalls(arkClass.getSignature(), callerMethod, calleeNode.getID(), invokeStmt);
+                const isSdkClass = this.scene.hasSdkFile(arkClass.getSignature().getDeclaringFileSignature());
+                const isInstanced = this.instancedClasses.has(arkClass.getSignature());
+                if (isSdkClass || isInstanced) {
+                    resolveResult.push(this.cg.getCallSiteManager().newCallSite(invokeStmt, undefined, calleeNode.getID(), callerMethod));
                 }
                 else {
-                    resolveResult.push(this.cg.getCallSiteManager().newCallSite(invokeStmt, undefined, calleeNode.getID(), callerMethod));
+                    this.addIgnoredCalls(arkClass.getSignature(), callerMethod, calleeNode.getID(), invokeStmt);
                 }
             });
         }
         return resolveResult;
     }
+    tryResolveAggressiveThisPruneCall(invokeStmt, methodName, callerMethod) {
+        const invokeExpr = invokeStmt.getInvokeExpr();
+        if (!this.enableThisPrune || !(invokeExpr instanceof Expr_1.ArkInstanceInvokeExpr)) {
+            return null;
+        }
+        const base = invokeExpr.getBase();
+        if (!base.getName || base.getName() !== TSConst_1.THIS_NAME) {
+            return null;
+        }
+        // Start from the declaring class of this call site, then walk up the super chain.
+        // Pick the first non-abstract implementation of `methodName`.
+        let curClass = invokeStmt.getCfg().getDeclaringMethod().getDeclaringArkClass();
+        while (curClass) {
+            const methodInClass = curClass.getMethodWithName(methodName);
+            if (methodInClass && !methodInClass.isAbstract()) {
+                return this.cg.getCallSiteManager().newCallSite(invokeStmt, undefined, this.cg.getCallGraphNodeByMethod(methodInClass.getSignature()).getID(), callerMethod);
+            }
+            curClass = curClass.getSuperClass();
+        }
+        return null;
+    }
+    resolveConstructorClassFromNew(methodName, invokeStmt, callerMethod) {
+        var _a;
+        if (methodName !== TSConst_1.CONSTRUCTOR_NAME) {
+            return null;
+        }
+        if (!(invokeStmt instanceof Stmt_1.ArkAssignStmt)) {
+            return null;
+        }
+        const leftOp = invokeStmt.getLeftOp();
+        if (!(leftOp instanceof Local_1.Local)) {
+            return null;
+        }
+        const declaringStmt = leftOp.getDeclaringStmt();
+        if (!(declaringStmt && declaringStmt instanceof Stmt_1.ArkAssignStmt)) {
+            return null;
+        }
+        const rightOp = declaringStmt.getRightOp();
+        if (!(rightOp instanceof Expr_1.ArkNewExpr)) {
+            return null;
+        }
+        const classSig = rightOp.getClassType().getClassSignature();
+        const constructorMethod = (_a = this.scene.getClass(classSig)) === null || _a === void 0 ? void 0 : _a.getMethodWithName(TSConst_1.CONSTRUCTOR_NAME);
+        if (!constructorMethod) {
+            return null;
+        }
+        return this.cg.getCallSiteManager().newCallSite(invokeStmt, undefined, this.cg.getCallGraphNodeByMethod(constructorMethod.getSignature()).getID(), callerMethod);
+    }
     preProcessMethod(funcID) {
         let newCallSites = [];
         let instancedClasses = this.collectInstancedClassesInMethod(funcID);
-        let newlyInstancedClasses = new Set(Array.from(instancedClasses).filter(item => !this.instancedClasses.has(item)));
+        let newlyInstancedClasses = new Set();
+        for (const sig of instancedClasses) {
+            if (!this.instancedClasses.has(sig)) {
+                newlyInstancedClasses.add(sig);
+            }
+        }
         newlyInstancedClasses.forEach(sig => {
             let ignoredCalls = this.ignoredCalls.get(sig);
             if (ignoredCalls) {
                 ignoredCalls.forEach(call => {
                     this.cg.addDynamicCallEdge(call.caller, call.callee, call.callStmt);
-                    newCallSites.push(this.cg.getCallSiteManager().newCallSite(call.callStmt, undefined, call.callee, call.caller));
+                    const newCallSite = this.cg.getCallSiteManager().newCallSite(call.callStmt, undefined, call.callee, call.caller);
+                    this.cg.addStmtToCallSiteMap(call.callStmt, newCallSite);
+                    this.cg.addMethodToCallSiteMap(call.callee, newCallSite);
+                    newCallSites.push(newCallSite);
                 });
             }
             this.instancedClasses.add(sig);
@@ -118,17 +218,20 @@ class RapidTypeAnalysis extends AbstractAnalysis_1.AbstractAnalysis {
         }
         let cfg = arkMethod.getCfg();
         if (!cfg) {
-            logger.error(`arkMethod ${arkMethod.getSignature().toString()} has no cfg`);
             return instancedClasses;
         }
         for (let stmt of cfg.getStmts()) {
-            let stmtExpr = stmt.getExprs()[0];
-            if (stmtExpr instanceof Expr_1.ArkNewExpr) {
-                let classSig = stmtExpr.getType().getClassSignature();
-                if (classSig != null) {
-                    // TODO: need to check if different stmt has single sig
-                    instancedClasses.add(classSig);
-                }
+            let stmtExpr;
+            if (stmt instanceof Stmt_1.ArkAssignStmt && stmt.getRightOp() instanceof Expr_1.ArkNewExpr) {
+                stmtExpr = stmt.getRightOp();
+            }
+            else {
+                continue;
+            }
+            let classSig = stmtExpr.getType().getClassSignature();
+            if (classSig != null) {
+                // TODO: need to check if different stmt has single sig
+                instancedClasses.add(classSig);
             }
         }
         return instancedClasses;

package/lib/callgraph/common/Statistics.d.ts

@@ -1,4 +1,4 @@
-import { CallGraphNodeKind } from '../model/CallGraph';
+import { CallGraph, CallGraphNodeKind } from '../model/CallGraph';
 import { PointerAnalysis } from '../pointerAnalysis/PointerAnalysis';
 declare abstract class StatTraits {
     TotalTime: number;
@@ -53,8 +53,10 @@ export declare class CGStat extends StatTraits {
     numIntrinsic: number;
     numConstructor: number;
     numBlank: number;
+    numTotalEdge: number;
     startStat(): void;
-    endStat(): void;
+    endStat(cg?: CallGraph): void;
+    private computeEdgeStat;
     addNodeStat(kind: CallGraphNodeKind): void;
     getStat(): string;
 }

package/lib/callgraph/common/Statistics.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Statistics.d.ts","sourceRoot":"","sources":["../../../src/callgraph/common/Statistics.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAiB,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,oCAAoC,CAAC;AAQrE,uBAAe,UAAU;IACrB,SAAS,EAAE,MAAM,CAAK;IACtB,SAAS,EAAE,MAAM,CAAK;IACtB,OAAO,EAAE,MAAM,CAAK;IAEb,OAAO,IAAI,MAAM;IAIjB,SAAS,IAAI,IAAI;CAG3B;AAED,qBAAa,OAAQ,SAAQ,UAAU;IACnC,GAAG,EAAE,eAAe,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAK;IAC7B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,iBAAiB,EAAE,MAAM,CAAK;IAC9B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,YAAY,EAAE,MAAM,CAAK;IACzB,WAAW,EAAE,MAAM,CAAK;IAExB,eAAe,EAAE,MAAM,CAAK;IAC5B,yBAAyB,EAAE,MAAM,CAAK;IACtC,oBAAoB,EAAE,MAAM,CAAK;IAGjC,sBAAsB,EAAE,MAAM,CAAK;IAEnC,uBAAuB,EAAE,MAAM,CAAK;IAEpC,wBAAwB,EAAE,MAAM,CAAK;IAErC,yBAAyB,EAAE,MAAM,CAAK;IACtC,gBAAgB,EAAE,MAAM,CAAK;IAE7B,SAAS,EAAE,MAAM,CAAK;IAEtB,aAAa,EAAE,GAAG,CAAC;IACnB,WAAW,EAAE,GAAG,CAAC;IACjB,OAAO,EAAE,MAAM,CAAK;IACpB,QAAQ,EAAE,MAAM,CAAK;gBAET,GAAG,EAAE,eAAe;IAKzB,SAAS,IAAI,IAAI;IAKjB,OAAO,IAAI,IAAI;IAUf,MAAM,IAAI,MAAM;IAIvB,OAAO,CAAC,cAAc;IAkCtB,OAAO,CAAC,yBAAyB;IA4BjC,OAAO,CAAC,oBAAoB;IAerB,OAAO,IAAI,MAAM;IAqBjB,SAAS,IAAI,IAAI;CAG3B;AAED,qBAAa,OAAQ,SAAQ,UAAU;IACnC,cAAc,EAAE,MAAM,CAAK;IAC3B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,YAAY,EAAE,MAAM,CAAK;IAElB,OAAO,IAAI,MAAM;IAUjB,SAAS,IAAI,IAAI;CAG3B;AAED,qBAAa,MAAO,SAAQ,UAAU;IAElC,YAAY,EAAE,MAAM,CAAK;IACzB,OAAO,EAAE,MAAM,CAAK;IACpB,UAAU,EAAE,MAAM,CAAK;IACvB,YAAY,EAAE,MAAM,CAAK;IACzB,cAAc,EAAE,MAAM,CAAK;IAC3B,QAAQ,EAAE,MAAM,CAAK;IAEd,SAAS,IAAI,IAAI;IAIjB,OAAO,IAAI,IAAI;IAKf,WAAW,CAAC,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAoB1C,OAAO,IAAI,MAAM;CAa3B"}
+{"version":3,"file":"Statistics.d.ts","sourceRoot":"","sources":["../../../src/callgraph/common/Statistics.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,SAAS,EAAiB,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACjF,OAAO,EAAE,eAAe,EAAE,MAAM,oCAAoC,CAAC;AAQrE,uBAAe,UAAU;IACrB,SAAS,EAAE,MAAM,CAAK;IACtB,SAAS,EAAE,MAAM,CAAK;IACtB,OAAO,EAAE,MAAM,CAAK;IAEb,OAAO,IAAI,MAAM;IAIjB,SAAS,IAAI,IAAI;CAG3B;AAED,qBAAa,OAAQ,SAAQ,UAAU;IACnC,GAAG,EAAE,eAAe,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAK;IAC7B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,iBAAiB,EAAE,MAAM,CAAK;IAC9B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,YAAY,EAAE,MAAM,CAAK;IACzB,WAAW,EAAE,MAAM,CAAK;IAExB,eAAe,EAAE,MAAM,CAAK;IAC5B,yBAAyB,EAAE,MAAM,CAAK;IACtC,oBAAoB,EAAE,MAAM,CAAK;IAGjC,sBAAsB,EAAE,MAAM,CAAK;IAEnC,uBAAuB,EAAE,MAAM,CAAK;IAEpC,wBAAwB,EAAE,MAAM,CAAK;IAErC,yBAAyB,EAAE,MAAM,CAAK;IACtC,gBAAgB,EAAE,MAAM,CAAK;IAE7B,SAAS,EAAE,MAAM,CAAK;IAEtB,aAAa,EAAE,GAAG,CAAC;IACnB,WAAW,EAAE,GAAG,CAAC;IACjB,OAAO,EAAE,MAAM,CAAK;IACpB,QAAQ,EAAE,MAAM,CAAK;gBAET,GAAG,EAAE,eAAe;IAKzB,SAAS,IAAI,IAAI;IAKjB,OAAO,IAAI,IAAI;IAUf,MAAM,IAAI,MAAM;IAIvB,OAAO,CAAC,cAAc;IAkCtB,OAAO,CAAC,yBAAyB;IA4BjC,OAAO,CAAC,oBAAoB;IAerB,OAAO,IAAI,MAAM;IAqBjB,SAAS,IAAI,IAAI;CAG3B;AAED,qBAAa,OAAQ,SAAQ,UAAU;IACnC,cAAc,EAAE,MAAM,CAAK;IAC3B,gBAAgB,EAAE,MAAM,CAAK;IAC7B,YAAY,EAAE,MAAM,CAAK;IAElB,OAAO,IAAI,MAAM;IAUjB,SAAS,IAAI,IAAI;CAG3B;AAED,qBAAa,MAAO,SAAQ,UAAU;IAElC,YAAY,EAAE,MAAM,CAAK;IACzB,OAAO,EAAE,MAAM,CAAK;IACpB,UAAU,EAAE,MAAM,CAAK;IACvB,YAAY,EAAE,MAAM,CAAK;IACzB,cAAc,EAAE,MAAM,CAAK;IAC3B,QAAQ,EAAE,MAAM,CAAK;IAGrB,YAAY,EAAE,MAAM,CAAK;IAElB,SAAS,IAAI,IAAI;IAIjB,OAAO,CAAC,EAAE,CAAC,EAAE,SAAS,GAAG,IAAI;IAQpC,OAAO,CAAC,eAAe;IAOhB,WAAW,CAAC,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAoB1C,OAAO,IAAI,MAAM;CAc3B"}

package/lib/callgraph/common/Statistics.js

@@ -225,13 +225,24 @@ class CGStat extends StatTraits {
         this.numIntrinsic = 0;
         this.numConstructor = 0;
         this.numBlank = 0;
+        // edge statistics
+        this.numTotalEdge = 0;
     }
     startStat() {
         this.startTime = new Date().getTime();
     }
-    endStat() {
+    endStat(cg) {
         this.endTime = new Date().getTime();
         this.TotalTime = (this.endTime - this.startTime) / 1000;
+        if (cg) {
+            this.computeEdgeStat(cg);
+        }
+    }
+    computeEdgeStat(cg) {
+        this.numTotalEdge = 0;
+        for (const _edge of cg.getCallEdges()) {
+            this.numTotalEdge++;
+        }
     }
     addNodeStat(kind) {
         switch (kind) {
@@ -262,7 +273,8 @@ ${'Intrinsic function'.padEnd(LABEL_WIDTH)}${this.numIntrinsic}
 ${'Constructor function'.padEnd(LABEL_WIDTH)}${this.numConstructor}
 ${'Virtual function'.padEnd(LABEL_WIDTH)}${this.numVirtual}
 ${'Blank function'.padEnd(LABEL_WIDTH)}${this.numBlank}
-${'Total'.padEnd(LABEL_WIDTH)}${this.numTotalNode}`;
+${'Total node'.padEnd(LABEL_WIDTH)}${this.numTotalNode}
+${'Total edge'.padEnd(LABEL_WIDTH)}${this.numTotalEdge}`;
     }
 }
 exports.CGStat = CGStat;

package/lib/callgraph/model/CallGraph.d.ts

@@ -15,13 +15,14 @@ export declare enum CallGraphNodeKind {
     blank = 4
 }
 export declare class CallGraphEdge extends BaseEdge {
-    private directCalls;
-    private specialCalls;
-    private indirectCalls;
+    private flags;
     constructor(src: CallGraphNode, dst: CallGraphNode);
-    addDirectCallSite(stmt: Stmt): void;
-    addSpecialCallSite(stmt: Stmt): void;
-    addInDirectCallSite(stmt: Stmt): void;
+    addDirectCallSite(_stmt: Stmt): void;
+    addSpecialCallSite(_stmt: Stmt): void;
+    addInDirectCallSite(_stmt: Stmt): void;
+    hasDirectCall(): boolean;
+    hasIndirectCall(): boolean;
+    hasSpecialCall(): boolean;
     getDotAttr(): string;
 }
 export declare class CallGraphNode extends BaseNode {
@@ -44,11 +45,12 @@ export declare class CallGraph extends BaseExplicitGraph {
     private callPairToEdgeMap;
     private methodToCallSiteMap;
     private entries;
-    private cgStat;
+    private cgStat?;
     private dummyMainMethodID;
-    constructor(s: Scene);
+    constructor(s: Scene, enableStatistics?: boolean);
     private getCallPairString;
     getCallEdgeByPair(srcID: NodeID, dstID: NodeID): CallGraphEdge | undefined;
+    getCallEdges(): Iterable<CallGraphEdge>;
     addCallGraphNode(method: Method, kind?: CallGraphNodeKind): CallGraphNode;
     removeCallGraphNode(nodeID: NodeID): void;
     getCallGraphNodeByMethod(method: Method): CallGraphNode;
@@ -67,7 +69,9 @@ export declare class CallGraph extends BaseExplicitGraph {
     getArkMethodByFuncID(id: FuncID): ArkMethod | null;
     getEntries(): FuncID[];
     setEntries(n: NodeID[]): void;
+    getCallPairEdges(): Map<string, CallGraphEdge>;
     dump(name: string, entry?: FuncID): void;
+    dump2Json(name: string): void;
     detectReachable(fromID: FuncID, dstID: FuncID): boolean;
     startStat(): void;
     endStat(): void;

package/lib/callgraph/model/CallGraph.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CallGraph.d.ts","sourceRoot":"","sources":["../../../src/callgraph/model/CallGraph.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAC5C,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAGvD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAGnG,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,eAAe,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAE3F,MAAM,MAAM,MAAM,GAAG,eAAe,CAAC;AACrC,MAAM,MAAM,MAAM,GAAG,MAAM,CAAC;AAG5B,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;AAE5C,oBAAY,iBAAiB;IACzB,IAAI,IAAA;IACJ,MAAM,IAAA;IACN,SAAS,IAAA;IACT,WAAW,IAAA;IACX,KAAK,IAAA;CACR;AAED,qBAAa,aAAc,SAAQ,QAAQ;IACvC,OAAO,CAAC,WAAW,CAAsB;IACzC,OAAO,CAAC,YAAY,CAAsB;IAC1C,OAAO,CAAC,aAAa,CAAsB;gBAG/B,GAAG,EAAE,aAAa,EAAE,GAAG,EAAE,aAAa;IAI3C,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAInC,kBAAkB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAIpC,mBAAmB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAIrC,UAAU,IAAI,MAAM;CAe9B;AAED,qBAAa,aAAc,SAAQ,QAAQ;IACvC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,WAAW,CAAkB;gBAEzB,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAE,iBAA0C;IAKzE,SAAS,IAAI,MAAM;IAInB,YAAY,CAAC,CAAC,EAAE,OAAO,GAAG,IAAI;IAI9B,WAAW,IAAI,OAAO;IAI7B,IAAW,aAAa,IAAI,OAAO,CAElC;IAEM,UAAU,IAAI,MAAM;IAIpB,WAAW,IAAI,MAAM;CAK/B;AAED,qBAAa,SAAU,SAAQ,iBAAiB;IAC5C,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,SAAS,CAAyB;IAC1C,OAAO,CAAC,iBAAiB,CAAoC;IAC7D,OAAO,CAAC,oBAAoB,CAAqC;IACjE,OAAO,CAAC,iBAAiB,CAAkC;IAC3D,OAAO,CAAC,iBAAiB,CAAyC;IAClE,OAAO,CAAC,mBAAmB,CAAyC;IACpE,OAAO,CAAC,OAAO,CAAY;IAC3B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,iBAAiB,CAAqB;gBAElC,CAAC,EAAE,KAAK;IAMpB,OAAO,CAAC,iBAAiB;IAIlB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAK1E,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAA0C,GAAG,aAAa;IAYjG,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IASzC,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAgBvD,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,GAAE,OAAc,GAAG,IAAI;IA2B9G,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAYzC,kBAAkB,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI;IAYnF,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,GAAG,IAAI;IAmB5E,oBAAoB,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,SAAS;IAIzD,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,GAAG,OAAO;IAUvD,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,QAAQ,EAAE;IAIzC,sBAAsB,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,GAAG,IAAI;IAQ1D,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,GAAG,CAAC,QAAQ,CAAC;IAWnE,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,IAAI,EAAE;IAU7D,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAkBvC,iBAAiB,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAQ5C,oBAAoB,CAAC,EAAE,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI;IAUlD,UAAU,IAAI,MAAM,EAAE;IAItB,UAAU,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI;IAI7B,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAQxC,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO;IA0BvD,SAAS,IAAI,IAAI;IAIjB,OAAO,IAAI,IAAI;IAIf,SAAS,IAAI,IAAI;IAIjB,OAAO,IAAI,MAAM;IAIjB,kBAAkB,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAInD,kBAAkB,IAAI,MAAM,GAAG,SAAS;IAIxC,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAYxC,YAAY,IAAI,MAAM;IAItB,kBAAkB,IAAI,eAAe;IAIrC,eAAe,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM;CAWnD"}
+{"version":3,"file":"CallGraph.d.ts","sourceRoot":"","sources":["../../../src/callgraph/model/CallGraph.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAC5C,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAGvD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAGnG,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,eAAe,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAG3F,MAAM,MAAM,MAAM,GAAG,eAAe,CAAC;AACrC,MAAM,MAAM,MAAM,GAAG,MAAM,CAAC;AAE5B,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;AAE5C,oBAAY,iBAAiB;IACzB,IAAI,IAAA;IACJ,MAAM,IAAA;IACN,SAAS,IAAA;IACT,WAAW,IAAA;IACX,KAAK,IAAA;CACR;AAMD,qBAAa,aAAc,SAAQ,QAAQ;IACvC,OAAO,CAAC,KAAK,CAAa;gBAEd,GAAG,EAAE,aAAa,EAAE,GAAG,EAAE,aAAa;IAI3C,iBAAiB,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI;IAIpC,kBAAkB,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI;IAIrC,mBAAmB,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI;IAItC,aAAa,IAAI,OAAO;IAIxB,eAAe,IAAI,OAAO;IAI1B,cAAc,IAAI,OAAO;IAIzB,UAAU,IAAI,MAAM;CAW9B;AAED,qBAAa,aAAc,SAAQ,QAAQ;IACvC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,WAAW,CAAkB;gBAEzB,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAE,iBAA0C;IAKzE,SAAS,IAAI,MAAM;IAInB,YAAY,CAAC,CAAC,EAAE,OAAO,GAAG,IAAI;IAI9B,WAAW,IAAI,OAAO;IAI7B,IAAW,aAAa,IAAI,OAAO,CAElC;IAEM,UAAU,IAAI,MAAM;IAIpB,WAAW,IAAI,MAAM;CAK/B;AAED,qBAAa,SAAU,SAAQ,iBAAiB;IAC5C,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,SAAS,CAAyB;IAC1C,OAAO,CAAC,iBAAiB,CAAoC;IAC7D,OAAO,CAAC,oBAAoB,CAAqC;IACjE,OAAO,CAAC,iBAAiB,CAAkC;IAC3D,OAAO,CAAC,iBAAiB,CAAyC;IAClE,OAAO,CAAC,mBAAmB,CAAyC;IACpE,OAAO,CAAC,OAAO,CAAY;IAC3B,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,iBAAiB,CAAqB;gBAElC,CAAC,EAAE,KAAK,EAAE,gBAAgB,GAAE,OAAe;IAQvD,OAAO,CAAC,iBAAiB;IAIlB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAK1E,YAAY,IAAI,QAAQ,CAAC,aAAa,CAAC;IAIvC,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAA0C,GAAG,aAAa;IAYjG,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IASzC,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAgBvD,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,GAAE,OAAc,GAAG,IAAI;IA2B9G,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAYzC,kBAAkB,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI;IAYnF,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,GAAG,IAAI;IAmB5E,oBAAoB,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,SAAS;IAIzD,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,GAAG,OAAO;IASvD,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,QAAQ,EAAE;IAIzC,sBAAsB,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,GAAG,IAAI;IAQ1D,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,GAAG,CAAC,QAAQ,CAAC;IAWnE,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,IAAI,EAAE;IAU7D,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAkBvC,iBAAiB,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAQ5C,oBAAoB,CAAC,EAAE,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI;IAUlD,UAAU,IAAI,MAAM,EAAE;IAItB,UAAU,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI;IAI7B,gBAAgB,IAAI,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC;IAI9C,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAQxC,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAK7B,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO;IA0BvD,SAAS,IAAI,IAAI;IAIjB,OAAO,IAAI,IAAI;IAIf,SAAS,IAAI,IAAI;IAIjB,OAAO,IAAI,MAAM;IAIjB,kBAAkB,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAInD,kBAAkB,IAAI,MAAM,GAAG,SAAS;IAIxC,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAYxC,YAAY,IAAI,MAAM;IAItB,kBAAkB,IAAI,eAAe;IAIrC,eAAe,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM;CAWnD"}

package/lib/callgraph/model/CallGraph.js

@@ -24,6 +24,7 @@ const Const_1 = require("../../core/common/Const");
 const CallSite_1 = require("./CallSite");
 Object.defineProperty(exports, "CallSite", { enumerable: true, get: function () { return CallSite_1.CallSite; } });
 Object.defineProperty(exports, "DynCallSite", { enumerable: true, get: function () { return CallSite_1.DynCallSite; } });
+const CGJsonPrinter_1 = require("../../save/CGJsonPrinter");
 var CallGraphNodeKind;
 (function (CallGraphNodeKind) {
     CallGraphNodeKind[CallGraphNodeKind["real"] = 0] = "real";
@@ -32,34 +33,40 @@ var CallGraphNodeKind;
     CallGraphNodeKind[CallGraphNodeKind["constructor"] = 3] = "constructor";
     CallGraphNodeKind[CallGraphNodeKind["blank"] = 4] = "blank";
 })(CallGraphNodeKind = exports.CallGraphNodeKind || (exports.CallGraphNodeKind = {}));
+const EDGE_FLAG_DIRECT = 1;
+const EDGE_FLAG_SPECIAL = 2;
+const EDGE_FLAG_INDIRECT = 4;
 class CallGraphEdge extends BaseExplicitGraph_1.BaseEdge {
-    // private callSiteID: CallSiteID;
     constructor(src, dst) {
         super(src, dst, 0);
-        this.directCalls = new Set();
-        this.specialCalls = new Set();
-        this.indirectCalls = new Set();
+        this.flags = 0;
     }
-    addDirectCallSite(stmt) {
-        this.directCalls.add(stmt);
+    addDirectCallSite(_stmt) {
+        this.flags |= EDGE_FLAG_DIRECT;
     }
-    addSpecialCallSite(stmt) {
-        this.specialCalls.add(stmt);
+    addSpecialCallSite(_stmt) {
+        this.flags |= EDGE_FLAG_SPECIAL;
     }
-    addInDirectCallSite(stmt) {
-        this.indirectCalls.add(stmt);
+    addInDirectCallSite(_stmt) {
+        this.flags |= EDGE_FLAG_INDIRECT;
+    }
+    hasDirectCall() {
+        return (this.flags & EDGE_FLAG_DIRECT) !== 0;
+    }
+    hasIndirectCall() {
+        return (this.flags & EDGE_FLAG_INDIRECT) !== 0;
+    }
+    hasSpecialCall() {
+        return (this.flags & EDGE_FLAG_SPECIAL) !== 0;
     }
     getDotAttr() {
-        const indirectCallNums = this.indirectCalls.size;
-        const directCallNums = this.directCalls.size;
-        const specialCallNums = this.specialCalls.size;
-        if (indirectCallNums !== 0 && directCallNums === 0) {
+        if (this.hasIndirectCall() && !this.hasDirectCall()) {
             return 'color=red';
         }
-        else if (specialCallNums !== 0) {
+        else if (this.hasSpecialCall()) {
             return 'color=yellow';
         }
-        else if (indirectCallNums === 0 && directCallNums !== 0) {
+        else if (this.hasDirectCall()) {
             return 'color=black';
         }
         else {
@@ -97,7 +104,7 @@ class CallGraphNode extends BaseExplicitGraph_1.BaseNode {
 }
 exports.CallGraphNode = CallGraphNode;
 class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
-    constructor(s) {
+    constructor(s, enableStatistics = false) {
         super();
         this.csManager = new CallSite_1.CallSiteManager();
         this.stmtToCallSitemap = new Map();
@@ -106,7 +113,9 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
         this.callPairToEdgeMap = new Map();
         this.methodToCallSiteMap = new Map();
         this.scene = s;
-        this.cgStat = new Statistics_1.CGStat();
+        if (enableStatistics) {
+            this.cgStat = new Statistics_1.CGStat();
+        }
     }
     getCallPairString(srcID, dstID) {
         return `${srcID}-${dstID}`;
@@ -115,14 +124,18 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
         let key = this.getCallPairString(srcID, dstID);
         return this.callPairToEdgeMap.get(key);
     }
+    getCallEdges() {
+        return this.callPairToEdgeMap.values();
+    }
     addCallGraphNode(method, kind = CallGraphNodeKind.real) {
+        var _a;
         let id = this.nodeNum;
         let cgNode = new CallGraphNode(id, method, kind);
         // check if sdk method
         cgNode.setSdkMethod(this.scene.hasSdkFile(method.getDeclaringClassSignature().getDeclaringFileSignature()));
         this.addNode(cgNode);
         this.methodToCGNodeMap.set(method.toString(), cgNode.getID());
-        this.cgStat.addNodeStat(kind);
+        (_a = this.cgStat) === null || _a === void 0 ? void 0 : _a.addNodeStat(kind);
         return cgNode;
     }
     removeCallGraphNode(nodeID) {
@@ -173,11 +186,12 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
         }
     }
     removeCallGraphEdge(nodeID) {
+        var _a, _b;
         let node = this.getNode(nodeID);
-        for (const inEdge of node.getIncomingEdge()) {
+        for (const inEdge of (_a = node.getIncomingEdge()) !== null && _a !== void 0 ? _a : []) {
             node.removeIncomingEdge(inEdge);
         }
-        for (const outEdge of node.getOutgoingEdges()) {
+        for (const outEdge of (_b = node.getOutgoingEdges()) !== null && _b !== void 0 ? _b : []) {
             node.removeIncomingEdge(outEdge);
         }
     }
@@ -212,10 +226,8 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
         return this.stmtToDynCallSitemap.get(stmt);
     }
     addStmtToCallSiteMap(stmt, cs) {
-        var _a;
         if (this.stmtToCallSitemap.has(stmt)) {
-            let callSites = (_a = this.stmtToCallSitemap.get(stmt)) !== null && _a !== void 0 ? _a : [];
-            this.stmtToCallSitemap.set(stmt, [...callSites, cs]);
+            this.stmtToCallSitemap.get(stmt).push(cs);
             return false;
         }
         this.stmtToCallSitemap.set(stmt, [cs]);
@@ -289,6 +301,9 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
     setEntries(n) {
         this.entries = n;
     }
+    getCallPairEdges() {
+        return this.callPairToEdgeMap;
+    }
     dump(name, entry) {
         let printer = new GraphPrinter_1.GraphPrinter(this);
         if (entry) {
@@ -296,7 +311,12 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
         }
         PrinterBuilder_1.PrinterBuilder.dump(printer, name);
     }
+    dump2Json(name) {
+        let printer = new CGJsonPrinter_1.CallGraphJsonPrinter(this);
+        PrinterBuilder_1.PrinterBuilder.dump(printer, name);
+    }
     detectReachable(fromID, dstID) {
+        var _a;
         let dWorklist = [];
         let travserdFuncs = new Set();
         dWorklist.push(fromID);
@@ -307,7 +327,7 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
             }
             travserdFuncs.add(nodeID);
             let node = this.getNode(nodeID);
-            for (let e of node.getOutgoingEdges()) {
+            for (let e of (_a = node.getOutgoingEdges()) !== null && _a !== void 0 ? _a : []) {
                 let dst = e.getDstID();
                 if (dst === dstID) {
                     return true;
@@ -318,16 +338,20 @@ class CallGraph extends BaseExplicitGraph_1.BaseExplicitGraph {
         return false;
     }
     startStat() {
-        this.cgStat.startStat();
+        var _a;
+        (_a = this.cgStat) === null || _a === void 0 ? void 0 : _a.startStat();
     }
     endStat() {
-        this.cgStat.endStat();
+        var _a;
+        (_a = this.cgStat) === null || _a === void 0 ? void 0 : _a.endStat(this);
     }
     printStat() {
-        this.cgStat.printStat();
+        var _a;
+        (_a = this.cgStat) === null || _a === void 0 ? void 0 : _a.printStat();
     }
     getStat() {
-        return this.cgStat.getStat();
+        var _a, _b;
+        return (_b = (_a = this.cgStat) === null || _a === void 0 ? void 0 : _a.getStat()) !== null && _b !== void 0 ? _b : '';
     }
     setDummyMainFuncID(dummyMainMethodID) {
         this.dummyMainMethodID = dummyMainMethodID;

package/lib/callgraph/model/builder/CallGraphBuilder.d.ts

@@ -10,6 +10,7 @@ export declare class CallGraphBuilder {
     buildDirectCallGraph(methods: ArkMethod[]): void;
     buildClassHierarchyCallGraph(entries: Method[], displayGeneratedMethod?: boolean): void;
     buildCHA4WholeProject(displayGeneratedMethod?: boolean): void;
+    buildRTA4WholeProject(displayGeneratedMethod?: boolean): void;
     buildRapidTypeCallGraph(entries: Method[], displayGeneratedMethod?: boolean): void;
     private getDCCallee;
     setEntries(): void;

package/lib/callgraph/model/builder/CallGraphBuilder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CallGraphBuilder.d.ts","sourceRoot":"","sources":["../../../../src/callgraph/model/builder/CallGraphBuilder.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,SAAS,EAAoC,MAAM,EAAE,MAAM,cAAc,CAAC;AACnF,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAKvC,OAAO,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAE1D,qBAAa,gBAAgB;IACzB,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,KAAK,CAAQ;gBAET,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK;IAK3B,4BAA4B,IAAI,IAAI;IAWpC,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,IAAI;IAgBxC,oBAAoB,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,IAAI;IA2BhD,4BAA4B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAW9F,qBAAqB,CAAC,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAKpE,uBAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAYhG,OAAO,CAAC,WAAW;IAIZ,UAAU,IAAI,IAAI;CAO5B"}
+{"version":3,"file":"CallGraphBuilder.d.ts","sourceRoot":"","sources":["../../../../src/callgraph/model/builder/CallGraphBuilder.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,SAAS,EAAoC,MAAM,EAAE,MAAM,cAAc,CAAC;AACnF,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAKvC,OAAO,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAE1D,qBAAa,gBAAgB;IACzB,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,KAAK,CAAQ;gBAET,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK;IAK3B,4BAA4B,IAAI,IAAI;IAWpC,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,IAAI;IAgBxC,oBAAoB,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,IAAI;IA2BhD,4BAA4B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAW9F,qBAAqB,CAAC,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAKpE,qBAAqB,CAAC,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAKpE,uBAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,sBAAsB,GAAE,OAAe,GAAG,IAAI;IAYhG,OAAO,CAAC,WAAW;IAIZ,UAAU,IAAI,IAAI;CAO5B"}

package/lib/callgraph/model/builder/CallGraphBuilder.js

@@ -87,13 +87,17 @@ class CallGraphBuilder {
         let classHierarchyAnalysis = new ClassHierarchyAnalysis_1.ClassHierarchyAnalysis(this.scene, this.cg, this);
         classHierarchyAnalysis.projectStart(displayGeneratedMethod);
     }
+    buildRTA4WholeProject(displayGeneratedMethod = false) {
+        let rapidTypeAnalysis = new RapidTypeAnalysis_1.RapidTypeAnalysis(this.scene, this.cg, this);
+        rapidTypeAnalysis.projectStart(displayGeneratedMethod);
+    }
     buildRapidTypeCallGraph(entries, displayGeneratedMethod = false) {
         let cgEntries = [];
         entries.forEach((entry) => {
             cgEntries.push(this.cg.getCallGraphNodeByMethod(entry).getID());
         });
         this.cg.setEntries(cgEntries);
-        let rapidTypeAnalysis = new RapidTypeAnalysis_1.RapidTypeAnalysis(this.scene, this.cg);
+        let rapidTypeAnalysis = new RapidTypeAnalysis_1.RapidTypeAnalysis(this.scene, this.cg, this);
         rapidTypeAnalysis.start(displayGeneratedMethod);
     }
     /// Get direct call callee