ConsGEdge.h

Go to the documentation of this file.

//===- ConsGEdge.h -- Constraint graph edge-----------------------------------//
//
//                     SVF: Static Value-Flow Analysis
//
// Copyright (C) <2013-2017>  <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/>.
//
//===----------------------------------------------------------------------===//
 
/*
 * ConsGEdge.h
 *
 *  Created on: Mar 19, 2014
 *      Author: Yulei Sui
 */
 
#ifndef CONSGEDGE_H_
#define CONSGEDGE_H_
 
#include "SVFIR/SVFIR.h"
#include "Util/WorkList.h"
 
#include <map>
#include <set>
 
namespace SVF
{
 
class ConstraintNode;
typedef GenericEdge<ConstraintNode> GenericConsEdgeTy;
class ConstraintEdge : public GenericConsEdgeTy
{
 
public:
    enum ConstraintEdgeK
    {
        Addr, Copy, Store, Load, NormalGep, VariantGep
    };
private:
    EdgeID edgeId;
public:
    ConstraintEdge(ConstraintNode* s, ConstraintNode* d, ConstraintEdgeK k, EdgeID id = 0) : GenericConsEdgeTy(s,d,k),edgeId(id)
    {
    }
    ~ConstraintEdge()
    {
    }
    inline EdgeID getEdgeID() const
    {
        return edgeId;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == Addr ||
               edge->getEdgeKind() == Copy ||
               edge->getEdgeKind() == Store ||
               edge->getEdgeKind() == Load ||
               edge->getEdgeKind() == NormalGep ||
               edge->getEdgeKind() == VariantGep;
    }
    typedef GenericNode<ConstraintNode,ConstraintEdge>::GEdgeSetTy ConstraintEdgeSetTy;
 
};
 
 
class AddrCGEdge: public ConstraintEdge
{
private:
    AddrCGEdge();                      
    AddrCGEdge(const AddrCGEdge &);  
    void operator=(const AddrCGEdge &); 
public:
 
    static inline bool classof(const AddrCGEdge *)
    {
        return true;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == Addr;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == Addr;
    }
 
    AddrCGEdge(ConstraintNode* s, ConstraintNode* d, EdgeID id);
};
 
 
class CopyCGEdge: public ConstraintEdge
{
private:
    CopyCGEdge();                      
    CopyCGEdge(const CopyCGEdge &);  
    void operator=(const CopyCGEdge &); 
public:
 
    static inline bool classof(const CopyCGEdge *)
    {
        return true;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == Copy;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == Copy;
    }
 
    CopyCGEdge(ConstraintNode* s, ConstraintNode* d, EdgeID id) : ConstraintEdge(s,d,Copy,id)
    {
    }
};
 
 
class StoreCGEdge: public ConstraintEdge
{
private:
    StoreCGEdge();                      
    StoreCGEdge(const StoreCGEdge &);  
    void operator=(const StoreCGEdge &); 
 
public:
 
    static inline bool classof(const StoreCGEdge *)
    {
        return true;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == Store;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == Store;
    }
 
    StoreCGEdge(ConstraintNode* s, ConstraintNode* d, EdgeID id) : ConstraintEdge(s,d,Store,id)
    {
    }
};
 
 
class LoadCGEdge: public ConstraintEdge
{
private:
    LoadCGEdge();                      
    LoadCGEdge(const LoadCGEdge &);  
    void operator=(const LoadCGEdge &); 
 
public:
 
    static inline bool classof(const LoadCGEdge *)
    {
        return true;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == Load;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == Load;
    }
 
    LoadCGEdge(ConstraintNode* s, ConstraintNode* d, EdgeID id) : ConstraintEdge(s,d,Load,id)
    {
    }
};
 
 
class GepCGEdge: public ConstraintEdge
{
private:
    GepCGEdge();                      
    GepCGEdge(const GepCGEdge &);  
    void operator=(const GepCGEdge &); 
 
protected:
 
    GepCGEdge(ConstraintNode* s, ConstraintNode* d, ConstraintEdgeK k, EdgeID id)
        : ConstraintEdge(s,d,k,id)
    {
 
    }
 
public:
 
    static inline bool classof(const GepCGEdge *)
    {
        return true;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == NormalGep ||
               edge->getEdgeKind() == VariantGep;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == NormalGep ||
               edge->getEdgeKind() == VariantGep;
    }
 
};
 
class NormalGepCGEdge : public GepCGEdge
{
private:
    NormalGepCGEdge();                      
    NormalGepCGEdge(const NormalGepCGEdge &);  
    void operator=(const NormalGepCGEdge &); 
 
    AccessPath ap;  
 
public:
 
    static inline bool classof(const NormalGepCGEdge *)
    {
        return true;
    }
    static inline bool classof(const GepCGEdge *edge)
    {
        return edge->getEdgeKind() == NormalGep;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == NormalGep;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == NormalGep;
    }
 
    NormalGepCGEdge(ConstraintNode* s, ConstraintNode* d, const AccessPath& ap,
                    EdgeID id)
        : GepCGEdge(s, d, NormalGep, id), ap(ap)
    {
    }
 
    inline const AccessPath& getAccessPath() const
    {
        return ap;
    }
 
    inline APOffset getConstantFieldIdx() const
    {
        return ap.getConstantStructFldIdx();
    }
 
};
 
class VariantGepCGEdge : public GepCGEdge
{
private:
    VariantGepCGEdge();                      
    VariantGepCGEdge(const VariantGepCGEdge &);  
    void operator=(const VariantGepCGEdge &); 
 
public:
 
    static inline bool classof(const VariantGepCGEdge *)
    {
        return true;
    }
    static inline bool classof(const GepCGEdge *edge)
    {
        return edge->getEdgeKind() == VariantGep;
    }
    static inline bool classof(const ConstraintEdge *edge)
    {
        return edge->getEdgeKind() == VariantGep;
    }
    static inline bool classof(const GenericConsEdgeTy *edge)
    {
        return edge->getEdgeKind() == VariantGep;
    }
 
    VariantGepCGEdge(ConstraintNode* s, ConstraintNode* d, EdgeID id)
        : GepCGEdge(s,d,VariantGep,id)
    {}
};
 
} // End namespace SVF
 
#endif /* CONSGEDGE_H_ */