CDGBuilder.h

Go to the documentation of this file.

//===----- CDGBuilder.h -- Control Dependence Graph Builder -------------//
//
//                     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/>.
//
//===----------------------------------------------------------------------===//
 
/*
 * CDGBuilder.h
 *
 *  Created on: Sep 27, 2023
 *      Author: Xiao Cheng
 */
#ifndef SVF_CDGBUILDER_H
#define SVF_CDGBUILDER_H
 
#include "Graphs/CDG.h"
#include "SVFIR/SVFValue.h"
 
// control dependence builder
namespace SVF
{
class CDGBuilder
{
public:
 
    CDGBuilder() : _controlDG(CDG::getCDG())
    {
 
    }
 
    ~CDGBuilder()
    {
 
    }
 
    void build();
 
    void buildControlDependence(const SVFModule *svfgModule);
 
    void buildICFGNodeControlMap();
 
 
 
private:
 
 
    static void
    extractBBS(const SVFFunction *func,
               Map<const SVFBasicBlock *, std::vector<const SVFBasicBlock *>> &res);
 
    void
    extractNodesBetweenPdomNodes(const SVFBasicBlock *succ, const SVFBasicBlock *LCA,
                                 std::vector<const SVFBasicBlock *> &tgtNodes);
 
 
    void dfsNodesBetweenPdomNodes(const SVFBasicBlock *cur,
                                  const SVFBasicBlock *tgt,
                                  std::vector<const SVFBasicBlock *> &path,
                                  std::vector<const SVFBasicBlock *> &tgtNodes,
                                  SVFLoopAndDomInfo *ld);
 
 
    s64_t getBBSuccessorBranchID(const SVFBasicBlock *BB, const SVFBasicBlock *Succ);
 
 
    inline void updateMap(const SVFBasicBlock *pred, const SVFBasicBlock *bb, s32_t pos)
    {
        _svfcontrolMap[pred][bb].insert(pos);
        _svfdependentOnMap[bb][pred].insert(pos);
    }
 
 
private:
    CDG *_controlDG;
    Map<const SVFBasicBlock *, Map<const SVFBasicBlock *, Set<s32_t>>> _svfcontrolMap; 
    Map<const SVFBasicBlock *, Map<const SVFBasicBlock *, Set<s32_t>>> _svfdependentOnMap; 
    Map<const ICFGNode *, Map<const ICFGNode *, Set<s32_t>>> _nodeControlMap; 
    Map<const ICFGNode *, Map<const ICFGNode *, Set<s32_t>>> _nodeDependentOnMap; 
};
}
 
 
#endif //SVF_CDGBUILDER_H