commit 7b545b57200ab960894e873716862cafbf9321f0
author Nobuaki Sukegawa <nsuke@apache.org> Mon Jan 11 13:46:04 2016 +0900
committer Nobuaki Sukegawa <nsuke@apache.org> Wed Jan 13 20:25:23 2016 +0900
tree 56b13f14a20877edf7f7d33fe05e62e309f877ca
parent 7e286b0d143be88adbd84f2e1cbfec66196a6a57

THRIFT-3532 Add configurable string and container read size limit to Python protocols

This closes #787

lib/py/src/protocol/TBase.py

diff --git a/lib/py/src/protocol/TBase.py b/lib/py/src/protocol/TBase.py
index 4f71e11..d106f4e 100644
--- a/lib/py/src/protocol/TBase.py
+++ b/lib/py/src/protocol/TBase.py
@@ -53,7 +53,9 @@
        fastbinary is not None):
       fastbinary.decode_binary(self,
                                iprot.trans,
-                               (self.__class__, self.thrift_spec))
+                               (self.__class__, self.thrift_spec),
+                               iprot.string_length_limit,
+                               iprot.container_length_limit)
       return
     iprot.readStruct(self, self.thrift_spec)
 
@@ -90,5 +92,7 @@
       self = cls()
       return fastbinary.decode_binary(None,
                                       iprot.trans,
-                                      (self.__class__, self.thrift_spec))
+                                      (self.__class__, self.thrift_spec),
+                                      iprot.string_length_limit,
+                                      iprot.container_length_limit)
     return iprot.readStruct(cls, cls.thrift_spec, True)