summaryrefslogtreecommitdiffstats
path: root/pyload/remote/thriftbackend/Protocol.py
diff options
context:
space:
mode:
Diffstat (limited to 'pyload/remote/thriftbackend/Protocol.py')
-rw-r--r--pyload/remote/thriftbackend/Protocol.py33
1 files changed, 33 insertions, 0 deletions
diff --git a/pyload/remote/thriftbackend/Protocol.py b/pyload/remote/thriftbackend/Protocol.py
new file mode 100644
index 000000000..3aefa167f
--- /dev/null
+++ b/pyload/remote/thriftbackend/Protocol.py
@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+
+import thrift
+
+
+class Protocol(thrift.protocol.TBinaryProtocol.thrift.protocol.TBinaryProtocol):
+
+ def writeString(self, str):
+ try:
+ str = str.encode("utf8", "ignore")
+ except Exception:
+ pass
+
+ self.writeI32(len(str))
+ self.trans.write(str)
+
+
+ def readString(self):
+ len = self.readI32()
+ str = self.trans.readAll(len)
+ try:
+ str = str.decode("utf8", "ignore")
+ except Exception:
+ pass
+
+ return str
+
+
+class ProtocolFactory(thrift.protocol.TBinaryProtocol.thrift.protocol.TBinaryProtocolFactory):
+
+ def getProtocol(self, trans):
+ prot = Protocol(trans, self.strictRead, self.strictWrite)
+ return prot