svf-lib 1.0.2452 → 1.0.2454

package/SVF-linux-x86_64/include/SVFIR/ObjTypeInfo.h

@@ -42,6 +42,7 @@ namespace SVF
 class ObjTypeInfo
 {
     friend class SymbolTableBuilder;
+    friend class GraphDBClient;
 
 public:
     typedef enum
@@ -148,6 +149,10 @@ public:
     {
         flags |= mask;
     }
+    inline u32_t getFlag() const
+    {
+        return flags;
+    }
     inline bool hasFlag(MEMTYPE mask)
     {
         return (flags & mask) == mask;

package/SVF-linux-x86_64/include/SVFIR/SVFIR.h

@@ -46,6 +46,8 @@ class SVFIR : public IRGraph
     friend class PAGBuilderFromFile;
     friend class TypeBasedHeapCloning;
     friend class BVDataPTAImpl;
+    friend class GraphDBClient;
+    friend class GraphDBSVFIRBuilder;
 
 public:
     typedef Set<const CallICFGNode*> CallSiteSet;
@@ -181,6 +183,11 @@ public:
         return callGraph;
     }
 
+    inline void setCallGraph(CallGraph* cg)
+    {
+        callGraph = cg;
+    }
+
     const FunObjVar* getFunObjVar(const std::string& name);
 
     inline const std::string& getModuleIdentifier() const
@@ -500,6 +507,22 @@ public:
     /// Print SVFIR
     void print();
 
+protected:
+    inline NodeID addBaseObjNode(BaseObjVar* node)
+    {
+        memToFieldsMap[node->getId()].set(node->getId());
+        return addObjNode(node);
+    }
+
+    NodeID addDummyObjNode(DummyObjVar* node);
+
+    NodeID addGepObjNode(GepObjVar* gepObj, NodeID base, const APOffset& apOffset);
+
+    inline void addGepValObjFromDB(NodeID curInstID, const GepValVar* gepValvar)
+    {
+        GepValObjMap[curInstID][std::make_pair(gepValvar->getBaseNode()->getId(), gepValvar->getAccessPath())] = gepValvar->getId();
+    }
+
 private:
 
     /// Map a SVFStatement type to a set of corresponding SVF statements
@@ -521,6 +544,11 @@ private:
     inline void addFunArgs(const FunObjVar* fun, const SVFVar* arg)
     {
         FunEntryICFGNode* funEntryBlockNode = icfg->getFunEntryICFGNode(fun);
+        addFunArgs(funEntryBlockNode, fun, arg);
+    }
+
+    inline void addFunArgs(FunEntryICFGNode* funEntryBlockNode, const FunObjVar* fun, const SVFVar* arg)
+    {
         funEntryBlockNode->addFormalParms(arg);
         funArgsListMap[fun].push_back(arg);
     }
@@ -528,6 +556,11 @@ private:
     inline void addFunRet(const FunObjVar* fun, const SVFVar* ret)
     {
         FunExitICFGNode* funExitBlockNode = icfg->getFunExitICFGNode(fun);
+        addFunRet(funExitBlockNode, fun, ret);
+    }
+
+    inline void addFunRet(FunExitICFGNode* funExitBlockNode, const FunObjVar* fun, const SVFVar* ret)
+    {
         funExitBlockNode->addFormalRet(ret);
         funRetMap[fun] = ret;
     }
@@ -557,7 +590,7 @@ private:
     /// Add a value (pointer) node
     inline NodeID addValNode(NodeID i, const SVFType* type, const ICFGNode* icfgNode)
     {
-        SVFVar *node = new ValVar(i, type, icfgNode, ValVar::ValNode);
+        ValVar *node = new ValVar(i, type, icfgNode, ValVar::ValNode);
         return addValNode(node);
     }
 
@@ -624,9 +657,8 @@ private:
      */
     inline NodeID addHeapObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         HeapObjVar *heapObj = new HeapObjVar(i, ti, node);
-        return addObjNode(heapObj);
+        return addBaseObjNode(heapObj);
     }
 
     /**
@@ -634,60 +666,52 @@ private:
      */
     inline NodeID addStackObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         StackObjVar *stackObj = new StackObjVar(i, ti, node);
-        return addObjNode(stackObj);
+        return addBaseObjNode(stackObj);
     }
 
     NodeID addFunObjNode(NodeID id,  ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[id].set(id);
         FunObjVar* funObj = new FunObjVar(id, ti, node);
-        return addObjNode(funObj);
+        return addBaseObjNode(funObj);
     }
 
 
     inline NodeID addConstantFPObjNode(NodeID i, ObjTypeInfo* ti, double dval, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         ConstFPObjVar* conObj = new ConstFPObjVar(i, dval, ti, node);
-        return addObjNode(conObj);
+        return addBaseObjNode(conObj);
     }
 
 
     inline NodeID addConstantIntObjNode(NodeID i, ObjTypeInfo* ti, const std::pair<s64_t, u64_t>& intValue, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         ConstIntObjVar* conObj =
             new ConstIntObjVar(i, intValue.first, intValue.second, ti, node);
-        return addObjNode(conObj);
+        return addBaseObjNode(conObj);
     }
 
 
     inline NodeID addConstantNullPtrObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         ConstNullPtrObjVar* conObj = new ConstNullPtrObjVar(i, ti, node);
-        return addObjNode(conObj);
+        return addBaseObjNode(conObj);
     }
 
     inline NodeID addGlobalObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         GlobalObjVar* gObj = new GlobalObjVar(i, ti, node);
-        return addObjNode(gObj);
+        return addBaseObjNode(gObj);
     }
     inline NodeID addConstantAggObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         ConstAggObjVar* conObj = new ConstAggObjVar(i, ti, node);
-        return addObjNode(conObj);
+        return addBaseObjNode(conObj);
     }
     inline NodeID addConstantDataObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         ConstDataObjVar* conObj = new ConstDataObjVar(i, ti, node);
-        return addObjNode(conObj);
+        return addBaseObjNode(conObj);
     }
 
     /// Add a unique return node for a procedure
@@ -710,9 +734,8 @@ private:
     /// Add a field-insensitive node, this method can only invoked by getFIGepObjNode
     NodeID addFIObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
     {
-        memToFieldsMap[i].set(i);
         BaseObjVar* baseObj = new BaseObjVar(i, ti, node);
-        return addObjNode(baseObj);
+        return addBaseObjNode(baseObj);
     }
 
 
@@ -730,11 +753,11 @@ private:
         {
             ObjTypeInfo* ti = createObjTypeInfo(type);
             idToObjTypeInfoMap()[i] = ti;
-            return addObjNode(new DummyObjVar(i, ti, nullptr));
+            return addDummyObjNode(new DummyObjVar(i, ti, nullptr));
         }
         else
         {
-            return addObjNode(new DummyObjVar(i, getObjTypeInfo(i), nullptr));
+            return addDummyObjNode(new DummyObjVar(i, getObjTypeInfo(i), nullptr));
         }
     }
 
@@ -753,23 +776,9 @@ private:
     //@}
 
     /// Add a value (pointer) node
-    inline NodeID addValNode(SVFVar *node)
-    {
-        assert(node && "node cannot be nullptr.");
-        assert(hasGNode(node->getId()) == false &&
-               "This NodeID clashes here. Please check NodeIDAllocator. Switch "
-               "Strategy::DBUG to SEQ or DENSE");
-        return addNode(node);
-    }
+    NodeID addValNode(ValVar* node);
     /// Add a memory obj node
-    inline NodeID addObjNode(SVFVar *node)
-    {
-        assert(node && "node cannot be nullptr.");
-        assert(hasGNode(node->getId()) == false &&
-               "This NodeID clashes here. Please check NodeIDAllocator. Switch "
-               "Strategy::DBUG to SEQ or DENSE");
-        return addNode(node);
-    }
+    NodeID addObjNode(ObjVar *node);
     /// Add a unique return node for a procedure
     inline NodeID addRetNode(const FunObjVar*, SVFVar *node)
     {
@@ -795,36 +804,49 @@ private:
     //@{
     /// Add Address edge
     AddrStmt* addAddrStmt(NodeID src, NodeID dst);
+    void addAddrStmt(AddrStmt* edge);
     /// Add Copy edge
     CopyStmt* addCopyStmt(NodeID src, NodeID dst, CopyStmt::CopyKind type);
+    void addCopyStmt(CopyStmt* edge);
 
     /// Add phi node information
     PhiStmt*  addPhiStmt(NodeID res, NodeID opnd, const ICFGNode* pred);
+    void addPhiStmt(PhiStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add SelectStmt
     SelectStmt*  addSelectStmt(NodeID res, NodeID op1, NodeID op2, NodeID cond);
+    void addSelectStmt(SelectStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add Copy edge
     CmpStmt* addCmpStmt(NodeID op1, NodeID op2, NodeID dst, u32_t predict);
+    void addCmpStmt(CmpStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add Copy edge
     BinaryOPStmt* addBinaryOPStmt(NodeID op1, NodeID op2, NodeID dst,
                                   u32_t opcode);
+    void addBinaryOPStmt(BinaryOPStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add Unary edge
     UnaryOPStmt* addUnaryOPStmt(NodeID src, NodeID dst, u32_t opcode);
+    void addUnaryOPStmt(UnaryOPStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add BranchStmt
     BranchStmt* addBranchStmt(NodeID br, NodeID cond,
                               const BranchStmt::SuccAndCondPairVec& succs);
+    void addBranchStmt(BranchStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add Load edge
     LoadStmt* addLoadStmt(NodeID src, NodeID dst);
+    void addLoadStmt(LoadStmt* edge);
     /// Add Store edge
     StoreStmt* addStoreStmt(NodeID src, NodeID dst, const ICFGNode* val);
+    void addStoreStmt(StoreStmt* edge, SVFVar* src, SVFVar* dst);
     /// Add Call edge
     CallPE* addCallPE(NodeID src, NodeID dst, const CallICFGNode* cs,
                       const FunEntryICFGNode* entry);
+    void addCallPE(CallPE* edge, SVFVar* src, SVFVar* dst);
     /// Add Return edge
     RetPE* addRetPE(NodeID src, NodeID dst, const CallICFGNode* cs,
                     const FunExitICFGNode* exit);
+    void addRetPE(RetPE* edge, SVFVar* src, SVFVar* dst);
     /// Add Gep edge
     GepStmt* addGepStmt(NodeID src, NodeID dst, const AccessPath& ap,
                         bool constGep);
+    void addGepStmt(GepStmt* edge);
     /// Add Offset(Gep) edge
     GepStmt* addNormalGepStmt(NodeID src, NodeID dst, const AccessPath& ap);
     /// Add Variant(Gep) edge

package/SVF-linux-x86_64/include/SVFIR/SVFStatements.h

@@ -51,6 +51,7 @@ class SVFBasicBlock;
 typedef GenericEdge<SVFVar> GenericPAGEdgeTy;
 class SVFStmt : public GenericPAGEdgeTy
 {
+    friend class GraphDBClient;
 
 public:
     /// Types of SVFIR statements
@@ -89,8 +90,54 @@ protected:
     {
     }
 
+    SVFStmt(SVFVar* s, SVFVar* d, GEdgeFlag k, EdgeID eid, SVFVar* value, ICFGNode* icfgNode, bool real = true);
+
+    /**
+     * Set the SVF BasicBlock for the new statements, this is used when loading statements from DB
+     */
+    inline void setBasicBlock(const SVFBasicBlock* bb)
+    {
+        basicBlock = bb;
+    }
+
+    /**
+     * set the call edge lanbel counter for the new statements, this is used when loading statements from DB
+     */
+    inline void setCallEdgeLabelCounter(u64_t counter)
+    {
+        callEdgeLabelCounter = counter;
+    }
+
+    /**
+     * set the store edge lanbel counter for the new statements, this is used when loading statements from DB
+     */
+    inline void setStoreEdgeLabelCounter(u64_t counter)
+    {
+        storeEdgeLabelCounter = counter;
+    }
+
+    /**
+     * set the multi operand edge lanbel counter for the new statements, this is used when loading statements from DB
+     */
+    inline void setMultiOpndLabelCounter(u64_t counter)
+    {
+        multiOpndLabelCounter = counter;
+    }
+
+    /**
+     * Add a call site Instruction to label mapping, this is used when loading statements from DB
+     */
+    static inline void addInst2Labeled(const ICFGNode* cs, u32_t label)
+    {
+        inst2LabelMap.emplace(cs, label);
+    }
+
+    static inline void addVar2Labeled(const SVFVar* var, u32_t label)
+    {
+        var2LabelMap.emplace(var, label);
+    }
+
 public:
-    static u32_t totalEdgeNum; ///< Total edge number
 
     /// Constructor
     SVFStmt(SVFVar* s, SVFVar* d, GEdgeFlag k, bool real = true);
@@ -222,6 +269,33 @@ private:
     static u64_t callEdgeLabelCounter;  ///< Call site Instruction counter
     static u64_t storeEdgeLabelCounter;  ///< Store Instruction counter
     static u64_t multiOpndLabelCounter;  ///< MultiOpndStmt counter
+
+public:
+    static inline const Inst2LabelMap* getInst2LabelMap()
+    {
+        return &inst2LabelMap;
+    }
+
+    static inline const Var2LabelMap* getVar2LabelMap()
+    {
+        return &var2LabelMap;
+    }
+
+    static inline const u64_t* getCallEdgeLabelCounter()
+    {
+        return &callEdgeLabelCounter;
+    }
+
+    static inline const u64_t* getStoreEdgeLabelCounter()
+    {
+        return &storeEdgeLabelCounter;
+    }
+
+    static inline const u64_t* getMultiOpndLabelCounter()
+    {
+        return &multiOpndLabelCounter;
+    }
+
 };
 
 /*
@@ -231,6 +305,7 @@ private:
 */
 class AssignStmt : public SVFStmt
 {
+    friend class GraphDBClient;
 
 private:
     AssignStmt();                      ///< place holder
@@ -303,6 +378,7 @@ public:
  */
 class AddrStmt: public AssignStmt
 {
+    friend class GraphDBClient;
 
 private:
     AddrStmt(const AddrStmt&);       ///< place holder
@@ -355,6 +431,8 @@ public:
 class CopyStmt: public AssignStmt
 {
 
+    friend class GraphDBClient;
+
 private:
     CopyStmt(const CopyStmt&);       ///< place holder
     void operator=(const CopyStmt&); ///< place holder
@@ -430,6 +508,7 @@ public:
     CopyStmt(SVFVar* s, SVFVar* d, CopyKind k) : AssignStmt(s, d, SVFStmt::Copy), copyKind(k) {}
 
     virtual const std::string toString() const override;
+
 private:
     u32_t copyKind;
 };
@@ -439,6 +518,7 @@ private:
  */
 class StoreStmt: public AssignStmt
 {
+    friend class GraphDBClient;
 
 private:
     StoreStmt(const StoreStmt&);      ///< place holder
@@ -465,6 +545,7 @@ public:
     StoreStmt(SVFVar* s, SVFVar* d, const ICFGNode* st);
 
     virtual const std::string toString() const override;
+
 };
 
 /*!
@@ -472,6 +553,7 @@ public:
  */
 class LoadStmt: public AssignStmt
 {
+    friend class GraphDBClient;
 
 private:
     LoadStmt(const LoadStmt&);       ///< place holder
@@ -505,6 +587,8 @@ public:
  */
 class GepStmt: public AssignStmt
 {
+    friend class GraphDBClient;
+
 
 private:
     GepStmt(const GepStmt &);  ///< place holder
@@ -578,6 +662,7 @@ public:
 
     virtual const std::string toString() const;
 
+
 };
 
 
@@ -586,6 +671,7 @@ public:
  */
 class CallPE: public AssignStmt
 {
+    friend class GraphDBClient;
 
 private:
     CallPE(const CallPE&);         ///< place holder
@@ -641,6 +727,7 @@ public:
  */
 class RetPE: public AssignStmt
 {
+    friend class GraphDBClient;
 
 private:
     RetPE(const RetPE&);          ///< place holder
@@ -689,6 +776,7 @@ public:
     //@}
 
     virtual const std::string toString() const override;
+
 };
 
 /*
@@ -696,6 +784,9 @@ public:
 */
 class MultiOpndStmt : public SVFStmt
 {
+    friend class GraphDBClient;
+
+
 
 public:
     typedef std::vector<SVFVar*> OPVars;
@@ -774,7 +865,7 @@ public:
  */
 class PhiStmt: public MultiOpndStmt
 {
-
+    friend class GraphDBClient;
 public:
     typedef std::vector<const ICFGNode*> OpICFGNodeVec;
 
@@ -806,8 +897,8 @@ public:
     //@}
 
     /// constructor
-    PhiStmt(SVFVar* s, const OPVars& opnds, const OpICFGNodeVec& icfgNodes)
-        : MultiOpndStmt(s, opnds, SVFStmt::Phi), opICFGNodes(icfgNodes)
+    PhiStmt(SVFVar* res, const OPVars& opnds, const OpICFGNodeVec& icfgNodes)
+        : MultiOpndStmt(res, opnds, SVFStmt::Phi), opICFGNodes(icfgNodes)
     {
         assert(opnds.size() == icfgNodes.size() &&
                "Numbers of operands and their ICFGNodes are not consistent?");
@@ -820,6 +911,18 @@ public:
                "Numbers of operands and their ICFGNodes are not consistent?");
     }
 
+    void setOpICFGNodeVec(OpICFGNodeVec& icfgNodes)
+    {
+        assert(opVars.size() == icfgNodes.size() &&
+               "Numbers of operands and their ICFGNodes are not consistent?");
+        opICFGNodes = icfgNodes;
+    }
+
+    inline const OpICFGNodeVec* getOpICFGNodeVec() const
+    {
+        return &opICFGNodes;
+    }
+
     /// Return the corresponding ICFGNode of this operand
     inline const ICFGNode* getOpICFGNode(u32_t op_idx) const
     {
@@ -831,6 +934,7 @@ public:
     bool isFunctionRetPhi() const;
 
     virtual const std::string toString() const override;
+
 };
 
 /*!
@@ -838,7 +942,7 @@ public:
  */
 class SelectStmt: public MultiOpndStmt
 {
-
+    friend class GraphDBClient;
 private:
     SelectStmt(const SelectStmt&);     ///< place holder
     void operator=(const SelectStmt&); ///< place holder
@@ -867,7 +971,7 @@ public:
     //@}
 
     /// constructor
-    SelectStmt(SVFVar* s, const OPVars& opnds, const SVFVar* cond);
+    SelectStmt(SVFVar* res, const OPVars& opnds, const SVFVar* cond);
     virtual const std::string toString() const override;
 
     inline const SVFVar* getCondition() const
@@ -882,6 +986,7 @@ public:
     {
         return getOpVar(1);
     }
+
 };
 
 /*!
@@ -889,7 +994,7 @@ public:
  */
 class CmpStmt: public MultiOpndStmt
 {
-
+    friend class GraphDBClient;
 private:
     CmpStmt(const CmpStmt&);        ///< place holder
     void operator=(const CmpStmt&); ///< place holder
@@ -956,7 +1061,7 @@ public:
     //@}
 
     /// constructor
-    CmpStmt(SVFVar* s, const OPVars& opnds, u32_t pre);
+    CmpStmt(SVFVar* res, const OPVars& opnds, u32_t pre);
 
     u32_t getPredicate() const
     {
@@ -964,6 +1069,7 @@ public:
     }
 
     virtual const std::string toString() const override;
+
 };
 
 /*!
@@ -971,7 +1077,7 @@ public:
  */
 class BinaryOPStmt: public MultiOpndStmt
 {
-
+    friend class GraphDBClient;
 private:
     BinaryOPStmt(const BinaryOPStmt&);   ///< place holder
     void operator=(const BinaryOPStmt&); ///< place holder
@@ -1022,7 +1128,7 @@ public:
     //@}
 
     /// constructor
-    BinaryOPStmt(SVFVar* s, const OPVars& opnds, u32_t oc);
+    BinaryOPStmt(SVFVar* res, const OPVars& opnds, u32_t oc);
 
     u32_t getOpcode() const
     {
@@ -1030,6 +1136,7 @@ public:
     }
 
     virtual const std::string toString() const override;
+
 };
 
 /*!
@@ -1037,6 +1144,7 @@ public:
  */
 class UnaryOPStmt: public SVFStmt
 {
+    friend class GraphDBClient;
 
 private:
     UnaryOPStmt(const UnaryOPStmt&);    ///< place holder
@@ -1093,6 +1201,7 @@ public:
     NodeID getResID() const;
 
     virtual const std::string toString() const override;
+
 };
 
 /*!
@@ -1100,6 +1209,7 @@ public:
  */
 class BranchStmt: public SVFStmt
 {
+    friend class GraphDBClient;
 
 public:
     typedef std::vector<std::pair<const ICFGNode*, s32_t>> SuccAndCondPairVec;
@@ -1179,6 +1289,7 @@ public:
     }
     //@}
     virtual const std::string toString() const override;
+
 };
 
 /*!
@@ -1186,6 +1297,7 @@ public:
  */
 class TDForkPE: public CallPE
 {
+    friend class GraphDBClient;
 
 private:
     TDForkPE(const TDForkPE&);       ///< place holder
@@ -1216,6 +1328,7 @@ public:
     }
 
     virtual const std::string toString() const;
+
 };
 
 /*!
@@ -1223,6 +1336,7 @@ public:
  */
 class TDJoinPE: public RetPE
 {
+    friend class GraphDBClient;
 
 private:
     TDJoinPE(const TDJoinPE&);       ///< place holder
@@ -1253,6 +1367,7 @@ public:
     }
 
     virtual const std::string toString() const;
+
 };
 
 } // End namespace SVF