mirror of
https://github.com/ahezard/twl_firm_patcher.git
synced 2025-06-19 05:45:34 -04:00
125 lines
4.1 KiB
Python
125 lines
4.1 KiB
Python
# -*- coding: utf8 -*-
|
|
# Patch an .nds (works with homebrew and ds demo only) to make it ready for make_cia
|
|
#
|
|
# 2016-02-28, Ahezard
|
|
#
|
|
# inspired by
|
|
# Apache Thunder .nds edited files and comments
|
|
# https://github.com/Relys/Project_CTR/blob/master/makerom/srl.h
|
|
# https://dsibrew.org/wiki/DSi_Cartridge_Header
|
|
# if the header size of the input nds file is 0x200 (homebrew)
|
|
# the header size of the output nds file will be patched to 0x4000 (normal ds/dsi header), 0x3E00 offset
|
|
|
|
from struct import *
|
|
from collections import namedtuple
|
|
from collections import OrderedDict
|
|
from pprint import pprint
|
|
import os, sys
|
|
import binascii
|
|
import argparse
|
|
import mmap
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='extract devLauncherSrl from the twl_bg.cxi')
|
|
parser.add_argument('--srl', help='srl file')
|
|
parser.add_argument('--out', help='output code file')
|
|
args = parser.parse_args()
|
|
|
|
#
|
|
# CRC16 MODULE
|
|
#
|
|
# includes CRC16 and CRC16 MODBUS
|
|
#
|
|
|
|
from ctypes import c_ushort
|
|
|
|
# from https://github.com/cristianav/PyCRC/blob/master/demo.py
|
|
class CRC16(object):
|
|
crc16_tab = []
|
|
|
|
# The CRC's are computed using polynomials. Here is the most used
|
|
# coefficient for CRC16
|
|
crc16_constant = 0xA001 # 40961
|
|
|
|
def __init__(self, modbus_flag=False):
|
|
# initialize the precalculated tables
|
|
if not len(self.crc16_tab):
|
|
self.init_crc16()
|
|
self.mdflag = bool(modbus_flag)
|
|
|
|
def calculate(self, input_data=None):
|
|
try:
|
|
is_string = isinstance(input_data, str)
|
|
is_bytes = isinstance(input_data, (bytes, bytearray))
|
|
|
|
if not is_string and not is_bytes:
|
|
raise Exception("Please provide a string or a byte sequence "
|
|
"as argument for calculation.")
|
|
|
|
crc_value = 0x0000 if not self.mdflag else 0xffff
|
|
|
|
for c in input_data:
|
|
d = ord(c) if is_string else c
|
|
tmp = crc_value ^ d
|
|
rotated = crc_value >> 8
|
|
crc_value = rotated ^ self.crc16_tab[(tmp & 0x00ff)]
|
|
|
|
return crc_value
|
|
except Exception as e:
|
|
print("EXCEPTION(calculate): {}".format(e))
|
|
|
|
def init_crc16(self):
|
|
"""The algorithm uses tables with precalculated values"""
|
|
for i in range(0, 256):
|
|
crc = c_ushort(i).value
|
|
for j in range(0, 8):
|
|
if crc & 0x0001:
|
|
crc = c_ushort(crc >> 1).value ^ self.crc16_constant
|
|
else:
|
|
crc = c_ushort(crc >> 1).value
|
|
self.crc16_tab.append(crc)
|
|
|
|
def getSize(fileobject):
|
|
current = fileobject.tell()
|
|
fileobject.seek(0,2) # move the cursor to the end of the file
|
|
size = fileobject.tell()
|
|
fileobject.seek(current,0)
|
|
return size
|
|
|
|
def skipUntilAddress(f_in,f_out, caddr, taddr):
|
|
chunk = f_in.read(taddr-caddr)
|
|
f_out.write(chunk)
|
|
|
|
def writeBlankuntilAddress(f_out, caddr, taddr):
|
|
f_out.write("\x00"*(taddr-caddr))
|
|
|
|
srl_fname=args.srl
|
|
|
|
# write the file
|
|
files = open(srl_fname, 'rb')
|
|
filew = open(args.out, "wb")
|
|
|
|
filesize=getSize(files)
|
|
|
|
# Sono patch for deblured scaling filter : https://gbatemp.net/threads/sharp-ds-i-mode-scaling-filters.542694/
|
|
srlsizeoffset=0x40000
|
|
pattern= "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x20\x00\x40\x00\x00\x00\x20\x00\x40\x00\x40\x00\x20\x00\x00\x00\x40\x00\x20\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
patch = "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x20\x00\x40\x00\x40\x00\x40\x00\x40\x00\x20\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
mm = mmap.mmap(files.fileno(), filesize-srlsizeoffset, None, mmap.ACCESS_READ, srlsizeoffset)
|
|
patternLoc=mm.find(pattern)
|
|
skipUntilAddress(files,filew,0,srlsizeoffset+patternLoc-0x3C)
|
|
files.read(len(patch))
|
|
filew.write(patch)
|
|
current=srlsizeoffset+len(patch)
|
|
|
|
print "pattern offset"
|
|
print srlsizeoffset+patternLoc
|
|
|
|
print "deblur patch offset"
|
|
print srlsizeoffset+patternLoc-0x3C
|
|
|
|
skipUntilAddress(files,filew,current,filesize)
|
|
|
|
filew.close()
|
|
files.close()
|
|
|