hbc/pywii/pywii-tools/tmdupdatecr.py

30 lines
587 B
Python
Raw Normal View History

2016-11-23 06:35:12 +01:00
#!/usr/bin/env python
import sys, os, os.path
import pywii as wii
from Crypto.Hash import SHA
wii.loadkeys()
args = sys.argv[1:]
tmdfile = args.pop(0)
indir = args.pop(0)
print "updating content records of TMD file %s" % tmdfile
tmd = wii.WiiTmd(open(tmdfile, "rb").read())
for i, cr in enumerate(tmd.get_content_records()):
data = open(os.path.join(indir, "%08X" % cr.cid), "rb").read()
cr.sha = SHA.new(data).digest()
cr.size = len(data)
tmd.update_content_record(i, cr)
tmd.null_signature()
tmd.brute_sha()
tmd.update()
f = open(tmdfile, "wb")
f.write(tmd.data)
f.close()