arkanalyzer 1.0.7 → 1.0.8

package/lib/core/common/TypeInference.js

@@ -62,26 +62,27 @@ class TypeInference {
         const stmts = arkField.getInitializer();
         let rightType;
         let fieldRef;
-        if (stmts) {
-            for (const stmt of stmts) {
-                this.resolveExprsInStmt(stmt, arkClass);
-                this.resolveFieldRefsInStmt(stmt, arkClass, arkClass.getMethodWithName('@instance_init'));
-                this.resolveArkAssignStmt(stmt, arkClass);
-            }
-            const lastStmt = stmts[stmts.length - 1];
-            if (lastStmt instanceof Stmt_1.ArkAssignStmt) {
-                rightType = lastStmt.getRightOp().getType();
-                if (lastStmt.getLeftOp() instanceof Ref_1.AbstractFieldRef) {
-                    fieldRef = lastStmt.getLeftOp();
-                }
+        const method = arkClass.getMethodWithName(Const_1.INSTANCE_INIT_METHOD_NAME);
+        for (const stmt of stmts) {
+            if (method) {
+                this.resolveExprsInStmt(stmt, method);
+                this.resolveFieldRefsInStmt(stmt, method);
             }
+            this.resolveArkAssignStmt(stmt, arkClass);
         }
         const beforeType = arkField.getType();
-        let fieldType;
-        if (arkField.getCategory() === ArkField_1.FieldCategory.ENUM_MEMBER) {
-            fieldType = new Type_1.ClassType(arkClass.getSignature());
+        if (!this.isUnclearType(beforeType)) {
+            return;
         }
-        else if (beforeType) {
+        const lastStmt = stmts[stmts.length - 1];
+        if (lastStmt instanceof Stmt_1.ArkAssignStmt) {
+            rightType = lastStmt.getRightOp().getType();
+            if (lastStmt.getLeftOp() instanceof Ref_1.AbstractFieldRef) {
+                fieldRef = lastStmt.getLeftOp();
+            }
+        }
+        let fieldType;
+        if (beforeType) {
             fieldType = this.inferUnclearedType(beforeType, arkClass, rightType);
         }
         if (fieldType) {
@@ -96,7 +97,7 @@ class TypeInference {
     static inferUnclearedType(leftOpType, declaringArkClass, rightType) {
         let type;
         if (leftOpType instanceof Type_1.UnclearReferenceType) {
-            type = this.inferUnclearReferenceType(leftOpType.getName(), declaringArkClass);
+            type = this.inferUnclearRefType(leftOpType, declaringArkClass);
         }
         else if (leftOpType instanceof Type_1.ClassType
             && leftOpType.getClassSignature().getDeclaringFileSignature().getFileName() === Const_1.UNKNOWN_FILE_NAME) {
@@ -116,10 +117,10 @@ class TypeInference {
                 else {
                     newType = optionType;
                 }
-                if (newType && newType !== optionType) {
+                if (newType) {
                     types[i] = newType;
                 }
-                if (rightType && newType && newType === rightType) {
+                if (rightType && newType && newType.constructor === rightType.constructor) {
                     leftOpType.setCurrType(rightType);
                     type = leftOpType;
                 }
@@ -145,24 +146,42 @@ class TypeInference {
         return type;
     }
     static inferTypeInMethod(arkMethod) {
-        var _a;
+        var _a, _b;
+        const arkClass = arkMethod.getDeclaringArkClass();
+        this.inferGenericType(arkMethod.getGenericTypes(), arkClass);
+        const signatures = [];
+        (_a = arkMethod.getDeclareSignatures()) === null || _a === void 0 ? void 0 : _a.forEach(m => signatures.push(m));
+        const impl = arkMethod.getImplementationSignature();
+        if (impl) {
+            signatures.push(impl);
+        }
+        signatures.forEach(s => {
+            s.getMethodSubSignature().getParameters().forEach(p => {
+                const type = TypeInference.inferUnclearedType(p.getType(), arkClass);
+                if (type) {
+                    p.setType(type);
+                }
+            });
+            const type = TypeInference.inferUnclearedType(s.getMethodSubSignature().getReturnType(), arkClass);
+            if (type) {
+                s.getMethodSubSignature().setReturnType(type);
+            }
+        });
         const body = arkMethod.getBody();
         if (!body) {
             logger.warn('empty body');
             return;
         }
-        const arkClass = arkMethod.getDeclaringArkClass();
-        (_a = body.getAliasTypeMap()) === null || _a === void 0 ? void 0 : _a.forEach((value) => this.inferUnclearedType(value[0], arkClass));
+        (_b = body.getAliasTypeMap()) === null || _b === void 0 ? void 0 : _b.forEach((value) => this.inferUnclearedType(value[0], arkClass));
         this.inferGenericType(arkMethod.getGenericTypes(), arkClass);
         const cfg = body.getCfg();
         for (const block of cfg.getBlocks()) {
             for (const stmt of block.getStmts()) {
-                this.resolveExprsInStmt(stmt, arkClass);
-                this.resolveFieldRefsInStmt(stmt, arkClass, arkMethod);
+                this.resolveExprsInStmt(stmt, arkMethod);
+                this.resolveFieldRefsInStmt(stmt, arkMethod);
                 this.resolveArkAssignStmt(stmt, arkClass);
             }
         }
-        this.inferMethodReturnType(arkMethod);
     }
     /**
      * @Deprecated
@@ -185,10 +204,10 @@ class TypeInference {
      * infer type for Exprs in stmt which invoke method.
      * such as ArkInstanceInvokeExpr ArkStaticInvokeExpr ArkNewExpr
      */
-    static resolveExprsInStmt(stmt, arkClass) {
+    static resolveExprsInStmt(stmt, arkMethod) {
         const exprs = stmt.getExprs();
         for (const expr of exprs) {
-            const newExpr = expr.inferType(arkClass);
+            const newExpr = expr.inferType(arkMethod);
             if (stmt.containsInvokeExpr() && expr instanceof Expr_1.ArkInstanceInvokeExpr && newExpr instanceof Expr_1.ArkStaticInvokeExpr) {
                 if (stmt instanceof Stmt_1.ArkAssignStmt && stmt.getRightOp() instanceof Expr_1.ArkInstanceInvokeExpr) {
                     stmt.setRightOp(newExpr);
@@ -202,57 +221,50 @@ class TypeInference {
     /**
      * infer type for fieldRefs in stmt.
      */
-    static resolveFieldRefsInStmt(stmt, arkClass, arkMethod) {
-        var _a;
+    static resolveFieldRefsInStmt(stmt, arkMethod) {
         for (const use of stmt.getUses()) {
             if (use instanceof Ref_1.AbstractRef) {
-                const fieldRef = use.inferType(arkClass);
-                if (fieldRef instanceof Ref_1.ArkStaticFieldRef && stmt instanceof Stmt_1.ArkAssignStmt) {
-                    if (stmt.getRightOp() instanceof Ref_1.ArkInstanceFieldRef) {
-                        stmt.setRightOp(fieldRef);
-                    }
-                    else {
-                        stmt.replaceUse(use, fieldRef);
-                        stmt.setRightOp(stmt.getRightOp());
-                    }
-                }
-                else if (use instanceof Ref_1.ArkInstanceFieldRef && fieldRef instanceof Ref_1.ArkArrayRef && stmt instanceof Stmt_1.ArkAssignStmt) {
-                    const index = fieldRef.getIndex();
-                    if (index instanceof Constant_1.Constant && index.getType() instanceof Type_1.StringType) {
-                        const local = (_a = arkMethod === null || arkMethod === void 0 ? void 0 : arkMethod.getBody()) === null || _a === void 0 ? void 0 : _a.getLocals().get(index.getValue());
-                        if (local) {
-                            fieldRef.setIndex(local);
-                        }
-                    }
-                    stmt.replaceUse(use, fieldRef);
-                    stmt.setRightOp(stmt.getRightOp());
-                }
+                this.processRef(use, stmt, arkMethod);
             }
         }
         const stmtDef = stmt.getDef();
         if (stmtDef && stmtDef instanceof Ref_1.AbstractRef) {
-            const fieldRef = stmtDef.inferType(arkClass);
+            const fieldRef = stmtDef.inferType(arkMethod);
             if (fieldRef instanceof Ref_1.ArkStaticFieldRef && stmt instanceof Stmt_1.ArkAssignStmt) {
                 stmt.setLeftOp(fieldRef);
             }
         }
     }
-    static parseArkExport2Type(arkExport) {
+    static processRef(use, stmt, arkMethod) {
         var _a;
+        const fieldRef = use.inferType(arkMethod);
+        if (fieldRef instanceof Ref_1.ArkStaticFieldRef && stmt instanceof Stmt_1.ArkAssignStmt) {
+            if (stmt.getRightOp() instanceof Ref_1.ArkInstanceFieldRef) {
+                stmt.setRightOp(fieldRef);
+            }
+            else {
+                stmt.replaceUse(use, fieldRef);
+                stmt.setRightOp(stmt.getRightOp());
+            }
+        }
+        else if (use instanceof Ref_1.ArkInstanceFieldRef && fieldRef instanceof Ref_1.ArkArrayRef && stmt instanceof Stmt_1.ArkAssignStmt) {
+            const index = fieldRef.getIndex();
+            if (index instanceof Constant_1.Constant && index.getType() instanceof Type_1.StringType) {
+                const local = (_a = arkMethod === null || arkMethod === void 0 ? void 0 : arkMethod.getBody()) === null || _a === void 0 ? void 0 : _a.getLocals().get(index.getValue());
+                if (local) {
+                    fieldRef.setIndex(local);
+                }
+            }
+            stmt.replaceUse(use, fieldRef);
+            stmt.setRightOp(stmt.getRightOp());
+        }
+    }
+    static parseArkExport2Type(arkExport) {
         if (!arkExport) {
             return null;
         }
         if (arkExport instanceof ArkClass_1.ArkClass) {
-            return new Type_1.ClassType(arkExport.getSignature(), (_a = arkExport.getGenericsTypes()) === null || _a === void 0 ? void 0 : _a.map((x, index) => {
-                let defaultType = x.getDefaultType();
-                if (defaultType instanceof Type_1.UnclearReferenceType) {
-                    const newType = TypeInference.inferUnclearReferenceType(defaultType.getName(), arkExport);
-                    if (newType) {
-                        defaultType = newType;
-                    }
-                }
-                return defaultType !== null && defaultType !== void 0 ? defaultType : Type_1.UndefinedType.getInstance();
-            }));
+            return new Type_1.ClassType(arkExport.getSignature(), arkExport.getGenericsTypes());
         }
         else if (arkExport instanceof ArkNamespace_1.ArkNamespace) {
             let namespaceType = new Type_1.AnnotationNamespaceType(arkExport.getName());
@@ -298,28 +310,44 @@ class TypeInference {
             }
         }
         const leftOp = stmt.getLeftOp();
-        if (leftOp instanceof Local_1.Local) {
-            const leftOpType = leftOp.getType();
-            let type = this.inferUnclearedType(leftOpType, arkClass, rightOp.getType());
-            if (type) {
-                leftOp.setType(type);
-            }
-            else if (rightOp instanceof Ref_1.ArkThisRef) {
-                leftOp.setType(rightOp.getType());
-            }
-            else if (this.isUnclearType(leftOpType) && !this.isUnclearType(stmt.getRightOp().getType())) {
-                leftOp.setType(stmt.getRightOp().getType());
+        let type = leftOp.getType();
+        if (this.isUnclearType(type)) {
+            type = this.inferUnclearedType(type, arkClass, rightOp.getType());
+        }
+        if (type instanceof Type_1.UnionType &&
+            !this.isUnclearType(rightOp.getType()) && !(leftOp instanceof Ref_1.ArkArrayRef)) {
+            const cur = type.getTypes().find(t => rightOp.getType().constructor === t.constructor);
+            if (cur) {
+                type.setCurrType(cur);
             }
         }
+        if (this.isUnclearType(type) && !this.isUnclearType(rightOp.getType())) {
+            type = rightOp.getType();
+        }
+        if (type && leftOp instanceof Local_1.Local) {
+            leftOp.setType(type);
+        }
+        else if (type && leftOp instanceof Ref_1.AbstractFieldRef) {
+            leftOp.getFieldSignature().setType(type);
+        }
     }
     static isUnclearType(type) {
         if (!type || type instanceof Type_1.UnknownType || type instanceof Type_1.UnclearReferenceType) {
             return true;
         }
         else if (type instanceof Type_1.ClassType
-            && type.getClassSignature().getDeclaringFileSignature().getFileName() === '_UnknownFileName') {
+            && type.getClassSignature().getDeclaringFileSignature().getFileName() === Const_1.UNKNOWN_FILE_NAME) {
             return true;
         }
+        else if (type instanceof Type_1.UnionType) {
+            return !!type.getTypes().find(t => t instanceof Type_1.UnclearReferenceType);
+        }
+        else if (type instanceof Type_1.ArrayType) {
+            return (type.getBaseType() instanceof Type_1.UnclearReferenceType);
+        }
+        else if (type instanceof Type_1.AliasType) {
+            return (type.getOriginalType() instanceof Type_1.UnclearReferenceType);
+        }
         return false;
     }
     static inferSimpleTypeInStmt(stmt) {
@@ -361,33 +389,50 @@ class TypeInference {
                 return new Type_1.UnclearReferenceType(typeStr);
         }
     }
-    static inferValueType(value, arkClass) {
+    static inferValueType(value, arkMethod) {
         if (value instanceof Ref_1.ArkInstanceFieldRef || value instanceof Expr_1.ArkInstanceInvokeExpr) {
-            this.inferValueType(value.getBase(), arkClass);
+            this.inferValueType(value.getBase(), arkMethod);
         }
         if (value instanceof Ref_1.AbstractRef || value instanceof Expr_1.AbstractExpr || value instanceof Local_1.Local) {
-            value.inferType(arkClass);
+            value.inferType(arkMethod);
         }
         return value.getType();
     }
     static inferMethodReturnType(method) {
-        const oldMethodSignature = method.getSignature();
-        const oldMethodSubSignature = oldMethodSignature.getMethodSubSignature();
         if (method.getName() === TSConst_1.CONSTRUCTOR_NAME) {
+            const oldMethodSignature = method.getSignature();
+            const oldMethodSubSignature = oldMethodSignature.getMethodSubSignature();
             const newReturnType = new Type_1.ClassType(method.getDeclaringArkClass().getSignature());
             const newMethodSubSignature = new ArkSignature_1.MethodSubSignature(oldMethodSubSignature.getMethodName(), oldMethodSubSignature.getParameters(), newReturnType, oldMethodSubSignature.isStatic());
-            method.setSignature(new ArkSignature_1.MethodSignature(oldMethodSignature.getDeclaringClassSignature(), newMethodSubSignature));
+            method.setImplementationSignature(new ArkSignature_1.MethodSignature(oldMethodSignature.getDeclaringClassSignature(), newMethodSubSignature));
             return;
         }
-        const returnType = method.getReturnType();
-        let inferType;
-        if (returnType instanceof Type_1.UnclearReferenceType) {
-            inferType = this.inferUnclearReferenceType(returnType.getName(), method.getDeclaringArkClass());
+        let implSignature = method.getImplementationSignature();
+        if (implSignature !== null) {
+            const newSignature = this.inferSignatureReturnType(implSignature, method.getDeclaringArkClass());
+            if (newSignature !== null) {
+                method.setImplementationSignature(newSignature);
+            }
         }
-        if (inferType) {
-            const newMethodSubSignature = new ArkSignature_1.MethodSubSignature(oldMethodSubSignature.getMethodName(), oldMethodSubSignature.getParameters(), inferType, oldMethodSubSignature.isStatic());
-            method.setSignature(new ArkSignature_1.MethodSignature(oldMethodSignature.getDeclaringClassSignature(), newMethodSubSignature));
+        let declareSignatures = method.getDeclareSignatures();
+        declareSignatures === null || declareSignatures === void 0 ? void 0 : declareSignatures.forEach((signature, index) => {
+            const newSignature = this.inferSignatureReturnType(signature, method.getDeclaringArkClass());
+            if (newSignature !== null) {
+                method.setDeclareSignatureWithIndex(newSignature, index);
+            }
+        });
+    }
+    static inferSignatureReturnType(oldSignature, declaringClass) {
+        const currReturnType = oldSignature.getType();
+        if (currReturnType instanceof Type_1.UnclearReferenceType) {
+            const newReturnType = this.inferUnclearReferenceType(currReturnType.getName(), declaringClass);
+            if (newReturnType !== null) {
+                const oldSubSignature = oldSignature.getMethodSubSignature();
+                const newMethodSubSignature = new ArkSignature_1.MethodSubSignature(oldSubSignature.getMethodName(), oldSubSignature.getParameters(), newReturnType, oldSubSignature.isStatic());
+                return new ArkSignature_1.MethodSignature(oldSignature.getDeclaringClassSignature(), newMethodSubSignature);
+            }
         }
+        return null;
     }
     static inferGenericType(types, arkClass) {
         types === null || types === void 0 ? void 0 : types.forEach(type => {
@@ -407,6 +452,27 @@ class TypeInference {
             }
         });
     }
+    static inferUnclearRefType(urType, arkClass) {
+        var _a;
+        const realTypes = urType.getGenericTypes();
+        this.inferRealGenericTypes(realTypes, arkClass);
+        if (urType.getName() === Builtin_1.Builtin.ARRAY) {
+            return new Type_1.ArrayType((_a = realTypes[0]) !== null && _a !== void 0 ? _a : Type_1.AnyType, 1);
+        }
+        const type = this.inferUnclearReferenceType(urType.getName(), arkClass);
+        if (realTypes.length === 0) {
+            return type;
+        }
+        if (type instanceof Type_1.ClassType) {
+            return new Type_1.ClassType(type.getClassSignature(), realTypes);
+        }
+        else if (type instanceof Type_1.FunctionType) {
+            return new Type_1.FunctionType(type.getMethodSignature(), realTypes);
+        }
+        else {
+            return new Type_1.UnclearReferenceType(urType.getName(), realTypes);
+        }
+    }
     static inferUnclearReferenceType(refName, arkClass) {
         var _a;
         if (!refName) {
@@ -465,6 +531,10 @@ class TypeInference {
                 }
                 return propertyAndType;
             }
+            if (arkClass.isAnonymousClass()) {
+                const fieldType = this.inferUnclearReferenceType(fieldName, arkClass);
+                return fieldType ? [null, fieldType] : null;
+            }
             const property = ModelUtils_1.ModelUtils.findPropertyInClass(fieldName, arkClass);
             let propertyType = null;
             if (property instanceof ArkField_1.ArkField) {
@@ -521,5 +591,66 @@ class TypeInference {
             }
         }
     }
+    static inferAnonymousClass(anon, declaredSignature, set = new Set()) {
+        var _a;
+        if (!anon) {
+            return;
+        }
+        const key = anon.getSignature().toString();
+        if (set.has(key)) {
+            return;
+        }
+        else {
+            set.add(key);
+        }
+        const scene = anon.getDeclaringArkFile().getScene();
+        const declaredClass = scene.getClass(declaredSignature);
+        if (!declaredClass) {
+            return;
+        }
+        for (const anonField of anon.getFields()) {
+            const property = ModelUtils_1.ModelUtils.findPropertyInClass(anon.getName(), declaredClass);
+            if (property instanceof ArkField_1.ArkField) {
+                TypeInference.assignAnonField(property, anonField, scene, set);
+            }
+        }
+        for (const anonMethod of anon.getMethods()) {
+            const methodSignature = (_a = declaredClass.getMethodWithName(anonMethod.getName())) === null || _a === void 0 ? void 0 : _a.matchMethodSignature(anonMethod.getSubSignature().getParameterTypes());
+            if (methodSignature) {
+                anonMethod.setImplementationSignature(methodSignature);
+            }
+        }
+    }
+    static assignAnonField(property, anonField, scene, set) {
+        function deepInfer(anonType, declaredSignature) {
+            if (anonType instanceof Type_1.ClassType && anonType.getClassSignature().getClassName().startsWith(Const_1.ANONYMOUS_CLASS_PREFIX)) {
+                TypeInference.inferAnonymousClass(scene.getClass(anonType.getClassSignature()), declaredSignature, set);
+            }
+        }
+        const type = property.getSignature().getType();
+        const lastStmt = anonField.getInitializer().at(-1);
+        if (lastStmt instanceof Stmt_1.ArkAssignStmt) {
+            const rightType = lastStmt.getRightOp().getType();
+            if (type instanceof Type_1.ClassType) {
+                deepInfer(rightType, type.getClassSignature());
+            }
+            else if (type instanceof Type_1.ArrayType && type.getBaseType() instanceof Type_1.ClassType &&
+                rightType instanceof Type_1.ArrayType) {
+                const baseType = rightType.getBaseType();
+                const classSignature = type.getBaseType().getClassSignature();
+                if (baseType instanceof Type_1.UnionType) {
+                    baseType.getTypes().forEach(t => deepInfer(t, classSignature));
+                }
+                else {
+                    deepInfer(rightType, classSignature);
+                }
+            }
+            const leftOp = lastStmt.getLeftOp();
+            if (leftOp instanceof Ref_1.AbstractFieldRef) {
+                leftOp.setFieldSignature(property.getSignature());
+            }
+        }
+        anonField.setSignature(property.getSignature());
+    }
 }
 exports.TypeInference = TypeInference;

package/lib/core/common/ValueUtil.d.ts

@@ -2,10 +2,6 @@ import { Constant } from '../base/Constant';
 export declare const EMPTY_STRING = "";
 export declare class ValueUtil {
     private static readonly NumberConstantCache;
-    private static readonly UndefinedConstant;
-    private static readonly NullConstant;
-    private static readonly TrueConstant;
-    private static readonly FalseConstant;
     private static readonly EMPTY_STRING_CONSTANT;
     static getOrCreateNumberConst(n: number): Constant;
     static createStringConst(str: string): Constant;

package/lib/core/common/ValueUtil.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ValueUtil.d.ts","sourceRoot":"","sources":["../../../src/core/common/ValueUtil.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAG5C,eAAO,MAAM,YAAY,KAAK,CAAC;AAE/B,qBAAa,SAAS;IAClB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAYxC;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAA0D;IACnG,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAgD;IACpF,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAmD;IACvF,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAoD;IACzF,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAwD;WAEvF,sBAAsB,CAAC,CAAC,EAAE,MAAM,GAAG,QAAQ;WAS3C,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,QAAQ;WAOxC,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,QAAQ;WAQlC,iBAAiB,IAAI,QAAQ;WAI7B,eAAe,IAAI,QAAQ;WAI3B,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,QAAQ;CAG7D"}
+{"version":3,"file":"ValueUtil.d.ts","sourceRoot":"","sources":["../../../src/core/common/ValueUtil.ts"],"names":[],"mappings":"AAeA,OAAO,EAEH,QAAQ,EAKX,MAAM,kBAAkB,CAAC;AAE1B,eAAO,MAAM,YAAY,KAAK,CAAC;AAE/B,qBAAa,SAAS;IAClB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAoC;IAC/E,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAoC;WAEnE,sBAAsB,CAAC,CAAC,EAAE,MAAM,GAAG,QAAQ;WAS3C,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,QAAQ;WAOxC,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,QAAQ;WAQlC,iBAAiB,IAAI,QAAQ;WAI7B,eAAe,IAAI,QAAQ;WAI3B,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,QAAQ;CAG7D"}

package/lib/core/common/ValueUtil.js

@@ -16,13 +16,12 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ValueUtil = exports.EMPTY_STRING = void 0;
 const Constant_1 = require("../base/Constant");
-const Type_1 = require("../base/Type");
 exports.EMPTY_STRING = '';
 class ValueUtil {
     static getOrCreateNumberConst(n) {
         let constant = this.NumberConstantCache.get(n);
         if (constant === undefined) {
-            constant = new Constant_1.Constant(n.toString(), Type_1.NumberType.getInstance());
+            constant = new Constant_1.NumberConstant(n);
             this.NumberConstantCache.set(n, constant);
         }
         return constant;
@@ -31,41 +30,25 @@ class ValueUtil {
         if (str === exports.EMPTY_STRING) {
             return this.EMPTY_STRING_CONSTANT;
         }
-        return new Constant_1.Constant(str, Type_1.StringType.getInstance());
+        return new Constant_1.StringConstant(str);
     }
     static createConst(str) {
         const n = Number(str);
         if (!isNaN(n)) {
             return this.getOrCreateNumberConst(n);
         }
-        return new Constant_1.Constant(str, Type_1.StringType.getInstance());
+        return new Constant_1.StringConstant(str);
     }
     static getUndefinedConst() {
-        return this.UndefinedConstant;
+        return Constant_1.UndefinedConstant.getInstance();
     }
     static getNullConstant() {
-        return this.NullConstant;
+        return Constant_1.NullConstant.getInstance();
     }
     static getBooleanConstant(value) {
-        return value ? this.TrueConstant : this.FalseConstant;
+        return Constant_1.BooleanConstant.getInstance(value);
     }
 }
 exports.ValueUtil = ValueUtil;
-ValueUtil.NumberConstantCache = new Map([
-    [0, new Constant_1.Constant('0', Type_1.NumberType.getInstance())],
-    [1, new Constant_1.Constant('1', Type_1.NumberType.getInstance())],
-    [2, new Constant_1.Constant('2', Type_1.NumberType.getInstance())],
-    [3, new Constant_1.Constant('3', Type_1.NumberType.getInstance())],
-    [4, new Constant_1.Constant('4', Type_1.NumberType.getInstance())],
-    [5, new Constant_1.Constant('5', Type_1.NumberType.getInstance())],
-    [6, new Constant_1.Constant('6', Type_1.NumberType.getInstance())],
-    [7, new Constant_1.Constant('7', Type_1.NumberType.getInstance())],
-    [8, new Constant_1.Constant('8', Type_1.NumberType.getInstance())],
-    [9, new Constant_1.Constant('9', Type_1.NumberType.getInstance())],
-    [10, new Constant_1.Constant('10', Type_1.NumberType.getInstance())],
-]);
-ValueUtil.UndefinedConstant = new Constant_1.Constant('undefined', Type_1.UndefinedType.getInstance());
-ValueUtil.NullConstant = new Constant_1.Constant('null', Type_1.NullType.getInstance());
-ValueUtil.TrueConstant = new Constant_1.Constant('true', Type_1.BooleanType.getInstance());
-ValueUtil.FalseConstant = new Constant_1.Constant('false', Type_1.BooleanType.getInstance());
-ValueUtil.EMPTY_STRING_CONSTANT = new Constant_1.Constant(exports.EMPTY_STRING, Type_1.StringType.getInstance());
+ValueUtil.NumberConstantCache = new Map();
+ValueUtil.EMPTY_STRING_CONSTANT = new Constant_1.StringConstant(exports.EMPTY_STRING);

package/lib/core/dataflow/DataflowSolver.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DataflowSolver.d.ts","sourceRoot":"","sources":["../../../src/core/dataflow/DataflowSolver.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEpC,OAAO,EAAE,aAAa,EAAoC,IAAI,EAAE,MAAM,cAAc,CAAC;AACrF,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAgB,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,kDAAkD,CAAC;AAa1F,KAAK,qBAAqB,CAAC,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC;AAE5C,8BAAsB,cAAc,CAAC,CAAC;IAElC,SAAS,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC;IACtC,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,SAAS,CAAC,WAAW,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACxC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;IACtB,SAAS,CAAC,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjE,SAAS,CAAC,UAAU,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,SAAS,CAAC,WAAW,EAAE,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC;IACvB,SAAS,CAAC,GAAG,EAAG,sBAAsB,CAAC;IACvC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,SAAS,CAAC,UAAU,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAa;gBAEvC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK;IAa9C,KAAK;IAKZ,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,GAAG,OAAO;IASlD,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI,EAAE;IAIzC,SAAS,CAAC,IAAI;IAgBd,SAAS,CAAC,mBAAmB;IAe7B,SAAS,CAAC,mBAAmB,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI;IAgBtD,SAAS,CAAC,cAAc;IAQxB,SAAS,CAAC,mBAAmB,CAAC,QAAQ,EAAE,aAAa,GAAG,GAAG,CAAC,SAAS,CAAC;IAatE,SAAS,CAAC,mBAAmB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAI/C,SAAS,CAAC,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAMlD,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAW9C,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAcrC,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IA0C3C,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAgB7C,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAkC3C,SAAS,CAAC,qBAAqB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI;IAgCpG,SAAS,CAAC,OAAO;IAiBjB,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO;IAc9C,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO;IAIvC,cAAc,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;CAG5C"}
+{"version":3,"file":"DataflowSolver.d.ts","sourceRoot":"","sources":["../../../src/core/dataflow/DataflowSolver.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEpC,OAAO,EAAE,aAAa,EAAoC,IAAI,EAAE,MAAM,cAAc,CAAC;AACrF,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAgB,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,kDAAkD,CAAC;AAa1F,KAAK,qBAAqB,CAAC,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC;AAE5C,8BAAsB,cAAc,CAAC,CAAC;IAElC,SAAS,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC;IACtC,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,SAAS,CAAC,WAAW,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACxC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;IACtB,SAAS,CAAC,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjE,SAAS,CAAC,UAAU,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,SAAS,CAAC,WAAW,EAAE,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC;IACvB,SAAS,CAAC,GAAG,EAAG,sBAAsB,CAAC;IACvC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,SAAS,CAAC,UAAU,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAa;gBAEvC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK;IAa9C,KAAK;IAKZ,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,GAAG,OAAO;IASlD,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI,EAAE;IAIzC,SAAS,CAAC,IAAI;IAcd,SAAS,CAAC,mBAAmB;IAe7B,SAAS,CAAC,mBAAmB,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI;IAgBtD,SAAS,CAAC,cAAc;IAQxB,SAAS,CAAC,mBAAmB,CAAC,QAAQ,EAAE,aAAa,GAAG,GAAG,CAAC,SAAS,CAAC;IAatE,SAAS,CAAC,mBAAmB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAI/C,SAAS,CAAC,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAMlD,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAW9C,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAcrC,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IA0C3C,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAgB7C,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAkC3C,SAAS,CAAC,qBAAqB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI;IAgCpG,SAAS,CAAC,OAAO;IAiBjB,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO;IAc9C,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO;IAIvC,cAAc,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;CAG5C"}

package/lib/core/dataflow/DataflowSolver.js

@@ -58,8 +58,6 @@ class DataflowSolver {
         this.pathEdgeSet.add(edge);
         // build CHA
         let cg = new CallGraph_1.CallGraph(this.scene);
-        // let cgBuilder = new CallGraphBuilder(cg, this.scene)
-        // cgBuilder.buildClassHierarchyCallGraph([this.problem.getEntryMethod().getSignature()])
         this.CHA = new ClassHierarchyAnalysis_1.ClassHierarchyAnalysis(this.scene, cg);
         this.buildStmtMapInClass();
         this.setCfg4AllStmt();
@@ -207,7 +205,7 @@ class DataflowSolver {
         let callEdgePoint = edge.edgeEnd;
         const invokeStmt = callEdgePoint.node;
         let callees;
-        if (this.scene.getFilesMap().get(invokeStmt.getInvokeExpr().getMethodSignature().getDeclaringClassSignature().getDeclaringFileSignature().toString())) {
+        if (this.scene.getFile(invokeStmt.getInvokeExpr().getMethodSignature().getDeclaringClassSignature().getDeclaringFileSignature())) {
             callees = this.getAllCalleeMethods(callEdgePoint.node);
         }
         else {
@@ -289,7 +287,7 @@ class DataflowSolver {
     isCallStatement(stmt) {
         for (const expr of stmt.getExprs()) {
             if (expr instanceof Expr_1.AbstractInvokeExpr) {
-                if (this.scene.getFilesMap().get(expr.getMethodSignature().getDeclaringClassSignature().getDeclaringFileSignature().toString())) {
+                if (this.scene.getFile(expr.getMethodSignature().getDeclaringClassSignature().getDeclaringFileSignature())) {
                     return true;
                 }
                 if (stmt instanceof Stmt_1.ArkInvokeStmt && (0, Util_1.getRecallMethodInParam)(stmt)) {

package/lib/core/dataflow/TiantAnalysis.js

@@ -46,6 +46,7 @@ const Ref_1 = require("../base/Ref");
 const DataflowSolver_1 = require("./DataflowSolver");
 const Expr_1 = require("../base/Expr");
 const Type_1 = require("../base/Type");
+const ArkSignature_1 = require("../model/ArkSignature");
 const fs = __importStar(require("fs"));
 const Util_1 = require("./Util");
 class TiantAnalysisChecker extends DataflowProblem_1.DataflowProblem {
@@ -259,7 +260,7 @@ class TiantAnalysisChecker extends DataflowProblem_1.DataflowProblem {
         const data = fs.readFileSync(path, 'utf-8');
         const objects = JSON.parse(data);
         for (const object of objects) {
-            const file = this.scene.getSdkArkFilesMap().get(object.file);
+            const file = this.scene.getFile(new ArkSignature_1.FileSignature(object.sdkName, object.file));
             if (!file) {
                 console.log("no file: " + object.file);
                 continue;

package/lib/core/dataflow/Util.d.ts

@@ -5,7 +5,7 @@ import { Local } from "../base/Local";
 import { AbstractRef } from "../base/Ref";
 export declare const INTERNAL_PARAMETER_SOURCE: string[];
 export declare const INTERNAL_SINK_METHOD: string[];
-export declare function Json2ArkMethod(str: string, scene: Scene): ArkMethod | null;
+export declare function Json2ArkMethod(sdkName: string, str: string, scene: Scene): ArkMethod | null;
 export declare function getRecallMethodInParam(stmt: ArkInvokeStmt): ArkMethod | null;
 export declare function LocalEqual(local1: Local, local2: Local): boolean;
 export declare function RefEqual(ref1: AbstractRef, ref2: AbstractRef): boolean;

package/lib/core/dataflow/Util.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Util.d.ts","sourceRoot":"","sources":["../../../src/core/dataflow/Util.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE/C,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,WAAW,EAA0C,MAAM,aAAa,CAAC;AAGlF,eAAO,MAAM,yBAAyB,EAAE,MAAM,EAE7C,CAAA;AAED,eAAO,MAAM,oBAAoB,EAAE,MAAM,EAMxC,CAAA;AAID,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG,IAAI,CAoF1E;AAGD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,aAAa,GAAG,SAAS,GAAG,IAAI,CAW5E;AAGD,wBAAgB,UAAU,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,GAAG,OAAO,CAQhE;AAED,wBAAgB,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,GAAG,OAAO,CAOtE"}
+{"version":3,"file":"Util.d.ts","sourceRoot":"","sources":["../../../src/core/dataflow/Util.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE/C,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEpC,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,WAAW,EAA0C,MAAM,aAAa,CAAC;AAGlF,eAAO,MAAM,yBAAyB,EAAE,MAAM,EAE7C,CAAA;AAED,eAAO,MAAM,oBAAoB,EAAE,MAAM,EAMxC,CAAA;AAID,wBAAgB,cAAc,CAAC,OAAO,EAAC,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG,IAAI,CAoF1F;AAGD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,aAAa,GAAG,SAAS,GAAG,IAAI,CAW5E;AAGD,wBAAgB,UAAU,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,GAAG,OAAO,CAQhE;AAED,wBAAgB,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,GAAG,OAAO,CAOtE"}

package/lib/core/dataflow/Util.js

@@ -16,19 +16,20 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.RefEqual = exports.LocalEqual = exports.getRecallMethodInParam = exports.Json2ArkMethod = exports.INTERNAL_SINK_METHOD = exports.INTERNAL_PARAMETER_SOURCE = void 0;
 const Type_1 = require("../base/Type");
+const ArkSignature_1 = require("../model/ArkSignature");
 const Ref_1 = require("../base/Ref");
 exports.INTERNAL_PARAMETER_SOURCE = [
     '@ohos.app.ability.Want.d.ts: Want'
 ];
 exports.INTERNAL_SINK_METHOD = [
-    'console.<@_UnknownProjectName/_UnknownFileName: .log()>',
-    'console.<@_UnknownProjectName/_UnknownFileName: .error()>',
-    'console.<@_UnknownProjectName/_UnknownFileName: .info()>',
-    'console.<@_UnknownProjectName/_UnknownFileName: .warn()>',
-    'console.<@_UnknownProjectName/_UnknownFileName: .assert()>'
+    'console.<@%unk/%unk: .log()>',
+    'console.<@%unk/%unk: .error()>',
+    'console.<@%unk/%unk: .info()>',
+    'console.<@%unk/%unk: .warn()>',
+    'console.<@%unk/%unk: .assert()>'
 ];
 const filenamePrefix = '@etsSdk/api/';
-function Json2ArkMethod(str, scene) {
+function Json2ArkMethod(sdkName, str, scene) {
     const mes = str.split(': ');
     const fileName = filenamePrefix + mes[0] + ': ';
     const otherMes = mes.slice(1).join(': ').split('.');
@@ -45,7 +46,7 @@ function Json2ArkMethod(str, scene) {
         }
         paramNames = otherMes[2].match(/\((.*?)\)/)[1].split(',').map((item) => item.replace(/\s/g, '')).filter((item) => item !== '');
     }
-    const file = scene.getSdkArkFilesMap().get(fileName);
+    const file = scene.getFile(new ArkSignature_1.FileSignature(sdkName, fileName));
     if (!file) {
         return null;
     }