BigfootDev/flatbuffers
1
0
Fork 1
mirror of https://github.com/google/flatbuffers.git synced 2026-06-25 10:18:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

2Gb buffer size checks fixed for Python Builder

Browse Source
This commit is contained in:
Frank Stein
2015-07-14 00:10:11 +03:00
parent 4798456df6
commit a1d801c375
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/flatbuffers/builder.py
View File

@@ -93,7 +93,7 @@ class Builder(object):
The internal buffer is grown as needed. 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." msg = "flatbuffers: Cannot create Builder larger than 2 gigabytes."
raise BuilderSizeError(msg) raise BuilderSizeError(msg)
@@ -104,6 +104,12 @@ class Builder(object):
self.objectEnd = None self.objectEnd = None
self.vtables = [] self.vtables = []
def MaxBufferSize(self):
"""
Maximum buffer size is 2Gb.
"""
return 2**31
def Output(self): def Output(self):
""" """
Output returns the portion of the buffer that has been used for Output returns the portion of the buffer that has been used for
@@ -238,7 +244,7 @@ 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) <= 2**20: if not len(self.Bytes) <= self.MaxBufferSize():
msg = "flatbuffers: cannot grow buffer beyond 2 gigabytes" msg = "flatbuffers: cannot grow buffer beyond 2 gigabytes"
raise BuilderSizeError(msg) raise BuilderSizeError(msg)