blob: 635093efff95b606d7b74a84ab9bf579f1336552 [file] [log] [blame]
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001/*
2 An implementation of Text I/O as defined by PEP 3116 - "New I/O"
Antoine Pitrou24f36292009-03-28 22:16:42 +00003
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00004 Classes defined here: TextIOBase, IncrementalNewlineDecoder, TextIOWrapper.
Antoine Pitrou24f36292009-03-28 22:16:42 +00005
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00006 Written by Amaury Forgeot d'Arc and Antoine Pitrou
7*/
8
9#define PY_SSIZE_T_CLEAN
10#include "Python.h"
11#include "structmember.h"
12#include "_iomodule.h"
13
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020014_Py_IDENTIFIER(close);
15_Py_IDENTIFIER(_dealloc_warn);
16_Py_IDENTIFIER(decode);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020017_Py_IDENTIFIER(fileno);
18_Py_IDENTIFIER(flush);
19_Py_IDENTIFIER(getpreferredencoding);
20_Py_IDENTIFIER(isatty);
Martin v. Löwis767046a2011-10-14 15:35:36 +020021_Py_IDENTIFIER(mode);
22_Py_IDENTIFIER(name);
23_Py_IDENTIFIER(raw);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020024_Py_IDENTIFIER(read);
Martin v. Löwis767046a2011-10-14 15:35:36 +020025_Py_IDENTIFIER(read1);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020026_Py_IDENTIFIER(readable);
27_Py_IDENTIFIER(replace);
28_Py_IDENTIFIER(reset);
29_Py_IDENTIFIER(seek);
30_Py_IDENTIFIER(seekable);
31_Py_IDENTIFIER(setstate);
32_Py_IDENTIFIER(tell);
33_Py_IDENTIFIER(writable);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +020034
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000035/* TextIOBase */
36
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000037PyDoc_STRVAR(textiobase_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000038 "Base class for text I/O.\n"
39 "\n"
40 "This class provides a character and line based interface to stream\n"
41 "I/O. There is no readinto method because Python's character strings\n"
42 "are immutable. There is no public constructor.\n"
43 );
44
45static PyObject *
46_unsupported(const char *message)
47{
Antoine Pitrou712cb732013-12-21 15:51:54 +010048 _PyIO_State *state = IO_STATE();
49 if (state != NULL)
50 PyErr_SetString(state->unsupported_operation, message);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000051 return NULL;
52}
53
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000054PyDoc_STRVAR(textiobase_detach_doc,
Benjamin Petersond2e0c792009-05-01 20:40:59 +000055 "Separate the underlying buffer from the TextIOBase and return it.\n"
56 "\n"
57 "After the underlying buffer has been detached, the TextIO is in an\n"
58 "unusable state.\n"
59 );
60
61static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000062textiobase_detach(PyObject *self)
Benjamin Petersond2e0c792009-05-01 20:40:59 +000063{
64 return _unsupported("detach");
65}
66
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000067PyDoc_STRVAR(textiobase_read_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000068 "Read at most n characters from stream.\n"
69 "\n"
70 "Read from underlying buffer until we have n characters or we hit EOF.\n"
71 "If n is negative or omitted, read until EOF.\n"
72 );
73
74static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000075textiobase_read(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000076{
77 return _unsupported("read");
78}
79
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000080PyDoc_STRVAR(textiobase_readline_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000081 "Read until newline or EOF.\n"
82 "\n"
83 "Returns an empty string if EOF is hit immediately.\n"
84 );
85
86static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000087textiobase_readline(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000088{
89 return _unsupported("readline");
90}
91
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000092PyDoc_STRVAR(textiobase_write_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000093 "Write string to stream.\n"
94 "Returns the number of characters written (which is always equal to\n"
95 "the length of the string).\n"
96 );
97
98static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000099textiobase_write(PyObject *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000100{
101 return _unsupported("write");
102}
103
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000104PyDoc_STRVAR(textiobase_encoding_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000105 "Encoding of the text stream.\n"
106 "\n"
107 "Subclasses should override.\n"
108 );
109
110static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000111textiobase_encoding_get(PyObject *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000112{
113 Py_RETURN_NONE;
114}
115
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000116PyDoc_STRVAR(textiobase_newlines_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000117 "Line endings translated so far.\n"
118 "\n"
119 "Only line endings translated during reading are considered.\n"
120 "\n"
121 "Subclasses should override.\n"
122 );
123
124static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000125textiobase_newlines_get(PyObject *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000126{
127 Py_RETURN_NONE;
128}
129
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000130PyDoc_STRVAR(textiobase_errors_doc,
Benjamin Peterson0926ad12009-06-06 18:02:12 +0000131 "The error setting of the decoder or encoder.\n"
132 "\n"
133 "Subclasses should override.\n"
134 );
135
136static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000137textiobase_errors_get(PyObject *self, void *context)
Benjamin Peterson0926ad12009-06-06 18:02:12 +0000138{
139 Py_RETURN_NONE;
140}
141
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000142
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000143static PyMethodDef textiobase_methods[] = {
144 {"detach", (PyCFunction)textiobase_detach, METH_NOARGS, textiobase_detach_doc},
145 {"read", textiobase_read, METH_VARARGS, textiobase_read_doc},
146 {"readline", textiobase_readline, METH_VARARGS, textiobase_readline_doc},
147 {"write", textiobase_write, METH_VARARGS, textiobase_write_doc},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000148 {NULL, NULL}
149};
150
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000151static PyGetSetDef textiobase_getset[] = {
152 {"encoding", (getter)textiobase_encoding_get, NULL, textiobase_encoding_doc},
153 {"newlines", (getter)textiobase_newlines_get, NULL, textiobase_newlines_doc},
154 {"errors", (getter)textiobase_errors_get, NULL, textiobase_errors_doc},
Benjamin Peterson1fea3212009-04-19 03:15:20 +0000155 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000156};
157
158PyTypeObject PyTextIOBase_Type = {
159 PyVarObject_HEAD_INIT(NULL, 0)
160 "_io._TextIOBase", /*tp_name*/
161 0, /*tp_basicsize*/
162 0, /*tp_itemsize*/
163 0, /*tp_dealloc*/
164 0, /*tp_print*/
165 0, /*tp_getattr*/
166 0, /*tp_setattr*/
167 0, /*tp_compare */
168 0, /*tp_repr*/
169 0, /*tp_as_number*/
170 0, /*tp_as_sequence*/
171 0, /*tp_as_mapping*/
172 0, /*tp_hash */
173 0, /*tp_call*/
174 0, /*tp_str*/
175 0, /*tp_getattro*/
176 0, /*tp_setattro*/
177 0, /*tp_as_buffer*/
Antoine Pitrou796564c2013-07-30 19:59:21 +0200178 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE
179 | Py_TPFLAGS_HAVE_FINALIZE, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000180 textiobase_doc, /* tp_doc */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000181 0, /* tp_traverse */
182 0, /* tp_clear */
183 0, /* tp_richcompare */
184 0, /* tp_weaklistoffset */
185 0, /* tp_iter */
186 0, /* tp_iternext */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000187 textiobase_methods, /* tp_methods */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000188 0, /* tp_members */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000189 textiobase_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000190 &PyIOBase_Type, /* tp_base */
191 0, /* tp_dict */
192 0, /* tp_descr_get */
193 0, /* tp_descr_set */
194 0, /* tp_dictoffset */
195 0, /* tp_init */
196 0, /* tp_alloc */
197 0, /* tp_new */
Antoine Pitrou796564c2013-07-30 19:59:21 +0200198 0, /* tp_free */
199 0, /* tp_is_gc */
200 0, /* tp_bases */
201 0, /* tp_mro */
202 0, /* tp_cache */
203 0, /* tp_subclasses */
204 0, /* tp_weaklist */
205 0, /* tp_del */
206 0, /* tp_version_tag */
207 0, /* tp_finalize */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000208};
209
210
211/* IncrementalNewlineDecoder */
212
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000213PyDoc_STRVAR(incrementalnewlinedecoder_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000214 "Codec used when reading a file in universal newlines mode. It wraps\n"
215 "another incremental decoder, translating \\r\\n and \\r into \\n. It also\n"
216 "records the types of newlines encountered. When used with\n"
217 "translate=False, it ensures that the newline sequence is returned in\n"
218 "one piece. When used with decoder=None, it expects unicode strings as\n"
219 "decode input and translates newlines without first invoking an external\n"
220 "decoder.\n"
221 );
222
223typedef struct {
224 PyObject_HEAD
225 PyObject *decoder;
226 PyObject *errors;
Antoine Pitrouca767bd2009-09-21 21:37:02 +0000227 signed int pendingcr: 1;
228 signed int translate: 1;
229 unsigned int seennl: 3;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000230} nldecoder_object;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000231
232static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000233incrementalnewlinedecoder_init(nldecoder_object *self,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000234 PyObject *args, PyObject *kwds)
235{
236 PyObject *decoder;
237 int translate;
238 PyObject *errors = NULL;
239 char *kwlist[] = {"decoder", "translate", "errors", NULL};
240
241 if (!PyArg_ParseTupleAndKeywords(args, kwds, "Oi|O:IncrementalNewlineDecoder",
242 kwlist, &decoder, &translate, &errors))
243 return -1;
244
245 self->decoder = decoder;
246 Py_INCREF(decoder);
247
248 if (errors == NULL) {
249 self->errors = PyUnicode_FromString("strict");
250 if (self->errors == NULL)
251 return -1;
252 }
253 else {
254 Py_INCREF(errors);
255 self->errors = errors;
256 }
257
258 self->translate = translate;
259 self->seennl = 0;
260 self->pendingcr = 0;
261
262 return 0;
263}
264
265static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000266incrementalnewlinedecoder_dealloc(nldecoder_object *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000267{
268 Py_CLEAR(self->decoder);
269 Py_CLEAR(self->errors);
270 Py_TYPE(self)->tp_free((PyObject *)self);
271}
272
Serhiy Storchaka94dc6732013-02-03 17:03:31 +0200273static int
274check_decoded(PyObject *decoded)
275{
276 if (decoded == NULL)
277 return -1;
278 if (!PyUnicode_Check(decoded)) {
279 PyErr_Format(PyExc_TypeError,
280 "decoder should return a string result, not '%.200s'",
281 Py_TYPE(decoded)->tp_name);
282 Py_DECREF(decoded);
283 return -1;
284 }
Serhiy Storchakad03ce4a2013-02-03 17:07:32 +0200285 if (PyUnicode_READY(decoded) < 0) {
286 Py_DECREF(decoded);
287 return -1;
288 }
Serhiy Storchaka94dc6732013-02-03 17:03:31 +0200289 return 0;
290}
291
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000292#define SEEN_CR 1
293#define SEEN_LF 2
294#define SEEN_CRLF 4
295#define SEEN_ALL (SEEN_CR | SEEN_LF | SEEN_CRLF)
296
297PyObject *
Antoine Pitrou09fcb722013-10-23 19:20:21 +0200298_PyIncrementalNewlineDecoder_decode(PyObject *myself,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000299 PyObject *input, int final)
300{
301 PyObject *output;
302 Py_ssize_t output_len;
Antoine Pitrou09fcb722013-10-23 19:20:21 +0200303 nldecoder_object *self = (nldecoder_object *) myself;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000304
305 if (self->decoder == NULL) {
306 PyErr_SetString(PyExc_ValueError,
307 "IncrementalNewlineDecoder.__init__ not called");
308 return NULL;
309 }
310
311 /* decode input (with the eventual \r from a previous pass) */
312 if (self->decoder != Py_None) {
313 output = PyObject_CallMethodObjArgs(self->decoder,
314 _PyIO_str_decode, input, final ? Py_True : Py_False, NULL);
315 }
316 else {
317 output = input;
318 Py_INCREF(output);
319 }
320
Serhiy Storchaka94dc6732013-02-03 17:03:31 +0200321 if (check_decoded(output) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000322 return NULL;
323
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200324 output_len = PyUnicode_GET_LENGTH(output);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000325 if (self->pendingcr && (final || output_len > 0)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200326 /* Prefix output with CR */
327 int kind;
328 PyObject *modified;
329 char *out;
330
331 modified = PyUnicode_New(output_len + 1,
332 PyUnicode_MAX_CHAR_VALUE(output));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000333 if (modified == NULL)
334 goto error;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200335 kind = PyUnicode_KIND(modified);
336 out = PyUnicode_DATA(modified);
337 PyUnicode_WRITE(kind, PyUnicode_DATA(modified), 0, '\r');
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200338 memcpy(out + kind, PyUnicode_DATA(output), kind * output_len);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000339 Py_DECREF(output);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200340 output = modified; /* output remains ready */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000341 self->pendingcr = 0;
342 output_len++;
343 }
344
345 /* retain last \r even when not translating data:
346 * then readline() is sure to get \r\n in one pass
347 */
348 if (!final) {
Antoine Pitrou24f36292009-03-28 22:16:42 +0000349 if (output_len > 0
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200350 && PyUnicode_READ_CHAR(output, output_len - 1) == '\r')
351 {
352 PyObject *modified = PyUnicode_Substring(output, 0, output_len -1);
353 if (modified == NULL)
354 goto error;
355 Py_DECREF(output);
356 output = modified;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000357 self->pendingcr = 1;
358 }
359 }
360
361 /* Record which newlines are read and do newline translation if desired,
362 all in one pass. */
363 {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200364 void *in_str;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000365 Py_ssize_t len;
366 int seennl = self->seennl;
367 int only_lf = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200368 int kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000369
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200370 in_str = PyUnicode_DATA(output);
371 len = PyUnicode_GET_LENGTH(output);
372 kind = PyUnicode_KIND(output);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000373
374 if (len == 0)
375 return output;
376
377 /* If, up to now, newlines are consistently \n, do a quick check
378 for the \r *byte* with the libc's optimized memchr.
379 */
380 if (seennl == SEEN_LF || seennl == 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200381 only_lf = (memchr(in_str, '\r', kind * len) == NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000382 }
383
Antoine Pitrou66913e22009-03-06 23:40:56 +0000384 if (only_lf) {
385 /* If not already seen, quick scan for a possible "\n" character.
386 (there's nothing else to be done, even when in translation mode)
387 */
388 if (seennl == 0 &&
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200389 memchr(in_str, '\n', kind * len) != NULL) {
Antoine Pitrouc28e2e52011-11-13 03:53:42 +0100390 if (kind == PyUnicode_1BYTE_KIND)
391 seennl |= SEEN_LF;
392 else {
393 Py_ssize_t i = 0;
394 for (;;) {
395 Py_UCS4 c;
396 /* Fast loop for non-control characters */
397 while (PyUnicode_READ(kind, in_str, i) > '\n')
398 i++;
399 c = PyUnicode_READ(kind, in_str, i++);
400 if (c == '\n') {
401 seennl |= SEEN_LF;
402 break;
403 }
404 if (i >= len)
405 break;
Antoine Pitrou66913e22009-03-06 23:40:56 +0000406 }
Antoine Pitrou66913e22009-03-06 23:40:56 +0000407 }
408 }
409 /* Finished: we have scanned for newlines, and none of them
410 need translating */
411 }
412 else if (!self->translate) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200413 Py_ssize_t i = 0;
Antoine Pitrou66913e22009-03-06 23:40:56 +0000414 /* We have already seen all newline types, no need to scan again */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000415 if (seennl == SEEN_ALL)
416 goto endscan;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000417 for (;;) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200418 Py_UCS4 c;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000419 /* Fast loop for non-control characters */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200420 while (PyUnicode_READ(kind, in_str, i) > '\r')
421 i++;
422 c = PyUnicode_READ(kind, in_str, i++);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000423 if (c == '\n')
424 seennl |= SEEN_LF;
425 else if (c == '\r') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200426 if (PyUnicode_READ(kind, in_str, i) == '\n') {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000427 seennl |= SEEN_CRLF;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200428 i++;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000429 }
430 else
431 seennl |= SEEN_CR;
432 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200433 if (i >= len)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000434 break;
435 if (seennl == SEEN_ALL)
436 break;
437 }
438 endscan:
439 ;
440 }
Antoine Pitrou66913e22009-03-06 23:40:56 +0000441 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200442 void *translated;
443 int kind = PyUnicode_KIND(output);
444 void *in_str = PyUnicode_DATA(output);
445 Py_ssize_t in, out;
446 /* XXX: Previous in-place translation here is disabled as
447 resizing is not possible anymore */
448 /* We could try to optimize this so that we only do a copy
449 when there is something to translate. On the other hand,
450 we already know there is a \r byte, so chances are high
451 that something needs to be done. */
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200452 translated = PyMem_Malloc(kind * len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200453 if (translated == NULL) {
454 PyErr_NoMemory();
455 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000456 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200457 in = out = 0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000458 for (;;) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200459 Py_UCS4 c;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000460 /* Fast loop for non-control characters */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200461 while ((c = PyUnicode_READ(kind, in_str, in++)) > '\r')
462 PyUnicode_WRITE(kind, translated, out++, c);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000463 if (c == '\n') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200464 PyUnicode_WRITE(kind, translated, out++, c);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000465 seennl |= SEEN_LF;
466 continue;
467 }
468 if (c == '\r') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200469 if (PyUnicode_READ(kind, in_str, in) == '\n') {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000470 in++;
471 seennl |= SEEN_CRLF;
472 }
473 else
474 seennl |= SEEN_CR;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200475 PyUnicode_WRITE(kind, translated, out++, '\n');
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000476 continue;
477 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200478 if (in > len)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000479 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200480 PyUnicode_WRITE(kind, translated, out++, c);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000481 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200482 Py_DECREF(output);
483 output = PyUnicode_FromKindAndData(kind, translated, out);
Antoine Pitrouc1b0bfd2011-11-12 22:34:28 +0100484 PyMem_Free(translated);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200485 if (!output)
Ross Lagerwall0f9eec12012-04-07 07:09:57 +0200486 return NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000487 }
488 self->seennl |= seennl;
489 }
490
491 return output;
492
493 error:
494 Py_DECREF(output);
495 return NULL;
496}
497
498static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000499incrementalnewlinedecoder_decode(nldecoder_object *self,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000500 PyObject *args, PyObject *kwds)
501{
502 char *kwlist[] = {"input", "final", NULL};
503 PyObject *input;
504 int final = 0;
505
506 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|i:IncrementalNewlineDecoder",
507 kwlist, &input, &final))
508 return NULL;
509 return _PyIncrementalNewlineDecoder_decode((PyObject *) self, input, final);
510}
511
512static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000513incrementalnewlinedecoder_getstate(nldecoder_object *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000514{
515 PyObject *buffer;
516 unsigned PY_LONG_LONG flag;
517
518 if (self->decoder != Py_None) {
519 PyObject *state = PyObject_CallMethodObjArgs(self->decoder,
520 _PyIO_str_getstate, NULL);
521 if (state == NULL)
522 return NULL;
523 if (!PyArg_Parse(state, "(OK)", &buffer, &flag)) {
524 Py_DECREF(state);
525 return NULL;
526 }
527 Py_INCREF(buffer);
528 Py_DECREF(state);
529 }
530 else {
531 buffer = PyBytes_FromString("");
532 flag = 0;
533 }
534 flag <<= 1;
535 if (self->pendingcr)
536 flag |= 1;
537 return Py_BuildValue("NK", buffer, flag);
538}
539
540static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000541incrementalnewlinedecoder_setstate(nldecoder_object *self, PyObject *state)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000542{
543 PyObject *buffer;
544 unsigned PY_LONG_LONG flag;
545
546 if (!PyArg_Parse(state, "(OK)", &buffer, &flag))
547 return NULL;
548
549 self->pendingcr = (int) flag & 1;
550 flag >>= 1;
551
552 if (self->decoder != Py_None)
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200553 return _PyObject_CallMethodId(self->decoder,
554 &PyId_setstate, "((OK))", buffer, flag);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000555 else
556 Py_RETURN_NONE;
557}
558
559static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000560incrementalnewlinedecoder_reset(nldecoder_object *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000561{
562 self->seennl = 0;
563 self->pendingcr = 0;
564 if (self->decoder != Py_None)
565 return PyObject_CallMethodObjArgs(self->decoder, _PyIO_str_reset, NULL);
566 else
567 Py_RETURN_NONE;
568}
569
570static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000571incrementalnewlinedecoder_newlines_get(nldecoder_object *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000572{
573 switch (self->seennl) {
574 case SEEN_CR:
575 return PyUnicode_FromString("\r");
576 case SEEN_LF:
577 return PyUnicode_FromString("\n");
578 case SEEN_CRLF:
579 return PyUnicode_FromString("\r\n");
580 case SEEN_CR | SEEN_LF:
581 return Py_BuildValue("ss", "\r", "\n");
582 case SEEN_CR | SEEN_CRLF:
583 return Py_BuildValue("ss", "\r", "\r\n");
584 case SEEN_LF | SEEN_CRLF:
585 return Py_BuildValue("ss", "\n", "\r\n");
586 case SEEN_CR | SEEN_LF | SEEN_CRLF:
587 return Py_BuildValue("sss", "\r", "\n", "\r\n");
588 default:
589 Py_RETURN_NONE;
590 }
591
592}
593
594
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000595static PyMethodDef incrementalnewlinedecoder_methods[] = {
596 {"decode", (PyCFunction)incrementalnewlinedecoder_decode, METH_VARARGS|METH_KEYWORDS},
597 {"getstate", (PyCFunction)incrementalnewlinedecoder_getstate, METH_NOARGS},
598 {"setstate", (PyCFunction)incrementalnewlinedecoder_setstate, METH_O},
599 {"reset", (PyCFunction)incrementalnewlinedecoder_reset, METH_NOARGS},
Benjamin Peterson1fea3212009-04-19 03:15:20 +0000600 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000601};
602
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000603static PyGetSetDef incrementalnewlinedecoder_getset[] = {
604 {"newlines", (getter)incrementalnewlinedecoder_newlines_get, NULL, NULL},
Benjamin Peterson1fea3212009-04-19 03:15:20 +0000605 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000606};
607
608PyTypeObject PyIncrementalNewlineDecoder_Type = {
609 PyVarObject_HEAD_INIT(NULL, 0)
610 "_io.IncrementalNewlineDecoder", /*tp_name*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000611 sizeof(nldecoder_object), /*tp_basicsize*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000612 0, /*tp_itemsize*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000613 (destructor)incrementalnewlinedecoder_dealloc, /*tp_dealloc*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000614 0, /*tp_print*/
615 0, /*tp_getattr*/
616 0, /*tp_setattr*/
617 0, /*tp_compare */
618 0, /*tp_repr*/
619 0, /*tp_as_number*/
620 0, /*tp_as_sequence*/
621 0, /*tp_as_mapping*/
622 0, /*tp_hash */
623 0, /*tp_call*/
624 0, /*tp_str*/
625 0, /*tp_getattro*/
626 0, /*tp_setattro*/
627 0, /*tp_as_buffer*/
628 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000629 incrementalnewlinedecoder_doc, /* tp_doc */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000630 0, /* tp_traverse */
631 0, /* tp_clear */
632 0, /* tp_richcompare */
633 0, /*tp_weaklistoffset*/
634 0, /* tp_iter */
635 0, /* tp_iternext */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000636 incrementalnewlinedecoder_methods, /* tp_methods */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000637 0, /* tp_members */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000638 incrementalnewlinedecoder_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000639 0, /* tp_base */
640 0, /* tp_dict */
641 0, /* tp_descr_get */
642 0, /* tp_descr_set */
643 0, /* tp_dictoffset */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000644 (initproc)incrementalnewlinedecoder_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000645 0, /* tp_alloc */
646 PyType_GenericNew, /* tp_new */
647};
648
649
650/* TextIOWrapper */
651
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000652PyDoc_STRVAR(textiowrapper_doc,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000653 "Character and line based layer over a BufferedIOBase object, buffer.\n"
654 "\n"
655 "encoding gives the name of the encoding that the stream will be\n"
Victor Stinnerf86a5e82012-06-05 13:43:22 +0200656 "decoded or encoded with. It defaults to locale.getpreferredencoding(False).\n"
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000657 "\n"
Andrew Kuchlingc7b6c502013-06-16 12:58:48 -0400658 "errors determines the strictness of encoding and decoding (see\n"
659 "help(codecs.Codec) or the documentation for codecs.register) and\n"
660 "defaults to \"strict\".\n"
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000661 "\n"
Antoine Pitrou0c1c0d42012-08-04 00:55:38 +0200662 "newline controls how line endings are handled. It can be None, '',\n"
663 "'\\n', '\\r', and '\\r\\n'. It works as follows:\n"
664 "\n"
665 "* On input, if newline is None, universal newlines mode is\n"
666 " enabled. Lines in the input can end in '\\n', '\\r', or '\\r\\n', and\n"
667 " these are translated into '\\n' before being returned to the\n"
668 " caller. If it is '', universal newline mode is enabled, but line\n"
669 " endings are returned to the caller untranslated. If it has any of\n"
670 " the other legal values, input lines are only terminated by the given\n"
671 " string, and the line ending is returned to the caller untranslated.\n"
672 "\n"
673 "* On output, if newline is None, any '\\n' characters written are\n"
674 " translated to the system default line separator, os.linesep. If\n"
Ezio Melotti16d2b472012-09-18 07:20:18 +0300675 " newline is '' or '\\n', no translation takes place. If newline is any\n"
Victor Stinner401e17d2012-08-04 01:18:56 +0200676 " of the other legal values, any '\\n' characters written are translated\n"
677 " to the given string.\n"
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000678 "\n"
679 "If line_buffering is True, a call to flush is implied when a call to\n"
680 "write contains a newline character."
681 );
682
683typedef PyObject *
684 (*encodefunc_t)(PyObject *, PyObject *);
685
686typedef struct
687{
688 PyObject_HEAD
689 int ok; /* initialized? */
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000690 int detached;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000691 Py_ssize_t chunk_size;
692 PyObject *buffer;
693 PyObject *encoding;
694 PyObject *encoder;
695 PyObject *decoder;
696 PyObject *readnl;
697 PyObject *errors;
698 const char *writenl; /* utf-8 encoded, NULL stands for \n */
699 char line_buffering;
Antoine Pitroue96ec682011-07-23 21:46:35 +0200700 char write_through;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000701 char readuniversal;
702 char readtranslate;
703 char writetranslate;
704 char seekable;
Antoine Pitroue96ec682011-07-23 21:46:35 +0200705 char has_read1;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000706 char telling;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200707 char finalizing;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000708 /* Specialized encoding func (see below) */
709 encodefunc_t encodefunc;
Antoine Pitroue4501852009-05-14 18:55:55 +0000710 /* Whether or not it's the start of the stream */
711 char encoding_start_of_stream;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000712
713 /* Reads and writes are internally buffered in order to speed things up.
714 However, any read will first flush the write buffer if itsn't empty.
Antoine Pitrou24f36292009-03-28 22:16:42 +0000715
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000716 Please also note that text to be written is first encoded before being
717 buffered. This is necessary so that encoding errors are immediately
718 reported to the caller, but it unfortunately means that the
719 IncrementalEncoder (whose encode() method is always written in Python)
720 becomes a bottleneck for small writes.
721 */
722 PyObject *decoded_chars; /* buffer for text returned from decoder */
723 Py_ssize_t decoded_chars_used; /* offset into _decoded_chars for read() */
724 PyObject *pending_bytes; /* list of bytes objects waiting to be
725 written, or NULL */
726 Py_ssize_t pending_bytes_count;
Antoine Pitrou211b81d2011-02-25 20:27:33 +0000727
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000728 /* snapshot is either None, or a tuple (dec_flags, next_input) where
729 * dec_flags is the second (integer) item of the decoder state and
730 * next_input is the chunk of input bytes that comes next after the
731 * snapshot point. We use this to reconstruct decoder states in tell().
732 */
Antoine Pitrou211b81d2011-02-25 20:27:33 +0000733 PyObject *snapshot;
734 /* Bytes-to-characters ratio for the current chunk. Serves as input for
735 the heuristic in tell(). */
736 double b2cratio;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000737
738 /* Cache raw object if it's a FileIO object */
739 PyObject *raw;
740
741 PyObject *weakreflist;
742 PyObject *dict;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000743} textio;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000744
745
746/* A couple of specialized cases in order to bypass the slow incremental
747 encoding methods for the most popular encodings. */
748
749static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000750ascii_encode(textio *self, PyObject *text)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000751{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200752 return _PyUnicode_AsASCIIString(text, PyBytes_AS_STRING(self->errors));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000753}
754
755static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000756utf16be_encode(textio *self, PyObject *text)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000757{
Antoine Pitrou5c398e82011-11-13 04:11:37 +0100758 return _PyUnicode_EncodeUTF16(text,
759 PyBytes_AS_STRING(self->errors), 1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000760}
761
762static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000763utf16le_encode(textio *self, PyObject *text)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000764{
Antoine Pitrou5c398e82011-11-13 04:11:37 +0100765 return _PyUnicode_EncodeUTF16(text,
766 PyBytes_AS_STRING(self->errors), -1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000767}
768
769static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000770utf16_encode(textio *self, PyObject *text)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000771{
Antoine Pitroue4501852009-05-14 18:55:55 +0000772 if (!self->encoding_start_of_stream) {
773 /* Skip the BOM and use native byte ordering */
Christian Heimes743e0cd2012-10-17 23:52:17 +0200774#if PY_BIG_ENDIAN
Antoine Pitroue4501852009-05-14 18:55:55 +0000775 return utf16be_encode(self, text);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000776#else
Antoine Pitroue4501852009-05-14 18:55:55 +0000777 return utf16le_encode(self, text);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000778#endif
Antoine Pitroue4501852009-05-14 18:55:55 +0000779 }
Antoine Pitrou5c398e82011-11-13 04:11:37 +0100780 return _PyUnicode_EncodeUTF16(text,
781 PyBytes_AS_STRING(self->errors), 0);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000782}
783
Antoine Pitroue4501852009-05-14 18:55:55 +0000784static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000785utf32be_encode(textio *self, PyObject *text)
Antoine Pitroue4501852009-05-14 18:55:55 +0000786{
Antoine Pitrou5c398e82011-11-13 04:11:37 +0100787 return _PyUnicode_EncodeUTF32(text,
788 PyBytes_AS_STRING(self->errors), 1);
Antoine Pitroue4501852009-05-14 18:55:55 +0000789}
790
791static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000792utf32le_encode(textio *self, PyObject *text)
Antoine Pitroue4501852009-05-14 18:55:55 +0000793{
Antoine Pitrou5c398e82011-11-13 04:11:37 +0100794 return _PyUnicode_EncodeUTF32(text,
795 PyBytes_AS_STRING(self->errors), -1);
Antoine Pitroue4501852009-05-14 18:55:55 +0000796}
797
798static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000799utf32_encode(textio *self, PyObject *text)
Antoine Pitroue4501852009-05-14 18:55:55 +0000800{
801 if (!self->encoding_start_of_stream) {
802 /* Skip the BOM and use native byte ordering */
Christian Heimes743e0cd2012-10-17 23:52:17 +0200803#if PY_BIG_ENDIAN
Antoine Pitroue4501852009-05-14 18:55:55 +0000804 return utf32be_encode(self, text);
805#else
806 return utf32le_encode(self, text);
807#endif
808 }
Antoine Pitrou5c398e82011-11-13 04:11:37 +0100809 return _PyUnicode_EncodeUTF32(text,
810 PyBytes_AS_STRING(self->errors), 0);
Antoine Pitroue4501852009-05-14 18:55:55 +0000811}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000812
813static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000814utf8_encode(textio *self, PyObject *text)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000815{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200816 return _PyUnicode_AsUTF8String(text, PyBytes_AS_STRING(self->errors));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000817}
818
819static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000820latin1_encode(textio *self, PyObject *text)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000821{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200822 return _PyUnicode_AsLatin1String(text, PyBytes_AS_STRING(self->errors));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000823}
824
825/* Map normalized encoding names onto the specialized encoding funcs */
826
827typedef struct {
828 const char *name;
829 encodefunc_t encodefunc;
830} encodefuncentry;
831
Antoine Pitrou24f36292009-03-28 22:16:42 +0000832static encodefuncentry encodefuncs[] = {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000833 {"ascii", (encodefunc_t) ascii_encode},
834 {"iso8859-1", (encodefunc_t) latin1_encode},
Antoine Pitroue4501852009-05-14 18:55:55 +0000835 {"utf-8", (encodefunc_t) utf8_encode},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000836 {"utf-16-be", (encodefunc_t) utf16be_encode},
837 {"utf-16-le", (encodefunc_t) utf16le_encode},
838 {"utf-16", (encodefunc_t) utf16_encode},
Antoine Pitroue4501852009-05-14 18:55:55 +0000839 {"utf-32-be", (encodefunc_t) utf32be_encode},
840 {"utf-32-le", (encodefunc_t) utf32le_encode},
841 {"utf-32", (encodefunc_t) utf32_encode},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000842 {NULL, NULL}
843};
844
845
846static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000847textiowrapper_init(textio *self, PyObject *args, PyObject *kwds)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000848{
849 char *kwlist[] = {"buffer", "encoding", "errors",
Antoine Pitroue96ec682011-07-23 21:46:35 +0200850 "newline", "line_buffering", "write_through",
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000851 NULL};
Nick Coghlana9b15242014-02-04 22:11:18 +1000852 PyObject *buffer, *raw, *codec_info = NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000853 char *encoding = NULL;
854 char *errors = NULL;
855 char *newline = NULL;
Antoine Pitroue96ec682011-07-23 21:46:35 +0200856 int line_buffering = 0, write_through = 0;
Antoine Pitrou712cb732013-12-21 15:51:54 +0100857 _PyIO_State *state = NULL;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000858
859 PyObject *res;
860 int r;
861
862 self->ok = 0;
Benjamin Petersond2e0c792009-05-01 20:40:59 +0000863 self->detached = 0;
Antoine Pitroue96ec682011-07-23 21:46:35 +0200864 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|zzzii:fileio",
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000865 kwlist, &buffer, &encoding, &errors,
Antoine Pitroue96ec682011-07-23 21:46:35 +0200866 &newline, &line_buffering, &write_through))
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000867 return -1;
868
869 if (newline && newline[0] != '\0'
870 && !(newline[0] == '\n' && newline[1] == '\0')
871 && !(newline[0] == '\r' && newline[1] == '\0')
872 && !(newline[0] == '\r' && newline[1] == '\n' && newline[2] == '\0')) {
873 PyErr_Format(PyExc_ValueError,
874 "illegal newline value: %s", newline);
875 return -1;
876 }
877
878 Py_CLEAR(self->buffer);
879 Py_CLEAR(self->encoding);
880 Py_CLEAR(self->encoder);
881 Py_CLEAR(self->decoder);
882 Py_CLEAR(self->readnl);
883 Py_CLEAR(self->decoded_chars);
884 Py_CLEAR(self->pending_bytes);
885 Py_CLEAR(self->snapshot);
886 Py_CLEAR(self->errors);
887 Py_CLEAR(self->raw);
888 self->decoded_chars_used = 0;
889 self->pending_bytes_count = 0;
890 self->encodefunc = NULL;
Antoine Pitrou211b81d2011-02-25 20:27:33 +0000891 self->b2cratio = 0.0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000892
893 if (encoding == NULL) {
894 /* Try os.device_encoding(fileno) */
895 PyObject *fileno;
Antoine Pitrou712cb732013-12-21 15:51:54 +0100896 state = IO_STATE();
897 if (state == NULL)
898 goto error;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200899 fileno = _PyObject_CallMethodId(buffer, &PyId_fileno, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000900 /* Ignore only AttributeError and UnsupportedOperation */
901 if (fileno == NULL) {
902 if (PyErr_ExceptionMatches(PyExc_AttributeError) ||
903 PyErr_ExceptionMatches(state->unsupported_operation)) {
904 PyErr_Clear();
905 }
906 else {
907 goto error;
908 }
909 }
910 else {
Serhiy Storchaka78980432013-01-15 01:12:17 +0200911 int fd = _PyLong_AsInt(fileno);
Brett Cannonefb00c02012-02-29 18:31:31 -0500912 Py_DECREF(fileno);
913 if (fd == -1 && PyErr_Occurred()) {
914 goto error;
915 }
916
917 self->encoding = _Py_device_encoding(fd);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000918 if (self->encoding == NULL)
919 goto error;
920 else if (!PyUnicode_Check(self->encoding))
921 Py_CLEAR(self->encoding);
922 }
923 }
924 if (encoding == NULL && self->encoding == NULL) {
Antoine Pitrou932ff832013-08-01 21:04:50 +0200925 PyObject *locale_module = _PyIO_get_locale_module(state);
926 if (locale_module == NULL)
927 goto catch_ImportError;
928 self->encoding = _PyObject_CallMethodId(
929 locale_module, &PyId_getpreferredencoding, "O", Py_False);
930 Py_DECREF(locale_module);
931 if (self->encoding == NULL) {
932 catch_ImportError:
933 /*
934 Importing locale can raise a ImportError because of
935 _functools, and locale.getpreferredencoding can raise a
936 ImportError if _locale is not available. These will happen
937 during module building.
938 */
939 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
940 PyErr_Clear();
941 self->encoding = PyUnicode_FromString("ascii");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000942 }
Antoine Pitrou932ff832013-08-01 21:04:50 +0200943 else
944 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000945 }
Antoine Pitrou932ff832013-08-01 21:04:50 +0200946 else if (!PyUnicode_Check(self->encoding))
947 Py_CLEAR(self->encoding);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000948 }
Victor Stinnerf6c57832010-05-19 01:17:01 +0000949 if (self->encoding != NULL) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000950 encoding = _PyUnicode_AsString(self->encoding);
Victor Stinnerf6c57832010-05-19 01:17:01 +0000951 if (encoding == NULL)
952 goto error;
953 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000954 else if (encoding != NULL) {
955 self->encoding = PyUnicode_FromString(encoding);
956 if (self->encoding == NULL)
957 goto error;
958 }
959 else {
960 PyErr_SetString(PyExc_IOError,
961 "could not determine default encoding");
962 }
963
Nick Coghlana9b15242014-02-04 22:11:18 +1000964 /* Check we have been asked for a real text encoding */
965 codec_info = _PyCodec_LookupTextEncoding(encoding, "codecs.open()");
966 if (codec_info == NULL) {
967 Py_CLEAR(self->encoding);
968 goto error;
969 }
970
971 /* XXX: Failures beyond this point have the potential to leak elements
972 * of the partially constructed object (like self->encoding)
973 */
974
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000975 if (errors == NULL)
976 errors = "strict";
977 self->errors = PyBytes_FromString(errors);
978 if (self->errors == NULL)
979 goto error;
980
981 self->chunk_size = 8192;
982 self->readuniversal = (newline == NULL || newline[0] == '\0');
983 self->line_buffering = line_buffering;
Antoine Pitroue96ec682011-07-23 21:46:35 +0200984 self->write_through = write_through;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000985 self->readtranslate = (newline == NULL);
986 if (newline) {
987 self->readnl = PyUnicode_FromString(newline);
988 if (self->readnl == NULL)
Nick Coghlana9b15242014-02-04 22:11:18 +1000989 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000990 }
991 self->writetranslate = (newline == NULL || newline[0] != '\0');
992 if (!self->readuniversal && self->readnl) {
993 self->writenl = _PyUnicode_AsString(self->readnl);
Victor Stinnerf6c57832010-05-19 01:17:01 +0000994 if (self->writenl == NULL)
995 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000996 if (!strcmp(self->writenl, "\n"))
997 self->writenl = NULL;
998 }
999#ifdef MS_WINDOWS
1000 else
1001 self->writenl = "\r\n";
1002#endif
1003
1004 /* Build the decoder object */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001005 res = _PyObject_CallMethodId(buffer, &PyId_readable, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001006 if (res == NULL)
1007 goto error;
1008 r = PyObject_IsTrue(res);
1009 Py_DECREF(res);
1010 if (r == -1)
1011 goto error;
1012 if (r == 1) {
Nick Coghlana9b15242014-02-04 22:11:18 +10001013 self->decoder = _PyCodecInfo_GetIncrementalDecoder(codec_info,
1014 errors);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001015 if (self->decoder == NULL)
1016 goto error;
1017
1018 if (self->readuniversal) {
1019 PyObject *incrementalDecoder = PyObject_CallFunction(
1020 (PyObject *)&PyIncrementalNewlineDecoder_Type,
1021 "Oi", self->decoder, (int)self->readtranslate);
1022 if (incrementalDecoder == NULL)
1023 goto error;
1024 Py_CLEAR(self->decoder);
1025 self->decoder = incrementalDecoder;
1026 }
1027 }
1028
1029 /* Build the encoder object */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001030 res = _PyObject_CallMethodId(buffer, &PyId_writable, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001031 if (res == NULL)
1032 goto error;
1033 r = PyObject_IsTrue(res);
1034 Py_DECREF(res);
1035 if (r == -1)
1036 goto error;
1037 if (r == 1) {
Nick Coghlana9b15242014-02-04 22:11:18 +10001038 self->encoder = _PyCodecInfo_GetIncrementalEncoder(codec_info,
1039 errors);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001040 if (self->encoder == NULL)
1041 goto error;
1042 /* Get the normalized named of the codec */
Nick Coghlana9b15242014-02-04 22:11:18 +10001043 res = _PyObject_GetAttrId(codec_info, &PyId_name);
Benjamin Peterson2cfca792009-06-06 20:46:48 +00001044 if (res == NULL) {
1045 if (PyErr_ExceptionMatches(PyExc_AttributeError))
1046 PyErr_Clear();
1047 else
1048 goto error;
1049 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001050 else if (PyUnicode_Check(res)) {
1051 encodefuncentry *e = encodefuncs;
1052 while (e->name != NULL) {
1053 if (!PyUnicode_CompareWithASCIIString(res, e->name)) {
1054 self->encodefunc = e->encodefunc;
1055 break;
1056 }
1057 e++;
1058 }
1059 }
1060 Py_XDECREF(res);
1061 }
1062
Nick Coghlana9b15242014-02-04 22:11:18 +10001063 /* Finished sorting out the codec details */
1064 Py_DECREF(codec_info);
1065
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001066 self->buffer = buffer;
1067 Py_INCREF(buffer);
Antoine Pitrou24f36292009-03-28 22:16:42 +00001068
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001069 if (Py_TYPE(buffer) == &PyBufferedReader_Type ||
1070 Py_TYPE(buffer) == &PyBufferedWriter_Type ||
1071 Py_TYPE(buffer) == &PyBufferedRandom_Type) {
Martin v. Löwis767046a2011-10-14 15:35:36 +02001072 raw = _PyObject_GetAttrId(buffer, &PyId_raw);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001073 /* Cache the raw FileIO object to speed up 'closed' checks */
Benjamin Peterson2cfca792009-06-06 20:46:48 +00001074 if (raw == NULL) {
1075 if (PyErr_ExceptionMatches(PyExc_AttributeError))
1076 PyErr_Clear();
1077 else
1078 goto error;
1079 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001080 else if (Py_TYPE(raw) == &PyFileIO_Type)
1081 self->raw = raw;
1082 else
1083 Py_DECREF(raw);
1084 }
1085
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001086 res = _PyObject_CallMethodId(buffer, &PyId_seekable, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001087 if (res == NULL)
1088 goto error;
Antoine Pitrou6f430e42012-08-15 23:18:25 +02001089 r = PyObject_IsTrue(res);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001090 Py_DECREF(res);
Antoine Pitrou6f430e42012-08-15 23:18:25 +02001091 if (r < 0)
1092 goto error;
1093 self->seekable = self->telling = r;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001094
Martin v. Löwis767046a2011-10-14 15:35:36 +02001095 self->has_read1 = _PyObject_HasAttrId(buffer, &PyId_read1);
Antoine Pitroue96ec682011-07-23 21:46:35 +02001096
Antoine Pitroue4501852009-05-14 18:55:55 +00001097 self->encoding_start_of_stream = 0;
1098 if (self->seekable && self->encoder) {
1099 PyObject *cookieObj;
1100 int cmp;
1101
1102 self->encoding_start_of_stream = 1;
1103
1104 cookieObj = PyObject_CallMethodObjArgs(buffer, _PyIO_str_tell, NULL);
1105 if (cookieObj == NULL)
1106 goto error;
1107
1108 cmp = PyObject_RichCompareBool(cookieObj, _PyIO_zero, Py_EQ);
1109 Py_DECREF(cookieObj);
1110 if (cmp < 0) {
1111 goto error;
1112 }
1113
1114 if (cmp == 0) {
1115 self->encoding_start_of_stream = 0;
1116 res = PyObject_CallMethodObjArgs(self->encoder, _PyIO_str_setstate,
1117 _PyIO_zero, NULL);
1118 if (res == NULL)
1119 goto error;
1120 Py_DECREF(res);
1121 }
1122 }
1123
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001124 self->ok = 1;
1125 return 0;
1126
1127 error:
Nick Coghlana9b15242014-02-04 22:11:18 +10001128 Py_XDECREF(codec_info);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001129 return -1;
1130}
1131
1132static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001133_textiowrapper_clear(textio *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001134{
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001135 self->ok = 0;
1136 Py_CLEAR(self->buffer);
1137 Py_CLEAR(self->encoding);
1138 Py_CLEAR(self->encoder);
1139 Py_CLEAR(self->decoder);
1140 Py_CLEAR(self->readnl);
1141 Py_CLEAR(self->decoded_chars);
1142 Py_CLEAR(self->pending_bytes);
1143 Py_CLEAR(self->snapshot);
1144 Py_CLEAR(self->errors);
1145 Py_CLEAR(self->raw);
1146 return 0;
1147}
1148
1149static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001150textiowrapper_dealloc(textio *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001151{
Antoine Pitrou796564c2013-07-30 19:59:21 +02001152 self->finalizing = 1;
1153 if (_PyIOBase_finalize((PyObject *) self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001154 return;
Antoine Pitrou796564c2013-07-30 19:59:21 +02001155 _textiowrapper_clear(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001156 _PyObject_GC_UNTRACK(self);
1157 if (self->weakreflist != NULL)
1158 PyObject_ClearWeakRefs((PyObject *)self);
1159 Py_CLEAR(self->dict);
1160 Py_TYPE(self)->tp_free((PyObject *)self);
1161}
1162
1163static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001164textiowrapper_traverse(textio *self, visitproc visit, void *arg)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001165{
1166 Py_VISIT(self->buffer);
1167 Py_VISIT(self->encoding);
1168 Py_VISIT(self->encoder);
1169 Py_VISIT(self->decoder);
1170 Py_VISIT(self->readnl);
1171 Py_VISIT(self->decoded_chars);
1172 Py_VISIT(self->pending_bytes);
1173 Py_VISIT(self->snapshot);
1174 Py_VISIT(self->errors);
1175 Py_VISIT(self->raw);
1176
1177 Py_VISIT(self->dict);
1178 return 0;
1179}
1180
1181static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001182textiowrapper_clear(textio *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001183{
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001184 if (_textiowrapper_clear(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001185 return -1;
1186 Py_CLEAR(self->dict);
1187 return 0;
1188}
1189
1190static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001191textiowrapper_closed_get(textio *self, void *context);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001192
1193/* This macro takes some shortcuts to make the common case faster. */
1194#define CHECK_CLOSED(self) \
1195 do { \
1196 int r; \
1197 PyObject *_res; \
1198 if (Py_TYPE(self) == &PyTextIOWrapper_Type) { \
1199 if (self->raw != NULL) \
1200 r = _PyFileIO_closed(self->raw); \
1201 else { \
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001202 _res = textiowrapper_closed_get(self, NULL); \
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001203 if (_res == NULL) \
1204 return NULL; \
1205 r = PyObject_IsTrue(_res); \
1206 Py_DECREF(_res); \
1207 if (r < 0) \
1208 return NULL; \
1209 } \
1210 if (r > 0) { \
1211 PyErr_SetString(PyExc_ValueError, \
1212 "I/O operation on closed file."); \
1213 return NULL; \
1214 } \
1215 } \
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001216 else if (_PyIOBase_check_closed((PyObject *)self, Py_True) == NULL) \
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001217 return NULL; \
1218 } while (0)
1219
1220#define CHECK_INITIALIZED(self) \
1221 if (self->ok <= 0) { \
Benjamin Petersond2e0c792009-05-01 20:40:59 +00001222 if (self->detached) { \
1223 PyErr_SetString(PyExc_ValueError, \
1224 "underlying buffer has been detached"); \
1225 } else { \
1226 PyErr_SetString(PyExc_ValueError, \
1227 "I/O operation on uninitialized object"); \
1228 } \
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001229 return NULL; \
1230 }
1231
1232#define CHECK_INITIALIZED_INT(self) \
1233 if (self->ok <= 0) { \
Benjamin Petersond2e0c792009-05-01 20:40:59 +00001234 if (self->detached) { \
1235 PyErr_SetString(PyExc_ValueError, \
1236 "underlying buffer has been detached"); \
1237 } else { \
1238 PyErr_SetString(PyExc_ValueError, \
1239 "I/O operation on uninitialized object"); \
1240 } \
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001241 return -1; \
1242 }
1243
1244
Benjamin Petersond2e0c792009-05-01 20:40:59 +00001245static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001246textiowrapper_detach(textio *self)
Benjamin Petersond2e0c792009-05-01 20:40:59 +00001247{
1248 PyObject *buffer, *res;
1249 CHECK_INITIALIZED(self);
1250 res = PyObject_CallMethodObjArgs((PyObject *)self, _PyIO_str_flush, NULL);
1251 if (res == NULL)
1252 return NULL;
1253 Py_DECREF(res);
1254 buffer = self->buffer;
1255 self->buffer = NULL;
1256 self->detached = 1;
1257 self->ok = 0;
1258 return buffer;
1259}
1260
Antoine Pitrou24f36292009-03-28 22:16:42 +00001261/* Flush the internal write buffer. This doesn't explicitly flush the
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001262 underlying buffered object, though. */
1263static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001264_textiowrapper_writeflush(textio *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001265{
Amaury Forgeot d'Arcccd686a2009-08-29 23:00:38 +00001266 PyObject *pending, *b, *ret;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001267
1268 if (self->pending_bytes == NULL)
1269 return 0;
Amaury Forgeot d'Arcccd686a2009-08-29 23:00:38 +00001270
1271 pending = self->pending_bytes;
1272 Py_INCREF(pending);
1273 self->pending_bytes_count = 0;
1274 Py_CLEAR(self->pending_bytes);
1275
1276 b = _PyBytes_Join(_PyIO_empty_bytes, pending);
1277 Py_DECREF(pending);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001278 if (b == NULL)
1279 return -1;
Gregory P. Smithb9817b02013-02-01 13:03:39 -08001280 ret = NULL;
1281 do {
1282 ret = PyObject_CallMethodObjArgs(self->buffer,
1283 _PyIO_str_write, b, NULL);
1284 } while (ret == NULL && _PyIO_trap_eintr());
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001285 Py_DECREF(b);
1286 if (ret == NULL)
1287 return -1;
1288 Py_DECREF(ret);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001289 return 0;
1290}
1291
1292static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001293textiowrapper_write(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001294{
1295 PyObject *ret;
1296 PyObject *text; /* owned reference */
1297 PyObject *b;
1298 Py_ssize_t textlen;
1299 int haslf = 0;
Antoine Pitrouc644e7c2014-05-09 00:24:50 +02001300 int needflush = 0, text_needflush = 0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001301
1302 CHECK_INITIALIZED(self);
1303
1304 if (!PyArg_ParseTuple(args, "U:write", &text)) {
1305 return NULL;
1306 }
1307
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001308 if (PyUnicode_READY(text) == -1)
1309 return NULL;
1310
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001311 CHECK_CLOSED(self);
1312
Antoine Pitrou0d739d72010-09-05 23:01:12 +00001313 if (self->encoder == NULL)
1314 return _unsupported("not writable");
Benjamin Peterson81971ea2009-05-14 22:01:31 +00001315
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001316 Py_INCREF(text);
1317
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001318 textlen = PyUnicode_GET_LENGTH(text);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001319
1320 if ((self->writetranslate && self->writenl != NULL) || self->line_buffering)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001321 if (PyUnicode_FindChar(text, '\n', 0, PyUnicode_GET_LENGTH(text), 1) != -1)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001322 haslf = 1;
1323
1324 if (haslf && self->writetranslate && self->writenl != NULL) {
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001325 PyObject *newtext = _PyObject_CallMethodId(
1326 text, &PyId_replace, "ss", "\n", self->writenl);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001327 Py_DECREF(text);
1328 if (newtext == NULL)
1329 return NULL;
1330 text = newtext;
1331 }
1332
Antoine Pitroue96ec682011-07-23 21:46:35 +02001333 if (self->write_through)
Antoine Pitrouc644e7c2014-05-09 00:24:50 +02001334 text_needflush = 1;
1335 if (self->line_buffering &&
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001336 (haslf ||
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001337 PyUnicode_FindChar(text, '\r', 0, PyUnicode_GET_LENGTH(text), 1) != -1))
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001338 needflush = 1;
1339
1340 /* XXX What if we were just reading? */
Antoine Pitroue4501852009-05-14 18:55:55 +00001341 if (self->encodefunc != NULL) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001342 b = (*self->encodefunc)((PyObject *) self, text);
Antoine Pitroue4501852009-05-14 18:55:55 +00001343 self->encoding_start_of_stream = 0;
1344 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001345 else
1346 b = PyObject_CallMethodObjArgs(self->encoder,
1347 _PyIO_str_encode, text, NULL);
1348 Py_DECREF(text);
1349 if (b == NULL)
1350 return NULL;
1351
1352 if (self->pending_bytes == NULL) {
1353 self->pending_bytes = PyList_New(0);
1354 if (self->pending_bytes == NULL) {
1355 Py_DECREF(b);
1356 return NULL;
1357 }
1358 self->pending_bytes_count = 0;
1359 }
1360 if (PyList_Append(self->pending_bytes, b) < 0) {
1361 Py_DECREF(b);
1362 return NULL;
1363 }
1364 self->pending_bytes_count += PyBytes_GET_SIZE(b);
1365 Py_DECREF(b);
Antoine Pitrouc644e7c2014-05-09 00:24:50 +02001366 if (self->pending_bytes_count > self->chunk_size || needflush ||
1367 text_needflush) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001368 if (_textiowrapper_writeflush(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001369 return NULL;
1370 }
Antoine Pitrou24f36292009-03-28 22:16:42 +00001371
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001372 if (needflush) {
1373 ret = PyObject_CallMethodObjArgs(self->buffer, _PyIO_str_flush, NULL);
1374 if (ret == NULL)
1375 return NULL;
1376 Py_DECREF(ret);
1377 }
1378
1379 Py_CLEAR(self->snapshot);
1380
1381 if (self->decoder) {
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001382 ret = _PyObject_CallMethodId(self->decoder, &PyId_reset, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001383 if (ret == NULL)
1384 return NULL;
1385 Py_DECREF(ret);
1386 }
1387
1388 return PyLong_FromSsize_t(textlen);
1389}
1390
1391/* Steal a reference to chars and store it in the decoded_char buffer;
1392 */
1393static void
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001394textiowrapper_set_decoded_chars(textio *self, PyObject *chars)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001395{
1396 Py_CLEAR(self->decoded_chars);
1397 self->decoded_chars = chars;
1398 self->decoded_chars_used = 0;
1399}
1400
1401static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001402textiowrapper_get_decoded_chars(textio *self, Py_ssize_t n)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001403{
1404 PyObject *chars;
1405 Py_ssize_t avail;
1406
1407 if (self->decoded_chars == NULL)
1408 return PyUnicode_FromStringAndSize(NULL, 0);
1409
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001410 /* decoded_chars is guaranteed to be "ready". */
1411 avail = (PyUnicode_GET_LENGTH(self->decoded_chars)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001412 - self->decoded_chars_used);
1413
1414 assert(avail >= 0);
1415
1416 if (n < 0 || n > avail)
1417 n = avail;
1418
1419 if (self->decoded_chars_used > 0 || n < avail) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001420 chars = PyUnicode_Substring(self->decoded_chars,
1421 self->decoded_chars_used,
1422 self->decoded_chars_used + n);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001423 if (chars == NULL)
1424 return NULL;
1425 }
1426 else {
1427 chars = self->decoded_chars;
1428 Py_INCREF(chars);
1429 }
1430
1431 self->decoded_chars_used += n;
1432 return chars;
1433}
1434
1435/* Read and decode the next chunk of data from the BufferedReader.
1436 */
1437static int
Antoine Pitroue5324562011-11-19 00:39:01 +01001438textiowrapper_read_chunk(textio *self, Py_ssize_t size_hint)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001439{
1440 PyObject *dec_buffer = NULL;
1441 PyObject *dec_flags = NULL;
1442 PyObject *input_chunk = NULL;
1443 PyObject *decoded_chars, *chunk_size;
Antoine Pitrou211b81d2011-02-25 20:27:33 +00001444 Py_ssize_t nbytes, nchars;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001445 int eof;
1446
1447 /* The return value is True unless EOF was reached. The decoded string is
1448 * placed in self._decoded_chars (replacing its previous value). The
1449 * entire input chunk is sent to the decoder, though some of it may remain
1450 * buffered in the decoder, yet to be converted.
1451 */
1452
1453 if (self->decoder == NULL) {
Antoine Pitrou0d739d72010-09-05 23:01:12 +00001454 _unsupported("not readable");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001455 return -1;
1456 }
1457
1458 if (self->telling) {
1459 /* To prepare for tell(), we need to snapshot a point in the file
1460 * where the decoder's input buffer is empty.
1461 */
1462
1463 PyObject *state = PyObject_CallMethodObjArgs(self->decoder,
1464 _PyIO_str_getstate, NULL);
1465 if (state == NULL)
1466 return -1;
1467 /* Given this, we know there was a valid snapshot point
1468 * len(dec_buffer) bytes ago with decoder state (b'', dec_flags).
1469 */
1470 if (PyArg_Parse(state, "(OO)", &dec_buffer, &dec_flags) < 0) {
1471 Py_DECREF(state);
1472 return -1;
1473 }
1474 Py_INCREF(dec_buffer);
1475 Py_INCREF(dec_flags);
1476 Py_DECREF(state);
1477 }
1478
1479 /* Read a chunk, decode it, and put the result in self._decoded_chars. */
Antoine Pitroue5324562011-11-19 00:39:01 +01001480 if (size_hint > 0) {
Victor Stinnerf8facac2011-11-22 02:30:47 +01001481 size_hint = (Py_ssize_t)(Py_MAX(self->b2cratio, 1.0) * size_hint);
Antoine Pitroue5324562011-11-19 00:39:01 +01001482 }
1483 chunk_size = PyLong_FromSsize_t(Py_MAX(self->chunk_size, size_hint));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001484 if (chunk_size == NULL)
1485 goto fail;
1486 input_chunk = PyObject_CallMethodObjArgs(self->buffer,
Antoine Pitroue96ec682011-07-23 21:46:35 +02001487 (self->has_read1 ? _PyIO_str_read1: _PyIO_str_read),
1488 chunk_size, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001489 Py_DECREF(chunk_size);
1490 if (input_chunk == NULL)
1491 goto fail;
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02001492 if (!PyBytes_Check(input_chunk)) {
1493 PyErr_Format(PyExc_TypeError,
1494 "underlying %s() should have returned a bytes object, "
1495 "not '%.200s'", (self->has_read1 ? "read1": "read"),
1496 Py_TYPE(input_chunk)->tp_name);
1497 goto fail;
1498 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001499
Antoine Pitrou211b81d2011-02-25 20:27:33 +00001500 nbytes = PyBytes_Size(input_chunk);
1501 eof = (nbytes == 0);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001502
1503 if (Py_TYPE(self->decoder) == &PyIncrementalNewlineDecoder_Type) {
1504 decoded_chars = _PyIncrementalNewlineDecoder_decode(
1505 self->decoder, input_chunk, eof);
1506 }
1507 else {
1508 decoded_chars = PyObject_CallMethodObjArgs(self->decoder,
1509 _PyIO_str_decode, input_chunk, eof ? Py_True : Py_False, NULL);
1510 }
1511
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02001512 if (check_decoded(decoded_chars) < 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001513 goto fail;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001514 textiowrapper_set_decoded_chars(self, decoded_chars);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001515 nchars = PyUnicode_GET_LENGTH(decoded_chars);
Antoine Pitrou211b81d2011-02-25 20:27:33 +00001516 if (nchars > 0)
1517 self->b2cratio = (double) nbytes / nchars;
1518 else
1519 self->b2cratio = 0.0;
1520 if (nchars > 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001521 eof = 0;
1522
1523 if (self->telling) {
1524 /* At the snapshot point, len(dec_buffer) bytes before the read, the
1525 * next input to be decoded is dec_buffer + input_chunk.
1526 */
1527 PyObject *next_input = PyNumber_Add(dec_buffer, input_chunk);
1528 if (next_input == NULL)
1529 goto fail;
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02001530 if (!PyBytes_Check(next_input)) {
1531 PyErr_Format(PyExc_TypeError,
1532 "decoder getstate() should have returned a bytes "
1533 "object, not '%.200s'",
1534 Py_TYPE(next_input)->tp_name);
1535 Py_DECREF(next_input);
1536 goto fail;
1537 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001538 Py_DECREF(dec_buffer);
1539 Py_CLEAR(self->snapshot);
1540 self->snapshot = Py_BuildValue("NN", dec_flags, next_input);
1541 }
1542 Py_DECREF(input_chunk);
1543
1544 return (eof == 0);
1545
1546 fail:
1547 Py_XDECREF(dec_buffer);
1548 Py_XDECREF(dec_flags);
1549 Py_XDECREF(input_chunk);
1550 return -1;
1551}
1552
1553static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001554textiowrapper_read(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001555{
1556 Py_ssize_t n = -1;
1557 PyObject *result = NULL, *chunks = NULL;
1558
1559 CHECK_INITIALIZED(self);
1560
Benjamin Petersonbf5ff762009-12-13 19:25:34 +00001561 if (!PyArg_ParseTuple(args, "|O&:read", &_PyIO_ConvertSsize_t, &n))
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001562 return NULL;
1563
1564 CHECK_CLOSED(self);
1565
Antoine Pitrou0d739d72010-09-05 23:01:12 +00001566 if (self->decoder == NULL)
1567 return _unsupported("not readable");
Benjamin Petersona1b49012009-03-31 23:11:32 +00001568
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001569 if (_textiowrapper_writeflush(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001570 return NULL;
1571
1572 if (n < 0) {
1573 /* Read everything */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001574 PyObject *bytes = _PyObject_CallMethodId(self->buffer, &PyId_read, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001575 PyObject *decoded;
1576 if (bytes == NULL)
1577 goto fail;
Victor Stinnerfd821132011-05-25 22:01:33 +02001578
1579 if (Py_TYPE(self->decoder) == &PyIncrementalNewlineDecoder_Type)
1580 decoded = _PyIncrementalNewlineDecoder_decode(self->decoder,
1581 bytes, 1);
1582 else
1583 decoded = PyObject_CallMethodObjArgs(
1584 self->decoder, _PyIO_str_decode, bytes, Py_True, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001585 Py_DECREF(bytes);
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02001586 if (check_decoded(decoded) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001587 goto fail;
1588
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001589 result = textiowrapper_get_decoded_chars(self, -1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001590
1591 if (result == NULL) {
1592 Py_DECREF(decoded);
1593 return NULL;
1594 }
1595
1596 PyUnicode_AppendAndDel(&result, decoded);
1597 if (result == NULL)
1598 goto fail;
1599
1600 Py_CLEAR(self->snapshot);
1601 return result;
1602 }
1603 else {
1604 int res = 1;
1605 Py_ssize_t remaining = n;
1606
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001607 result = textiowrapper_get_decoded_chars(self, n);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001608 if (result == NULL)
1609 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001610 if (PyUnicode_READY(result) == -1)
1611 goto fail;
1612 remaining -= PyUnicode_GET_LENGTH(result);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001613
1614 /* Keep reading chunks until we have n characters to return */
1615 while (remaining > 0) {
Antoine Pitroue5324562011-11-19 00:39:01 +01001616 res = textiowrapper_read_chunk(self, remaining);
Gregory P. Smith51359922012-06-23 23:55:39 -07001617 if (res < 0) {
1618 /* NOTE: PyErr_SetFromErrno() calls PyErr_CheckSignals()
1619 when EINTR occurs so we needn't do it ourselves. */
1620 if (_PyIO_trap_eintr()) {
1621 continue;
1622 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001623 goto fail;
Gregory P. Smith51359922012-06-23 23:55:39 -07001624 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001625 if (res == 0) /* EOF */
1626 break;
1627 if (chunks == NULL) {
1628 chunks = PyList_New(0);
1629 if (chunks == NULL)
1630 goto fail;
1631 }
Antoine Pitroue5324562011-11-19 00:39:01 +01001632 if (PyUnicode_GET_LENGTH(result) > 0 &&
1633 PyList_Append(chunks, result) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001634 goto fail;
1635 Py_DECREF(result);
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001636 result = textiowrapper_get_decoded_chars(self, remaining);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001637 if (result == NULL)
1638 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001639 remaining -= PyUnicode_GET_LENGTH(result);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001640 }
1641 if (chunks != NULL) {
1642 if (result != NULL && PyList_Append(chunks, result) < 0)
1643 goto fail;
1644 Py_CLEAR(result);
1645 result = PyUnicode_Join(_PyIO_empty_str, chunks);
1646 if (result == NULL)
1647 goto fail;
1648 Py_CLEAR(chunks);
1649 }
1650 return result;
1651 }
1652 fail:
1653 Py_XDECREF(result);
1654 Py_XDECREF(chunks);
1655 return NULL;
1656}
1657
1658
Victor Stinnerf7b8cb62011-09-29 03:28:17 +02001659/* NOTE: `end` must point to the real end of the Py_UCS4 storage,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001660 that is to the NUL character. Otherwise the function will produce
1661 incorrect results. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001662static char *
1663find_control_char(int kind, char *s, char *end, Py_UCS4 ch)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001664{
Antoine Pitrouc28e2e52011-11-13 03:53:42 +01001665 if (kind == PyUnicode_1BYTE_KIND) {
1666 assert(ch < 256);
1667 return (char *) memchr((void *) s, (char) ch, end - s);
1668 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001669 for (;;) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001670 while (PyUnicode_READ(kind, s, 0) > ch)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001671 s += kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001672 if (PyUnicode_READ(kind, s, 0) == ch)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001673 return s;
1674 if (s == end)
1675 return NULL;
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001676 s += kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001677 }
1678}
1679
1680Py_ssize_t
1681_PyIO_find_line_ending(
1682 int translated, int universal, PyObject *readnl,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001683 int kind, char *start, char *end, Py_ssize_t *consumed)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001684{
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001685 Py_ssize_t len = ((char*)end - (char*)start)/kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001686
1687 if (translated) {
1688 /* Newlines are already translated, only search for \n */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001689 char *pos = find_control_char(kind, start, end, '\n');
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001690 if (pos != NULL)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001691 return (pos - start)/kind + 1;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001692 else {
1693 *consumed = len;
1694 return -1;
1695 }
1696 }
1697 else if (universal) {
1698 /* Universal newline search. Find any of \r, \r\n, \n
1699 * The decoder ensures that \r\n are not split in two pieces
1700 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001701 char *s = start;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001702 for (;;) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001703 Py_UCS4 ch;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001704 /* Fast path for non-control chars. The loop always ends
Victor Stinnerf7b8cb62011-09-29 03:28:17 +02001705 since the Unicode string is NUL-terminated. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001706 while (PyUnicode_READ(kind, s, 0) > '\r')
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001707 s += kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001708 if (s >= end) {
1709 *consumed = len;
1710 return -1;
1711 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001712 ch = PyUnicode_READ(kind, s, 0);
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001713 s += kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001714 if (ch == '\n')
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001715 return (s - start)/kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001716 if (ch == '\r') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001717 if (PyUnicode_READ(kind, s, 0) == '\n')
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001718 return (s - start)/kind + 1;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001719 else
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001720 return (s - start)/kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001721 }
1722 }
1723 }
1724 else {
1725 /* Non-universal mode. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001726 Py_ssize_t readnl_len = PyUnicode_GET_LENGTH(readnl);
1727 char *nl = PyUnicode_DATA(readnl);
1728 /* Assume that readnl is an ASCII character. */
1729 assert(PyUnicode_KIND(readnl) == PyUnicode_1BYTE_KIND);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001730 if (readnl_len == 1) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001731 char *pos = find_control_char(kind, start, end, nl[0]);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001732 if (pos != NULL)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001733 return (pos - start)/kind + 1;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001734 *consumed = len;
1735 return -1;
1736 }
1737 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001738 char *s = start;
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001739 char *e = end - (readnl_len - 1)*kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001740 char *pos;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001741 if (e < s)
1742 e = s;
1743 while (s < e) {
1744 Py_ssize_t i;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001745 char *pos = find_control_char(kind, s, end, nl[0]);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001746 if (pos == NULL || pos >= e)
1747 break;
1748 for (i = 1; i < readnl_len; i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001749 if (PyUnicode_READ(kind, pos, i) != nl[i])
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001750 break;
1751 }
1752 if (i == readnl_len)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001753 return (pos - start)/kind + readnl_len;
1754 s = pos + kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001755 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001756 pos = find_control_char(kind, e, end, nl[0]);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001757 if (pos == NULL)
1758 *consumed = len;
1759 else
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001760 *consumed = (pos - start)/kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001761 return -1;
1762 }
1763 }
1764}
1765
1766static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001767_textiowrapper_readline(textio *self, Py_ssize_t limit)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001768{
1769 PyObject *line = NULL, *chunks = NULL, *remaining = NULL;
1770 Py_ssize_t start, endpos, chunked, offset_to_buffer;
1771 int res;
1772
1773 CHECK_CLOSED(self);
1774
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001775 if (_textiowrapper_writeflush(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001776 return NULL;
1777
1778 chunked = 0;
1779
1780 while (1) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001781 char *ptr;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001782 Py_ssize_t line_len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001783 int kind;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001784 Py_ssize_t consumed = 0;
1785
1786 /* First, get some data if necessary */
1787 res = 1;
1788 while (!self->decoded_chars ||
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001789 !PyUnicode_GET_LENGTH(self->decoded_chars)) {
Antoine Pitroue5324562011-11-19 00:39:01 +01001790 res = textiowrapper_read_chunk(self, 0);
Gregory P. Smith51359922012-06-23 23:55:39 -07001791 if (res < 0) {
1792 /* NOTE: PyErr_SetFromErrno() calls PyErr_CheckSignals()
1793 when EINTR occurs so we needn't do it ourselves. */
1794 if (_PyIO_trap_eintr()) {
1795 continue;
1796 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001797 goto error;
Gregory P. Smith51359922012-06-23 23:55:39 -07001798 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001799 if (res == 0)
1800 break;
1801 }
1802 if (res == 0) {
1803 /* end of file */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001804 textiowrapper_set_decoded_chars(self, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001805 Py_CLEAR(self->snapshot);
1806 start = endpos = offset_to_buffer = 0;
1807 break;
1808 }
1809
1810 if (remaining == NULL) {
1811 line = self->decoded_chars;
1812 start = self->decoded_chars_used;
1813 offset_to_buffer = 0;
1814 Py_INCREF(line);
1815 }
1816 else {
1817 assert(self->decoded_chars_used == 0);
1818 line = PyUnicode_Concat(remaining, self->decoded_chars);
1819 start = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001820 offset_to_buffer = PyUnicode_GET_LENGTH(remaining);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001821 Py_CLEAR(remaining);
1822 if (line == NULL)
1823 goto error;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001824 if (PyUnicode_READY(line) == -1)
1825 goto error;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001826 }
1827
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001828 ptr = PyUnicode_DATA(line);
1829 line_len = PyUnicode_GET_LENGTH(line);
1830 kind = PyUnicode_KIND(line);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001831
1832 endpos = _PyIO_find_line_ending(
1833 self->readtranslate, self->readuniversal, self->readnl,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001834 kind,
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001835 ptr + kind * start,
1836 ptr + kind * line_len,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001837 &consumed);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001838 if (endpos >= 0) {
1839 endpos += start;
1840 if (limit >= 0 && (endpos - start) + chunked >= limit)
1841 endpos = start + limit - chunked;
1842 break;
1843 }
1844
1845 /* We can put aside up to `endpos` */
1846 endpos = consumed + start;
1847 if (limit >= 0 && (endpos - start) + chunked >= limit) {
1848 /* Didn't find line ending, but reached length limit */
1849 endpos = start + limit - chunked;
1850 break;
1851 }
1852
1853 if (endpos > start) {
1854 /* No line ending seen yet - put aside current data */
1855 PyObject *s;
1856 if (chunks == NULL) {
1857 chunks = PyList_New(0);
1858 if (chunks == NULL)
1859 goto error;
1860 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001861 s = PyUnicode_Substring(line, start, endpos);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001862 if (s == NULL)
1863 goto error;
1864 if (PyList_Append(chunks, s) < 0) {
1865 Py_DECREF(s);
1866 goto error;
1867 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001868 chunked += PyUnicode_GET_LENGTH(s);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001869 Py_DECREF(s);
1870 }
1871 /* There may be some remaining bytes we'll have to prepend to the
1872 next chunk of data */
1873 if (endpos < line_len) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001874 remaining = PyUnicode_Substring(line, endpos, line_len);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001875 if (remaining == NULL)
1876 goto error;
1877 }
1878 Py_CLEAR(line);
1879 /* We have consumed the buffer */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001880 textiowrapper_set_decoded_chars(self, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001881 }
1882
1883 if (line != NULL) {
1884 /* Our line ends in the current buffer */
1885 self->decoded_chars_used = endpos - offset_to_buffer;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001886 if (start > 0 || endpos < PyUnicode_GET_LENGTH(line)) {
1887 PyObject *s = PyUnicode_Substring(line, start, endpos);
1888 Py_CLEAR(line);
1889 if (s == NULL)
1890 goto error;
1891 line = s;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001892 }
1893 }
1894 if (remaining != NULL) {
1895 if (chunks == NULL) {
1896 chunks = PyList_New(0);
1897 if (chunks == NULL)
1898 goto error;
1899 }
1900 if (PyList_Append(chunks, remaining) < 0)
1901 goto error;
1902 Py_CLEAR(remaining);
1903 }
1904 if (chunks != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001905 if (line != NULL) {
1906 if (PyList_Append(chunks, line) < 0)
1907 goto error;
1908 Py_DECREF(line);
1909 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001910 line = PyUnicode_Join(_PyIO_empty_str, chunks);
1911 if (line == NULL)
1912 goto error;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001913 Py_CLEAR(chunks);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001914 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001915 if (line == NULL) {
1916 Py_INCREF(_PyIO_empty_str);
1917 line = _PyIO_empty_str;
1918 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001919
1920 return line;
1921
1922 error:
1923 Py_XDECREF(chunks);
1924 Py_XDECREF(remaining);
1925 Py_XDECREF(line);
1926 return NULL;
1927}
1928
1929static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001930textiowrapper_readline(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001931{
1932 Py_ssize_t limit = -1;
1933
1934 CHECK_INITIALIZED(self);
1935 if (!PyArg_ParseTuple(args, "|n:readline", &limit)) {
1936 return NULL;
1937 }
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001938 return _textiowrapper_readline(self, limit);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001939}
1940
1941/* Seek and Tell */
1942
1943typedef struct {
1944 Py_off_t start_pos;
1945 int dec_flags;
1946 int bytes_to_feed;
1947 int chars_to_skip;
1948 char need_eof;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001949} cookie_type;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001950
1951/*
1952 To speed up cookie packing/unpacking, we store the fields in a temporary
1953 string and call _PyLong_FromByteArray() or _PyLong_AsByteArray (resp.).
1954 The following macros define at which offsets in the intermediary byte
1955 string the various CookieStruct fields will be stored.
1956 */
1957
1958#define COOKIE_BUF_LEN (sizeof(Py_off_t) + 3 * sizeof(int) + sizeof(char))
1959
Christian Heimes743e0cd2012-10-17 23:52:17 +02001960#if PY_BIG_ENDIAN
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001961/* We want the least significant byte of start_pos to also be the least
1962 significant byte of the cookie, which means that in big-endian mode we
1963 must copy the fields in reverse order. */
1964
1965# define OFF_START_POS (sizeof(char) + 3 * sizeof(int))
1966# define OFF_DEC_FLAGS (sizeof(char) + 2 * sizeof(int))
1967# define OFF_BYTES_TO_FEED (sizeof(char) + sizeof(int))
1968# define OFF_CHARS_TO_SKIP (sizeof(char))
1969# define OFF_NEED_EOF 0
1970
1971#else
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001972/* Little-endian mode: the least significant byte of start_pos will
1973 naturally end up the least significant byte of the cookie. */
1974
1975# define OFF_START_POS 0
1976# define OFF_DEC_FLAGS (sizeof(Py_off_t))
1977# define OFF_BYTES_TO_FEED (sizeof(Py_off_t) + sizeof(int))
1978# define OFF_CHARS_TO_SKIP (sizeof(Py_off_t) + 2 * sizeof(int))
1979# define OFF_NEED_EOF (sizeof(Py_off_t) + 3 * sizeof(int))
1980
1981#endif
1982
1983static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00001984textiowrapper_parse_cookie(cookie_type *cookie, PyObject *cookieObj)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001985{
1986 unsigned char buffer[COOKIE_BUF_LEN];
1987 PyLongObject *cookieLong = (PyLongObject *)PyNumber_Long(cookieObj);
1988 if (cookieLong == NULL)
1989 return -1;
1990
1991 if (_PyLong_AsByteArray(cookieLong, buffer, sizeof(buffer),
Christian Heimes743e0cd2012-10-17 23:52:17 +02001992 PY_LITTLE_ENDIAN, 0) < 0) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001993 Py_DECREF(cookieLong);
1994 return -1;
1995 }
1996 Py_DECREF(cookieLong);
1997
Antoine Pitrou2db74c22009-03-06 21:49:02 +00001998 memcpy(&cookie->start_pos, buffer + OFF_START_POS, sizeof(cookie->start_pos));
1999 memcpy(&cookie->dec_flags, buffer + OFF_DEC_FLAGS, sizeof(cookie->dec_flags));
2000 memcpy(&cookie->bytes_to_feed, buffer + OFF_BYTES_TO_FEED, sizeof(cookie->bytes_to_feed));
2001 memcpy(&cookie->chars_to_skip, buffer + OFF_CHARS_TO_SKIP, sizeof(cookie->chars_to_skip));
2002 memcpy(&cookie->need_eof, buffer + OFF_NEED_EOF, sizeof(cookie->need_eof));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002003
2004 return 0;
2005}
2006
2007static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002008textiowrapper_build_cookie(cookie_type *cookie)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002009{
2010 unsigned char buffer[COOKIE_BUF_LEN];
2011
Antoine Pitrou2db74c22009-03-06 21:49:02 +00002012 memcpy(buffer + OFF_START_POS, &cookie->start_pos, sizeof(cookie->start_pos));
2013 memcpy(buffer + OFF_DEC_FLAGS, &cookie->dec_flags, sizeof(cookie->dec_flags));
2014 memcpy(buffer + OFF_BYTES_TO_FEED, &cookie->bytes_to_feed, sizeof(cookie->bytes_to_feed));
2015 memcpy(buffer + OFF_CHARS_TO_SKIP, &cookie->chars_to_skip, sizeof(cookie->chars_to_skip));
2016 memcpy(buffer + OFF_NEED_EOF, &cookie->need_eof, sizeof(cookie->need_eof));
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002017
Christian Heimes743e0cd2012-10-17 23:52:17 +02002018 return _PyLong_FromByteArray(buffer, sizeof(buffer),
2019 PY_LITTLE_ENDIAN, 0);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002020}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002021
2022static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002023_textiowrapper_decoder_setstate(textio *self, cookie_type *cookie)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002024{
2025 PyObject *res;
2026 /* When seeking to the start of the stream, we call decoder.reset()
2027 rather than decoder.getstate().
2028 This is for a few decoders such as utf-16 for which the state value
2029 at start is not (b"", 0) but e.g. (b"", 2) (meaning, in the case of
2030 utf-16, that we are expecting a BOM).
2031 */
2032 if (cookie->start_pos == 0 && cookie->dec_flags == 0)
2033 res = PyObject_CallMethodObjArgs(self->decoder, _PyIO_str_reset, NULL);
2034 else
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002035 res = _PyObject_CallMethodId(self->decoder, &PyId_setstate,
2036 "((yi))", "", cookie->dec_flags);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002037 if (res == NULL)
2038 return -1;
2039 Py_DECREF(res);
2040 return 0;
2041}
2042
Antoine Pitroue4501852009-05-14 18:55:55 +00002043static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002044_textiowrapper_encoder_setstate(textio *self, cookie_type *cookie)
Antoine Pitroue4501852009-05-14 18:55:55 +00002045{
2046 PyObject *res;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002047 /* Same as _textiowrapper_decoder_setstate() above. */
Antoine Pitroue4501852009-05-14 18:55:55 +00002048 if (cookie->start_pos == 0 && cookie->dec_flags == 0) {
2049 res = PyObject_CallMethodObjArgs(self->encoder, _PyIO_str_reset, NULL);
2050 self->encoding_start_of_stream = 1;
2051 }
2052 else {
2053 res = PyObject_CallMethodObjArgs(self->encoder, _PyIO_str_setstate,
2054 _PyIO_zero, NULL);
2055 self->encoding_start_of_stream = 0;
2056 }
2057 if (res == NULL)
2058 return -1;
2059 Py_DECREF(res);
2060 return 0;
2061}
2062
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002063static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002064textiowrapper_seek(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002065{
2066 PyObject *cookieObj, *posobj;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002067 cookie_type cookie;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002068 int whence = 0;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002069 PyObject *res;
2070 int cmp;
2071
2072 CHECK_INITIALIZED(self);
2073
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002074 if (!PyArg_ParseTuple(args, "O|i:seek", &cookieObj, &whence))
2075 return NULL;
2076 CHECK_CLOSED(self);
2077
2078 Py_INCREF(cookieObj);
2079
2080 if (!self->seekable) {
Antoine Pitrou0d739d72010-09-05 23:01:12 +00002081 _unsupported("underlying stream is not seekable");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002082 goto fail;
2083 }
2084
2085 if (whence == 1) {
2086 /* seek relative to current position */
Antoine Pitroue4501852009-05-14 18:55:55 +00002087 cmp = PyObject_RichCompareBool(cookieObj, _PyIO_zero, Py_EQ);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002088 if (cmp < 0)
2089 goto fail;
2090
2091 if (cmp == 0) {
Antoine Pitrou0d739d72010-09-05 23:01:12 +00002092 _unsupported("can't do nonzero cur-relative seeks");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002093 goto fail;
2094 }
2095
2096 /* Seeking to the current position should attempt to
2097 * sync the underlying buffer with the current position.
2098 */
2099 Py_DECREF(cookieObj);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002100 cookieObj = _PyObject_CallMethodId((PyObject *)self, &PyId_tell, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002101 if (cookieObj == NULL)
2102 goto fail;
2103 }
2104 else if (whence == 2) {
2105 /* seek relative to end of file */
Antoine Pitroue4501852009-05-14 18:55:55 +00002106 cmp = PyObject_RichCompareBool(cookieObj, _PyIO_zero, Py_EQ);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002107 if (cmp < 0)
2108 goto fail;
2109
2110 if (cmp == 0) {
Antoine Pitrou0d739d72010-09-05 23:01:12 +00002111 _unsupported("can't do nonzero end-relative seeks");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002112 goto fail;
2113 }
2114
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002115 res = _PyObject_CallMethodId((PyObject *)self, &PyId_flush, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002116 if (res == NULL)
2117 goto fail;
2118 Py_DECREF(res);
2119
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002120 textiowrapper_set_decoded_chars(self, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002121 Py_CLEAR(self->snapshot);
2122 if (self->decoder) {
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002123 res = _PyObject_CallMethodId(self->decoder, &PyId_reset, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002124 if (res == NULL)
2125 goto fail;
2126 Py_DECREF(res);
2127 }
2128
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002129 res = _PyObject_CallMethodId(self->buffer, &PyId_seek, "ii", 0, 2);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002130 Py_XDECREF(cookieObj);
2131 return res;
2132 }
2133 else if (whence != 0) {
2134 PyErr_Format(PyExc_ValueError,
2135 "invalid whence (%d, should be 0, 1 or 2)", whence);
2136 goto fail;
2137 }
2138
Antoine Pitroue4501852009-05-14 18:55:55 +00002139 cmp = PyObject_RichCompareBool(cookieObj, _PyIO_zero, Py_LT);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002140 if (cmp < 0)
2141 goto fail;
2142
2143 if (cmp == 1) {
2144 PyErr_Format(PyExc_ValueError,
2145 "negative seek position %R", cookieObj);
2146 goto fail;
2147 }
2148
2149 res = PyObject_CallMethodObjArgs((PyObject *)self, _PyIO_str_flush, NULL);
2150 if (res == NULL)
2151 goto fail;
2152 Py_DECREF(res);
2153
2154 /* The strategy of seek() is to go back to the safe start point
2155 * and replay the effect of read(chars_to_skip) from there.
2156 */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002157 if (textiowrapper_parse_cookie(&cookie, cookieObj) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002158 goto fail;
2159
2160 /* Seek back to the safe start point. */
2161 posobj = PyLong_FromOff_t(cookie.start_pos);
2162 if (posobj == NULL)
2163 goto fail;
2164 res = PyObject_CallMethodObjArgs(self->buffer,
2165 _PyIO_str_seek, posobj, NULL);
2166 Py_DECREF(posobj);
2167 if (res == NULL)
2168 goto fail;
2169 Py_DECREF(res);
2170
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002171 textiowrapper_set_decoded_chars(self, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002172 Py_CLEAR(self->snapshot);
2173
2174 /* Restore the decoder to its state from the safe start point. */
2175 if (self->decoder) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002176 if (_textiowrapper_decoder_setstate(self, &cookie) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002177 goto fail;
2178 }
2179
2180 if (cookie.chars_to_skip) {
2181 /* Just like _read_chunk, feed the decoder and save a snapshot. */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002182 PyObject *input_chunk = _PyObject_CallMethodId(
2183 self->buffer, &PyId_read, "i", cookie.bytes_to_feed);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002184 PyObject *decoded;
2185
2186 if (input_chunk == NULL)
2187 goto fail;
2188
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02002189 if (!PyBytes_Check(input_chunk)) {
2190 PyErr_Format(PyExc_TypeError,
2191 "underlying read() should have returned a bytes "
2192 "object, not '%.200s'",
2193 Py_TYPE(input_chunk)->tp_name);
2194 Py_DECREF(input_chunk);
2195 goto fail;
2196 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002197
2198 self->snapshot = Py_BuildValue("iN", cookie.dec_flags, input_chunk);
2199 if (self->snapshot == NULL) {
2200 Py_DECREF(input_chunk);
2201 goto fail;
2202 }
2203
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002204 decoded = _PyObject_CallMethodId(self->decoder, &PyId_decode,
2205 "Oi", input_chunk, (int)cookie.need_eof);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002206
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02002207 if (check_decoded(decoded) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002208 goto fail;
2209
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002210 textiowrapper_set_decoded_chars(self, decoded);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002211
2212 /* Skip chars_to_skip of the decoded characters. */
Victor Stinner9e30aa52011-11-21 02:49:52 +01002213 if (PyUnicode_GetLength(self->decoded_chars) < cookie.chars_to_skip) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002214 PyErr_SetString(PyExc_IOError, "can't restore logical file position");
2215 goto fail;
2216 }
2217 self->decoded_chars_used = cookie.chars_to_skip;
2218 }
2219 else {
2220 self->snapshot = Py_BuildValue("iy", cookie.dec_flags, "");
2221 if (self->snapshot == NULL)
2222 goto fail;
2223 }
2224
Antoine Pitroue4501852009-05-14 18:55:55 +00002225 /* Finally, reset the encoder (merely useful for proper BOM handling) */
2226 if (self->encoder) {
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002227 if (_textiowrapper_encoder_setstate(self, &cookie) < 0)
Antoine Pitroue4501852009-05-14 18:55:55 +00002228 goto fail;
2229 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002230 return cookieObj;
2231 fail:
2232 Py_XDECREF(cookieObj);
2233 return NULL;
2234
2235}
2236
2237static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002238textiowrapper_tell(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002239{
2240 PyObject *res;
2241 PyObject *posobj = NULL;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002242 cookie_type cookie = {0,0,0,0,0};
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002243 PyObject *next_input;
2244 Py_ssize_t chars_to_skip, chars_decoded;
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002245 Py_ssize_t skip_bytes, skip_back;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002246 PyObject *saved_state = NULL;
2247 char *input, *input_end;
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002248 char *dec_buffer;
2249 Py_ssize_t dec_buffer_len;
2250 int dec_flags;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002251
2252 CHECK_INITIALIZED(self);
2253 CHECK_CLOSED(self);
2254
2255 if (!self->seekable) {
Antoine Pitrou0d739d72010-09-05 23:01:12 +00002256 _unsupported("underlying stream is not seekable");
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002257 goto fail;
2258 }
2259 if (!self->telling) {
2260 PyErr_SetString(PyExc_IOError,
2261 "telling position disabled by next() call");
2262 goto fail;
2263 }
2264
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002265 if (_textiowrapper_writeflush(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002266 return NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002267 res = _PyObject_CallMethodId((PyObject *)self, &PyId_flush, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002268 if (res == NULL)
2269 goto fail;
2270 Py_DECREF(res);
2271
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002272 posobj = _PyObject_CallMethodId(self->buffer, &PyId_tell, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002273 if (posobj == NULL)
2274 goto fail;
2275
2276 if (self->decoder == NULL || self->snapshot == NULL) {
Victor Stinner9e30aa52011-11-21 02:49:52 +01002277 assert (self->decoded_chars == NULL || PyUnicode_GetLength(self->decoded_chars) == 0);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002278 return posobj;
2279 }
2280
2281#if defined(HAVE_LARGEFILE_SUPPORT)
2282 cookie.start_pos = PyLong_AsLongLong(posobj);
2283#else
2284 cookie.start_pos = PyLong_AsLong(posobj);
2285#endif
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002286 Py_DECREF(posobj);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002287 if (PyErr_Occurred())
2288 goto fail;
2289
2290 /* Skip backward to the snapshot point (see _read_chunk). */
2291 if (!PyArg_Parse(self->snapshot, "(iO)", &cookie.dec_flags, &next_input))
2292 goto fail;
2293
2294 assert (PyBytes_Check(next_input));
2295
2296 cookie.start_pos -= PyBytes_GET_SIZE(next_input);
2297
2298 /* How many decoded characters have been used up since the snapshot? */
2299 if (self->decoded_chars_used == 0) {
2300 /* We haven't moved from the snapshot point. */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002301 return textiowrapper_build_cookie(&cookie);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002302 }
2303
2304 chars_to_skip = self->decoded_chars_used;
2305
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002306 /* Decoder state will be restored at the end */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002307 saved_state = PyObject_CallMethodObjArgs(self->decoder,
2308 _PyIO_str_getstate, NULL);
2309 if (saved_state == NULL)
2310 goto fail;
2311
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002312#define DECODER_GETSTATE() do { \
2313 PyObject *_state = PyObject_CallMethodObjArgs(self->decoder, \
2314 _PyIO_str_getstate, NULL); \
2315 if (_state == NULL) \
2316 goto fail; \
2317 if (!PyArg_Parse(_state, "(y#i)", &dec_buffer, &dec_buffer_len, &dec_flags)) { \
2318 Py_DECREF(_state); \
2319 goto fail; \
2320 } \
2321 Py_DECREF(_state); \
2322 } while (0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002323
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002324#define DECODER_DECODE(start, len, res) do { \
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002325 PyObject *_decoded = _PyObject_CallMethodId( \
2326 self->decoder, &PyId_decode, "y#", start, len); \
Serhiy Storchakad03ce4a2013-02-03 17:07:32 +02002327 if (check_decoded(_decoded) < 0) \
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002328 goto fail; \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002329 res = PyUnicode_GET_LENGTH(_decoded); \
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002330 Py_DECREF(_decoded); \
2331 } while (0)
2332
2333 /* Fast search for an acceptable start point, close to our
2334 current pos */
2335 skip_bytes = (Py_ssize_t) (self->b2cratio * chars_to_skip);
2336 skip_back = 1;
2337 assert(skip_back <= PyBytes_GET_SIZE(next_input));
2338 input = PyBytes_AS_STRING(next_input);
2339 while (skip_bytes > 0) {
2340 /* Decode up to temptative start point */
2341 if (_textiowrapper_decoder_setstate(self, &cookie) < 0)
2342 goto fail;
2343 DECODER_DECODE(input, skip_bytes, chars_decoded);
2344 if (chars_decoded <= chars_to_skip) {
2345 DECODER_GETSTATE();
2346 if (dec_buffer_len == 0) {
2347 /* Before pos and no bytes buffered in decoder => OK */
2348 cookie.dec_flags = dec_flags;
2349 chars_to_skip -= chars_decoded;
2350 break;
2351 }
2352 /* Skip back by buffered amount and reset heuristic */
2353 skip_bytes -= dec_buffer_len;
2354 skip_back = 1;
2355 }
2356 else {
2357 /* We're too far ahead, skip back a bit */
2358 skip_bytes -= skip_back;
2359 skip_back *= 2;
2360 }
2361 }
2362 if (skip_bytes <= 0) {
2363 skip_bytes = 0;
2364 if (_textiowrapper_decoder_setstate(self, &cookie) < 0)
2365 goto fail;
2366 }
2367
2368 /* Note our initial start point. */
2369 cookie.start_pos += skip_bytes;
Victor Stinner9a282972013-06-24 23:01:33 +02002370 cookie.chars_to_skip = Py_SAFE_DOWNCAST(chars_to_skip, Py_ssize_t, int);
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002371 if (chars_to_skip == 0)
2372 goto finally;
2373
2374 /* We should be close to the desired position. Now feed the decoder one
2375 * byte at a time until we reach the `chars_to_skip` target.
2376 * As we go, note the nearest "safe start point" before the current
2377 * location (a point where the decoder has nothing buffered, so seek()
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002378 * can safely start from there and advance to this location).
2379 */
2380 chars_decoded = 0;
2381 input = PyBytes_AS_STRING(next_input);
2382 input_end = input + PyBytes_GET_SIZE(next_input);
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002383 input += skip_bytes;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002384 while (input < input_end) {
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002385 Py_ssize_t n;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002386
Serhiy Storchakaec67d182013-08-20 20:04:47 +03002387 DECODER_DECODE(input, (Py_ssize_t)1, n);
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002388 /* We got n chars for 1 byte */
2389 chars_decoded += n;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002390 cookie.bytes_to_feed += 1;
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002391 DECODER_GETSTATE();
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002392
2393 if (dec_buffer_len == 0 && chars_decoded <= chars_to_skip) {
2394 /* Decoder buffer is empty, so this is a safe start point. */
2395 cookie.start_pos += cookie.bytes_to_feed;
2396 chars_to_skip -= chars_decoded;
2397 cookie.dec_flags = dec_flags;
2398 cookie.bytes_to_feed = 0;
2399 chars_decoded = 0;
2400 }
2401 if (chars_decoded >= chars_to_skip)
2402 break;
2403 input++;
2404 }
2405 if (input == input_end) {
2406 /* We didn't get enough decoded data; signal EOF to get more. */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002407 PyObject *decoded = _PyObject_CallMethodId(
2408 self->decoder, &PyId_decode, "yi", "", /* final = */ 1);
Serhiy Storchaka94dc6732013-02-03 17:03:31 +02002409 if (check_decoded(decoded) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002410 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002411 chars_decoded += PyUnicode_GET_LENGTH(decoded);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002412 Py_DECREF(decoded);
2413 cookie.need_eof = 1;
2414
2415 if (chars_decoded < chars_to_skip) {
2416 PyErr_SetString(PyExc_IOError,
2417 "can't reconstruct logical file position");
2418 goto fail;
2419 }
2420 }
2421
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002422finally:
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002423 res = _PyObject_CallMethodId(self->decoder, &PyId_setstate, "(O)", saved_state);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002424 Py_DECREF(saved_state);
2425 if (res == NULL)
2426 return NULL;
2427 Py_DECREF(res);
2428
2429 /* The returned cookie corresponds to the last safe start point. */
2430 cookie.chars_to_skip = Py_SAFE_DOWNCAST(chars_to_skip, Py_ssize_t, int);
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002431 return textiowrapper_build_cookie(&cookie);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002432
Antoine Pitrou211b81d2011-02-25 20:27:33 +00002433fail:
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002434 if (saved_state) {
2435 PyObject *type, *value, *traceback;
2436 PyErr_Fetch(&type, &value, &traceback);
2437
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002438 res = _PyObject_CallMethodId(self->decoder, &PyId_setstate, "(O)", saved_state);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002439 Py_DECREF(saved_state);
2440 if (res == NULL)
2441 return NULL;
2442 Py_DECREF(res);
2443
2444 PyErr_Restore(type, value, traceback);
2445 }
2446 return NULL;
2447}
2448
2449static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002450textiowrapper_truncate(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002451{
2452 PyObject *pos = Py_None;
2453 PyObject *res;
2454
2455 CHECK_INITIALIZED(self)
2456 if (!PyArg_ParseTuple(args, "|O:truncate", &pos)) {
2457 return NULL;
2458 }
2459
2460 res = PyObject_CallMethodObjArgs((PyObject *) self, _PyIO_str_flush, NULL);
2461 if (res == NULL)
2462 return NULL;
2463 Py_DECREF(res);
2464
Antoine Pitrou905a2ff2010-01-31 22:47:27 +00002465 return PyObject_CallMethodObjArgs(self->buffer, _PyIO_str_truncate, pos, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002466}
2467
Benjamin Petersonc4c0eae2009-03-09 00:07:03 +00002468static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002469textiowrapper_repr(textio *self)
Benjamin Petersonc4c0eae2009-03-09 00:07:03 +00002470{
Antoine Pitroua4815ca2011-01-09 20:38:15 +00002471 PyObject *nameobj, *modeobj, *res, *s;
Antoine Pitrou716c4442009-05-23 19:04:03 +00002472
2473 CHECK_INITIALIZED(self);
2474
Antoine Pitroua4815ca2011-01-09 20:38:15 +00002475 res = PyUnicode_FromString("<_io.TextIOWrapper");
2476 if (res == NULL)
2477 return NULL;
Martin v. Löwis767046a2011-10-14 15:35:36 +02002478 nameobj = _PyObject_GetAttrId((PyObject *) self, &PyId_name);
Antoine Pitrou716c4442009-05-23 19:04:03 +00002479 if (nameobj == NULL) {
2480 if (PyErr_ExceptionMatches(PyExc_AttributeError))
2481 PyErr_Clear();
2482 else
Antoine Pitroua4815ca2011-01-09 20:38:15 +00002483 goto error;
Antoine Pitrou716c4442009-05-23 19:04:03 +00002484 }
2485 else {
Antoine Pitroua4815ca2011-01-09 20:38:15 +00002486 s = PyUnicode_FromFormat(" name=%R", nameobj);
Antoine Pitrou716c4442009-05-23 19:04:03 +00002487 Py_DECREF(nameobj);
Antoine Pitroua4815ca2011-01-09 20:38:15 +00002488 if (s == NULL)
2489 goto error;
2490 PyUnicode_AppendAndDel(&res, s);
2491 if (res == NULL)
2492 return NULL;
Antoine Pitrou716c4442009-05-23 19:04:03 +00002493 }
Martin v. Löwis767046a2011-10-14 15:35:36 +02002494 modeobj = _PyObject_GetAttrId((PyObject *) self, &PyId_mode);
Antoine Pitroua4815ca2011-01-09 20:38:15 +00002495 if (modeobj == NULL) {
2496 if (PyErr_ExceptionMatches(PyExc_AttributeError))
2497 PyErr_Clear();
2498 else
2499 goto error;
2500 }
2501 else {
2502 s = PyUnicode_FromFormat(" mode=%R", modeobj);
2503 Py_DECREF(modeobj);
2504 if (s == NULL)
2505 goto error;
2506 PyUnicode_AppendAndDel(&res, s);
2507 if (res == NULL)
2508 return NULL;
2509 }
2510 s = PyUnicode_FromFormat("%U encoding=%R>",
2511 res, self->encoding);
2512 Py_DECREF(res);
2513 return s;
2514error:
2515 Py_XDECREF(res);
2516 return NULL;
Benjamin Petersonc4c0eae2009-03-09 00:07:03 +00002517}
2518
2519
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002520/* Inquiries */
2521
2522static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002523textiowrapper_fileno(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002524{
2525 CHECK_INITIALIZED(self);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002526 return _PyObject_CallMethodId(self->buffer, &PyId_fileno, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002527}
2528
2529static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002530textiowrapper_seekable(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002531{
2532 CHECK_INITIALIZED(self);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002533 return _PyObject_CallMethodId(self->buffer, &PyId_seekable, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002534}
2535
2536static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002537textiowrapper_readable(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002538{
2539 CHECK_INITIALIZED(self);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002540 return _PyObject_CallMethodId(self->buffer, &PyId_readable, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002541}
2542
2543static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002544textiowrapper_writable(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002545{
2546 CHECK_INITIALIZED(self);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002547 return _PyObject_CallMethodId(self->buffer, &PyId_writable, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002548}
2549
2550static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002551textiowrapper_isatty(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002552{
2553 CHECK_INITIALIZED(self);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002554 return _PyObject_CallMethodId(self->buffer, &PyId_isatty, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002555}
2556
2557static PyObject *
Antoine Pitrou243757e2010-11-05 21:15:39 +00002558textiowrapper_getstate(textio *self, PyObject *args)
2559{
2560 PyErr_Format(PyExc_TypeError,
2561 "cannot serialize '%s' object", Py_TYPE(self)->tp_name);
2562 return NULL;
2563}
2564
2565static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002566textiowrapper_flush(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002567{
2568 CHECK_INITIALIZED(self);
2569 CHECK_CLOSED(self);
2570 self->telling = self->seekable;
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002571 if (_textiowrapper_writeflush(self) < 0)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002572 return NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002573 return _PyObject_CallMethodId(self->buffer, &PyId_flush, NULL);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002574}
2575
2576static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002577textiowrapper_close(textio *self, PyObject *args)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002578{
2579 PyObject *res;
Antoine Pitrou6be88762010-05-03 16:48:20 +00002580 int r;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002581 CHECK_INITIALIZED(self);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002582
Antoine Pitrou6be88762010-05-03 16:48:20 +00002583 res = textiowrapper_closed_get(self, NULL);
2584 if (res == NULL)
2585 return NULL;
2586 r = PyObject_IsTrue(res);
2587 Py_DECREF(res);
2588 if (r < 0)
2589 return NULL;
Victor Stinnerf6c57832010-05-19 01:17:01 +00002590
Antoine Pitrou6be88762010-05-03 16:48:20 +00002591 if (r > 0) {
2592 Py_RETURN_NONE; /* stream already closed */
2593 }
2594 else {
Benjamin Peterson68623612012-12-20 11:53:11 -06002595 PyObject *exc = NULL, *val, *tb;
Antoine Pitrou796564c2013-07-30 19:59:21 +02002596 if (self->finalizing) {
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002597 res = _PyObject_CallMethodId(self->buffer, &PyId__dealloc_warn, "O", self);
Antoine Pitroue033e062010-10-29 10:38:18 +00002598 if (res)
2599 Py_DECREF(res);
2600 else
2601 PyErr_Clear();
2602 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002603 res = _PyObject_CallMethodId((PyObject *)self, &PyId_flush, NULL);
Benjamin Peterson68623612012-12-20 11:53:11 -06002604 if (res == NULL)
2605 PyErr_Fetch(&exc, &val, &tb);
Antoine Pitrou6be88762010-05-03 16:48:20 +00002606 else
2607 Py_DECREF(res);
2608
Benjamin Peterson68623612012-12-20 11:53:11 -06002609 res = _PyObject_CallMethodId(self->buffer, &PyId_close, NULL);
2610 if (exc != NULL) {
2611 if (res != NULL) {
2612 Py_CLEAR(res);
2613 PyErr_Restore(exc, val, tb);
2614 }
2615 else {
Serhiy Storchaka76d3f142014-06-11 07:18:53 +03002616 PyObject *exc2, *val2, *tb2;
2617 PyErr_Fetch(&exc2, &val2, &tb2);
Serhiy Storchaka8a8f7f92014-06-09 09:13:04 +03002618 PyErr_NormalizeException(&exc, &val, &tb);
Benjamin Peterson68623612012-12-20 11:53:11 -06002619 Py_DECREF(exc);
2620 Py_XDECREF(tb);
Serhiy Storchaka76d3f142014-06-11 07:18:53 +03002621 PyErr_NormalizeException(&exc2, &val2, &tb2);
Benjamin Peterson68623612012-12-20 11:53:11 -06002622 PyException_SetContext(val2, val);
Serhiy Storchaka76d3f142014-06-11 07:18:53 +03002623 PyErr_Restore(exc2, val2, tb2);
Benjamin Peterson68623612012-12-20 11:53:11 -06002624 }
2625 }
2626 return res;
Antoine Pitrou6be88762010-05-03 16:48:20 +00002627 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002628}
2629
2630static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002631textiowrapper_iternext(textio *self)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002632{
2633 PyObject *line;
2634
2635 CHECK_INITIALIZED(self);
2636
2637 self->telling = 0;
2638 if (Py_TYPE(self) == &PyTextIOWrapper_Type) {
2639 /* Skip method call overhead for speed */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002640 line = _textiowrapper_readline(self, -1);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002641 }
2642 else {
2643 line = PyObject_CallMethodObjArgs((PyObject *)self,
2644 _PyIO_str_readline, NULL);
2645 if (line && !PyUnicode_Check(line)) {
2646 PyErr_Format(PyExc_IOError,
2647 "readline() should have returned an str object, "
2648 "not '%.200s'", Py_TYPE(line)->tp_name);
2649 Py_DECREF(line);
2650 return NULL;
2651 }
2652 }
2653
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002654 if (line == NULL || PyUnicode_READY(line) == -1)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002655 return NULL;
2656
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002657 if (PyUnicode_GET_LENGTH(line) == 0) {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002658 /* Reached EOF or would have blocked */
2659 Py_DECREF(line);
2660 Py_CLEAR(self->snapshot);
2661 self->telling = self->seekable;
2662 return NULL;
2663 }
2664
2665 return line;
2666}
2667
2668static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002669textiowrapper_name_get(textio *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002670{
2671 CHECK_INITIALIZED(self);
Martin v. Löwis767046a2011-10-14 15:35:36 +02002672 return _PyObject_GetAttrId(self->buffer, &PyId_name);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002673}
2674
2675static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002676textiowrapper_closed_get(textio *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002677{
2678 CHECK_INITIALIZED(self);
2679 return PyObject_GetAttr(self->buffer, _PyIO_str_closed);
2680}
2681
2682static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002683textiowrapper_newlines_get(textio *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002684{
2685 PyObject *res;
2686 CHECK_INITIALIZED(self);
2687 if (self->decoder == NULL)
2688 Py_RETURN_NONE;
2689 res = PyObject_GetAttr(self->decoder, _PyIO_str_newlines);
2690 if (res == NULL) {
Benjamin Peterson2cfca792009-06-06 20:46:48 +00002691 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
2692 PyErr_Clear();
2693 Py_RETURN_NONE;
2694 }
2695 else {
2696 return NULL;
2697 }
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002698 }
2699 return res;
2700}
2701
2702static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002703textiowrapper_errors_get(textio *self, void *context)
Benjamin Peterson0926ad12009-06-06 18:02:12 +00002704{
2705 CHECK_INITIALIZED(self);
2706 return PyUnicode_FromString(PyBytes_AS_STRING(self->errors));
2707}
2708
2709static PyObject *
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002710textiowrapper_chunk_size_get(textio *self, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002711{
2712 CHECK_INITIALIZED(self);
2713 return PyLong_FromSsize_t(self->chunk_size);
2714}
2715
2716static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002717textiowrapper_chunk_size_set(textio *self, PyObject *arg, void *context)
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002718{
2719 Py_ssize_t n;
2720 CHECK_INITIALIZED_INT(self);
Antoine Pitroucb4ae812011-07-13 21:07:49 +02002721 n = PyNumber_AsSsize_t(arg, PyExc_ValueError);
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002722 if (n == -1 && PyErr_Occurred())
2723 return -1;
2724 if (n <= 0) {
2725 PyErr_SetString(PyExc_ValueError,
2726 "a strictly positive integer is required");
2727 return -1;
2728 }
2729 self->chunk_size = n;
2730 return 0;
2731}
2732
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002733static PyMethodDef textiowrapper_methods[] = {
2734 {"detach", (PyCFunction)textiowrapper_detach, METH_NOARGS},
2735 {"write", (PyCFunction)textiowrapper_write, METH_VARARGS},
2736 {"read", (PyCFunction)textiowrapper_read, METH_VARARGS},
2737 {"readline", (PyCFunction)textiowrapper_readline, METH_VARARGS},
2738 {"flush", (PyCFunction)textiowrapper_flush, METH_NOARGS},
2739 {"close", (PyCFunction)textiowrapper_close, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002740
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002741 {"fileno", (PyCFunction)textiowrapper_fileno, METH_NOARGS},
2742 {"seekable", (PyCFunction)textiowrapper_seekable, METH_NOARGS},
2743 {"readable", (PyCFunction)textiowrapper_readable, METH_NOARGS},
2744 {"writable", (PyCFunction)textiowrapper_writable, METH_NOARGS},
2745 {"isatty", (PyCFunction)textiowrapper_isatty, METH_NOARGS},
Antoine Pitrou243757e2010-11-05 21:15:39 +00002746 {"__getstate__", (PyCFunction)textiowrapper_getstate, METH_NOARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002747
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002748 {"seek", (PyCFunction)textiowrapper_seek, METH_VARARGS},
2749 {"tell", (PyCFunction)textiowrapper_tell, METH_NOARGS},
2750 {"truncate", (PyCFunction)textiowrapper_truncate, METH_VARARGS},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002751 {NULL, NULL}
2752};
2753
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002754static PyMemberDef textiowrapper_members[] = {
2755 {"encoding", T_OBJECT, offsetof(textio, encoding), READONLY},
2756 {"buffer", T_OBJECT, offsetof(textio, buffer), READONLY},
2757 {"line_buffering", T_BOOL, offsetof(textio, line_buffering), READONLY},
Antoine Pitrou796564c2013-07-30 19:59:21 +02002758 {"_finalizing", T_BOOL, offsetof(textio, finalizing), 0},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002759 {NULL}
2760};
2761
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002762static PyGetSetDef textiowrapper_getset[] = {
2763 {"name", (getter)textiowrapper_name_get, NULL, NULL},
2764 {"closed", (getter)textiowrapper_closed_get, NULL, NULL},
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002765/* {"mode", (getter)TextIOWrapper_mode_get, NULL, NULL},
2766*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002767 {"newlines", (getter)textiowrapper_newlines_get, NULL, NULL},
2768 {"errors", (getter)textiowrapper_errors_get, NULL, NULL},
2769 {"_CHUNK_SIZE", (getter)textiowrapper_chunk_size_get,
2770 (setter)textiowrapper_chunk_size_set, NULL},
Benjamin Peterson1fea3212009-04-19 03:15:20 +00002771 {NULL}
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002772};
2773
2774PyTypeObject PyTextIOWrapper_Type = {
2775 PyVarObject_HEAD_INIT(NULL, 0)
2776 "_io.TextIOWrapper", /*tp_name*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002777 sizeof(textio), /*tp_basicsize*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002778 0, /*tp_itemsize*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002779 (destructor)textiowrapper_dealloc, /*tp_dealloc*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002780 0, /*tp_print*/
2781 0, /*tp_getattr*/
Benjamin Petersonc4c0eae2009-03-09 00:07:03 +00002782 0, /*tps_etattr*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002783 0, /*tp_compare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002784 (reprfunc)textiowrapper_repr,/*tp_repr*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002785 0, /*tp_as_number*/
2786 0, /*tp_as_sequence*/
2787 0, /*tp_as_mapping*/
2788 0, /*tp_hash */
2789 0, /*tp_call*/
2790 0, /*tp_str*/
2791 0, /*tp_getattro*/
2792 0, /*tp_setattro*/
2793 0, /*tp_as_buffer*/
2794 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE
Antoine Pitrou796564c2013-07-30 19:59:21 +02002795 | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_HAVE_FINALIZE, /*tp_flags*/
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002796 textiowrapper_doc, /* tp_doc */
2797 (traverseproc)textiowrapper_traverse, /* tp_traverse */
2798 (inquiry)textiowrapper_clear, /* tp_clear */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002799 0, /* tp_richcompare */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002800 offsetof(textio, weakreflist), /*tp_weaklistoffset*/
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002801 0, /* tp_iter */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002802 (iternextfunc)textiowrapper_iternext, /* tp_iternext */
2803 textiowrapper_methods, /* tp_methods */
2804 textiowrapper_members, /* tp_members */
2805 textiowrapper_getset, /* tp_getset */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002806 0, /* tp_base */
2807 0, /* tp_dict */
2808 0, /* tp_descr_get */
2809 0, /* tp_descr_set */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +00002810 offsetof(textio, dict), /*tp_dictoffset*/
2811 (initproc)textiowrapper_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002812 0, /* tp_alloc */
2813 PyType_GenericNew, /* tp_new */
Antoine Pitrou796564c2013-07-30 19:59:21 +02002814 0, /* tp_free */
2815 0, /* tp_is_gc */
2816 0, /* tp_bases */
2817 0, /* tp_mro */
2818 0, /* tp_cache */
2819 0, /* tp_subclasses */
2820 0, /* tp_weaklist */
2821 0, /* tp_del */
2822 0, /* tp_version_tag */
2823 0, /* tp_finalize */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00002824};