summaryrefslogtreecommitdiff
path: root/msgpack/pack.h
diff options
context:
space:
mode:
Diffstat (limited to 'msgpack/pack.h')
-rw-r--r--msgpack/pack.h29
1 files changed, 18 insertions, 11 deletions
diff --git a/msgpack/pack.h b/msgpack/pack.h
index ea97601..d7e0867 100644
--- a/msgpack/pack.h
+++ b/msgpack/pack.h
@@ -25,7 +25,9 @@ extern "C" {
#endif
typedef struct msgpack_packer {
- PyObject* writer;
+ char *buf;
+ size_t length;
+ size_t buf_size;
} msgpack_packer;
typedef struct Packer Packer;
@@ -64,16 +66,21 @@ static inline int msgpack_pack_raw_body(msgpack_packer* pk, const void* b, size_
static inline int msgpack_pack_write(msgpack_packer* pk, const char *data, size_t l)
{
- PyObject *buf, *ret;
-
- buf = PyBuffer_FromMemory((void*)data, l);
- //buf = PyString_FromStringAndSize(data, l);
- if (buf == NULL) return -1;
-
- ret = PyObject_CallFunctionObjArgs(pk->writer, buf, NULL);
- Py_DECREF(buf);
- if (ret == NULL) return -1;
- Py_DECREF(ret);
+ char* buf = pk->buf;
+ size_t bs = pk->buf_size;
+ size_t len = pk->length;
+
+ if (len + l > bs) {
+ bs = (len + l) * 2;
+ buf = realloc(pk->buf, bs);
+ if (!buf) return -1;
+ }
+ memcpy(buf + len, data, l);
+ len += l;
+
+ pk->buf = buf;
+ pk->buf_size = bs;
+ pk->length = len;
return 0;
}