Steensgaard.h

Go to the documentation of this file.

/*
 * Steensgaard.h
 *
 *  Created on: 2 Feb. 2021
 *      Author: Yulei Sui
 */
 
#ifndef INCLUDE_WPA_STEENSGAARD_H_
#define INCLUDE_WPA_STEENSGAARD_H_
 
#include "WPA/Andersen.h"
 
namespace SVF
{
typedef WPASolver<ConstraintGraph*> WPAConstraintSolver;
 
class Steensgaard : public AndersenBase
{
 
public:
    typedef Map<NodeID, NodeID> NodeToEquivClassMap;
    typedef Map<NodeID, Set<NodeID>> NodeToSubsMap;
 
    Steensgaard(SVFIR* _pag) : AndersenBase(_pag, Steensgaard_WPA, true) {}
 
    static Steensgaard* createSteensgaard(SVFIR* _pag)
    {
        if (steens == nullptr)
        {
            steens = new Steensgaard(_pag);
            steens->analyze();
            return steens;
        }
        return steens;
    }
    static void releaseSteensgaard()
    {
        if (steens)
            delete steens;
        steens = nullptr;
    }
 
    virtual void solveWorklist() override;
 
    void processAllAddr();
 
    void ecUnion(NodeID id, NodeID ec);
 
 
    static inline bool classof(const Steensgaard*)
    {
        return true;
    }
    static inline bool classof(const AndersenBase* pta)
    {
        return (pta->getAnalysisTy() == Steensgaard_WPA);
    }
    static inline bool classof(const PointerAnalysis* pta)
    {
        return (pta->getAnalysisTy() == Steensgaard_WPA);
    }
 
    virtual inline const PointsTo& getPts(NodeID id) override
    {
        return getPTDataTy()->getPts(getEC(id));
    }
    virtual inline bool unionPts(NodeID id, const PointsTo& target) override
    {
        id = getEC(id);
        return getPTDataTy()->unionPts(id, target);
    }
    virtual inline bool unionPts(NodeID id, NodeID ptd) override
    {
        id = getEC(id);
        ptd = getEC(ptd);
        return getPTDataTy()->unionPts(id, ptd);
    }
 
    inline NodeID getEC(NodeID id) const
    {
        NodeToEquivClassMap::const_iterator it = nodeToECMap.find(id);
        if (it == nodeToECMap.end())
            return id;
        else
            return it->second;
    }
    inline NodeID sccRepNode(NodeID id) const override
    {
        return getEC(id);
    }
    void setEC(NodeID node, NodeID rep);
 
    inline Set<NodeID>& getSubNodes(NodeID id)
    {
        nodeToSubsMap[id].insert(id);
        return nodeToSubsMap[id];
    }
    inline void addSubNode(NodeID node, NodeID sub)
    {
        nodeToSubsMap[node].insert(sub);
    }
 
    virtual inline bool addCopyEdge(NodeID src, NodeID dst) override
    {
        return consCG->addCopyCGEdge(src, dst);
    }
 
private:
    static Steensgaard* steens; // static instance
    NodeToEquivClassMap nodeToECMap;
    NodeToSubsMap nodeToSubsMap;
};
 
} // namespace SVF
 
#endif /* INCLUDE_WPA_STEENSGAARD_H_ */