DataDirEntryWrapper.cpp

Go to the documentation of this file.

#include "pe/DataDirEntryWrapper.h"
#include "pe/PEFile.h"
 
DataDirEntryWrapper::DataDirEntryWrapper(PEFile* pe, pe:: dir_entry v_entryType)
    :  PENodeWrapper(pe), entryType(v_entryType)
{
    wrap();
}
 
IMAGE_DATA_DIRECTORY* DataDirEntryWrapper::getDataDirectory()
{
 
    if (m_PE == NULL) return NULL;
 
    IMAGE_DATA_DIRECTORY *d = m_PE->getDataDirectory();
    return d;
}
 
offset_t DataDirEntryWrapper::getDirEntryAddress()
{
    if (this->entryType >= pe::DIR_ENTRIES_COUNT) return INVALID_ADDR;
 
    IMAGE_DATA_DIRECTORY *d = getDataDirectory();
    if (!d) return INVALID_ADDR;
 
    offset_t rva = static_cast<offset_t>(d[this->entryType].VirtualAddress);
    if (rva == 0) return INVALID_ADDR;
    return rva;
}
 
bufsize_t DataDirEntryWrapper::getDirEntrySize()
{
    if (this->entryType >= pe::DIR_ENTRIES_COUNT) return 0;
 
    IMAGE_DATA_DIRECTORY *d = getDataDirectory();
    if (!d) return 0;
 
    bufsize_t size = static_cast<bufsize_t>(d[this->entryType].Size);
    return size;
}