ICFGNode.h

Go to the documentation of this file.

//===- ICFGNode.h -- ICFG node------------------------------------------------//
//
//                     SVF: Static Value-Flow Analysis
//
// Copyright (C) <2013-2018>  <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/>.
//
//===----------------------------------------------------------------------===//
 
/*
 * ICFGNode.h
 *
 *  Created on: Sep 11, 2018
 *      Author: Yulei
 */
 
#ifndef ICFGNODE_H_
#define ICFGNODE_H_
 
#include "Util/SVFUtil.h"
#include "Graphs/BasicBlockG.h"
#include "Graphs/GenericGraph.h"
#include "Graphs/ICFGEdge.h"
 
namespace SVF
{
 
class ICFGNode;
class RetICFGNode;
class CallPE;
class RetPE;
class SVFStmt;
class SVFVar;
class VFGNode;
 
typedef GenericNode<ICFGNode, ICFGEdge> GenericICFGNodeTy;
 
class ICFGNode : public GenericICFGNodeTy
{
 
public:
 
    typedef ICFGEdge::ICFGEdgeSetTy::iterator iterator;
    typedef ICFGEdge::ICFGEdgeSetTy::const_iterator const_iterator;
    typedef Set<const CallPE *> CallPESet;
    typedef Set<const RetPE *> RetPESet;
    typedef std::list<const VFGNode*> VFGNodeList;
    typedef std::list<const SVFStmt*> SVFStmtList;
    typedef GNodeK ICFGNodeK;
 
public:
    ICFGNode(NodeID i, GNodeK k) : GenericICFGNodeTy(i, k), fun(nullptr), bb(nullptr)
    {
    }
 
    virtual const FunObjVar* getFun() const
    {
        return fun;
    }
 
    virtual const SVFBasicBlock* getBB() const
    {
        return bb;
    }
 
 
 
    friend OutStream &operator<<(OutStream &o, const ICFGNode &node)
    {
        o << node.toString();
        return o;
    }
 
    inline void addVFGNode(const VFGNode *vfgNode)
    {
        VFGNodes.push_back(vfgNode);
    }
 
    inline const VFGNodeList& getVFGNodes() const
    {
        return VFGNodes;
    }
 
    inline void addSVFStmt(const SVFStmt *edge)
    {
        pagEdges.push_back(edge);
    }
 
    inline const SVFStmtList& getSVFStmts() const
    {
        return pagEdges;
    }
 
    virtual const std::string toString() const;
 
 
 
    void dump() const;
 
 
    static inline bool classof(const ICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const GenericICFGNodeTy* node)
    {
        return isICFGNodeKinds(node->getNodeKind());
    }
 
    static inline bool classof(const SVFValue* node)
    {
        return isICFGNodeKinds(node->getNodeKind());
    }
 
protected:
    const FunObjVar* fun;
    const SVFBasicBlock* bb;
    VFGNodeList VFGNodes; //< a list of VFGNodes
    SVFStmtList pagEdges; //< a list of PAGEdges
 
};
 
class GlobalICFGNode : public ICFGNode
{
 
public:
    GlobalICFGNode(NodeID id) : ICFGNode(id, GlobalBlock)
    {
    }
 
 
    static inline bool classof(const GlobalICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const ICFGNode *node)
    {
        return node->getNodeKind() == GlobalBlock;
    }
 
    static inline bool classof(const GenericICFGNodeTy *node)
    {
        return node->getNodeKind() == GlobalBlock;
    }
 
    const std::string toString() const override;
 
    const std::string getSourceLoc() const override
    {
        return "Global ICFGNode";
    }
};
 
class IntraICFGNode : public ICFGNode
{
    friend class GraphDBClient;
 
private:
    bool isRet;
 
public:
    IntraICFGNode(NodeID id, const SVFBasicBlock* b, bool isReturn) : ICFGNode(id, IntraBlock), isRet(isReturn)
    {
        fun = b->getFunction();
        bb = b;
    }
 
 
    static inline bool classof(const IntraICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const ICFGNode *node)
    {
        return node->getNodeKind() == IntraBlock;
    }
 
    static inline bool classof(const GenericICFGNodeTy *node)
    {
        return node->getNodeKind() == IntraBlock;
    }
 
    const std::string toString() const override;
 
    inline bool isRetInst() const
    {
        return isRet;
    }
};
 
class InterICFGNode : public ICFGNode
{
 
public:
    InterICFGNode(NodeID id, ICFGNodeK k) : ICFGNode(id, k)
    {
    }
 
 
    static inline bool classof(const InterICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const ICFGNode* node)
    {
        return isInterICFGNodeKind(node->getNodeKind());
    }
 
    static inline bool classof(const GenericICFGNodeTy* node)
    {
        return isInterICFGNodeKind(node->getNodeKind());
    }
 
    static inline bool classof(const SVFValue* node)
    {
        return isInterICFGNodeKind(node->getNodeKind());
    }
 
};
 
 
 
 
class FunEntryICFGNode : public InterICFGNode
{
    friend class GraphDBClient;
 
public:
    typedef std::vector<const SVFVar *> FormalParmNodeVec;
private:
    FormalParmNodeVec FPNodes;
 
public:
    FunEntryICFGNode(NodeID id, const FunObjVar* f);
 
    inline const FunObjVar* getFun() const override
    {
        return fun;
    }
 
    inline const FormalParmNodeVec &getFormalParms() const
    {
        return FPNodes;
    }
 
    inline void addFormalParms(const SVFVar *fp)
    {
        FPNodes.push_back(fp);
    }
 
 
    static inline bool classof(const FunEntryICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const InterICFGNode *node)
    {
        return node->getNodeKind() == FunEntryBlock;
    }
 
    static inline bool classof(const ICFGNode *node)
    {
        return node->getNodeKind() == FunEntryBlock;
    }
 
    static inline bool classof(const GenericICFGNodeTy *node)
    {
        return node->getNodeKind() == FunEntryBlock;
    }
 
    static inline bool classof(const SVFValue*node)
    {
        return node->getNodeKind() == FunEntryBlock;
    }
 
    const std::string toString() const override;
 
    const std::string getSourceLoc() const override;
};
 
class FunExitICFGNode : public InterICFGNode
{
    friend class GraphDBClient;
 
private:
    const SVFVar *formalRet;
 
public:
    FunExitICFGNode(NodeID id, const FunObjVar* f, const SVFBasicBlock* b)
        : InterICFGNode(id, FunExitBlock), formalRet(nullptr)
    {
        this->fun = f;
        this->bb = b;
    }
 
    inline const FunObjVar* getFun() const override
    {
        return fun;
    }
 
    inline const SVFVar *getFormalRet() const
    {
        return formalRet;
    }
 
    inline void addFormalRet(const SVFVar *fr)
    {
        formalRet = fr;
    }
 
 
    static inline bool classof(const FunEntryICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const ICFGNode *node)
    {
        return node->getNodeKind() == FunExitBlock;
    }
 
    static inline bool classof(const InterICFGNode *node)
    {
        return node->getNodeKind() == FunExitBlock;
    }
 
    static inline bool classof(const GenericICFGNodeTy *node)
    {
        return node->getNodeKind() == FunExitBlock;
    }
 
    static inline bool classof(const SVFValue*node)
    {
        return node->getNodeKind() == FunExitBlock;
    }
 
    const std::string toString() const override;
 
    const std::string getSourceLoc() const override;
};
 
class CallICFGNode : public InterICFGNode
{
    friend class GraphDBClient;
 
public:
    typedef std::vector<const ValVar *> ActualParmNodeVec;
 
protected:
    const RetICFGNode* ret;
    ActualParmNodeVec APNodes;      
    const FunObjVar* calledFunc;  
    bool isvararg;                  
    bool isVirCallInst;             
    SVFVar* vtabPtr;                
    s32_t virtualFunIdx;            
    std::string funNameOfVcall;     
    const SVFVar* indFunPtr;
 
 
public:
    CallICFGNode(NodeID id, const SVFBasicBlock* b, const SVFType* ty,
                 const FunObjVar* cf, bool iv, bool ivc, s32_t vfi,
                 const std::string& fnv)
        : InterICFGNode(id, FunCallBlock), ret(nullptr), calledFunc(cf),
          isvararg(iv), isVirCallInst(ivc), vtabPtr(nullptr),
          virtualFunIdx(vfi), funNameOfVcall(fnv)
    {
        fun = b->getFunction();
        bb = b;
        type = ty;
    }
 
    inline const RetICFGNode* getRetICFGNode() const
    {
        assert(ret && "RetICFGNode not set?");
        return ret;
    }
 
    inline void setRetICFGNode(const RetICFGNode* r)
    {
        ret = r;
    }
 
    inline const FunObjVar* getCaller() const
    {
        return getFun();
    }
 
    inline const SVFBasicBlock* getParent() const
    {
        return getBB();
    }
 
    inline bool isIndirectCall() const
    {
        return nullptr == calledFunc;
    }
 
    inline const ActualParmNodeVec &getActualParms() const
    {
        return APNodes;
    }
 
    inline void addActualParms(const ValVar *ap)
    {
        APNodes.push_back(ap);
    }
 
    inline const ValVar* getArgument(u32_t ArgNo) const
    {
        return getActualParms()[ArgNo];
    }
 
    inline u32_t arg_size() const
    {
        return APNodes.size();
    }
    inline bool arg_empty() const
    {
        return APNodes.empty();
    }
 
    inline u32_t getNumArgOperands() const
    {
        return arg_size();
    }
    inline const FunObjVar* getCalledFunction() const
    {
        return calledFunc;
    }
 
    inline bool isVarArg() const
    {
        return isvararg;
    }
    inline bool isVirtualCall() const
    {
        return isVirCallInst;
    }
 
    inline void setVtablePtr(SVFVar* v)
    {
        vtabPtr = v;
    }
 
    inline const SVFVar* getVtablePtr() const
    {
        assert(isVirtualCall() && "not a virtual call?");
        return vtabPtr;
    }
 
 
    inline s32_t getFunIdxInVtable() const
    {
        assert(isVirtualCall() && "not a virtual call?");
        assert(virtualFunIdx >=0 && "virtual function idx is less than 0? not set yet?");
        return virtualFunIdx;
    }
 
    inline const std::string& getFunNameOfVirtualCall() const
    {
        assert(isVirtualCall() && "not a virtual call?");
        return funNameOfVcall;
    }
 
 
    static inline bool classof(const CallICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const ICFGNode *node)
    {
        return node->getNodeKind() == FunCallBlock;
    }
 
    static inline bool classof(const InterICFGNode *node)
    {
        return node->getNodeKind() == FunCallBlock;
    }
 
    static inline bool classof(const GenericICFGNodeTy *node)
    {
        return node->getNodeKind() == FunCallBlock;
    }
 
    static inline bool classof(const SVFValue*node)
    {
        return node->getNodeKind() == FunCallBlock;
    }
 
    const std::string toString() const override;
 
    const std::string getSourceLoc() const override
    {
        return "CallICFGNode: " + ICFGNode::getSourceLoc();
    }
 
    inline void setIndFunPtr(const SVFVar* indFun)
    {
        assert(isIndirectCall() && "not a indirect call?");
        indFunPtr = indFun;
    }
 
    inline const SVFVar* getIndFunPtr() const
    {
        assert(isIndirectCall() && "not a indirect call?");
        return indFunPtr;
    }
};
 
 
class RetICFGNode : public InterICFGNode
{
    friend class GraphDBClient;
 
protected:
    inline void setCallBlockNode(const CallICFGNode* cb)
    {
        callBlockNode = cb;
    }
 
private:
    const SVFVar *actualRet;
    const CallICFGNode* callBlockNode;
 
public:
    RetICFGNode(NodeID id, CallICFGNode* cb) :
        InterICFGNode(id, FunRetBlock), actualRet(nullptr), callBlockNode(cb)
    {
        if (nullptr != cb)
        {
            fun = cb->getFun();
            bb = cb->getBB();
            type = cb->getType();
        }
    }
 
    inline const CallICFGNode* getCallICFGNode() const
    {
        return callBlockNode;
    }
    inline const SVFVar *getActualRet() const
    {
        return actualRet;
    }
 
    inline void addActualRet(const SVFVar *ar)
    {
        actualRet = ar;
    }
 
 
    static inline bool classof(const RetICFGNode *)
    {
        return true;
    }
 
    static inline bool classof(const InterICFGNode *node)
    {
        return node->getNodeKind() == FunRetBlock;
    }
 
    static inline bool classof(const ICFGNode *node)
    {
        return node->getNodeKind() == FunRetBlock;
    }
 
    static inline bool classof(const GenericICFGNodeTy *node)
    {
        return node->getNodeKind() == FunRetBlock;
    }
    static inline bool classof(const SVFValue*node)
    {
        return node->getNodeKind() == FunRetBlock;
    }
 
    const std::string toString() const override;
 
    const std::string getSourceLoc() const override
    {
        return "RetICFGNode: " + ICFGNode::getSourceLoc();
    }
};
 
} // End namespace SVF
 
#endif /* ICFGNode_H_ */