def get_rel(self, rel_path):
rel_path = rel_path.replace("\\", "/")
if rel_path in self.rels_by_path:
return self.rels_by_path[rel_path]
else:
if not rel_path.startswith("files/rels/"):
raise Exception("Invalid REL path: %s" % rel_path)
rel_name = os.path.basename(rel_path)
rels_arc = self.get_arc("files/RELS.arc")
rel_file_entry = rels_arc.get_file_entry(rel_name)
if rel_file_entry:
rel_file_entry.decompress_data_if_necessary()
data = rel_file_entry.data
else:
data = self.gcm.read_file_data(rel_path)
rel = REL()
rel.read(data)
self.rels_by_path[rel_path] = rel
return rel
def convert_elf_to_rel(in_elf_path,
out_rel_path,
rel_id,
actor_profile_name,
rels_arc_path=None):
elf = ELF()
elf.read_from_file(in_elf_path)
rel = REL()
rel.id = rel_id
# First convert the sections we want to include in the REL from ELF sections to REL sections.
section_name_to_rel_section = {}
elf_section_index_to_rel_section = {}
for elf_section_index, elf_section in enumerate(elf.sections):
if elf_section.name in ALLOWED_SECTIONS or elf_section.type == ELFSectionType.SHT_NULL:
# Sections we will add to the REL.
rel_section = RELSection()
rel_section.data = make_copy_data(elf_section.data)
rel.sections.append(rel_section)
if elf_section.flags & ELFSectionFlags.SHF_EXECINSTR.value:
rel_section.is_executable = True
section_name_to_rel_section[elf_section.name] = rel_section
elf_section_index_to_rel_section[elf_section_index] = rel_section
if elf_section.type == ELFSectionType.SHT_NULL:
rel_section.is_uninitialized = True
rel_section.is_bss = False
else:
rel_section.is_uninitialized = False
rel_section.is_bss = False
# TODO: bss support
# Then generate the relocations.
for elf_section in elf.sections:
if elf_section.type == ELFSectionType.SHT_RELA:
assert elf_section.name.startswith(".rela")
relocated_section_name = elf_section.name[len(".rela"):]
if relocated_section_name not in section_name_to_rel_section:
# Ignored section
continue
rel_section = section_name_to_rel_section[relocated_section_name]
section_index = rel.sections.index(rel_section)
for elf_relocation in elf.relocations[elf_section.name]:
rel_relocation = RELRelocation()
elf_symbol = elf.symbols[".symtab"][
elf_relocation.symbol_index]
rel_relocation.relocation_type = RELRelocationType(
elf_relocation.type.value)
#print("%X" % elf_symbol.section_index)
if elf_symbol.section_index == 0:
raise Exception(
"Unresolved external symbol in main.dol: %s" %
elf_symbol.name)
if elf_symbol.section_index == ELFSymbolSpecialSection.SHN_ABS.value:
# Symbol is located in main.dol.
module_num = 0
# I don't think this value is used for dol relocations.
# In vanilla, this was written as 4 for some reason?
rel_relocation.section_num_to_relocate_against = 0
elif elf_symbol.section_index >= 0xFF00:
raise Exception(
"Special section number not implemented: %04X" %
elf_symbol.section_index)
else:
# Symbol is located in the current REL.
# TODO: support for relocating against other rels besides the current one
module_num = rel.id
section_name_to_relocate_against = elf.sections[
elf_symbol.section_index].name
if section_name_to_relocate_against not in section_name_to_rel_section:
raise Exception(
"Section name \"%s\" could not be found for symbol \"%s\""
% (section_name_to_relocate_against,
elf_symbol.name))
rel_section_to_relocate_against = section_name_to_rel_section[
section_name_to_relocate_against]
rel_section_index_to_relocate_against = rel.sections.index(
rel_section_to_relocate_against)
rel_relocation.section_num_to_relocate_against = rel_section_index_to_relocate_against
rel_relocation.symbol_address = elf_symbol.address + elf_relocation.addend
rel_relocation.relocation_offset = elf_relocation.relocation_offset
rel_relocation.curr_section_num = section_index
if module_num not in rel.relocation_entries_for_module:
rel.relocation_entries_for_module[module_num] = []
rel.relocation_entries_for_module[module_num].append(
rel_relocation)
symbol = elf.symbols_by_name[".symtab"]["_prolog"]
rel_section = elf_section_index_to_rel_section[symbol.section_index]
rel_section_index = rel.sections.index(rel_section)
rel.prolog_section = rel_section_index
rel.prolog_offset = symbol.address
symbol = elf.symbols_by_name[".symtab"]["_epilog"]
rel_section = elf_section_index_to_rel_section[symbol.section_index]
rel_section_index = rel.sections.index(rel_section)
rel.epilog_section = rel_section_index
rel.epilog_offset = symbol.address
symbol = elf.symbols_by_name[".symtab"]["_unresolved"]
rel_section = elf_section_index_to_rel_section[symbol.section_index]
rel_section_index = rel.sections.index(rel_section)
rel.unresolved_section = rel_section_index
rel.unresolved_offset = symbol.address
rel.save_to_file(out_rel_path)
# TODO: instead of replacing a REL, add a new REL!
if rels_arc_path is not None and os.path.isfile(rels_arc_path):
with open(rels_arc_path, "rb") as f:
data = BytesIO(f.read())
rels_arc = RARC()
rels_arc.read(data)
# Update the profile list to properly reference the profile in the custom REL.
# Then insert the custom REL and the updated profile list into RELS.arc and save it for quick testing.
profile_list_data = rels_arc.get_file_entry(
"f_pc_profile_lst.rel").data
profile_list = REL()
profile_list.read(profile_list_data)
if rel.id not in profile_list.relocation_entries_for_module:
raise Exception(
"REL ID %02X could not be found in the profile list." % rel.id)
# TODO: add new REL instead of replacing!
if actor_profile_name not in elf.symbols_by_name[".symtab"]:
raise Exception(
"Could not find the actor profile. The specified symbol name for it was \"%s\", but that symbol could not be found in the ELF."
% actor_profile_name)
profile_symbol = elf.symbols_by_name[".symtab"][actor_profile_name]
profile_list.relocation_entries_for_module[
rel.id][0].symbol_address = profile_symbol.address
section_with_profile = section_name_to_rel_section[".rodata"]
new_section_index_with_profile = rel.sections.index(
section_with_profile)
profile_list.relocation_entries_for_module[rel.id][
0].section_num_to_relocate_against = new_section_index_with_profile
profile_list.save_changes(preserve_section_data_offsets=True)
# Insert the RELs.
rels_arc.get_file_entry(
"f_pc_profile_lst.rel").data = profile_list.data
found_rel = False
for file_entry in rels_arc.file_entries:
if file_entry.is_dir:
continue
file_entry.decompress_data_if_necessary()
if read_u32(file_entry.data, 0) == rel.id:
file_entry.data = rel.data
found_rel = True
break
if not found_rel:
raise Exception("Failed to find REL to replace with ID 0x%03X" %
rel.id)
rels_arc.save_changes()
with open(rels_arc_path, "wb") as f:
f.write(read_all_bytes(rels_arc.data))