decode_supout.py

#!/usr/bin/python
# (C) Kirils Solovjovs, 2015

import sys,os,base64,zlib,re

#           111111 11112222
#01234567 89012345 67890123
#cdefgh56 78abGH12 34ABCDEF  (from)
#abcdefgh 12345678 ABCDEFGH  (to)
#revtribitmap=[2,3,4,5,6,7,12,13,14,15,0,1,22,23,8,9,10,11,16,17,18,19,20,21]
tribitmap=[10,11,0,1,2,3,4,5,14,15,16,17,6,7,8,9,18,19,20,21,22,23,12,13]

def tribit(content):
	#origlen=len(content)
	#while len(content)%3:
	#	content= content + "\x00"
		
	result=""
	for i in xrange(0, len(content) - 1,3):	
		goodtribit=0
		badtribit=ord(content[i])*0x10000+ord(content[i+1])*0x100+ord(content[i+2])
		for mangle in tribitmap:
			goodtribit = (goodtribit<<1) + (1 if ((badtribit & (0x800000>>mangle))>0) else 0)
			
		for move in [16,8,0]:
			result=result+chr((goodtribit >> move)& 0xff)

	return result
	#return result[0:origlen]

if len(sys.argv) > 1:
	if len(sys.argv) > 2:
		dir = sys.argv[2]
	else:
		dir = sys.argv[1]+"_contents/"
else:
	raise Exception("Usage: "+sys.argv[0]+" <supout.rif> [output_folder]")


if not os.access(sys.argv[1], os.R_OK):
	raise Exception("Can't read file "+sys.argv[1])

if not os.path.exists(dir):
    os.makedirs(dir)

if not os.access(dir, os.W_OK):
	raise Exception("Directory "+dir+" not writeable")
	
if os.listdir(dir)!=[]:
	raise Exception("Directory "+dir+" not empty")

i=0
with open(sys.argv[1], 'r') as my_file:
    sections=my_file.read().replace("--BEGIN ROUTEROS SUPOUT SECTION\r\n",":").replace("--END ROUTEROS SUPOUT SECTION\r\n",":").replace("::",":").split(":")
    for sect in sections:
			if len(sect.strip())>0:
				i= i + 1
				
				out = tribit(base64.b64decode(sect.replace("=","A")))
				
				[name,zipped]=out.split("\x00",1);
				print '%02d' % i,name.ljust(23),
				if not i%3:
					print
					
				res=zlib.decompress(zipped)
				fo = open(dir+"/"+str(i).zfill(2)+"_"+re.sub('[^a-z0-9\.-]','_',name), "wb")
				fo.write(res);
				fo.close()