From 8f0096eb5e8344efe283e82200ba682c6c70fedd Mon Sep 17 00:00:00 2001
From: david <david@optixx.org>
Date: Thu, 9 Jul 2009 11:58:28 +0200
Subject: [PATCH] add rom inject

---
 scripts/rom_inject.py | 71 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 scripts/rom_inject.py

diff --git a/scripts/rom_inject.py b/scripts/rom_inject.py
new file mode 100644
index 0000000..413f475
--- /dev/null
+++ b/scripts/rom_inject.py
@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+import os
+import sys
+import struct
+
+def usage():
+    print "%s <rom> <code> <addr>" % sys.argv[0]
+    sys.exit(0)
+
+
+def error(msg):
+    print "ERROR: %s" % msg
+    sys.exit(-1)
+
+def main():
+
+
+
+    if len(sys.argv) != 4:
+        usage()
+    if not os.path.isfile(sys.argv[1]):
+        error("Can't open %s file" % sys.argv[1])
+    if not os.path.isfile(sys.argv[2]):
+        error("Can't open %s file" % sys.argv[2])
+      
+    rom = open(sys.argv[1],"r").read()
+    code = open(sys.argv[2],"r").read()
+    out_filename = sys.argv[1].split(".")[0] + ".inject"
+    out = open(out_filename,"w")
+    
+    irq_vector = 0x7FE0
+    bank_size = 1 << 15
+    code_len = len(code)
+    rom_len = len(rom)
+    bank_cnt = rom_len / bank_size
+    try:
+      addr = int(sys.argv[3],16)
+    except:
+        error("Expect %s in hex" % sys.argv[3])
+
+    addr_patch = struct.pack(">H",addr)
+
+
+    if  addr >  bank_size:
+        error("Addr 0x%04x is not within first bank" % addr)
+
+    if  addr + code_len >  ( bank_size - 256 ):
+        error("Code is %s bytes, and so too big to fit from 0x%04x into first bank" % (code_len,addr))
+
+    print "Rom size:        0x%08x (%i)" % (rom_len,rom_len)
+    print "Code size:       0x%04x (%i)" % (code_len,code_len)
+    print "Banks:           0x%04x (%i)" % (bank_cnt,bank_cnt)
+    print "Patch addr:      0x%08x"  %  addr
+    print "IRQ addr:        0x%08x"  %  irq_vector
+
+
+    out.write(rom[:addr])
+    out.write(code)
+    out.write(rom[addr+code_len:irq_vector])
+    out.write(addr_patch)
+    out.write(rom[irq_vector+2:])
+    print "%i Bytes written to %s" % (out.tell(),out_filename)
+    out.close()
+    print "Apply checksum fix"
+    cmd = "ucon64 -snes -chk %s 2>&1 > /dev/null" % out_filename
+    os.system(cmd)
+
+
+if __name__ == '__main__':
+    main()
+