blob: 14a4356bc7badc35b5dd940958b1f67cf324023a [file] [log] [blame]
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +00001/*
2 An implementation of the new I/O lib as defined by PEP 3116 - "New I/O"
3
4 Classes defined here: UnsupportedOperation, BlockingIOError.
5 Functions defined here: open().
6
7 Mostly written by Amaury Forgeot d'Arc
8*/
9
10#define PY_SSIZE_T_CLEAN
11#include "Python.h"
12#include "structmember.h"
13#include "_iomodule.h"
14
15#ifdef HAVE_SYS_TYPES_H
16#include <sys/types.h>
17#endif /* HAVE_SYS_TYPES_H */
18
19#ifdef HAVE_SYS_STAT_H
20#include <sys/stat.h>
21#endif /* HAVE_SYS_STAT_H */
22
23
24/* Various interned strings */
25
26PyObject *_PyIO_str_close;
27PyObject *_PyIO_str_closed;
28PyObject *_PyIO_str_decode;
29PyObject *_PyIO_str_encode;
30PyObject *_PyIO_str_fileno;
31PyObject *_PyIO_str_flush;
32PyObject *_PyIO_str_getstate;
33PyObject *_PyIO_str_isatty;
34PyObject *_PyIO_str_newlines;
35PyObject *_PyIO_str_nl;
36PyObject *_PyIO_str_read;
37PyObject *_PyIO_str_read1;
38PyObject *_PyIO_str_readable;
39PyObject *_PyIO_str_readinto;
40PyObject *_PyIO_str_readline;
41PyObject *_PyIO_str_reset;
42PyObject *_PyIO_str_seek;
43PyObject *_PyIO_str_seekable;
Antoine Pitroue4501852009-05-14 18:55:55 +000044PyObject *_PyIO_str_setstate;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000045PyObject *_PyIO_str_tell;
46PyObject *_PyIO_str_truncate;
47PyObject *_PyIO_str_writable;
48PyObject *_PyIO_str_write;
49
50PyObject *_PyIO_empty_str;
51PyObject *_PyIO_empty_bytes;
Antoine Pitroue4501852009-05-14 18:55:55 +000052PyObject *_PyIO_zero;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000053
54
55PyDoc_STRVAR(module_doc,
56"The io module provides the Python interfaces to stream handling. The\n"
57"builtin open function is defined in this module.\n"
58"\n"
59"At the top of the I/O hierarchy is the abstract base class IOBase. It\n"
60"defines the basic interface to a stream. Note, however, that there is no\n"
61"seperation between reading and writing to streams; implementations are\n"
62"allowed to throw an IOError if they do not support a given operation.\n"
63"\n"
64"Extending IOBase is RawIOBase which deals simply with the reading and\n"
Benjamin Peterson8f2b6652009-04-05 00:46:27 +000065"writing of raw bytes to a stream. FileIO subclasses RawIOBase to provide\n"
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000066"an interface to OS files.\n"
67"\n"
68"BufferedIOBase deals with buffering on a raw byte stream (RawIOBase). Its\n"
69"subclasses, BufferedWriter, BufferedReader, and BufferedRWPair buffer\n"
70"streams that are readable, writable, and both respectively.\n"
71"BufferedRandom provides a buffered interface to random access\n"
72"streams. BytesIO is a simple stream of in-memory bytes.\n"
73"\n"
74"Another IOBase subclass, TextIOBase, deals with the encoding and decoding\n"
75"of streams into text. TextIOWrapper, which extends it, is a buffered text\n"
76"interface to a buffered raw stream (`BufferedIOBase`). Finally, StringIO\n"
77"is a in-memory stream for text.\n"
78"\n"
79"Argument names are not part of the specification, and only the arguments\n"
80"of open() are intended to be used as keyword arguments.\n"
81"\n"
82"data:\n"
83"\n"
84"DEFAULT_BUFFER_SIZE\n"
85"\n"
86" An int containing the default buffer size used by the module's buffered\n"
87" I/O classes. open() uses the file's blksize (as obtained by os.stat) if\n"
88" possible.\n"
89 );
90
91
92/*
93 * BlockingIOError extends IOError
94 */
95
96static int
Benjamin Peterson680bf1a2009-06-12 02:07:12 +000097blockingioerror_init(PyBlockingIOErrorObject *self, PyObject *args,
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +000098 PyObject *kwds)
99{
100 PyObject *myerrno = NULL, *strerror = NULL;
101 PyObject *baseargs = NULL;
102 Py_ssize_t written = 0;
103
104 assert(PyTuple_Check(args));
105
106 self->written = 0;
107 if (!PyArg_ParseTuple(args, "OO|n:BlockingIOError",
108 &myerrno, &strerror, &written))
109 return -1;
110
111 baseargs = PyTuple_Pack(2, myerrno, strerror);
112 if (baseargs == NULL)
113 return -1;
114 /* This will take care of initializing of myerrno and strerror members */
115 if (((PyTypeObject *)PyExc_IOError)->tp_init(
116 (PyObject *)self, baseargs, kwds) == -1) {
117 Py_DECREF(baseargs);
118 return -1;
119 }
120 Py_DECREF(baseargs);
121
122 self->written = written;
123 return 0;
124}
125
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000126static PyMemberDef blockingioerror_members[] = {
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000127 {"characters_written", T_PYSSIZET, offsetof(PyBlockingIOErrorObject, written), 0},
128 {NULL} /* Sentinel */
129};
130
131static PyTypeObject _PyExc_BlockingIOError = {
132 PyVarObject_HEAD_INIT(NULL, 0)
133 "BlockingIOError", /*tp_name*/
134 sizeof(PyBlockingIOErrorObject), /*tp_basicsize*/
135 0, /*tp_itemsize*/
136 0, /*tp_dealloc*/
137 0, /*tp_print*/
138 0, /*tp_getattr*/
139 0, /*tp_setattr*/
140 0, /*tp_compare */
141 0, /*tp_repr*/
142 0, /*tp_as_number*/
143 0, /*tp_as_sequence*/
144 0, /*tp_as_mapping*/
145 0, /*tp_hash */
146 0, /*tp_call*/
147 0, /*tp_str*/
148 0, /*tp_getattro*/
149 0, /*tp_setattro*/
150 0, /*tp_as_buffer*/
151 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
152 PyDoc_STR("Exception raised when I/O would block "
153 "on a non-blocking I/O stream"), /* tp_doc */
154 0, /* tp_traverse */
155 0, /* tp_clear */
156 0, /* tp_richcompare */
157 0, /* tp_weaklistoffset */
158 0, /* tp_iter */
159 0, /* tp_iternext */
160 0, /* tp_methods */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000161 blockingioerror_members, /* tp_members */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000162 0, /* tp_getset */
163 0, /* tp_base */
164 0, /* tp_dict */
165 0, /* tp_descr_get */
166 0, /* tp_descr_set */
167 0, /* tp_dictoffset */
Benjamin Peterson680bf1a2009-06-12 02:07:12 +0000168 (initproc)blockingioerror_init, /* tp_init */
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000169 0, /* tp_alloc */
170 0, /* tp_new */
171};
172PyObject *PyExc_BlockingIOError = (PyObject *)&_PyExc_BlockingIOError;
173
174
175/*
176 * The main open() function
177 */
178PyDoc_STRVAR(open_doc,
179"Open file and return a stream. Raise IOError upon failure.\n"
180"\n"
181"file is either a text or byte string giving the name (and the path\n"
182"if the file isn't in the current working directory) of the file to\n"
183"be opened or an integer file descriptor of the file to be\n"
184"wrapped. (If a file descriptor is given, it is closed when the\n"
185"returned I/O object is closed, unless closefd is set to False.)\n"
186"\n"
187"mode is an optional string that specifies the mode in which the file\n"
188"is opened. It defaults to 'r' which means open for reading in text\n"
189"mode. Other common values are 'w' for writing (truncating the file if\n"
190"it already exists), and 'a' for appending (which on some Unix systems,\n"
191"means that all writes append to the end of the file regardless of the\n"
192"current seek position). In text mode, if encoding is not specified the\n"
193"encoding used is platform dependent. (For reading and writing raw\n"
194"bytes use binary mode and leave encoding unspecified.) The available\n"
195"modes are:\n"
196"\n"
197"========= ===============================================================\n"
198"Character Meaning\n"
199"--------- ---------------------------------------------------------------\n"
200"'r' open for reading (default)\n"
201"'w' open for writing, truncating the file first\n"
202"'a' open for writing, appending to the end of the file if it exists\n"
203"'b' binary mode\n"
204"'t' text mode (default)\n"
205"'+' open a disk file for updating (reading and writing)\n"
206"'U' universal newline mode (for backwards compatibility; unneeded\n"
207" for new code)\n"
208"========= ===============================================================\n"
209"\n"
210"The default mode is 'rt' (open for reading text). For binary random\n"
211"access, the mode 'w+b' opens and truncates the file to 0 bytes, while\n"
212"'r+b' opens the file without truncation.\n"
213"\n"
214"Python distinguishes between files opened in binary and text modes,\n"
215"even when the underlying operating system doesn't. Files opened in\n"
216"binary mode (appending 'b' to the mode argument) return contents as\n"
217"bytes objects without any decoding. In text mode (the default, or when\n"
218"'t' is appended to the mode argument), the contents of the file are\n"
219"returned as strings, the bytes having been first decoded using a\n"
220"platform-dependent encoding or using the specified encoding if given.\n"
221"\n"
Antoine Pitroud5587bc2009-12-19 21:08:31 +0000222"buffering is an optional integer used to set the buffering policy.\n"
223"Pass 0 to switch buffering off (only allowed in binary mode), 1 to select\n"
224"line buffering (only usable in text mode), and an integer > 1 to indicate\n"
225"the size of a fixed-size chunk buffer. When no buffering argument is\n"
226"given, the default buffering policy works as follows:\n"
227"\n"
228"* Binary files are buffered in fixed-size chunks; the size of the buffer\n"
229" is chosen using a heuristic trying to determine the underlying device's\n"
230" \"block size\" and falling back on `io.DEFAULT_BUFFER_SIZE`.\n"
231" On many systems, the buffer will typically be 4096 or 8192 bytes long.\n"
232"\n"
233"* \"Interactive\" text files (files for which isatty() returns True)\n"
234" use line buffering. Other text files use the policy described above\n"
235" for binary files.\n"
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000236"\n"
237"encoding is the name of the encoding used to decode or encode the\n"
238"file. This should only be used in text mode. The default encoding is\n"
239"platform dependent, but any encoding supported by Python can be\n"
240"passed. See the codecs module for the list of supported encodings.\n"
241"\n"
242"errors is an optional string that specifies how encoding errors are to\n"
243"be handled---this argument should not be used in binary mode. Pass\n"
244"'strict' to raise a ValueError exception if there is an encoding error\n"
245"(the default of None has the same effect), or pass 'ignore' to ignore\n"
246"errors. (Note that ignoring encoding errors can lead to data loss.)\n"
247"See the documentation for codecs.register for a list of the permitted\n"
248"encoding error strings.\n"
249"\n"
250"newline controls how universal newlines works (it only applies to text\n"
251"mode). It can be None, '', '\\n', '\\r', and '\\r\\n'. It works as\n"
252"follows:\n"
253"\n"
254"* On input, if newline is None, universal newlines mode is\n"
255" enabled. Lines in the input can end in '\\n', '\\r', or '\\r\\n', and\n"
256" these are translated into '\\n' before being returned to the\n"
257" caller. If it is '', universal newline mode is enabled, but line\n"
258" endings are returned to the caller untranslated. If it has any of\n"
259" the other legal values, input lines are only terminated by the given\n"
260" string, and the line ending is returned to the caller untranslated.\n"
261"\n"
262"* On output, if newline is None, any '\\n' characters written are\n"
263" translated to the system default line separator, os.linesep. If\n"
264" newline is '', no translation takes place. If newline is any of the\n"
265" other legal values, any '\\n' characters written are translated to\n"
266" the given string.\n"
267"\n"
268"If closefd is False, the underlying file descriptor will be kept open\n"
269"when the file is closed. This does not work when a file name is given\n"
270"and must be True in that case.\n"
271"\n"
272"open() returns a file object whose type depends on the mode, and\n"
273"through which the standard file operations such as reading and writing\n"
274"are performed. When open() is used to open a file in a text mode ('w',\n"
275"'r', 'wt', 'rt', etc.), it returns a TextIOWrapper. When used to open\n"
276"a file in a binary mode, the returned class varies: in read binary\n"
277"mode, it returns a BufferedReader; in write binary and append binary\n"
278"modes, it returns a BufferedWriter, and in read/write mode, it returns\n"
279"a BufferedRandom.\n"
280"\n"
281"It is also possible to use a string or bytearray as a file for both\n"
282"reading and writing. For strings StringIO can be used like a file\n"
283"opened in a text mode, and for bytes a BytesIO can be used like a file\n"
284"opened in a binary mode.\n"
285 );
286
287static PyObject *
288io_open(PyObject *self, PyObject *args, PyObject *kwds)
289{
290 char *kwlist[] = {"file", "mode", "buffering",
291 "encoding", "errors", "newline",
292 "closefd", NULL};
293 PyObject *file;
294 char *mode = "r";
295 int buffering = -1, closefd = 1;
296 char *encoding = NULL, *errors = NULL, *newline = NULL;
297 unsigned i;
298
299 int reading = 0, writing = 0, appending = 0, updating = 0;
300 int text = 0, binary = 0, universal = 0;
301
302 char rawmode[5], *m;
303 int line_buffering, isatty;
304
305 PyObject *raw, *modeobj = NULL, *buffer = NULL, *wrapper = NULL;
306
307 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|sizzzi:open", kwlist,
308 &file, &mode, &buffering,
309 &encoding, &errors, &newline,
310 &closefd)) {
311 return NULL;
312 }
313
314 if (!PyUnicode_Check(file) &&
315 !PyBytes_Check(file) &&
316 !PyNumber_Check(file)) {
317 PyErr_Format(PyExc_TypeError, "invalid file: %R", file);
318 return NULL;
319 }
320
321 /* Decode mode */
322 for (i = 0; i < strlen(mode); i++) {
323 char c = mode[i];
324
325 switch (c) {
326 case 'r':
327 reading = 1;
328 break;
329 case 'w':
330 writing = 1;
331 break;
332 case 'a':
333 appending = 1;
334 break;
335 case '+':
336 updating = 1;
337 break;
338 case 't':
339 text = 1;
340 break;
341 case 'b':
342 binary = 1;
343 break;
344 case 'U':
345 universal = 1;
346 reading = 1;
347 break;
348 default:
349 goto invalid_mode;
350 }
351
352 /* c must not be duplicated */
353 if (strchr(mode+i+1, c)) {
354 invalid_mode:
355 PyErr_Format(PyExc_ValueError, "invalid mode: '%s'", mode);
356 return NULL;
357 }
358
359 }
360
361 m = rawmode;
362 if (reading) *(m++) = 'r';
363 if (writing) *(m++) = 'w';
364 if (appending) *(m++) = 'a';
365 if (updating) *(m++) = '+';
366 *m = '\0';
367
368 /* Parameters validation */
369 if (universal) {
370 if (writing || appending) {
371 PyErr_SetString(PyExc_ValueError,
372 "can't use U and writing mode at once");
373 return NULL;
374 }
375 reading = 1;
376 }
377
378 if (text && binary) {
379 PyErr_SetString(PyExc_ValueError,
380 "can't have text and binary mode at once");
381 return NULL;
382 }
383
384 if (reading + writing + appending > 1) {
385 PyErr_SetString(PyExc_ValueError,
386 "must have exactly one of read/write/append mode");
387 return NULL;
388 }
389
390 if (binary && encoding != NULL) {
391 PyErr_SetString(PyExc_ValueError,
392 "binary mode doesn't take an encoding argument");
393 return NULL;
394 }
395
396 if (binary && errors != NULL) {
397 PyErr_SetString(PyExc_ValueError,
398 "binary mode doesn't take an errors argument");
399 return NULL;
400 }
401
402 if (binary && newline != NULL) {
403 PyErr_SetString(PyExc_ValueError,
404 "binary mode doesn't take a newline argument");
405 return NULL;
406 }
407
408 /* Create the Raw file stream */
409 raw = PyObject_CallFunction((PyObject *)&PyFileIO_Type,
410 "Osi", file, rawmode, closefd);
411 if (raw == NULL)
412 return NULL;
413
414 modeobj = PyUnicode_FromString(mode);
415 if (modeobj == NULL)
416 goto error;
417
418 /* buffering */
419 {
420 PyObject *res = PyObject_CallMethod(raw, "isatty", NULL);
421 if (res == NULL)
422 goto error;
423 isatty = PyLong_AsLong(res);
424 Py_DECREF(res);
425 if (isatty == -1 && PyErr_Occurred())
426 goto error;
427 }
428
429 if (buffering == 1 || (buffering < 0 && isatty)) {
430 buffering = -1;
431 line_buffering = 1;
432 }
433 else
434 line_buffering = 0;
435
436 if (buffering < 0) {
437 buffering = DEFAULT_BUFFER_SIZE;
438#ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
439 {
440 struct stat st;
441 long fileno;
442 PyObject *res = PyObject_CallMethod(raw, "fileno", NULL);
443 if (res == NULL)
444 goto error;
445
446 fileno = PyLong_AsLong(res);
447 Py_DECREF(res);
448 if (fileno == -1 && PyErr_Occurred())
449 goto error;
450
451 if (fstat(fileno, &st) >= 0)
452 buffering = st.st_blksize;
453 }
454#endif
455 }
456 if (buffering < 0) {
457 PyErr_SetString(PyExc_ValueError,
458 "invalid buffering size");
459 goto error;
460 }
461
462 /* if not buffering, returns the raw file object */
463 if (buffering == 0) {
464 if (!binary) {
465 PyErr_SetString(PyExc_ValueError,
466 "can't have unbuffered text I/O");
467 goto error;
468 }
469
470 Py_DECREF(modeobj);
471 return raw;
472 }
473
474 /* wraps into a buffered file */
475 {
476 PyObject *Buffered_class;
477
478 if (updating)
479 Buffered_class = (PyObject *)&PyBufferedRandom_Type;
480 else if (writing || appending)
481 Buffered_class = (PyObject *)&PyBufferedWriter_Type;
482 else if (reading)
483 Buffered_class = (PyObject *)&PyBufferedReader_Type;
484 else {
485 PyErr_Format(PyExc_ValueError,
486 "unknown mode: '%s'", mode);
487 goto error;
488 }
489
490 buffer = PyObject_CallFunction(Buffered_class, "Oi", raw, buffering);
491 }
492 Py_CLEAR(raw);
493 if (buffer == NULL)
494 goto error;
495
496
497 /* if binary, returns the buffered file */
498 if (binary) {
499 Py_DECREF(modeobj);
500 return buffer;
501 }
502
503 /* wraps into a TextIOWrapper */
504 wrapper = PyObject_CallFunction((PyObject *)&PyTextIOWrapper_Type,
505 "Osssi",
506 buffer,
507 encoding, errors, newline,
508 line_buffering);
509 Py_CLEAR(buffer);
510 if (wrapper == NULL)
511 goto error;
512
513 if (PyObject_SetAttrString(wrapper, "mode", modeobj) < 0)
514 goto error;
515 Py_DECREF(modeobj);
516 return wrapper;
517
518 error:
519 Py_XDECREF(raw);
520 Py_XDECREF(modeobj);
521 Py_XDECREF(buffer);
522 Py_XDECREF(wrapper);
523 return NULL;
524}
525
526/*
527 * Private helpers for the io module.
528 */
529
530Py_off_t
531PyNumber_AsOff_t(PyObject *item, PyObject *err)
532{
533 Py_off_t result;
534 PyObject *runerr;
535 PyObject *value = PyNumber_Index(item);
536 if (value == NULL)
537 return -1;
538
539 /* We're done if PyLong_AsSsize_t() returns without error. */
540 result = PyLong_AsOff_t(value);
541 if (result != -1 || !(runerr = PyErr_Occurred()))
542 goto finish;
543
544 /* Error handling code -- only manage OverflowError differently */
545 if (!PyErr_GivenExceptionMatches(runerr, PyExc_OverflowError))
546 goto finish;
547
548 PyErr_Clear();
549 /* If no error-handling desired then the default clipping
550 is sufficient.
551 */
552 if (!err) {
553 assert(PyLong_Check(value));
554 /* Whether or not it is less than or equal to
555 zero is determined by the sign of ob_size
556 */
557 if (_PyLong_Sign(value) < 0)
558 result = PY_OFF_T_MIN;
559 else
560 result = PY_OFF_T_MAX;
561 }
562 else {
563 /* Otherwise replace the error with caller's error object. */
564 PyErr_Format(err,
565 "cannot fit '%.200s' into an offset-sized integer",
566 item->ob_type->tp_name);
567 }
568
569 finish:
570 Py_DECREF(value);
571 return result;
572}
573
Benjamin Petersonbf5ff762009-12-13 19:25:34 +0000574
575/* Basically the "n" format code with the ability to turn None into -1. */
576int
577_PyIO_ConvertSsize_t(PyObject *obj, void *result) {
578 Py_ssize_t limit;
579 if (obj == Py_None) {
580 limit = -1;
581 }
582 else if (PyNumber_Check(obj)) {
583 limit = PyNumber_AsSsize_t(obj, PyExc_OverflowError);
584 if (limit == -1 && PyErr_Occurred())
585 return 0;
586 }
587 else {
588 PyErr_Format(PyExc_TypeError,
589 "integer argument expected, got '%.200s'",
590 Py_TYPE(obj)->tp_name);
591 return 0;
592 }
593 *((Py_ssize_t *)result) = limit;
594 return 1;
595}
596
597
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000598static int
599iomodule_traverse(PyObject *mod, visitproc visit, void *arg) {
600 _PyIO_State *state = IO_MOD_STATE(mod);
601 if (!state->initialized)
602 return 0;
603 Py_VISIT(state->os_module);
604 if (state->locale_module != NULL) {
605 Py_VISIT(state->locale_module);
606 }
607 Py_VISIT(state->unsupported_operation);
608 return 0;
609}
610
Benjamin Petersonbf5ff762009-12-13 19:25:34 +0000611
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000612static int
613iomodule_clear(PyObject *mod) {
614 _PyIO_State *state = IO_MOD_STATE(mod);
615 if (!state->initialized)
616 return 0;
617 Py_CLEAR(state->os_module);
618 if (state->locale_module != NULL)
619 Py_CLEAR(state->locale_module);
620 Py_CLEAR(state->unsupported_operation);
621 return 0;
622}
623
624static void
625iomodule_free(PyObject *mod) {
626 iomodule_clear(mod);
627}
628
Benjamin Petersonbf5ff762009-12-13 19:25:34 +0000629
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000630/*
631 * Module definition
632 */
633
634static PyMethodDef module_methods[] = {
635 {"open", (PyCFunction)io_open, METH_VARARGS|METH_KEYWORDS, open_doc},
636 {NULL, NULL}
637};
638
639struct PyModuleDef _PyIO_Module = {
640 PyModuleDef_HEAD_INIT,
641 "io",
642 module_doc,
643 sizeof(_PyIO_State),
644 module_methods,
645 NULL,
646 iomodule_traverse,
647 iomodule_clear,
648 (freefunc)iomodule_free,
649};
650
651PyMODINIT_FUNC
652PyInit__io(void)
653{
654 PyObject *m = PyModule_Create(&_PyIO_Module);
655 _PyIO_State *state = NULL;
656 if (m == NULL)
657 return NULL;
658 state = IO_MOD_STATE(m);
659 state->initialized = 0;
660
661 /* put os in the module state */
662 state->os_module = PyImport_ImportModule("os");
663 if (state->os_module == NULL)
664 goto fail;
665
666#define ADD_TYPE(type, name) \
667 if (PyType_Ready(type) < 0) \
668 goto fail; \
669 Py_INCREF(type); \
670 if (PyModule_AddObject(m, name, (PyObject *)type) < 0) { \
671 Py_DECREF(type); \
672 goto fail; \
673 }
674
675 /* DEFAULT_BUFFER_SIZE */
676 if (PyModule_AddIntMacro(m, DEFAULT_BUFFER_SIZE) < 0)
677 goto fail;
678
679 /* UnsupportedOperation inherits from ValueError and IOError */
680 state->unsupported_operation = PyObject_CallFunction(
681 (PyObject *)&PyType_Type, "s(OO){}",
682 "UnsupportedOperation", PyExc_ValueError, PyExc_IOError);
683 if (state->unsupported_operation == NULL)
684 goto fail;
685 Py_INCREF(state->unsupported_operation);
686 if (PyModule_AddObject(m, "UnsupportedOperation",
687 state->unsupported_operation) < 0)
688 goto fail;
689
690 /* BlockingIOError */
691 _PyExc_BlockingIOError.tp_base = (PyTypeObject *) PyExc_IOError;
692 ADD_TYPE(&_PyExc_BlockingIOError, "BlockingIOError");
693
694 /* Concrete base types of the IO ABCs.
695 (the ABCs themselves are declared through inheritance in io.py)
696 */
697 ADD_TYPE(&PyIOBase_Type, "_IOBase");
698 ADD_TYPE(&PyRawIOBase_Type, "_RawIOBase");
699 ADD_TYPE(&PyBufferedIOBase_Type, "_BufferedIOBase");
700 ADD_TYPE(&PyTextIOBase_Type, "_TextIOBase");
701
702 /* Implementation of concrete IO objects. */
703 /* FileIO */
704 PyFileIO_Type.tp_base = &PyRawIOBase_Type;
705 ADD_TYPE(&PyFileIO_Type, "FileIO");
706
707 /* BytesIO */
708 PyBytesIO_Type.tp_base = &PyBufferedIOBase_Type;
709 ADD_TYPE(&PyBytesIO_Type, "BytesIO");
710
711 /* StringIO */
712 PyStringIO_Type.tp_base = &PyTextIOBase_Type;
713 ADD_TYPE(&PyStringIO_Type, "StringIO");
714
715 /* BufferedReader */
716 PyBufferedReader_Type.tp_base = &PyBufferedIOBase_Type;
717 ADD_TYPE(&PyBufferedReader_Type, "BufferedReader");
718
719 /* BufferedWriter */
720 PyBufferedWriter_Type.tp_base = &PyBufferedIOBase_Type;
721 ADD_TYPE(&PyBufferedWriter_Type, "BufferedWriter");
722
723 /* BufferedRWPair */
724 PyBufferedRWPair_Type.tp_base = &PyBufferedIOBase_Type;
725 ADD_TYPE(&PyBufferedRWPair_Type, "BufferedRWPair");
726
727 /* BufferedRandom */
728 PyBufferedRandom_Type.tp_base = &PyBufferedIOBase_Type;
729 ADD_TYPE(&PyBufferedRandom_Type, "BufferedRandom");
730
731 /* TextIOWrapper */
732 PyTextIOWrapper_Type.tp_base = &PyTextIOBase_Type;
733 ADD_TYPE(&PyTextIOWrapper_Type, "TextIOWrapper");
734
735 /* IncrementalNewlineDecoder */
736 ADD_TYPE(&PyIncrementalNewlineDecoder_Type, "IncrementalNewlineDecoder");
737
738 /* Interned strings */
739 if (!(_PyIO_str_close = PyUnicode_InternFromString("close")))
740 goto fail;
741 if (!(_PyIO_str_closed = PyUnicode_InternFromString("closed")))
742 goto fail;
743 if (!(_PyIO_str_decode = PyUnicode_InternFromString("decode")))
744 goto fail;
745 if (!(_PyIO_str_encode = PyUnicode_InternFromString("encode")))
746 goto fail;
747 if (!(_PyIO_str_fileno = PyUnicode_InternFromString("fileno")))
748 goto fail;
749 if (!(_PyIO_str_flush = PyUnicode_InternFromString("flush")))
750 goto fail;
751 if (!(_PyIO_str_getstate = PyUnicode_InternFromString("getstate")))
752 goto fail;
753 if (!(_PyIO_str_isatty = PyUnicode_InternFromString("isatty")))
754 goto fail;
755 if (!(_PyIO_str_newlines = PyUnicode_InternFromString("newlines")))
756 goto fail;
757 if (!(_PyIO_str_nl = PyUnicode_InternFromString("\n")))
758 goto fail;
759 if (!(_PyIO_str_read = PyUnicode_InternFromString("read")))
760 goto fail;
761 if (!(_PyIO_str_read1 = PyUnicode_InternFromString("read1")))
762 goto fail;
763 if (!(_PyIO_str_readable = PyUnicode_InternFromString("readable")))
764 goto fail;
765 if (!(_PyIO_str_readinto = PyUnicode_InternFromString("readinto")))
766 goto fail;
767 if (!(_PyIO_str_readline = PyUnicode_InternFromString("readline")))
768 goto fail;
769 if (!(_PyIO_str_reset = PyUnicode_InternFromString("reset")))
770 goto fail;
771 if (!(_PyIO_str_seek = PyUnicode_InternFromString("seek")))
772 goto fail;
773 if (!(_PyIO_str_seekable = PyUnicode_InternFromString("seekable")))
774 goto fail;
Antoine Pitroue4501852009-05-14 18:55:55 +0000775 if (!(_PyIO_str_setstate = PyUnicode_InternFromString("setstate")))
776 goto fail;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000777 if (!(_PyIO_str_tell = PyUnicode_InternFromString("tell")))
778 goto fail;
779 if (!(_PyIO_str_truncate = PyUnicode_InternFromString("truncate")))
780 goto fail;
781 if (!(_PyIO_str_write = PyUnicode_InternFromString("write")))
782 goto fail;
783 if (!(_PyIO_str_writable = PyUnicode_InternFromString("writable")))
784 goto fail;
785
786 if (!(_PyIO_empty_str = PyUnicode_FromStringAndSize(NULL, 0)))
787 goto fail;
788 if (!(_PyIO_empty_bytes = PyBytes_FromStringAndSize(NULL, 0)))
789 goto fail;
Antoine Pitroue4501852009-05-14 18:55:55 +0000790 if (!(_PyIO_zero = PyLong_FromLong(0L)))
791 goto fail;
Benjamin Peterson4fa88fa2009-03-04 00:14:51 +0000792
793 state->initialized = 1;
794
795 return m;
796
797 fail:
798 Py_XDECREF(state->os_module);
799 Py_XDECREF(state->unsupported_operation);
800 Py_DECREF(m);
801 return NULL;
802}