Home | History | Annotate | Download | only in workloads
      1 #
      2 # Uncompress a file generated via atrace -z
      3 #
      4 # Usage: python atrace-uncompress.py infile > outfile
      5 #
      6 import sys, zlib
      7 
      8 def main():
      9 
     10 	if len(sys.argv) != 2:
     11 		print >> sys.stderr, ('Usage: %s inputfile' % sys.argv[0])
     12 		sys.exit(1)
     13 
     14 	infile = open(sys.argv[1], "rb")
     15 	out = infile.read()
     16 	parts = out.split('\nTRACE:', 1)
     17 
     18 	data = ''.join(parts[1])
     19 
     20 	# Remove CR characters
     21 	if data.startswith('\r\n'):
     22 		data = data.replace('\r\n', '\n')
     23 
     24 	# Skip the initial newline.
     25 	data = data[1:]
     26 
     27 	if not data:
     28 		print >> sys.stderr, ('No trace data found')
     29 		sys.exit(1)
     30 
     31 	out = zlib.decompress(data)
     32 	print(out)
     33 
     34 if __name__ == '__main__':
     35 	main()
     36