arkanalyzer 1.0.13 → 1.0.15

package/lib/core/dataflow/TiantAnalysis.js

@@ -1,340 +0,0 @@
-"use strict";
-/*
- * Copyright (c) 2024 Huawei Device Co., Ltd.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TiantAnalysisSolver = exports.TiantAnalysisChecker = void 0;
-const DataflowProblem_1 = require("./DataflowProblem");
-const Local_1 = require("../base/Local");
-const Stmt_1 = require("../base/Stmt");
-const Constant_1 = require("../base/Constant");
-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 {
-    constructor(stmt, method) {
-        super();
-        this.zeroValue = new Constant_1.Constant('zeroValue', Type_1.UndefinedType.getInstance());
-        this.classMap = new Map();
-        this.globalVariableMap = new Map();
-        this.sources = [];
-        this.sinks = [];
-        this.entryPoint = stmt;
-        this.entryMethod = method;
-        this.scene = method.getDeclaringArkFile().getScene();
-        this.globalVariableMap = this.scene.getGlobalVariableMap();
-    }
-    getEntryPoint() {
-        return this.entryPoint;
-    }
-    getEntryMethod() {
-        return this.entryMethod;
-    }
-    callSource(val) {
-        if (val instanceof Expr_1.AbstractInvokeExpr) {
-            for (const source of this.sources) {
-                if (source.getSignature() === val.getMethodSignature()) {
-                    return true;
-                }
-            }
-        }
-        return false;
-    }
-    setSources(methods) {
-        this.sources = methods;
-    }
-    setSinks(methods) {
-        this.sinks = methods;
-    }
-    getNormalFlowFunction(srcStmt, tgtStmt) {
-        let checkerInstance = this;
-        return new class {
-            getDataFacts(dataFact) {
-                let ret = new Set();
-                if (checkerInstance.getEntryPoint() === srcStmt && checkerInstance.getZeroValue() === dataFact) {
-                    let entryMethod = checkerInstance.getEntryMethod();
-                    const parameters = [...entryMethod.getCfg().getBlocks()][0].getStmts().slice(0, entryMethod.getParameters().length);
-                    for (let i = 0; i < parameters.length; i++) {
-                        const para = parameters[i].getDef();
-                        if (para)
-                            ret.add(para);
-                    }
-                    ret.add(checkerInstance.getZeroValue());
-                    // 加入所有的全局变量和静态属性（may analysis）
-                    const staticFields = entryMethod.getDeclaringArkClass().getStaticFields(checkerInstance.classMap);
-                    for (const field of staticFields) {
-                        if (field.getInitializer() === undefined) {
-                            ret.add(new Ref_1.ArkStaticFieldRef(field.getSignature()));
-                        }
-                    }
-                    for (const local of entryMethod.getDeclaringArkClass().getGlobalVariable(checkerInstance.globalVariableMap)) {
-                        ret.add(local);
-                    }
-                    return ret;
-                }
-                if (!checkerInstance.factEqual(srcStmt.getDef(), dataFact)) {
-                    if (!(dataFact instanceof Local_1.Local && dataFact.getName() === srcStmt.getDef().toString()))
-                        ret.add(dataFact);
-                }
-                if (srcStmt instanceof Stmt_1.ArkAssignStmt) {
-                    let stmt = srcStmt;
-                    let assigned = stmt.getLeftOp();
-                    let rightOp = stmt.getRightOp();
-                    if (checkerInstance.getZeroValue() === dataFact) {
-                        if (checkerInstance.callSource(rightOp)) {
-                            ret.add(assigned);
-                        }
-                    }
-                    else if (checkerInstance.factEqual(rightOp, dataFact) || rightOp.getType() instanceof Type_1.UndefinedType) {
-                        ret.add(assigned);
-                        if (assigned instanceof Ref_1.ArkInstanceFieldRef) {
-                        }
-                    }
-                    else if (dataFact instanceof Ref_1.ArkInstanceFieldRef && rightOp === dataFact.getBase()) {
-                        const field = new Ref_1.ArkInstanceFieldRef(srcStmt.getLeftOp(), dataFact.getFieldSignature());
-                        ret.add(field);
-                    }
-                }
-                return ret;
-            }
-        };
-    }
-    getCallFlowFunction(srcStmt, method) {
-        let checkerInstance = this;
-        return new class {
-            getDataFacts(dataFact) {
-                const ret = new Set();
-                if (checkerInstance.getZeroValue() === dataFact) {
-                    ret.add(checkerInstance.getZeroValue());
-                    // 加上调用函数能访问到的所有静态变量，如果不考虑多线程，加上所有变量，考虑则要统计之前已经处理过的变量并排除
-                    for (const field of method.getDeclaringArkClass().getStaticFields(checkerInstance.classMap)) {
-                        if (field.getInitializer() === undefined) {
-                            ret.add(new Ref_1.ArkStaticFieldRef(field.getSignature()));
-                        }
-                    }
-                    for (const local of method.getDeclaringArkClass().getGlobalVariable(checkerInstance.globalVariableMap)) {
-                        ret.add(local);
-                    }
-                }
-                else {
-                    const callExpr = srcStmt.getExprs()[0];
-                    if (callExpr instanceof Expr_1.ArkInstanceInvokeExpr && dataFact instanceof Ref_1.ArkInstanceFieldRef && callExpr.getBase().getName() === dataFact.getBase().getName()) {
-                        // todo:base转this
-                        const _this = [...srcStmt.getCfg().getBlocks()][0].getStmts()[0].getDef();
-                        const thisRef = new Ref_1.ArkInstanceFieldRef(_this, dataFact.getFieldSignature());
-                        ret.add(thisRef);
-                    }
-                    else if (callExpr instanceof Expr_1.ArkStaticInvokeExpr && dataFact instanceof Ref_1.ArkStaticFieldRef && callExpr.getMethodSignature().getDeclaringClassSignature() === dataFact.getFieldSignature().getDeclaringSignature()) {
-                        ret.add(dataFact);
-                    }
-                    for (const sink of checkerInstance.sinks) {
-                        if (callExpr.getMethodSignature() === sink.getSignature()) {
-                            for (const param of callExpr.getArgs()) {
-                                if (checkerInstance.factEqual(param, dataFact)) {
-                                    console.log("source: " + dataFact);
-                                    console.log("sink: " + srcStmt.getOriginPositionInfo().toString() + ", " + srcStmt.toString());
-                                }
-                            }
-                        }
-                    }
-                }
-                const callStmt = srcStmt;
-                const args = callStmt.getInvokeExpr().getArgs();
-                for (let i = 0; i < args.length; i++) {
-                    if (args[i] === dataFact || checkerInstance.callSource(args[i]) && checkerInstance.getZeroValue() === dataFact) {
-                        const realParameter = [...method.getCfg().getBlocks()][0].getStmts()[i].getDef();
-                        if (realParameter)
-                            ret.add(realParameter);
-                    }
-                    else if (dataFact instanceof Ref_1.ArkInstanceFieldRef && dataFact.getBase().getName() === args[i].toString()) {
-                        const realParameter = [...method.getCfg().getBlocks()][0].getStmts()[i].getDef();
-                        if (realParameter) {
-                            const retRef = new Ref_1.ArkInstanceFieldRef(realParameter, dataFact.getFieldSignature());
-                            ret.add(retRef);
-                        }
-                    }
-                }
-                return ret;
-            }
-        };
-    }
-    getExitToReturnFlowFunction(srcStmt, tgtStmt, callStmt) {
-        let checkerInstance = this;
-        return new class {
-            getDataFacts(dataFact) {
-                let ret = new Set();
-                if (dataFact === checkerInstance.getZeroValue()) {
-                    ret.add(checkerInstance.getZeroValue());
-                }
-                if (dataFact instanceof Ref_1.ArkInstanceFieldRef && dataFact.getBase().getName() === "this") {
-                    // todo:this转base。
-                    const expr = callStmt.getExprs()[0];
-                    if (expr instanceof Expr_1.ArkInstanceInvokeExpr) {
-                        const fieldRef = new Ref_1.ArkInstanceFieldRef(expr.getBase(), dataFact.getFieldSignature());
-                        ret.add(fieldRef);
-                    }
-                }
-                if (!(callStmt instanceof Stmt_1.ArkAssignStmt)) {
-                    return ret;
-                }
-                if (srcStmt instanceof Stmt_1.ArkReturnStmt) {
-                    let ass = callStmt;
-                    let leftOp = ass.getLeftOp();
-                    let retVal = srcStmt.getOp();
-                    if (dataFact === checkerInstance.getZeroValue()) {
-                        ret.add(checkerInstance.getZeroValue());
-                        if (checkerInstance.callSource(retVal) || checkerInstance.callSource(ass.getRightOp())) {
-                            ret.add(leftOp);
-                        }
-                    }
-                    else if (retVal === dataFact) {
-                        ret.add(leftOp);
-                    }
-                }
-                return ret;
-            }
-        };
-    }
-    getCallToReturnFlowFunction(srcStmt, tgtStmt) {
-        let checkerInstance = this;
-        return new class {
-            getDataFacts(dataFact) {
-                const ret = new Set();
-                if (checkerInstance.getZeroValue() === dataFact) {
-                    ret.add(checkerInstance.getZeroValue());
-                }
-                const defValue = srcStmt.getDef();
-                if (!(defValue && defValue === dataFact)) {
-                    ret.add(dataFact);
-                }
-                return ret;
-            }
-        };
-    }
-    createZeroValue() {
-        return this.zeroValue;
-    }
-    getZeroValue() {
-        return this.zeroValue;
-    }
-    Json2ArkMethod(path) {
-        let arkMethods = [];
-        const data = fs.readFileSync(path, 'utf-8');
-        const objects = JSON.parse(data);
-        for (const object of objects) {
-            const file = this.scene.getFile(new ArkSignature_1.FileSignature(object.sdkName, object.file));
-            if (!file) {
-                console.log("no file: " + object.file);
-                continue;
-            }
-            let arkClass = null;
-            if (object.namespace === "_") {
-                for (const clas of file.getClasses()) {
-                    if (clas.getName() === object.class) {
-                        arkClass = clas;
-                        break;
-                    }
-                }
-            }
-            else {
-                let arkNamespace = null;
-                for (const ns of file.getNamespaces()) {
-                    if (ns.getName() === object.namespace) {
-                        arkNamespace = ns;
-                        break;
-                    }
-                }
-                if (arkNamespace) {
-                    for (const clas of arkNamespace.getClasses()) {
-                        if (clas.getName() === object.class) {
-                            arkClass = clas;
-                            break;
-                        }
-                    }
-                }
-                else {
-                    console.log("no namespace: " + object.namespace);
-                    continue;
-                }
-            }
-            if (!arkClass) {
-                console.log("no class: " + object.class);
-                continue;
-            }
-            else {
-                let arkMethod = null;
-                for (const method of arkClass.getMethods()) {
-                    if (method.getName() === object.method) {
-                        arkMethod = method;
-                        break;
-                    }
-                }
-                if (arkMethod) {
-                    arkMethods.push(arkMethod);
-                }
-                else {
-                    console.log("no method: " + object.method);
-                    continue;
-                }
-            }
-        }
-        return arkMethods;
-    }
-    factEqual(d1, d2) {
-        if (d1 instanceof Constant_1.Constant && d2 instanceof Constant_1.Constant) {
-            return d1 === d2;
-        }
-        else if (d1 instanceof Local_1.Local && d2 instanceof Local_1.Local) {
-            return (0, Util_1.LocalEqual)(d1, d2);
-        }
-        else if (d1 instanceof Ref_1.AbstractRef && d2 instanceof Ref_1.AbstractRef) {
-            return (0, Util_1.RefEqual)(d1, d2);
-        }
-        return false;
-    }
-}
-exports.TiantAnalysisChecker = TiantAnalysisChecker;
-class TiantAnalysisSolver extends DataflowSolver_1.DataflowSolver {
-    constructor(problem, scene) {
-        super(problem, scene);
-    }
-}
-exports.TiantAnalysisSolver = TiantAnalysisSolver;