1 files changed, 9 insertions, 6 deletions

diff --git a/reversing_tools/parse_mitm.py b/reversing_tools/parse_mitm.py
index ef61298..e59ce73 100755
--- a/reversing_tools/parse_mitm.py
+++ b/reversing_tools/parse_mitm.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 
-import sys, os
+import sys
+import os
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'python'))
 
 import json
@@ -11,7 +12,7 @@ from inform import InformSerializer
 
 def make_serializer(from_file):
     with open(from_file) as fp:
-        keystore = { i['mac']: i['x_authkey'] for i in json.load(fp) }
+        keystore = {i['mac']: i['x_authkey'] for i in json.load(fp)}
 
     return InformSerializer("", keystore)
 
@@ -23,14 +24,16 @@ def dumps_pretty(obj):
 if __name__ == "__main__":
     ser = make_serializer("data/devices.json")
 
-    with open('data/mitm/reboot.txt', 'rb') as fp, open('test.out', 'w') as fp2:
+    with open('data/mitm/onoff.txt', 'rb') as fp, open('test.out', 'w') as fp2:
         read = FlowReader(fp)
 
         for rec in read.stream():
             res = ser.parse(StringIO(rec.response.content))
             req = ser.parse(StringIO(rec.request.content))
 
-            print dumps_pretty(req.payload)
-            print dumps_pretty(res.payload)
-            print
+            # print req.payload
+            if res.payload['_type'] == 'cmd':
+                print dumps_pretty(res.payload)
+            # print dumps_pretty(req.payload)
+            # print dumps_pretty(res.payload)
             print