SoR EBOOT Tool v1

import struct
import pdb

PTR_COL = 6
s_list = ['世界地図', 'ロンドエール', '円卓の間', '自室', '地下室',
          'フューミの村', '村はずれ', '取引所', '村長の家', 'アーレンティール',
          '森の入口', '交易所', '大樹の館', 'エセルガルド', '城砦周辺',
          '闇の行商人', '正門前', 'ミストアイル', '氷の断崖',
          'ドワーフの道具屋', '穴ぐら', 'グロムバルク', '城門', '商店', '王宮']
s_list1 = [True] * len(s_list)
s_list2 = [[]] * len(s_list)
s_list3 = [False] * len(s_list)
block_list = [0x14f46c, 0x14f6fc, 0x14f944, 0x14fc54,
              0x1501bc, 0x1504bc, 0x150c5c, 0x150fa4, 0x1511d4, 0x151464,
              0x153edc, 0x154194, 0x15447c, 0x154854, 0x154b24, 0x154e84,
              0x1550b4, 0x1554b4, 0x1556e4, 0x156324, 0x15658c, 0x156894,
              0x156c4c, 0x156f24, 0x1572c4, 0x15858c, 0x158d44, 0x158fbc,
              0x15938c, 0x15961c, 0x159acc, 0x159d2c, 0x159f6c, 0x15a1dc,
              0x15a424, 0x15a664, 0x15a8bc, 0x15ac8c, 0x15aebc, 0x15b0ec,
              0x15b444, 0x15b674, 0x15b8ac, 0x15d2c8, 0x15d56c, 0x15d79c,
              0x15eae4, 0x15f00c, 0x15fc94, 0x15ffd4, 0x1602a4, 0x160cec,
              0x161074, 0x1612cc, 0x1614fc, 0x16180c, 0x161a84, 0x161cb4,
              0x161eec, 0x16213c, 0x1623ac, 0x16279c, 0x162d64, 0x16334c,
              0x16383c, 0x163a6c, 0x164104, 0x164764, 0x1649bc, 0x164c4c,
              0x164e94, 0x1650c4, 0x1652f4, 0x16586c, 0x1662b4, 0x16650c,
              0x16673c, 0x16696c, 0x166c74, 0x166ea4, 0x1670d4, 0x167624,
              0x167b4c, 0x167dbc, 0x167fec, 0x16846c, 0x168ab4, 0x168ebc,
              0x16917c, 0x169424, 0x1696d4, 0x1779fc, 0x177c6c, 0x177ebc,
              0x1781c4, 0x178584, 0x178834, 0x178b44, 0x178d74, 0x178fc4,
              0x1791f4, 0x1795ec, 0x1798e4, 0x179d5c, 0x179f8c, 0x17a1bc,
              0x17a3ec, 0x17a634, 0x17a874, 0x17ab0c, 0x17ad5c, 0x17af8c,
              0x17b1bc, 0x17b46c, 0x17b6a4, 0x17bad4, 0x17bea4, 0x17c42c,
              0x17c65c, 0x17c894, 0x17cac4, 0x17cd24, 0x17cfbc, 0x17d23c,
              0x17d46c, 0x17d71c, 0x17dad4, 0x17dd94, 0x17e1ac, 0x17e474,
              0x17e774, 0x17e9ac, 0x17ebec, 0x17efb4, 0x17f22c, 0x17f494,
              0x180afc, 0x180d3c, 0x180fdc, 0x1813c4, 0x18162c]
blockiter = iter(block_list)

def update_ptr(ptr, orig_pos, new_pos):
    if ptr[0] == '^':
        MSBflag = True
        ptr = ptr[1:]
    else:
        MSBflag = False
    ptr = int(ptr, 16)
    if ptr < 0x200000:
        filedata[ptr:ptr+4] = struct.pack('<I', new_pos)
    else:
##        print(hex(addr))
        orig_pos -= 0xC0
        ptr -= 0x8803F40
        new_pos -= 0x8804000
        if new_pos & 0xFFFF >= 0x8000:
            new_pos += 0x10000
        filedata[ptr:ptr+2] = struct.pack('<H', new_pos >> 16)
        if MSBflag:
            return
        ptr = filedata.find(struct.pack('<H', orig_pos & 0xFFFF), ptr)
        filedata[ptr:ptr+2] = struct.pack('<H', new_pos & 0xFFFF)

def eboot_transform():
    '''unwind EBOOT relocations'''
    with open('ULJS00548.BIN', 'rb') as f:
        filedata = bytearray(f.read())
    pos = 0x312b04
    end = pos + 0x6de90
    count = 0
    while pos < end:
        if struct.unpack('<I', filedata[pos+4:pos+8])[0] == 2:
            tgt = struct.unpack('<I', filedata[pos:pos+4])[0] + 0xC0
            val = struct.unpack('<I', filedata[tgt:tgt+4])[0]
            filedata[tgt:tgt+4] = struct.pack('<I', val + 0x8804000)
            filedata[pos:pos+8] = b'\x00' * 8
        pos += 8
    return filedata

filedata = eboot_transform()    #Unwinding eboot relocations
#Blank out the 2nd block
filedata[0x14EEC0:0x14EFF0] = b'\x00' * (0x14EFF0 - 0x14EEC0)
count = 0
insert_point = 0x14E944         #First string goes here
block_end = insert_point + 300  #When to switch to the 2nd block
#Read input file data
inputdata = []
with open('all.tsv', 'r', encoding='utf-8') as f:
    for line in f:
        ptrs = []
        line = line.rstrip('\r\n').split('\t')
        if line[0] != '':           #New entry
#Read in the pointer list
            c = PTR_COL
            while c < len(line):
                if line[c] == '':
                    break
                else:
                    ptrs.append(line[c])
                    c += 1
#Add entry to list. Format = addr, origtext, newtext, ptrs
            inputdata.append([int(line[0], 16), line[4], line[5], ptrs])
        else:                       #Continuation entry
#Add data to previous entry
            inputdata[-1][1] += '\n' + line[4]
            inputdata[-1][2] += '\n' + line[5]
deleted = 0
for i, (addr, orig_text, text, ptrs) in enumerate(list(inputdata)):
    if orig_text in s_list:         #Entry is in the list at the top
                                    #These repeat many times, like over 100
#Delete from list. We don't want to come back to it later
        del inputdata[i - deleted]
        deleted += 1
        index = s_list.index(orig_text)     #Get index number
        flag = s_list1[index]               #Get flag. True = Not processed yet
        if flag:                                        #Not processed yet
            s_list1[index] = False                      #Set flag
            if insert_point + len(text) < block_end:    #Check for end of block
                pass
            else:                                       #Switch to 2nd block
                insert_point = 0x14EEC4
                block_end = insert_point + 300
            text = text.encode('cp932')                 #Convert to bytes
            text += b'\x00' * (4 - len(text) % 4)       #Pad to word-aligned
#Insert the text
            filedata[insert_point:insert_point+len(text)] = text
#Set the pointer for where this text can be found
            s_list2[index] = insert_point + 0x8803F40
            insert_point += len(text)                   #Advance pointer
        else:                                           #Already processed
            if s_list3[index]:                          #3rd (or later time)
                end = filedata.find(b'\x00', addr)      #Blank out the string
                filedata[addr:end] = b'\x00' * (end-addr)
            else:                                       #2nd time
                s_list3[index] = True                   #Set flag
        for ptr in ptrs:                                #Update pointers
            update_ptr(ptr, addr, s_list2[index])
insert_point = next(blockiter)
block_end = insert_point + 300
for i, (addr, orig_text, text, ptrs) in enumerate(reversed(inputdata)):
    if text != orig_text and text != '':
        text = text.encode('cp932')
        end = filedata.find(b'\x00', addr)              #Compute available space
        while filedata[end] == 0:
            end += 1
        avail_space = end - 1 - addr
        if len(text) <= avail_space:                    #Enough space
            text += b'\x00' * (4 - len(text) % 4)       #Pad to word-aligned
            filedata[addr:addr+len(text)] = text        #Insert text
        else:                                           #Not enough space
            if ptrs == []:
                print(hex(addr))
                continue        #Need pointers to move if not enough space
            filedata[addr:end] = b'\x00' * (end-addr)   #Blank out the string
            text += b'\x00' * (4 - len(text) % 4)       #Pad to word-aligned
            if insert_point + len(text) < block_end:    #Enough space in block
                pass
            else:                                       #Not enough, go to next
                insert_point = next(blockiter)          #block
                block_end = insert_point + 300
            filedata[insert_point:insert_point+len(text)] = text
            for ptr in ptrs:                            #Update pointers
                update_ptr(ptr, addr, insert_point + 0x8803F40)
            insert_point += len(text)                   #Advance pointer


with open('EBOOT.BIN', 'wb') as f:
    f.write(filedata)