#define PY_SSIZE_T_CLEAN

#include "Python.h"

#include "longintrepr.h"

#include "code.h"

#include "marshal.h"

#define ABS(x) ((x) < 0 ? -(x) : (x))

#if defined(MS_WINDOWS) && defined(_DEBUG)

#define MAX_MARSHAL_STACK_DEPTH 1000

#define MAX_MARSHAL_STACK_DEPTH 2000

#define TYPE_NULL '0'

#define TYPE_NONE 'N'

#define TYPE_FALSE 'F'

#define TYPE_TRUE 'T'

#define TYPE_STOPITER 'S'

#define TYPE_ELLIPSIS '.'

#define TYPE_INT 'i'

#define TYPE_INT64 'I'

#define TYPE_FLOAT 'f'

#define TYPE_BINARY_FLOAT 'g'

#define TYPE_COMPLEX 'x'

#define TYPE_BINARY_COMPLEX 'y'

#define TYPE_LONG 'l'

#define TYPE_STRING 's'

#define TYPE_INTERNED 't'

#define TYPE_STRINGREF 'R'

#define TYPE_TUPLE '('

#define TYPE_LIST '['

#define TYPE_DICT '{'

#define TYPE_CODE 'c'

#define TYPE_UNICODE 'u'

#define TYPE_UNKNOWN '?'

#define TYPE_SET '<'

#define TYPE_FROZENSET '>'

#define WFERR_OK 0

#define WFERR_UNMARSHALLABLE 1

#define WFERR_NESTEDTOODEEP 2

#define WFERR_NOMEMORY 3

typedef struct {

FILE *fp;

int error; /* see WFERR_* values */

int depth;

/* If fp == NULL, the following are valid: */

PyObject *str;

char *ptr;

char *end;

PyObject *strings; /* dict on marshal, list on unmarshal */

int version;

} WFILE;

#define w_byte(c, p) if (((p)->fp)) putc((c), (p)->fp); \

else if ((p)->ptr != (p)->end) *(p)->ptr++ = (c); \

else w_more(c, p)

static void

w_more(int c, WFILE *p)

{

Py_ssize_t size, newsize;

if (p->str == NULL)

return; /* An error already occurred */

size = PyString_Size(p->str);

newsize = size + size + 1024;

if (newsize > 32*1024*1024) {

newsize = size + (size >> 3); /* 12.5% overallocation */

}

if (_PyString_Resize(&p->str, newsize) != 0) {

p->ptr = p->end = NULL;

}

else {

p->ptr = PyString_AS_STRING((PyStringObject *)p->str) + size;

p->end =

PyString_AS_STRING((PyStringObject *)p->str) + newsize;

*p->ptr++ = Py_SAFE_DOWNCAST(c, int, char);

}

}

static void

w_string(const char *s, Py_ssize_t n, WFILE *p)

{

if (p->fp != NULL) {

fwrite(s, 1, n, p->fp);

}

else {

while (--n >= 0) {

w_byte(*s, p);

s++;

}

}

}

static void

w_short(int x, WFILE *p)

{

w_byte((char)( x & 0xff), p);

w_byte((char)((x>> 8) & 0xff), p);

}

static void

w_long(long x, WFILE *p)

{

w_byte((char)( x & 0xff), p);

w_byte((char)((x>> 8) & 0xff), p);

w_byte((char)((x>>16) & 0xff), p);

w_byte((char)((x>>24) & 0xff), p);

}

#if SIZEOF_LONG > 4

static void

w_long64(long x, WFILE *p)

{

w_long(x, p);

w_long(x>>32, p);

}

#define SIZE32_MAX 0x7FFFFFFF

#if SIZEOF_SIZE_T > 4

# define W_SIZE(n, p) do { \

if ((n) > SIZE32_MAX) { \

(p)->depth--; \

(p)->error = WFERR_UNMARSHALLABLE; \

return; \

} \

w_long((long)(n), p); \

} while(0)

# define W_SIZE w_long

static void

w_pstring(const char *s, Py_ssize_t n, WFILE *p)

{

W_SIZE(n, p);

w_string(s, n, p);

}

#define PyLong_MARSHAL_SHIFT 15

#define PyLong_MARSHAL_BASE ((short)1 << PyLong_MARSHAL_SHIFT)

#define PyLong_MARSHAL_MASK (PyLong_MARSHAL_BASE - 1)

#if PyLong_SHIFT % PyLong_MARSHAL_SHIFT != 0

#error "PyLong_SHIFT must be a multiple of PyLong_MARSHAL_SHIFT"

#define PyLong_MARSHAL_RATIO (PyLong_SHIFT / PyLong_MARSHAL_SHIFT)

static void

w_PyLong(const PyLongObject *ob, WFILE *p)

{

Py_ssize_t i, j, n, l;

digit d;

w_byte(TYPE_LONG, p);

if (Py_SIZE(ob) == 0) {

w_long((long)0, p);

return;

}

/* set l to number of base PyLong_MARSHAL_BASE digits */

n = ABS(Py_SIZE(ob));

l = (n-1) * PyLong_MARSHAL_RATIO;

d = ob->ob_digit[n-1];

assert(d != 0); /* a PyLong is always normalized */

do {

d >>= PyLong_MARSHAL_SHIFT;

l++;

} while (d != 0);

if (l > SIZE32_MAX) {

p->depth--;

p->error = WFERR_UNMARSHALLABLE;

return;

}

w_long((long)(Py_SIZE(ob) > 0 ? l : -l), p);

for (i=0; i < n-1; i++) {

d = ob->ob_digit[i];

for (j=0; j < PyLong_MARSHAL_RATIO; j++) {

w_short(d & PyLong_MARSHAL_MASK, p);

d >>= PyLong_MARSHAL_SHIFT;

}

assert (d == 0);

}

d = ob->ob_digit[n-1];

do {

w_short(d & PyLong_MARSHAL_MASK, p);

d >>= PyLong_MARSHAL_SHIFT;

} while (d != 0);

}

static void

w_object(PyObject *v, WFILE *p)

{

Py_ssize_t i, n;

p->depth++;

if (p->depth > MAX_MARSHAL_STACK_DEPTH) {

p->error = WFERR_NESTEDTOODEEP;

}

else if (v == NULL) {

w_byte(TYPE_NULL, p);

}

else if (v == Py_None) {

w_byte(TYPE_NONE, p);

}

else if (v == PyExc_StopIteration) {

w_byte(TYPE_STOPITER, p);

}

else if (v == Py_Ellipsis) {

w_byte(TYPE_ELLIPSIS, p);

}

else if (v == Py_False) {

w_byte(TYPE_FALSE, p);

}

else if (v == Py_True) {

w_byte(TYPE_TRUE, p);

}

else if (PyInt_CheckExact(v)) {

long x = PyInt_AS_LONG((PyIntObject *)v);

#if SIZEOF_LONG > 4

long y = Py_ARITHMETIC_RIGHT_SHIFT(long, x, 31);

if (y && y != -1) {

w_byte(TYPE_INT64, p);

w_long64(x, p);

}

else

{

w_byte(TYPE_INT, p);

w_long(x, p);

}

}

else if (PyLong_CheckExact(v)) {

PyLongObject *ob = (PyLongObject *)v;

w_PyLong(ob, p);

}

else if (PyFloat_CheckExact(v)) {

if (p->version > 1) {

unsigned char buf[8];

if (_PyFloat_Pack8(PyFloat_AsDouble(v),

buf, 1) < 0) {

p->error = WFERR_UNMARSHALLABLE;

return;

}

w_byte(TYPE_BINARY_FLOAT, p);

w_string((char*)buf, 8, p);

}

else {

char *buf = PyOS_double_to_string(PyFloat_AS_DOUBLE(v),

'g', 17, 0, NULL);

if (!buf) {

p->error = WFERR_NOMEMORY;

return;

}

n = strlen(buf);

w_byte(TYPE_FLOAT, p);

w_byte((int)n, p);

w_string(buf, n, p);

PyMem_Free(buf);

}

}

#ifndef WITHOUT_COMPLEX

else if (PyComplex_CheckExact(v)) {

if (p->version > 1) {

unsigned char buf[8];

if (_PyFloat_Pack8(PyComplex_RealAsDouble(v),

buf, 1) < 0) {

p->error = WFERR_UNMARSHALLABLE;

return;

}

w_byte(TYPE_BINARY_COMPLEX, p);

w_string((char*)buf, 8, p);

if (_PyFloat_Pack8(PyComplex_ImagAsDouble(v),

buf, 1) < 0) {

p->error = WFERR_UNMARSHALLABLE;

return;

}

w_string((char*)buf, 8, p);

}

else {

char *buf;

w_byte(TYPE_COMPLEX, p);

buf = PyOS_double_to_string(PyComplex_RealAsDouble(v),

'g', 17, 0, NULL);

if (!buf) {

p->error = WFERR_NOMEMORY;

return;

}

n = strlen(buf);

w_byte((int)n, p);

w_string(buf, n, p);

PyMem_Free(buf);

buf = PyOS_double_to_string(PyComplex_ImagAsDouble(v),

'g', 17, 0, NULL);

if (!buf) {

p->error = WFERR_NOMEMORY;

return;

}

n = strlen(buf);

w_byte((int)n, p);

w_string(buf, n, p);

PyMem_Free(buf);

}

}

else if (PyString_CheckExact(v)) {

if (p->strings && PyString_CHECK_INTERNED(v)) {

PyObject *o = PyDict_GetItem(p->strings, v);

if (o) {

long w = PyInt_AsLong(o);

w_byte(TYPE_STRINGREF, p);

w_long(w, p);

goto exit;

}

else {

int ok;

o = PyInt_FromSsize_t(PyDict_Size(p->strings));

ok = o &&

PyDict_SetItem(p->strings, v, o) >= 0;

Py_XDECREF(o);

if (!ok) {

p->depth--;

p->error = WFERR_UNMARSHALLABLE;

return;

}

w_byte(TYPE_INTERNED, p);

}

}

else {

w_byte(TYPE_STRING, p);

}

w_pstring(PyBytes_AS_STRING(v), PyString_GET_SIZE(v), p);

}

#ifdef Py_USING_UNICODE

else if (PyUnicode_CheckExact(v)) {

PyObject *utf8;

utf8 = PyUnicode_AsUTF8String(v);

if (utf8 == NULL) {

p->depth--;

p->error = WFERR_UNMARSHALLABLE;

return;

}

w_byte(TYPE_UNICODE, p);

w_pstring(PyString_AS_STRING(utf8), PyString_GET_SIZE(utf8), p);

Py_DECREF(utf8);

}

else if (PyTuple_CheckExact(v)) {

w_byte(TYPE_TUPLE, p);

n = PyTuple_Size(v);

W_SIZE(n, p);

for (i = 0; i < n; i++) {

w_object(PyTuple_GET_ITEM(v, i), p);

}

}

else if (PyList_CheckExact(v)) {

w_byte(TYPE_LIST, p);

n = PyList_GET_SIZE(v);

W_SIZE(n, p);

for (i = 0; i < n; i++) {

w_object(PyList_GET_ITEM(v, i), p);

}

}

else if (PyDict_CheckExact(v)) {

Py_ssize_t pos;

PyObject *key, *value;

w_byte(TYPE_DICT, p);

/* This one is NULL object terminated! */

pos = 0;

while (PyDict_Next(v, &pos, &key, &value)) {

w_object(key, p);

w_object(value, p);

}

w_object((PyObject *)NULL, p);

}

else if (PyAnySet_CheckExact(v)) {

PyObject *value, *it;

if (PyObject_TypeCheck(v, &PySet_Type))

w_byte(TYPE_SET, p);

else

w_byte(TYPE_FROZENSET, p);

n = PyObject_Size(v);

if (n == -1) {

p->depth--;

p->error = WFERR_UNMARSHALLABLE;

return;

}

W_SIZE(n, p);

it = PyObject_GetIter(v);

if (it == NULL) {

p->depth--;

p->error = WFERR_UNMARSHALLABLE;

return;

}

while ((value = PyIter_Next(it)) != NULL) {

w_object(value, p);

Py_DECREF(value);

}

Py_DECREF(it);

if (PyErr_Occurred()) {

p->depth--;

p->error = WFERR_UNMARSHALLABLE;

return;

}

}

else if (PyCode_Check(v)) {

PyCodeObject *co = (PyCodeObject *)v;

w_byte(TYPE_CODE, p);

w_long(co->co_argcount, p);

w_long(co->co_nlocals, p);

w_long(co->co_stacksize, p);

w_long(co->co_flags, p);

w_object(co->co_code, p);

w_object(co->co_consts, p);

w_object(co->co_names, p);

w_object(co->co_varnames, p);

w_object(co->co_freevars, p);

w_object(co->co_cellvars, p);

w_object(co->co_filename, p);

w_object(co->co_name, p);

w_long(co->co_firstlineno, p);

w_object(co->co_lnotab, p);

}

else if (PyObject_CheckReadBuffer(v)) {

/* Write unknown buffer-style objects as a string */

char *s;

PyBufferProcs *pb = v->ob_type->tp_as_buffer;

w_byte(TYPE_STRING, p);

n = (*pb->bf_getreadbuffer)(v, 0, (void **)&s);

w_pstring(s, n, p);

}

else {

w_byte(TYPE_UNKNOWN, p);

p->error = WFERR_UNMARSHALLABLE;

}

exit:

p->depth--;

}

PyMarshal_WriteLongToFile(long x, FILE *fp, int version)

{

WFILE wf;

wf.fp = fp;

wf.str = NULL;

wf.ptr = NULL;

wf.end = NULL;

wf.error = WFERR_OK;

wf.depth = 0;

wf.strings = NULL;

wf.version = version;

w_long(x, &wf);

}

PyMarshal_WriteObjectToFile(PyObject *x, FILE *fp, int version)

{

WFILE wf;

wf.fp = fp;

wf.str = NULL;

wf.ptr = NULL;

wf.end = NULL;

wf.error = WFERR_OK;

wf.depth = 0;

wf.strings = (version > 0) ? PyDict_New() : NULL;

wf.version = version;

w_object(x, &wf);

Py_XDECREF(wf.strings);

}

typedef WFILE RFILE; /* Same struct with different invariants */

#define rs_byte(p) (((p)->ptr < (p)->end) ? (unsigned char)*(p)->ptr++ : EOF)

#define r_byte(p) ((p)->fp ? getc((p)->fp) : rs_byte(p))

static Py_ssize_t

r_string(char *s, Py_ssize_t n, RFILE *p)

{

if (p->fp != NULL)

/* The result fits into int because it must be <=n. */

return fread(s, 1, n, p->fp);

if (p->end - p->ptr < n)

n = p->end - p->ptr;

memcpy(s, p->ptr, n);

p->ptr += n;

return n;

}

static int

r_short(RFILE *p)

{

register short x;

x = r_byte(p);

x |= r_byte(p) << 8;

/* Sign-extension, in case short greater than 16 bits */

x |= -(x & 0x8000);

return x;

}

static long

r_long(RFILE *p)

{

register long x;

register FILE *fp = p->fp;

if (fp) {

x = getc(fp);

x |= (long)getc(fp) << 8;

x |= (long)getc(fp) << 16;

x |= (long)getc(fp) << 24;

}

else {

x = rs_byte(p);

x |= (long)rs_byte(p) << 8;

x |= (long)rs_byte(p) << 16;

x |= (long)rs_byte(p) << 24;

}

#if SIZEOF_LONG > 4

/* Sign extension for 64-bit machines */

x |= -(x & 0x80000000L);

return x;

}

static PyObject *

r_long64(RFILE *p)

{

long lo4 = r_long(p);

long hi4 = r_long(p);

#if SIZEOF_LONG > 4

long x = (hi4 << 32) | (lo4 & 0xFFFFFFFFL);

return PyInt_FromLong(x);

unsigned char buf[8];

int one = 1;

int is_little_endian = (int)*(char*)&one;

if (is_little_endian) {

memcpy(buf, &lo4, 4);

memcpy(buf+4, &hi4, 4);

}

else {

memcpy(buf, &hi4, 4);

memcpy(buf+4, &lo4, 4);

}

return _PyLong_FromByteArray(buf, 8, is_little_endian, 1);

}

static PyObject *

r_PyLong(RFILE *p)

{

PyLongObject *ob;

long n, size, i;

int j, md, shorts_in_top_digit;

digit d;

n = r_long(p);

if (n == 0)

return (PyObject *)_PyLong_New(0);

if (n < -SIZE32_MAX || n > SIZE32_MAX) {

PyErr_SetString(PyExc_ValueError,

"bad marshal data (long size out of range)");

return NULL;

}

size = 1 + (ABS(n) - 1) / PyLong_MARSHAL_RATIO;

shorts_in_top_digit = 1 + (ABS(n) - 1) % PyLong_MARSHAL_RATIO;

ob = _PyLong_New(size);

if (ob == NULL)

return NULL;

Py_SIZE(ob) = n > 0 ? size : -size;

for (i = 0; i < size-1; i++) {

d = 0;

for (j=0; j < PyLong_MARSHAL_RATIO; j++) {

md = r_short(p);

if (md < 0 || md > PyLong_MARSHAL_BASE)

goto bad_digit;

d += (digit)md << j*PyLong_MARSHAL_SHIFT;

}

ob->ob_digit[i] = d;

}

d = 0;

for (j=0; j < shorts_in_top_digit; j++) {

md = r_short(p);

if (md < 0 || md > PyLong_MARSHAL_BASE)

goto bad_digit;

/* topmost marshal digit should be nonzero */

if (md == 0 && j == shorts_in_top_digit - 1) {

Py_DECREF(ob);

PyErr_SetString(PyExc_ValueError,

"bad marshal data (unnormalized long data)");

return NULL;

}

d += (digit)md << j*PyLong_MARSHAL_SHIFT;

}

/* top digit should be nonzero, else the resulting PyLong won't be

ob->ob_digit[size-1] = d;

return (PyObject *)ob;

bad_digit:

Py_DECREF(ob);

PyErr_SetString(PyExc_ValueError,

"bad marshal data (digit out of range in long)");

return NULL;

}

static PyObject *

r_object(RFILE *p)

{

/* NULL is a valid return value, it does not necessarily means that

PyObject *v, *v2;

long i, n;

int type = r_byte(p);

PyObject *retval;

p->depth++;

if (p->depth > MAX_MARSHAL_STACK_DEPTH) {

p->depth--;

PyErr_SetString(PyExc_ValueError, "recursion limit exceeded");

return NULL;

}

switch (type) {

case EOF:

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

case TYPE_NULL:

retval = NULL;

break;

case TYPE_NONE:

Py_INCREF(Py_None);

retval = Py_None;

break;

case TYPE_STOPITER:

Py_INCREF(PyExc_StopIteration);

retval = PyExc_StopIteration;

break;

case TYPE_ELLIPSIS:

Py_INCREF(Py_Ellipsis);

retval = Py_Ellipsis;

break;

case TYPE_FALSE:

Py_INCREF(Py_False);

retval = Py_False;

break;

case TYPE_TRUE:

Py_INCREF(Py_True);

retval = Py_True;

break;

case TYPE_INT:

retval = PyInt_FromLong(r_long(p));

break;

case TYPE_INT64:

retval = r_long64(p);

break;

case TYPE_LONG:

retval = r_PyLong(p);

break;

case TYPE_FLOAT:

{

char buf[256];

double dx;

n = r_byte(p);

if (n == EOF || r_string(buf, n, p) != n) {

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

buf[n] = '\0';

dx = PyOS_string_to_double(buf, NULL, NULL);

if (dx == -1.0 && PyErr_Occurred()) {

retval = NULL;

break;

}

retval = PyFloat_FromDouble(dx);

break;

}

case TYPE_BINARY_FLOAT:

{

unsigned char buf[8];

double x;

if (r_string((char*)buf, 8, p) != 8) {

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

x = _PyFloat_Unpack8(buf, 1);

if (x == -1.0 && PyErr_Occurred()) {

retval = NULL;

break;

}

retval = PyFloat_FromDouble(x);

break;

}

#ifndef WITHOUT_COMPLEX

case TYPE_COMPLEX:

{

char buf[256];

Py_complex c;

n = r_byte(p);

if (n == EOF || r_string(buf, n, p) != n) {

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

buf[n] = '\0';

c.real = PyOS_string_to_double(buf, NULL, NULL);

if (c.real == -1.0 && PyErr_Occurred()) {

retval = NULL;

break;

}

n = r_byte(p);

if (n == EOF || r_string(buf, n, p) != n) {

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

buf[n] = '\0';

c.imag = PyOS_string_to_double(buf, NULL, NULL);

if (c.imag == -1.0 && PyErr_Occurred()) {

retval = NULL;

break;

}

retval = PyComplex_FromCComplex(c);

break;

}

case TYPE_BINARY_COMPLEX:

{

unsigned char buf[8];

Py_complex c;

if (r_string((char*)buf, 8, p) != 8) {

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

c.real = _PyFloat_Unpack8(buf, 1);

if (c.real == -1.0 && PyErr_Occurred()) {

retval = NULL;

break;

}

if (r_string((char*)buf, 8, p) != 8) {

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

c.imag = _PyFloat_Unpack8(buf, 1);

if (c.imag == -1.0 && PyErr_Occurred()) {

retval = NULL;

break;

}

retval = PyComplex_FromCComplex(c);

break;

}

case TYPE_INTERNED:

case TYPE_STRING:

n = r_long(p);

if (n < 0 || n > SIZE32_MAX) {

PyErr_SetString(PyExc_ValueError, "bad marshal data (string size out of range)");

retval = NULL;

break;

}

v = PyString_FromStringAndSize((char *)NULL, n);

if (v == NULL) {

retval = NULL;

break;

}

if (r_string(PyString_AS_STRING(v), n, p) != n) {

Py_DECREF(v);

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

if (type == TYPE_INTERNED) {

PyString_InternInPlace(&v);

if (PyList_Append(p->strings, v) < 0) {

retval = NULL;

break;

}

}

retval = v;

break;

case TYPE_STRINGREF:

n = r_long(p);

if (n < 0 || n >= PyList_GET_SIZE(p->strings)) {

PyErr_SetString(PyExc_ValueError, "bad marshal data (string ref out of range)");

retval = NULL;

break;

}

v = PyList_GET_ITEM(p->strings, n);

Py_INCREF(v);

retval = v;

break;

#ifdef Py_USING_UNICODE

case TYPE_UNICODE:

{

char *buffer;

n = r_long(p);

if (n < 0 || n > SIZE32_MAX) {

PyErr_SetString(PyExc_ValueError, "bad marshal data (unicode size out of range)");

retval = NULL;

break;

}

buffer = PyMem_NEW(char, n);

if (buffer == NULL) {

retval = PyErr_NoMemory();

break;

}

if (r_string(buffer, n, p) != n) {

PyMem_DEL(buffer);

PyErr_SetString(PyExc_EOFError,

"EOF read where object expected");

retval = NULL;

break;

}

v = PyUnicode_DecodeUTF8(buffer, n, NULL);

PyMem_DEL(buffer);

retval = v;

break;

}

case TYPE_TUPLE:

n = r_long(p);

if (n < 0 || n > SIZE32_MAX) {

PyErr_SetString(PyExc_ValueError, "bad marshal data (tuple size out of range)");

retval = NULL;

break;

}

v = PyTuple_New(n);

if (v == NULL) {

retval = NULL;

break;

}

for (i = 0; i < n; i++) {

v2 = r_object(p);

if ( v2 == NULL ) {

if (!PyErr_Occurred())

PyErr_SetString(PyExc_TypeError,

"NULL object in marshal data for tuple");

Py_DECREF(v);

v = NULL;

break;

}

PyTuple_SET_ITEM(v, i, v2);

}

retval = v;

break;

case TYPE_LIST:

n = r_long(p);

if (n < 0 || n > SIZE32_MAX) {

PyErr_SetString(PyExc_ValueError, "bad marshal data (list size out of range)");

retval = NULL;

break;

}

v = PyList_New(n);

if (v == NULL) {

retval = NULL;

break;

}

for (i = 0; i < n; i++) {

v2 = r_object(p);

if ( v2 == NULL ) {

if (!PyErr_Occurred())

PyErr_SetString(PyExc_TypeError,

"NULL object in marshal data for list");

Py_DECREF(v);

v = NULL;

break;

}

PyList_SET_ITEM(v, i, v2);

}

retval = v;

break;

case TYPE_DICT:

v = PyDict_New();

if (v == NULL) {

retval = NULL;

break;

}

for (;;) {

PyObject *key, *val;

key = r_object(p);

if (key == NULL)

break;

val = r_object(p);

if (val != NULL)

PyDict_SetItem(v, key, val);

Py_DECREF(key);

Py_XDECREF(val);

}

if (PyErr_Occurred()) {

Py_DECREF(v);

v = NULL;

}

retval = v;

break;

case TYPE_SET:

case TYPE_FROZENSET:

n = r_long(p);

if (n < 0 || n > SIZE32_MAX) {

PyErr_SetString(PyExc_ValueError, "bad marshal data (set size out of range)");

retval = NULL;

break;

}

v = (type == TYPE_SET) ? PySet_New(NULL) : PyFrozenSet_New(NULL);

if (v == NULL) {

retval = NULL;

break;

}

for (i = 0; i < n; i++) {

v2 = r_object(p);

if ( v2 == NULL ) {

if (!PyErr_Occurred())

PyErr_SetString(PyExc_TypeError,

"NULL object in marshal data for set");

Py_DECREF(v);

v = NULL;

break;

}

if (PySet_Add(v, v2) == -1) {

Py_DECREF(v);

Py_DECREF(v2);

v = NULL;

break;

}

Py_DECREF(v2);

}

retval = v;

break;

case TYPE_CODE:

if (PyEval_GetRestricted()) {

PyErr_SetString(PyExc_RuntimeError,

"cannot unmarshal code objects in "

"restricted execution mode");