Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2026-02-06 17:59:52 +00:00
Code Issues Projects Releases Packages Wiki Activity

raise ValueError when read_size > max_buffer_size.

Browse source
This commit is contained in:
INADA Naoki 2012-07-20 02:02:54 +09:00
parent e5462ff72f
commit 53ca2bb648
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
msgpack/_msgpack.pyx
View file

@ -338,14 +338,15 @@ cdef class Unpacker(object):
self.file_like_read = file_like.read
if not PyCallable_Check(self.file_like_read):
raise ValueError("`file_like.read` must be a callable.")
if not max_buffer_size:
max_buffer_size = INT_MAX
self.max_buffer_size = max_buffer_size
if read_size > max_buffer_size:
raise ValueError("read_size should be less or equal to max_buffer_size")
self.read_size = read_size
self.buf = <char*>malloc(read_size)
if self.buf == NULL:
raise MemoryError("Unable to allocate internal buffer.")
if max_buffer_size:
self.max_buffer_size = max_buffer_size
else:
self.max_buffer_size = INT_MAX
self.buf_size = read_size
self.buf_head = 0
self.buf_tail = 0