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