blob: a8631e0bc61d4258dc84b1f72cb54b3a8fc41d22 [file] [log] [blame]
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001/*
2 An implementation of Buffered I/O as defined by PEP 3116 - "New I/O"
Antoine Pitrou3486a982011-05-12 01:57:53 +02003
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00004 Classes defined here: BufferedIOBase, BufferedReader, BufferedWriter,
5 BufferedRandom.
Antoine Pitrou3486a982011-05-12 01:57:53 +02006
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00007 Written by Amaury Forgeot d'Arc and Antoine Pitrou
8*/
9
10#define PY_SSIZE_T_CLEAN
11#include "Python.h"
12#include "structmember.h"
13#include "pythread.h"
14#include "_iomodule.h"
15
16/*
17 * BufferedIOBase class, inherits from IOBase.
18 */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000019PyDoc_STRVAR(bufferediobase_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000020 "Base class for buffered IO objects.\n"
21 "\n"
22 "The main difference with RawIOBase is that the read() method\n"
23 "supports omitting the size argument, and does not have a default\n"
24 "implementation that defers to readinto().\n"
25 "\n"
26 "In addition, read(), readinto() and write() may raise\n"
27 "BlockingIOError if the underlying raw stream is in non-blocking\n"
28 "mode and not ready; unlike their raw counterparts, they will never\n"
29 "return None.\n"
30 "\n"
31 "A typical implementation should not inherit from a RawIOBase\n"
32 "implementation, but wrap one.\n"
33 );
34
35static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000036bufferediobase_readinto(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000037{
38 Py_buffer buf;
39 Py_ssize_t len;
40 PyObject *data;
41
42 if (!PyArg_ParseTuple(args, "w*:readinto", &buf)) {
43 return NULL;
44 }
45
46 data = PyObject_CallMethod(self, "read", "n", buf.len);
47 if (data == NULL)
48 goto error;
49
50 if (!PyBytes_Check(data)) {
51 Py_DECREF(data);
52 PyErr_SetString(PyExc_TypeError, "read() should return bytes");
53 goto error;
54 }
55
56 len = Py_SIZE(data);
57 memcpy(buf.buf, PyBytes_AS_STRING(data), len);
58
59 PyBuffer_Release(&buf);
60 Py_DECREF(data);
61
62 return PyLong_FromSsize_t(len);
63
64 error:
65 PyBuffer_Release(&buf);
66 return NULL;
67}
68
69static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000070bufferediobase_unsupported(const char *message)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000071{
72 PyErr_SetString(IO_STATE->unsupported_operation, message);
73 return NULL;
74}
75
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000076PyDoc_STRVAR(bufferediobase_detach_doc,
Benjamin Petersond2e0c792009-05-01 20:40:59 +000077 "Disconnect this buffer from its underlying raw stream and return it.\n"
78 "\n"
79 "After the raw stream has been detached, the buffer is in an unusable\n"
80 "state.\n");
81
82static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000083bufferediobase_detach(PyObject *self)
Benjamin Petersond2e0c792009-05-01 20:40:59 +000084{
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000085 return bufferediobase_unsupported("detach");
Benjamin Petersond2e0c792009-05-01 20:40:59 +000086}
87
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000088PyDoc_STRVAR(bufferediobase_read_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000089 "Read and return up to n bytes.\n"
90 "\n"
91 "If the argument is omitted, None, or negative, reads and\n"
92 "returns all data until EOF.\n"
93 "\n"
94 "If the argument is positive, and the underlying raw stream is\n"
95 "not 'interactive', multiple raw reads may be issued to satisfy\n"
96 "the byte count (unless EOF is reached first). But for\n"
97 "interactive raw streams (as well as sockets and pipes), at most\n"
98 "one raw read will be issued, and a short result does not imply\n"
99 "that EOF is imminent.\n"
100 "\n"
101 "Returns an empty bytes object on EOF.\n"
102 "\n"
103 "Returns None if the underlying raw stream was open in non-blocking\n"
104 "mode and no data is available at the moment.\n");
105
106static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000107bufferediobase_read(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000108{
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000109 return bufferediobase_unsupported("read");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000110}
111
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000112PyDoc_STRVAR(bufferediobase_read1_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000113 "Read and return up to n bytes, with at most one read() call\n"
114 "to the underlying raw stream. A short result does not imply\n"
115 "that EOF is imminent.\n"
116 "\n"
117 "Returns an empty bytes object on EOF.\n");
118
119static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000120bufferediobase_read1(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000121{
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000122 return bufferediobase_unsupported("read1");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000123}
124
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000125PyDoc_STRVAR(bufferediobase_write_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000126 "Write the given buffer to the IO stream.\n"
127 "\n"
128 "Returns the number of bytes written, which is never less than\n"
129 "len(b).\n"
130 "\n"
131 "Raises BlockingIOError if the buffer is full and the\n"
132 "underlying raw stream cannot accept more data at the moment.\n");
133
134static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000135bufferediobase_write(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000136{
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000137 return bufferediobase_unsupported("write");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000138}
139
140
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000141static PyMethodDef bufferediobase_methods[] = {
142 {"detach", (PyCFunction)bufferediobase_detach, METH_NOARGS, bufferediobase_detach_doc},
143 {"read", bufferediobase_read, METH_VARARGS, bufferediobase_read_doc},
144 {"read1", bufferediobase_read1, METH_VARARGS, bufferediobase_read1_doc},
145 {"readinto", bufferediobase_readinto, METH_VARARGS, NULL},
146 {"write", bufferediobase_write, METH_VARARGS, bufferediobase_write_doc},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000147 {NULL, NULL}
148};
149
150PyTypeObject PyBufferedIOBase_Type = {
151 PyVarObject_HEAD_INIT(NULL, 0)
152 "_io._BufferedIOBase", /*tp_name*/
153 0, /*tp_basicsize*/
154 0, /*tp_itemsize*/
155 0, /*tp_dealloc*/
156 0, /*tp_print*/
157 0, /*tp_getattr*/
158 0, /*tp_setattr*/
159 0, /*tp_compare */
160 0, /*tp_repr*/
161 0, /*tp_as_number*/
162 0, /*tp_as_sequence*/
163 0, /*tp_as_mapping*/
164 0, /*tp_hash */
165 0, /*tp_call*/
166 0, /*tp_str*/
167 0, /*tp_getattro*/
168 0, /*tp_setattro*/
169 0, /*tp_as_buffer*/
170 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000171 bufferediobase_doc, /* tp_doc */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000172 0, /* tp_traverse */
173 0, /* tp_clear */
174 0, /* tp_richcompare */
175 0, /* tp_weaklistoffset */
176 0, /* tp_iter */
177 0, /* tp_iternext */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000178 bufferediobase_methods, /* tp_methods */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000179 0, /* tp_members */
180 0, /* tp_getset */
181 &PyIOBase_Type, /* tp_base */
182 0, /* tp_dict */
183 0, /* tp_descr_get */
184 0, /* tp_descr_set */
185 0, /* tp_dictoffset */
186 0, /* tp_init */
187 0, /* tp_alloc */
188 0, /* tp_new */
189};
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000190
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000191
192typedef struct {
193 PyObject_HEAD
194
195 PyObject *raw;
196 int ok; /* Initialized? */
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000197 int detached;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000198 int readable;
199 int writable;
Antoine Pitroue033e062010-10-29 10:38:18 +0000200 int deallocating;
Antoine Pitrou3486a982011-05-12 01:57:53 +0200201
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000202 /* True if this is a vanilla Buffered object (rather than a user derived
203 class) *and* the raw stream is a vanilla FileIO object. */
204 int fast_closed_checks;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000205
206 /* Absolute position inside the raw stream (-1 if unknown). */
207 Py_off_t abs_pos;
208
209 /* A static buffer of size `buffer_size` */
210 char *buffer;
211 /* Current logical position in the buffer. */
212 Py_off_t pos;
213 /* Position of the raw stream in the buffer. */
214 Py_off_t raw_pos;
215
216 /* Just after the last buffered byte in the buffer, or -1 if the buffer
217 isn't ready for reading. */
218 Py_off_t read_end;
219
220 /* Just after the last byte actually written */
221 Py_off_t write_pos;
222 /* Just after the last byte waiting to be written, or -1 if the buffer
223 isn't ready for writing. */
224 Py_off_t write_end;
225
Georg Brandldfd73442009-04-05 11:47:34 +0000226#ifdef WITH_THREAD
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000227 PyThread_type_lock lock;
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000228 volatile long owner;
Georg Brandldfd73442009-04-05 11:47:34 +0000229#endif
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000230
231 Py_ssize_t buffer_size;
232 Py_ssize_t buffer_mask;
233
234 PyObject *dict;
235 PyObject *weakreflist;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000236} buffered;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000237
238/*
239 Implementation notes:
Antoine Pitrou3486a982011-05-12 01:57:53 +0200240
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000241 * BufferedReader, BufferedWriter and BufferedRandom try to share most
242 methods (this is helped by the members `readable` and `writable`, which
243 are initialized in the respective constructors)
244 * They also share a single buffer for reading and writing. This enables
245 interleaved reads and writes without flushing. It also makes the logic
246 a bit trickier to get right.
247 * The absolute position of the raw stream is cached, if possible, in the
248 `abs_pos` member. It must be updated every time an operation is done
249 on the raw stream. If not sure, it can be reinitialized by calling
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000250 _buffered_raw_tell(), which queries the raw stream (_buffered_raw_seek()
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000251 also does it). To read it, use RAW_TELL().
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000252 * Three helpers, _bufferedreader_raw_read, _bufferedwriter_raw_write and
253 _bufferedwriter_flush_unlocked do a lot of useful housekeeping.
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000254
255 NOTE: we should try to maintain block alignment of reads and writes to the
256 raw stream (according to the buffer size), but for now it is only done
257 in read() and friends.
Antoine Pitrou3486a982011-05-12 01:57:53 +0200258
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000259*/
260
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000261/* These macros protect the buffered object against concurrent operations. */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000262
Georg Brandldfd73442009-04-05 11:47:34 +0000263#ifdef WITH_THREAD
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000264
265static int
266_enter_buffered_busy(buffered *self)
267{
268 if (self->owner == PyThread_get_thread_ident()) {
269 PyErr_Format(PyExc_RuntimeError,
270 "reentrant call inside %R", self);
271 return 0;
Antoine Pitrou5800b272009-11-01 12:05:48 +0000272 }
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000273 Py_BEGIN_ALLOW_THREADS
274 PyThread_acquire_lock(self->lock, 1);
275 Py_END_ALLOW_THREADS
276 return 1;
277}
278
279#define ENTER_BUFFERED(self) \
280 ( (PyThread_acquire_lock(self->lock, 0) ? \
281 1 : _enter_buffered_busy(self)) \
282 && (self->owner = PyThread_get_thread_ident(), 1) )
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000283
284#define LEAVE_BUFFERED(self) \
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000285 do { \
286 self->owner = 0; \
287 PyThread_release_lock(self->lock); \
288 } while(0);
289
Georg Brandldfd73442009-04-05 11:47:34 +0000290#else
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000291#define ENTER_BUFFERED(self) 1
Georg Brandldfd73442009-04-05 11:47:34 +0000292#define LEAVE_BUFFERED(self)
293#endif
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000294
295#define CHECK_INITIALIZED(self) \
296 if (self->ok <= 0) { \
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000297 if (self->detached) { \
298 PyErr_SetString(PyExc_ValueError, \
299 "raw stream has been detached"); \
300 } else { \
301 PyErr_SetString(PyExc_ValueError, \
302 "I/O operation on uninitialized object"); \
303 } \
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000304 return NULL; \
305 }
306
307#define CHECK_INITIALIZED_INT(self) \
308 if (self->ok <= 0) { \
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000309 if (self->detached) { \
310 PyErr_SetString(PyExc_ValueError, \
311 "raw stream has been detached"); \
312 } else { \
313 PyErr_SetString(PyExc_ValueError, \
314 "I/O operation on uninitialized object"); \
315 } \
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000316 return -1; \
317 }
318
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000319#define IS_CLOSED(self) \
320 (self->fast_closed_checks \
321 ? _PyFileIO_closed(self->raw) \
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000322 : buffered_closed(self))
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000323
324#define CHECK_CLOSED(self, error_msg) \
325 if (IS_CLOSED(self)) { \
326 PyErr_SetString(PyExc_ValueError, error_msg); \
327 return NULL; \
328 }
329
330
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000331#define VALID_READ_BUFFER(self) \
332 (self->readable && self->read_end != -1)
333
334#define VALID_WRITE_BUFFER(self) \
335 (self->writable && self->write_end != -1)
336
337#define ADJUST_POSITION(self, _new_pos) \
338 do { \
339 self->pos = _new_pos; \
340 if (VALID_READ_BUFFER(self) && self->read_end < self->pos) \
341 self->read_end = self->pos; \
342 } while(0)
343
344#define READAHEAD(self) \
345 ((self->readable && VALID_READ_BUFFER(self)) \
346 ? (self->read_end - self->pos) : 0)
347
348#define RAW_OFFSET(self) \
349 (((VALID_READ_BUFFER(self) || VALID_WRITE_BUFFER(self)) \
350 && self->raw_pos >= 0) ? self->raw_pos - self->pos : 0)
351
352#define RAW_TELL(self) \
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000353 (self->abs_pos != -1 ? self->abs_pos : _buffered_raw_tell(self))
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000354
355#define MINUS_LAST_BLOCK(self, size) \
356 (self->buffer_mask ? \
357 (size & ~self->buffer_mask) : \
358 (self->buffer_size * (size / self->buffer_size)))
359
360
361static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000362buffered_dealloc(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000363{
Antoine Pitroue033e062010-10-29 10:38:18 +0000364 self->deallocating = 1;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000365 if (self->ok && _PyIOBase_finalize((PyObject *) self) < 0)
366 return;
367 _PyObject_GC_UNTRACK(self);
368 self->ok = 0;
369 if (self->weakreflist != NULL)
370 PyObject_ClearWeakRefs((PyObject *)self);
371 Py_CLEAR(self->raw);
372 if (self->buffer) {
373 PyMem_Free(self->buffer);
374 self->buffer = NULL;
375 }
Georg Brandldfd73442009-04-05 11:47:34 +0000376#ifdef WITH_THREAD
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000377 if (self->lock) {
378 PyThread_free_lock(self->lock);
379 self->lock = NULL;
380 }
Georg Brandldfd73442009-04-05 11:47:34 +0000381#endif
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000382 Py_CLEAR(self->dict);
383 Py_TYPE(self)->tp_free((PyObject *)self);
384}
385
386static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000387buffered_traverse(buffered *self, visitproc visit, void *arg)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000388{
389 Py_VISIT(self->raw);
390 Py_VISIT(self->dict);
391 return 0;
392}
393
394static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000395buffered_clear(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000396{
397 if (self->ok && _PyIOBase_finalize((PyObject *) self) < 0)
398 return -1;
399 self->ok = 0;
400 Py_CLEAR(self->raw);
401 Py_CLEAR(self->dict);
402 return 0;
403}
404
Antoine Pitroue033e062010-10-29 10:38:18 +0000405/* Because this can call arbitrary code, it shouldn't be called when
406 the refcount is 0 (that is, not directly from tp_dealloc unless
407 the refcount has been temporarily re-incremented). */
Matthias Klosebee33162010-11-16 20:07:51 +0000408static PyObject *
Antoine Pitroue033e062010-10-29 10:38:18 +0000409buffered_dealloc_warn(buffered *self, PyObject *source)
410{
411 if (self->ok && self->raw) {
412 PyObject *r;
413 r = PyObject_CallMethod(self->raw, "_dealloc_warn", "O", source);
414 if (r)
415 Py_DECREF(r);
416 else
417 PyErr_Clear();
418 }
419 Py_RETURN_NONE;
420}
421
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000422/*
423 * _BufferedIOMixin methods
424 * This is not a class, just a collection of methods that will be reused
425 * by BufferedReader and BufferedWriter
426 */
427
428/* Flush and close */
429
430static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000431buffered_simple_flush(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000432{
433 CHECK_INITIALIZED(self)
434 return PyObject_CallMethodObjArgs(self->raw, _PyIO_str_flush, NULL);
435}
436
437static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000438buffered_closed(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000439{
440 int closed;
441 PyObject *res;
442 CHECK_INITIALIZED_INT(self)
443 res = PyObject_GetAttr(self->raw, _PyIO_str_closed);
444 if (res == NULL)
445 return -1;
446 closed = PyObject_IsTrue(res);
447 Py_DECREF(res);
448 return closed;
449}
450
451static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000452buffered_closed_get(buffered *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000453{
454 CHECK_INITIALIZED(self)
455 return PyObject_GetAttr(self->raw, _PyIO_str_closed);
456}
457
458static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000459buffered_close(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000460{
461 PyObject *res = NULL;
462 int r;
463
464 CHECK_INITIALIZED(self)
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000465 if (!ENTER_BUFFERED(self))
466 return NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000467
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000468 r = buffered_closed(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000469 if (r < 0)
470 goto end;
471 if (r > 0) {
472 res = Py_None;
473 Py_INCREF(res);
474 goto end;
475 }
Antoine Pitroue033e062010-10-29 10:38:18 +0000476
477 if (self->deallocating) {
478 PyObject *r = buffered_dealloc_warn(self, (PyObject *) self);
479 if (r)
480 Py_DECREF(r);
481 else
482 PyErr_Clear();
483 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000484 /* flush() will most probably re-take the lock, so drop it first */
485 LEAVE_BUFFERED(self)
486 res = PyObject_CallMethodObjArgs((PyObject *)self, _PyIO_str_flush, NULL);
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000487 if (!ENTER_BUFFERED(self))
488 return NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000489 if (res == NULL) {
Antoine Pitrou6be88762010-05-03 16:48:20 +0000490 goto end;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000491 }
492 Py_XDECREF(res);
493
494 res = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_close, NULL);
495
496end:
497 LEAVE_BUFFERED(self)
498 return res;
499}
500
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000501/* detach */
502
503static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000504buffered_detach(buffered *self, PyObject *args)
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000505{
506 PyObject *raw, *res;
507 CHECK_INITIALIZED(self)
508 res = PyObject_CallMethodObjArgs((PyObject *)self, _PyIO_str_flush, NULL);
509 if (res == NULL)
510 return NULL;
511 Py_DECREF(res);
512 raw = self->raw;
513 self->raw = NULL;
514 self->detached = 1;
515 self->ok = 0;
516 return raw;
517}
518
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000519/* Inquiries */
520
521static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000522buffered_seekable(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000523{
524 CHECK_INITIALIZED(self)
525 return PyObject_CallMethodObjArgs(self->raw, _PyIO_str_seekable, NULL);
526}
527
528static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000529buffered_readable(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000530{
531 CHECK_INITIALIZED(self)
532 return PyObject_CallMethodObjArgs(self->raw, _PyIO_str_readable, NULL);
533}
534
535static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000536buffered_writable(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000537{
538 CHECK_INITIALIZED(self)
539 return PyObject_CallMethodObjArgs(self->raw, _PyIO_str_writable, NULL);
540}
541
542static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000543buffered_name_get(buffered *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000544{
545 CHECK_INITIALIZED(self)
546 return PyObject_GetAttrString(self->raw, "name");
547}
548
549static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000550buffered_mode_get(buffered *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000551{
552 CHECK_INITIALIZED(self)
553 return PyObject_GetAttrString(self->raw, "mode");
554}
555
556/* Lower-level APIs */
557
558static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000559buffered_fileno(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000560{
561 CHECK_INITIALIZED(self)
562 return PyObject_CallMethodObjArgs(self->raw, _PyIO_str_fileno, NULL);
563}
564
565static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000566buffered_isatty(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000567{
568 CHECK_INITIALIZED(self)
569 return PyObject_CallMethodObjArgs(self->raw, _PyIO_str_isatty, NULL);
570}
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000571
Antoine Pitrou243757e2010-11-05 21:15:39 +0000572/* Serialization */
573
574static PyObject *
575buffered_getstate(buffered *self, PyObject *args)
576{
577 PyErr_Format(PyExc_TypeError,
578 "cannot serialize '%s' object", Py_TYPE(self)->tp_name);
579 return NULL;
580}
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000581
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000582/* Forward decls */
583static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000584_bufferedwriter_flush_unlocked(buffered *, int);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000585static Py_ssize_t
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000586_bufferedreader_fill_buffer(buffered *self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000587static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000588_bufferedreader_reset_buf(buffered *self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000589static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000590_bufferedwriter_reset_buf(buffered *self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000591static PyObject *
Victor Stinnerbc93a112011-06-01 00:01:24 +0200592_bufferedreader_peek_unlocked(buffered *self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000593static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000594_bufferedreader_read_all(buffered *self);
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000595static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000596_bufferedreader_read_fast(buffered *self, Py_ssize_t);
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000597static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000598_bufferedreader_read_generic(buffered *self, Py_ssize_t);
Antoine Pitrou3486a982011-05-12 01:57:53 +0200599static Py_ssize_t
600_bufferedreader_raw_read(buffered *self, char *start, Py_ssize_t len);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000601
602/*
603 * Helpers
604 */
605
606/* Returns the address of the `written` member if a BlockingIOError was
607 raised, NULL otherwise. The error is always re-raised. */
608static Py_ssize_t *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000609_buffered_check_blocking_error(void)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000610{
611 PyObject *t, *v, *tb;
612 PyBlockingIOErrorObject *err;
613
614 PyErr_Fetch(&t, &v, &tb);
615 if (v == NULL || !PyErr_GivenExceptionMatches(v, PyExc_BlockingIOError)) {
616 PyErr_Restore(t, v, tb);
617 return NULL;
618 }
619 err = (PyBlockingIOErrorObject *) v;
620 /* TODO: sanity check (err->written >= 0) */
621 PyErr_Restore(t, v, tb);
622 return &err->written;
623}
624
625static Py_off_t
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000626_buffered_raw_tell(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000627{
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000628 Py_off_t n;
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000629 PyObject *res;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000630 res = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_tell, NULL);
631 if (res == NULL)
632 return -1;
633 n = PyNumber_AsOff_t(res, PyExc_ValueError);
634 Py_DECREF(res);
635 if (n < 0) {
636 if (!PyErr_Occurred())
637 PyErr_Format(PyExc_IOError,
Mark Dickinson1a0aaaa2009-11-24 20:54:11 +0000638 "Raw stream returned invalid position %" PY_PRIdOFF,
Antoine Pitrou3486a982011-05-12 01:57:53 +0200639 (PY_OFF_T_COMPAT)n);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000640 return -1;
641 }
642 self->abs_pos = n;
643 return n;
644}
645
646static Py_off_t
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000647_buffered_raw_seek(buffered *self, Py_off_t target, int whence)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000648{
649 PyObject *res, *posobj, *whenceobj;
650 Py_off_t n;
651
652 posobj = PyLong_FromOff_t(target);
653 if (posobj == NULL)
654 return -1;
655 whenceobj = PyLong_FromLong(whence);
656 if (whenceobj == NULL) {
657 Py_DECREF(posobj);
658 return -1;
659 }
660 res = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_seek,
661 posobj, whenceobj, NULL);
662 Py_DECREF(posobj);
663 Py_DECREF(whenceobj);
664 if (res == NULL)
665 return -1;
666 n = PyNumber_AsOff_t(res, PyExc_ValueError);
667 Py_DECREF(res);
668 if (n < 0) {
669 if (!PyErr_Occurred())
670 PyErr_Format(PyExc_IOError,
Mark Dickinson1a0aaaa2009-11-24 20:54:11 +0000671 "Raw stream returned invalid position %" PY_PRIdOFF,
Antoine Pitrou3486a982011-05-12 01:57:53 +0200672 (PY_OFF_T_COMPAT)n);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000673 return -1;
674 }
675 self->abs_pos = n;
676 return n;
677}
678
679static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000680_buffered_init(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000681{
682 Py_ssize_t n;
683 if (self->buffer_size <= 0) {
684 PyErr_SetString(PyExc_ValueError,
685 "buffer size must be strictly positive");
686 return -1;
687 }
688 if (self->buffer)
689 PyMem_Free(self->buffer);
690 self->buffer = PyMem_Malloc(self->buffer_size);
691 if (self->buffer == NULL) {
692 PyErr_NoMemory();
693 return -1;
694 }
Georg Brandldfd73442009-04-05 11:47:34 +0000695#ifdef WITH_THREAD
Antoine Pitrouc881f152010-08-01 16:53:42 +0000696 if (self->lock)
697 PyThread_free_lock(self->lock);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000698 self->lock = PyThread_allocate_lock();
699 if (self->lock == NULL) {
700 PyErr_SetString(PyExc_RuntimeError, "can't allocate read lock");
701 return -1;
702 }
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000703 self->owner = 0;
Georg Brandldfd73442009-04-05 11:47:34 +0000704#endif
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000705 /* Find out whether buffer_size is a power of 2 */
706 /* XXX is this optimization useful? */
707 for (n = self->buffer_size - 1; n & 1; n >>= 1)
708 ;
709 if (n == 0)
710 self->buffer_mask = self->buffer_size - 1;
711 else
712 self->buffer_mask = 0;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000713 if (_buffered_raw_tell(self) == -1)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000714 PyErr_Clear();
715 return 0;
716}
717
Antoine Pitrou707ce822011-02-25 21:24:11 +0000718/* Return 1 if an EnvironmentError with errno == EINTR is set (and then
719 clears the error indicator), 0 otherwise.
720 Should only be called when PyErr_Occurred() is true.
721*/
722static int
723_trap_eintr(void)
724{
725 static PyObject *eintr_int = NULL;
726 PyObject *typ, *val, *tb;
727 PyEnvironmentErrorObject *env_err;
728
729 if (eintr_int == NULL) {
730 eintr_int = PyLong_FromLong(EINTR);
731 assert(eintr_int != NULL);
732 }
733 if (!PyErr_ExceptionMatches(PyExc_EnvironmentError))
734 return 0;
735 PyErr_Fetch(&typ, &val, &tb);
736 PyErr_NormalizeException(&typ, &val, &tb);
737 env_err = (PyEnvironmentErrorObject *) val;
738 assert(env_err != NULL);
739 if (env_err->myerrno != NULL &&
740 PyObject_RichCompareBool(env_err->myerrno, eintr_int, Py_EQ) > 0) {
741 Py_DECREF(typ);
742 Py_DECREF(val);
743 Py_XDECREF(tb);
744 return 1;
745 }
746 /* This silences any error set by PyObject_RichCompareBool() */
747 PyErr_Restore(typ, val, tb);
748 return 0;
749}
750
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000751/*
752 * Shared methods and wrappers
753 */
754
755static PyObject *
Antoine Pitroue05565e2011-08-20 14:39:23 +0200756buffered_flush_and_rewind_unlocked(buffered *self)
757{
758 PyObject *res;
759
760 res = _bufferedwriter_flush_unlocked(self, 0);
761 if (res == NULL)
762 return NULL;
763 Py_DECREF(res);
764
765 if (self->readable) {
766 /* Rewind the raw stream so that its position corresponds to
767 the current logical position. */
768 Py_off_t n;
769 n = _buffered_raw_seek(self, -RAW_OFFSET(self), 1);
770 _bufferedreader_reset_buf(self);
771 if (n == -1)
772 return NULL;
773 }
774 Py_RETURN_NONE;
775}
776
777static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000778buffered_flush(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000779{
780 PyObject *res;
781
782 CHECK_INITIALIZED(self)
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000783 CHECK_CLOSED(self, "flush of closed file")
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000784
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000785 if (!ENTER_BUFFERED(self))
786 return NULL;
Antoine Pitroue05565e2011-08-20 14:39:23 +0200787 res = buffered_flush_and_rewind_unlocked(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000788 LEAVE_BUFFERED(self)
789
790 return res;
791}
792
793static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000794buffered_peek(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000795{
796 Py_ssize_t n = 0;
797 PyObject *res = NULL;
798
799 CHECK_INITIALIZED(self)
800 if (!PyArg_ParseTuple(args, "|n:peek", &n)) {
801 return NULL;
802 }
803
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000804 if (!ENTER_BUFFERED(self))
805 return NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000806
807 if (self->writable) {
Antoine Pitroue05565e2011-08-20 14:39:23 +0200808 res = buffered_flush_and_rewind_unlocked(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000809 if (res == NULL)
810 goto end;
811 Py_CLEAR(res);
812 }
Victor Stinnerbc93a112011-06-01 00:01:24 +0200813 res = _bufferedreader_peek_unlocked(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000814
815end:
816 LEAVE_BUFFERED(self)
817 return res;
818}
819
820static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000821buffered_read(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000822{
823 Py_ssize_t n = -1;
824 PyObject *res;
825
826 CHECK_INITIALIZED(self)
Benjamin Petersonbf5ff762009-12-13 19:25:34 +0000827 if (!PyArg_ParseTuple(args, "|O&:read", &_PyIO_ConvertSsize_t, &n)) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000828 return NULL;
829 }
830 if (n < -1) {
831 PyErr_SetString(PyExc_ValueError,
832 "read length must be positive or -1");
833 return NULL;
834 }
835
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000836 CHECK_CLOSED(self, "read of closed file")
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000837
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000838 if (n == -1) {
839 /* The number of bytes is unspecified, read until the end of stream */
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000840 if (!ENTER_BUFFERED(self))
841 return NULL;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000842 res = _bufferedreader_read_all(self);
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000843 }
844 else {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000845 res = _bufferedreader_read_fast(self, n);
Antoine Pitroue05565e2011-08-20 14:39:23 +0200846 if (res != Py_None)
847 return res;
848 Py_DECREF(res);
849 if (!ENTER_BUFFERED(self))
850 return NULL;
851 res = _bufferedreader_read_generic(self, n);
Antoine Pitrou711af3a2009-04-11 15:39:24 +0000852 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000853
Antoine Pitroue05565e2011-08-20 14:39:23 +0200854 LEAVE_BUFFERED(self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000855 return res;
856}
857
858static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000859buffered_read1(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000860{
861 Py_ssize_t n, have, r;
862 PyObject *res = NULL;
863
864 CHECK_INITIALIZED(self)
865 if (!PyArg_ParseTuple(args, "n:read1", &n)) {
866 return NULL;
867 }
868
869 if (n < 0) {
870 PyErr_SetString(PyExc_ValueError,
871 "read length must be positive");
872 return NULL;
873 }
874 if (n == 0)
875 return PyBytes_FromStringAndSize(NULL, 0);
876
Antoine Pitrouf3b68b32010-12-03 18:41:39 +0000877 if (!ENTER_BUFFERED(self))
878 return NULL;
Antoine Pitrou3486a982011-05-12 01:57:53 +0200879
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000880 /* Return up to n bytes. If at least one byte is buffered, we
881 only return buffered bytes. Otherwise, we do one raw read. */
882
883 /* XXX: this mimicks the io.py implementation but is probably wrong.
884 If we need to read from the raw stream, then we could actually read
885 all `n` bytes asked by the caller (and possibly more, so as to fill
886 our buffer for the next reads). */
887
888 have = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
889 if (have > 0) {
890 if (n > have)
891 n = have;
892 res = PyBytes_FromStringAndSize(self->buffer + self->pos, n);
893 if (res == NULL)
894 goto end;
895 self->pos += n;
896 goto end;
897 }
898
Antoine Pitroue05565e2011-08-20 14:39:23 +0200899 if (self->writable) {
900 res = buffered_flush_and_rewind_unlocked(self);
901 if (res == NULL)
902 goto end;
903 Py_DECREF(res);
904 }
905
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000906 /* Fill the buffer from the raw stream, and copy it to the result. */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000907 _bufferedreader_reset_buf(self);
908 r = _bufferedreader_fill_buffer(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000909 if (r == -1)
910 goto end;
911 if (r == -2)
912 r = 0;
913 if (n > r)
914 n = r;
915 res = PyBytes_FromStringAndSize(self->buffer, n);
916 if (res == NULL)
917 goto end;
918 self->pos = n;
919
920end:
921 LEAVE_BUFFERED(self)
922 return res;
923}
924
925static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000926buffered_readinto(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000927{
Antoine Pitrou3486a982011-05-12 01:57:53 +0200928 Py_buffer buf;
929 Py_ssize_t n, written = 0, remaining;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000930 PyObject *res = NULL;
931
932 CHECK_INITIALIZED(self)
Antoine Pitrou3486a982011-05-12 01:57:53 +0200933
934 if (!PyArg_ParseTuple(args, "w*:readinto", &buf))
935 return NULL;
936
937 n = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
938 if (n > 0) {
939 if (n >= buf.len) {
940 memcpy(buf.buf, self->buffer + self->pos, buf.len);
941 self->pos += buf.len;
942 res = PyLong_FromSsize_t(buf.len);
943 goto end_unlocked;
944 }
945 memcpy(buf.buf, self->buffer + self->pos, n);
946 self->pos += n;
947 written = n;
948 }
949
950 if (!ENTER_BUFFERED(self))
951 goto end_unlocked;
952
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000953 if (self->writable) {
Antoine Pitroue8bb1a02011-08-20 14:52:04 +0200954 res = buffered_flush_and_rewind_unlocked(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000955 if (res == NULL)
956 goto end;
Antoine Pitrou3486a982011-05-12 01:57:53 +0200957 Py_CLEAR(res);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000958 }
Antoine Pitrou3486a982011-05-12 01:57:53 +0200959
960 _bufferedreader_reset_buf(self);
961 self->pos = 0;
962
963 for (remaining = buf.len - written;
964 remaining > 0;
965 written += n, remaining -= n) {
966 /* If remaining bytes is larger than internal buffer size, copy
967 * directly into caller's buffer. */
968 if (remaining > self->buffer_size) {
Antoine Pitrou4e19e112011-05-12 02:07:00 +0200969 n = _bufferedreader_raw_read(self, (char *) buf.buf + written,
970 remaining);
Antoine Pitrou3486a982011-05-12 01:57:53 +0200971 }
972 else {
973 n = _bufferedreader_fill_buffer(self);
974 if (n > 0) {
975 if (n > remaining)
976 n = remaining;
Antoine Pitrou4e19e112011-05-12 02:07:00 +0200977 memcpy((char *) buf.buf + written,
978 self->buffer + self->pos, n);
Antoine Pitrou3486a982011-05-12 01:57:53 +0200979 self->pos += n;
980 continue; /* short circuit */
981 }
982 }
983 if (n == 0 || (n == -2 && written > 0))
984 break;
985 if (n < 0) {
986 if (n == -2) {
987 Py_INCREF(Py_None);
988 res = Py_None;
989 }
990 goto end;
991 }
992 }
993 res = PyLong_FromSsize_t(written);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000994
995end:
Antoine Pitrou3486a982011-05-12 01:57:53 +0200996 LEAVE_BUFFERED(self);
997end_unlocked:
998 PyBuffer_Release(&buf);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000999 return res;
1000}
1001
1002static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001003_buffered_readline(buffered *self, Py_ssize_t limit)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001004{
1005 PyObject *res = NULL;
1006 PyObject *chunks = NULL;
1007 Py_ssize_t n, written = 0;
1008 const char *start, *s, *end;
1009
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001010 CHECK_CLOSED(self, "readline of closed file")
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001011
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001012 /* First, try to find a line in the buffer. This can run unlocked because
1013 the calls to the C API are simple enough that they can't trigger
1014 any thread switch. */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001015 n = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
1016 if (limit >= 0 && n > limit)
1017 n = limit;
1018 start = self->buffer + self->pos;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001019 s = memchr(start, '\n', n);
1020 if (s != NULL) {
1021 res = PyBytes_FromStringAndSize(start, s - start + 1);
1022 if (res != NULL)
1023 self->pos += s - start + 1;
1024 goto end_unlocked;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001025 }
1026 if (n == limit) {
1027 res = PyBytes_FromStringAndSize(start, n);
1028 if (res != NULL)
1029 self->pos += n;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001030 goto end_unlocked;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001031 }
1032
Antoine Pitrouf3b68b32010-12-03 18:41:39 +00001033 if (!ENTER_BUFFERED(self))
1034 goto end_unlocked;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001035
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001036 /* Now we try to get some more from the raw stream */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001037 chunks = PyList_New(0);
1038 if (chunks == NULL)
1039 goto end;
1040 if (n > 0) {
1041 res = PyBytes_FromStringAndSize(start, n);
1042 if (res == NULL)
1043 goto end;
1044 if (PyList_Append(chunks, res) < 0) {
1045 Py_CLEAR(res);
1046 goto end;
1047 }
1048 Py_CLEAR(res);
1049 written += n;
Antoine Pitroue05565e2011-08-20 14:39:23 +02001050 self->pos += n;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001051 if (limit >= 0)
1052 limit -= n;
1053 }
Antoine Pitroue05565e2011-08-20 14:39:23 +02001054 if (self->writable) {
1055 PyObject *r = buffered_flush_and_rewind_unlocked(self);
1056 if (r == NULL)
1057 goto end;
1058 Py_DECREF(r);
1059 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001060
1061 for (;;) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001062 _bufferedreader_reset_buf(self);
1063 n = _bufferedreader_fill_buffer(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001064 if (n == -1)
1065 goto end;
1066 if (n <= 0)
1067 break;
1068 if (limit >= 0 && n > limit)
1069 n = limit;
1070 start = self->buffer;
1071 end = start + n;
1072 s = start;
1073 while (s < end) {
1074 if (*s++ == '\n') {
1075 res = PyBytes_FromStringAndSize(start, s - start);
1076 if (res == NULL)
1077 goto end;
1078 self->pos = s - start;
1079 goto found;
1080 }
1081 }
1082 res = PyBytes_FromStringAndSize(start, n);
1083 if (res == NULL)
1084 goto end;
1085 if (n == limit) {
1086 self->pos = n;
1087 break;
1088 }
1089 if (PyList_Append(chunks, res) < 0) {
1090 Py_CLEAR(res);
1091 goto end;
1092 }
1093 Py_CLEAR(res);
1094 written += n;
1095 if (limit >= 0)
1096 limit -= n;
1097 }
1098found:
1099 if (res != NULL && PyList_Append(chunks, res) < 0) {
1100 Py_CLEAR(res);
1101 goto end;
1102 }
1103 Py_CLEAR(res);
1104 res = _PyBytes_Join(_PyIO_empty_bytes, chunks);
1105
1106end:
1107 LEAVE_BUFFERED(self)
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001108end_unlocked:
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001109 Py_XDECREF(chunks);
1110 return res;
1111}
1112
1113static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001114buffered_readline(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001115{
1116 Py_ssize_t limit = -1;
1117
1118 CHECK_INITIALIZED(self)
Benjamin Petersonbf5ff762009-12-13 19:25:34 +00001119 if (!PyArg_ParseTuple(args, "|O&:readline", &_PyIO_ConvertSsize_t, &limit))
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001120 return NULL;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001121 return _buffered_readline(self, limit);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001122}
1123
1124
1125static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001126buffered_tell(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001127{
1128 Py_off_t pos;
1129
1130 CHECK_INITIALIZED(self)
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001131 pos = _buffered_raw_tell(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001132 if (pos == -1)
1133 return NULL;
1134 pos -= RAW_OFFSET(self);
1135 /* TODO: sanity check (pos >= 0) */
1136 return PyLong_FromOff_t(pos);
1137}
1138
1139static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001140buffered_seek(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001141{
1142 Py_off_t target, n;
1143 int whence = 0;
1144 PyObject *targetobj, *res = NULL;
1145
1146 CHECK_INITIALIZED(self)
1147 if (!PyArg_ParseTuple(args, "O|i:seek", &targetobj, &whence)) {
1148 return NULL;
1149 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001150 if (whence < 0 || whence > 2) {
1151 PyErr_Format(PyExc_ValueError,
1152 "whence must be between 0 and 2, not %d", whence);
1153 return NULL;
1154 }
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001155
1156 CHECK_CLOSED(self, "seek of closed file")
1157
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001158 target = PyNumber_AsOff_t(targetobj, PyExc_ValueError);
1159 if (target == -1 && PyErr_Occurred())
1160 return NULL;
1161
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001162 if (whence != 2 && self->readable) {
1163 Py_off_t current, avail;
1164 /* Check if seeking leaves us inside the current buffer,
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001165 so as to return quickly if possible. Also, we needn't take the
1166 lock in this fast path.
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001167 Don't know how to do that when whence == 2, though. */
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001168 /* NOTE: RAW_TELL() can release the GIL but the object is in a stable
1169 state at this point. */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001170 current = RAW_TELL(self);
1171 avail = READAHEAD(self);
1172 if (avail > 0) {
1173 Py_off_t offset;
1174 if (whence == 0)
1175 offset = target - (current - RAW_OFFSET(self));
1176 else
1177 offset = target;
1178 if (offset >= -self->pos && offset <= avail) {
1179 self->pos += offset;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001180 return PyLong_FromOff_t(current - avail + offset);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001181 }
1182 }
1183 }
1184
Antoine Pitrouf3b68b32010-12-03 18:41:39 +00001185 if (!ENTER_BUFFERED(self))
1186 return NULL;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001187
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001188 /* Fallback: invoke raw seek() method and clear buffer */
1189 if (self->writable) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001190 res = _bufferedwriter_flush_unlocked(self, 0);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001191 if (res == NULL)
1192 goto end;
1193 Py_CLEAR(res);
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001194 _bufferedwriter_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001195 }
1196
1197 /* TODO: align on block boundary and read buffer if needed? */
1198 if (whence == 1)
1199 target -= RAW_OFFSET(self);
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001200 n = _buffered_raw_seek(self, target, whence);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001201 if (n == -1)
1202 goto end;
1203 self->raw_pos = -1;
1204 res = PyLong_FromOff_t(n);
1205 if (res != NULL && self->readable)
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001206 _bufferedreader_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001207
1208end:
1209 LEAVE_BUFFERED(self)
1210 return res;
1211}
1212
1213static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001214buffered_truncate(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001215{
1216 PyObject *pos = Py_None;
1217 PyObject *res = NULL;
1218
1219 CHECK_INITIALIZED(self)
1220 if (!PyArg_ParseTuple(args, "|O:truncate", &pos)) {
1221 return NULL;
1222 }
1223
Antoine Pitrouf3b68b32010-12-03 18:41:39 +00001224 if (!ENTER_BUFFERED(self))
1225 return NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001226
1227 if (self->writable) {
Antoine Pitroue05565e2011-08-20 14:39:23 +02001228 res = buffered_flush_and_rewind_unlocked(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001229 if (res == NULL)
1230 goto end;
1231 Py_CLEAR(res);
1232 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001233 res = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_truncate, pos, NULL);
1234 if (res == NULL)
1235 goto end;
1236 /* Reset cached position */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001237 if (_buffered_raw_tell(self) == -1)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001238 PyErr_Clear();
1239
1240end:
1241 LEAVE_BUFFERED(self)
1242 return res;
1243}
1244
1245static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001246buffered_iternext(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001247{
1248 PyObject *line;
1249 PyTypeObject *tp;
1250
1251 CHECK_INITIALIZED(self);
1252
1253 tp = Py_TYPE(self);
1254 if (tp == &PyBufferedReader_Type ||
1255 tp == &PyBufferedRandom_Type) {
1256 /* Skip method call overhead for speed */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001257 line = _buffered_readline(self, -1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001258 }
1259 else {
1260 line = PyObject_CallMethodObjArgs((PyObject *)self,
1261 _PyIO_str_readline, NULL);
1262 if (line && !PyBytes_Check(line)) {
1263 PyErr_Format(PyExc_IOError,
1264 "readline() should have returned a bytes object, "
1265 "not '%.200s'", Py_TYPE(line)->tp_name);
1266 Py_DECREF(line);
1267 return NULL;
1268 }
1269 }
1270
1271 if (line == NULL)
1272 return NULL;
1273
1274 if (PyBytes_GET_SIZE(line) == 0) {
1275 /* Reached EOF or would have blocked */
1276 Py_DECREF(line);
1277 return NULL;
1278 }
1279
1280 return line;
1281}
1282
Antoine Pitrou716c4442009-05-23 19:04:03 +00001283static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001284buffered_repr(buffered *self)
Antoine Pitrou716c4442009-05-23 19:04:03 +00001285{
1286 PyObject *nameobj, *res;
1287
1288 nameobj = PyObject_GetAttrString((PyObject *) self, "name");
1289 if (nameobj == NULL) {
1290 if (PyErr_ExceptionMatches(PyExc_AttributeError))
1291 PyErr_Clear();
1292 else
1293 return NULL;
1294 res = PyUnicode_FromFormat("<%s>", Py_TYPE(self)->tp_name);
1295 }
1296 else {
1297 res = PyUnicode_FromFormat("<%s name=%R>",
1298 Py_TYPE(self)->tp_name, nameobj);
1299 Py_DECREF(nameobj);
1300 }
1301 return res;
1302}
1303
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001304/*
1305 * class BufferedReader
1306 */
1307
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001308PyDoc_STRVAR(bufferedreader_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001309 "Create a new buffered reader using the given readable raw IO object.");
1310
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001311static void _bufferedreader_reset_buf(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001312{
1313 self->read_end = -1;
1314}
1315
1316static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001317bufferedreader_init(buffered *self, PyObject *args, PyObject *kwds)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001318{
1319 char *kwlist[] = {"raw", "buffer_size", NULL};
1320 Py_ssize_t buffer_size = DEFAULT_BUFFER_SIZE;
1321 PyObject *raw;
1322
1323 self->ok = 0;
Benjamin Petersond2e0c792009-05-01 20:40:59 +00001324 self->detached = 0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001325
1326 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|n:BufferedReader", kwlist,
1327 &raw, &buffer_size)) {
1328 return -1;
1329 }
1330
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001331 if (_PyIOBase_check_readable(raw, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001332 return -1;
1333
1334 Py_CLEAR(self->raw);
1335 Py_INCREF(raw);
1336 self->raw = raw;
1337 self->buffer_size = buffer_size;
1338 self->readable = 1;
1339 self->writable = 0;
1340
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001341 if (_buffered_init(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001342 return -1;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001343 _bufferedreader_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001344
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001345 self->fast_closed_checks = (Py_TYPE(self) == &PyBufferedReader_Type &&
1346 Py_TYPE(raw) == &PyFileIO_Type);
1347
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001348 self->ok = 1;
1349 return 0;
1350}
1351
1352static Py_ssize_t
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001353_bufferedreader_raw_read(buffered *self, char *start, Py_ssize_t len)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001354{
1355 Py_buffer buf;
1356 PyObject *memobj, *res;
1357 Py_ssize_t n;
1358 /* NOTE: the buffer needn't be released as its object is NULL. */
1359 if (PyBuffer_FillInfo(&buf, NULL, start, len, 0, PyBUF_CONTIG) == -1)
1360 return -1;
1361 memobj = PyMemoryView_FromBuffer(&buf);
1362 if (memobj == NULL)
1363 return -1;
Antoine Pitrou707ce822011-02-25 21:24:11 +00001364 /* NOTE: PyErr_SetFromErrno() calls PyErr_CheckSignals() when EINTR
1365 occurs so we needn't do it ourselves.
1366 We then retry reading, ignoring the signal if no handler has
1367 raised (see issue #10956).
1368 */
1369 do {
1370 res = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_readinto, memobj, NULL);
1371 } while (res == NULL && _trap_eintr());
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001372 Py_DECREF(memobj);
1373 if (res == NULL)
1374 return -1;
1375 if (res == Py_None) {
1376 /* Non-blocking stream would have blocked. Special return code! */
1377 Py_DECREF(res);
1378 return -2;
1379 }
1380 n = PyNumber_AsSsize_t(res, PyExc_ValueError);
1381 Py_DECREF(res);
1382 if (n < 0 || n > len) {
1383 PyErr_Format(PyExc_IOError,
1384 "raw readinto() returned invalid length %zd "
1385 "(should have been between 0 and %zd)", n, len);
1386 return -1;
1387 }
1388 if (n > 0 && self->abs_pos != -1)
1389 self->abs_pos += n;
1390 return n;
1391}
1392
1393static Py_ssize_t
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001394_bufferedreader_fill_buffer(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001395{
1396 Py_ssize_t start, len, n;
1397 if (VALID_READ_BUFFER(self))
1398 start = Py_SAFE_DOWNCAST(self->read_end, Py_off_t, Py_ssize_t);
1399 else
1400 start = 0;
1401 len = self->buffer_size - start;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001402 n = _bufferedreader_raw_read(self, self->buffer + start, len);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001403 if (n <= 0)
1404 return n;
1405 self->read_end = start + n;
1406 self->raw_pos = start + n;
1407 return n;
1408}
1409
1410static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001411_bufferedreader_read_all(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001412{
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001413 Py_ssize_t current_size;
Victor Stinnerb57f1082011-05-26 00:19:38 +02001414 PyObject *res, *data = NULL, *chunk, *chunks;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001415
1416 /* First copy what we have in the current buffer. */
1417 current_size = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
1418 if (current_size) {
1419 data = PyBytes_FromStringAndSize(
1420 self->buffer + self->pos, current_size);
Victor Stinnerb57f1082011-05-26 00:19:38 +02001421 if (data == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001422 return NULL;
Antoine Pitroue05565e2011-08-20 14:39:23 +02001423 self->pos += current_size;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001424 }
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001425 /* We're going past the buffer's bounds, flush it */
1426 if (self->writable) {
Antoine Pitroue05565e2011-08-20 14:39:23 +02001427 res = buffered_flush_and_rewind_unlocked(self);
Victor Stinnerb57f1082011-05-26 00:19:38 +02001428 if (res == NULL)
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001429 return NULL;
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001430 Py_CLEAR(res);
1431 }
Antoine Pitroue05565e2011-08-20 14:39:23 +02001432 _bufferedreader_reset_buf(self);
Victor Stinnerb57f1082011-05-26 00:19:38 +02001433
1434 if (PyObject_HasAttr(self->raw, _PyIO_str_readall)) {
1435 chunk = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_readall, NULL);
1436 if (chunk == NULL)
1437 return NULL;
1438 if (chunk != Py_None && !PyBytes_Check(chunk)) {
1439 Py_XDECREF(data);
1440 Py_DECREF(chunk);
1441 PyErr_SetString(PyExc_TypeError, "readall() should return bytes");
1442 return NULL;
1443 }
1444 if (chunk == Py_None) {
1445 if (current_size == 0)
1446 return chunk;
1447 else {
1448 Py_DECREF(chunk);
1449 return data;
1450 }
1451 }
1452 else if (current_size) {
1453 PyBytes_Concat(&data, chunk);
1454 Py_DECREF(chunk);
1455 if (data == NULL)
1456 return NULL;
1457 return data;
1458 } else
1459 return chunk;
1460 }
1461
1462 chunks = PyList_New(0);
1463 if (chunks == NULL)
1464 return NULL;
1465
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001466 while (1) {
1467 if (data) {
1468 if (PyList_Append(chunks, data) < 0) {
1469 Py_DECREF(data);
1470 Py_DECREF(chunks);
1471 return NULL;
1472 }
1473 Py_DECREF(data);
1474 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001475
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001476 /* Read until EOF or until read() would block. */
1477 data = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_read, NULL);
1478 if (data == NULL) {
1479 Py_DECREF(chunks);
1480 return NULL;
1481 }
1482 if (data != Py_None && !PyBytes_Check(data)) {
1483 Py_DECREF(data);
1484 Py_DECREF(chunks);
1485 PyErr_SetString(PyExc_TypeError, "read() should return bytes");
1486 return NULL;
1487 }
1488 if (data == Py_None || PyBytes_GET_SIZE(data) == 0) {
1489 if (current_size == 0) {
1490 Py_DECREF(chunks);
1491 return data;
1492 }
1493 else {
1494 res = _PyBytes_Join(_PyIO_empty_bytes, chunks);
1495 Py_DECREF(data);
1496 Py_DECREF(chunks);
1497 return res;
1498 }
1499 }
1500 current_size += PyBytes_GET_SIZE(data);
1501 if (self->abs_pos != -1)
1502 self->abs_pos += PyBytes_GET_SIZE(data);
1503 }
1504}
1505
1506/* Read n bytes from the buffer if it can, otherwise return None.
1507 This function is simple enough that it can run unlocked. */
1508static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001509_bufferedreader_read_fast(buffered *self, Py_ssize_t n)
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001510{
1511 Py_ssize_t current_size;
1512
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001513 current_size = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
1514 if (n <= current_size) {
1515 /* Fast path: the data to read is fully buffered. */
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001516 PyObject *res = PyBytes_FromStringAndSize(self->buffer + self->pos, n);
1517 if (res != NULL)
1518 self->pos += n;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001519 return res;
1520 }
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001521 Py_RETURN_NONE;
1522}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001523
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001524/* Generic read function: read from the stream until enough bytes are read,
1525 * or until an EOF occurs or until read() would block.
1526 */
1527static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001528_bufferedreader_read_generic(buffered *self, Py_ssize_t n)
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001529{
1530 PyObject *res = NULL;
1531 Py_ssize_t current_size, remaining, written;
1532 char *out;
1533
1534 current_size = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
1535 if (n <= current_size)
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001536 return _bufferedreader_read_fast(self, n);
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001537
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001538 res = PyBytes_FromStringAndSize(NULL, n);
1539 if (res == NULL)
1540 goto error;
1541 out = PyBytes_AS_STRING(res);
1542 remaining = n;
1543 written = 0;
1544 if (current_size > 0) {
1545 memcpy(out, self->buffer + self->pos, current_size);
1546 remaining -= current_size;
1547 written += current_size;
Antoine Pitroue05565e2011-08-20 14:39:23 +02001548 self->pos += current_size;
1549 }
1550 /* Flush the write buffer if necessary */
1551 if (self->writable) {
1552 PyObject *r = buffered_flush_and_rewind_unlocked(self);
1553 if (r == NULL)
1554 goto error;
1555 Py_DECREF(r);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001556 }
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001557 _bufferedreader_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001558 while (remaining > 0) {
1559 /* We want to read a whole block at the end into buffer.
1560 If we had readv() we could do this in one pass. */
1561 Py_ssize_t r = MINUS_LAST_BLOCK(self, remaining);
1562 if (r == 0)
1563 break;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001564 r = _bufferedreader_raw_read(self, out + written, r);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001565 if (r == -1)
1566 goto error;
1567 if (r == 0 || r == -2) {
1568 /* EOF occurred or read() would block. */
1569 if (r == 0 || written > 0) {
1570 if (_PyBytes_Resize(&res, written))
1571 goto error;
1572 return res;
1573 }
1574 Py_DECREF(res);
1575 Py_INCREF(Py_None);
1576 return Py_None;
1577 }
1578 remaining -= r;
1579 written += r;
1580 }
1581 assert(remaining <= self->buffer_size);
1582 self->pos = 0;
1583 self->raw_pos = 0;
1584 self->read_end = 0;
Antoine Pitrou32cfede2010-08-11 13:31:33 +00001585 /* NOTE: when the read is satisfied, we avoid issuing any additional
1586 reads, which could block indefinitely (e.g. on a socket).
1587 See issue #9550. */
1588 while (remaining > 0 && self->read_end < self->buffer_size) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001589 Py_ssize_t r = _bufferedreader_fill_buffer(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001590 if (r == -1)
1591 goto error;
1592 if (r == 0 || r == -2) {
1593 /* EOF occurred or read() would block. */
1594 if (r == 0 || written > 0) {
1595 if (_PyBytes_Resize(&res, written))
1596 goto error;
1597 return res;
1598 }
1599 Py_DECREF(res);
1600 Py_INCREF(Py_None);
1601 return Py_None;
1602 }
1603 if (remaining > r) {
1604 memcpy(out + written, self->buffer + self->pos, r);
1605 written += r;
1606 self->pos += r;
1607 remaining -= r;
1608 }
1609 else if (remaining > 0) {
1610 memcpy(out + written, self->buffer + self->pos, remaining);
1611 written += remaining;
1612 self->pos += remaining;
1613 remaining = 0;
1614 }
1615 if (remaining == 0)
1616 break;
1617 }
1618
1619 return res;
1620
1621error:
1622 Py_XDECREF(res);
1623 return NULL;
1624}
1625
1626static PyObject *
Victor Stinnerbc93a112011-06-01 00:01:24 +02001627_bufferedreader_peek_unlocked(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001628{
1629 Py_ssize_t have, r;
1630
1631 have = Py_SAFE_DOWNCAST(READAHEAD(self), Py_off_t, Py_ssize_t);
1632 /* Constraints:
1633 1. we don't want to advance the file position.
1634 2. we don't want to lose block alignment, so we can't shift the buffer
1635 to make some place.
1636 Therefore, we either return `have` bytes (if > 0), or a full buffer.
1637 */
1638 if (have > 0) {
1639 return PyBytes_FromStringAndSize(self->buffer + self->pos, have);
1640 }
1641
1642 /* Fill the buffer from the raw stream, and copy it to the result. */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001643 _bufferedreader_reset_buf(self);
1644 r = _bufferedreader_fill_buffer(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001645 if (r == -1)
1646 return NULL;
1647 if (r == -2)
1648 r = 0;
1649 self->pos = 0;
1650 return PyBytes_FromStringAndSize(self->buffer, r);
1651}
1652
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001653static PyMethodDef bufferedreader_methods[] = {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001654 /* BufferedIOMixin methods */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001655 {"detach", (PyCFunction)buffered_detach, METH_NOARGS},
1656 {"flush", (PyCFunction)buffered_simple_flush, METH_NOARGS},
1657 {"close", (PyCFunction)buffered_close, METH_NOARGS},
1658 {"seekable", (PyCFunction)buffered_seekable, METH_NOARGS},
1659 {"readable", (PyCFunction)buffered_readable, METH_NOARGS},
1660 {"writable", (PyCFunction)buffered_writable, METH_NOARGS},
1661 {"fileno", (PyCFunction)buffered_fileno, METH_NOARGS},
1662 {"isatty", (PyCFunction)buffered_isatty, METH_NOARGS},
Antoine Pitroue033e062010-10-29 10:38:18 +00001663 {"_dealloc_warn", (PyCFunction)buffered_dealloc_warn, METH_O},
Antoine Pitrou243757e2010-11-05 21:15:39 +00001664 {"__getstate__", (PyCFunction)buffered_getstate, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001665
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001666 {"read", (PyCFunction)buffered_read, METH_VARARGS},
1667 {"peek", (PyCFunction)buffered_peek, METH_VARARGS},
1668 {"read1", (PyCFunction)buffered_read1, METH_VARARGS},
Antoine Pitrou3486a982011-05-12 01:57:53 +02001669 {"readinto", (PyCFunction)buffered_readinto, METH_VARARGS},
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001670 {"readline", (PyCFunction)buffered_readline, METH_VARARGS},
1671 {"seek", (PyCFunction)buffered_seek, METH_VARARGS},
1672 {"tell", (PyCFunction)buffered_tell, METH_NOARGS},
1673 {"truncate", (PyCFunction)buffered_truncate, METH_VARARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001674 {NULL, NULL}
1675};
1676
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001677static PyMemberDef bufferedreader_members[] = {
Antoine Pitrou7f8f4182010-12-21 21:20:59 +00001678 {"raw", T_OBJECT, offsetof(buffered, raw), READONLY},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001679 {NULL}
1680};
1681
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001682static PyGetSetDef bufferedreader_getset[] = {
1683 {"closed", (getter)buffered_closed_get, NULL, NULL},
1684 {"name", (getter)buffered_name_get, NULL, NULL},
1685 {"mode", (getter)buffered_mode_get, NULL, NULL},
Benjamin Peterson1fea3212009-04-19 03:15:20 +00001686 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001687};
1688
1689
1690PyTypeObject PyBufferedReader_Type = {
1691 PyVarObject_HEAD_INIT(NULL, 0)
1692 "_io.BufferedReader", /*tp_name*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001693 sizeof(buffered), /*tp_basicsize*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001694 0, /*tp_itemsize*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001695 (destructor)buffered_dealloc, /*tp_dealloc*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001696 0, /*tp_print*/
1697 0, /*tp_getattr*/
1698 0, /*tp_setattr*/
1699 0, /*tp_compare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001700 (reprfunc)buffered_repr, /*tp_repr*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001701 0, /*tp_as_number*/
1702 0, /*tp_as_sequence*/
1703 0, /*tp_as_mapping*/
1704 0, /*tp_hash */
1705 0, /*tp_call*/
1706 0, /*tp_str*/
1707 0, /*tp_getattro*/
1708 0, /*tp_setattro*/
1709 0, /*tp_as_buffer*/
1710 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE
1711 | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001712 bufferedreader_doc, /* tp_doc */
1713 (traverseproc)buffered_traverse, /* tp_traverse */
1714 (inquiry)buffered_clear, /* tp_clear */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001715 0, /* tp_richcompare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001716 offsetof(buffered, weakreflist), /*tp_weaklistoffset*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001717 0, /* tp_iter */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001718 (iternextfunc)buffered_iternext, /* tp_iternext */
1719 bufferedreader_methods, /* tp_methods */
1720 bufferedreader_members, /* tp_members */
1721 bufferedreader_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001722 0, /* tp_base */
1723 0, /* tp_dict */
1724 0, /* tp_descr_get */
1725 0, /* tp_descr_set */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001726 offsetof(buffered, dict), /* tp_dictoffset */
1727 (initproc)bufferedreader_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001728 0, /* tp_alloc */
1729 PyType_GenericNew, /* tp_new */
1730};
1731
1732
Benjamin Peterson59406a92009-03-26 17:10:29 +00001733
1734static int
1735complain_about_max_buffer_size(void)
1736{
1737 if (PyErr_WarnEx(PyExc_DeprecationWarning,
1738 "max_buffer_size is deprecated", 1) < 0)
1739 return 0;
1740 return 1;
1741}
1742
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001743/*
1744 * class BufferedWriter
1745 */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001746PyDoc_STRVAR(bufferedwriter_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001747 "A buffer for a writeable sequential RawIO object.\n"
1748 "\n"
1749 "The constructor creates a BufferedWriter for the given writeable raw\n"
1750 "stream. If the buffer_size is not given, it defaults to\n"
1751 "DEFAULT_BUFFER_SIZE. max_buffer_size isn't used anymore.\n"
1752 );
1753
1754static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001755_bufferedwriter_reset_buf(buffered *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001756{
1757 self->write_pos = 0;
1758 self->write_end = -1;
1759}
1760
1761static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001762bufferedwriter_init(buffered *self, PyObject *args, PyObject *kwds)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001763{
1764 /* TODO: properly deprecate max_buffer_size */
1765 char *kwlist[] = {"raw", "buffer_size", "max_buffer_size", NULL};
1766 Py_ssize_t buffer_size = DEFAULT_BUFFER_SIZE;
Benjamin Peterson59406a92009-03-26 17:10:29 +00001767 Py_ssize_t max_buffer_size = -234;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001768 PyObject *raw;
1769
1770 self->ok = 0;
Benjamin Petersond2e0c792009-05-01 20:40:59 +00001771 self->detached = 0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001772
1773 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|nn:BufferedReader", kwlist,
1774 &raw, &buffer_size, &max_buffer_size)) {
1775 return -1;
1776 }
1777
Benjamin Peterson59406a92009-03-26 17:10:29 +00001778 if (max_buffer_size != -234 && !complain_about_max_buffer_size())
1779 return -1;
1780
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001781 if (_PyIOBase_check_writable(raw, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001782 return -1;
1783
1784 Py_CLEAR(self->raw);
1785 Py_INCREF(raw);
1786 self->raw = raw;
1787 self->readable = 0;
1788 self->writable = 1;
1789
1790 self->buffer_size = buffer_size;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001791 if (_buffered_init(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001792 return -1;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001793 _bufferedwriter_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001794 self->pos = 0;
1795
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001796 self->fast_closed_checks = (Py_TYPE(self) == &PyBufferedWriter_Type &&
1797 Py_TYPE(raw) == &PyFileIO_Type);
1798
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001799 self->ok = 1;
1800 return 0;
1801}
1802
1803static Py_ssize_t
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001804_bufferedwriter_raw_write(buffered *self, char *start, Py_ssize_t len)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001805{
1806 Py_buffer buf;
1807 PyObject *memobj, *res;
1808 Py_ssize_t n;
1809 /* NOTE: the buffer needn't be released as its object is NULL. */
1810 if (PyBuffer_FillInfo(&buf, NULL, start, len, 1, PyBUF_CONTIG_RO) == -1)
1811 return -1;
1812 memobj = PyMemoryView_FromBuffer(&buf);
1813 if (memobj == NULL)
1814 return -1;
Antoine Pitrou707ce822011-02-25 21:24:11 +00001815 /* NOTE: PyErr_SetFromErrno() calls PyErr_CheckSignals() when EINTR
1816 occurs so we needn't do it ourselves.
1817 We then retry writing, ignoring the signal if no handler has
1818 raised (see issue #10956).
1819 */
1820 do {
1821 res = PyObject_CallMethodObjArgs(self->raw, _PyIO_str_write, memobj, NULL);
1822 } while (res == NULL && _trap_eintr());
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001823 Py_DECREF(memobj);
1824 if (res == NULL)
1825 return -1;
1826 n = PyNumber_AsSsize_t(res, PyExc_ValueError);
1827 Py_DECREF(res);
1828 if (n < 0 || n > len) {
1829 PyErr_Format(PyExc_IOError,
1830 "raw write() returned invalid length %zd "
1831 "(should have been between 0 and %zd)", n, len);
1832 return -1;
1833 }
1834 if (n > 0 && self->abs_pos != -1)
1835 self->abs_pos += n;
1836 return n;
1837}
1838
1839/* `restore_pos` is 1 if we need to restore the raw stream position at
1840 the end, 0 otherwise. */
1841static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001842_bufferedwriter_flush_unlocked(buffered *self, int restore_pos)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001843{
1844 Py_ssize_t written = 0;
1845 Py_off_t n, rewind;
1846
1847 if (!VALID_WRITE_BUFFER(self) || self->write_pos == self->write_end)
1848 goto end;
1849 /* First, rewind */
1850 rewind = RAW_OFFSET(self) + (self->pos - self->write_pos);
1851 if (rewind != 0) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001852 n = _buffered_raw_seek(self, -rewind, 1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001853 if (n < 0) {
1854 goto error;
1855 }
1856 self->raw_pos -= rewind;
1857 }
1858 while (self->write_pos < self->write_end) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001859 n = _bufferedwriter_raw_write(self,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001860 self->buffer + self->write_pos,
1861 Py_SAFE_DOWNCAST(self->write_end - self->write_pos,
1862 Py_off_t, Py_ssize_t));
1863 if (n == -1) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001864 Py_ssize_t *w = _buffered_check_blocking_error();
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001865 if (w == NULL)
1866 goto error;
1867 self->write_pos += *w;
1868 self->raw_pos = self->write_pos;
1869 written += *w;
1870 *w = written;
1871 /* Already re-raised */
1872 goto error;
1873 }
1874 self->write_pos += n;
1875 self->raw_pos = self->write_pos;
1876 written += Py_SAFE_DOWNCAST(n, Py_off_t, Py_ssize_t);
Antoine Pitroub46b9d52010-08-21 19:09:32 +00001877 /* Partial writes can return successfully when interrupted by a
1878 signal (see write(2)). We must run signal handlers before
1879 blocking another time, possibly indefinitely. */
1880 if (PyErr_CheckSignals() < 0)
1881 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001882 }
1883
1884 if (restore_pos) {
1885 Py_off_t forward = rewind - written;
1886 if (forward != 0) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001887 n = _buffered_raw_seek(self, forward, 1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001888 if (n < 0) {
1889 goto error;
1890 }
1891 self->raw_pos += forward;
1892 }
1893 }
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001894 _bufferedwriter_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001895
1896end:
1897 Py_RETURN_NONE;
1898
1899error:
1900 return NULL;
1901}
1902
1903static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001904bufferedwriter_write(buffered *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001905{
1906 PyObject *res = NULL;
1907 Py_buffer buf;
Amaury Forgeot d'Arce1b60d42009-10-05 21:09:00 +00001908 Py_ssize_t written, avail, remaining;
1909 Py_off_t offset;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001910
1911 CHECK_INITIALIZED(self)
1912 if (!PyArg_ParseTuple(args, "y*:write", &buf)) {
1913 return NULL;
1914 }
1915
Antoine Pitrou711af3a2009-04-11 15:39:24 +00001916 if (IS_CLOSED(self)) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001917 PyErr_SetString(PyExc_ValueError, "write to closed file");
1918 PyBuffer_Release(&buf);
1919 return NULL;
1920 }
1921
Antoine Pitrouf3b68b32010-12-03 18:41:39 +00001922 if (!ENTER_BUFFERED(self)) {
1923 PyBuffer_Release(&buf);
1924 return NULL;
1925 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001926
1927 /* Fast path: the data to write can be fully buffered. */
1928 if (!VALID_READ_BUFFER(self) && !VALID_WRITE_BUFFER(self)) {
1929 self->pos = 0;
1930 self->raw_pos = 0;
1931 }
1932 avail = Py_SAFE_DOWNCAST(self->buffer_size - self->pos, Py_off_t, Py_ssize_t);
1933 if (buf.len <= avail) {
1934 memcpy(self->buffer + self->pos, buf.buf, buf.len);
Antoine Pitrou7c404892011-05-13 00:13:33 +02001935 if (!VALID_WRITE_BUFFER(self) || self->write_pos > self->pos) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001936 self->write_pos = self->pos;
1937 }
1938 ADJUST_POSITION(self, self->pos + buf.len);
1939 if (self->pos > self->write_end)
1940 self->write_end = self->pos;
1941 written = buf.len;
1942 goto end;
1943 }
1944
1945 /* First write the current buffer */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001946 res = _bufferedwriter_flush_unlocked(self, 0);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001947 if (res == NULL) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001948 Py_ssize_t *w = _buffered_check_blocking_error();
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001949 if (w == NULL)
1950 goto error;
1951 if (self->readable)
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001952 _bufferedreader_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001953 /* Make some place by shifting the buffer. */
1954 assert(VALID_WRITE_BUFFER(self));
1955 memmove(self->buffer, self->buffer + self->write_pos,
1956 Py_SAFE_DOWNCAST(self->write_end - self->write_pos,
1957 Py_off_t, Py_ssize_t));
1958 self->write_end -= self->write_pos;
1959 self->raw_pos -= self->write_pos;
1960 self->pos -= self->write_pos;
1961 self->write_pos = 0;
1962 avail = Py_SAFE_DOWNCAST(self->buffer_size - self->write_end,
1963 Py_off_t, Py_ssize_t);
1964 if (buf.len <= avail) {
1965 /* Everything can be buffered */
1966 PyErr_Clear();
1967 memcpy(self->buffer + self->write_end, buf.buf, buf.len);
1968 self->write_end += buf.len;
1969 written = buf.len;
1970 goto end;
1971 }
1972 /* Buffer as much as possible. */
1973 memcpy(self->buffer + self->write_end, buf.buf, avail);
1974 self->write_end += avail;
1975 /* Already re-raised */
1976 *w = avail;
1977 goto error;
1978 }
1979 Py_CLEAR(res);
1980
Antoine Pitroua0ceb732009-08-06 20:29:56 +00001981 /* Adjust the raw stream position if it is away from the logical stream
1982 position. This happens if the read buffer has been filled but not
1983 modified (and therefore _bufferedwriter_flush_unlocked() didn't rewind
1984 the raw stream by itself).
1985 Fixes issue #6629.
1986 */
Amaury Forgeot d'Arce1b60d42009-10-05 21:09:00 +00001987 offset = RAW_OFFSET(self);
1988 if (offset != 0) {
1989 if (_buffered_raw_seek(self, -offset, 1) < 0)
Antoine Pitroua0ceb732009-08-06 20:29:56 +00001990 goto error;
Amaury Forgeot d'Arce1b60d42009-10-05 21:09:00 +00001991 self->raw_pos -= offset;
Antoine Pitroua0ceb732009-08-06 20:29:56 +00001992 }
1993
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001994 /* Then write buf itself. At this point the buffer has been emptied. */
1995 remaining = buf.len;
1996 written = 0;
1997 while (remaining > self->buffer_size) {
Amaury Forgeot d'Arce1b60d42009-10-05 21:09:00 +00001998 Py_ssize_t n = _bufferedwriter_raw_write(
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001999 self, (char *) buf.buf + written, buf.len - written);
2000 if (n == -1) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002001 Py_ssize_t *w = _buffered_check_blocking_error();
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002002 if (w == NULL)
2003 goto error;
2004 written += *w;
2005 remaining -= *w;
2006 if (remaining > self->buffer_size) {
2007 /* Can't buffer everything, still buffer as much as possible */
2008 memcpy(self->buffer,
2009 (char *) buf.buf + written, self->buffer_size);
2010 self->raw_pos = 0;
2011 ADJUST_POSITION(self, self->buffer_size);
2012 self->write_end = self->buffer_size;
2013 *w = written + self->buffer_size;
2014 /* Already re-raised */
2015 goto error;
2016 }
2017 PyErr_Clear();
2018 break;
2019 }
2020 written += n;
2021 remaining -= n;
Antoine Pitroub46b9d52010-08-21 19:09:32 +00002022 /* Partial writes can return successfully when interrupted by a
2023 signal (see write(2)). We must run signal handlers before
2024 blocking another time, possibly indefinitely. */
2025 if (PyErr_CheckSignals() < 0)
2026 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002027 }
2028 if (self->readable)
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002029 _bufferedreader_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002030 if (remaining > 0) {
2031 memcpy(self->buffer, (char *) buf.buf + written, remaining);
2032 written += remaining;
2033 }
2034 self->write_pos = 0;
2035 /* TODO: sanity check (remaining >= 0) */
2036 self->write_end = remaining;
2037 ADJUST_POSITION(self, remaining);
2038 self->raw_pos = 0;
2039
2040end:
2041 res = PyLong_FromSsize_t(written);
2042
2043error:
2044 LEAVE_BUFFERED(self)
2045 PyBuffer_Release(&buf);
2046 return res;
2047}
2048
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002049static PyMethodDef bufferedwriter_methods[] = {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002050 /* BufferedIOMixin methods */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002051 {"close", (PyCFunction)buffered_close, METH_NOARGS},
2052 {"detach", (PyCFunction)buffered_detach, METH_NOARGS},
2053 {"seekable", (PyCFunction)buffered_seekable, METH_NOARGS},
2054 {"readable", (PyCFunction)buffered_readable, METH_NOARGS},
2055 {"writable", (PyCFunction)buffered_writable, METH_NOARGS},
2056 {"fileno", (PyCFunction)buffered_fileno, METH_NOARGS},
2057 {"isatty", (PyCFunction)buffered_isatty, METH_NOARGS},
Antoine Pitroue033e062010-10-29 10:38:18 +00002058 {"_dealloc_warn", (PyCFunction)buffered_dealloc_warn, METH_O},
Antoine Pitrou243757e2010-11-05 21:15:39 +00002059 {"__getstate__", (PyCFunction)buffered_getstate, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002060
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002061 {"write", (PyCFunction)bufferedwriter_write, METH_VARARGS},
2062 {"truncate", (PyCFunction)buffered_truncate, METH_VARARGS},
2063 {"flush", (PyCFunction)buffered_flush, METH_NOARGS},
2064 {"seek", (PyCFunction)buffered_seek, METH_VARARGS},
2065 {"tell", (PyCFunction)buffered_tell, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002066 {NULL, NULL}
2067};
2068
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002069static PyMemberDef bufferedwriter_members[] = {
Antoine Pitrou7f8f4182010-12-21 21:20:59 +00002070 {"raw", T_OBJECT, offsetof(buffered, raw), READONLY},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002071 {NULL}
2072};
2073
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002074static PyGetSetDef bufferedwriter_getset[] = {
2075 {"closed", (getter)buffered_closed_get, NULL, NULL},
2076 {"name", (getter)buffered_name_get, NULL, NULL},
2077 {"mode", (getter)buffered_mode_get, NULL, NULL},
Benjamin Peterson1fea3212009-04-19 03:15:20 +00002078 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002079};
2080
2081
2082PyTypeObject PyBufferedWriter_Type = {
2083 PyVarObject_HEAD_INIT(NULL, 0)
2084 "_io.BufferedWriter", /*tp_name*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002085 sizeof(buffered), /*tp_basicsize*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002086 0, /*tp_itemsize*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002087 (destructor)buffered_dealloc, /*tp_dealloc*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002088 0, /*tp_print*/
2089 0, /*tp_getattr*/
2090 0, /*tp_setattr*/
2091 0, /*tp_compare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002092 (reprfunc)buffered_repr, /*tp_repr*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002093 0, /*tp_as_number*/
2094 0, /*tp_as_sequence*/
2095 0, /*tp_as_mapping*/
2096 0, /*tp_hash */
2097 0, /*tp_call*/
2098 0, /*tp_str*/
2099 0, /*tp_getattro*/
2100 0, /*tp_setattro*/
2101 0, /*tp_as_buffer*/
2102 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE
2103 | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002104 bufferedwriter_doc, /* tp_doc */
2105 (traverseproc)buffered_traverse, /* tp_traverse */
2106 (inquiry)buffered_clear, /* tp_clear */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002107 0, /* tp_richcompare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002108 offsetof(buffered, weakreflist), /*tp_weaklistoffset*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002109 0, /* tp_iter */
2110 0, /* tp_iternext */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002111 bufferedwriter_methods, /* tp_methods */
2112 bufferedwriter_members, /* tp_members */
2113 bufferedwriter_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002114 0, /* tp_base */
2115 0, /* tp_dict */
2116 0, /* tp_descr_get */
2117 0, /* tp_descr_set */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002118 offsetof(buffered, dict), /* tp_dictoffset */
2119 (initproc)bufferedwriter_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002120 0, /* tp_alloc */
2121 PyType_GenericNew, /* tp_new */
2122};
2123
2124
2125
2126/*
2127 * BufferedRWPair
2128 */
2129
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002130PyDoc_STRVAR(bufferedrwpair_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002131 "A buffered reader and writer object together.\n"
2132 "\n"
2133 "A buffered reader object and buffered writer object put together to\n"
2134 "form a sequential IO object that can read and write. This is typically\n"
2135 "used with a socket or two-way pipe.\n"
2136 "\n"
2137 "reader and writer are RawIOBase objects that are readable and\n"
2138 "writeable respectively. If the buffer_size is omitted it defaults to\n"
Benjamin Peterson59406a92009-03-26 17:10:29 +00002139 "DEFAULT_BUFFER_SIZE.\n"
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002140 );
2141
2142/* XXX The usefulness of this (compared to having two separate IO objects) is
2143 * questionable.
2144 */
2145
2146typedef struct {
2147 PyObject_HEAD
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002148 buffered *reader;
2149 buffered *writer;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002150 PyObject *dict;
2151 PyObject *weakreflist;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002152} rwpair;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002153
2154static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002155bufferedrwpair_init(rwpair *self, PyObject *args, PyObject *kwds)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002156{
2157 PyObject *reader, *writer;
2158 Py_ssize_t buffer_size = DEFAULT_BUFFER_SIZE;
Benjamin Peterson59406a92009-03-26 17:10:29 +00002159 Py_ssize_t max_buffer_size = -234;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002160
2161 if (!PyArg_ParseTuple(args, "OO|nn:BufferedRWPair", &reader, &writer,
2162 &buffer_size, &max_buffer_size)) {
2163 return -1;
2164 }
2165
Benjamin Peterson59406a92009-03-26 17:10:29 +00002166 if (max_buffer_size != -234 && !complain_about_max_buffer_size())
2167 return -1;
2168
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002169 if (_PyIOBase_check_readable(reader, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002170 return -1;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002171 if (_PyIOBase_check_writable(writer, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002172 return -1;
2173
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002174 self->reader = (buffered *) PyObject_CallFunction(
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002175 (PyObject *) &PyBufferedReader_Type, "On", reader, buffer_size);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002176 if (self->reader == NULL)
2177 return -1;
2178
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002179 self->writer = (buffered *) PyObject_CallFunction(
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002180 (PyObject *) &PyBufferedWriter_Type, "On", writer, buffer_size);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002181 if (self->writer == NULL) {
2182 Py_CLEAR(self->reader);
2183 return -1;
2184 }
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002185
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002186 return 0;
2187}
2188
2189static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002190bufferedrwpair_traverse(rwpair *self, visitproc visit, void *arg)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002191{
2192 Py_VISIT(self->dict);
2193 return 0;
2194}
2195
2196static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002197bufferedrwpair_clear(rwpair *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002198{
2199 Py_CLEAR(self->reader);
2200 Py_CLEAR(self->writer);
2201 Py_CLEAR(self->dict);
2202 return 0;
2203}
2204
2205static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002206bufferedrwpair_dealloc(rwpair *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002207{
2208 _PyObject_GC_UNTRACK(self);
2209 Py_CLEAR(self->reader);
2210 Py_CLEAR(self->writer);
2211 Py_CLEAR(self->dict);
2212 Py_TYPE(self)->tp_free((PyObject *) self);
2213}
2214
2215static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002216_forward_call(buffered *self, const char *name, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002217{
2218 PyObject *func = PyObject_GetAttrString((PyObject *)self, name);
2219 PyObject *ret;
2220
2221 if (func == NULL) {
2222 PyErr_SetString(PyExc_AttributeError, name);
2223 return NULL;
2224 }
2225
2226 ret = PyObject_CallObject(func, args);
2227 Py_DECREF(func);
2228 return ret;
2229}
2230
2231static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002232bufferedrwpair_read(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002233{
2234 return _forward_call(self->reader, "read", args);
2235}
2236
2237static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002238bufferedrwpair_peek(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002239{
2240 return _forward_call(self->reader, "peek", args);
2241}
2242
2243static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002244bufferedrwpair_read1(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002245{
2246 return _forward_call(self->reader, "read1", args);
2247}
2248
2249static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002250bufferedrwpair_readinto(rwpair *self, PyObject *args)
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002251{
2252 return _forward_call(self->reader, "readinto", args);
2253}
2254
2255static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002256bufferedrwpair_write(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002257{
2258 return _forward_call(self->writer, "write", args);
2259}
2260
2261static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002262bufferedrwpair_flush(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002263{
2264 return _forward_call(self->writer, "flush", args);
2265}
2266
2267static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002268bufferedrwpair_readable(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002269{
2270 return _forward_call(self->reader, "readable", args);
2271}
2272
2273static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002274bufferedrwpair_writable(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002275{
2276 return _forward_call(self->writer, "writable", args);
2277}
2278
2279static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002280bufferedrwpair_close(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002281{
2282 PyObject *ret = _forward_call(self->writer, "close", args);
2283 if (ret == NULL)
2284 return NULL;
2285 Py_DECREF(ret);
2286
2287 return _forward_call(self->reader, "close", args);
2288}
2289
2290static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002291bufferedrwpair_isatty(rwpair *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002292{
2293 PyObject *ret = _forward_call(self->writer, "isatty", args);
2294
2295 if (ret != Py_False) {
2296 /* either True or exception */
2297 return ret;
2298 }
2299 Py_DECREF(ret);
2300
2301 return _forward_call(self->reader, "isatty", args);
2302}
2303
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002304static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002305bufferedrwpair_closed_get(rwpair *self, void *context)
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002306{
2307 return PyObject_GetAttr((PyObject *) self->writer, _PyIO_str_closed);
2308}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002309
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002310static PyMethodDef bufferedrwpair_methods[] = {
2311 {"read", (PyCFunction)bufferedrwpair_read, METH_VARARGS},
2312 {"peek", (PyCFunction)bufferedrwpair_peek, METH_VARARGS},
2313 {"read1", (PyCFunction)bufferedrwpair_read1, METH_VARARGS},
2314 {"readinto", (PyCFunction)bufferedrwpair_readinto, METH_VARARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002315
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002316 {"write", (PyCFunction)bufferedrwpair_write, METH_VARARGS},
2317 {"flush", (PyCFunction)bufferedrwpair_flush, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002318
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002319 {"readable", (PyCFunction)bufferedrwpair_readable, METH_NOARGS},
2320 {"writable", (PyCFunction)bufferedrwpair_writable, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002321
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002322 {"close", (PyCFunction)bufferedrwpair_close, METH_NOARGS},
2323 {"isatty", (PyCFunction)bufferedrwpair_isatty, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002324
Antoine Pitrou243757e2010-11-05 21:15:39 +00002325 {"__getstate__", (PyCFunction)buffered_getstate, METH_NOARGS},
2326
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002327 {NULL, NULL}
2328};
2329
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002330static PyGetSetDef bufferedrwpair_getset[] = {
2331 {"closed", (getter)bufferedrwpair_closed_get, NULL, NULL},
Benjamin Peterson1fea3212009-04-19 03:15:20 +00002332 {NULL}
Antoine Pitroucf4c7492009-04-19 00:09:36 +00002333};
2334
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002335PyTypeObject PyBufferedRWPair_Type = {
2336 PyVarObject_HEAD_INIT(NULL, 0)
2337 "_io.BufferedRWPair", /*tp_name*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002338 sizeof(rwpair), /*tp_basicsize*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002339 0, /*tp_itemsize*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002340 (destructor)bufferedrwpair_dealloc, /*tp_dealloc*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002341 0, /*tp_print*/
2342 0, /*tp_getattr*/
2343 0, /*tp_setattr*/
2344 0, /*tp_compare */
2345 0, /*tp_repr*/
2346 0, /*tp_as_number*/
2347 0, /*tp_as_sequence*/
2348 0, /*tp_as_mapping*/
2349 0, /*tp_hash */
2350 0, /*tp_call*/
2351 0, /*tp_str*/
2352 0, /*tp_getattro*/
2353 0, /*tp_setattro*/
2354 0, /*tp_as_buffer*/
2355 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE
2356 | Py_TPFLAGS_HAVE_GC, /* tp_flags */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002357 bufferedrwpair_doc, /* tp_doc */
2358 (traverseproc)bufferedrwpair_traverse, /* tp_traverse */
2359 (inquiry)bufferedrwpair_clear, /* tp_clear */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002360 0, /* tp_richcompare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002361 offsetof(rwpair, weakreflist), /*tp_weaklistoffset*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002362 0, /* tp_iter */
2363 0, /* tp_iternext */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002364 bufferedrwpair_methods, /* tp_methods */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002365 0, /* tp_members */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002366 bufferedrwpair_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002367 0, /* tp_base */
2368 0, /* tp_dict */
2369 0, /* tp_descr_get */
2370 0, /* tp_descr_set */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002371 offsetof(rwpair, dict), /* tp_dictoffset */
2372 (initproc)bufferedrwpair_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002373 0, /* tp_alloc */
2374 PyType_GenericNew, /* tp_new */
2375};
2376
2377
2378
2379/*
2380 * BufferedRandom
2381 */
2382
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002383PyDoc_STRVAR(bufferedrandom_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002384 "A buffered interface to random access streams.\n"
2385 "\n"
2386 "The constructor creates a reader and writer for a seekable stream,\n"
2387 "raw, given in the first argument. If the buffer_size is omitted it\n"
2388 "defaults to DEFAULT_BUFFER_SIZE. max_buffer_size isn't used anymore.\n"
2389 );
2390
2391static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002392bufferedrandom_init(buffered *self, PyObject *args, PyObject *kwds)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002393{
2394 char *kwlist[] = {"raw", "buffer_size", "max_buffer_size", NULL};
2395 Py_ssize_t buffer_size = DEFAULT_BUFFER_SIZE;
Benjamin Peterson59406a92009-03-26 17:10:29 +00002396 Py_ssize_t max_buffer_size = -234;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002397 PyObject *raw;
2398
2399 self->ok = 0;
Benjamin Petersond2e0c792009-05-01 20:40:59 +00002400 self->detached = 0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002401
2402 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|nn:BufferedReader", kwlist,
2403 &raw, &buffer_size, &max_buffer_size)) {
2404 return -1;
2405 }
2406
Benjamin Peterson59406a92009-03-26 17:10:29 +00002407 if (max_buffer_size != -234 && !complain_about_max_buffer_size())
2408 return -1;
2409
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002410 if (_PyIOBase_check_seekable(raw, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002411 return -1;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002412 if (_PyIOBase_check_readable(raw, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002413 return -1;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002414 if (_PyIOBase_check_writable(raw, Py_True) == NULL)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002415 return -1;
2416
2417 Py_CLEAR(self->raw);
2418 Py_INCREF(raw);
2419 self->raw = raw;
2420 self->buffer_size = buffer_size;
2421 self->readable = 1;
2422 self->writable = 1;
2423
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002424 if (_buffered_init(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002425 return -1;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002426 _bufferedreader_reset_buf(self);
2427 _bufferedwriter_reset_buf(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002428 self->pos = 0;
2429
Antoine Pitrou711af3a2009-04-11 15:39:24 +00002430 self->fast_closed_checks = (Py_TYPE(self) == &PyBufferedRandom_Type &&
2431 Py_TYPE(raw) == &PyFileIO_Type);
2432
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002433 self->ok = 1;
2434 return 0;
2435}
2436
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002437static PyMethodDef bufferedrandom_methods[] = {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002438 /* BufferedIOMixin methods */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002439 {"close", (PyCFunction)buffered_close, METH_NOARGS},
2440 {"detach", (PyCFunction)buffered_detach, METH_NOARGS},
2441 {"seekable", (PyCFunction)buffered_seekable, METH_NOARGS},
2442 {"readable", (PyCFunction)buffered_readable, METH_NOARGS},
2443 {"writable", (PyCFunction)buffered_writable, METH_NOARGS},
2444 {"fileno", (PyCFunction)buffered_fileno, METH_NOARGS},
2445 {"isatty", (PyCFunction)buffered_isatty, METH_NOARGS},
Antoine Pitroue033e062010-10-29 10:38:18 +00002446 {"_dealloc_warn", (PyCFunction)buffered_dealloc_warn, METH_O},
Antoine Pitrou243757e2010-11-05 21:15:39 +00002447 {"__getstate__", (PyCFunction)buffered_getstate, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002448
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002449 {"flush", (PyCFunction)buffered_flush, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002450
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002451 {"seek", (PyCFunction)buffered_seek, METH_VARARGS},
2452 {"tell", (PyCFunction)buffered_tell, METH_NOARGS},
2453 {"truncate", (PyCFunction)buffered_truncate, METH_VARARGS},
2454 {"read", (PyCFunction)buffered_read, METH_VARARGS},
2455 {"read1", (PyCFunction)buffered_read1, METH_VARARGS},
2456 {"readinto", (PyCFunction)buffered_readinto, METH_VARARGS},
2457 {"readline", (PyCFunction)buffered_readline, METH_VARARGS},
2458 {"peek", (PyCFunction)buffered_peek, METH_VARARGS},
2459 {"write", (PyCFunction)bufferedwriter_write, METH_VARARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002460 {NULL, NULL}
2461};
2462
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002463static PyMemberDef bufferedrandom_members[] = {
Antoine Pitrou7f8f4182010-12-21 21:20:59 +00002464 {"raw", T_OBJECT, offsetof(buffered, raw), READONLY},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002465 {NULL}
2466};
2467
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002468static PyGetSetDef bufferedrandom_getset[] = {
2469 {"closed", (getter)buffered_closed_get, NULL, NULL},
2470 {"name", (getter)buffered_name_get, NULL, NULL},
2471 {"mode", (getter)buffered_mode_get, NULL, NULL},
Benjamin Peterson1fea3212009-04-19 03:15:20 +00002472 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002473};
2474
2475
2476PyTypeObject PyBufferedRandom_Type = {
2477 PyVarObject_HEAD_INIT(NULL, 0)
2478 "_io.BufferedRandom", /*tp_name*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002479 sizeof(buffered), /*tp_basicsize*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002480 0, /*tp_itemsize*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002481 (destructor)buffered_dealloc, /*tp_dealloc*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002482 0, /*tp_print*/
2483 0, /*tp_getattr*/
2484 0, /*tp_setattr*/
2485 0, /*tp_compare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002486 (reprfunc)buffered_repr, /*tp_repr*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002487 0, /*tp_as_number*/
2488 0, /*tp_as_sequence*/
2489 0, /*tp_as_mapping*/
2490 0, /*tp_hash */
2491 0, /*tp_call*/
2492 0, /*tp_str*/
2493 0, /*tp_getattro*/
2494 0, /*tp_setattro*/
2495 0, /*tp_as_buffer*/
2496 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE
2497 | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002498 bufferedrandom_doc, /* tp_doc */
2499 (traverseproc)buffered_traverse, /* tp_traverse */
2500 (inquiry)buffered_clear, /* tp_clear */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002501 0, /* tp_richcompare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002502 offsetof(buffered, weakreflist), /*tp_weaklistoffset*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002503 0, /* tp_iter */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002504 (iternextfunc)buffered_iternext, /* tp_iternext */
2505 bufferedrandom_methods, /* tp_methods */
2506 bufferedrandom_members, /* tp_members */
2507 bufferedrandom_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002508 0, /* tp_base */
2509 0, /*tp_dict*/
2510 0, /* tp_descr_get */
2511 0, /* tp_descr_set */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002512 offsetof(buffered, dict), /*tp_dictoffset*/
2513 (initproc)bufferedrandom_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002514 0, /* tp_alloc */
2515 PyType_GenericNew, /* tp_new */
2516};