You've already forked godot
mirror of
https://github.com/godotengine/godot.git
synced 2025-11-04 12:00:25 +00:00
Merge pull request #108455 from bruvzg/win_exp_symbols
[Windows export] Move debug symbol sections on export.
This commit is contained in:
@@ -574,9 +574,20 @@ Error TemplateModifier::_modify_template(const Ref<EditorExportPreset> &p_preset
|
||||
|
||||
Vector<SectionEntry> section_entries = _get_section_entries(template_file);
|
||||
ERR_FAIL_COND_V(section_entries.size() < 2, ERR_CANT_OPEN);
|
||||
ERR_FAIL_COND_V(section_entries[section_entries.size() - 2].name != String(".rsrc"), ERR_CANT_OPEN);
|
||||
ERR_FAIL_COND_V(section_entries[section_entries.size() - 1].name != String(".reloc"), ERR_CANT_OPEN);
|
||||
// TODO fail on not sorted by physical address?
|
||||
|
||||
// Find resource (".rsrc") and relocation (".reloc") sections, usually last two, but ".debug_*" sections (referenced as "/[n]"), symbol table, and string table can follow.
|
||||
int resource_index = section_entries.size() - 2;
|
||||
int relocations_index = section_entries.size() - 1;
|
||||
for (int i = 0; i < section_entries.size(); i++) {
|
||||
if (section_entries[i].name == ".rsrc") {
|
||||
resource_index = i;
|
||||
} else if (section_entries[i].name == ".reloc") {
|
||||
relocations_index = i;
|
||||
}
|
||||
}
|
||||
|
||||
ERR_FAIL_COND_V(section_entries[resource_index].name != ".rsrc", ERR_CANT_OPEN);
|
||||
ERR_FAIL_COND_V(section_entries[relocations_index].name != ".reloc", ERR_CANT_OPEN);
|
||||
|
||||
uint64_t original_template_size = template_file->get_length();
|
||||
|
||||
@@ -584,27 +595,58 @@ Error TemplateModifier::_modify_template(const Ref<EditorExportPreset> &p_preset
|
||||
|
||||
VersionInfo version_info = _create_version_info(_get_strings(p_preset));
|
||||
|
||||
SectionEntry resources_section_entry = section_entries.get(section_entries.size() - 2);
|
||||
SectionEntry &resources_section_entry = section_entries.write[resource_index];
|
||||
uint32_t old_resources_size_of_raw_data = resources_section_entry.size_of_raw_data;
|
||||
Vector<uint8_t> resources = _create_resources(resources_section_entry.virtual_address, group_icon, version_info);
|
||||
resources_section_entry.virtual_size = resources.size();
|
||||
resources.resize_initialized(_snap(resources.size(), BLOCK_SIZE));
|
||||
resources_section_entry.size_of_raw_data = resources.size();
|
||||
|
||||
SectionEntry relocations_section_entry = section_entries.get(section_entries.size() - 1);
|
||||
uint32_t old_relocations_virtual_address = relocations_section_entry.virtual_address;
|
||||
template_file->seek(relocations_section_entry.pointer_to_raw_data);
|
||||
Vector<uint8_t> relocations = template_file->get_buffer(relocations_section_entry.size_of_raw_data);
|
||||
relocations_section_entry.pointer_to_raw_data = resources_section_entry.pointer_to_raw_data + resources_section_entry.size_of_raw_data;
|
||||
relocations_section_entry.virtual_address = resources_section_entry.virtual_address + _snap(resources_section_entry.virtual_size, PE_PAGE_SIZE);
|
||||
int32_t raw_size_delta = resources_section_entry.size_of_raw_data - old_resources_size_of_raw_data;
|
||||
uint32_t old_last_section_virtual_address = section_entries.get(section_entries.size() - 1).virtual_address;
|
||||
|
||||
// Some data (e.g. DWARF debug symbols) can be placed after the last section.
|
||||
uint32_t old_footer_offset = section_entries.get(section_entries.size() - 1).pointer_to_raw_data + section_entries.get(section_entries.size() - 1).size_of_raw_data;
|
||||
|
||||
// Copy and update sections after ".rsrc".
|
||||
Vector<Vector<uint8_t>> moved_section_data;
|
||||
uint32_t prev_virtual_address = resources_section_entry.virtual_address;
|
||||
uint32_t prev_virtual_size = resources_section_entry.virtual_size;
|
||||
for (int i = resource_index + 1; i < section_entries.size(); i++) {
|
||||
SectionEntry §ion_entry = section_entries.write[i];
|
||||
template_file->seek(section_entry.pointer_to_raw_data);
|
||||
Vector<uint8_t> data = template_file->get_buffer(section_entry.size_of_raw_data);
|
||||
moved_section_data.push_back(data);
|
||||
section_entry.pointer_to_raw_data += raw_size_delta;
|
||||
section_entry.virtual_address = prev_virtual_address + _snap(prev_virtual_size, PE_PAGE_SIZE);
|
||||
prev_virtual_address = section_entry.virtual_address;
|
||||
prev_virtual_size = section_entry.virtual_size;
|
||||
}
|
||||
|
||||
// Copy COFF symbol table and string table after the last section.
|
||||
uint32_t footer_size = template_file->get_length() - old_footer_offset;
|
||||
template_file->seek(old_footer_offset);
|
||||
Vector<uint8_t> footer;
|
||||
if (footer_size > 0) {
|
||||
footer = template_file->get_buffer(footer_size);
|
||||
}
|
||||
|
||||
uint32_t pe_header_offset = _get_pe_header_offset(template_file);
|
||||
|
||||
// Update symbol table pointer.
|
||||
template_file->seek(pe_header_offset + 12);
|
||||
uint32_t symbols_offset = template_file->get_32();
|
||||
if (symbols_offset > resources_section_entry.pointer_to_raw_data) {
|
||||
template_file->seek(pe_header_offset + 12);
|
||||
template_file->store_32(symbols_offset + raw_size_delta);
|
||||
}
|
||||
|
||||
template_file->seek(pe_header_offset + MAGIC_NUMBER_OFFSET);
|
||||
uint16_t magic_number = template_file->get_16();
|
||||
ERR_FAIL_COND_V_MSG(magic_number != 0x10b && magic_number != 0x20b, ERR_CANT_OPEN, vformat("Magic number has wrong value: %04x", magic_number));
|
||||
bool pe32plus = magic_number == 0x20b;
|
||||
|
||||
// Update image size.
|
||||
template_file->seek(pe_header_offset + SIZE_OF_INITIALIZED_DATA_OFFSET);
|
||||
uint32_t size_of_initialized_data = template_file->get_32();
|
||||
size_of_initialized_data += resources_section_entry.size_of_raw_data - old_resources_size_of_raw_data;
|
||||
@@ -613,30 +655,43 @@ Error TemplateModifier::_modify_template(const Ref<EditorExportPreset> &p_preset
|
||||
|
||||
template_file->seek(pe_header_offset + SIZE_OF_IMAGE_OFFSET);
|
||||
uint32_t size_of_image = template_file->get_32();
|
||||
size_of_image += relocations_section_entry.virtual_address - old_relocations_virtual_address;
|
||||
size_of_image += section_entries.get(section_entries.size() - 1).virtual_address - old_last_section_virtual_address;
|
||||
template_file->seek(pe_header_offset + SIZE_OF_IMAGE_OFFSET);
|
||||
template_file->store_32(size_of_image);
|
||||
|
||||
uint32_t optional_header_offset = pe_header_offset + COFF_HEADER_SIZE;
|
||||
|
||||
// Update resource section size.
|
||||
template_file->seek(optional_header_offset + (pe32plus ? 132 : 116));
|
||||
template_file->store_32(resources_section_entry.virtual_size);
|
||||
|
||||
// Update relocation section size and pointer.
|
||||
template_file->seek(optional_header_offset + (pe32plus ? 152 : 136));
|
||||
template_file->store_32(relocations_section_entry.virtual_address);
|
||||
template_file->store_32(relocations_section_entry.virtual_size);
|
||||
template_file->store_32(section_entries[relocations_index].virtual_address);
|
||||
template_file->store_32(section_entries[relocations_index].virtual_size);
|
||||
|
||||
template_file->seek(optional_header_offset + (pe32plus ? 240 : 224) + SectionEntry::SIZE * (section_entries.size() - 2));
|
||||
template_file->seek(optional_header_offset + (pe32plus ? 240 : 224) + SectionEntry::SIZE * resource_index);
|
||||
template_file->store_buffer(resources_section_entry.save());
|
||||
template_file->store_buffer(relocations_section_entry.save());
|
||||
for (int i = resource_index + 1; i < section_entries.size(); i++) {
|
||||
template_file->seek(optional_header_offset + (pe32plus ? 240 : 224) + SectionEntry::SIZE * i);
|
||||
template_file->store_buffer(section_entries[i].save());
|
||||
}
|
||||
|
||||
// Write new resource section.
|
||||
template_file->seek(resources_section_entry.pointer_to_raw_data);
|
||||
template_file->store_buffer(resources);
|
||||
template_file->store_buffer(relocations);
|
||||
// Write the rest of sections.
|
||||
for (const Vector<uint8_t> &data : moved_section_data) {
|
||||
template_file->store_buffer(data);
|
||||
}
|
||||
// Write footer data.
|
||||
if (footer_size > 0) {
|
||||
template_file->store_buffer(footer);
|
||||
}
|
||||
|
||||
if (template_file->get_position() < original_template_size) {
|
||||
template_file->close();
|
||||
_truncate(p_template_path, relocations_section_entry.pointer_to_raw_data + relocations_section_entry.size_of_raw_data);
|
||||
_truncate(p_template_path, section_entries.get(section_entries.size() - 1).pointer_to_raw_data + section_entries.get(section_entries.size() - 1).size_of_raw_data + footer_size);
|
||||
}
|
||||
|
||||
return OK;
|
||||
|
||||
Reference in New Issue
Block a user