Assembler.h

#ifndef ROSE_BinaryAnalysis_Assembler_H
#define ROSE_BinaryAnalysis_Assembler_H
 
#include <featureTests.h>
#ifdef ROSE_ENABLE_BINARY_ANALYSIS
 
#include <Rose/Exception.h>
 
namespace Rose {
namespace BinaryAnalysis {
 
class Assembler {
public:
    class Exception: public Rose::Exception {
    public:
        Exception(const std::string &reason, SgAsmInstruction *insn)
            : Rose::Exception(reason), insn(insn)
            {}
        Exception(const std::string &reason)
            : Rose::Exception(reason), insn(NULL)
            {}
        void print(std::ostream&) const;
        friend std::ostream& operator<<(std::ostream&, const Exception&);
 
        SgAsmInstruction *insn;         
    };
 
    enum EncodingType
        {
        ET_SHORTEST,            
        ET_LONGEST,             
        ET_MATCHES             
    };
 
    Assembler()
        : p_debug(NULL), p_encoding_type(ET_SHORTEST)
        {}
 
    virtual ~Assembler() {}
 
    static Assembler *create(SgAsmInterpretation *interp);
 
    static Assembler *create(SgAsmGenericHeader*);
 
    /*==========================================================================================================================
     * Main public assembly methods
     *========================================================================================================================== */
public:
    virtual SgUnsignedCharList assembleOne(SgAsmInstruction *insn) = 0;
 
    SgUnsignedCharList assembleBlock(SgAsmBlock*);
 
    SgUnsignedCharList assembleBlock(const std::vector<SgAsmInstruction*> &insns, rose_addr_t starting_rva);
 
    virtual SgUnsignedCharList assembleProgram(const std::string &source) = 0;
 
    /*==========================================================================================================================
     * Assembler properties and settings
     *========================================================================================================================== */
public:
    void set_encoding_type(EncodingType et) {
        p_encoding_type = et;
    }
 
    EncodingType get_encoding_type() const {
        return p_encoding_type;
    }
 
    void set_debug(FILE *f) {
        p_debug = f;
    }
 
    FILE *get_debug() const {
        return p_debug;
    }
 
    /*==========================================================================================================================
     * Data members
     *========================================================================================================================== */
protected:
    FILE *p_debug;                                      
    EncodingType p_encoding_type;                       
};
 
} // namespace
} // namespace
 
#endif
#endif