2Gb buffer size checks fixed for Python Builder

1 files changed, 8 insertions, 2 deletions

diff --git a/python/flatbuffers/builder.py b/python/flatbuffers/builder.py
index 068c413f..4c78d7a6 100644
--- a/python/flatbuffers/builder.py
+++ b/python/flatbuffers/builder.py
@@ -93,7 +93,7 @@ class Builder(object):
         The internal buffer is grown as needed.
         """
 
-        if not (0 <= initialSize < (2**UOffsetTFlags.bytewidth - 1)):
+        if not (0 <= initialSize <= self.MaxBufferSize()):
             msg = "flatbuffers: Cannot create Builder larger than 2 gigabytes."
             raise BuilderSizeError(msg)
 
@@ -104,6 +104,12 @@ class Builder(object):
         self.objectEnd = None
         self.vtables = []
 
+    def MaxBufferSize(self):
+        """
+        Maximum buffer size is 2Gb.
+        """
+        return 2**31
+
     def Output(self):
         """
         Output returns the portion of the buffer that has been used for
@@ -238,7 +244,7 @@ class Builder(object):
     def growByteBuffer(self):
         """Doubles the size of the byteslice, and copies the old data towards
            the end of the new buffer (since we build the buffer backwards)."""
-        if not len(self.Bytes) <= 2**20:
+        if not len(self.Bytes) <= self.MaxBufferSize():
             msg = "flatbuffers: cannot grow buffer beyond 2 gigabytes"
             raise BuilderSizeError(msg)