Incorrect buffer size check fixed.

This commit is contained in:
Frank Stein 2015-08-14 00:50:39 +03:00
parent 7bcbb19569
commit 33e4ab65e9
1 changed files with 2 additions and 2 deletions

View File

@ -245,11 +245,11 @@ class Builder(object):
def growByteBuffer(self): def growByteBuffer(self):
"""Doubles the size of the byteslice, and copies the old data towards """Doubles the size of the byteslice, and copies the old data towards
the end of the new buffer (since we build the buffer backwards).""" the end of the new buffer (since we build the buffer backwards)."""
if not len(self.Bytes) <= Builder.MAX_BUFFER_SIZE: if len(self.Bytes) == Builder.MAX_BUFFER_SIZE:
msg = "flatbuffers: cannot grow buffer beyond 2 gigabytes" msg = "flatbuffers: cannot grow buffer beyond 2 gigabytes"
raise BuilderSizeError(msg) raise BuilderSizeError(msg)
newSize = min( len(self.Bytes) * 2, Builder.MAX_BUFFER_SIZE ) newSize = min(len(self.Bytes) * 2, Builder.MAX_BUFFER_SIZE)
if newSize == 0: if newSize == 0:
newSize = 1 newSize = 1
bytes2 = bytearray(newSize) bytes2 = bytearray(newSize)