Issue #8468: bz2.BZ2File() accepts str with surrogates and bytes filenames
diff --git a/Modules/bz2module.c b/Modules/bz2module.c
index 550f1cf..f0ddf5c 100644
--- a/Modules/bz2module.c
+++ b/Modules/bz2module.c
@@ -1162,6 +1162,7 @@
{
static char *kwlist[] = {"filename", "mode", "buffering",
"compresslevel", 0};
+ PyObject *name_obj = NULL;
char *name;
char *mode = "r";
int buffering = -1;
@@ -1171,14 +1172,17 @@
self->size = -1;
- if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s|sii:BZ2File",
- kwlist, &name, &mode, &buffering,
+ if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O&|sii:BZ2File",
+ kwlist, PyUnicode_FSConverter, &name_obj,
+ &mode, &buffering,
&compresslevel))
return -1;
+ name = PyBytes_AsString(name_obj);
if (compresslevel < 1 || compresslevel > 9) {
PyErr_SetString(PyExc_ValueError,
"compresslevel must be between 1 and 9");
+ Py_DECREF(name_obj);
return -1;
}
@@ -1202,6 +1206,7 @@
if (error) {
PyErr_Format(PyExc_ValueError,
"invalid mode char %c", *mode);
+ Py_DECREF(name_obj);
return -1;
}
mode++;
@@ -1216,6 +1221,7 @@
mode = (mode_char == 'r') ? "rb" : "wb";
self->rawfp = fopen(name, mode);
+ Py_DECREF(name_obj);
if (self->rawfp == NULL) {
PyErr_SetFromErrno(PyExc_IOError);
return -1;