From: Georg Brandl <georg@python.org>
Date: Thu, 1 Jun 2006 06:39:19 +0000 (+0000)
Subject: Correctly unpickle 2.4 exceptions via __setstate__ (patch #1498571)
X-Git-Tag: v2.5b1~313
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=85ac8508346db10592afde6c9ded8785942f60b7;p=thirdparty%2FPython%2Fcpython.git

Correctly unpickle 2.4 exceptions via __setstate__ (patch #1498571)
---

diff --git a/Objects/exceptions.c b/Objects/exceptions.c
index fbf10fedab07..acc7da50c161 100644
--- a/Objects/exceptions.c
+++ b/Objects/exceptions.c
@@ -150,6 +150,29 @@ BaseException_reduce(PyBaseExceptionObject *self)
         return PyTuple_Pack(2, self->ob_type, self->args);
 }
 
+/*
+ * Needed for backward compatibility, since exceptions used to store
+ * all their attributes in the __dict__. Code is taken from cPickle's
+ * load_build function.
+ */
+static PyObject *
+BaseException_setstate(PyObject *self, PyObject *state)
+{
+    PyObject *d_key, *d_value;
+    Py_ssize_t i = 0;
+
+    if (state != Py_None) {
+        if (!PyDict_Check(state)) {
+            PyErr_SetString(PyExc_TypeError, "state is not a dictionary");
+            return NULL;
+        }
+        while (PyDict_Next(state, &i, &d_key, &d_value)) {
+            if (PyObject_SetAttr(self, d_key, d_value) < 0)
+                return NULL;
+        }
+    }
+    Py_RETURN_NONE;
+}
 
 #ifdef Py_USING_UNICODE
 /* while this method generates fairly uninspired output, it a least
@@ -168,6 +191,7 @@ BaseException_unicode(PyBaseExceptionObject *self)
 
 static PyMethodDef BaseException_methods[] = {
    {"__reduce__", (PyCFunction)BaseException_reduce, METH_NOARGS },
+   {"__setstate__", (PyCFunction)BaseException_setstate, METH_O },
 #ifdef Py_USING_UNICODE
    {"__unicode__", (PyCFunction)BaseException_unicode, METH_NOARGS },
 #endif