more fixes

1 files changed, 30 insertions, 0 deletions

diff --git a/module/remote/thriftbackend/Protocol.py b/module/remote/thriftbackend/Protocol.py
new file mode 100644
index 000000000..c42d01459
--- /dev/null
+++ b/module/remote/thriftbackend/Protocol.py
@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+
+from thrift.protocol import TBinaryProtocol
+
+class Protocol(TBinaryProtocol.TBinaryProtocol):
+    def writeString(self, str):
+        try:
+            str = str.encode("utf8", "ignore")
+        except Exception, e:
+            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:
+            pass
+
+        return str
+
+
+class ProtocolFactory(TBinaryProtocol.TBinaryProtocolFactory):
+
+    def getProtocol(self, trans):
+        prot = Protocol(trans, self.strictRead, self.strictWrite)
+        return prot
\ No newline at end of file