mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-03 08:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Issue #13393: BufferedReader.read1() now asks the full requested size to

Browse source 
the raw stream instead of limiting itself to the buffer size.
This commit is contained in:
Antoine Pitrou 2011-11-16 00:56:10 +01:00
parent 2821644d8a
commit 56a220ae4e
2 changed files with 19 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

49
Modules/_io/bufferedio.c
View file

@ -889,51 +889,34 @@ buffered_read1(buffered *self, PyObject *args)
if (n == 0)
return PyBytes_FromStringAndSize(NULL, 0);
if (!ENTER_BUFFERED(self))
return NULL;
/* Return up to n bytes. If at least one byte is buffered, we
only return buffered bytes. Otherwise, we do one raw read. */
/* XXX: this mimicks the io.py implementation but is probably wrong.
If we need to read from the raw stream, then we could actually read
all `n` bytes asked by the caller (and possibly more, so as to fill
our buffer for the next reads). */
have = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
if (have > 0) {
if (n > have)
n = have;
res = PyBytes_FromStringAndSize(self->buffer + self->pos, n);
if (res == NULL)
goto end;
self->pos += n;
goto end;
n = Py_MIN(have, n);
res = _bufferedreader_read_fast(self, n);
assert(res != Py_None);
return res;
}
if (self->writable) {
res = buffered_flush_and_rewind_unlocked(self);
if (res == NULL)
goto end;
res = PyBytes_FromStringAndSize(NULL, n);
if (res == NULL)
return NULL;
if (!ENTER_BUFFERED(self)) {
Py_DECREF(res);
return NULL;
}
/* Fill the buffer from the raw stream, and copy it to the result. */
_bufferedreader_reset_buf(self);
r = _bufferedreader_fill_buffer(self);
if (r == -1)
goto end;
r = _bufferedreader_raw_read(self, PyBytes_AS_STRING(res), n);
LEAVE_BUFFERED(self)
if (r == -1) {
Py_DECREF(res);
return NULL;
}
if (r == -2)
r = 0;
if (n > r)
n = r;
res = PyBytes_FromStringAndSize(self->buffer, n);
if (res == NULL)
goto end;
self->pos = n;
end:
LEAVE_BUFFERED(self)
_PyBytes_Resize(&res, r);
return res;
}