arkanalyzer 1.0.7 → 1.0.9

package/lib/Scene.js

@@ -77,7 +77,6 @@ class Scene {
         this.realProjectDir = '';
         this.moduleScenesMap = new Map();
         this.modulePath2NameMap = new Map();
-        this.globalImportInfos = [];
         this.moduleSdkMap = new Map();
         this.projectSdkMap = new Map();
         // values that are visible in curr scope
@@ -95,6 +94,24 @@ class Scene {
         this.ohPkgContent = {};
         this.buildStage = SceneBuildStage.BUILD_INIT;
     }
+    getOptions() {
+        return this.options;
+    }
+    clear() {
+        this.projectFiles = [];
+        this.moduleScenesMap.clear();
+        this.modulePath2NameMap.clear();
+        this.moduleSdkMap.clear();
+        this.projectSdkMap.clear();
+        this.filesMap.clear();
+        this.namespacesMap.clear();
+        this.classesMap.clear();
+        this.methodsMap.clear();
+        this.sdkArkFilesMap.clear();
+        this.sdkGlobalMap.clear();
+        this.ohPkgContentMap.clear();
+        this.ohPkgContent = {};
+    }
     getStage() {
         return this.buildStage;
     }
@@ -106,21 +123,20 @@ class Scene {
      * @example
      * 1. Build Scene object from scene config
 
-    ```typescript
-    // build config
-    const projectDir = ... ...;
-    const sceneConfig = new SceneConfig();
-    sceneConfig.buildFromProjectDir(projectDir);
+     ```typescript
+     // build config
+     const projectDir = ... ...;
+     const sceneConfig = new SceneConfig();
+     sceneConfig.buildFromProjectDir(projectDir);
 
-    // build scene
-    const scene = new Scene();
-    scene.buildSceneFromProjectDir(sceneConfig);
-    ```
-    */
+     // build scene
+     const scene = new Scene();
+     scene.buildSceneFromProjectDir(sceneConfig);
+     ```
+     */
     buildSceneFromProjectDir(sceneConfig) {
         this.buildBasicInfo(sceneConfig);
         this.genArkFiles();
-        this.collectProjectImportInfos();
     }
     /**
      * Set the basic information of the scene using a config,
@@ -129,6 +145,7 @@ class Scene {
      */
     buildBasicInfo(sceneConfig) {
         var _a;
+        this.options = sceneConfig.getOptions();
         this.projectName = sceneConfig.getTargetProjectName();
         this.realProjectDir = fs_1.default.realpathSync(sceneConfig.getTargetProjectDirectory());
         this.projectFiles = sceneConfig.getProjectFiles();
@@ -178,6 +195,9 @@ class Scene {
                 }
             }
         });
+        this.sdkArkFilesMap.forEach(file => {
+            this.inferFile(file);
+        });
     }
     addDefaultConstructors() {
         for (const file of this.getFiles()) {
@@ -211,21 +231,22 @@ class Scene {
             let arkFile = new ArkFile_1.ArkFile();
             arkFile.setScene(this);
             (0, ArkFileBuilder_1.buildArkFileFromFile)(file, this.realProjectDir, arkFile, this.projectName);
-            this.filesMap.set(arkFile.getFileSignature().toString(), arkFile);
+            this.filesMap.set(arkFile.getFileSignature().toMapKey(), arkFile);
         });
         this.buildAllMethodBody();
         this.addDefaultConstructors();
     }
     buildSdk(sdkName, sdkPath) {
-        const allFiles = (0, getAllFiles_1.getAllFiles)(sdkPath, ['.ets', '.ts']);
+        const allFiles = (0, getAllFiles_1.getAllFiles)(sdkPath, ['.ets', '.ts'], this.options.ignoreFileNames);
         allFiles.forEach((file) => {
             logger.info('=== parse sdk file:', file);
             let arkFile = new ArkFile_1.ArkFile();
             arkFile.setScene(this);
             (0, ArkFileBuilder_1.buildArkFileFromFile)(file, path_1.default.normalize(sdkPath), arkFile, sdkName);
             ModelUtils_1.ModelUtils.getAllClassesInFile(arkFile).forEach(cls => { var _a; return (_a = cls.getDefaultArkMethod()) === null || _a === void 0 ? void 0 : _a.buildBody(); });
-            const fileSig = arkFile.getFileSignature().toString();
+            const fileSig = arkFile.getFileSignature().toMapKey();
             this.sdkArkFilesMap.set(fileSig, arkFile);
+            ModelUtils_1.ModelUtils.buildGlobalMap(arkFile, this.sdkGlobalMap);
         });
     }
     /**
@@ -241,8 +262,8 @@ class Scene {
             }
         });
         this.modulePath2NameMap.forEach((value, key) => {
-            let moduleScene = new ModuleScene();
-            moduleScene.ModuleScenBuilder(value, key, this);
+            let moduleScene = new ModuleScene(this);
+            moduleScene.ModuleSceneBuilder(value, key);
             this.moduleScenesMap.set(value, moduleScene);
         });
         this.buildAllMethodBody();
@@ -286,8 +307,8 @@ class Scene {
                 });
             }
         }
-        let moduleScene = new ModuleScene();
-        moduleScene.ModuleScenBuilder(moduleName, modulePath, this);
+        let moduleScene = new ModuleScene(this);
+        moduleScene.ModuleSceneBuilder(moduleName, modulePath);
         this.moduleScenesMap.set(moduleName, moduleScene);
         this.buildAllMethodBody();
     }
@@ -326,23 +347,29 @@ class Scene {
      * @example
      * 1. get ArkFile based on file signature.
 
-    ```typescript
-    if (...) {
-        const fromSignature = new FileSignature();
-        fromSignature.setProjectName(im.getDeclaringArkFile().getProjectName());
-        fromSignature.setFileName(fileName);
-        return scene.getFile(fromSignature);
-    }
-    ```
+     ```typescript
+     if (...) {
+     const fromSignature = new FileSignature();
+     fromSignature.setProjectName(im.getDeclaringArkFile().getProjectName());
+     fromSignature.setFileName(fileName);
+     return scene.getFile(fromSignature);
+     }
+     ```
      */
     getFile(fileSignature) {
         if (this.projectName === fileSignature.getProjectName()) {
-            return this.filesMap.get(fileSignature.toString()) || null;
+            return this.filesMap.get(fileSignature.toMapKey()) || null;
         }
         else {
-            return this.sdkArkFilesMap.get(fileSignature.toString()) || null;
+            return this.sdkArkFilesMap.get(fileSignature.toMapKey()) || null;
         }
     }
+    setFile(file) {
+        this.filesMap.set(file.getFileSignature().toMapKey(), file);
+    }
+    hasSdkFile(fileSignature) {
+        return this.sdkArkFilesMap.has(fileSignature.toMapKey());
+    }
     /**
      * Get files of a {@link Scene}. Generally, a project includes several ets/ts files that define the different class. We need to generate {@link ArkFile} objects from these ets/ts files.
      * @returns The array of {@link ArkFile} from `scene.filesMap.values()`.
@@ -373,8 +400,8 @@ class Scene {
     getFiles() {
         return Array.from(this.filesMap.values());
     }
-    getSdkArkFilesMap() {
-        return this.sdkArkFilesMap;
+    getSdkArkFiles() {
+        return Array.from(this.sdkArkFilesMap.values());
     }
     getModuleSdkMap() {
         return this.moduleSdkMap;
@@ -382,15 +409,17 @@ class Scene {
     getProjectSdkMap() {
         return this.projectSdkMap;
     }
-    getFilesMap() {
-        return this.filesMap;
-    }
     getNamespace(namespaceSignature) {
+        var _a;
         if (this.projectName === namespaceSignature.getDeclaringFileSignature().getProjectName()) {
-            return this.getNamespacesMap().get(namespaceSignature.toString()) || null;
+            return this.getNamespacesMap().get(namespaceSignature.toMapKey()) || null;
         }
         else {
-            const arkFile = this.sdkArkFilesMap.get(namespaceSignature.getDeclaringFileSignature().toString());
+            const arkFile = this.getFile(namespaceSignature.getDeclaringFileSignature());
+            const parentNs = namespaceSignature.getDeclaringNamespaceSignature();
+            if (parentNs) {
+                return ((_a = arkFile === null || arkFile === void 0 ? void 0 : arkFile.getNamespace(parentNs)) === null || _a === void 0 ? void 0 : _a.getNamespace(namespaceSignature)) || null;
+            }
             return (arkFile === null || arkFile === void 0 ? void 0 : arkFile.getNamespace(namespaceSignature)) || null;
         }
     }
@@ -398,7 +427,7 @@ class Scene {
         if (this.namespacesMap.size === 0) {
             for (const file of this.getFiles()) {
                 ModelUtils_1.ModelUtils.getAllNamespacesInFile(file).forEach((namespace) => {
-                    this.namespacesMap.set(namespace.getNamespaceSignature().toString(), namespace);
+                    this.namespacesMap.set(namespace.getNamespaceSignature().toMapKey(), namespace);
                 });
             }
         }
@@ -415,10 +444,10 @@ class Scene {
     getClass(classSignature, refresh) {
         var _a;
         if (this.projectName === classSignature.getDeclaringFileSignature().getProjectName()) {
-            return this.getClassesMap(refresh).get(classSignature.toString()) || null;
+            return this.getClassesMap(refresh).get(classSignature.toMapKey()) || null;
         }
         else {
-            const arkFile = this.sdkArkFilesMap.get(classSignature.getDeclaringFileSignature().toString());
+            const arkFile = this.getFile(classSignature.getDeclaringFileSignature());
             const namespaceSignature = classSignature.getDeclaringNamespaceSignature();
             if (namespaceSignature) {
                 return ((_a = arkFile === null || arkFile === void 0 ? void 0 : arkFile.getNamespace(namespaceSignature)) === null || _a === void 0 ? void 0 : _a.getClass(classSignature)) || null;
@@ -431,12 +460,12 @@ class Scene {
             this.classesMap.clear();
             for (const file of this.getFiles()) {
                 for (const cls of file.getClasses()) {
-                    this.classesMap.set(cls.getSignature().toString(), cls);
+                    this.classesMap.set(cls.getSignature().toMapKey(), cls);
                 }
             }
             for (const namespace of this.getNamespacesMap().values()) {
                 for (const cls of namespace.getClasses()) {
-                    this.classesMap.set(cls.getSignature().toString(), cls);
+                    this.classesMap.set(cls.getSignature().toMapKey(), cls);
                 }
             }
         }
@@ -448,7 +477,7 @@ class Scene {
     getMethod(methodSignature, refresh) {
         var _a;
         if (this.projectName === methodSignature.getDeclaringClassSignature().getDeclaringFileSignature().getProjectName()) {
-            return this.getMethodsMap(refresh).get(methodSignature.toString()) || null;
+            return this.getMethodsMap(refresh).get(methodSignature.toMapKey()) || null;
         }
         else {
             return ((_a = this.getClass(methodSignature.getDeclaringClassSignature())) === null || _a === void 0 ? void 0 : _a.getMethod(methodSignature)) || null;
@@ -459,7 +488,7 @@ class Scene {
             this.methodsMap.clear();
             for (const cls of this.getClassesMap().values()) {
                 for (const method of cls.getMethods(true)) {
-                    this.methodsMap.set(method.getSignature().toString(), method);
+                    this.methodsMap.set(method.getSignature().toMapKey(), method);
                 }
             }
         }
@@ -482,19 +511,31 @@ class Scene {
      * @example
      * 1. get method from getMethod.
 
-    ```typescript
-    const methodSignatures = this.CHA.resolveCall(xxx, yyy);
-    for (const methodSignature of methodSignatures) {
-        const method = this.scene.getMethod(methodSignature);
-        ... ...
-    }
-    ```
+     ```typescript
+     const methodSignatures = this.CHA.resolveCall(xxx, yyy);
+     for (const methodSignature of methodSignatures) {
+     const method = this.scene.getMethod(methodSignature);
+     ... ...
+     }
+     ```
      */
     getMethods(refresh) {
         return Array.from(this.getMethodsMap(refresh).values());
     }
     addToMethodsMap(method) {
-        this.methodsMap.set(method.getSignature().toString(), method);
+        this.methodsMap.set(method.getSignature().toMapKey(), method);
+    }
+    removeMethod(method) {
+        return this.methodsMap.delete(method.getSignature().toMapKey());
+    }
+    removeClass(arkClass) {
+        return this.classesMap.delete(arkClass.getSignature().toMapKey());
+    }
+    removeNamespace(namespace) {
+        return this.namespacesMap.delete(namespace.getSignature().toMapKey());
+    }
+    removeFile(file) {
+        return this.filesMap.delete(file.getFileSignature().toMapKey());
     }
     hasMainMethod() {
         return false;
@@ -534,29 +575,28 @@ class Scene {
      * @example
      * 1. Infer the type of each class field and method field.
 
-    ```typescript
-    const scene = new Scene();
-    scene.buildSceneFromProjectDir(sceneConfig);
-    scene.inferTypes();
-    ```
+     ```typescript
+     const scene = new Scene();
+     scene.buildSceneFromProjectDir(sceneConfig);
+     scene.inferTypes();
+     ```
      */
     inferTypes() {
-        this.sdkArkFilesMap.forEach(file => {
-            ModelUtils_1.ModelUtils.buildGlobalMap(file, this.sdkGlobalMap);
-        });
-        this.filesMap.forEach(file => {
-            ModelUtils_1.ModelUtils.getAllClassesInFile(file).forEach(arkClass => {
-                TypeInference_1.TypeInference.inferGenericType(arkClass.getGenericsTypes(), arkClass);
-                arkClass.getFields().forEach(arkField => TypeInference_1.TypeInference.inferTypeInArkField(arkField));
-                const defaultArkMethod = arkClass.getDefaultArkMethod();
-                if (defaultArkMethod) {
-                    TypeInference_1.TypeInference.inferTypeInMethod(defaultArkMethod);
-                }
-                arkClass.getMethods().forEach(arkMethod => TypeInference_1.TypeInference.inferTypeInMethod(arkMethod));
-            });
-        });
+        this.filesMap.forEach(file => this.inferFile(file));
+        this.getClasses(true);
         this.getMethodsMap(true);
     }
+    inferFile(file) {
+        ModelUtils_1.ModelUtils.getAllClassesInFile(file).forEach(arkClass => {
+            TypeInference_1.TypeInference.inferGenericType(arkClass.getGenericsTypes(), arkClass);
+            const defaultArkMethod = arkClass.getDefaultArkMethod();
+            if (defaultArkMethod) {
+                TypeInference_1.TypeInference.inferTypeInMethod(defaultArkMethod);
+            }
+            arkClass.getFields().forEach(arkField => TypeInference_1.TypeInference.inferTypeInArkField(arkField));
+            arkClass.getMethods().forEach(arkMethod => TypeInference_1.TypeInference.inferTypeInMethod(arkMethod));
+        });
+    }
     /**
      * Iterate all assignment statements in methods,
      * and set the type of left operand based on the type of right operand
@@ -565,11 +605,11 @@ class Scene {
      * @example
      * 1. Infer simple type when scene building.
 
-    ```typescript
-    let scene = new Scene();
-    scene.buildSceneFromProjectDir(config);
-    scene.inferSimpleTypes();
-    ```
+     ```typescript
+     let scene = new Scene();
+     scene.buildSceneFromProjectDir(config);
+     scene.inferSimpleTypes();
+     ```
      */
     inferSimpleTypes() {
         for (let arkFile of this.getFiles()) {
@@ -580,13 +620,6 @@ class Scene {
             }
         }
     }
-    collectProjectImportInfos() {
-        this.getFiles().forEach((arkFile) => {
-            arkFile.getImportInfos().forEach((importInfo) => {
-                this.globalImportInfos.push(importInfo);
-            });
-        });
-    }
     getClassMap() {
         var _a, _b, _c, _d;
         const classMap = new Map();
@@ -669,7 +702,7 @@ class Scene {
                     try {
                         // 遗留问题：只统计了项目文件的namespace，没统计sdk文件内部的引入
                         const importNameSpaceClasses = classMap.get(importNameSpace.getNamespaceSignature());
-                        importClasses.push(...importNameSpaceClasses.filter(c => !importClasses.includes(c) && c.getName() !== '_DEFAULT_ARK_CLASS'));
+                        importClasses.push(...importNameSpaceClasses.filter(c => !importClasses.includes(c) && c.getName() !== Const_1.DEFAULT_ARK_CLASS_NAME));
                     }
                     catch (_e) {
                     }
@@ -681,14 +714,14 @@ class Scene {
             const namespaceStack = [...file.getNamespaces()];
             for (const ns of namespaceStack) {
                 const nsClasses = classMap.get(ns.getNamespaceSignature());
-                nsClasses.push(...fileClasses.filter(c => !nsClasses.includes(c) && c.getName() !== '_DEFAULT_ARK_CLASS'));
+                nsClasses.push(...fileClasses.filter(c => !nsClasses.includes(c) && c.getName() !== Const_1.DEFAULT_ARK_CLASS_NAME));
             }
             while (namespaceStack.length > 0) {
                 const ns = namespaceStack.shift();
                 const nsClasses = classMap.get(ns.getNamespaceSignature());
                 for (const nsns of ns.getNamespaces()) {
                     const nsnsClasses = classMap.get(nsns.getNamespaceSignature());
-                    nsnsClasses.push(...nsClasses.filter(c => !nsnsClasses.includes(c) && c.getName() !== '_DEFAULT_ARK_CLASS'));
+                    nsnsClasses.push(...nsClasses.filter(c => !nsnsClasses.includes(c) && c.getName() !== Const_1.DEFAULT_ARK_CLASS_NAME));
                     namespaceStack.push(nsns);
                 }
             }
@@ -780,7 +813,7 @@ class Scene {
                     try {
                         // 遗留问题：只统计了项目文件，没统计sdk文件内部的引入
                         const importNameSpaceClasses = globalVariableMap.get(importNameSpace.getNamespaceSignature());
-                        importLocals.push(...importNameSpaceClasses.filter(c => !importLocals.includes(c) && c.getName() !== '_DEFAULT_ARK_CLASS'));
+                        importLocals.push(...importNameSpaceClasses.filter(c => !importLocals.includes(c) && c.getName() !== Const_1.DEFAULT_ARK_CLASS_NAME));
                     }
                     catch (_h) {
                     }
@@ -837,17 +870,16 @@ class Scene {
 }
 exports.Scene = Scene;
 class ModuleScene {
-    constructor() {
-        this.projectScene = new Scene();
+    constructor(projectScene) {
         this.moduleName = '';
         this.modulePath = '';
         this.moduleOhPkgFilePath = '';
         this.ohPkgContent = {};
+        this.projectScene = projectScene;
     }
-    ModuleScenBuilder(moduleName, modulePath, projectScene, recursively = false) {
+    ModuleSceneBuilder(moduleName, modulePath, recursively = false) {
         this.moduleName = moduleName;
         this.modulePath = modulePath;
-        this.projectScene = projectScene;
         this.getModuleOhPkgFilePath();
         if (this.moduleOhPkgFilePath) {
             this.ohPkgContent = (0, json5parser_1.fetchDependenciesFromFile)(this.moduleOhPkgFilePath);
@@ -883,14 +915,13 @@ class ModuleScene {
         return this.ohPkgContent;
     }
     genArkFiles() {
-        (0, getAllFiles_1.getAllFiles)(this.modulePath, ['.ets', '.ts']).forEach((file) => {
+        (0, getAllFiles_1.getAllFiles)(this.modulePath, ['.ets', '.ts'], this.projectScene.getOptions().ignoreFileNames).forEach((file) => {
             logger.info('=== parse file:', file);
             let arkFile = new ArkFile_1.ArkFile();
             arkFile.setScene(this.projectScene);
             arkFile.setModuleScene(this);
             (0, ArkFileBuilder_1.buildArkFileFromFile)(file, this.projectScene.getRealProjectDir(), arkFile, this.projectScene.getProjectName());
-            const fileSig = arkFile.getFileSignature().toString();
-            this.projectScene.getFilesMap().set(fileSig, arkFile);
+            this.projectScene.setFile(arkFile);
         });
     }
 }

package/lib/callgraph/algorithm/AbstractAnalysis.js

@@ -76,7 +76,7 @@ class AbstractAnalysis {
         while (this.workList.length !== 0) {
             const method = this.workList.shift();
             const cgNode = this.cg.getNode(method);
-            if (this.processedMethod.has(method) || cgNode.getIsSdkMethod()) {
+            if (this.processedMethod.has(method) || cgNode.isSdkMethod()) {
                 continue;
             }
             // pre process for RTA only

package/lib/callgraph/common/Statistics.js

@@ -108,7 +108,7 @@ class PTAStat {
         let cg = this.pta.getCallGraph();
         this.pta.getUnhandledFuncs().forEach(funcID => {
             let cgNode = cg.getNode(funcID);
-            if (cgNode.getIsSdkMethod()) {
+            if (cgNode.isSdkMethod()) {
                 return;
             }
             let f = cg.getArkMethodByFuncID(funcID);

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

@@ -44,12 +44,12 @@ export declare class CallGraphEdge extends BaseEdge {
 }
 export declare class CallGraphNode extends BaseNode {
     private method;
-    private isSdkMethod;
+    private ifSdkMethod;
     private isBlank;
     constructor(id: number, m: Method, k?: CallGraphNodeKind);
     getMethod(): Method;
     setSdkMethod(v: boolean): void;
-    getIsSdkMethod(): boolean;
+    isSdkMethod(): boolean;
     get isBlankMethod(): boolean;
     set isBlankMethod(is: boolean);
     getDotAttr(): string;
@@ -90,5 +90,6 @@ export declare class CallGraph extends BaseGraph {
     printStat(): void;
     setDummyMainFuncID(dummyMainMethodID: number): void;
     getDummyMainFuncID(): FuncID | undefined;
+    isUnknownMethod(funcID: FuncID): boolean;
 }
 //# sourceMappingURL=CallGraph.d.ts.map

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,CAAA;AAC/D,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAA;AAC3C,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAA;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAGvD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAEpE,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAEvD,MAAM,MAAM,MAAM,GAAG,eAAe,CAAC;AACrC,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC;AAChC,MAAM,MAAM,MAAM,GAAG,MAAM,CAAC;AAG5B,oBAAY,iBAAiB;IACzB,IAAI,IAAA;IAAE,MAAM,IAAA;IAAE,SAAS,IAAA;IAAE,WAAW,IAAA;CACvC;AAED,qBAAa,QAAQ;IACV,QAAQ,EAAE,IAAI,CAAC;IACf,IAAI,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;gBAEhB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM;CAMtE;AAED,qBAAa,WAAW;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,IAAI,CAAC;IACf,IAAI,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;IAC1B,sBAAsB,EAAE,MAAM,GAAG,SAAS,CAAC;gBAEtC,MAAM,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,SAAS,EAAE,MAAM,GAAG,SAAS;CAM7F;AAED,qBAAa,UAAW,SAAQ,QAAQ;IAC7B,GAAG,EAAE,SAAS,CAAC;gBAEV,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,QAAQ;CAI1C;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;IAI5B,kBAAkB,CAAC,IAAI,EAAE,IAAI;IAI7B,mBAAmB,CAAC,IAAI,EAAE,IAAI;IAI9B,UAAU,IAAI,MAAM;CAkB9B;AAED,qBAAa,aAAc,SAAQ,QAAQ;IACvC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,OAAO,CAAkB;gBAErB,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAE,iBAA0C;IAKzE,SAAS,IAAI,MAAM;IAInB,YAAY,CAAC,CAAC,EAAE,OAAO,GAAG,IAAI;IAI9B,cAAc,IAAI,OAAO;IAIhC,IAAW,aAAa,IAAI,OAAO,CAElC;IAED,IAAW,aAAa,CAAC,EAAE,EAAE,OAAO,EAEnC;IAEM,UAAU,IAAI,MAAM;IAOpB,WAAW,IAAI,MAAM;CAK/B;AAED,qBAAa,SAAU,SAAQ,SAAS;IACpC,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,eAAe,CAAwC;IAC/D,OAAO,CAAC,eAAe,CAAwC;IAC/D,OAAO,CAAC,iBAAiB,CAAkC;IAC3D,OAAO,CAAC,oBAAoB,CAAqC;IACjE,OAAO,CAAC,iBAAiB,CAAkC;IAC3D,OAAO,CAAC,iBAAiB,CAAyC;IAClE,OAAO,CAAC,WAAW,CAAa;IAGhC,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;IAmBjG,mBAAmB,CAAC,MAAM,EAAE,MAAM;IASlC,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;IAkC9G,mBAAmB,CAAC,MAAM,EAAE,MAAM;IAYlC,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;IAcrE,oBAAoB,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,SAAS;IAIzD,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,GAAG,OAAO;IAQvD,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,QAAQ,GAAG,SAAS;IAInD,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAkBvC,iBAAiB,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAS5C,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,kBAAkB,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAInD,kBAAkB,IAAI,MAAM,GAAG,SAAS;CAGlD"}
+{"version":3,"file":"CallGraph.d.ts","sourceRoot":"","sources":["../../../src/callgraph/model/CallGraph.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAA;AAC/D,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAA;AAC3C,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAA;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAGvD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAEpE,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAGvD,MAAM,MAAM,MAAM,GAAG,eAAe,CAAC;AACrC,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC;AAChC,MAAM,MAAM,MAAM,GAAG,MAAM,CAAC;AAG5B,oBAAY,iBAAiB;IACzB,IAAI,IAAA;IAAE,MAAM,IAAA;IAAE,SAAS,IAAA;IAAE,WAAW,IAAA;CACvC;AAED,qBAAa,QAAQ;IACV,QAAQ,EAAE,IAAI,CAAC;IACf,IAAI,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;gBAEhB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM;CAMtE;AAED,qBAAa,WAAW;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,IAAI,CAAC;IACf,IAAI,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;IAC1B,sBAAsB,EAAE,MAAM,GAAG,SAAS,CAAC;gBAEtC,MAAM,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,SAAS,EAAE,MAAM,GAAG,SAAS;CAM7F;AAED,qBAAa,UAAW,SAAQ,QAAQ;IAC7B,GAAG,EAAE,SAAS,CAAC;gBAEV,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,QAAQ;CAI1C;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;IAI5B,kBAAkB,CAAC,IAAI,EAAE,IAAI;IAI7B,mBAAmB,CAAC,IAAI,EAAE,IAAI;IAI9B,UAAU,IAAI,MAAM;CAkB9B;AAED,qBAAa,aAAc,SAAQ,QAAQ;IACvC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,OAAO,CAAkB;gBAErB,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;IAED,IAAW,aAAa,CAAC,EAAE,EAAE,OAAO,EAEnC;IAEM,UAAU,IAAI,MAAM;IAOpB,WAAW,IAAI,MAAM;CAK/B;AAED,qBAAa,SAAU,SAAQ,SAAS;IACpC,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,eAAe,CAAwC;IAC/D,OAAO,CAAC,eAAe,CAAwC;IAC/D,OAAO,CAAC,iBAAiB,CAAkC;IAC3D,OAAO,CAAC,oBAAoB,CAAqC;IACjE,OAAO,CAAC,iBAAiB,CAAkC;IAC3D,OAAO,CAAC,iBAAiB,CAAyC;IAClE,OAAO,CAAC,WAAW,CAAa;IAGhC,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;IAmBjG,mBAAmB,CAAC,MAAM,EAAE,MAAM;IASlC,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;IAkC9G,mBAAmB,CAAC,MAAM,EAAE,MAAM;IAYlC,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;IAcrE,oBAAoB,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,SAAS;IAIzD,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,GAAG,OAAO;IAQvD,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,QAAQ,GAAG,SAAS;IAInD,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAkBvC,iBAAiB,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAS5C,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,kBAAkB,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAInD,kBAAkB,IAAI,MAAM,GAAG,SAAS;IAIxC,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;CAWlD"}

package/lib/callgraph/model/CallGraph.js

@@ -19,6 +19,7 @@ const GraphPrinter_1 = require("../../save/GraphPrinter");
 const PrinterBuilder_1 = require("../../save/PrinterBuilder");
 const BaseGraph_1 = require("./BaseGraph");
 const Statistics_1 = require("../common/Statistics");
+const Const_1 = require("../../core/common/Const");
 var CallGraphNodeKind;
 (function (CallGraphNodeKind) {
     CallGraphNodeKind[CallGraphNodeKind["real"] = 0] = "real";
@@ -93,7 +94,7 @@ exports.CallGraphEdge = CallGraphEdge;
 class CallGraphNode extends BaseGraph_1.BaseNode {
     constructor(id, m, k = CallGraphNodeKind.real) {
         super(id, k);
-        this.isSdkMethod = false;
+        this.ifSdkMethod = false;
         this.isBlank = false;
         this.method = m;
     }
@@ -101,10 +102,10 @@ class CallGraphNode extends BaseGraph_1.BaseNode {
         return this.method;
     }
     setSdkMethod(v) {
-        this.isSdkMethod = v;
+        this.ifSdkMethod = v;
     }
-    getIsSdkMethod() {
-        return this.isSdkMethod;
+    isSdkMethod() {
+        return this.ifSdkMethod;
     }
     get isBlankMethod() {
         return this.isBlank;
@@ -149,7 +150,7 @@ class CallGraph extends BaseGraph_1.BaseGraph {
         let id = this.nodeNum;
         let cgNode = new CallGraphNode(id, method, kind);
         // check if sdk method
-        cgNode.setSdkMethod(this.scene.getSdkArkFilesMap().has(method.getDeclaringClassSignature().getDeclaringFileSignature().toString()));
+        cgNode.setSdkMethod(this.scene.hasSdkFile(method.getDeclaringClassSignature().getDeclaringFileSignature()));
         let arkMethod = this.scene.getMethod(method);
         if (!arkMethod || !arkMethod.getCfg()) {
             cgNode.isBlankMethod = true;
@@ -334,5 +335,14 @@ class CallGraph extends BaseGraph_1.BaseGraph {
     getDummyMainFuncID() {
         return this.dummyMainMethodID;
     }
+    isUnknownMethod(funcID) {
+        let method = this.getMethodByFuncID(funcID);
+        if (method) {
+            if (!(method.getDeclaringClassSignature().getDeclaringFileSignature().getFileName() === Const_1.UNKNOWN_FILE_NAME)) {
+                return false;
+            }
+        }
+        return true;
+    }
 }
 exports.CallGraph = CallGraph;

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

@@ -35,7 +35,7 @@ class CallGraphBuilder {
         for (const method of methods) {
             let m = method.getSignature();
             let kind = CallGraph_1.CallGraphNodeKind.real;
-            if (method.isGenerated()) { // || method.getName() === '_DEFAULT_ARK_METHOD') {
+            if (method.isGenerated()) {
                 kind = CallGraph_1.CallGraphNodeKind.intrinsic;
             }
             if (method.getName() === 'constructor') {