SVFIR.h

Go to the documentation of this file.

//===- SVFIR.h -- IR of SVF ---------------------------------------------//
//
//                     SVF: Static Value-Flow Analysis
//
// Copyright (C) <2013->  <Yulei Sui>
//
 
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
 
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
//===----------------------------------------------------------------------===//
 
/*
 * SVFIR.h
 *
 *  Created on: 31, 12, 2021
 *      Author: Yulei Sui
 */
 
#ifndef INCLUDE_SVFIR_H_
#define INCLUDE_SVFIR_H_
 
#include "Graphs/IRGraph.h"
 
namespace SVF
{
class CommonCHGraph;
class SVFIR : public IRGraph
{
    friend class SVFIRBuilder;
    friend class ExternalPAG;
    friend class PAGBuilderFromFile;
    friend class TypeBasedHeapCloning;
    friend class BVDataPTAImpl;
    friend class GraphDBClient;
    friend class GraphDBSVFIRBuilder;
 
public:
    typedef Set<const CallICFGNode*> CallSiteSet;
    typedef OrderedMap<const CallICFGNode*,NodeID> CallSiteToFunPtrMap;
    typedef Map<NodeID,CallSiteSet> FunPtrToCallSitesMap;
    typedef Map<NodeID,NodeBS> MemObjToFieldsMap;
    typedef std::vector<const SVFStmt*> SVFStmtList;
    typedef std::vector<const ValVar*> ValVarList;
    typedef Map<const SVFVar*,PhiStmt*> PHINodeMap;
    typedef Map<const FunObjVar*,ValVarList> FunToArgsListMap;
    typedef Map<const CallICFGNode*,ValVarList> CSToArgsListMap;
    typedef Map<const RetICFGNode*,const ValVar*> CSToRetMap;
    typedef Map<const FunObjVar*,const ValVar*> FunToRetMap;
    typedef Map<const FunObjVar*,SVFStmtSet> FunToPAGEdgeSetMap;
    typedef Map<const ICFGNode*,SVFStmtList> ICFGNode2SVFStmtsMap;
    typedef Map<NodeID, NodeID> NodeToNodeMap;
    typedef std::pair<NodeID, APOffset> GepOffset;
    typedef std::pair<NodeID, AccessPath> NodeAccessPath;
    typedef Map<GepOffset,NodeID> OffsetToGepVarMap;
    typedef Map<NodeAccessPath,NodeID> NodeAccessPathMap;
    typedef Map<NodeID, NodeAccessPathMap> GepValueVarMap;
    typedef std::pair<const SVFType*, std::vector<AccessPath>> SVFTypeLocSetsPair;
    typedef Map<NodeID, SVFTypeLocSetsPair> TypeLocSetsMap;
    typedef Map<NodePair,NodeID> NodePairSetMap;
 
private:
    ICFGNode2SVFStmtsMap icfgNode2SVFStmtsMap;  
    ICFGNode2SVFStmtsMap icfgNode2PTASVFStmtsMap;   
    GepValueVarMap GepValObjMap;    
    TypeLocSetsMap typeLocSetsMap;  
    OffsetToGepVarMap GepObjVarMap; 
    MemObjToFieldsMap memToFieldsMap;   
    SVFStmtSet globSVFStmtSet;  
    PHINodeMap phiNodeMap;  
    FunToArgsListMap funArgsListMap;    
    CSToArgsListMap callSiteArgsListMap;    
    CSToRetMap callSiteRetMap;  
    FunToRetMap funRetMap;  
    CallSiteToFunPtrMap indCallSiteToFunPtrMap; 
    FunPtrToCallSitesMap funPtrToCallSitesMap;  
    OrderedNodeSet candidatePointers;
    ICFG* icfg; // ICFG
    CommonCHGraph* chgraph; // class hierarchy graph
    CallSiteSet callSiteSet; 
    CallGraph* callGraph; 
 
    static std::unique_ptr<SVFIR> pag;  
    static std::string pagReadFromTxt;
 
    std::string moduleIdentifier;
 
    SVFIR(bool buildFromFile);
 
    void destroy();
 
public:
 
 
    static inline SVFIR* getPAG(bool buildFromFile = false)
    {
        if (pag == nullptr)
        {
            pag = std::unique_ptr<SVFIR>(new SVFIR(buildFromFile));
        }
        return pag.get();
    }
    static void releaseSVFIR()
    {
        pag = nullptr;
    }
 
    inline const SVFVar* getSVFVar(NodeID id) const
    {
        return getGNode(id);
    }
    inline const ValVar* getValVar(NodeID id) const
    {
        if(const SVFVar* var = getSVFVar(id))
            return SVFUtil::dyn_cast<ValVar>(var);
        else
        {
            assert(false && "the Node is not a ValVar");
            return nullptr;
        }
    }
    inline const ObjVar* getObjVar(NodeID id) const
    {
        if(const SVFVar* var = getSVFVar(id))
            return SVFUtil::dyn_cast<ObjVar>(var);
        else
        {
            assert(false && "the Node is not an ObjVar");
            return nullptr;
        }
    }
    inline const BaseObjVar* getBaseObjVar(NodeID id) const
    {
        if(const SVFVar* var = getSVFVar(id))
            return SVFUtil::dyn_cast<BaseObjVar>(var);
        else
        {
            assert(false && "the Node is not a BaseObjVar");
            return nullptr;
        }
    }
    inline const GepObjVar* getGepObjVar(NodeID id) const
    {
        if(const SVFVar* var = getSVFVar(id))
            return SVFUtil::dyn_cast<GepObjVar>(var);
        else
        {
            assert(false && "the Node is not a GepObjVar");
            return nullptr;
        }
    }
    inline bool isObjVar(NodeID id) const
    {
        return SVFUtil::isa<ObjVar>(getSVFVar(id));
    }
    inline bool isBaseObjVar(NodeID id) const
    {
        return SVFUtil::isa<BaseObjVar>(getSVFVar(id));
    }
    inline bool isGepObjVar(NodeID id) const
    {
        return SVFUtil::isa<GepObjVar>(getSVFVar(id));
    }
    inline const IDToNodeMapTy& getSVFVarMap() const
    {
        return IDToNodeMap;
    }
 
    inline MemObjToFieldsMap& getMemToFieldsMap()
    {
        return memToFieldsMap;
    }
    inline OffsetToGepVarMap& getGepObjNodeMap()
    {
        return GepObjVarMap;
    }
    inline OrderedNodeSet& getAllValidPtrs()
    {
        return candidatePointers;
    }
    void initialiseCandidatePointers();
 
    virtual ~SVFIR()
    {
        destroy();
    }
 
 
    static void handleBlackHole(bool b);
 
    inline void setICFG(ICFG* i)
    {
        icfg = i;
    }
    inline ICFG* getICFG() const
    {
        assert(icfg->totalICFGNode>0 && "empty ICFG! Build SVF IR first!");
        return icfg;
    }
    inline void setCHG(CommonCHGraph* c)
    {
        chgraph = c;
    }
    inline CommonCHGraph* getCHG()
    {
        assert(chgraph && "empty ICFG! Build SVF IR first!");
        return chgraph;
    }
 
    inline const CallGraph* getCallGraph()
    {
        assert(callGraph && "empty CallGraph! Build SVF IR first!");
        return callGraph;
    }
 
    inline void setCallGraph(CallGraph* cg)
    {
        callGraph = cg;
    }
 
    const FunObjVar* getFunObjVar(const std::string& name);
 
    inline const std::string& getModuleIdentifier() const
    {
        if (pagReadFromTxt.empty())
        {
            assert(!moduleIdentifier.empty() &&
                   "No module found! Reading from a file other than LLVM-IR?");
            return moduleIdentifier;
        }
        else
        {
            return pagReadFromTxt;
        }
    }
 
    static inline std::string pagFileName()
    {
        return pagReadFromTxt;
    }
 
    static inline bool pagReadFromTXT()
    {
        return !pagReadFromTxt.empty();
    }
 
    static inline void setPagFromTXT(const std::string& txt)
    {
        pagReadFromTxt = txt;
    }
 
    inline void setModuleIdentifier(const std::string& moduleIdentifier)
    {
        this->moduleIdentifier = moduleIdentifier;
    }
 
 
 
    inline SVFStmt::SVFStmtSetTy& getSVFStmtSet(SVFStmt::PEDGEK kind)
    {
        return KindToSVFStmtSetMap[kind];
    }
    inline SVFStmt::SVFStmtSetTy& getPTASVFStmtSet(SVFStmt::PEDGEK kind)
    {
        return KindToPTASVFStmtSetMap[kind];
    }
    inline bool hasSVFStmtList(const ICFGNode* inst) const
    {
        return icfgNode2SVFStmtsMap.find(inst) != icfgNode2SVFStmtsMap.end();
    }
    inline bool hasPTASVFStmtList(const ICFGNode* inst) const
    {
        return icfgNode2PTASVFStmtsMap.find(inst) !=
               icfgNode2PTASVFStmtsMap.end();
    }
    inline SVFStmtList& getSVFStmtList(const ICFGNode* inst)
    {
        return icfgNode2SVFStmtsMap[inst];
    }
    inline SVFStmtList& getPTASVFStmtList(const ICFGNode* inst)
    {
        return icfgNode2PTASVFStmtsMap[inst];
    }
    inline void addToSVFStmtList(ICFGNode* inst, SVFStmt* edge)
    {
        edge->setICFGNode(inst);
        icfgNode2SVFStmtsMap[inst].push_back(edge);
        if (edge->isPTAEdge())
            icfgNode2PTASVFStmtsMap[inst].push_back(edge);
    }
    inline void addToTypeLocSetsMap(NodeID argId, SVFTypeLocSetsPair& locSets)
    {
        typeLocSetsMap[argId]=locSets;
    }
    inline SVFTypeLocSetsPair& getTypeLocSetsMap(NodeID argId)
    {
        return typeLocSetsMap[argId];
    }
    inline SVFStmtSet& getGlobalSVFStmtSet()
    {
        return globSVFStmtSet;
    }
    inline const CallSiteSet& getCallSiteSet() const
    {
        return callSiteSet;
    }
    inline bool isPhiNode(const SVFVar* node) const
    {
        return phiNodeMap.find(node) != phiNodeMap.end();
    }
 
    inline bool hasFunArgsList(const FunObjVar* func) const
    {
        return (funArgsListMap.find(func) != funArgsListMap.end());
    }
    inline FunToArgsListMap& getFunArgsMap()
    {
        return funArgsListMap;
    }
    inline const ValVarList& getFunArgsList(const FunObjVar*  func) const
    {
        FunToArgsListMap::const_iterator it = funArgsListMap.find(func);
        assert(it != funArgsListMap.end() && "this function doesn't have arguments");
        return it->second;
    }
    inline bool hasCallSiteArgsMap(const CallICFGNode* cs) const
    {
        return (callSiteArgsListMap.find(cs) != callSiteArgsListMap.end());
    }
    inline CSToArgsListMap& getCallSiteArgsMap()
    {
        return callSiteArgsListMap;
    }
    inline const ValVarList& getCallSiteArgsList(const CallICFGNode* cs) const
    {
        CSToArgsListMap::const_iterator it = callSiteArgsListMap.find(cs);
        assert(it != callSiteArgsListMap.end() && "this call site doesn't have arguments");
        return it->second;
    }
    inline CSToRetMap& getCallSiteRets()
    {
        return callSiteRetMap;
    }
    inline const ValVar* getCallSiteRet(const RetICFGNode* cs) const
    {
        CSToRetMap::const_iterator it = callSiteRetMap.find(cs);
        assert(it != callSiteRetMap.end() && "this call site doesn't have return");
        return it->second;
    }
    inline bool callsiteHasRet(const RetICFGNode* cs) const
    {
        return callSiteRetMap.find(cs) != callSiteRetMap.end();
    }
    inline FunToRetMap& getFunRets()
    {
        return funRetMap;
    }
    inline const ValVar* getFunRet(const FunObjVar*  func) const
    {
        FunToRetMap::const_iterator it = funRetMap.find(func);
        assert(it != funRetMap.end() && "this function doesn't have return");
        return it->second;
    }
    inline bool funHasRet(const FunObjVar* func) const
    {
        return funRetMap.find(func) != funRetMap.end();
    }
 
 
    inline u32_t getFieldValNodeNum() const
    {
        return GepValObjMap.size();
    }
    inline u32_t getFieldObjNodeNum() const
    {
        return GepObjVarMap.size();
    }
 
    NodeID getGepValVar(NodeID curInst, NodeID base,
                        const AccessPath& ap) const;
 
 
    inline const CallSiteToFunPtrMap& getIndirectCallsites() const
    {
        return indCallSiteToFunPtrMap;
    }
    inline NodeID getFunPtr(const CallICFGNode* cs) const
    {
        CallSiteToFunPtrMap::const_iterator it = indCallSiteToFunPtrMap.find(cs);
        assert(it!=indCallSiteToFunPtrMap.end() && "indirect callsite not have a function pointer?");
        return it->second;
    }
    inline const CallSiteSet& getIndCallSites(NodeID funPtr) const
    {
        FunPtrToCallSitesMap::const_iterator it = funPtrToCallSitesMap.find(funPtr);
        assert(it!=funPtrToCallSitesMap.end() && "function pointer not used at any indirect callsite?");
        return it->second;
    }
    inline bool isIndirectCallSites(const CallICFGNode* cs) const
    {
        return (indCallSiteToFunPtrMap.find(cs) != indCallSiteToFunPtrMap.end());
    }
    inline bool isFunPtr(NodeID id) const
    {
        return (funPtrToCallSitesMap.find(id) != funPtrToCallSitesMap.end());
    }
 
    inline SVFStmt* getIntraPAGEdge(NodeID src, NodeID dst, SVFStmt::PEDGEK kind)
    {
        return getIntraPAGEdge(getGNode(src), getGNode(dst), kind);
    }
    inline SVFStmt* getIntraPAGEdge(SVFVar* src, SVFVar* dst, SVFStmt::PEDGEK kind)
    {
        SVFStmt edge(src, dst, kind, false);
        const SVFStmt::SVFStmtSetTy& edgeSet = getSVFStmtSet(kind);
        SVFStmt::SVFStmtSetTy::const_iterator it = edgeSet.find(&edge);
        assert(it != edgeSet.end() && "can not find pag edge");
        return (*it);
    }
 
 
    inline const BaseObjVar* getBaseObject(NodeID id) const
    {
        const SVFVar* node = getSVFVar(id);
        if(const GepObjVar* gepObjVar = SVFUtil::dyn_cast<GepObjVar>(node))
            return SVFUtil::dyn_cast<BaseObjVar>(
                       getSVFVar(gepObjVar->getBaseNode()));
        else
            return SVFUtil::dyn_cast<BaseObjVar>(node);
    }
 
    inline const ValVar* getBaseValVar(NodeID id) const
    {
        const SVFVar* node = getSVFVar(id);
        if(const GepValVar* gepVar = SVFUtil::dyn_cast<GepValVar>(node))
            return gepVar->getBaseNode();
        else
            return SVFUtil::dyn_cast<ValVar>(node);
    }
 
    NodeID getGepObjVar(const BaseObjVar* baseObj, const APOffset& ap);
    NodeID getGepObjVar(NodeID id, const APOffset& ap) ;
 
    inline NodeID getFIObjVar(const BaseObjVar* obj) const
    {
        return obj->getId();
    }
    inline NodeID getFIObjVar(NodeID id) const
    {
        return getBaseObjVarID(id);
    }
 
 
 
    inline bool isBlkObjOrConstantObj(NodeID id) const
    {
        return (isBlkObj(id) || isConstantObj(id));
    }
 
    inline bool isConstantObj(NodeID id) const
    {
        const BaseObjVar* obj = getBaseObject(id);
        assert(obj && "not an object node?");
        return isConstantSym(id) ||
               obj->isConstDataOrConstGlobal();
    }
 
 
 
    inline NodeID getBaseObjVarID(NodeID id) const
    {
        return getBaseObject(id)->getId();
    }
 
 
    NodeBS& getAllFieldsObjVars(const BaseObjVar* obj);
    NodeBS& getAllFieldsObjVars(NodeID id);
    NodeBS getFieldsAfterCollapse(NodeID id);
    inline NodeID addDummyValNode()
    {
        return addDummyValNode(NodeIDAllocator::get()->allocateValueId(), nullptr);
    }
    inline NodeID addDummyObjNode(const SVFType* type)
    {
        return addDummyObjNode(NodeIDAllocator::get()->allocateObjectId(), type);
    }
 
    bool isValidPointer(NodeID nodeId) const;
 
    bool isValidTopLevelPtr(const SVFVar* node);
 
    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:
 
    inline void addToStmt2TypeMap(SVFStmt* edge)
    {
        bool added = KindToSVFStmtSetMap[edge->getEdgeKind()].insert(edge).second;
        (void)added; // Suppress warning of unused variable under release build
        assert(added && "duplicated edge, not added!!!");
        if (edge->isPTAEdge() || (SVFUtil::isa<CopyStmt>(edge) && SVFUtil::cast<CopyStmt>(edge)->isInt2Ptr()))
        {
            totalPTAPAGEdge++;
            KindToPTASVFStmtSetMap[edge->getEdgeKind()].insert(edge);
        }
    }
 
 
    inline void addFunArgs(const FunObjVar* fun, const ValVar* arg)
    {
        FunEntryICFGNode* funEntryBlockNode = icfg->getFunEntryICFGNode(fun);
        addFunArgs(funEntryBlockNode, fun, arg);
    }
 
    inline void addFunArgs(FunEntryICFGNode* funEntryBlockNode, const FunObjVar* fun, const ValVar* arg)
    {
        funEntryBlockNode->addFormalParms(arg);
        funArgsListMap[fun].push_back(arg);
    }
    inline void addFunRet(const FunObjVar* fun, const ValVar* ret)
    {
        FunExitICFGNode* funExitBlockNode = icfg->getFunExitICFGNode(fun);
        addFunRet(funExitBlockNode, fun, ret);
    }
 
    inline void addFunRet(FunExitICFGNode* funExitBlockNode, const FunObjVar* fun, const ValVar* ret)
    {
        funExitBlockNode->addFormalRet(ret);
        funRetMap[fun] = ret;
    }
    inline void addCallSiteArgs(CallICFGNode* callBlockNode,const ValVar* arg)
    {
        callBlockNode->addActualParms(arg);
        callSiteArgsListMap[callBlockNode].push_back(arg);
    }
    inline void addCallSiteRets(RetICFGNode* retBlockNode,const ValVar* arg)
    {
        retBlockNode->addActualRet(arg);
        callSiteRetMap[retBlockNode]= arg;
    }
    inline void addIndirectCallsites(const CallICFGNode* cs,NodeID funPtr)
    {
        bool added = indCallSiteToFunPtrMap.emplace(cs, funPtr).second;
        (void) added;
        funPtrToCallSitesMap[funPtr].insert(cs);
        assert(added && "adding the same indirect callsite twice?");
    }
 
 
 
    inline NodeID addValNode(NodeID i, const SVFType* type, const ICFGNode* icfgNode)
    {
        ValVar *node = new ValVar(i, type, icfgNode, ValVar::ValNode);
        return addValNode(node);
    }
 
    NodeID addFunValNode(NodeID i, const ICFGNode* icfgNode, const FunObjVar* funObjVar, const SVFType* type)
    {
        FunValVar* node = new FunValVar(i, icfgNode, funObjVar, type);
        return addValNode(node);
    }
 
    NodeID addArgValNode(NodeID i, u32_t argNo, const ICFGNode* icfgNode, const FunObjVar* callGraphNode, const SVFType* type)
    {
        ArgValVar* node =
            new ArgValVar(i, argNo, icfgNode, callGraphNode, type);
        return addValNode(node);
    }
 
    inline NodeID addConstantFPValNode(const NodeID i, double dval,
                                       const ICFGNode* icfgNode, const SVFType* type)
    {
        SVFVar* node = new ConstFPValVar(i, dval, icfgNode, type);
        return addNode(node);
    }
 
    inline NodeID addConstantIntValNode(NodeID i, const std::pair<s64_t, u64_t>& intValue,
                                        const ICFGNode* icfgNode, const SVFType* type)
    {
        SVFVar* node = new ConstIntValVar(i, intValue.first, intValue.second, icfgNode, type);
        return addNode(node);
    }
 
    inline NodeID addConstantNullPtrValNode(const NodeID i, const ICFGNode* icfgNode, const SVFType* type)
    {
        SVFVar* node = new ConstNullPtrValVar(i, icfgNode, type);
        return addNode(node);
    }
 
    inline NodeID addGlobalValNode(const NodeID i, const ICFGNode* icfgNode, const SVFType* svfType)
    {
        SVFVar* node = new GlobalValVar(i, icfgNode, svfType);
        return addNode(node);
    }
 
    inline NodeID addConstantAggValNode(const NodeID i, const ICFGNode* icfgNode, const SVFType* svfType)
    {
        SVFVar* node = new ConstAggValVar(i, icfgNode, svfType);
        return addNode(node);
    }
 
    inline NodeID addConstantDataValNode(const NodeID i, const ICFGNode* icfgNode, const SVFType* type)
    {
        SVFVar* node = new ConstDataValVar(i, icfgNode, type);
        return addNode(node);
    }
 
 
    inline NodeID addObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        return addFIObjNode( i, ti, node);
    }
 
    inline NodeID addHeapObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        HeapObjVar *heapObj = new HeapObjVar(i, ti, node);
        return addBaseObjNode(heapObj);
    }
 
    inline NodeID addStackObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        StackObjVar *stackObj = new StackObjVar(i, ti, node);
        return addBaseObjNode(stackObj);
    }
 
    NodeID addFunObjNode(NodeID id,  ObjTypeInfo* ti, const ICFGNode* node)
    {
        FunObjVar* funObj = new FunObjVar(id, ti, node);
        return addBaseObjNode(funObj);
    }
 
 
    inline NodeID addConstantFPObjNode(NodeID i, ObjTypeInfo* ti, double dval, const ICFGNode* node)
    {
        ConstFPObjVar* conObj = new ConstFPObjVar(i, dval, ti, node);
        return addBaseObjNode(conObj);
    }
 
 
    inline NodeID addConstantIntObjNode(NodeID i, ObjTypeInfo* ti, const std::pair<s64_t, u64_t>& intValue, const ICFGNode* node)
    {
        ConstIntObjVar* conObj =
            new ConstIntObjVar(i, intValue.first, intValue.second, ti, node);
        return addBaseObjNode(conObj);
    }
 
 
    inline NodeID addConstantNullPtrObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        ConstNullPtrObjVar* conObj = new ConstNullPtrObjVar(i, ti, node);
        return addBaseObjNode(conObj);
    }
 
    inline NodeID addGlobalObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        GlobalObjVar* gObj = new GlobalObjVar(i, ti, node);
        return addBaseObjNode(gObj);
    }
    inline NodeID addConstantAggObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        ConstAggObjVar* conObj = new ConstAggObjVar(i, ti, node);
        return addBaseObjNode(conObj);
    }
    inline NodeID addConstantDataObjNode(const NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        ConstDataObjVar* conObj = new ConstDataObjVar(i, ti, node);
        return addBaseObjNode(conObj);
    }
 
    inline NodeID addRetNode(NodeID i, const FunObjVar* callGraphNode, const SVFType* type, const ICFGNode* icn)
    {
        SVFVar *node = new RetValPN(i, callGraphNode, type, icn);
        return addRetNode(callGraphNode, node);
    }
    inline NodeID addVarargNode(NodeID i, const FunObjVar* val, const SVFType* type, const ICFGNode* n)
    {
        SVFVar *node = new VarArgValPN(i, val, type, n);
        return addNode(node);
    }
 
    NodeID addGepValNode(NodeID curInst, const ValVar* base, const AccessPath& ap, NodeID i, const SVFType* type, const ICFGNode* node);
    NodeID addGepObjNode(const BaseObjVar* baseObj, const APOffset& apOffset, const NodeID gepId);
    NodeID addFIObjNode(NodeID i, ObjTypeInfo* ti, const ICFGNode* node)
    {
        BaseObjVar* baseObj = new BaseObjVar(i, ti, node);
        return addBaseObjNode(baseObj);
    }
 
 
 
 
    inline NodeID addDummyValNode(NodeID i, const ICFGNode* node)
    {
        return addValNode(new DummyValVar(i, node));
    }
    inline NodeID addDummyObjNode(NodeID i, const SVFType* type)
    {
        if (idToObjTypeInfoMap().find(i) == idToObjTypeInfoMap().end())
        {
            ObjTypeInfo* ti = createObjTypeInfo(type);
            idToObjTypeInfoMap()[i] = ti;
            return addDummyObjNode(new DummyObjVar(i, ti, nullptr));
        }
        else
        {
            return addDummyObjNode(new DummyObjVar(i, getObjTypeInfo(i), nullptr));
        }
    }
 
    inline NodeID addBlackholeObjNode()
    {
        return addObjNode(new DummyObjVar(getBlackHoleNode(), getObjTypeInfo(getBlackHoleNode()), nullptr));
    }
    inline NodeID addConstantObjNode()
    {
        return addObjNode(new DummyObjVar(getConstantNode(), getObjTypeInfo(getConstantNode()), nullptr));
    }
    inline NodeID addBlackholePtrNode()
    {
        return addDummyValNode(getBlkPtr(), nullptr);
    }
 
    NodeID addValNode(ValVar* node);
    NodeID addObjNode(ObjVar *node);
    inline NodeID addRetNode(const FunObjVar*, SVFVar *node)
    {
        return addNode(node);
    }
    inline NodeID addVarargNode(const FunObjVar*, SVFVar *node)
    {
        return addNode(node);
    }
 
    inline void addGlobalPAGEdge(const SVFStmt* edge)
    {
        globSVFStmtSet.insert(edge);
    }
    inline void addCallSite(const CallICFGNode* call)
    {
        callSiteSet.insert(call);
    }
 
 
    AddrStmt* addAddrStmt(NodeID src, NodeID dst);
    void addAddrStmt(AddrStmt* edge);
    CopyStmt* addCopyStmt(NodeID src, NodeID dst, CopyStmt::CopyKind type);
    void addCopyStmt(CopyStmt* edge);
 
    PhiStmt*  addPhiStmt(NodeID res, NodeID opnd, const ICFGNode* pred);
    void addPhiStmt(PhiStmt* edge, SVFVar* src, SVFVar* dst);
    SelectStmt*  addSelectStmt(NodeID res, NodeID op1, NodeID op2, NodeID cond);
    void addSelectStmt(SelectStmt* edge, SVFVar* src, SVFVar* dst);
    CmpStmt* addCmpStmt(NodeID op1, NodeID op2, NodeID dst, u32_t predict);
    void addCmpStmt(CmpStmt* edge, SVFVar* src, SVFVar* dst);
    BinaryOPStmt* addBinaryOPStmt(NodeID op1, NodeID op2, NodeID dst,
                                  u32_t opcode);
    void addBinaryOPStmt(BinaryOPStmt* edge, SVFVar* src, SVFVar* dst);
    UnaryOPStmt* addUnaryOPStmt(NodeID src, NodeID dst, u32_t opcode);
    void addUnaryOPStmt(UnaryOPStmt* edge, SVFVar* src, SVFVar* dst);
    BranchStmt* addBranchStmt(NodeID br, NodeID cond,
                              const BranchStmt::SuccAndCondPairVec& succs);
    void addBranchStmt(BranchStmt* edge, SVFVar* src, SVFVar* dst);
    LoadStmt* addLoadStmt(NodeID src, NodeID dst);
    void addLoadStmt(LoadStmt* edge);
    StoreStmt* addStoreStmt(NodeID src, NodeID dst, const ICFGNode* val);
    void addStoreStmt(StoreStmt* edge, SVFVar* src, SVFVar* dst);
    CallPE* addCallPE(NodeID src, NodeID dst, const CallICFGNode* cs,
                      const FunEntryICFGNode* entry);
    void addCallPE(CallPE* edge, SVFVar* src, SVFVar* dst);
    RetPE* addRetPE(NodeID src, NodeID dst, const CallICFGNode* cs,
                    const FunExitICFGNode* exit);
    void addRetPE(RetPE* edge, SVFVar* src, SVFVar* dst);
    GepStmt* addGepStmt(NodeID src, NodeID dst, const AccessPath& ap,
                        bool constGep);
    void addGepStmt(GepStmt* edge);
    GepStmt* addNormalGepStmt(NodeID src, NodeID dst, const AccessPath& ap);
    GepStmt* addVariantGepStmt(NodeID src, NodeID dst, const AccessPath& ap);
    TDForkPE* addThreadForkPE(NodeID src, NodeID dst, const CallICFGNode* cs,
                              const FunEntryICFGNode* entry);
    TDJoinPE* addThreadJoinPE(NodeID src, NodeID dst, const CallICFGNode* cs,
                              const FunExitICFGNode* exit);
 
    SVFStmt* addBlackHoleAddrStmt(NodeID node);
};
 
typedef SVFIR PAG;
 
} // End namespace SVF
 
 
 
#endif /* INCLUDE_SVFIR_H_ */