// This function writes into a given place in memory (pbRsrcSec) the edited resource section
void CResourceEditor::WriteRsrcSec(BYTE* pbRsrcSec) {
BYTE* seeker = pbRsrcSec;
queue<CResourceDirectory*> qDirs; // Used to scan the tree by level
queue<CResourceDataEntry*> qDataEntries; // Used for writing the data entries
queue<CResourceDataEntry*> qDataEntries2; // Used for writing raw resources data
queue<CResourceDirectoryEntry*> qStrings; // Used for writing resources' names
qDirs.push(m_cResDir);
while (!qDirs.empty()) {
CResourceDirectory* crd = qDirs.front();
IMAGE_RESOURCE_DIRECTORY rdDir = crd->GetInfo();
rdDir.NumberOfNamedEntries = ConvertEndianness(rdDir.NumberOfNamedEntries);
rdDir.NumberOfIdEntries = ConvertEndianness(rdDir.NumberOfIdEntries);
CopyMemory(seeker, &rdDir, sizeof(IMAGE_RESOURCE_DIRECTORY));
crd->m_ulWrittenAt = (ULONG_PTR)(seeker);
seeker += sizeof(IMAGE_RESOURCE_DIRECTORY);
for (unsigned int i = 0; i < crd->CountEntries(); i++) {
if (crd->GetEntry(i)->HasName())
qStrings.push(crd->GetEntry(i));
if (crd->GetEntry(i)->IsDataDirectory())
qDirs.push(crd->GetEntry(i)->GetSubDirectory());
else {
qDataEntries.push(crd->GetEntry(i)->GetDataEntry());
qDataEntries2.push(crd->GetEntry(i)->GetDataEntry());
}
MY_IMAGE_RESOURCE_DIRECTORY_ENTRY rDirE;
ZeroMemory(&rDirE, sizeof(rDirE));
rDirE.UOffset.DirectoryOffset.DataIsDirectory = crd->GetEntry(i)->IsDataDirectory();
rDirE.UName.Id = crd->GetEntry(i)->HasName() ? 0 : crd->GetEntry(i)->GetId();
rDirE.UName.Id = ConvertEndianness(rDirE.UName.Id);
rDirE.UName.NameString.NameIsString = (crd->GetEntry(i)->HasName()) ? 1 : 0;
CopyMemory(seeker, &rDirE, sizeof(MY_IMAGE_RESOURCE_DIRECTORY_ENTRY));
crd->GetEntry(i)->m_ulWrittenAt = (ULONG_PTR)(seeker);
seeker += sizeof(MY_IMAGE_RESOURCE_DIRECTORY_ENTRY);
}
qDirs.pop();
}
/*
* Write IMAGE_RESOURCE_DATA_ENTRYs.
*/
while (!qDataEntries.empty()) {
CResourceDataEntry* cRDataE = qDataEntries.front();
IMAGE_RESOURCE_DATA_ENTRY rDataE = {0,};
rDataE.CodePage = ConvertEndianness(cRDataE->GetCodePage());
rDataE.Size = ConvertEndianness(cRDataE->GetSize());
CopyMemory(seeker, &rDataE, sizeof(IMAGE_RESOURCE_DATA_ENTRY));
cRDataE->m_ulWrittenAt = (ULONG_PTR)(seeker);
seeker += sizeof(IMAGE_RESOURCE_DATA_ENTRY);
qDataEntries.pop();
}
/*
* Write strings
*/
while (!qStrings.empty()) {
CResourceDirectoryEntry* cRDirE = qStrings.front();
PMY_IMAGE_RESOURCE_DIRECTORY_ENTRY(cRDirE->m_ulWrittenAt)->UName.NameString.NameOffset = ConvertEndianness((DWORD) (seeker - pbRsrcSec));
const WINWCHAR* szName = cRDirE->GetName();
WORD iLen = (WORD) WinWStrLen(szName) + 1;
*(WORD*)seeker = ConvertEndianness(iLen);
CopyMemory(seeker + sizeof(WORD), szName, iLen*sizeof(WINWCHAR));
seeker += RALIGN(iLen * sizeof(WINWCHAR) + sizeof(WORD), 4);
qStrings.pop();
}
/*
* Write raw resource data and set offsets in IMAGE_RESOURCE_DATA_ENTRYs.
*/
while (!qDataEntries2.empty()) {
CResourceDataEntry* cRDataE = qDataEntries2.front();
CopyMemory(seeker, cRDataE->GetData(), cRDataE->GetSize());
PIMAGE_RESOURCE_DATA_ENTRY(cRDataE->m_ulWrittenAt)->OffsetToData = ConvertEndianness((DWORD)(seeker - pbRsrcSec) + m_dwResourceSectionVA);
seeker += RALIGN(cRDataE->GetSize(), 8);
qDataEntries2.pop();
}
/*
* Set all of the directory entries offsets.
*/
SetOffsets(m_cResDir, (ULONG_PTR)(pbRsrcSec));
}
// This function writes into a given place in memory (pbRsrcSec) the edited resource section
void CResourceEditor::WriteRsrcSec(BYTE* pbRsrcSec) {
BYTE* seeker = pbRsrcSec;
queue<CResourceDirectory*> qDirs; // Used to scan the tree by level
queue<CResourceDataEntry*> qDataEntries; // Used for writing the data entries
queue<CResourceDataEntry*> qDataEntries2; // Used for writing raw resources data
queue<CResourceDirectoryEntry*> qStrings; // Used for writing resources' names
qDirs.push(m_cResDir);
while (!qDirs.empty()) {
CResourceDirectory* crd = qDirs.front();
IMAGE_RESOURCE_DIRECTORY rdDir = crd->GetInfo();
CopyMemory(seeker, &rdDir, sizeof(IMAGE_RESOURCE_DIRECTORY));
crd->m_dwWrittenAt = DWORD(seeker);
seeker += sizeof(IMAGE_RESOURCE_DIRECTORY);
for (int i = 0; i < crd->CountEntries(); i++) {
if (crd->GetEntry(i)->HasName())
qStrings.push(crd->GetEntry(i));
if (crd->GetEntry(i)->IsDataDirectory())
qDirs.push(crd->GetEntry(i)->GetSubDirectory());
else {
qDataEntries.push(crd->GetEntry(i)->GetDataEntry());
qDataEntries2.push(crd->GetEntry(i)->GetDataEntry());
}
MY_IMAGE_RESOURCE_DIRECTORY_ENTRY rDirE;
ZeroMemory(&rDirE, sizeof(rDirE));
rDirE.DirectoryOffset.DataIsDirectory = crd->GetEntry(i)->IsDataDirectory();
rDirE.Id = (crd->GetEntry(i)->HasName()) ? 0 : crd->GetEntry(i)->GetId();
rDirE.NameString.NameIsString = (crd->GetEntry(i)->HasName()) ? 1 : 0;
CopyMemory(seeker, &rDirE, sizeof(MY_IMAGE_RESOURCE_DIRECTORY_ENTRY));
crd->GetEntry(i)->m_dwWrittenAt = DWORD(seeker);
seeker += sizeof(MY_IMAGE_RESOURCE_DIRECTORY_ENTRY);
}
qDirs.pop();
}
/*
* Write IMAGE_RESOURCE_DATA_ENTRYs.
*/
while (!qDataEntries.empty()) {
CResourceDataEntry* cRDataE = qDataEntries.front();
IMAGE_RESOURCE_DATA_ENTRY rDataE = {0,};
rDataE.CodePage = cRDataE->GetCodePage();
rDataE.Size = cRDataE->GetSize();
CopyMemory(seeker, &rDataE, sizeof(IMAGE_RESOURCE_DATA_ENTRY));
cRDataE->m_dwWrittenAt = DWORD(seeker);
seeker += sizeof(IMAGE_RESOURCE_DATA_ENTRY);
qDataEntries.pop();
}
/*
* Write strings
*/
while (!qStrings.empty()) {
CResourceDirectoryEntry* cRDirE = qStrings.front();
PMY_IMAGE_RESOURCE_DIRECTORY_ENTRY(cRDirE->m_dwWrittenAt)->NameString.NameOffset = DWORD(seeker) - DWORD(pbRsrcSec);
char* szName = cRDirE->GetName();
WORD iLen = strlen(szName) + 1;
WCHAR *szwName = new WCHAR[iLen];
// MultiByteToWideChar return value includes the null char, so -1
iLen = MultiByteToWideChar(CP_ACP, 0, szName, iLen, szwName, iLen) - 1;
if (iLen == (WORD) -1)
throw runtime_error("Unicode conversion failed");
*(WORD*)seeker = iLen;
seeker += sizeof(WORD);
CopyMemory(seeker, szwName, iLen*sizeof(WCHAR));
seeker += iLen*sizeof(WCHAR);
// Even though the number of chars is predefined a null termination is required
*(WORD*)seeker = 0;
seeker += sizeof(WORD);
delete [] szName;
delete [] szwName;
qStrings.pop();
}
/*
* Write raw resource data and set offsets in IMAGE_RESOURCE_DATA_ENTRYs.
*/
while (!qDataEntries2.empty()) {
CResourceDataEntry* cRDataE = qDataEntries2.front();
CopyMemory(seeker, cRDataE->GetData(), cRDataE->GetSize());
PIMAGE_RESOURCE_DATA_ENTRY(cRDataE->m_dwWrittenAt)->OffsetToData = seeker - pbRsrcSec + m_dwResourceSectionVA;
seeker += RALIGN(cRDataE->GetSize(), 8);
qDataEntries2.pop();
}
/*
* Set all of the directory entries offsets.
*/
SetOffsets(m_cResDir, DWORD(pbRsrcSec));
}