Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Refactor: replace query with parameter
Pass module state to _PyIOBase_check_seekable()
  • Loading branch information
erlend-aasland committed May 9, 2023
commit c68d657173994b1b3d7be543ae6bd789838f313f
3 changes: 2 additions & 1 deletion Modules/_io/_iomodule.h
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,8 @@ extern PyType_Spec winconsoleio_spec;
struct _io_state; // Forward decl.
Comment thread
erlend-aasland marked this conversation as resolved.
Outdated
extern PyObject* _PyIOBase_check_readable(PyObject *self, PyObject *args);
extern PyObject* _PyIOBase_check_writable(PyObject *self, PyObject *args);
extern PyObject* _PyIOBase_check_seekable(PyObject *self, PyObject *args);
extern PyObject* _PyIOBase_check_seekable(struct _io_state *state,
PyObject *self, PyObject *args);
extern PyObject* _PyIOBase_check_closed(PyObject *self, PyObject *args);

/* Helper for finalization.
Expand Down
9 changes: 6 additions & 3 deletions Modules/_io/bufferedio.c
Original file line number Diff line number Diff line change
Expand Up @@ -1224,8 +1224,10 @@ _io__Buffered_seek_impl(buffered *self, PyObject *targetobj, int whence)

CHECK_CLOSED(self, "seek of closed file")

if (_PyIOBase_check_seekable(self->raw, Py_True) == NULL)
_PyIO_State *state = IO_STATE();
if (_PyIOBase_check_seekable(state, self->raw, Py_True) == NULL) {
return NULL;
}

target = PyNumber_AsOff_t(targetobj, PyExc_ValueError);
if (target == -1 && PyErr_Occurred())
Expand Down Expand Up @@ -2293,8 +2295,10 @@ _io_BufferedRandom___init___impl(buffered *self, PyObject *raw,
self->ok = 0;
self->detached = 0;

if (_PyIOBase_check_seekable(raw, Py_True) == NULL)
_PyIO_State *state = find_io_state_by_def(Py_TYPE(self));
if (_PyIOBase_check_seekable(state, raw, Py_True) == NULL) {
return -1;
}
if (_PyIOBase_check_readable(raw, Py_True) == NULL)
return -1;
if (_PyIOBase_check_writable(raw, Py_True) == NULL)
Expand All @@ -2312,7 +2316,6 @@ _io_BufferedRandom___init___impl(buffered *self, PyObject *raw,
_bufferedwriter_reset_buf(self);
self->pos = 0;

_PyIO_State *state = find_io_state_by_def(Py_TYPE(self));
self->fast_closed_checks = (Py_IS_TYPE(self, state->PyBufferedRandom_Type) &&
Py_IS_TYPE(raw, state->PyFileIO_Type));

Expand Down
6 changes: 3 additions & 3 deletions Modules/_io/iobase.c
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,8 @@ _PyIOBase_check_closed(PyObject *self, PyObject *args)
static PyObject *
iobase_check_seekable(PyObject *self, PyObject *args)
{
return _PyIOBase_check_seekable(self, args);
_PyIO_State *state = IO_STATE();
return _PyIOBase_check_seekable(state, self, args);
}

static PyObject *
Expand Down Expand Up @@ -390,14 +391,13 @@ _io__IOBase_seekable_impl(PyObject *self)
}

PyObject *
_PyIOBase_check_seekable(PyObject *self, PyObject *args)
_PyIOBase_check_seekable(_PyIO_State *state, PyObject *self, PyObject *args)
{
PyObject *res = PyObject_CallMethodNoArgs(self, &_Py_ID(seekable));
if (res == NULL)
return NULL;
if (res != Py_True) {
Py_CLEAR(res);
_PyIO_State *state = IO_STATE();
iobase_unsupported(state, "File or stream is not seekable.");
return NULL;
}
Expand Down