From 127b2ef2d568a929d08f07219a45283cb546ab00 Mon Sep 17 00:00:00 2001 From: Jeremy Hylton Date: Wed, 12 Apr 2000 22:04:01 +0000 Subject: [PATCH] raise TypeError when bad argument passed to cStringIO.StringIO --- Modules/cStringIO.c | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/Modules/cStringIO.c b/Modules/cStringIO.c index 6a5efd666a3..e816178dc9b 100644 --- a/Modules/cStringIO.c +++ b/Modules/cStringIO.c @@ -578,9 +578,14 @@ newIobject(PyObject *s) { Iobject *self; char *buf; int size; - - UNLESS(buf=PyString_AsString(s)) return NULL; - UNLESS(-1 != (size=PyString_Size(s))) return NULL; + + if (!PyString_Check(s)) { + PyErr_Format(PyExc_TypeError, "expected string, %.200s found", + s->ob_type->tp_name); + return NULL; + } + buf = PyString_AS_STRING(s); + size = PyString_GET_SIZE(s); UNLESS(self = PyObject_NEW(Iobject, &Itype)) return NULL; Py_INCREF(s); self->buf=buf; @@ -603,7 +608,8 @@ static PyObject * IO_StringIO(PyObject *self, PyObject *args) { PyObject *s=0; - UNLESS(PyArg_ParseTuple(args, "|O:StringIO", &s)) return NULL; + if (!PyArg_ParseTuple(args, "|O:StringIO", &s)) + return NULL; if(s) return newIobject(s); return newOobject(128); }