MTAAnnotator.h

Go to the documentation of this file.

/*
 * MTAAnnotator.h
 *
 *  Created on: May 4, 2014
 *      Author: Peng Di
 */
 
#ifndef MTAANNOTATOR_H_
#define MTAANNOTATOR_H_
 
#include "SVF-LLVM/BasicTypes.h"
#include "SVF-LLVM/LLVMModule.h"
#include "SVFIR/SVFValue.h"
#include "Util/Annotator.h"
#include "MTA/MHP.h"
 
namespace SVF
{
 
class MTAAnnotator: public Annotator
{
 
public:
    typedef Set<const Instruction*> InstSet;
    MTAAnnotator(): mhp(nullptr),lsa(nullptr)
    {
        numOfAllSt = 0;
        numOfAllLd = 0;
        numOfNonLocalSt = 0;
        numOfNonLocalLd = 0;
        numOfAliasSt = 0;
        numOfAliasLd = 0;
        numOfMHPSt = 0;
        numOfMHPLd = 0;
        numOfAnnotatedSt = 0;
        numOfAnnotatedLd = 0;
    }
    virtual ~MTAAnnotator()
    {
 
    }
 
    void annotateDRCheck(Instruction* inst);
 
    void initialize(MHP* mhp, LockAnalysis* lsa);
 
    void pruneThreadLocal(PointerAnalysis*pta);
 
    void pruneAliasMHP(PointerAnalysis* pta);
 
    void performAnnotate();
 
    void collectLoadStoreInst(SVFModule* mod);
 
    const Value* getStoreOperand(const Instruction* inst);
    const Value* getLoadOperand(const Instruction* inst);
 
 
    inline bool isMemset(const Instruction* I)
    {
        const SVFInstruction* svfInst = LLVMModuleSet::getLLVMModuleSet()->getSVFInstruction(I);
        const SVFFunction* F =SVFUtil::getCallee(svfInst);
        return F && F->getName().find("llvm.memset") != std::string::npos;
    }
 
    inline bool isMemcpy(const Instruction* I)
    {
        const SVFInstruction* svfInst = LLVMModuleSet::getLLVMModuleSet()->getSVFInstruction(I);
        const SVFFunction* F = SVFUtil::getCallee(svfInst);
        return F && F->getName().find("llvm.memcpy") != std::string::npos;
    }
 
private:
    MHP* mhp;
    LockAnalysis* lsa;
    InstSet loadset;
    InstSet storeset;
 
 
//  RCMemoryPartitioning mp;
 
public:
 
    u32_t numOfAllSt;
    u32_t numOfAllLd;
    u32_t numOfNonLocalSt;
    u32_t numOfNonLocalLd;
    u32_t numOfAliasSt;
    u32_t numOfAliasLd;
    u32_t numOfMHPSt;
    u32_t numOfMHPLd;
    u32_t numOfAnnotatedSt;
    u32_t numOfAnnotatedLd;
 
 
    static const u32_t ANNO_MHP= 0x04;
    static const u32_t ANNO_ALIAS= 0x02;
    static const u32_t ANNO_LOCAL= 0x01;
};
 
} // End namespace SVF
 
#endif /* MTAANNOTATOR_H_ */