From a8c20fedaf8e11462aa67f11d041143afaca2f20 Mon Sep 17 00:00:00 2001
From: Yannick Ulrich <yannick.ulrich@psi.ch>
Date: Tue, 18 Feb 2020 21:32:21 +0100
Subject: [PATCH] Compress in #17: format for iSsrf
This is taken from
https://mathematica.stackexchange.com/a/104696
---
pymule/compress.py | 24 ++++++++++++++++++++++++
1 file changed, 24 insertions(+)
diff --git a/pymule/compress.py b/pymule/compress.py
index 6691199..b6789c5 100644
--- a/pymule/compress.py
+++ b/pymule/compress.py
@@ -1,6 +1,7 @@
# https://mathematica.stackexchange.com/a/104696
import zlib
import base64
+import struct
import StringIO
@@ -11,3 +12,26 @@ def decomp(b):
if fp.read(4) != "!boR":
raise KeyError("Header is not matching")
return fp
+
+
+def parse(fp):
+ ans = []
+ cur = []
+ while True:
+ cmd = fp.read(1)
+ if cmd == '':
+ ans.append(cur)
+ return ans
+ elif cmd == 'i':
+ cur.append(struct.unpack('<i', fp.read(4))[0])
+ elif cmd == 'S' or cmd == 's':
+ le = struct.unpack('<i', fp.read(4))[0]
+ cur.append(fp.read(le))
+ elif cmd == 'r':
+ cur.append(struct.unpack('<d', fp.read(8))[0])
+ elif cmd == 'f':
+ le = struct.unpack('<i', fp.read(4))[0]
+ ans.append(cur[:])
+ cur = []
+ else:
+ raise KeyError("Unknown cmd %s" % cmd)
--
GitLab