2008-07-11 16:51:25 +00:00
|
|
|
/*
|
|
|
|
|
* DebugDirectory.cpp - Part of the PeLib library.
|
|
|
|
|
*
|
|
|
|
|
* Copyright (c) 2004 - 2005 Sebastian Porst (webmaster@the-interweb.com)
|
|
|
|
|
* All rights reserved.
|
|
|
|
|
*
|
|
|
|
|
* This software is licensed under the zlib/libpng License.
|
|
|
|
|
* For more details see http://www.opensource.org/licenses/zlib-license.php
|
|
|
|
|
* or the license information file (license.htm) in the root directory
|
|
|
|
|
* of PeLib.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include "PeLibInc.h"
|
|
|
|
|
#include "DebugDirectory.h"
|
|
|
|
|
|
|
|
|
|
namespace PeLib
|
|
|
|
|
{
|
|
|
|
|
void DebugDirectory::clear()
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo.clear();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::vector<PELIB_IMG_DEBUG_DIRECTORY> DebugDirectory::read(InputBuffer& ibBuffer, unsigned int uiSize)
|
|
|
|
|
{
|
|
|
|
|
std::vector<PELIB_IMG_DEBUG_DIRECTORY> currDebugInfo;
|
|
|
|
|
|
|
|
|
|
PELIB_IMG_DEBUG_DIRECTORY iddCurr;
|
|
|
|
|
|
|
|
|
|
for (unsigned int i=0;i<uiSize/PELIB_IMAGE_DEBUG_DIRECTORY::size();i++)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
ibBuffer >> iddCurr.idd.Characteristics;
|
|
|
|
|
ibBuffer >> iddCurr.idd.TimeDateStamp;
|
|
|
|
|
ibBuffer >> iddCurr.idd.MajorVersion;
|
|
|
|
|
ibBuffer >> iddCurr.idd.MinorVersion;
|
|
|
|
|
ibBuffer >> iddCurr.idd.Type;
|
|
|
|
|
ibBuffer >> iddCurr.idd.SizeOfData;
|
|
|
|
|
ibBuffer >> iddCurr.idd.AddressOfRawData;
|
|
|
|
|
ibBuffer >> iddCurr.idd.PointerToRawData;
|
|
|
|
|
|
|
|
|
|
currDebugInfo.push_back(iddCurr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return currDebugInfo;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int DebugDirectory::read(unsigned char* buffer, unsigned int buffersize)
|
|
|
|
|
{
|
|
|
|
|
// XXX: Note, debug data is not read at all. This might or might not change
|
|
|
|
|
// in the future.
|
|
|
|
|
|
|
|
|
|
std::vector<byte> vDebugDirectory(buffer, buffer + buffersize);
|
|
|
|
|
|
|
|
|
|
InputBuffer ibBuffer(vDebugDirectory);
|
|
|
|
|
|
|
|
|
|
std::vector<PELIB_IMG_DEBUG_DIRECTORY> currDebugInfo = read(ibBuffer, buffersize);
|
|
|
|
|
|
|
|
|
|
std::swap(currDebugInfo, m_vDebugInfo);
|
|
|
|
|
|
|
|
|
|
return NO_ERROR;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @param strFilename Name of the file which will be read.
|
|
|
|
|
* @param uiOffset File offset of the Debug directory.
|
|
|
|
|
* @param uiSize Size of the Debug directory.
|
|
|
|
|
**/
|
|
|
|
|
int DebugDirectory::read(const std::string& strFilename, unsigned int uiOffset, unsigned int uiSize)
|
|
|
|
|
{
|
|
|
|
|
std::ifstream ifFile(strFilename.c_str(), std::ios::binary);
|
|
|
|
|
unsigned int ulFileSize = fileSize(ifFile);
|
|
|
|
|
|
|
|
|
|
if (!ifFile)
|
|
|
|
|
{
|
|
|
|
|
return ERROR_OPENING_FILE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (ulFileSize < uiOffset + uiSize)
|
|
|
|
|
{
|
|
|
|
|
return ERROR_INVALID_FILE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ifFile.seekg(uiOffset, std::ios::beg);
|
|
|
|
|
|
|
|
|
|
std::vector<byte> vDebugDirectory(uiSize);
|
|
|
|
|
ifFile.read(reinterpret_cast<char*>(&vDebugDirectory[0]), uiSize);
|
|
|
|
|
|
|
|
|
|
InputBuffer ibBuffer(vDebugDirectory);
|
|
|
|
|
|
|
|
|
|
std::vector<PELIB_IMG_DEBUG_DIRECTORY> currDebugInfo = read(ibBuffer, uiSize);
|
|
|
|
|
|
|
|
|
|
for (unsigned int i=0;i<currDebugInfo.size();i++)
|
|
|
|
|
{
|
|
|
|
|
ifFile.seekg(currDebugInfo[i].idd.PointerToRawData, std::ios::beg);
|
|
|
|
|
currDebugInfo[i].data.resize(currDebugInfo[i].idd.SizeOfData);
|
|
|
|
|
ifFile.read(reinterpret_cast<char*>(&currDebugInfo[i].data[0]), currDebugInfo[i].idd.SizeOfData);
|
|
|
|
|
if (!ifFile) return ERROR_INVALID_FILE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::swap(currDebugInfo, m_vDebugInfo);
|
|
|
|
|
|
|
|
|
|
return NO_ERROR;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Rebuilds the current debug directory.
|
|
|
|
|
* @param vBuffer Buffer where the rebuilt directory is stored.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::rebuild(std::vector<byte>& vBuffer) const
|
|
|
|
|
{
|
|
|
|
|
OutputBuffer obBuffer(vBuffer);
|
|
|
|
|
|
|
|
|
|
for (unsigned int i=0;i<m_vDebugInfo.size();i++)
|
|
|
|
|
{
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.Characteristics;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.TimeDateStamp;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.MajorVersion;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.MinorVersion;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.Type;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.SizeOfData;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.AddressOfRawData;
|
|
|
|
|
obBuffer << m_vDebugInfo[i].idd.PointerToRawData;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @return Size of the debug directory.
|
|
|
|
|
**/
|
|
|
|
|
unsigned int DebugDirectory::size() const
|
|
|
|
|
{
|
|
|
|
|
return static_cast<unsigned int>(m_vDebugInfo.size()) * PELIB_IMAGE_DEBUG_DIRECTORY::size();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @param strFilename Name of the file which will be written.
|
|
|
|
|
* @param uiOffset File offset where the debug directory will be stored.
|
|
|
|
|
**/
|
|
|
|
|
int DebugDirectory::write(const std::string& strFilename, unsigned int uiOffset) const
|
|
|
|
|
{
|
|
|
|
|
std::fstream ofFile(strFilename.c_str(), std::ios_base::in);
|
|
|
|
|
|
|
|
|
|
if (!ofFile)
|
|
|
|
|
{
|
|
|
|
|
ofFile.clear();
|
|
|
|
|
ofFile.open(strFilename.c_str(), std::ios_base::out | std::ios_base::binary);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
ofFile.close();
|
|
|
|
|
ofFile.open(strFilename.c_str(), std::ios_base::in | std::ios_base::out | std::ios_base::binary);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!ofFile)
|
|
|
|
|
{
|
|
|
|
|
return ERROR_OPENING_FILE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ofFile.seekp(uiOffset, std::ios::beg);
|
|
|
|
|
|
|
|
|
|
std::vector<unsigned char> vBuffer;
|
|
|
|
|
rebuild(vBuffer);
|
|
|
|
|
|
|
|
|
|
ofFile.write(reinterpret_cast<const char*>(&vBuffer[0]), static_cast<unsigned int>(vBuffer.size()));
|
|
|
|
|
|
|
|
|
|
ofFile.close();
|
|
|
|
|
|
|
|
|
|
return NO_ERROR;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @return Number of debug structures in the current Debug directory.
|
|
|
|
|
**/
|
|
|
|
|
unsigned int DebugDirectory::calcNumberOfEntries() const
|
|
|
|
|
{
|
|
|
|
|
return static_cast<unsigned int>(m_vDebugInfo.size());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Adds a new debug structure to the debug directory. The initial values of all members of the structure
|
|
|
|
|
* are undefined.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::addEntry()
|
|
|
|
|
{
|
|
|
|
|
PELIB_IMG_DEBUG_DIRECTORY p;
|
|
|
|
|
m_vDebugInfo.push_back(p);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Removes a debug structure from the current debug directory. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::removeEntry(unsigned int uiIndex)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo.erase(m_vDebugInfo.begin() + uiIndex);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the Characteristics value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return Characteristics value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
dword DebugDirectory::getCharacteristics(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.Characteristics;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the TimeDateStamp value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return TimeDateStamp value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
dword DebugDirectory::getTimeDateStamp(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.TimeDateStamp;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the MajorVersion value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return MajorVersion value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
word DebugDirectory::getMajorVersion(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.MajorVersion;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the MinorVersion value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return MinorVersion value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
word DebugDirectory::getMinorVersion(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.MinorVersion;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the Type value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return Type value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
dword DebugDirectory::getType(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.Type;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the SizeOfData value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return SizeOfData value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
dword DebugDirectory::getSizeOfData(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.SizeOfData;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the AddressOfRawData value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return AddressOfRawData value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
dword DebugDirectory::getAddressOfRawData(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.AddressOfRawData;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Returns the PointerToRawData value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @return PointerToRawData value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
dword DebugDirectory::getPointerToRawData(unsigned int uiIndex) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[uiIndex].idd.PointerToRawData;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::vector<byte> DebugDirectory::getData(unsigned int index) const
|
|
|
|
|
{
|
|
|
|
|
return m_vDebugInfo[index].data;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the Characteristics value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param dwValue New value of the Characteristics value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setCharacteristics(unsigned int uiIndex, dword dwValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.Characteristics = dwValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the TimeDateStamp value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param dwValue New value of the TimeDateStamp value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setTimeDateStamp(unsigned int uiIndex, dword dwValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.TimeDateStamp = dwValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the MajorVersion value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param wValue New value of the MajorVersion value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setMajorVersion(unsigned int uiIndex, word wValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.MajorVersion = wValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the MinorVersion value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param wValue New value of the MinorVersion value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setMinorVersion(unsigned int uiIndex, word wValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.MinorVersion = wValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the Type value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param dwValue New value of the Type value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setType(unsigned int uiIndex, dword dwValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.Type = dwValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the SizeOfData value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param dwValue New value of the SizeOfData value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setSizeOfData(unsigned int uiIndex, dword dwValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.SizeOfData = dwValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the AddressOfRawData value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param dwValue New value of the AddressOfRawData value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setAddressOfRawData(unsigned int uiIndex, dword dwValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.AddressOfRawData = dwValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Changes the PointerToRawData value of a debug structure. If an invalid structure is specified
|
|
|
|
|
* by the parameter uiIndex the result will be undefined behaviour.
|
|
|
|
|
* @param uiIndex Identifies the debug structure.
|
|
|
|
|
* @param dwValue New value of the PointerToRawData value of the debug structure.
|
|
|
|
|
**/
|
|
|
|
|
void DebugDirectory::setPointerToRawData(unsigned int uiIndex, dword dwValue)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[uiIndex].idd.PointerToRawData = dwValue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void DebugDirectory::setData(unsigned int index, const std::vector<byte>& data)
|
|
|
|
|
{
|
|
|
|
|
m_vDebugInfo[index].data = data;
|
|
|
|
|
}
|
|
|
|
|
}
|
