Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2025-10-26 23:24:10 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix stream unpacker broken.

Browse source
This commit is contained in:
inada-n 2009-12-16 22:05:31 +09:00
parent e72aa01fe5
commit 35a4d612c9
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
msgpack/_msgpack.pyx
View file

@ -233,7 +233,9 @@ cdef class Unpacker(object):
if self.buf: if self.buf:
free(self.buf); free(self.buf);
def __init__(self, file_like=None, int read_size=1024*1024): def __init__(self, file_like=None, int read_size=0):
if read_size == 0:
read_size = 1024*1024
self.file_like = file_like self.file_like = file_like
self.read_size = read_size self.read_size = read_size
self.waiting_bytes = [] self.waiting_bytes = []
@ -309,6 +311,7 @@ cdef class Unpacker(object):
self.fill_buffer() self.fill_buffer()
ret = template_execute(&self.ctx, self.buf, self.buf_tail, &self.buf_head) ret = template_execute(&self.ctx, self.buf, self.buf_tail, &self.buf_head)
if ret == 1: if ret == 1:
template_init(&self.ctx)
return template_data(&self.ctx) return template_data(&self.ctx)
elif ret == 0: elif ret == 0:
if self.file_like is not None: if self.file_like is not None:
@ -319,3 +322,10 @@ cdef class Unpacker(object):
def __iter__(self): def __iter__(self):
return UnpackIterator(self) return UnpackIterator(self)
# for debug.
#def _buf(self):
# return PyString_FromStringAndSize(self.buf, self.buf_tail)
#def _off(self):
# return self.buf_head