From 86cea24d2bd727324a458a4ed282175353427028 Mon Sep 17 00:00:00 2001
From: Andrew Lamoureux <andrew@vector35.com>
Date: Thu, 21 Mar 2019 20:43:40 -0400
Subject: kaitai: recognize zip, rar, gzip

---
 python/examples/kaitai/kshelpers.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'python/examples')

diff --git a/python/examples/kaitai/kshelpers.py b/python/examples/kaitai/kshelpers.py
index d2e2f441..43b8e525 100644
--- a/python/examples/kaitai/kshelpers.py
+++ b/python/examples/kaitai/kshelpers.py
@@ -48,6 +48,12 @@ def idData(dataSample, length):
 		result = 'gif'
 	if dataSample[0:2] in [b'BM', b'BA', b'CI', b'CP', b'IC', b'PT'] and struct.unpack('<I', dataSample[2:6])[0]==length:
 		result = 'bmp'
+	if dataSample[0:2] == b'PK' and dataSample[2:4] in [b'\x01\x02', b'\x03\x04', b'\x05\x06']:
+		result = 'zip'
+	if dataSample[0:6] == b'Rar!\x1a\x07':
+		result = 'rar'
+	if dataSample[0:2] == b'\x1f\x8b' and dataSample[2:3]==b'\x08':
+		result = 'gzip'
 
 	#print('idData() returning \'%s\'' % result)
 	return result
-- 
cgit v1.3.1