mirror of
https://github.com/fail0verflow/hbc.git
synced 2024-11-29 04:54:17 +01:00
33 lines
653 B
Python
Executable File
33 lines
653 B
Python
Executable File
#!/usr/bin/env python2
|
|
|
|
import sys, os, os.path
|
|
import pywii as wii
|
|
try:
|
|
from Cryptodome.Hash import SHA
|
|
except ImportError:
|
|
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()
|
|
|