#include <stddef.h>

#include <stdlib.h>

#include "sysdep.h"

#include <limits.h>

#include <string.h>

#ifdef __cplusplus

extern "C" {

#ifdef _MSC_VER

#define inline __inline

typedef struct msgpack_packer {

} msgpack_packer;

typedef struct Packer Packer;

static inline int msgpack_pack_write(msgpack_packer* pk, const char *data, size_t l)

{

char* buf = pk->buf;

size_t bs = pk->buf_size;

size_t len = pk->length;

if (len + l > bs) {

bs = (len + l) * 2;

buf = (char*)PyMem_Realloc(buf, bs);

if (!buf) {

PyErr_NoMemory();

return -1;

}

}

memcpy(buf + len, data, l);

len += l;

pk->buf = buf;

pk->buf_size = bs;

pk->length = len;

return 0;

}

#define msgpack_pack_append_buffer(user, buf, len) \

return msgpack_pack_write(user, (const char*)buf, len)

#include "pack_template.h"

static inline int

msgpack_pack_unicode(msgpack_packer *pk, PyObject *o, long long limit)

{

#if PY_MAJOR_VERSION >= 3

assert(PyUnicode_Check(o));

Py_ssize_t len;

const char* buf = PyUnicode_AsUTF8AndSize(o, &len);

if (buf == NULL)

return -1;

if (len > limit) {

return -2;

}

int ret = msgpack_pack_raw(pk, len);

if (ret) return ret;

return msgpack_pack_raw_body(pk, buf, len);

PyObject *bytes;

Py_ssize_t len;

int ret;

// py2

bytes = PyUnicode_AsUTF8String(o);

if (bytes == NULL)

return -1;

len = PyString_GET_SIZE(bytes);

if (len > limit) {

Py_DECREF(bytes);

return -2;

}

ret = msgpack_pack_raw(pk, len);

if (ret) {

Py_DECREF(bytes);

return -1;

}

ret = msgpack_pack_raw_body(pk, PyString_AS_STRING(bytes), len);

Py_DECREF(bytes);

return ret;

}

#ifdef __cplusplus

}