Merge branch 'master' of git@github.com:icefire/Wii.py

experimental/bmg.py

@@ -11,6 +11,13 @@ def nullterm(str_plus):
     else:
         return str_plus
         
+def denullterm(str_plus):
+    z = str_plus.find('\r\n')
+    if z != -1:
+        return str_plus[:z]
+    else:
+        return str_plus
+        
 class BMG(object):
 	class BMG_DAT1(Struct):
 		__endian__ = Struct.BE
@@ -21,6 +28,10 @@ class BMG(object):
 			return_string  = "Magic: %s\n" % self.magic
 			return_string += "Length: %08x\n " % self.length
 			return return_string
+		def write(self, file):
+			file.write(self.magic)
+			file.write(struct.pack('>I', self.length))
+			return
 
 	class BMG_INF1(Struct):
 		__endian__ = Struct.BE
@@ -35,6 +46,12 @@ class BMG(object):
 			return_string += "Count: %04x\n" % self.count
 			return_string += "Unknown: %04x\n" % self.unknown01
 			return return_string
+		def write(self, file):
+			file.write(self.magic)
+			file.write(struct.pack('>I', self.length))
+			file.write(struct.pack('>H', self.count))
+			file.write(struct.pack('>H', self.unknown01))
+			return
 
 	class BMG_Header(Struct):
 		__endian__ = Struct.BE
@@ -61,122 +78,180 @@ class BMG(object):
 			return_string += "Unknown05: %08x\n" % self.unknown05
 			return_string += "Unknown06: %08x\n" % self.unknown06
 			return return_string
+		def write(self, file):
+			file.write(self.magic)
+			file.write(self.magic2)
+			file.write(struct.pack('>I', self.length))
+			file.write(struct.pack('>I', self.chunk_cnt))
+			file.write(struct.pack('>B', self.unknown01))
+			file.write(struct.pack('>B', self.unknown02))
+			file.write(struct.pack('>H', self.unknown03))
+			file.write(struct.pack('>I', self.unknown04))
+			file.write(struct.pack('>I', self.unknown05))
+			file.write(struct.pack('>I', self.unknown06))
+			return
 
-	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):
+	def Unpack(self, data, out_file, debug):
+		file = open(out_file, 'wb')
+		if file:
 			pos = 0
 			header = self.BMG_Header()
 			header.unpack(data[pos:pos+len(header)])
 			pos += len(header)
+			if debug == True:
 				print header
-
 				print "\n%08x\n" % pos
 
 			info = self.BMG_INF1()
 			info.unpack(data[pos:pos+len(info)])
 			pos += len(info)
+			if debug == True:
 				print info
 
 			unknown = Struct.uint32(data[pos:pos+4], endian='>')
 			pos += 4
+			if debug == True:
 				print "Unknown: %08x" % unknown
-
 				print "\n%08x\n" % pos
 
 			offset_list = []
 			for x in xrange(info.count):
 				offset = Struct.uint32(data[pos:pos+4], endian='>')
 				pos +=  4
+				if debug == True:
 					print "Offset: %08x" % offset
 				offset_list.append(offset)
 
 			while pos % 0x10:
 				padding = Struct.uint32(data[pos:pos+4], endian='>')
 				pos += 4
+				if debug == True:
 					print "Padding: %08x" % padding
 
+			if debug == True:
 				print "\n%08x\n" % pos
 
 			dat1 = self.BMG_DAT1()
 			dat1.unpack(data[pos:pos+len(dat1)])
 			pos += len(dat1)
+			if debug == True:
 				print dat1
 
+			if debug == True:
 				print "\n%08x\n" % pos
 			temp = pos
 
 			unknown = Struct.uint16(data[pos:pos+2], endian='>')
 			pos += 2
+			if debug == True:
 				print "Unknown: %04x" % unknown
 
 			for x in xrange(info.count):
 				pos = temp + offset_list[x]
 				string = nullterm(data[pos:])
 				string = unicode(string, 'utf_16_be')
-			print "String: %s" % string
+				file.write(string.encode('utf-8'))
-
+				file.write('\r')
-		#''' TEST DATA
+				file.write('\n')
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+				if debug == True:
-		pos += 4
+					print "String: %s" % string.encode('utf-8')
-		print "Unknown: %08x" % unknown
+			file.close()
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+		else:
-		pos += 4
+			print "Could not open file for writing"
-		print "Unknown: %08x" % unknown
+			sys.exit(1)
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+	def write(self, data, out_file):
-		pos += 4
+		file = open(out_file, 'wb')
-		print "Unknown: %08x" % unknown
+		if file:
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			pos = 0
-		pos += 4
+			count = 0
-		print "Unknown: %08x" % unknown
+			strings_list = []
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			while pos < len(data):
-		pos += 4
+				temp = denullterm(data[pos:])
-		print "Unknown: %08x" % unknown
+				pos += len(temp) + 2
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+				string = unicode(temp, 'utf-8')
-		pos += 4
+				strings_list.append(string)
-		print "Unknown: %08x" % unknown
+				count += 1
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			offsets_list = []
-		pos += 4
+			offset = 2
-		print "Unknown: %08x" % unknown
+			for x in xrange(count):
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+				offsets_list.append(offset)
-		pos += 4
+				offset += len(strings_list[x]) * 2 + 2
-		print "Unknown: %08x" % unknown
+			dat1 = self.BMG_DAT1()
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			dat1.length = offset + 8 + (0x10 - (offset % 0x10))
-		pos += 4
+			dat_pad_to_add = 0x10 - (offset % 0x10)
-		print "Unknown: %08x" % unknown
+			info = self.BMG_INF1()
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			info.length = 0x10 + 4 * count + (0x10 - ((4 * count) % 0x10))
-		pos += 4
+			info_pad_to_add = 0x10 - ((4*count) % 0x10)
-		print "Unknown: %08x" % unknown
+			header = self.BMG_Header()
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			header.magic = "MESG"
-		pos += 4
+			header.magic2 = "bmg1"
-		print "Unknown: %08x" % unknown
+			header.length = info.length + dat1.length + 0x20
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			header.chunk_cnt = 2
-		pos += 4
+			header.unknown01 = 2
-		print "Unknown: %08x" % unknown
+			header.unknown02 = 0
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			header.unknown03 = 0
-		pos += 4
+			header.unknown04 = 0
-		print "Unknown: %08x" % unknown
+			header.unknown05 = 0
-		unknown = Struct.uint32(data[pos:pos+4], endian='>')
+			header.unknown06 = 0
-		pos += 4
+			header.write(file)
-		print "Unknown: %08x" % unknown
+			info.magic = "INF1"
-		#END TEST DATA '''
+			info.count = count
+			info.unknown01 = 4
+			info.write(file)
+			unknown = 0
+			file.write(struct.pack('>I', unknown))
+			offset_list = []
+			for x in offsets_list:
+				file.write(struct.pack('>I', x))
+			padding = 0
+			for x in xrange(info_pad_to_add):
+				file.write(struct.pack('>B', padding))
+			dat1.magic = "DAT1"
+			dat1.write(file)
+			unknown = 0
+			file.write(struct.pack('>H', unknown))
+			for x in strings_list:
+				file.write(x.encode('utf_16_be'))
+				file.write('\x00\x00')
+			file.write("\n")
+			padding = 0
+			for x in xrange(dat_pad_to_add):
+				file.write(struct.pack('>B', padding))
+			file.close()
+		else:
+			print "Could not open file for writing"
+			sys.exit(1)
 
 def main():
 	if len(sys.argv) == 1:
-		print 'Usage: python bmg.py'
+		print 'Usage: python bmg.py -r <filename.bmg> <filename.hog>'
+		print '                  == OR ==                           '
+		print 'Usage: python bmg.py -w> <filename.hog> <filename.bmg'
 		sys.exit(1)
-	f = open(sys.argv[1], 'rb')
+	if sys.argv[1] == "-r":
+		f = open(sys.argv[2], 'rb')
 		if f:
 			bmg_buffer = f.read()
 			f.close()
 		else:
 			print 'Could not open file for reading'
 			sys.exit(1)
-
+		bmg = BMG(bmg_buffer, sys.argv[3], debug=False)
-	bmg = BMG(bmg_buffer)
+	elif sys.argv[1] == "-w":
+		f = open(sys.argv[2], 'rb')
+		if f:
+			bmg_buffer = f.read()
+			f.close()
+		else:
+			print "Could not open file for reading"
+			sys.exit(1)
+		bmg = BMG()
+		bmg.write(bmg_buffer, sys.argv[3])
 
 if __name__ == "__main__":
 	main()

experimental/znd.py

@@ -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):
+	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
+			if debug == True:
 				print "Count: %08x" % count
-
 				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)
 
+			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')
-			print "String: %s" % string
+				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 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)
-	znd = ZND(znd_buffer)
+	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 'Usage: python znd.py -r <filename.znd> <filename.hog>'
+		print '                 == OR ==                            '
+		print 'Usage: python znd.py -w <filename.hog> <filename.znd>'
+		sys.exit(1)
 
 if __name__ == "__main__":
 	main()