Kaydet (Commit) 2f80d96c authored tarafından Guido van Rossum's avatar Guido van Rossum

Patch by Stephen Turner, who writes:

"""
It fixes a memory corruption error resulting from BadPickleGet
exceptions in load_get, load_binget and load_long_binget.  This was
initially reported on c.l.py as a problem with Cookie.py; see the thread
titled "python core dump (SIGBUS) on Solaris" for more details.

If PyDict_GetItem(self->memo, py_key) call failed, then py_key was being
Py_DECREF'd out of existence before call was made to
PyErr_SetObject(BadPickleGet, py_key).

The bug can be duplicated as follows:

import cPickle
cPickle.loads('garyp')

This raises a BadPickleGet exception whose value is a freed object.  A
core dump will soon follow.
"""

Jim Fulton approves of the patch.
üst 0eb55ac9
/* /*
* cPickle.c,v 1.70 1999/06/15 14:09:35 jim Exp * cPickle.c,v 1.71 1999/07/11 13:30:34 jim Exp
* *
* Copyright (c) 1996-1998, Digital Creations, Fredericksburg, VA, USA. * Copyright (c) 1996-1998, Digital Creations, Fredericksburg, VA, USA.
* All rights reserved. * All rights reserved.
...@@ -49,7 +49,7 @@ ...@@ -49,7 +49,7 @@
static char cPickle_module_documentation[] = static char cPickle_module_documentation[] =
"C implementation and optimization of the Python pickle module\n" "C implementation and optimization of the Python pickle module\n"
"\n" "\n"
"cPickle.c,v 1.70 1999/06/15 14:09:35 jim Exp\n" "cPickle.c,v 1.71 1999/07/11 13:30:34 jim Exp\n"
; ;
#include "Python.h" #include "Python.h"
...@@ -3017,6 +3017,7 @@ load_get(Unpicklerobject *self) { ...@@ -3017,6 +3017,7 @@ load_get(Unpicklerobject *self) {
PyObject *py_str = 0, *value = 0; PyObject *py_str = 0, *value = 0;
int len; int len;
char *s; char *s;
int rc;
if ((len = (*self->readline_func)(self, &s)) < 0) return -1; if ((len = (*self->readline_func)(self, &s)) < 0) return -1;
if (len < 2) return bad_readline(); if (len < 2) return bad_readline();
...@@ -3024,14 +3025,16 @@ load_get(Unpicklerobject *self) { ...@@ -3024,14 +3025,16 @@ load_get(Unpicklerobject *self) {
UNLESS (py_str = PyString_FromStringAndSize(s, len - 1)) return -1; UNLESS (py_str = PyString_FromStringAndSize(s, len - 1)) return -1;
value = PyDict_GetItem(self->memo, py_str); value = PyDict_GetItem(self->memo, py_str);
Py_DECREF(py_str);
if (! value) { if (! value) {
PyErr_SetObject(BadPickleGet, py_str); PyErr_SetObject(BadPickleGet, py_str);
return -1; rc = -1;
} } else {
PDATA_APPEND(self->stack, value, -1);
rc = 0;
}
PDATA_APPEND(self->stack, value, -1); Py_DECREF(py_str);
return 0; return rc;
} }
...@@ -3040,6 +3043,7 @@ load_binget(Unpicklerobject *self) { ...@@ -3040,6 +3043,7 @@ load_binget(Unpicklerobject *self) {
PyObject *py_key = 0, *value = 0; PyObject *py_key = 0, *value = 0;
unsigned char key; unsigned char key;
char *s; char *s;
int rc;
if ((*self->read_func)(self, &s, 1) < 0) return -1; if ((*self->read_func)(self, &s, 1) < 0) return -1;
...@@ -3047,14 +3051,16 @@ load_binget(Unpicklerobject *self) { ...@@ -3047,14 +3051,16 @@ load_binget(Unpicklerobject *self) {
UNLESS (py_key = PyInt_FromLong((long)key)) return -1; UNLESS (py_key = PyInt_FromLong((long)key)) return -1;
value = PyDict_GetItem(self->memo, py_key); value = PyDict_GetItem(self->memo, py_key);
Py_DECREF(py_key);
if (! value) { if (! value) {
PyErr_SetObject(BadPickleGet, py_key); PyErr_SetObject(BadPickleGet, py_key);
return -1; rc = -1;
} } else {
PDATA_APPEND(self->stack, value, -1);
rc = 0;
}
PDATA_APPEND(self->stack, value, -1); Py_DECREF(py_key);
return 0; return rc;
} }
...@@ -3063,6 +3069,7 @@ load_long_binget(Unpicklerobject *self) { ...@@ -3063,6 +3069,7 @@ load_long_binget(Unpicklerobject *self) {
PyObject *py_key = 0, *value = 0; PyObject *py_key = 0, *value = 0;
unsigned char c, *s; unsigned char c, *s;
long key; long key;
int rc;
if ((*self->read_func)(self, &s, 4) < 0) return -1; if ((*self->read_func)(self, &s, 4) < 0) return -1;
...@@ -3078,14 +3085,16 @@ load_long_binget(Unpicklerobject *self) { ...@@ -3078,14 +3085,16 @@ load_long_binget(Unpicklerobject *self) {
UNLESS (py_key = PyInt_FromLong((long)key)) return -1; UNLESS (py_key = PyInt_FromLong((long)key)) return -1;
value = PyDict_GetItem(self->memo, py_key); value = PyDict_GetItem(self->memo, py_key);
Py_DECREF(py_key);
if (! value) { if (! value) {
PyErr_SetObject(BadPickleGet, py_key); PyErr_SetObject(BadPickleGet, py_key);
return -1; rc = -1;
} } else {
PDATA_APPEND(self->stack, value, -1);
rc = 0;
}
PDATA_APPEND(self->stack, value, -1); Py_DECREF(py_key);
return 0; return rc;
} }
...@@ -4360,7 +4369,7 @@ init_stuff(PyObject *module, PyObject *module_dict) { ...@@ -4360,7 +4369,7 @@ init_stuff(PyObject *module, PyObject *module_dict) {
DL_EXPORT(void) DL_EXPORT(void)
initcPickle() { initcPickle() {
PyObject *m, *d, *v; PyObject *m, *d, *v;
char *rev="1.70"; char *rev="1.71";
PyObject *format_version; PyObject *format_version;
PyObject *compatible_formats; PyObject *compatible_formats;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment