Skip to content

Commit

Permalink
pythongh-104645: fix error handling in marshal tests (pythonGH-104646)
Browse files Browse the repository at this point in the history
(cherry picked from commit ac56a85)

Co-authored-by: Irit Katriel <1055913+iritkatriel@users.noreply.github.com>
  • Loading branch information
iritkatriel authored and miss-islington committed May 19, 2023
1 parent 702ce87 commit e1768d9
Showing 1 changed file with 14 additions and 18 deletions.
32 changes: 14 additions & 18 deletions Modules/_testcapimodule.c
Original file line number Diff line number Diff line change
Expand Up @@ -4639,10 +4639,9 @@ pymarshal_write_long_to_file(PyObject* self, PyObject *args)
}

PyMarshal_WriteLongToFile(value, fp, version);
assert(!PyErr_Occurred());

fclose(fp);
if (PyErr_Occurred())
return NULL;
Py_RETURN_NONE;
}

Expand All @@ -4665,10 +4664,9 @@ pymarshal_write_object_to_file(PyObject* self, PyObject *args)
}

PyMarshal_WriteObjectToFile(obj, fp, version);
assert(!PyErr_Occurred());

fclose(fp);
if (PyErr_Occurred())
return NULL;
Py_RETURN_NONE;
}

Expand Down Expand Up @@ -4726,48 +4724,46 @@ pymarshal_read_long_from_file(PyObject* self, PyObject *args)
static PyObject*
pymarshal_read_last_object_from_file(PyObject* self, PyObject *args)
{
PyObject *obj;
long pos;
PyObject *filename;
FILE *fp;

if (!PyArg_ParseTuple(args, "O:pymarshal_read_last_object_from_file", &filename))
return NULL;

fp = _Py_fopen_obj(filename, "rb");
FILE *fp = _Py_fopen_obj(filename, "rb");
if (fp == NULL) {
PyErr_SetFromErrno(PyExc_OSError);
return NULL;
}

obj = PyMarshal_ReadLastObjectFromFile(fp);
pos = ftell(fp);
PyObject *obj = PyMarshal_ReadLastObjectFromFile(fp);
long pos = ftell(fp);

fclose(fp);
if (obj == NULL) {
return NULL;
}
return Py_BuildValue("Nl", obj, pos);
}

static PyObject*
pymarshal_read_object_from_file(PyObject* self, PyObject *args)
{
PyObject *obj;
long pos;
PyObject *filename;
FILE *fp;

if (!PyArg_ParseTuple(args, "O:pymarshal_read_object_from_file", &filename))
return NULL;

fp = _Py_fopen_obj(filename, "rb");
FILE *fp = _Py_fopen_obj(filename, "rb");
if (fp == NULL) {
PyErr_SetFromErrno(PyExc_OSError);
return NULL;
}

obj = PyMarshal_ReadObjectFromFile(fp);
pos = ftell(fp);
PyObject *obj = PyMarshal_ReadObjectFromFile(fp);
long pos = ftell(fp);

fclose(fp);
if (obj == NULL) {
return NULL;
}
return Py_BuildValue("Nl", obj, pos);
}

Expand Down

0 comments on commit e1768d9

Please sign in to comment.