From c91fcaa43b5348ffad993b7e7da251f41abb46d2 Mon Sep 17 00:00:00 2001 From: Guido van Rossum Date: Mon, 29 Mar 1999 20:00:14 +0000 Subject: Protection against picling to/from closed (real) file. The problem was reported by Moshe Zadka. --- Modules/cPickle.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/Modules/cPickle.c b/Modules/cPickle.c index 3bf9ba1..614ff06 100644 --- a/Modules/cPickle.c +++ b/Modules/cPickle.c @@ -2072,6 +2072,10 @@ newPicklerobject(PyObject *file, int bin) { if (PyFile_Check(file)) { self->fp = PyFile_AsFile(file); + if (self->fp == NULL) { + PyErr_SetString(PyExc_IOError, "output file closed"); + return NULL; + } self->write_func = write_file; } else if (PycStringIO_OutputCheck(file)) { @@ -3897,6 +3901,10 @@ newUnpicklerobject(PyObject *f) { /* Set read, readline based on type of f */ if (PyFile_Check(f)) { self->fp = PyFile_AsFile(f); + if (self->fp == NULL) { + PyErr_SetString(PyExc_IOError, "input file closed"); + return NULL; + } self->read_func = read_file; self->readline_func = readline_file; } -- cgit v0.12