/* $Id$ Copyright Copyright 1996 Digital Creations, L.C., 910 Princess Anne Street, Suite 300, Fredericksburg, Virginia 22401 U.S.A. All rights reserved. Copyright in this software is owned by DCLC, unless otherwise indicated. Permission to use, copy and distribute this software is hereby granted, provided that the above copyright notice appear in all copies and that both that copyright notice and this permission notice appear. Note that any product, process or technology described in this software may be the subject of other Intellectual Property rights reserved by Digital Creations, L.C. and are not licensed hereunder. Trademarks Digital Creations & DCLC, are trademarks of Digital Creations, L.C.. All other trademarks are owned by their respective companies. No Warranty The software is provided "as is" without warranty of any kind, either express or implied, including, but not limited to, the implied warranties of merchantability, fitness for a particular purpose, or non-infringement. This software could include technical inaccuracies or typographical errors. Changes are periodically made to the software; these changes will be incorporated in new editions of the software. DCLC may make improvements and/or changes in this software at any time without notice. Limitation Of Liability In no event will DCLC be liable for direct, indirect, special, incidental, economic, cover, or consequential damages arising out of the use of or inability to use this software even if advised of the possibility of such damages. Some states do not allow the exclusion or limitation of implied warranties or limitation of liability for incidental or consequential damages, so the above limitation or exclusion may not apply to you. If you have questions regarding this software, contact: Jim Fulton, jim@digicool.com Digital Creations L.C. (540) 371-6909 */ static char cPickle_module_documentation[] = "" ; #include "Python.h" #include "cStringIO.h" #include "graminit.h" #include "mymath.h" #include #define UNLESS(E) if (!(E)) #define DEL_LIST_SLICE(list, from, to) (PyList_SetSlice(list, from, to, NULL)) #define WRITE_BUF_SIZE 256 #define MARK '(' #define STOP '.' #define POP '0' #define POP_MARK '1' #define DUP '2' #define FLOAT 'F' #ifdef FORMAT_1_3 #define BINFLOAT 'G' #endif #define INT 'I' #define BININT 'J' #define BININT1 'K' #define LONG 'L' #define BININT2 'M' #define NONE 'N' #define PERSID 'P' #define BINPERSID 'Q' #define REDUCE 'R' #define STRING 'S' #define BINSTRING 'T' #define SHORT_BINSTRING 'U' #define APPEND 'a' #define BUILD 'b' #define GLOBAL 'c' #define DICT 'd' #define EMPTY_DICT '}' #define APPENDS 'e' #define GET 'g' #define BINGET 'h' #define INST 'i' #define LONG_BINGET 'j' #define LIST 'l' #define EMPTY_LIST ']' #define OBJ 'o' #define PUT 'p' #define BINPUT 'q' #define LONG_BINPUT 'r' #define SETITEM 's' #define TUPLE 't' #define EMPTY_TUPLE ')' #define SETITEMS 'u' static char MARKv = MARK; /* atol function from string module */ static PyObject *atol_func; static PyObject *PicklingError; static PyObject *UnpicklingError; static PyObject *dispatch_table; static PyObject *safe_constructors; static PyObject *class_map; static PyObject *empty_tuple; static PyObject *__class___str, *__getinitargs___str, *__dict___str, *__getstate___str, *__setstate___str, *__name___str, *__reduce___str, *write_str, *__safe_for_unpickling___str, *append_str, *read_str, *readline_str; /* __builtins__ module */ static PyObject *builtins; static int save(); static int put2(); typedef struct { PyObject_HEAD FILE *fp; PyObject *write; PyObject *file; PyObject *memo; PyObject *arg; PyObject *pers_func; PyObject *inst_pers_func; int bin; int (*write_func)(); char *write_buf; int buf_size; } Picklerobject; static PyTypeObject Picklertype; typedef struct { PyObject_HEAD FILE *fp; PyObject *file; PyObject *readline; PyObject *read; PyObject *memo; PyObject *arg; PyObject *stack; PyObject *mark; PyObject *pers_func; PyObject *last_string; int *marks; int num_marks; int marks_size; int (*read_func)(); int (*readline_func)(); int buf_size; char *buf; } Unpicklerobject; static PyTypeObject Unpicklertype; int cPickle_PyMapping_HasKey(o, key) PyObject *o; PyObject *key; { PyObject *v; if (v = PyObject_GetItem(o,key)) { Py_DECREF(v); return 1; } PyErr_Clear(); return 0; } #define PyErr_Format PyErr_JFFormat static PyObject * #ifdef HAVE_STDARG_PROTOTYPES /* VARARGS 2 */ PyErr_Format(PyObject *ErrType, char *stringformat, char *format, ...) #else /* VARARGS */ PyErr_Format(va_alist) va_dcl #endif { va_list va; PyObject *args=0, *retval=0; #ifdef HAVE_STDARG_PROTOTYPES va_start(va, format); #else PyObject *ErrType; char *stringformat, *format; va_start(va); ErrType = va_arg(va, PyObject *); stringformat = va_arg(va, char *); format = va_arg(va, char *); #endif if(format) args = Py_VaBuildValue(format, va); va_end(va); if(format && ! args) return NULL; if(stringformat && !(retval=PyString_FromString(stringformat))) return NULL; if(retval) { if(args) { PyObject *v; v=PyString_Format(retval, args); Py_DECREF(retval); Py_DECREF(args); if(! v) return NULL; retval=v; } } else if(args) retval=args; else { PyErr_SetObject(ErrType,Py_None); return NULL; } PyErr_SetObject(ErrType,retval); Py_DECREF(retval); return NULL; } static int write_file(Picklerobject *self, char *s, int n) { if (s == NULL) { return 0; } if ((int)fwrite(s, sizeof(char), n, self->fp) != n) { PyErr_SetFromErrno(PyExc_IOError); return -1; } return n; } static int write_cStringIO(Picklerobject *self, char *s, int n) { if (s == NULL) { return 0; } if (PycStringIO->cwrite((PyObject *)self->file, s, n) != n) { return -1; } return n; } static int write_other(Picklerobject *self, char *s, int n) { PyObject *py_str = 0, *junk = 0; int res = -1; if (s == NULL) { UNLESS(self->buf_size) return 0; UNLESS(py_str = PyString_FromStringAndSize(self->write_buf, self->buf_size)) goto finally; } else { if (self->buf_size && (n + self->buf_size) > WRITE_BUF_SIZE) { if (write_other(self, NULL, 0) < 0) goto finally; } if (n > WRITE_BUF_SIZE) { UNLESS(py_str = PyString_FromStringAndSize(s, n)) goto finally; } else { memcpy(self->write_buf + self->buf_size, s, n); self->buf_size += n; res = n; goto finally; } } UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(py_str); if (PyTuple_SetItem(self->arg, 0, py_str) < 0) goto finally; UNLESS(junk = PyObject_CallObject(self->write, self->arg)) goto finally; Py_DECREF(junk); self->buf_size = 0; res = n; finally: Py_XDECREF(py_str); return res; } static int read_file(Unpicklerobject *self, char **s, int n) { if (self->buf_size == 0) { int size; size = ((n < 32) ? 32 : n); UNLESS(self->buf = (char *)malloc(size * sizeof(char))) { PyErr_NoMemory(); return -1; } self->buf_size = size; } else if (n > self->buf_size) { UNLESS(self->buf = (char *)realloc(self->buf, n * sizeof(char))) { PyErr_NoMemory(); return -1; } self->buf_size = n; } if ((int)fread(self->buf, sizeof(char), n, self->fp) != n) { if (feof(self->fp)) { PyErr_SetNone(PyExc_EOFError); return -1; } PyErr_SetFromErrno(PyExc_IOError); return -1; } *s = self->buf; return n; } static int readline_file(Unpicklerobject *self, char **s) { int i; if (self->buf_size == 0) { UNLESS(self->buf = (char *)malloc(40 * sizeof(char))) { PyErr_NoMemory(); return -1; } self->buf_size = 40; } i = 0; while (1) { for (; i < (self->buf_size - 1); i++) { if (feof(self->fp) || (self->buf[i] = getc(self->fp)) == '\n') { self->buf[i + 1] = '\0'; *s = self->buf; return i + 1; } } UNLESS(self->buf = (char *)realloc(self->buf, (self->buf_size * 2) * sizeof(char))) { PyErr_NoMemory(); return -1; } self->buf_size *= 2; } } static int read_cStringIO(Unpicklerobject *self, char **s, int n) { char *ptr; if (PycStringIO->cread((PyObject *)self->file, &ptr, n) != n) { PyErr_SetNone(PyExc_EOFError); return -1; } *s = ptr; return n; } static int readline_cStringIO(Unpicklerobject *self, char **s) { int n; char *ptr; if ((n = PycStringIO->creadline((PyObject *)self->file, &ptr)) < 0) { return -1; } *s = ptr; return n; } static int read_other(Unpicklerobject *self, char **s, int n) { PyObject *bytes, *str; int res = -1; UNLESS(bytes = PyInt_FromLong(n)) { if (!PyErr_Occurred()) PyErr_SetNone(PyExc_EOFError); goto finally; } UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(bytes); if (PyTuple_SetItem(self->arg, 0, bytes) < 0) goto finally; UNLESS(str = PyObject_CallObject(self->read, self->arg)) goto finally; Py_XDECREF(self->last_string); self->last_string = str; *s = PyString_AsString(str); res = n; finally: Py_XDECREF(bytes); return res; } static int readline_other(Unpicklerobject *self, char **s) { PyObject *str; int str_size; UNLESS(str = PyObject_CallObject(self->readline, empty_tuple)) { return -1; } str_size = PyString_Size(str); Py_XDECREF(self->last_string); self->last_string = str; *s = PyString_AsString(str); return str_size; } static char * strndup(char *s, int l) { char *r; UNLESS(r=malloc((l+1)*sizeof(char))) return (char*)PyErr_NoMemory(); memcpy(r,s,l); r[l]=0; return r; } static int get(Picklerobject *self, PyObject *id) { PyObject *value = 0; long c_value; char s[30]; int len; UNLESS(value = PyDict_GetItem(self->memo, id)) return -1; UNLESS(value = PyTuple_GetItem(value, 0)) return -1; c_value = PyInt_AsLong(value); if (!self->bin) { s[0] = GET; sprintf(s + 1, "%ld\n", c_value); len = strlen(s); } else { if (c_value < 256) { s[0] = BINGET; s[1] = (int)(c_value & 0xff); len = 2; } else { s[0] = LONG_BINGET; s[1] = (int)(c_value & 0xff); s[2] = (int)((c_value >> 8) & 0xff); s[3] = (int)((c_value >> 16) & 0xff); s[4] = (int)((c_value >> 24) & 0xff); len = 5; } } if ((*self->write_func)(self, s, len) < 0) return -1; return 0; } static int put(Picklerobject *self, PyObject *ob) { if (ob->ob_refcnt < 2) return 0; return put2(self, ob); } static int put2(Picklerobject *self, PyObject *ob) { char c_str[30]; int p, len, res = -1; PyObject *py_ob_id = 0, *memo_len = 0, *t = 0; if ((p = PyDict_Size(self->memo)) < 0) goto finally; if (!self->bin) { c_str[0] = PUT; sprintf(c_str + 1, "%d\n", p); len = strlen(c_str); } else { if (p >= 256) { c_str[0] = LONG_BINPUT; c_str[1] = (int)(p & 0xff); c_str[2] = (int)((p >> 8) & 0xff); c_str[3] = (int)((p >> 16) & 0xff); c_str[4] = (int)((p >> 24) & 0xff); len = 5; } else { c_str[0] = BINPUT; c_str[1] = p; len = 2; } } if ((*self->write_func)(self, c_str, len) < 0) goto finally; UNLESS(py_ob_id = PyInt_FromLong((long)ob)) goto finally; UNLESS(memo_len = PyInt_FromLong(p)) goto finally; UNLESS(t = PyTuple_New(2)) goto finally; PyTuple_SET_ITEM(t, 0, memo_len); Py_INCREF(memo_len); PyTuple_SET_ITEM(t, 1, ob); Py_INCREF(ob); if (PyDict_SetItem(self->memo, py_ob_id, t) < 0) goto finally; res = 0; finally: Py_XDECREF(py_ob_id); Py_XDECREF(memo_len); Py_XDECREF(t); return res; } static PyObject * whichmodule(PyObject *global, PyObject *global_name) { int i, j; PyObject *module = 0, *modules_dict = 0, *global_name_attr = 0, *name = 0; if (module = PyDict_GetItem(class_map, global)) { Py_INCREF(module); return module; } else { PyErr_Clear(); } UNLESS(modules_dict = PySys_GetObject("modules")) return NULL; i = 0; while (j = PyDict_Next(modules_dict, &i, &name, &module)) { UNLESS(global_name_attr = PyObject_GetAttr(module, global_name)) { PyErr_Clear(); continue; } if (global_name_attr != global) { Py_DECREF(global_name_attr); continue; } Py_DECREF(global_name_attr); break; } if (!j) { PyErr_Format(PicklingError, "Could not find module for %s.", "O", global_name); return NULL; } PyDict_SetItem(class_map, global, name); Py_INCREF(name); return name; } static int save_none(Picklerobject *self, PyObject *args) { static char none = NONE; if ((*self->write_func)(self, &none, 1) < 0) return -1; return 0; } static int save_int(Picklerobject *self, PyObject *args) { char c_str[32]; long l = PyInt_AS_LONG((PyIntObject *)args); int len = 0; if (!self->bin || sizeof(long) == 8 && (l >> 32)) { /* Save extra-long ints in non-binary mode, so that we can use python long parsing code to restore, if necessary. */ c_str[0] = INT; sprintf(c_str + 1, "%ld\n", l); if ((*self->write_func)(self, c_str, strlen(c_str)) < 0) return -1; } else { c_str[1] = (int)( l & 0xff); c_str[2] = (int)((l >> 8) & 0xff); c_str[3] = (int)((l >> 16) & 0xff); c_str[4] = (int)((l >> 24) & 0xff); if ((c_str[4] == 0) && (c_str[3] == 0)) { if (c_str[2] == 0) { c_str[0] = BININT1; len = 2; } else { c_str[0] = BININT2; len = 3; } } else { c_str[0] = BININT; len = 5; } if ((*self->write_func)(self, c_str, len) < 0) return -1; } return 0; } static int save_long(Picklerobject *self, PyObject *args) { int size, res = -1; PyObject *repr = 0; static char l = LONG; UNLESS(repr = PyObject_Repr(args)) goto finally; if ((size = PyString_Size(repr)) < 0) goto finally; if ((*self->write_func)(self, &l, 1) < 0) goto finally; if ((*self->write_func)(self, PyString_AS_STRING((PyStringObject *)repr), size) < 0) goto finally; if ((*self->write_func)(self, "\n", 1) < 0) goto finally; res = 0; finally: Py_XDECREF(repr); return res; } static int save_float(Picklerobject *self, PyObject *args) { double x = PyFloat_AS_DOUBLE((PyFloatObject *)args); #ifdef FORMAT_1_3 if (self->bin) { int s, e, i = -1; double f; long fhi, flo; char str[9], *p = str; *p = BINFLOAT; p++; if (x < 0) { s = 1; x = -x; } else s = 0; f = frexp(x, &e); /* Normalize f to be in the range [1.0, 2.0) */ if (0.5 <= f && f < 1.0) { f *= 2.0; e--; } else if (f == 0.0) { e = 0; } else { PyErr_SetString(PyExc_SystemError, "frexp() result out of range"); return -1; } if (e >= 1024) { /* XXX 1024 itself is reserved for Inf/NaN */ PyErr_SetString(PyExc_OverflowError, "float too large to pack with d format"); return -1; } else if (e < -1022) { /* Gradual underflow */ f = ldexp(f, 1022 + e); e = 0; } else { e += 1023; f -= 1.0; /* Get rid of leading 1 */ } /* fhi receives the high 28 bits; flo the low 24 bits (== 52 bits) */ f *= 268435456.0; /* 2**28 */ fhi = (long) floor(f); /* Truncate */ f -= (double)fhi; f *= 16777216.0; /* 2**24 */ flo = (long) floor(f + 0.5); /* Round */ /* First byte */ *p = (s<<7) | (e>>4); p++; /* Second byte */ *p = ((e&0xF)<<4) | (fhi>>24); p++; /* Third byte */ *p = (fhi>>16) & 0xFF; p++; /* Fourth byte */ *p = (fhi>>8) & 0xFF; p++; /* Fifth byte */ *p = fhi & 0xFF; p++; /* Sixth byte */ *p = (flo>>16) & 0xFF; p++; /* Seventh byte */ *p = (flo>>8) & 0xFF; p++; /* Eighth byte */ *p = flo & 0xFF; if ((*self->write_func)(self, str, 9) < 0) return -1; } else #endif { char c_str[250]; c_str[0] = FLOAT; sprintf(c_str + 1, "%f\n", x); if ((*self->write_func)(self, c_str, strlen(c_str)) < 0) return -1; } return 0; } static int save_string(Picklerobject *self, PyObject *args) { int size, len; size = PyString_Size(args); if (!self->bin) { PyObject *repr; char *repr_str; static char string = STRING; UNLESS(repr = PyObject_Repr(args)) return -1; repr_str = PyString_AS_STRING((PyStringObject *)repr); len = PyString_Size(repr); if ((*self->write_func)(self, &string, 1) < 0) return -1; if ((*self->write_func)(self, repr_str, len) < 0) return -1; if ((*self->write_func)(self, "\n", 1) < 0) return -1; Py_XDECREF(repr); } else { int i; char c_str[5]; size = PyString_Size(args); if (size < 256) { c_str[0] = SHORT_BINSTRING; c_str[1] = size; len = 2; } else { c_str[0] = BINSTRING; for (i = 1; i < 5; i++) c_str[i] = (int)(size >> ((i - 1) * 8)); len = 5; } if ((*self->write_func)(self, c_str, len) < 0) return -1; if ((*self->write_func)(self, PyString_AS_STRING((PyStringObject *)args), size) < 0) return -1; } if (size > 1) if (put(self, args) < 0) return -1; return 0; } static int save_tuple(Picklerobject *self, PyObject *args) { PyObject *element = 0, *py_tuple_id = 0; int len, i, has_key, res = -1; static char tuple = TUPLE; if ((*self->write_func)(self, &MARKv, 1) < 0) goto finally; if ((len = PyTuple_Size(args)) < 0) goto finally; for (i = 0; i < len; i++) { UNLESS(element = PyTuple_GET_ITEM((PyTupleObject *)args, i)) goto finally; if (save(self, element, 0) < 0) goto finally; } UNLESS(py_tuple_id = PyInt_FromLong((long)args)) goto finally; if (len) { if (has_key = cPickle_PyMapping_HasKey(self->memo, py_tuple_id) < 0) goto finally; if (has_key) { if (self->bin) { static char pop_mark = POP_MARK; if ((*self->write_func)(self, &pop_mark, 1) < 0) goto finally; } else { static char pop = POP; for (i = 0; i <= len; i++) { if ((*self->write_func)(self, &pop, 1) < 0) goto finally; } } if (get(self, py_tuple_id) < 0) goto finally; res = 0; goto finally; } } if ((*self->write_func)(self, &tuple, 1) < 0) { goto finally; } if (put(self, args) < 0) goto finally; res = 0; finally: Py_XDECREF(py_tuple_id); return res; } static int save_empty_tuple(Picklerobject *self, PyObject *args) { static char tuple = EMPTY_TUPLE; return (*self->write_func)(self, &tuple, 1); } static int save_list(Picklerobject *self, PyObject *args) { PyObject *element = 0; int s_len, len, i, using_appends, res = -1; char s[3]; static char append = APPEND, appends = APPENDS; if(self->bin) { s[0] = EMPTY_LIST; s_len = 1; } else { s[0] = MARK; s[1] = LIST; s_len = 2; } if ((len = PyList_Size(args)) < 0) goto finally; if ((*self->write_func)(self, s, s_len) < 0) goto finally; if (len == 0) { if (put(self, args) < 0) goto finally; } else { if (put2(self, args) < 0) goto finally; } if (using_appends = (self->bin && (len > 1))) if ((*self->write_func)(self, &MARKv, 1) < 0) goto finally; for (i = 0; i < len; i++) { UNLESS(element = PyList_GET_ITEM((PyListObject *)args, i)) goto finally; if (save(self, element, 0) < 0) goto finally; if (!using_appends) { if ((*self->write_func)(self, &append, 1) < 0) goto finally; } } if (using_appends) { if ((*self->write_func)(self, &appends, 1) < 0) goto finally; } res = 0; finally: return res; } static int save_dict(Picklerobject *self, PyObject *args) { PyObject *key = 0, *value = 0; int i, len, res = -1, using_setitems; char s[3]; static char setitem = SETITEM, setitems = SETITEMS; if (self->bin) { s[0] = EMPTY_DICT; len = 1; } else { s[0] = MARK; s[1] = DICT; len = 2; } if ((*self->write_func)(self, s, len) < 0) goto finally; if ((len = PyDict_Size(args)) < 0) goto finally; if (len == 0) { if (put(self, args) < 0) goto finally; } else { if (put2(self, args) < 0) goto finally; } if (using_setitems = (self->bin && (PyDict_Size(args) > 1))) if ((*self->write_func)(self, &MARKv, 1) < 0) goto finally; i = 0; while (PyDict_Next(args, &i, &key, &value)) { if (save(self, key, 0) < 0) goto finally; if (save(self, value, 0) < 0) goto finally; if (!using_setitems) { if ((*self->write_func)(self, &setitem, 1) < 0) goto finally; } } if (using_setitems) { if ((*self->write_func)(self, &setitems, 1) < 0) goto finally; } res = 0; finally: return res; } static int save_inst(Picklerobject *self, PyObject *args) { PyObject *class = 0, *module = 0, *name = 0, *state = 0, *getinitargs_func = 0, *getstate_func = 0, *class_args = 0; char *module_str, *name_str; int module_size, name_size, res = -1; static char inst = INST, obj = OBJ, build = BUILD; if ((*self->write_func)(self, &MARKv, 1) < 0) goto finally; UNLESS(class = PyObject_GetAttr(args, __class___str)) goto finally; if (self->bin) { if (save(self, class, 0) < 0) goto finally; } if (getinitargs_func = PyObject_GetAttr(args, __getinitargs___str)) { PyObject *element = 0; int i, len; UNLESS(class_args = PyObject_CallObject(getinitargs_func, empty_tuple)) goto finally; if ((len = PyObject_Length(class_args)) < 0) goto finally; for (i = 0; i < len; i++) { UNLESS(element = PySequence_GetItem(class_args, i)) goto finally; if (save(self, element, 0) < 0) { Py_DECREF(element); goto finally; } Py_DECREF(element); } } else { PyErr_Clear(); } if (!self->bin) { UNLESS(name = ((PyClassObject *)class)->cl_name) { PyErr_SetString(PicklingError, "class has no name"); goto finally; } UNLESS(module = whichmodule(class, name)) goto finally; module_str = PyString_AS_STRING((PyStringObject *)module); module_size = PyString_Size(module); name_str = PyString_AS_STRING((PyStringObject *)name); name_size = PyString_Size(name); if ((*self->write_func)(self, &inst, 1) < 0) goto finally; if ((*self->write_func)(self, module_str, module_size) < 0) goto finally; if ((*self->write_func)(self, "\n", 1) < 0) goto finally; if ((*self->write_func)(self, name_str, name_size) < 0) goto finally; if ((*self->write_func)(self, "\n", 1) < 0) goto finally; } else if ((*self->write_func)(self, &obj, 1) < 0) { goto finally; } if (getstate_func = PyObject_GetAttr(args, __getstate___str)) { UNLESS(state = PyObject_CallObject(getstate_func, empty_tuple)) goto finally; } else { PyErr_Clear(); UNLESS(state = PyObject_GetAttr(args, __dict___str)) { PyErr_Clear(); res = 0; goto finally; } } if (!PyDict_Check(state)) { if (put2(self, args) < 0) goto finally; } else { if (put(self, args) < 0) goto finally; } if (save(self, state, 0) < 0) goto finally; if ((*self->write_func)(self, &build, 1) < 0) goto finally; res = 0; finally: Py_XDECREF(module); Py_XDECREF(class); Py_XDECREF(state); Py_XDECREF(getinitargs_func); Py_XDECREF(getstate_func); Py_XDECREF(class_args); return res; } static int save_global(Picklerobject *self, PyObject *args, PyObject *name) { PyObject *global_name = 0, *module = 0; char *name_str, *module_str; int module_size, name_size, res = -1; static char global = GLOBAL; if (name) { global_name = name; Py_INCREF(global_name); } else { UNLESS(global_name = PyObject_GetAttr(args, __name___str)) goto finally; } UNLESS(module = whichmodule(args, global_name)) goto finally; module_str = PyString_AS_STRING((PyStringObject *)module); module_size = PyString_Size(module); name_str = PyString_AS_STRING((PyStringObject *)global_name); name_size = PyString_Size(global_name); if ((*self->write_func)(self, &global, 1) < 0) goto finally; if ((*self->write_func)(self, module_str, module_size) < 0) goto finally; if ((*self->write_func)(self, "\n", 1) < 0) goto finally; if ((*self->write_func)(self, name_str, name_size) < 0) goto finally; if ((*self->write_func)(self, "\n", 1) < 0) goto finally; if (put(self, args) < 0) goto finally; res = 0; finally: Py_XDECREF(module); Py_XDECREF(global_name); return res; } static int save_pers(Picklerobject *self, PyObject *args, PyObject *f) { PyObject *pid = 0; int size, res = -1; static char persid = PERSID, binpersid = BINPERSID; UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(args); if (PyTuple_SetItem(self->arg, 0, args) < 0) goto finally; UNLESS(pid = PyObject_CallObject(f, self->arg)) goto finally; if (pid != Py_None) { if (!self->bin) { if (!PyString_Check(pid)) { PyErr_SetString(PicklingError, "persistent id must be string"); goto finally; } if ((*self->write_func)(self, &persid, 1) < 0) goto finally; if ((size = PyString_Size(pid)) < 0) goto finally; if ((*self->write_func)(self, PyString_AS_STRING((PyStringObject *)pid), size) < 0) goto finally; if ((*self->write_func)(self, "\n", 1) < 0) goto finally; res = 1; goto finally; } else if (save(self, pid, 1) >= 0) { if ((*self->write_func)(self, &binpersid, 1) < 0) res = -1; else res = 1; } goto finally; } res = 0; finally: Py_XDECREF(pid); return res; } static int save_reduce(Picklerobject *self, PyObject *callable, PyObject *tup, PyObject *state, PyObject *ob) { static char reduce = REDUCE, build = BUILD; if (save(self, callable, 0) < 0) return -1; if (save(self, tup, 0) < 0) return -1; if ((*self->write_func)(self, &reduce, 1) < 0) return -1; if (ob != NULL) { if (state && !PyDict_Check(state)) { if (put2(self, ob) < 0) return -1; } else { if (put(self, ob) < 0) return -1; } } if (state) { if (save(self, state, 0) < 0) return -1; if ((*self->write_func)(self, &build, 1) < 0) return -1; } return 0; } static int save(Picklerobject *self, PyObject *args, int pers_save) { PyTypeObject *type; PyObject *py_ob_id = 0, *__reduce__ = 0, *t = 0, *arg_tup = 0, *callable = 0, *state = 0, *junk = 0; int res = -1, tmp, size; if (!pers_save && self->pers_func) { if ((tmp = save_pers(self, args, self->pers_func)) != 0) { res = tmp; goto finally; } } if (args == Py_None) { res = save_none(self, args); goto finally; } type = args->ob_type; switch (type->tp_name[0]) { case 'i': if (type == &PyInt_Type) { res = save_int(self, args); goto finally; } break; case 'l': if (type == &PyLong_Type) { res = save_long(self, args); goto finally; } break; case 'f': if (type == &PyFloat_Type) { res = save_float(self, args); goto finally; } break; case 't': if (type == &PyTuple_Type && PyTuple_Size(args)==0) { if(self->bin) res = save_empty_tuple(self, args); else res = save_tuple(self, args); goto finally; } case 's': if ((type == &PyString_Type) && (PyString_Size(args) < 2)) { res = save_string(self, args); goto finally; } } if (args->ob_refcnt > 1) { long ob_id; int has_key; ob_id = (long)args; UNLESS(py_ob_id = PyInt_FromLong(ob_id)) goto finally; if ((has_key = cPickle_PyMapping_HasKey(self->memo, py_ob_id)) < 0) goto finally; if (has_key) { if (get(self, py_ob_id) < 0) goto finally; res = 0; goto finally; } } switch (type->tp_name[0]) { case 's': if (type == &PyString_Type) { res = save_string(self, args); goto finally; } case 't': if (type == &PyTuple_Type) { res = save_tuple(self, args); goto finally; } case 'l': if (type == &PyList_Type) { res = save_list(self, args); goto finally; } case 'd': if (type == &PyDict_Type) { res = save_dict(self, args); goto finally; } case 'i': if (type == &PyInstance_Type) { res = save_inst(self, args); goto finally; } case 'c': if (type == &PyClass_Type) { res = save_global(self, args, NULL); goto finally; } case 'f': if (type == &PyFunction_Type) { res = save_global(self, args, NULL); goto finally; } case 'b': if (type == &PyCFunction_Type) { res = save_global(self, args, NULL); goto finally; } } if (!pers_save && self->inst_pers_func) { if ((tmp = save_pers(self, args, self->inst_pers_func)) != 0) { res = tmp; goto finally; } } if (__reduce__ = PyDict_GetItem(dispatch_table, (PyObject *)type)) { Py_INCREF(__reduce__); UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(args); if (PyTuple_SetItem(self->arg, 0, args) < 0) goto finally; UNLESS(t = PyObject_CallObject(__reduce__, self->arg)) goto finally; } else { PyErr_Clear(); if (__reduce__ = PyObject_GetAttr(args, __reduce___str)) { UNLESS(t = PyObject_CallObject(__reduce__, empty_tuple)) goto finally; } else { PyErr_Clear(); } } if (t) { if (PyString_Check(t)) { res = save_global(self, args, t); goto finally; } if (!PyTuple_Check(t)) { PyErr_Format(PicklingError, "Value returned by %s must " "be a tuple", "O", __reduce__); goto finally; } size = PyTuple_Size(t); if ((size != 3) && (size != 2)) { PyErr_Format(PicklingError, "tuple returned by %s must " "contain only two or three elements", "O", __reduce__); goto finally; } callable = PyTuple_GET_ITEM(t, 0); arg_tup = PyTuple_GET_ITEM(t, 1); if (size > 2) { state = PyTuple_GET_ITEM(t, 2); } UNLESS(PyTuple_Check(arg_tup)) { PyErr_Format(PicklingError, "Second element of tuple " "returned by %s must be a tuple", "O", __reduce__); goto finally; } res = save_reduce(self, callable, arg_tup, state, args); goto finally; } /* if (PyObject_HasAttrString(args, "__class__")) { res = save_inst(self, args); goto finally; } */ PyErr_Format(PicklingError, "Cannot pickle %s objects.", "O", (PyObject *)type); finally: Py_XDECREF(py_ob_id); Py_XDECREF(__reduce__); Py_XDECREF(t); return res; } static int dump(Picklerobject *self, PyObject *args) { static char stop = STOP; if (save(self, args, 0) < 0) return -1; if ((*self->write_func)(self, &stop, 1) < 0) return -1; if ((*self->write_func)(self, NULL, 0) < 0) return -1; return 0; } static PyObject * Pickler_dump(Picklerobject *self, PyObject *args) { PyObject *ob; UNLESS(PyArg_ParseTuple(args, "O", &ob)) return NULL; if (dump(self, ob) < 0) return NULL; Py_INCREF(Py_None); return Py_None; } static PyObject * dump_special(Picklerobject *self, PyObject *args) { static char stop = STOP; PyObject *callable, *arg_tup, *state = NULL; UNLESS(PyArg_ParseTuple(args, "OO|O", &callable, &arg_tup, &state)) return NULL; UNLESS(PyTuple_Check(arg_tup)) { PyErr_SetString(PicklingError, "Second arg to dump_special must " "be tuple"); return NULL; } if (save_reduce(self, callable, arg_tup, state, NULL) < 0) return NULL; if ((*self->write_func)(self, &stop, 1) < 0) return NULL; if ((*self->write_func)(self, NULL, 0) < 0) return NULL; Py_INCREF(Py_None); return Py_None; } static struct PyMethodDef Pickler_methods[] = { {"dump", (PyCFunction)Pickler_dump, 1, ""}, {"dump_special", (PyCFunction)dump_special, 1, ""}, {NULL, NULL} /* sentinel */ }; static Picklerobject * newPicklerobject(PyObject *file, int bin) { Picklerobject *self; UNLESS(self = PyObject_NEW(Picklerobject, &Picklertype)) return NULL; self->fp = NULL; self->write = NULL; self->memo = NULL; self->arg = NULL; self->pers_func = NULL; self->inst_pers_func = NULL; self->write_buf = NULL; self->bin = bin; self->buf_size = 0; Py_INCREF(file); self->file = file; UNLESS(self->memo = PyDict_New()) { Py_XDECREF((PyObject *)self); return NULL; } if (PyFile_Check(file)) { self->fp = PyFile_AsFile(file); self->write_func = write_file; } else if (PycStringIO_OutputCheck(file)) { self->write_func = write_cStringIO; } else { self->write_func = write_other; UNLESS(self->write = PyObject_GetAttr(file, write_str)) { PyErr_Clear(); PyErr_SetString(PyExc_TypeError, "argument must have 'write' " "attribute"); Py_XDECREF((PyObject *)self); return NULL; } UNLESS(self->write_buf = (char *)malloc(WRITE_BUF_SIZE * sizeof(char))) { PyErr_NoMemory(); Py_XDECREF((PyObject *)self); return NULL; } } return self; } static PyObject * get_Pickler(PyObject *self, PyObject *args) { PyObject *file; int bin = 0; UNLESS(PyArg_ParseTuple(args, "O|i", &file, &bin)) return NULL; return (PyObject *)newPicklerobject(file, bin); } static void Pickler_dealloc(Picklerobject *self) { Py_XDECREF(self->write); Py_XDECREF(self->memo); Py_XDECREF(self->arg); Py_XDECREF(self->file); Py_XDECREF(self->pers_func); Py_XDECREF(self->inst_pers_func); if (self->write_buf) { free(self->write_buf); } PyMem_DEL(self); } static PyObject * Pickler_getattr(Picklerobject *self, char *name) { if (strcmp(name, "persistent_id") == 0) { if (!self->pers_func) { PyErr_SetString(PyExc_AttributeError, name); return NULL; } Py_INCREF(self->pers_func); return self->pers_func; } if (strcmp(name, "memo") == 0) { if (!self->memo) { PyErr_SetString(PyExc_AttributeError, name); return NULL; } Py_INCREF(self->memo); return self->memo; } if (strcmp(name, "PicklingError") == 0) { Py_INCREF(PicklingError); return PicklingError; } return Py_FindMethod(Pickler_methods, (PyObject *)self, name); } int Pickler_setattr(Picklerobject *self, char *name, PyObject *value) { if (strcmp(name, "persistent_id") == 0) { Py_XDECREF(self->pers_func); self->pers_func = value; Py_INCREF(value); return 0; } if (strcmp(name, "inst_persistent_id") == 0) { Py_XDECREF(self->inst_pers_func); self->inst_pers_func = value; Py_INCREF(value); return 0; } PyErr_SetString(PyExc_AttributeError, name); return -1; } static char Picklertype__doc__[] = ""; static PyTypeObject Picklertype_value() { PyTypeObject Picklertype = { PyObject_HEAD_INIT(&PyType_Type) 0, /*ob_size*/ "Pickler", /*tp_name*/ sizeof(Picklerobject), /*tp_basicsize*/ 0, /*tp_itemsize*/ /* methods */ (destructor)Pickler_dealloc, /*tp_dealloc*/ (printfunc)0, /*tp_print*/ (getattrfunc)Pickler_getattr, /*tp_getattr*/ (setattrfunc)Pickler_setattr, /*tp_setattr*/ (cmpfunc)0, /*tp_compare*/ (reprfunc)0, /*tp_repr*/ 0, /*tp_as_number*/ 0, /*tp_as_sequence*/ 0, /*tp_as_mapping*/ (hashfunc)0, /*tp_hash*/ (ternaryfunc)0, /*tp_call*/ (reprfunc)0, /*tp_str*/ /* Space for future expansion */ 0L,0L,0L,0L, Picklertype__doc__ /* Documentation string */ }; return Picklertype; } PyObject * PyImport_ImportModuleNi(char *module_name) { char *import_str; int size, i; PyObject *import; static PyObject *eval_dict = 0; size = strlen(module_name); for (i = 0; i < size; i++) { if (((module_name[i] < 'A') || (module_name[i] > 'z')) && (module_name[i] != '_')) { PyErr_SetString(PyExc_ImportError, "module name contains " "invalid characters."); return NULL; } } UNLESS(import_str = (char *)malloc((strlen(module_name) + 15) * sizeof(char))) { PyErr_NoMemory(); return NULL; } sprintf(import_str, "__import__('%s')", module_name); UNLESS(eval_dict) UNLESS(eval_dict = Py_BuildValue("{sO}", "__builtins__", builtins)) return NULL; if (!(import = PyRun_String(import_str, eval_input, eval_dict, eval_dict))) { free(import_str); return NULL; } free(import_str); return import; } static PyObject * find_class(PyObject *py_module_name, PyObject *py_class_name) { PyObject *import = 0, *class = 0, *t = 0; char *module_name, *class_name; PyObject *res = NULL; static PyObject *eval_dict = 0; module_name = PyString_AS_STRING((PyStringObject *)py_module_name); class_name = PyString_AS_STRING((PyStringObject *)py_class_name); UNLESS(t = PyTuple_New(2)) goto finally; PyTuple_SET_ITEM((PyTupleObject *)t, 0, py_module_name); Py_INCREF(py_module_name); PyTuple_SET_ITEM((PyTupleObject *)t, 1, py_class_name); Py_INCREF(py_class_name); if (class = PyDict_GetItem(class_map, t)) { res = class; Py_INCREF(class); goto finally; } PyErr_Clear(); if (!(import = PyImport_ImportModuleNi(module_name)) || !(class = PyObject_GetAttr(import, py_class_name))) { PyErr_Format(PyExc_SystemError, "Failed to import global %s " "from module %s", "ss", class_name, module_name); goto finally; } if (PyDict_SetItem(class_map, t, class) < 0) goto finally; res = class; finally: Py_XDECREF(import); Py_XDECREF(t); return res; } static int marker(Unpicklerobject *self) { if (self->num_marks < 1) { PyErr_SetString(UnpicklingError, "could not find MARK"); return -1; } return self->marks[--self->num_marks]; } static int load_none(Unpicklerobject *self) { if (PyList_Append(self->stack, Py_None) < 0) return -1; return 0; } static int load_int(Unpicklerobject *self) { PyObject *py_int = 0; char *endptr, *s; int len, res = -1; long l; if ((len = (*self->readline_func)(self, &s)) < 0) return -1; UNLESS(s=strndup(s,len)) return -1; errno = 0; l = strtol(s, &endptr, 0); if (errno || (*endptr != '\n') || (endptr[1] != '\0')) { /* Hm, maybe we've got something long. Let's try reading it as a Python long object. */ errno=0; UNLESS(py_int=PyLong_FromString(s,&endptr,0)) goto finally; if ((*endptr != '\n') || (endptr[1] != '\0')) { PyErr_SetString(PyExc_ValueError, "could not convert string to int"); goto finally; } } else { UNLESS(py_int = PyInt_FromLong(l)) goto finally; } if (PyList_Append(self->stack, py_int) < 0) goto finally; res = 0; finally: free(s); Py_XDECREF(py_int); return res; } static long calc_binint(char *s, int x) { unsigned char c; int i; long l; for (i = 0, l = 0L; i < x; i++) { c = (unsigned char)s[i]; l |= (long)c << (i * 8); } return l; } static int load_binintx(Unpicklerobject *self, char *s, int x) { PyObject *py_int = 0; long l; l = calc_binint(s, x); UNLESS(py_int = PyInt_FromLong(l)) return -1; if (PyList_Append(self->stack, py_int) < 0) { Py_DECREF(py_int); return -1; } Py_DECREF(py_int); return 0; } static int load_binint(Unpicklerobject *self) { char *s; if ((*self->read_func)(self, &s, 4) < 0) return -1; return load_binintx(self, s, 4); } static int load_binint1(Unpicklerobject *self) { char *s; if ((*self->read_func)(self, &s, 1) < 0) return -1; return load_binintx(self, s, 1); } static int load_binint2(Unpicklerobject *self) { char *s; if ((*self->read_func)(self, &s, 2) < 0) return -1; return load_binintx(self, s, 2); } static int load_long(Unpicklerobject *self) { PyObject *l = 0; char *end, *s; int len, res = -1; static PyObject *arg = 0; if ((len = (*self->readline_func)(self, &s)) < 0) return -1; UNLESS(s=strndup(s,len)) return -1; UNLESS(l = PyLong_FromString(s, &end, 0)) goto finally; if (PyList_Append(self->stack, l) < 0) goto finally; res = 0; finally: free(s); Py_XDECREF(l); return res; } static int load_float(Unpicklerobject *self) { PyObject *py_float = 0; char *endptr, *s; int len, res = -1; double d; if ((len = (*self->readline_func)(self, &s)) < 0) return -1; UNLESS(s=strndup(s,len)) return -1; errno = 0; d = strtod(s, &endptr); if (errno || (endptr[0] != '\n') || (endptr[1] != '\0')) { PyErr_SetString(PyExc_ValueError, "could not convert string to float"); goto finally; } UNLESS(py_float = PyFloat_FromDouble(d)) goto finally; if (PyList_Append(self->stack, py_float) < 0) goto finally; res = 0; finally: free(s); Py_XDECREF(py_float); return res; } #ifdef FORMAT_1_3 static int load_binfloat(Unpicklerobject *self) { PyObject *py_float = 0; int s, e, res = -1; long fhi, flo; double x; char *p; if ((*self->read_func)(self, &p, 8) < 0) return -1; /* First byte */ s = (*p>>7) & 1; e = (*p & 0x7F) << 4; p++; /* Second byte */ e |= (*p>>4) & 0xF; fhi = (*p & 0xF) << 24; p++; /* Third byte */ fhi |= (*p & 0xFF) << 16; p++; /* Fourth byte */ fhi |= (*p & 0xFF) << 8; p++; /* Fifth byte */ fhi |= *p & 0xFF; p++; /* Sixth byte */ flo = (*p & 0xFF) << 16; p++; /* Seventh byte */ flo |= (*p & 0xFF) << 8; p++; /* Eighth byte */ flo |= *p & 0xFF; x = (double)fhi + (double)flo / 16777216.0; /* 2**24 */ x /= 268435456.0; /* 2**28 */ /* XXX This sadly ignores Inf/NaN */ if (e == 0) e = -1022; else { x += 1.0; e -= 1023; } x = ldexp(x, e); if (s) x = -x; UNLESS(py_float = PyFloat_FromDouble(x)) goto finally; if (PyList_Append(self->stack, py_float) < 0) goto finally; res = 0; finally: Py_XDECREF(py_float); return res; } #endif static int load_string(Unpicklerobject *self) { PyObject *str = 0; int len, res = -1; char *s; static PyObject *eval_dict = 0; if ((len = (*self->readline_func)(self, &s)) < 0) return -1; UNLESS(s=strndup(s,len)) return -1; UNLESS(eval_dict) UNLESS(eval_dict = Py_BuildValue("{s{}}", "__builtins__")) goto finally; UNLESS(str = PyRun_String(s, eval_input, eval_dict, eval_dict)) goto finally; if (PyList_Append(self->stack, str) < 0) goto finally; res = 0; finally: free(s); Py_XDECREF(str); return res; } static int load_binstring(Unpicklerobject *self) { PyObject *py_string = 0; long l; int res = -1; char *s; if ((*self->read_func)(self, &s, 4) < 0) goto finally; l = calc_binint(s, 4); if ((*self->read_func)(self, &s, l) < 0) goto finally; UNLESS(py_string = PyString_FromStringAndSize(s, l)) goto finally; if (PyList_Append(self->stack, py_string) < 0) goto finally; res = 0; finally: Py_XDECREF(py_string); return res; } static int load_short_binstring(Unpicklerobject *self) { PyObject *py_string = 0; unsigned char l; int res = -1; char *s; if ((*self->read_func)(self, &s, 1) < 0) return -1; l = (unsigned char)s[0]; if ((*self->read_func)(self, &s, l) < 0) goto finally; UNLESS(py_string = PyString_FromStringAndSize(s, l)) goto finally; if (PyList_Append(self->stack, py_string) < 0) goto finally; res = 0; finally: Py_XDECREF(py_string); return res; } static int load_tuple(Unpicklerobject *self) { PyObject *tup = 0, *slice = 0, *list = 0; int i, j, res = -1; if ((i = marker(self)) < 0) goto finally; if ((j = PyList_Size(self->stack)) < 0) goto finally; UNLESS(slice = PyList_GetSlice(self->stack, i, j)) goto finally; UNLESS(tup = PySequence_Tuple(slice)) goto finally; UNLESS(list = PyList_New(1)) goto finally; Py_INCREF(tup); if (PyList_SetItem(list, 0, tup) < 0) goto finally; if (PyList_SetSlice(self->stack, i, j, list) < 0) goto finally; res = 0; finally: Py_XDECREF(tup); Py_XDECREF(list); Py_XDECREF(slice); return res; } static int load_empty_tuple(Unpicklerobject *self) { PyObject *tup = 0; int res; UNLESS(tup=PyTuple_New(0)) return -1; res=PyList_Append(self->stack, tup); Py_DECREF(tup); return res; } static int load_empty_list(Unpicklerobject *self) { PyObject *list = 0; int res; UNLESS(list=PyList_New(0)) return -1; res=PyList_Append(self->stack, list); Py_DECREF(list); return res; } static int load_empty_dict(Unpicklerobject *self) { PyObject *dict = 0; int res; UNLESS(dict=PyDict_New()) return -1; res=PyList_Append(self->stack, dict); Py_DECREF(dict); return res; } static int load_list(Unpicklerobject *self) { PyObject *list = 0, *slice = 0; int i, j, l, res = -1; if ((i = marker(self)) < 0) goto finally; if ((j = PyList_Size(self->stack)) < 0) goto finally; UNLESS(slice = PyList_GetSlice(self->stack, i, j)) goto finally; if((l=PyList_Size(slice)) < 0) goto finally; if(l) { UNLESS(list = PyList_New(1)) goto finally; Py_INCREF(slice); if (PyList_SetItem(list, 0, slice) < 0) goto finally; if (PyList_SetSlice(self->stack, i, j, list) < 0) goto finally; } else { if(PyList_Append(self->stack,slice) < 0) goto finally; } res = 0; finally: Py_XDECREF(list); Py_XDECREF(slice); return res; } static int load_dict(Unpicklerobject *self) { PyObject *list = 0, *dict = 0, *key = 0, *value = 0; int i, j, k, res = -1; if ((i = marker(self)) < 0) goto finally; if ((j = PyList_Size(self->stack)) < 0) goto finally; UNLESS(dict = PyDict_New()) goto finally; for (k = i; k < j; k += 2) { UNLESS(key = PyList_GET_ITEM((PyListObject *)self->stack, k)) goto finally; UNLESS(value = PyList_GET_ITEM((PyListObject *)self->stack, k + 1)) goto finally; if (PyDict_SetItem(dict, key, value) < 0) goto finally; } if(j) { UNLESS(list = PyList_New(1)) goto finally; Py_INCREF(dict); if (PyList_SetItem(list, 0, dict) < 0) goto finally; if (PyList_SetSlice(self->stack, i, j, list) < 0) goto finally; } else if(PyList_Append(self->stack, dict) < 0) goto finally; res = 0; finally: Py_XDECREF(dict); Py_XDECREF(list); return res; } static PyObject * Instance_New(PyObject *cls, PyObject *args) { int has_key; PyObject *safe=0, *r=0; if (PyClass_Check(cls)) if(r=PyInstance_New(cls, args, NULL)) return r; else goto err; if ((has_key = cPickle_PyMapping_HasKey(safe_constructors, cls)) < 0) goto err; if (!has_key) if(!(safe = PyObject_GetAttr(cls, __safe_for_unpickling___str)) || !PyObject_IsTrue(safe)) { PyErr_Format(UnpicklingError, "%s is not safe for unpickling", "O", cls); Py_XDECREF(safe); return NULL; } if(r=PyObject_CallObject(cls, args)) return r; err: { PyObject *tp, *v, *tb; PyErr_Fetch(&tp, &v, &tb); if(r=Py_BuildValue("OOO",v,cls,args)) { Py_XDECREF(v); v=r; } PyErr_Restore(tp,v,tb); } return NULL; } static int load_obj(Unpicklerobject *self) { PyObject *class = 0, *slice = 0, *tup = 0, *obj = 0; int i, len, res = -1; if ((i = marker(self)) < 0) goto finally; class = PyList_GET_ITEM((PyListObject *)self->stack, i); Py_INCREF(class); if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(slice = PyList_GetSlice(self->stack, i + 1, len)) goto finally; UNLESS(tup = PySequence_Tuple(slice)) goto finally; UNLESS(obj = Instance_New(class, tup)) goto finally; if (DEL_LIST_SLICE(self->stack, i, len) < 0) goto finally; if (PyList_Append(self->stack, obj) < 0) goto finally; res = 0; finally: Py_XDECREF(class); Py_XDECREF(slice); Py_XDECREF(tup); Py_XDECREF(obj); return res; } static int load_inst(Unpicklerobject *self) { PyObject *arg_tup = 0, *arg_slice = 0, *class = 0, *obj = 0, *module_name = 0, *class_name = 0; int i, j, len, res = -1; char *s; if ((i = marker(self)) < 0) goto finally; if ((j = PyList_Size(self->stack)) < 0) goto finally; UNLESS(arg_slice = PyList_GetSlice(self->stack, i, j)) goto finally; UNLESS(arg_tup = PySequence_Tuple(arg_slice)) goto finally; if (DEL_LIST_SLICE(self->stack, i, j) < 0) goto finally; if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(module_name = PyString_FromStringAndSize(s, len - 1)) goto finally; if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(class_name = PyString_FromStringAndSize(s, len - 1)) goto finally; UNLESS(class = find_class(module_name, class_name)) goto finally; UNLESS(obj = Instance_New(class, arg_tup)) goto finally; if (PyList_Append(self->stack, obj) < 0) goto finally; res = 0; finally: Py_XDECREF(class); Py_XDECREF(arg_slice); Py_XDECREF(arg_tup); Py_XDECREF(obj); Py_XDECREF(module_name); Py_XDECREF(class_name); return res; } static int load_global(Unpicklerobject *self) { PyObject *class = 0, *module_name = 0, *class_name = 0; int res = -1, len; char *s; if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(module_name = PyString_FromStringAndSize(s, len - 1)) goto finally; if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(class_name = PyString_FromStringAndSize(s, len - 1)) goto finally; UNLESS(class = find_class(module_name, class_name)) goto finally; if (PyList_Append(self->stack, class) < 0) goto finally; res = 0; finally: Py_XDECREF(class); Py_XDECREF(module_name); Py_XDECREF(class_name); return res; } static int load_persid(Unpicklerobject *self) { PyObject *pid = 0, *pers_load_val = 0; int len, res = -1; char *s; if (self->pers_func) { if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(pid = PyString_FromStringAndSize(s, len - 1)) goto finally; UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(pid); if (PyTuple_SetItem(self->arg, 0, pid) < 0) goto finally; UNLESS(pers_load_val = PyObject_CallObject(self->pers_func, self->arg)) goto finally; if (PyList_Append(self->stack, pers_load_val) < 0) goto finally; } res = 0; finally: Py_XDECREF(pid); Py_XDECREF(pers_load_val); return res; } static int load_binpersid(Unpicklerobject *self) { PyObject *pid = 0, *pers_load_val = 0; int len, res = -1; if (self->pers_func) { if ((len = PyList_Size(self->stack)) < 0) goto finally; pid = PyList_GET_ITEM((PyListObject *)self->stack, len - 1); Py_INCREF(pid); if (DEL_LIST_SLICE(self->stack, len - 1, len) < 0) goto finally; UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(pid); if (PyTuple_SetItem(self->arg, 0, pid) < 0) goto finally; UNLESS(pers_load_val = PyObject_CallObject(self->pers_func, self->arg)) goto finally; if (PyList_Append(self->stack, pers_load_val) < 0) goto finally; } res = 0; finally: Py_XDECREF(pid); Py_XDECREF(pers_load_val); return res; } static int load_pop(Unpicklerobject *self) { int len; if ((len = PyList_Size(self->stack)) < 0) return -1; if ((self->num_marks > 0) && (self->marks[self->num_marks - 1] == len)) self->num_marks--; else if (DEL_LIST_SLICE(self->stack, len - 1, len) < 0) return -1; return 0; } static int load_pop_mark(Unpicklerobject *self) { int i, len; if ((i = marker(self)) < 0) return -1; if ((len = PyList_Size(self->stack)) < 0) return -1; if (DEL_LIST_SLICE(self->stack, i, len) < 0) return -1; return 0; } static int load_dup(Unpicklerobject *self) { PyObject *last; int len; if ((len = PyList_Size(self->stack)) < 0) return -1; UNLESS(last = PyList_GetItem(self->stack, len - 1)) return -1; if (PyList_Append(self->stack, last) < 0) return -1; return 0; } static int load_get(Unpicklerobject *self) { PyObject *py_str = 0, *value = 0; int len, res = -1; char *s; if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(py_str = PyString_FromStringAndSize(s, len - 1)) goto finally; UNLESS(value = PyDict_GetItem(self->memo, py_str)) goto finally; if (PyList_Append(self->stack, value) < 0) goto finally; res = 0; finally: Py_XDECREF(py_str); return res; } static int load_binget(Unpicklerobject *self) { PyObject *py_key = 0, *value = 0; unsigned char key; int res = -1; char *s; if ((*self->read_func)(self, &s, 1) < 0) goto finally; key = (unsigned char)s[0]; UNLESS(py_key = PyInt_FromLong((long)key)) goto finally; UNLESS(value = PyDict_GetItem(self->memo, py_key)) goto finally; if (PyList_Append(self->stack, value) < 0) goto finally; res = 0; finally: Py_XDECREF(py_key); return res; } static int load_long_binget(Unpicklerobject *self) { PyObject *py_key = 0, *value = 0; unsigned char c, *s; long key; int res = -1; if ((*self->read_func)(self, &s, 4) < 0) goto finally; c = (unsigned char)s[0]; key = (long)c; c = (unsigned char)s[1]; key |= (long)c << 8; c = (unsigned char)s[2]; key |= (long)c << 16; c = (unsigned char)s[3]; key |= (long)c << 24; UNLESS(py_key = PyInt_FromLong(key)) goto finally; UNLESS(value = PyDict_GetItem(self->memo, py_key)) goto finally; if (PyList_Append(self->stack, value) < 0) goto finally; res = 0; finally: Py_XDECREF(py_key); return res; } static int load_put(Unpicklerobject *self) { PyObject *py_str = 0, *value = 0; int len, res = -1; char *s; if ((len = (*self->readline_func)(self, &s)) < 0) goto finally; UNLESS(py_str = PyString_FromStringAndSize(s, len - 1)) goto finally; if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(value = PyList_GetItem(self->stack, len - 1)) goto finally; if (PyDict_SetItem(self->memo, py_str, value) < 0) goto finally; res = 0; finally: Py_XDECREF(py_str); return res; } static int load_binput(Unpicklerobject *self) { PyObject *py_key = 0, *value = 0; unsigned char key, *s; int len, res = -1; if ((*self->read_func)(self, &s, 1) < 0) goto finally; key = (unsigned char)s[0]; UNLESS(py_key = PyInt_FromLong((long)key)) goto finally; if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(value = PyList_GetItem(self->stack, len - 1)) goto finally; if (PyDict_SetItem(self->memo, py_key, value) < 0) goto finally; res = 0; finally: Py_XDECREF(py_key); return res; } static int load_long_binput(Unpicklerobject *self) { PyObject *py_key = 0, *value = 0; long key; unsigned char c, *s; int len, res = -1; if ((*self->read_func)(self, &s, 4) < 0) goto finally; c = (unsigned char)s[0]; key = (long)c; c = (unsigned char)s[1]; key |= (long)c << 8; c = (unsigned char)s[2]; key |= (long)c << 16; c = (unsigned char)s[3]; key |= (long)c << 24; UNLESS(py_key = PyInt_FromLong(key)) goto finally; if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(value = PyList_GetItem(self->stack, len - 1)) goto finally; if (PyDict_SetItem(self->memo, py_key, value) < 0) goto finally; res = 0; finally: Py_XDECREF(py_key); return res; } static int do_append(Unpicklerobject *self, int x) { PyObject *value = 0, *list = 0, *append_method = 0; int len, i; if ((len = PyList_Size(self->stack)) < 0) return -1; UNLESS(list = PyList_GetItem(self->stack, x - 1)) goto err; if (PyList_Check(list)) { PyObject *slice = 0; int list_len; UNLESS(slice = PyList_GetSlice(self->stack, x, len)) return -1; list_len = PyList_Size(list); if (PyList_SetSlice(list, list_len, list_len, slice) < 0) { Py_DECREF(slice); return -1; } Py_DECREF(slice); } else { UNLESS(append_method = PyObject_GetAttr(list, append_str)) return -1; for (i = x; i < len; i++) { PyObject *junk; UNLESS(value = PyList_GetItem(self->stack, i)) return -1; UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto err; Py_INCREF(value); if (PyTuple_SetItem(self->arg, 0, value) < 0) goto err; UNLESS(junk = PyObject_CallObject(append_method, self->arg)) goto err; Py_DECREF(junk); } } if (DEL_LIST_SLICE(self->stack, x, len) < 0) goto err; Py_XDECREF(append_method); return 0; err: Py_XDECREF(append_method); return -1; } static int load_append(Unpicklerobject *self) { return do_append(self, PyList_Size(self->stack) - 1); } static int load_appends(Unpicklerobject *self) { return do_append(self, marker(self)); } static int do_setitems(Unpicklerobject *self, int x) { PyObject *value = 0, *key = 0, *dict = 0; int len, i, res = -1; if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(dict = PyList_GetItem(self->stack, x - 1)) goto finally; for (i = x; i < len; i += 2) { UNLESS(key = PyList_GetItem(self->stack, i)) goto finally; UNLESS(value = PyList_GetItem(self->stack, i + 1)) goto finally; if (PyObject_SetItem(dict, key, value) < 0) goto finally; } if (DEL_LIST_SLICE(self->stack, x, len) < 0) goto finally; res = 0; finally: return res; } static int load_setitem(Unpicklerobject *self) { return do_setitems(self, PyList_Size(self->stack) - 2); } static int load_setitems(Unpicklerobject *self) { return do_setitems(self, marker(self)); } static int load_build(Unpicklerobject *self) { PyObject *value = 0, *inst = 0, *instdict = 0, *d_key = 0, *d_value = 0, *junk = 0, *__setstate__ = 0; int len, i, res = -1; if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(value = PyList_GetItem(self->stack, len - 1)) goto finally; Py_INCREF(value); if (DEL_LIST_SLICE(self->stack, len - 1, len) < 0) goto finally; UNLESS(inst = PyList_GetItem(self->stack, len - 2)) goto finally; UNLESS(__setstate__ = PyObject_GetAttr(inst, __setstate___str)) { PyErr_Clear(); UNLESS(instdict = PyObject_GetAttr(inst, __dict___str)) goto finally; i = 0; while (PyDict_Next(value, &i, &d_key, &d_value)) { if (PyObject_SetItem(instdict, d_key, d_value) < 0) goto finally; } } else { UNLESS(self->arg) UNLESS(self->arg = PyTuple_New(1)) goto finally; Py_INCREF(value); if (PyTuple_SetItem(self->arg, 0, value) < 0) goto finally; UNLESS(junk = PyObject_CallObject(__setstate__, self->arg)) goto finally; Py_DECREF(junk); } res = 0; finally: Py_XDECREF(value); Py_XDECREF(instdict); Py_XDECREF(__setstate__); return res; } static int load_mark(Unpicklerobject *self) { int len; if ((len = PyList_Size(self->stack)) < 0) return -1; if (!self->marks_size) { self->marks_size = 20; UNLESS(self->marks = (int *)malloc(self->marks_size * sizeof(int))) { PyErr_NoMemory(); return -1; } } else if ((self->num_marks + 1) >= self->marks_size) { UNLESS(self->marks = (int *)realloc(self->marks, (self->marks_size + 20) * sizeof(int))) { PyErr_NoMemory(); return -1; } self->marks_size += 20; } self->marks[self->num_marks++] = len; return 0; } static int load_reduce(Unpicklerobject *self) { PyObject *callable = 0, *arg_tup = 0, *ob = 0; int len, res = -1; if ((len = PyList_Size(self->stack)) < 0) goto finally; UNLESS(arg_tup = PyList_GetItem(self->stack, len - 1)) goto finally; UNLESS(callable = PyList_GetItem(self->stack, len - 2)) goto finally; UNLESS(ob = Instance_New(callable, arg_tup)) goto finally; if (PyList_Append(self->stack, ob) < 0) goto finally; if (DEL_LIST_SLICE(self->stack, len - 2, len) < 0) goto finally; res = 0; finally: Py_XDECREF(ob); return res; } static PyObject * load(Unpicklerobject *self) { PyObject *stack = 0, *err = 0, *exc = 0, *val = 0, *tb = 0; int len; char *s; UNLESS(stack = PyList_New(0)) goto err; self->stack = stack; self->num_marks = 0; while (1) { if ((*self->read_func)(self, &s, 1) < 0) break; switch (s[0]) { case NONE: if (load_none(self) < 0) break; continue; case BININT: if (load_binint(self) < 0) break; continue; case BININT1: if (load_binint1(self) < 0) break; continue; case BININT2: if (load_binint2(self) < 0) break; continue; case INT: if (load_int(self) < 0) break; continue; case LONG: if (load_long(self) < 0) break; continue; case FLOAT: if (load_float(self) < 0) break; continue; #ifdef FORMAT_1_3 case BINFLOAT: if (load_binfloat(self) < 0) break; continue; #endif case BINSTRING: if (load_binstring(self) < 0) break; continue; case SHORT_BINSTRING: if (load_short_binstring(self) < 0) break; continue; case STRING: if (load_string(self) < 0) break; continue; case EMPTY_TUPLE: if (load_empty_tuple(self) < 0) break; continue; case TUPLE: if (load_tuple(self) < 0) break; continue; case EMPTY_LIST: if (load_empty_list(self) < 0) break; continue; case LIST: if (load_list(self) < 0) break; continue; case EMPTY_DICT: if (load_empty_dict(self) < 0) break; continue; case DICT: if (load_dict(self) < 0) break; continue; case OBJ: if (load_obj(self) < 0) break; continue; case INST: if (load_inst(self) < 0) break; continue; case GLOBAL: if (load_global(self) < 0) break; continue; case APPEND: if (load_append(self) < 0) break; continue; case APPENDS: if (load_appends(self) < 0) break; continue; case BUILD: if (load_build(self) < 0) break; continue; case DUP: if (load_dup(self) < 0) break; continue; case BINGET: if (load_binget(self) < 0) break; continue; case LONG_BINGET: if (load_long_binget(self) < 0) break; continue; case GET: if (load_get(self) < 0) break; continue; case MARK: if (load_mark(self) < 0) break; continue; case BINPUT: if (load_binput(self) < 0) break; continue; case LONG_BINPUT: if (load_long_binput(self) < 0) break; continue; case PUT: if (load_put(self) < 0) break; continue; case POP: if (load_pop(self) < 0) break; continue; case POP_MARK: if (load_pop_mark(self) < 0) break; continue; case SETITEM: if (load_setitem(self) < 0) break; continue; case SETITEMS: if (load_setitems(self) < 0) break; continue; case STOP: break; case PERSID: if (load_persid(self) < 0) break; continue; case BINPERSID: if (load_binpersid(self) < 0) break; continue; case REDUCE: if (load_reduce(self) < 0) break; continue; default: PyErr_Format(UnpicklingError, "invalid load key, '%s'.", "c", s[0]); goto err; } break; } if ((err = PyErr_Occurred()) == PyExc_EOFError) { PyErr_SetNone(PyExc_EOFError); goto err; } if (err) goto err; if ((len = PyList_Size(stack)) < 0) goto err; UNLESS(val = PyList_GetItem(stack, len - 1)) goto err; Py_INCREF(val); Py_DECREF(stack); self->stack=NULL; return val; err: self->stack=NULL; Py_XDECREF(stack); return NULL; } static PyObject * Unpickler_load(Unpicklerobject *self, PyObject *args) { UNLESS(PyArg_ParseTuple(args, "")) return NULL; return load(self); } static struct PyMethodDef Unpickler_methods[] = { {"load", (PyCFunction)Unpickler_load, 1, ""}, {NULL, NULL} /* sentinel */ }; static Unpicklerobject * newUnpicklerobject(PyObject *f) { Unpicklerobject *self; UNLESS(self = PyObject_NEW(Unpicklerobject, &Unpicklertype)) return NULL; self->file = NULL; self->arg = NULL; self->stack = NULL; self->pers_func = NULL; self->last_string = NULL; self->marks = NULL; self->num_marks = 0; self->marks_size = 0; self->buf_size = 0; self->read = NULL; self->readline = NULL; UNLESS(self->memo = PyDict_New()) { Py_XDECREF((PyObject *)self); return NULL; } Py_INCREF(f); self->file = f; /* Set read, readline based on type of f */ if (PyFile_Check(f)) { self->fp = PyFile_AsFile(f); self->read_func = read_file; self->readline_func = readline_file; } else if (PycStringIO_InputCheck(f)) { self->fp = NULL; self->read_func = read_cStringIO; self->readline_func = readline_cStringIO; } else { self->fp = NULL; self->read_func = read_other; self->readline_func = readline_other; UNLESS((self->readline = PyObject_GetAttr(f, readline_str)) && (self->read = PyObject_GetAttr(f, read_str))) { PyErr_Clear(); PyErr_SetString( PyExc_TypeError, "argument must have 'read' and " "'readline' attributes" ); Py_XDECREF((PyObject *)self); return NULL; } } return self; } static PyObject * get_Unpickler(PyObject *self, PyObject *args) { PyObject *file; UNLESS(PyArg_ParseTuple(args, "O", &file)) return NULL; return (PyObject *)newUnpicklerobject(file); } static void Unpickler_dealloc(Unpicklerobject *self) { Py_XDECREF(self->readline); Py_XDECREF(self->read); Py_XDECREF(self->file); Py_XDECREF(self->memo); Py_XDECREF(self->stack); Py_XDECREF(self->pers_func); Py_XDECREF(self->arg); Py_XDECREF(self->last_string); if (self->marks) { free(self->marks); } if (self->buf_size) { free(self->buf); } PyMem_DEL(self); } static PyObject * Unpickler_getattr(Unpicklerobject *self, char *name) { if (!strcmp(name, "persistent_load")) { if (!self->pers_func) { PyErr_SetString(PyExc_AttributeError, name); return NULL; } Py_INCREF(self->pers_func); return self->pers_func; } if (!strcmp(name, "memo")) { if (!self->memo) { PyErr_SetString(PyExc_AttributeError, name); return NULL; } Py_INCREF(self->memo); return self->memo; } if (!strcmp(name, "stack")) { if (!self->stack) { PyErr_SetString(PyExc_AttributeError, name); return NULL; } Py_INCREF(self->stack); return self->stack; } if (!strcmp(name, "UnpicklingError")) { Py_INCREF(UnpicklingError); return UnpicklingError; } return Py_FindMethod(Unpickler_methods, (PyObject *)self, name); } static int Unpickler_setattr(Unpicklerobject *self, char *name, PyObject *value) { if (!strcmp(name, "persistent_load")) { Py_XDECREF(self->pers_func); self->pers_func = value; Py_INCREF(value); return 0; } PyErr_SetString(PyExc_AttributeError, name); return -1; } static PyObject * cpm_dump(PyObject *self, PyObject *args) { PyObject *ob, *file, *res = NULL; Picklerobject *pickler = 0; int bin = 0; UNLESS(PyArg_ParseTuple(args, "OO|i", &ob, &file, &bin)) goto finally; UNLESS(pickler = newPicklerobject(file, bin)) goto finally; if (dump(pickler, ob) < 0) goto finally; Py_INCREF(Py_None); res = Py_None; finally: Py_XDECREF(pickler); return res; } static PyObject * cpm_dumps(PyObject *self, PyObject *args) { PyObject *ob, *file = 0, *res = NULL; Picklerobject *pickler = 0; int bin = 0; UNLESS(PyArg_ParseTuple(args, "O|i", &ob, &bin)) goto finally; UNLESS(file = PycStringIO->NewOutput(128)) goto finally; UNLESS(pickler = newPicklerobject(file, bin)) goto finally; if (dump(pickler, ob) < 0) goto finally; res = PycStringIO->cgetvalue(file); finally: Py_XDECREF(pickler); Py_XDECREF(file); return res; } static PyObject * cpm_load(PyObject *self, PyObject *args) { Unpicklerobject *unpickler = 0; PyObject *ob, *res = NULL; UNLESS(PyArg_ParseTuple(args, "O", &ob)) goto finally; UNLESS(unpickler = newUnpicklerobject(ob)) goto finally; res = load(unpickler); finally: Py_XDECREF(unpickler); return res; } static PyObject * cpm_loads(PyObject *self, PyObject *args) { PyObject *ob, *file = 0, *res = NULL; Unpicklerobject *unpickler = 0; UNLESS(PyArg_ParseTuple(args, "O", &ob)) goto finally; UNLESS(file = PycStringIO->NewInput(ob)) goto finally; UNLESS(unpickler = newUnpicklerobject(file)) goto finally; res = load(unpickler); finally: Py_XDECREF(file); Py_XDECREF(unpickler); return res; } static char Unpicklertype__doc__[] = ""; static PyTypeObject Unpicklertype_value() { PyTypeObject Unpicklertype = { PyObject_HEAD_INIT(&PyType_Type) 0, /*ob_size*/ "Unpickler", /*tp_name*/ sizeof(Unpicklerobject), /*tp_basicsize*/ 0, /*tp_itemsize*/ /* methods */ (destructor)Unpickler_dealloc, /*tp_dealloc*/ (printfunc)0, /*tp_print*/ (getattrfunc)Unpickler_getattr, /*tp_getattr*/ (setattrfunc)Unpickler_setattr, /*tp_setattr*/ (cmpfunc)0, /*tp_compare*/ (reprfunc)0, /*tp_repr*/ 0, /*tp_as_number*/ 0, /*tp_as_sequence*/ 0, /*tp_as_mapping*/ (hashfunc)0, /*tp_hash*/ (ternaryfunc)0, /*tp_call*/ (reprfunc)0, /*tp_str*/ /* Space for future expansion */ 0L,0L,0L,0L, Unpicklertype__doc__ /* Documentation string */ }; return Unpicklertype; } static struct PyMethodDef cPickle_methods[] = { {"dump", (PyCFunction)cpm_dump, 1, ""}, {"dumps", (PyCFunction)cpm_dumps, 1, ""}, {"load", (PyCFunction)cpm_load, 1, ""}, {"loads", (PyCFunction)cpm_loads, 1, ""}, {"Pickler", (PyCFunction)get_Pickler, 1, ""}, {"Unpickler", (PyCFunction)get_Unpickler, 1, ""}, { NULL, NULL } }; #define CHECK_FOR_ERRORS(MESS) \ if(PyErr_Occurred()) { \ PyObject *__sys_exc_type, *__sys_exc_value, *__sys_exc_traceback; \ PyErr_Fetch( &__sys_exc_type, &__sys_exc_value, &__sys_exc_traceback); \ fprintf(stderr, # MESS ":\n\t"); \ PyObject_Print(__sys_exc_type, stderr,0); \ fprintf(stderr,", "); \ PyObject_Print(__sys_exc_value, stderr,0); \ fprintf(stderr,"\n"); \ fflush(stderr); \ Py_FatalError(# MESS); \ } static int init_stuff(PyObject *module, PyObject *module_dict) { PyObject *string, *copy_reg; #define INIT_STR(S) UNLESS(S ## _str=PyString_FromString(#S)) return -1; INIT_STR(__class__); INIT_STR(__getinitargs__); INIT_STR(__dict__); INIT_STR(__getstate__); INIT_STR(__setstate__); INIT_STR(__name__); INIT_STR(__reduce__); INIT_STR(write); INIT_STR(__safe_for_unpickling__); INIT_STR(append); INIT_STR(read); INIT_STR(readline); UNLESS(builtins = PyImport_ImportModule("__builtin__")) return -1; UNLESS(copy_reg = PyImport_ImportModule("copy_reg")) return -1; UNLESS(dispatch_table = PyObject_GetAttrString(copy_reg, "dispatch_table")) return -1; UNLESS(safe_constructors = PyObject_GetAttrString(copy_reg, "safe_constructors")) return -1; Py_DECREF(copy_reg); UNLESS(string = PyImport_ImportModule("string")) return -1; UNLESS(atol_func = PyObject_GetAttrString(string, "atol")) return -1; Py_DECREF(string); UNLESS(empty_tuple = PyTuple_New(0)) return -1; UNLESS(class_map = PyDict_New()) return -1; UNLESS(PicklingError = PyString_FromString("cPickle.PicklingError")) return -1; if (PyDict_SetItemString(module_dict, "PicklingError", PicklingError) < 0) return -1; UNLESS(UnpicklingError = PyString_FromString("cPickle.UnpicklingError")) return -1; if (PyDict_SetItemString(module_dict, "UnpicklingError", UnpicklingError) < 0) return -1; PycString_IMPORT; return 0; } /* Initialization function for the module (*must* be called initcPickle) */ void initcPickle() { PyObject *m, *d; char *rev="$Revision$"; PyObject *format_version; PyObject *compatible_formats; /* Create the module and add the functions */ m = Py_InitModule4("cPickle", cPickle_methods, cPickle_module_documentation, (PyObject*)NULL,PYTHON_API_VERSION); Picklertype=Picklertype_value(); Unpicklertype=Unpicklertype_value(); /* Add some symbolic constants to the module */ d = PyModule_GetDict(m); PyDict_SetItemString(d,"__version__", PyString_FromStringAndSize(rev+11,strlen(rev+11)-2)); #ifdef FORMAT_1_3 format_version = PyString_FromString("1.3"); compatible_formats = Py_BuildValue("[sss]", "1.0", "1.1", "1.2"); #else format_version = PyString_FromString("1.2"); compatible_formats = Py_BuildValue("[ss]", "1.0", "1.1"); #endif PyDict_SetItemString(d, "format_version", format_version); PyDict_SetItemString(d, "compatible_formats", compatible_formats); init_stuff(m, d); CHECK_FOR_ERRORS("can't initialize module cPickle"); } /**************************************************************************** $Log$ Revision 2.4 1997/04/09 17:47:47 guido Give PyErr_Format a new name and make it static. Revision 2.3 1997/04/09 17:36:32 guido Jim Fulton's version 2.2. Revision 1.36 1997/03/11 22:05:02 chris write POP rather than POPMARK in non-binary mode use put2() in save_reduce() and save_inst() only if state is not a dictionary removed put2() from save_tuple() Revision 1.35 1997/03/11 20:03:30 jim Added log comment at bottom. ****************************************************************************/