AsmUnparser.h

Go to the documentation of this file.

#ifndef ROSE_AsmUnparser_H
#define ROSE_AsmUnparser_H

#include <ostream>

#include "callbacks.h"          /* Needed for ROSE_Callbacks::List<> */
#include "BinaryControlFlow.h"

class SgAsmInstruction;
class SgAsmBlock;
class SgAsmFunction;
class SgAsmInterpretation;


class AsmUnparser {
public:
    enum Organization {
        ORGANIZED_BY_AST,               
        ORGANIZED_BY_ADDRESS            
    };

    typedef BinaryAnalysis::ControlFlow::Graph CFG;
    typedef boost::graph_traits<CFG>::vertex_descriptor CFG_Vertex;
    typedef std::map<SgAsmBlock*, CFG_Vertex> CFG_BlockMap;

    class UnparserCallback {
    public:
        struct GeneralArgs {
            GeneralArgs(AsmUnparser *unparser, std::ostream &output)
                : unparser(unparser), output(output) {}
            AsmUnparser *unparser;      
            std::ostream &output;       
        };
            
        struct InsnArgs: public GeneralArgs {
            InsnArgs(AsmUnparser *unparser, std::ostream &output, SgAsmInstruction *insn, size_t position_in_block)
                : GeneralArgs(unparser, output), insn(insn), position_in_block(position_in_block) {}
            SgAsmInstruction *get_node() const { return insn; } 
            SgAsmInstruction *insn;     
            size_t position_in_block;   
        };

        struct BasicBlockArgs: public GeneralArgs {
            BasicBlockArgs(AsmUnparser *unparser, std::ostream &output, SgAsmBlock *block,
                           const std::vector<SgAsmInstruction*> &insns)
                : GeneralArgs(unparser, output), block(block), insns(insns) {}
            SgAsmBlock *get_node() const { return block; } 
            SgAsmBlock *block;          
            const std::vector<SgAsmInstruction*> &insns; 
        };

        struct StaticDataArgs: public GeneralArgs {
            StaticDataArgs(AsmUnparser *unparser, std::ostream &output, SgAsmStaticData *data, size_t position_in_block)
                : GeneralArgs(unparser, output), data(data), position_in_block(position_in_block) {}
            SgAsmStaticData *get_node() const { return data; } 
            SgAsmStaticData *data;      
            size_t position_in_block;   
        };

        struct DataBlockArgs: public GeneralArgs {
            DataBlockArgs(AsmUnparser *unparser, std::ostream &output, SgAsmBlock *block,
                          const std::vector<SgAsmStaticData*> &datalist)
                : GeneralArgs(unparser, output), block(block), datalist(datalist) {}
            SgAsmBlock *get_node() const { return block; } 
            SgAsmBlock *block;          
            const std::vector<SgAsmStaticData*> &datalist; 
        };

        struct FunctionArgs: public GeneralArgs {
            FunctionArgs(AsmUnparser *unparser, std::ostream &output, SgAsmFunction *func)
                : GeneralArgs(unparser, output), func(func) {}
            SgAsmFunction *get_node() const { return func; } 
            SgAsmFunction *func;        
        };

        struct InterpretationArgs: public GeneralArgs {
            InterpretationArgs(AsmUnparser *unparser, std::ostream &output, SgAsmInterpretation *interp)
                : GeneralArgs(unparser, output), interp(interp) {}
            SgAsmInterpretation *get_node() const { return interp; } 
            SgAsmInterpretation *interp; 
        };
        
        virtual ~UnparserCallback() {}

        virtual bool operator()(bool enabled, const InsnArgs&)           { abort(); }
        virtual bool operator()(bool enabled, const BasicBlockArgs&)     { abort(); }
        virtual bool operator()(bool enabled, const StaticDataArgs&)     { abort(); }
        virtual bool operator()(bool enabled, const DataBlockArgs&)      { abort(); }
        virtual bool operator()(bool enabled, const FunctionArgs&)       { abort(); }
        virtual bool operator()(bool enabled, const InterpretationArgs&) { abort(); }
    };

    /**************************************************************************************************************************
     *                                  Instruction Callbacks
     **************************************************************************************************************************/

    class InsnBlockSeparation: public UnparserCallback {
    public:
        SgAsmBlock *prev_block;
        InsnBlockSeparation(): prev_block(NULL) {}
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnFuncEntry: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnAddress: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnRawBytes: public UnparserCallback {
    public:
        HexdumpFormat fmt;
        InsnRawBytes() {
            fmt.width = 8;              /* Max instruction bytes per line of output. */
            fmt.pad_chars = true;       /* Show ASCII characters as well as bytes. */
        }
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnBlockEntry: public UnparserCallback {
    public:
        bool show_function;             
        bool show_reasons;              
        InsnBlockEntry(): show_function(true), show_reasons(true) {}
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnBody: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnNoEffect: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnComment: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    class InsnLineTermination: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InsnArgs &args);
    };

    /**************************************************************************************************************************
     *                                  Basic Block Callbacks
     **************************************************************************************************************************/

    class BasicBlockReasons: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockPredecessors: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockNoopUpdater: public UnparserCallback {
    public:
        bool debug;             
        BasicBlockNoopUpdater(): debug(false) {}
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockNoopWarning: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockBody: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockSuccessors: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockLineTermination: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const BasicBlockArgs &args);
    };

    class BasicBlockCleanup: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled/*ignored*/, const BasicBlockArgs &args);
    };

    /**************************************************************************************************************************
     *                                  Static Data Callbacks
     **************************************************************************************************************************/

    class StaticDataBlockSeparation: public UnparserCallback {
    public:
        SgAsmBlock *prev_block;
        StaticDataBlockSeparation(): prev_block(NULL) {}
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    class StaticDataRawBytes: public UnparserCallback {
    public:
        bool show_address;              /* Should we show the address in the left margin? */
        bool show_offset;               /* If show_address is true, then should we display START+OFFSET or just an address? */

        HexdumpFormat fmt;
        StaticDataRawBytes() {
            show_address = true;
            show_offset = false;
            fmt.prefix = NULL;          /* Adjusted by the callback each time called. */
            fmt.multiline = false;      /* Do not emit prefix before first line and linefeed after last line. */
            fmt.width = 8;              /* Max data bytes per line of output. */
            fmt.pad_chars = true;       /* Show ASCII characters as well as bytes. */
        }
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    class StaticDataBlockEntry: public UnparserCallback {
    public:
        bool show_function;             
        bool show_reasons;              
        StaticDataBlockEntry(): show_function(true), show_reasons(true) {}
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    class StaticDataDetails: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    class StaticDataComment: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    class StaticDataLineTermination: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    /**************************************************************************************************************************
     *                                  Data Block Callbacks
     **************************************************************************************************************************/

    class DataBlockTitle: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const StaticDataArgs &args);
    };

    class DataBlockBody: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const DataBlockArgs &args);
    };

    class DataBlockLineTermination: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const DataBlockArgs &args);
    };

    /**************************************************************************************************************************
     *                                  Function Callbacks
     **************************************************************************************************************************/

    class FunctionEntryAddress: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionSeparator: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionReasons: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionName: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionLineTermination: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionComment: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionAttributes: public UnparserCallback {
    public:
        std::string prefix;
        FunctionAttributes();
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    class FunctionBody: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const FunctionArgs &args);
    };

    /**************************************************************************************************************************
     *                                  Interpretation Callbacks
     **************************************************************************************************************************/

    class InterpName: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InterpretationArgs &args);
    };

    class InterpBody: public UnparserCallback {
    public:
        virtual bool operator()(bool enabled, const InterpretationArgs &args);
    };

    /**************************************************************************************************************************
     *                                  Built-in Callbacks
     * Don't forget to modify init() to actually add these to callback lists if desired.
     **************************************************************************************************************************/

    InsnBlockSeparation insnBlockSeparation;
    InsnFuncEntry insnFuncEntry;
    InsnAddress insnAddress;
    InsnRawBytes insnRawBytes;
    InsnBlockEntry insnBlockEntry;
    InsnBody insnBody;
    InsnNoEffect insnNoEffect;
    InsnComment insnComment;
    InsnLineTermination insnLineTermination;

    BasicBlockReasons basicBlockReasons;
    BasicBlockPredecessors basicBlockPredecessors;
    BasicBlockNoopUpdater basicBlockNoopUpdater;
    BasicBlockNoopWarning basicBlockNoopWarning;
    BasicBlockBody basicBlockBody;
    BasicBlockSuccessors basicBlockSuccessors;
    BasicBlockLineTermination basicBlockLineTermination;
    BasicBlockCleanup basicBlockCleanup;

    StaticDataBlockSeparation staticDataBlockSeparation;
    StaticDataRawBytes staticDataRawBytes;
    StaticDataBlockEntry staticDataBlockEntry;
    StaticDataDetails staticDataDetails;
    StaticDataComment staticDataComment;
    StaticDataLineTermination staticDataLineTermination;

    DataBlockBody dataBlockBody;
    DataBlockLineTermination dataBlockLineTermination;

    FunctionEntryAddress functionEntryAddress;
    FunctionSeparator functionSeparator;
    FunctionReasons functionReasons;
    FunctionName functionName;
    FunctionLineTermination functionLineTermination;
    FunctionComment functionComment;
    FunctionAttributes functionAttributes;
    FunctionBody functionBody;

    InterpName interpName;
    InterpBody interpBody;
    /**************************************************************************************************************************
     *                                  Public methods
     **************************************************************************************************************************/

    AsmUnparser() {
        init();
    }

    virtual ~AsmUnparser() {}

    virtual Organization get_organization() const { return organization; }
    virtual void set_organization(Organization organization) { this->organization = organization; }
    virtual bool is_unparsable_node(SgNode *node);

    virtual SgNode *find_unparsable_node(SgNode *ast);

    virtual std::vector<SgNode*> find_unparsable_nodes(SgNode *ast);
    
    std::vector<bool> insn_is_noop;

    virtual size_t unparse(std::ostream&, SgNode *ast);

    virtual bool unparse_one_node(std::ostream&, SgNode*);

    virtual bool unparse_insn(bool enabled, std::ostream&, SgAsmInstruction*, size_t position_in_block=(size_t)-1);
    virtual bool unparse_basicblock(bool enabled, std::ostream&, SgAsmBlock*);
    virtual bool unparse_staticdata(bool enabled, std::ostream&, SgAsmStaticData*, size_t position_in_block=(size_t)-1);
    virtual bool unparse_datablock(bool enabled, std::ostream&, SgAsmBlock*);
    virtual bool unparse_function(bool enabled, std::ostream&, SgAsmFunction*);
    virtual bool unparse_interpretation(bool enabled, std::ostream&, SgAsmInterpretation*);
    typedef std::map<uint64_t, std::string> LabelMap;

    void add_function_labels(SgNode *ast);

    void add_control_flow_graph(const BinaryAnalysis::ControlFlow::Graph &cfg);

protected:
    struct CallbackLists {
        ROSE_Callbacks::List<UnparserCallback> unparse;                 
        ROSE_Callbacks::List<UnparserCallback> pre;                     
        ROSE_Callbacks::List<UnparserCallback> post;                    
        void clear() {
            unparse.clear();
            pre.clear();
            post.clear();
        }
    };

    CallbackLists insn_callbacks;                       
    CallbackLists basicblock_callbacks;                 
    CallbackLists staticdata_callbacks;                 
    CallbackLists datablock_callbacks;                  
    CallbackLists function_callbacks;                   
    CallbackLists interp_callbacks;                     
    LabelMap labels;

    Organization organization;

    virtual void init();

    CFG cfg;

    CFG_BlockMap cfg_blockmap;
};

#endif

---