sanchayanmaity/gem5
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

util: Add support for detection of gzipped packet traces

Browse source 
This patch adds support for automatically detecting a gzipped packet
trace, thus accepting either a compressed or uncompressed trace.
This commit is contained in:
Andreas Hansson 2014-03-23 11:11:46 -04:00
parent 9f018d2f5a
commit 4dbffeb987
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
util/decode_packet_trace.py
View file

@ -48,6 +48,7 @@
# r,128,64,4000,0 # r,128,64,4000,0
# w,232123,64,500000,0 # w,232123,64,500000,0
import gzip
import protolib import protolib
import sys import sys
@ -81,7 +82,18 @@ def main():
exit(-1) exit(-1)
try: try:
proto_in = open(sys.argv[1], 'rb') # First see if this file is gzipped
try:
# Opening the file works even if it is not a gzip file
proto_in = gzip.open(sys.argv[1], 'rb')
# Force a check of the magic number by seeking in the
# file. If we do not do it here the error will occur when
# reading the first message.
proto_in.seek(1)
proto_in.seek(0)
except IOError:
proto_in = open(sys.argv[1], 'rb')
except IOError: except IOError:
print "Failed to open ", sys.argv[1], " for reading" print "Failed to open ", sys.argv[1], " for reading"
exit(-1) exit(-1)
@ -96,7 +108,7 @@ def main():
magic_number = proto_in.read(4) magic_number = proto_in.read(4)
if magic_number != "gem5": if magic_number != "gem5":
print "Unrecognized file" print "Unrecognized file", sys.argv[1]
exit(-1) exit(-1)
print "Parsing packet header" print "Parsing packet header"