MegaEvolution/insert.py at master · Touched/MegaEvolution · 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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#!/usr/bin/python3

import os
import subprocess
import sys
import shutil
import binascii
import textwrap
import sys
import argparse
import patch

if sys.version_info < (3, 4):
        print('Python 3.4 or later is required.')
        sys.exit(1)

# Parse arguments
parser = argparse.ArgumentParser()
parser.add_argument('--no-patch', action='store_true',
                    help='disable uninstall patch generation')
parser.add_argument('--offset', metavar='offset', type=int,
                    help='offset to insert at', default=0x800000)
parser.add_argument('--input', metavar='file',
                    help='input filename', default='BPRE0.gba')
parser.add_argument('--output', metavar='file',
                    help='output filename', default='test.gba')
parser.add_argument('--debug', action='store_true',
                    help='print symbol table')

args = parser.parse_args()

OBJCOPY = 'arm-none-eabi-objcopy'
OBJDUMP = 'arm-none-eabi-objdump'
NM = 'arm-none-eabi-nm'
AS = 'arm-none-eabi-as'
CC = 'arm-none-eabi-gcc'
CXX = 'arm-none-eabi-g++'

def get_text_section():
        # Dump sections
        out = subprocess.check_output([OBJDUMP, '-t', 'build/linked.o'])
        lines = out.decode().split('\n')

        # Find text section
        text = filter(lambda x: x.strip().endswith('.text'), lines)
        section = list(text)[0]

        # Get the offset
        offset = int(section.split(' ')[0], 16)

        return offset

def symbols(subtract=0):
        out = subprocess.check_output([NM, 'build/linked.o'])
        lines = out.decode().split('\n')

        name = ''

        ret = {}
        for line in lines:
                parts = line.strip().split()

                if (len(parts) < 3):
                        continue

                if (parts[1].lower() != 't'):
                        continue

                offset = int(parts[0], 16)
                ret[parts[2]] = offset - subtract

        return ret

def insert(rom):
        where = args.offset
        rom.seek(where)
        with open('build/output.bin', 'rb') as binary:
                data = binary.read()
        rom.write(data)
        return where

def hook(rom, space, hook_at, register=0):
        # Align 2
        if hook_at & 1:
            hook_at -= 1

        rom.seek(hook_at)

        register &= 7

        if hook_at % 4:
            data = bytes([0x01, 0x48 | register, 0x00 | (register << 3), 0x47, 0x0, 0x0])
        else:
            data = bytes([0x00, 0x48 | register, 0x00 | (register << 3), 0x47])

        space += 0x08000001
        data += (space.to_bytes(4, 'little'))
        rom.write(bytes(data))

shutil.copyfile(args.input, args.output)
with open(args.output, 'rb+') as rom:
        offset = get_text_section()
        table = symbols(offset)
        where = insert(rom)

        # Adjust symbol table
        for entry in table:
                table[entry] += where

        hook(rom, table['command'], 0x033224, 2)
        hook(rom, table['command'], 0x038A50, 2) # 08250A34
        hook(rom, table['move_button_hook'], 0x02EC10, 0)
        hook(rom, table['exit_battle_hook'], 0x0159DC, 0)
        hook(rom, table['faint_hook'], 0x0326C4, 3)
        hook(rom, table['level_string_hook'], 0x0483A4, 1)
        #hook(rom, table['mega_level_icon_hook'], 0x049E18, 1)
        hook(rom, table['load_graphics_hook'], 0x03495C, 1)
        #hook(rom, table['show_graphics_hook'], 0x047FD8, 1)
        # Add the fucking animation
        #loc = table['mega_animation_script_data'] + 0x08000000
        #move_index = 33 # tackle
        #rom.seek(move_index * 4 + 0x1C68F4)
        #rom.write(loc.to_bytes(4, 'little'))

        # AI
        hook(rom, table['ai_move_hook'], 0x038668, 0)

        # Stupid
        hook(rom, table['create_shaker_hook'], 0x04BE80, 3)
        hook(rom, table['objc_shaker_hook'], 0x04BEDC, 2)
        hook(rom, table['battle_menu_hook'], 0x02E438, 0)

        # Main
        #hook(rom, table['attack_canceller_hook'], 0x01DB00, 1)
        hook(rom, table['bc_pre_attacks_hook'], 0x0154A0, 0)

        if args.debug:
                width = max(map(len, table.keys())) + 1
                for key in sorted(table.keys()):
                        fstr = ('{:' + str(width) + '} {:08X}')
                        print(fstr.format(key + ':', table[key] + 0x08000000))

# Create a patch to remove these changes
if not args.no_patch:
        patch.create(args.output, args.input, 'uninstall.ips', patch.ips)