ControlFlow.h

Go to the documentation of this file.

/*
 * Copyright (C) 2022 The V-Gears Team
 *
 * This file is part of V-Gears
 *
 * V-Gears is free software: you can redistribute it and/or modify it under
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, version 3.0 (GPLv3) of the License.
 *
 * V-Gears 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 General Public License for more details.
 */
 
#pragma once
 
#include "Engine.h"
#include "Graph.h"
 
class ControlFlow {
 
    public:
 
        ControlFlow(InstVec& insts, Engine& engine);
 
        ControlFlow(const ControlFlow& control_flow) = delete;
 
        ControlFlow& operator = (const ControlFlow& control_flow) = delete;
 
        const Graph& GetGraph() const;
 
        void CreateGroups();
 
        const Graph& Analyze();
 
    private:
 
        Graph graph_;
 
        Engine& engine_;
 
        InstVec &insts_;
 
        std::map<uint32, GraphVertex> addr_map_;
 
        GraphVertex Find(const InstPtr inst);
 
        GraphVertex Find(ConstInstIterator it);
 
        GraphVertex Find(uint32 address);
 
        void Merge(GraphVertex graph_1, GraphVertex graph_2);
 
        void SetStackLevel(GraphVertex graph, int level);
 
        void DetectShortCircuit();
 
        void DetectWhile();
 
        void DetectDoWhile();
 
        void DetectBreak();
 
        void DetectContinue();
 
        bool ValidateBreakOrContinue(GroupPtr group, GroupPtr condition_group);
 
        void DetectIf();
 
        void DetectElse();
 
        bool ValidateElseBlock(GroupPtr if_group, GroupPtr start, GroupPtr end);
 
};