ResourceDirWrapper.h

Go to the documentation of this file.

#pragma once
 
#include "DataDirEntryWrapper.h"
#include "ResourceLeafWrapper.h"
 
#include "rsrc/ResourcesAlbum.h"
#include "rsrc/ResourceContentFactory.h"
 
#include <map>
#include <vector>
 
 
class ResourceDirWrapper : public DataDirEntryWrapper
{
public:
 
    enum ResourceDirFID {
        NONE = FIELD_NONE,
        CHARACTERISTIC,
        TIMESTAMP,
        MAJOR_VER,
        MINOR_VER,
        NAMED_ENTRIES_NUM,
        ID_ENTRIES_NUM,
        FIELD_COUNTER
    };
 
    ResourceDirWrapper(PEFile* pe, ResourcesAlbum *resAlbum = NULL, offset_t rawOffset = 0, long depth = 0, long topEntryId = TOP_ENTRY_ROOT)
        : DataDirEntryWrapper(pe, pe::DIR_RESOURCE),
        rawOff(rawOffset), dirDepth(depth), album(resAlbum), topEntryID(topEntryId)
    { 
        wrap();
    }
 
    bool wrap();
 
    virtual void* getPtr() { return resourceDir(); }
    virtual bufsize_t getSize();
 
    virtual QString getName() { return "Resources"; }
    virtual size_t getFieldsCount() { return FIELD_COUNTER; }
    virtual size_t getSubFieldsCount() { return 1; }
 
    virtual void* getFieldPtr(size_t fieldId, size_t subField);
    virtual bufsize_t getFieldSize(size_t fieldId, size_t subField);
    virtual QString getFieldName(size_t fieldId);
 
    bufsize_t getEntriesAreaSize()
    { 
        return static_cast<bufsize_t>(this->getEntriesCount()) * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY);
    }
    
    long getDepth() { return this->dirDepth; }
 
    IMAGE_RESOURCE_DIRECTORY* mainResourceDir();
    ResourcesAlbum* getAlbumPtr() { return album; }
 
private:
    IMAGE_RESOURCE_DIRECTORY* resourceDir();
    offset_t rawOff;
    long dirDepth;
    ResourcesAlbum *album;
    long topEntryID;
};
 
class ResourceEntryWrapper : public PENodeWrapper
{
public:
    // fields :
    enum FieldID {
        NONE = FIELD_NONE,
        NAME_ID_ADDR,
        OFFSET_TO_DATA,
        FIELD_COUNTER
    };
 
    static QString translateType(WORD id);
 
    ResourceEntryWrapper(PEFile *pe, ResourceDirWrapper *parentDir, size_t entryNumber, long topEntryId, ResourcesAlbum *resAlbum)
        : PENodeWrapper(pe, parentDir, entryNumber), topEntryID(topEntryId), album(resAlbum), childDir(NULL), childLeaf(NULL)
    {
        this->parentDir = parentDir;
        wrap();
    }
 
    virtual ~ResourceEntryWrapper() { clear(); }
 
    bool wrap();
    // full structure boundaries
    virtual void* getPtr() { return getEntryPtr(); }
    virtual bufsize_t getSize() { return sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY); }
 
    virtual QString getName() { return "Resource entry: "+ translateType(getID());}
    virtual size_t getFieldsCount() { return FIELD_COUNTER; }
    virtual size_t getSubFieldsCount() { return 1; }
 
    // specific field boundaries
    virtual void* getFieldPtr(size_t fieldId, size_t subField = FIELD_NONE);
    virtual QString getFieldName(size_t fieldId);
    virtual Executable::addr_type containsAddrType(size_t fieldId, size_t subField) { return Executable::NOT_ADDR; }
 
    bool isByName();
    bool isDir();
 
    WORD getID();
    offset_t getChildOffsetToDirectory(); //relative offset!
    offset_t getNameOffset();
 
    IMAGE_RESOURCE_DIRECTORY_STRING* getNameStr();
    offset_t getChildAddress();
 
    IMAGE_RESOURCE_DIRECTORY_ENTRY *getEntryPtr();
    ResourcesAlbum* getAlbumPtr() { return album; }
    long getTopEntryID() { return topEntryID; }
 
protected:
    virtual void clear();
 
private:
    long topEntryID;
    ResourcesAlbum *album;
 
    ResourceDirWrapper* parentDir, *childDir;
    ResourceLeafWrapper *childLeaf;
};