summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorINADA Naoki <inada-n@klab.com>2013-05-19 12:30:23 +0900
committerINADA Naoki <inada-n@klab.com>2013-05-19 12:30:23 +0900
commitbbe86e7a925f113337f6e24d3ff24ddc30137dd6 (patch)
tree89d7fc11e705996a155a9bc5026604cf841c39ea
parent08c56d66f6eecb99850897dd3c4987dea01e6d44 (diff)
downloadmsgpack-python-bbe86e7a925f113337f6e24d3ff24ddc30137dd6.tar.gz
Revert "Use new buffer interface."
This reverts commit 085db7f8dca2b4e2497bfb291238cd3ff2d06e28. Conflicts: msgpack/_unpacker.pyx
-rw-r--r--msgpack/_unpacker.pyx53
1 files changed, 28 insertions, 25 deletions
diff --git a/msgpack/_unpacker.pyx b/msgpack/_unpacker.pyx
index 06806bd..2644f15 100644
--- a/msgpack/_unpacker.pyx
+++ b/msgpack/_unpacker.pyx
@@ -2,6 +2,10 @@
#cython: embedsignature=True
from cpython cimport *
+cdef extern from "Python.h":
+ ctypedef char* const_void_ptr "const void*"
+ ctypedef struct PyObject
+ cdef int PyObject_AsReadBuffer(object o, const_void_ptr* buff, Py_ssize_t* buf_len) except -1
from libc.stdlib cimport *
from libc.string cimport *
@@ -15,8 +19,8 @@ from msgpack.exceptions import (
)
+
cdef extern from "unpack.h":
- ctypedef struct PyObject
ctypedef struct msgpack_user:
bint use_list
PyObject* object_hook
@@ -87,33 +91,32 @@ def unpackb(object packed, object object_hook=None, object list_hook=None,
cdef size_t off = 0
cdef int ret
- cdef Py_buffer buff
+ cdef char* buf
+ cdef Py_ssize_t buf_len
cdef char* cenc = NULL
cdef char* cerr = NULL
- PyObject_GetBuffer(packed, &buff, PyBUF_SIMPLE)
- try:
- if encoding is not None:
- if isinstance(encoding, unicode):
- encoding = encoding.encode('ascii')
- cenc = PyBytes_AsString(encoding)
-
- if unicode_errors is not None:
- if isinstance(unicode_errors, unicode):
- unicode_errors = unicode_errors.encode('ascii')
- cerr = PyBytes_AsString(unicode_errors)
-
- init_ctx(&ctx, object_hook, object_pairs_hook, list_hook, use_list, cenc, cerr)
- ret = unpack_construct(&ctx, <char*>buff.buf, buff.len, &off)
- if ret == 1:
- obj = unpack_data(&ctx)
- if off < buff.len:
- raise ExtraData(obj, PyBytes_FromStringAndSize(<char*>buff.buf+off, buff.len-off))
- return obj
- else:
- raise UnpackValueError("Unpack failed: error = %d" % (ret,))
- finally:
- PyBuffer_Release(&buff)
+ PyObject_AsReadBuffer(packed, <const_void_ptr*>&buf, &buf_len)
+
+ if encoding is not None:
+ if isinstance(encoding, unicode):
+ encoding = encoding.encode('ascii')
+ cenc = PyBytes_AsString(encoding)
+
+ if unicode_errors is not None:
+ if isinstance(unicode_errors, unicode):
+ unicode_errors = unicode_errors.encode('ascii')
+ cerr = PyBytes_AsString(unicode_errors)
+
+ init_ctx(&ctx, object_hook, object_pairs_hook, list_hook, use_list, cenc, cerr)
+ ret = unpack_construct(&ctx, buf, buf_len, &off)
+ if ret == 1:
+ obj = unpack_data(&ctx)
+ if off < buf_len:
+ raise ExtraData(obj, PyBytes_FromStringAndSize(buf+off, buf_len-off))
+ return obj
+ else:
+ raise UnpackValueError("Unpack failed: error = %d" % (ret,))
def unpack(object stream, object object_hook=None, object list_hook=None,