grp/Wii.py
1
Fork 0
mirror of https://github.com/grp/Wii.py.git synced 2025-06-18 14:55:35 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

added writing and finished znd

Browse Source
This commit is contained in:
megazig 2009-07-01 07:12:41 -05:00
parent 803f56f5c7
commit 88264d5c99
1 changed files with 93 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

123
experimental/znd.py
View File

@ -11,50 +11,113 @@ def nullterm(str_plus):
else:
return str_plus
def denullterm(str_plus):
z = str_plus.find('\x0a')
if z != -1:
return str_plus[:z]
else:
return str_plus
class ZND(object):
def __init__(self, data):
def __init__(self, data=None, out_file=None, debug=False):
self.data = []
if data != None:
self.Unpack(data)
self.Unpack(data, out_file, debug)
def Unpack(self, data):
pos = 0
def Unpack(self, data, out_file, debug=False):
file = open(out_file, 'wb')
if file:
pos = 0
count = Struct.uint32(data[pos:pos+4], endian='>')
pos += 4
print "Count: %08x" % count
print "\n%08x\n" % pos
offset_list = []
for x in xrange(count):
offset = Struct.uint32(data[pos:pos+4], endian='>')
count = Struct.uint32(data[pos:pos+4], endian='>')
pos += 4
print "Offset: %08x" % offset
offset_list.append(offset)
if debug == True:
print "Count: %08x" % count
print "\n%08x\n" % pos
print "\n%08x\n" % pos
offset_list = []
for x in xrange(count):
offset = Struct.uint32(data[pos:pos+4], endian='>')
pos += 4
if debug == True:
print "Offset: %08x" % offset
offset_list.append(offset)
for x in xrange(count):
pos = offset_list[x]
string = nullterm(data[pos:])
string = unicode(string, 'utf_16_be')
print "String: %s" % string
pos += len(string)
if debug == True:
print "\n%08x\n" % pos
for x in xrange(count):
pos = offset_list[x]
string = nullterm(data[pos:])
string = unicode(string, 'utf_16_be')
if debug == True:
print "String: %s" % string.encode('utf-8')
pos += len(string)
file.write(string.encode('utf-8'))
file.write('\n')
file.close()
else:
print "Could not open file for writing"
sys.exit(1)
def write(self, data, out_file):
file = open(out_file, 'wb')
if file:
pos = 0
count = 0
string_list = []
while pos < len(data):
temp = denullterm(data[pos:])
pos += len(temp) + 1
string = unicode(temp, 'utf-8')
string_list.append(string)
count += 1
file.write(struct.pack('>I', count))
pos = count * 4 + 4
for x in xrange(count):
file.write(struct.pack('>I', pos))
pos += len(string_list[x]) * 2 + 2
for x in xrange(count):
file.write(string_list[x].encode('utf_16_be'))
file.write('\x00\x00')
while pos % 0x20:
pos += 2
file.write('\x00\x00')
file.close()
else:
print "Could not open file for writing"
sys.exit(1)
def main():
if len(sys.argv) == 1:
print 'Usage: python znd.py <filename>'
print 'Usage: python znd.py -r <filename.znd> <filename.hog>'
print ' == OR == '
print 'Usage: python znd.py -w <filename.hog> <filename.znd>'
sys.exit(1)
f = open(sys.argv[1], 'rb')
if f:
znd_buffer = f.read()
f.close()
if sys.argv[1] == "-r":
f = open(sys.argv[2], 'rb')
if f:
znd_buffer = f.read()
f.close()
else:
print 'Could not open file for reading'
sys.exit(1)
znd = ZND(znd_buffer, sys.argv[3], debug=False)
elif sys.argv[1] == "-w":
f = open(sys.argv[2], 'rb')
if f:
buffer = f.read()
f.close()
else:
print 'Could not open file for reading'
sys.exit(1)
znd = ZND()
znd.write(buffer, sys.argv[3])
else:
print 'Could not open file for reading'
print 'Usage: python znd.py -r <filename.znd> <filename.hog>'
print ' == OR == '
print 'Usage: python znd.py -w <filename.hog> <filename.znd>'
sys.exit(1)
znd = ZND(znd_buffer)
if __name__ == "__main__":
main()