sceutils/module_info.py at master · olebeck/sceutils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import sys, json
from scetypes import ElfHeader, ElfPhdr, SceModuleInfo, SceModuleImports, SceModuleImports2
from util import c_str
from dataclasses import dataclass

@dataclass
class LibraryImport:
    name: str
    version: int
    functions: list[int]
    variables: list[int]

def read_module_info(fin) -> tuple[SceModuleInfo, list[LibraryImport]]:
    ehdr = ElfHeader.unpack(fin)
    if ehdr.e_machine == 0xf00d:
        return None, None
    segment_num = (ehdr.e_entry >> 30) & 0x3
    info_offset = ehdr.e_entry & 0x3fffffff
    fin.seek(ehdr.e_phoff + ElfPhdr.Size() * segment_num)
    phdr = ElfPhdr.unpack(fin)
    fin.seek(phdr.p_offset + info_offset)
    module_info = SceModuleInfo.unpack(fin)

    import_list = []
    fin.seek(phdr.p_offset + module_info.importsTop)
    imports_off = module_info.importsTop
    while imports_off < module_info.importsEnd:
        fin.seek(phdr.p_offset+imports_off)
        size = int.from_bytes(fin.read(2), "little")
        fin.seek(-2, 1)
        if size == 0x24:
            imports = SceModuleImports2.unpack(fin)
        elif size == 0x34:
            imports = SceModuleImports.unpack(fin)
        else:
            raise Exception(f"imports wrong size {size}")
        imports_off += size

        library_name = hex(imports.library_nid)
        if imports.library_name > phdr.p_vaddr:
            fin.seek(phdr.p_offset + imports.library_name - phdr.p_vaddr)
            library_name = c_str(fin.read(32))

        functions = []
        variables = []
        for (entry_table, nid_table, num, arr) in (
            (imports.func_entry_table, imports.func_nid_table, imports.num_functions, functions),
            (imports.var_entry_table, imports.var_nid_table, imports.num_vars, variables)
        ):
            if entry_table == 0:
                continue
            entry_table_offset = entry_table - phdr.p_vaddr
            nid_table_offset = nid_table - phdr.p_vaddr
            fin.seek(phdr.p_offset+entry_table_offset)
            entry_data = fin.read(num * 4)
            fin.seek(phdr.p_offset+nid_table_offset)
            nids_data = fin.read(num * 4)
            for i in range(num):
                nid = int.from_bytes(nids_data[i*4:(i+1)*4], "little")
                addr = int.from_bytes(entry_data[i*4:(i+1)*4], "little")
                arr.append(nid)
        import_list.append(LibraryImport(library_name, imports.version, functions, variables))
    return module_info, import_list

if __name__ == "__main__":
    with open(sys.argv[1], "rb") as fin:
        module_info, imports = read_module_info(fin)
    info = {
        "module_name": module_info.module_name,
        "module_nid": hex(module_info.module_nid),
        "imports": imports
    }
    print(json.dumps(info, indent=2))