Loader.h

Go to the documentation of this file.

#ifndef ROSE_LOADER_H
#define ROSE_LOADER_H

/* Increase ADDR if necessary to make it a multiple of ALMNT */
#define ALIGN_UP(ADDR,ALMNT)       ((((ADDR)+(ALMNT)-1)/(ALMNT))*(ALMNT))
#define ALIGN_DN(ADDR,ALMNT)       (((ADDR)/(ALMNT))*(ALMNT))


class Loader {
public:
    Loader(): p_debug(NULL) { ctor(); }
    virtual ~Loader() {}

    /*------------------------------------------------------------------------------------------------------------------------
     * Methods for registering and looking up loader implementations.
     *------------------------------------------------------------------------------------------------------------------------*/
public:

    static void register_subclass(Loader*);

    static Loader *find_loader(SgAsmGenericHeader*);

    virtual bool can_handle(SgAsmGenericHeader*) { return true; }

    static void initclass();

    /*------------------------------------------------------------------------------------------------------------------------
     * High-level functions for mapping.
     *------------------------------------------------------------------------------------------------------------------------*/
public:

    virtual MemoryMap *map_all_sections(MemoryMap *map, SgAsmGenericFile *file, bool allow_overmap=true) {
        return map_all_sections(map, file->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_all_sections(MemoryMap *map, SgAsmGenericHeader *fhdr, bool allow_overmap=true) {
        return map_all_sections(map, fhdr->get_sections()->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_all_sections(MemoryMap*, const SgAsmGenericSectionPtrList &sections, bool allow_overmap=true);



    virtual MemoryMap *map_code_sections(MemoryMap *map, SgAsmGenericFile *file, bool allow_overmap=true) {
        return map_code_sections(map, file->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_code_sections(MemoryMap *map, SgAsmGenericHeader *fhdr, bool allow_overmap=true) {
        return map_code_sections(map, fhdr->get_sections()->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_code_sections(MemoryMap*, const SgAsmGenericSectionPtrList &sections, bool allow_overmap=true);



    virtual MemoryMap *map_executable_sections(MemoryMap *map, SgAsmGenericFile *file, bool allow_overmap=true) {
        return map_executable_sections(map, file->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_executable_sections(MemoryMap *map, SgAsmGenericHeader *fhdr, bool allow_overmap=true) {
        return map_executable_sections(map, fhdr->get_sections()->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_executable_sections(MemoryMap*, const SgAsmGenericSectionPtrList &sections, bool allow_overmap=true);



    virtual MemoryMap *map_writable_sections(MemoryMap *map, SgAsmGenericFile *file, bool allow_overmap=true) {
        return map_writable_sections(map, file->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_writable_sections(MemoryMap *map, SgAsmGenericHeader *fhdr, bool allow_overmap=true) {
        return map_writable_sections(map, fhdr->get_sections()->get_sections(), allow_overmap);
    }

    virtual MemoryMap *map_writable_sections(MemoryMap*, const SgAsmGenericSectionPtrList &sections, bool allow_overmap=true);

    /*------------------------------------------------------------------------------------------------------------------------
     * Selectors
     *------------------------------------------------------------------------------------------------------------------------*/
public:

    enum Contribution {
        CONTRIBUTE_NONE,                
        CONTRIBUTE_ADD,                 
        CONTRIBUTE_SUB                  
    };

    class Selector {
    public:
        virtual ~Selector() {}
        virtual Contribution contributes(SgAsmGenericSection*) = 0;
    };

    /*------------------------------------------------------------------------------------------------------------------------
     * Helper methods. These are declared virtual but are often not overridden by subclasses.
     *------------------------------------------------------------------------------------------------------------------------*/
public:

    virtual rose_addr_t align_values(SgAsmGenericSection*, Contribution,
                                     rose_addr_t *va,     rose_addr_t *mem_size,
                                     rose_addr_t *offset, rose_addr_t *file_size,
                                     const MemoryMap *current);

    virtual SgAsmGenericSectionPtrList order_sections(const SgAsmGenericSectionPtrList&);

    virtual MemoryMap *create_map(MemoryMap *map, const SgAsmGenericSectionPtrList&, Selector*, bool allow_overmap=true);

    /*------------------------------------------------------------------------------------------------------------------------
     * Configuration methods.
     *------------------------------------------------------------------------------------------------------------------------*/

    void set_debug(FILE *f) { p_debug=f; }

    FILE *get_debug() const { return p_debug; }

private:
    void ctor();
    static std::vector<Loader*> p_registered;
    FILE *p_debug;
};

#endif

---