Commit 86b48cff authored by Jeremy Hylton's avatar Jeremy Hylton

Undo needless INCREF chicanery introduced by SF patch #450702.

    Apparently this patch (rev 2.41) replaced all the good old "s#"
    formats in PyArg_ParseTuple() with "S".  Then it did
    PyString_FromStringAndSize() to get back the values setup by the
    "s#" format.  It also incref'd and decref'd the string obtained by
    "S" even though the argument tuple had a reference to it.

Replace PyString_AsString() calls with PyString_AS_STRING().

    A good rule of thumb -- if you never check the return value of
    PyString_AsString() to see if it's NULL, you ought to be using the
    macro <wink>.
parent 7da83d32
......@@ -136,14 +136,9 @@ PyZlib_compress(PyObject *self, PyObject *args)
Byte *input, *output;
int length, level=Z_DEFAULT_COMPRESSION, err;
z_stream zst;
PyObject * inputString;
/* require Python string object, optional 'level' arg */
if (!PyArg_ParseTuple(args, "S|i:compress", &inputString, &level))
return NULL;
/* now get a pointer to the internal string */
if (PyString_AsStringAndSize(inputString, (char**)&input, &length) == -1)
if (!PyArg_ParseTuple(args, "s#|i:compress", &input, &length, &level))
return NULL;
zst.avail_out = length + length/1000 + 12 + 1;
......@@ -158,8 +153,6 @@ PyZlib_compress(PyObject *self, PyObject *args)
/* Past the point of no return. From here on out, we need to make sure
we clean up mallocs & INCREFs. */
Py_INCREF(inputString); /* increment so that we hold ref */
zst.zalloc = (alloc_func)NULL;
zst.zfree = (free_func)Z_NULL;
zst.next_out = (Byte *)output;
......@@ -203,7 +196,6 @@ PyZlib_compress(PyObject *self, PyObject *args)
error:
free(output);
Py_DECREF(inputString);
return ReturnVal;
}
......@@ -222,12 +214,9 @@ PyZlib_decompress(PyObject *self, PyObject *args)
int length, err;
int wsize=DEF_WBITS, r_strlen=DEFAULTALLOC;
z_stream zst;
PyObject * inputString;
if (!PyArg_ParseTuple(args, "S|ii:decompress",
&inputString, &wsize, &r_strlen))
return NULL;
if (PyString_AsStringAndSize(inputString, (char**)&input, &length) == -1)
if (!PyArg_ParseTuple(args, "s#|ii:decompress",
&input, &length, &wsize, &r_strlen))
return NULL;
if (r_strlen <= 0)
......@@ -239,14 +228,9 @@ PyZlib_decompress(PyObject *self, PyObject *args)
if (!(result_str = PyString_FromStringAndSize(NULL, r_strlen)))
return NULL;
/* Past the point of no return. From here on out, we need to make sure
we clean up mallocs & INCREFs. */
Py_INCREF(inputString); /* increment so that we hold ref */
zst.zalloc = (alloc_func)NULL;
zst.zfree = (free_func)Z_NULL;
zst.next_out = (Byte *)PyString_AsString(result_str);
zst.next_out = (Byte *)PyString_AS_STRING(result_str);
zst.next_in = (Byte *)input;
err = inflateInit2(&zst, wsize);
......@@ -291,7 +275,7 @@ PyZlib_decompress(PyObject *self, PyObject *args)
result_str = NULL;
goto error;
}
zst.next_out = (unsigned char *)PyString_AsString(result_str) \
zst.next_out = (unsigned char *)PyString_AS_STRING(result_str) \
+ r_strlen;
zst.avail_out = r_strlen;
r_strlen = r_strlen << 1;
......@@ -306,15 +290,13 @@ PyZlib_decompress(PyObject *self, PyObject *args)
err = inflateEnd(&zst);
if (err != Z_OK) {
zlib_error(zst, err, "while finishing data decompression");
return NULL;
goto error;
}
_PyString_Resize(&result_str, zst.total_out);
Py_DECREF(inputString);
return result_str;
error:
Py_DECREF(inputString);
Py_XDECREF(result_str);
return NULL;
}
......@@ -433,12 +415,8 @@ PyZlib_objcompress(compobject *self, PyObject *args)
PyObject *RetVal;
Byte *input;
unsigned long start_total_out;
PyObject *inputString;
if (!PyArg_ParseTuple(args, "S:compress", &inputString))
return NULL;
if (PyString_AsStringAndSize(inputString, (char**)&input, &inplen) == -1)
if (!PyArg_ParseTuple(args, "s#:compress", &input, &inplen))
return NULL;
if (!(RetVal = PyString_FromStringAndSize(NULL, length)))
......@@ -446,13 +424,11 @@ PyZlib_objcompress(compobject *self, PyObject *args)
ENTER_ZLIB
Py_INCREF(inputString);
start_total_out = self->zst.total_out;
self->zst.avail_in = inplen;
self->zst.next_in = input;
self->zst.avail_out = length;
self->zst.next_out = (unsigned char *)PyString_AsString(RetVal);
self->zst.next_out = (unsigned char *)PyString_AS_STRING(RetVal);
Py_BEGIN_ALLOW_THREADS
err = deflate(&(self->zst), Z_NO_FLUSH);
......@@ -465,7 +441,7 @@ PyZlib_objcompress(compobject *self, PyObject *args)
RetVal = NULL;
goto error;
}
self->zst.next_out = (unsigned char *)PyString_AsString(RetVal) \
self->zst.next_out = (unsigned char *)PyString_AS_STRING(RetVal) \
+ length;
self->zst.avail_out = length;
length = length << 1;
......@@ -490,7 +466,6 @@ PyZlib_objcompress(compobject *self, PyObject *args)
RetVal = NULL;
error:
Py_DECREF(inputString);
LEAVE_ZLIB
return RetVal;
}
......@@ -514,9 +489,9 @@ PyZlib_objdecompress(compobject *self, PyObject *args)
PyObject *RetVal;
Byte *input;
unsigned long start_total_out;
PyObject * inputString;
if (!PyArg_ParseTuple(args, "S|i:decompress", &inputString, &max_length))
if (!PyArg_ParseTuple(args, "s#|i:decompress", &input,
&inplen, &max_length))
return NULL;
if (max_length < 0) {
PyErr_SetString(PyExc_ValueError,
......@@ -524,9 +499,6 @@ PyZlib_objdecompress(compobject *self, PyObject *args)
return NULL;
}
if (PyString_AsStringAndSize(inputString, (char**)&input, &inplen) == -1)
return NULL;
/* limit amount of data allocated to max_length */
if (max_length && length > max_length)
length = max_length;
......@@ -535,13 +507,11 @@ PyZlib_objdecompress(compobject *self, PyObject *args)
ENTER_ZLIB
Py_INCREF(inputString);
start_total_out = self->zst.total_out;
self->zst.avail_in = inplen;
self->zst.next_in = input;
self->zst.avail_out = length;
self->zst.next_out = (unsigned char *)PyString_AsString(RetVal);
self->zst.next_out = (unsigned char *)PyString_AS_STRING(RetVal);
Py_BEGIN_ALLOW_THREADS
err = inflate(&(self->zst), Z_SYNC_FLUSH);
......@@ -567,7 +537,7 @@ PyZlib_objdecompress(compobject *self, PyObject *args)
RetVal = NULL;
goto error;
}
self->zst.next_out = (unsigned char *)PyString_AsString(RetVal) \
self->zst.next_out = (unsigned char *)PyString_AS_STRING(RetVal) \
+ old_length;
self->zst.avail_out = length - old_length;
......@@ -618,8 +588,6 @@ PyZlib_objdecompress(compobject *self, PyObject *args)
RetVal = NULL;
error:
Py_DECREF(inputString);
LEAVE_ZLIB
return RetVal;
......@@ -658,7 +626,7 @@ PyZlib_flush(compobject *self, PyObject *args)
start_total_out = self->zst.total_out;
self->zst.avail_in = 0;
self->zst.avail_out = length;
self->zst.next_out = (unsigned char *)PyString_AsString(RetVal);
self->zst.next_out = (unsigned char *)PyString_AS_STRING(RetVal);
Py_BEGIN_ALLOW_THREADS
err = deflate(&(self->zst), flushmode);
......@@ -671,7 +639,7 @@ PyZlib_flush(compobject *self, PyObject *args)
RetVal = NULL;
goto error;
}
self->zst.next_out = (unsigned char *)PyString_AsString(RetVal) \
self->zst.next_out = (unsigned char *)PyString_AS_STRING(RetVal) \
+ length;
self->zst.avail_out = length;
length = length << 1;
......
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