"add_extension", "remove_extension", "clear_extension_cache"]
dispatch_table = {}
-safe_constructors = {}
def pickle(ob_type, pickle_function, constructor_ob=None):
if type(ob_type) is _ClassType:
def constructor(object):
if not callable(object):
raise TypeError("constructors must be callable")
- safe_constructors[object] = 1
# Example: provide pickling support for complex numbers.
obj = base.__new__(cls, state)
base.__init__(obj, state)
return obj
-_reconstructor.__safe_for_unpickling__ = 1
_HEAPTYPE = 1<<9
static PyObject *dispatch_table;
-static PyObject *safe_constructors;
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,
+ *write_str, *append_str,
*read_str, *readline_str, *__main___str, *__basicnew___str,
- *copy_reg_str, *dispatch_table_str, *safe_constructors_str;
+ *copy_reg_str, *dispatch_table_str;
/*************************************************************************
Internal Data type for pickle data. */
int (*readline_func)(struct Unpicklerobject *, char **);
int buf_size;
char *buf;
- PyObject *safe_constructors;
PyObject *find_class;
} Unpicklerobject;
static PyObject *
Instance_New(PyObject *cls, PyObject *args)
{
- int has_key;
- PyObject *safe=0, *r=0;
+ PyObject *r = 0;
if (PyClass_Check(cls)) {
int l;
else goto err;
}
- /* Is safe_constructors always a dict? */
- has_key = cPickle_PyMapping_HasKey(safe_constructors, cls);
- if (!has_key) {
- safe = PyObject_GetAttr(cls, __safe_for_unpickling___str);
- if (!safe ||
- !PyObject_IsTrue(safe)) {
- cPickle_ErrFormat(UnpicklingError,
- "%s is not safe for unpickling",
- "O", cls);
- Py_XDECREF(safe);
- return NULL;
- }
- Py_DECREF(safe);
- }
-
if (args==Py_None) {
/* Special case, call cls.__basicnew__() */
PyObject *basicnew;
self->buf_size = 0;
self->read = NULL;
self->readline = NULL;
- self->safe_constructors = NULL;
self->find_class = NULL;
if (!( self->memo = PyDict_New()))
}
}
- if (PyEval_GetRestricted()) {
- /* Restricted execution, get private tables */
- PyObject *m;
-
- if (!( m=PyImport_Import(copy_reg_str))) goto err;
- self->safe_constructors=PyObject_GetAttr(m,
- safe_constructors_str);
- Py_DECREF(m);
- if (!( self->safe_constructors )) goto err;
- }
- else {
- self->safe_constructors=safe_constructors;
- Py_INCREF(safe_constructors);
- }
-
return self;
err:
Py_XDECREF(self->pers_func);
Py_XDECREF(self->arg);
Py_XDECREF(self->last_string);
- Py_XDECREF(self->safe_constructors);
if (self->marks) {
free(self->marks);
INIT_STR(__main__);
INIT_STR(__reduce__);
INIT_STR(write);
- INIT_STR(__safe_for_unpickling__);
INIT_STR(append);
INIT_STR(read);
INIT_STR(readline);
INIT_STR(copy_reg);
INIT_STR(dispatch_table);
- INIT_STR(safe_constructors);
INIT_STR(__basicnew__);
if (!( copy_reg = PyImport_ImportModule("copy_reg")))
return -1;
- /* These next few are special because we want to use different
- ones in restricted mode. */
+ /* This is special because we want to use a different
+ one in restricted mode. */
dispatch_table = PyObject_GetAttr(copy_reg, dispatch_table_str);
if (!dispatch_table)
return -1;
- if (!( safe_constructors = PyObject_GetAttr(copy_reg,
- safe_constructors_str)))
- return -1;
-
Py_DECREF(copy_reg);
/* Down to here ********************************** */
PyObject *d; /* its dict */
PyObject *x;
- /* Types that use __reduce__ for pickling need to set the following
- * magical attr in the type dict, with a true value.
- */
- PyObject *safepickle = PyString_FromString("__safe_for_unpickling__");
- if (safepickle == NULL)
- return;
-
m = Py_InitModule3("datetime", module_methods,
"Fast implementation of the datetime type.");
}
}
- /* tzinfo values */
- d = PyDateTime_TZInfoType.tp_dict;
-
- if (PyDict_SetItem(d, safepickle, Py_True) < 0)
- return;
-
/* timedelta values */
d = PyDateTime_DeltaType.tp_dict;
- if (PyDict_SetItem(d, safepickle, Py_True) < 0)
- return;
-
x = new_delta(0, 0, 1, 0);
if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
return;
/* date values */
d = PyDateTime_DateType.tp_dict;
- if (PyDict_SetItem(d, safepickle, Py_True) < 0)
- return;
-
x = new_date(1, 1, 1);
if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
return;
/* time values */
d = PyDateTime_TimeType.tp_dict;
- if (PyDict_SetItem(d, safepickle, Py_True) < 0)
- return;
-
x = new_time(0, 0, 0, 0, Py_None);
if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
return;
/* datetime values */
d = PyDateTime_DateTimeType.tp_dict;
- if (PyDict_SetItem(d, safepickle, Py_True) < 0)
- return;
-
x = new_datetime(1, 1, 1, 0, 0, 0, 0, Py_None);
if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
return;
return;
Py_DECREF(x);
- Py_DECREF(safepickle);
-
/* module initialization */
PyModule_AddIntConstant(m, "MINYEAR", MINYEAR);
PyModule_AddIntConstant(m, "MAXYEAR", MAXYEAR);