void PEResources::parseResourceLevel(Section §ion, uint32 offset, int level) {
_exe->seek(offset + 12);
uint16 entryCount = _exe->readUint16LE(); // named entry count
entryCount += _exe->readUint16LE(); // id entry count
for (uint32 i = 0; i < entryCount; i++) {
uint32 value = _exe->readUint32LE();
PEResourceID id;
if (value & 0x80000000) {
value &= 0x7fffffff;
uint32 startPos = _exe->pos();
_exe->seek(section.offset + (value & 0x7fffffff));
// Read in the name, UTF-16LE
uint16 nameLength = _exe->readUint16LE();
UString name;
name.readFixedUTF16LE(*_exe, nameLength);
_exe->seek(startPos);
id = name;
} else {
id = value;
}
uint32 nextOffset = _exe->readUint32LE();
uint32 lastOffset = _exe->pos();
if (level == 0)
_curType = id;
else if (level == 1)
_curName = id;
else if (level == 2)
_curLang = id;
if (level < 2) {
// Time to dive down further
parseResourceLevel(section, section.offset + (nextOffset & 0x7fffffff), level + 1);
} else {
_exe->seek(section.offset + nextOffset);
Resource resource;
resource.offset = _exe->readUint32LE() + section.offset - section.virtualAddress;
resource.size = _exe->readUint32LE();
//status("Found resource '%s' '%s' '%s' at %d of size %d", _curType.toString().c_str(),
// _curName.toString().c_str(), _curLang.toString().c_str(), resource.offset, resource.size);
_resources[_curType][_curName][_curLang] = resource;
}
_exe->seek(lastOffset);
}
}
bool PEResources::loadFromEXE(SeekableReadStream *stream) {
clear();
if (!stream)
return false;
if (stream->readUint16BE() != 'MZ')
return false;
stream->skip(58);
uint32 peOffset = stream->readUint32LE();
if (!peOffset || peOffset >= (uint32)stream->size())
return false;
stream->seek(peOffset);
if (stream->readUint32BE() != MKTAG('P', 'E', '\0', '\0'))
return false;
stream->skip(2);
uint16 sectionCount = stream->readUint16LE();
stream->skip(12);
uint16 optionalHeaderSize = stream->readUint16LE();
stream->skip(optionalHeaderSize + 2);
// Read in all the sections
for (uint16 i = 0; i < sectionCount; i++) {
char sectionName[9];
stream->read(sectionName, 8);
sectionName[8] = 0;
Section section;
stream->skip(4);
section.virtualAddress = stream->readUint32LE();
section.size = stream->readUint32LE();
section.offset = stream->readUint32LE();
stream->skip(16);
_sections[sectionName] = section;
}
// Currently, we require loading a resource section
if (_sections.find(".rsrc") == _sections.end()) {
clear();
return false;
}
_exe = stream;
Section &resSection = _sections[".rsrc"];
parseResourceLevel(resSection, resSection.offset, 0);
return true;
}
void PEResources::parseResourceLevel(Section §ion, uint32 offset, int level) {
_exe->seek(offset + 12);
uint16 namedEntryCount = _exe->readUint16LE();
uint16 intEntryCount = _exe->readUint16LE();
for (uint32 i = 0; i < (uint32)(namedEntryCount + intEntryCount); i++) {
uint32 value = _exe->readUint32LE();
WinResourceID id;
if (value & 0x80000000) {
value &= 0x7fffffff;
uint32 startPos = _exe->pos();
_exe->seek(section.offset + (value & 0x7fffffff));
// Read in the name, truncating from unicode to ascii
String name;
uint16 nameLength = _exe->readUint16LE();
while (nameLength--)
name += (char)(_exe->readUint16LE() & 0xff);
_exe->seek(startPos);
id = name;
} else {
id = value;
}
uint32 nextOffset = _exe->readUint32LE();
uint32 lastOffset = _exe->pos();
if (level == 0)
_curType = id;
else if (level == 1)
_curName = id;
else if (level == 2)
_curLang = id;
if (level < 2) {
// Time to dive down further
parseResourceLevel(section, section.offset + (nextOffset & 0x7fffffff), level + 1);
} else {
_exe->seek(section.offset + nextOffset);
Resource resource;
resource.offset = _exe->readUint32LE() + section.offset - section.virtualAddress;
resource.size = _exe->readUint32LE();
_resources[_curType][_curName][_curLang] = resource;
}
_exe->seek(lastOffset);
}
}
