From: Jeremy Hylton <jeremy@alum.mit.edu>
Date: Wed, 12 Apr 2000 22:04:01 +0000 (+0000)
Subject: raise TypeError when bad argument passed to cStringIO.StringIO
X-Git-Tag: v2.0b1~2046
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=127b2ef2d568a929d08f07219a45283cb546ab00;p=python

raise TypeError when bad argument passed to cStringIO.StringIO
---

diff --git a/Modules/cStringIO.c b/Modules/cStringIO.c
index 6a5efd666a..e816178dc9 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);
 }