811 lines
25 KiB
C
811 lines
25 KiB
C
/*
|
||
An implementation of the new I/O lib as defined by PEP 3116 - "New I/O"
|
||
|
||
Classes defined here: UnsupportedOperation, BlockingIOError.
|
||
Functions defined here: open().
|
||
|
||
Mostly written by Amaury Forgeot d'Arc
|
||
*/
|
||
|
||
#define PY_SSIZE_T_CLEAN
|
||
#include "Python.h"
|
||
#include "pycore_pystate.h" /* _PyInterpreterState_GET_UNSAFE() */
|
||
#include "structmember.h"
|
||
#include "_iomodule.h"
|
||
|
||
#ifdef HAVE_SYS_TYPES_H
|
||
#include <sys/types.h>
|
||
#endif /* HAVE_SYS_TYPES_H */
|
||
|
||
#ifdef HAVE_SYS_STAT_H
|
||
#include <sys/stat.h>
|
||
#endif /* HAVE_SYS_STAT_H */
|
||
|
||
#ifdef MS_WINDOWS
|
||
#include <windows.h>
|
||
#endif
|
||
|
||
/* Various interned strings */
|
||
|
||
PyObject *_PyIO_str_close = NULL;
|
||
PyObject *_PyIO_str_closed = NULL;
|
||
PyObject *_PyIO_str_decode = NULL;
|
||
PyObject *_PyIO_str_encode = NULL;
|
||
PyObject *_PyIO_str_fileno = NULL;
|
||
PyObject *_PyIO_str_flush = NULL;
|
||
PyObject *_PyIO_str_getstate = NULL;
|
||
PyObject *_PyIO_str_isatty = NULL;
|
||
PyObject *_PyIO_str_newlines = NULL;
|
||
PyObject *_PyIO_str_nl = NULL;
|
||
PyObject *_PyIO_str_peek = NULL;
|
||
PyObject *_PyIO_str_read = NULL;
|
||
PyObject *_PyIO_str_read1 = NULL;
|
||
PyObject *_PyIO_str_readable = NULL;
|
||
PyObject *_PyIO_str_readall = NULL;
|
||
PyObject *_PyIO_str_readinto = NULL;
|
||
PyObject *_PyIO_str_readline = NULL;
|
||
PyObject *_PyIO_str_reset = NULL;
|
||
PyObject *_PyIO_str_seek = NULL;
|
||
PyObject *_PyIO_str_seekable = NULL;
|
||
PyObject *_PyIO_str_setstate = NULL;
|
||
PyObject *_PyIO_str_tell = NULL;
|
||
PyObject *_PyIO_str_truncate = NULL;
|
||
PyObject *_PyIO_str_writable = NULL;
|
||
PyObject *_PyIO_str_write = NULL;
|
||
|
||
PyObject *_PyIO_empty_str = NULL;
|
||
PyObject *_PyIO_empty_bytes = NULL;
|
||
|
||
PyDoc_STRVAR(module_doc,
|
||
"The io module provides the Python interfaces to stream handling. The\n"
|
||
"builtin open function is defined in this module.\n"
|
||
"\n"
|
||
"At the top of the I/O hierarchy is the abstract base class IOBase. It\n"
|
||
"defines the basic interface to a stream. Note, however, that there is no\n"
|
||
"separation between reading and writing to streams; implementations are\n"
|
||
"allowed to raise an OSError if they do not support a given operation.\n"
|
||
"\n"
|
||
"Extending IOBase is RawIOBase which deals simply with the reading and\n"
|
||
"writing of raw bytes to a stream. FileIO subclasses RawIOBase to provide\n"
|
||
"an interface to OS files.\n"
|
||
"\n"
|
||
"BufferedIOBase deals with buffering on a raw byte stream (RawIOBase). Its\n"
|
||
"subclasses, BufferedWriter, BufferedReader, and BufferedRWPair buffer\n"
|
||
"streams that are readable, writable, and both respectively.\n"
|
||
"BufferedRandom provides a buffered interface to random access\n"
|
||
"streams. BytesIO is a simple stream of in-memory bytes.\n"
|
||
"\n"
|
||
"Another IOBase subclass, TextIOBase, deals with the encoding and decoding\n"
|
||
"of streams into text. TextIOWrapper, which extends it, is a buffered text\n"
|
||
"interface to a buffered raw stream (`BufferedIOBase`). Finally, StringIO\n"
|
||
"is an in-memory stream for text.\n"
|
||
"\n"
|
||
"Argument names are not part of the specification, and only the arguments\n"
|
||
"of open() are intended to be used as keyword arguments.\n"
|
||
"\n"
|
||
"data:\n"
|
||
"\n"
|
||
"DEFAULT_BUFFER_SIZE\n"
|
||
"\n"
|
||
" An int containing the default buffer size used by the module's buffered\n"
|
||
" I/O classes. open() uses the file's blksize (as obtained by os.stat) if\n"
|
||
" possible.\n"
|
||
);
|
||
|
||
|
||
/*
|
||
* The main open() function
|
||
*/
|
||
/*[clinic input]
|
||
module _io
|
||
|
||
_io.open
|
||
file: object
|
||
mode: str = "r"
|
||
buffering: int = -1
|
||
encoding: str(accept={str, NoneType}) = NULL
|
||
errors: str(accept={str, NoneType}) = NULL
|
||
newline: str(accept={str, NoneType}) = NULL
|
||
closefd: bool(accept={int}) = True
|
||
opener: object = None
|
||
|
||
Open file and return a stream. Raise OSError upon failure.
|
||
|
||
file is either a text or byte string giving the name (and the path
|
||
if the file isn't in the current working directory) of the file to
|
||
be opened or an integer file descriptor of the file to be
|
||
wrapped. (If a file descriptor is given, it is closed when the
|
||
returned I/O object is closed, unless closefd is set to False.)
|
||
|
||
mode is an optional string that specifies the mode in which the file
|
||
is opened. It defaults to 'r' which means open for reading in text
|
||
mode. Other common values are 'w' for writing (truncating the file if
|
||
it already exists), 'x' for creating and writing to a new file, and
|
||
'a' for appending (which on some Unix systems, means that all writes
|
||
append to the end of the file regardless of the current seek position).
|
||
In text mode, if encoding is not specified the encoding used is platform
|
||
dependent: locale.getpreferredencoding(False) is called to get the
|
||
current locale encoding. (For reading and writing raw bytes use binary
|
||
mode and leave encoding unspecified.) The available modes are:
|
||
|
||
========= ===============================================================
|
||
Character Meaning
|
||
--------- ---------------------------------------------------------------
|
||
'r' open for reading (default)
|
||
'w' open for writing, truncating the file first
|
||
'x' create a new file and open it for writing
|
||
'a' open for writing, appending to the end of the file if it exists
|
||
'b' binary mode
|
||
't' text mode (default)
|
||
'+' open a disk file for updating (reading and writing)
|
||
'U' universal newline mode (deprecated)
|
||
========= ===============================================================
|
||
|
||
The default mode is 'rt' (open for reading text). For binary random
|
||
access, the mode 'w+b' opens and truncates the file to 0 bytes, while
|
||
'r+b' opens the file without truncation. The 'x' mode implies 'w' and
|
||
raises an `FileExistsError` if the file already exists.
|
||
|
||
Python distinguishes between files opened in binary and text modes,
|
||
even when the underlying operating system doesn't. Files opened in
|
||
binary mode (appending 'b' to the mode argument) return contents as
|
||
bytes objects without any decoding. In text mode (the default, or when
|
||
't' is appended to the mode argument), the contents of the file are
|
||
returned as strings, the bytes having been first decoded using a
|
||
platform-dependent encoding or using the specified encoding if given.
|
||
|
||
'U' mode is deprecated and will raise an exception in future versions
|
||
of Python. It has no effect in Python 3. Use newline to control
|
||
universal newlines mode.
|
||
|
||
buffering is an optional integer used to set the buffering policy.
|
||
Pass 0 to switch buffering off (only allowed in binary mode), 1 to select
|
||
line buffering (only usable in text mode), and an integer > 1 to indicate
|
||
the size of a fixed-size chunk buffer. When no buffering argument is
|
||
given, the default buffering policy works as follows:
|
||
|
||
* Binary files are buffered in fixed-size chunks; the size of the buffer
|
||
is chosen using a heuristic trying to determine the underlying device's
|
||
"block size" and falling back on `io.DEFAULT_BUFFER_SIZE`.
|
||
On many systems, the buffer will typically be 4096 or 8192 bytes long.
|
||
|
||
* "Interactive" text files (files for which isatty() returns True)
|
||
use line buffering. Other text files use the policy described above
|
||
for binary files.
|
||
|
||
encoding is the name of the encoding used to decode or encode the
|
||
file. This should only be used in text mode. The default encoding is
|
||
platform dependent, but any encoding supported by Python can be
|
||
passed. See the codecs module for the list of supported encodings.
|
||
|
||
errors is an optional string that specifies how encoding errors are to
|
||
be handled---this argument should not be used in binary mode. Pass
|
||
'strict' to raise a ValueError exception if there is an encoding error
|
||
(the default of None has the same effect), or pass 'ignore' to ignore
|
||
errors. (Note that ignoring encoding errors can lead to data loss.)
|
||
See the documentation for codecs.register or run 'help(codecs.Codec)'
|
||
for a list of the permitted encoding error strings.
|
||
|
||
newline controls how universal newlines works (it only applies to text
|
||
mode). It can be None, '', '\n', '\r', and '\r\n'. It works as
|
||
follows:
|
||
|
||
* On input, if newline is None, universal newlines mode is
|
||
enabled. Lines in the input can end in '\n', '\r', or '\r\n', and
|
||
these are translated into '\n' before being returned to the
|
||
caller. If it is '', universal newline mode is enabled, but line
|
||
endings are returned to the caller untranslated. If it has any of
|
||
the other legal values, input lines are only terminated by the given
|
||
string, and the line ending is returned to the caller untranslated.
|
||
|
||
* On output, if newline is None, any '\n' characters written are
|
||
translated to the system default line separator, os.linesep. If
|
||
newline is '' or '\n', no translation takes place. If newline is any
|
||
of the other legal values, any '\n' characters written are translated
|
||
to the given string.
|
||
|
||
If closefd is False, the underlying file descriptor will be kept open
|
||
when the file is closed. This does not work when a file name is given
|
||
and must be True in that case.
|
||
|
||
A custom opener can be used by passing a callable as *opener*. The
|
||
underlying file descriptor for the file object is then obtained by
|
||
calling *opener* with (*file*, *flags*). *opener* must return an open
|
||
file descriptor (passing os.open as *opener* results in functionality
|
||
similar to passing None).
|
||
|
||
open() returns a file object whose type depends on the mode, and
|
||
through which the standard file operations such as reading and writing
|
||
are performed. When open() is used to open a file in a text mode ('w',
|
||
'r', 'wt', 'rt', etc.), it returns a TextIOWrapper. When used to open
|
||
a file in a binary mode, the returned class varies: in read binary
|
||
mode, it returns a BufferedReader; in write binary and append binary
|
||
modes, it returns a BufferedWriter, and in read/write mode, it returns
|
||
a BufferedRandom.
|
||
|
||
It is also possible to use a string or bytearray as a file for both
|
||
reading and writing. For strings StringIO can be used like a file
|
||
opened in a text mode, and for bytes a BytesIO can be used like a file
|
||
opened in a binary mode.
|
||
[clinic start generated code]*/
|
||
|
||
static PyObject *
|
||
_io_open_impl(PyObject *module, PyObject *file, const char *mode,
|
||
int buffering, const char *encoding, const char *errors,
|
||
const char *newline, int closefd, PyObject *opener)
|
||
/*[clinic end generated code: output=aefafc4ce2b46dc0 input=03da2940c8a65871]*/
|
||
{
|
||
unsigned i;
|
||
|
||
int creating = 0, reading = 0, writing = 0, appending = 0, updating = 0;
|
||
int text = 0, binary = 0, universal = 0;
|
||
|
||
char rawmode[6], *m;
|
||
int line_buffering, is_number;
|
||
long isatty = 0;
|
||
|
||
PyObject *raw, *modeobj = NULL, *buffer, *wrapper, *result = NULL, *path_or_fd = NULL;
|
||
|
||
_Py_IDENTIFIER(_blksize);
|
||
_Py_IDENTIFIER(isatty);
|
||
_Py_IDENTIFIER(mode);
|
||
_Py_IDENTIFIER(close);
|
||
|
||
is_number = PyNumber_Check(file);
|
||
|
||
if (is_number) {
|
||
path_or_fd = file;
|
||
Py_INCREF(path_or_fd);
|
||
} else {
|
||
path_or_fd = PyOS_FSPath(file);
|
||
if (path_or_fd == NULL) {
|
||
return NULL;
|
||
}
|
||
}
|
||
|
||
if (!is_number &&
|
||
!PyUnicode_Check(path_or_fd) &&
|
||
!PyBytes_Check(path_or_fd)) {
|
||
PyErr_Format(PyExc_TypeError, "invalid file: %R", file);
|
||
goto error;
|
||
}
|
||
|
||
/* Decode mode */
|
||
for (i = 0; i < strlen(mode); i++) {
|
||
char c = mode[i];
|
||
|
||
switch (c) {
|
||
case 'x':
|
||
creating = 1;
|
||
break;
|
||
case 'r':
|
||
reading = 1;
|
||
break;
|
||
case 'w':
|
||
writing = 1;
|
||
break;
|
||
case 'a':
|
||
appending = 1;
|
||
break;
|
||
case '+':
|
||
updating = 1;
|
||
break;
|
||
case 't':
|
||
text = 1;
|
||
break;
|
||
case 'b':
|
||
binary = 1;
|
||
break;
|
||
case 'U':
|
||
universal = 1;
|
||
reading = 1;
|
||
break;
|
||
default:
|
||
goto invalid_mode;
|
||
}
|
||
|
||
/* c must not be duplicated */
|
||
if (strchr(mode+i+1, c)) {
|
||
invalid_mode:
|
||
PyErr_Format(PyExc_ValueError, "invalid mode: '%s'", mode);
|
||
goto error;
|
||
}
|
||
|
||
}
|
||
|
||
m = rawmode;
|
||
if (creating) *(m++) = 'x';
|
||
if (reading) *(m++) = 'r';
|
||
if (writing) *(m++) = 'w';
|
||
if (appending) *(m++) = 'a';
|
||
if (updating) *(m++) = '+';
|
||
*m = '\0';
|
||
|
||
/* Parameters validation */
|
||
if (universal) {
|
||
if (creating || writing || appending || updating) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"mode U cannot be combined with 'x', 'w', 'a', or '+'");
|
||
goto error;
|
||
}
|
||
if (PyErr_WarnEx(PyExc_DeprecationWarning,
|
||
"'U' mode is deprecated", 1) < 0)
|
||
goto error;
|
||
reading = 1;
|
||
}
|
||
|
||
if (text && binary) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"can't have text and binary mode at once");
|
||
goto error;
|
||
}
|
||
|
||
if (creating + reading + writing + appending > 1) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"must have exactly one of create/read/write/append mode");
|
||
goto error;
|
||
}
|
||
|
||
if (binary && encoding != NULL) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"binary mode doesn't take an encoding argument");
|
||
goto error;
|
||
}
|
||
|
||
if (binary && errors != NULL) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"binary mode doesn't take an errors argument");
|
||
goto error;
|
||
}
|
||
|
||
if (binary && newline != NULL) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"binary mode doesn't take a newline argument");
|
||
goto error;
|
||
}
|
||
|
||
if (binary && buffering == 1) {
|
||
if (PyErr_WarnEx(PyExc_RuntimeWarning,
|
||
"line buffering (buffering=1) isn't supported in "
|
||
"binary mode, the default buffer size will be used",
|
||
1) < 0) {
|
||
goto error;
|
||
}
|
||
}
|
||
|
||
/* Create the Raw file stream */
|
||
{
|
||
PyObject *RawIO_class = (PyObject *)&PyFileIO_Type;
|
||
#ifdef MS_WINDOWS
|
||
PyConfig *config = &_PyInterpreterState_GET_UNSAFE()->config;
|
||
if (!config->legacy_windows_stdio && _PyIO_get_console_type(path_or_fd) != '\0') {
|
||
RawIO_class = (PyObject *)&PyWindowsConsoleIO_Type;
|
||
encoding = "utf-8";
|
||
}
|
||
#endif
|
||
raw = PyObject_CallFunction(RawIO_class,
|
||
"OsiO", path_or_fd, rawmode, closefd, opener);
|
||
}
|
||
|
||
if (raw == NULL)
|
||
goto error;
|
||
result = raw;
|
||
|
||
Py_DECREF(path_or_fd);
|
||
path_or_fd = NULL;
|
||
|
||
modeobj = PyUnicode_FromString(mode);
|
||
if (modeobj == NULL)
|
||
goto error;
|
||
|
||
/* buffering */
|
||
if (buffering < 0) {
|
||
PyObject *res = _PyObject_CallMethodId(raw, &PyId_isatty, NULL);
|
||
if (res == NULL)
|
||
goto error;
|
||
isatty = PyLong_AsLong(res);
|
||
Py_DECREF(res);
|
||
if (isatty == -1 && PyErr_Occurred())
|
||
goto error;
|
||
}
|
||
|
||
if (buffering == 1 || isatty) {
|
||
buffering = -1;
|
||
line_buffering = 1;
|
||
}
|
||
else
|
||
line_buffering = 0;
|
||
|
||
if (buffering < 0) {
|
||
PyObject *blksize_obj;
|
||
blksize_obj = _PyObject_GetAttrId(raw, &PyId__blksize);
|
||
if (blksize_obj == NULL)
|
||
goto error;
|
||
buffering = PyLong_AsLong(blksize_obj);
|
||
Py_DECREF(blksize_obj);
|
||
if (buffering == -1 && PyErr_Occurred())
|
||
goto error;
|
||
}
|
||
if (buffering < 0) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"invalid buffering size");
|
||
goto error;
|
||
}
|
||
|
||
/* if not buffering, returns the raw file object */
|
||
if (buffering == 0) {
|
||
if (!binary) {
|
||
PyErr_SetString(PyExc_ValueError,
|
||
"can't have unbuffered text I/O");
|
||
goto error;
|
||
}
|
||
|
||
Py_DECREF(modeobj);
|
||
return result;
|
||
}
|
||
|
||
/* wraps into a buffered file */
|
||
{
|
||
PyObject *Buffered_class;
|
||
|
||
if (updating)
|
||
Buffered_class = (PyObject *)&PyBufferedRandom_Type;
|
||
else if (creating || writing || appending)
|
||
Buffered_class = (PyObject *)&PyBufferedWriter_Type;
|
||
else if (reading)
|
||
Buffered_class = (PyObject *)&PyBufferedReader_Type;
|
||
else {
|
||
PyErr_Format(PyExc_ValueError,
|
||
"unknown mode: '%s'", mode);
|
||
goto error;
|
||
}
|
||
|
||
buffer = PyObject_CallFunction(Buffered_class, "Oi", raw, buffering);
|
||
}
|
||
if (buffer == NULL)
|
||
goto error;
|
||
result = buffer;
|
||
Py_DECREF(raw);
|
||
|
||
|
||
/* if binary, returns the buffered file */
|
||
if (binary) {
|
||
Py_DECREF(modeobj);
|
||
return result;
|
||
}
|
||
|
||
/* wraps into a TextIOWrapper */
|
||
wrapper = PyObject_CallFunction((PyObject *)&PyTextIOWrapper_Type,
|
||
"Osssi",
|
||
buffer,
|
||
encoding, errors, newline,
|
||
line_buffering);
|
||
if (wrapper == NULL)
|
||
goto error;
|
||
result = wrapper;
|
||
Py_DECREF(buffer);
|
||
|
||
if (_PyObject_SetAttrId(wrapper, &PyId_mode, modeobj) < 0)
|
||
goto error;
|
||
Py_DECREF(modeobj);
|
||
return result;
|
||
|
||
error:
|
||
if (result != NULL) {
|
||
PyObject *exc, *val, *tb, *close_result;
|
||
PyErr_Fetch(&exc, &val, &tb);
|
||
close_result = _PyObject_CallMethodId(result, &PyId_close, NULL);
|
||
_PyErr_ChainExceptions(exc, val, tb);
|
||
Py_XDECREF(close_result);
|
||
Py_DECREF(result);
|
||
}
|
||
Py_XDECREF(path_or_fd);
|
||
Py_XDECREF(modeobj);
|
||
return NULL;
|
||
}
|
||
|
||
/*[clinic input]
|
||
_io.open_code
|
||
|
||
path : unicode
|
||
|
||
Opens the provided file with the intent to import the contents.
|
||
|
||
This may perform extra validation beyond open(), but is otherwise interchangeable
|
||
with calling open(path, 'rb').
|
||
|
||
[clinic start generated code]*/
|
||
|
||
static PyObject *
|
||
_io_open_code_impl(PyObject *module, PyObject *path)
|
||
/*[clinic end generated code: output=2fe4ecbd6f3d6844 input=f5c18e23f4b2ed9f]*/
|
||
{
|
||
return PyFile_OpenCodeObject(path);
|
||
}
|
||
|
||
/*
|
||
* Private helpers for the io module.
|
||
*/
|
||
|
||
Py_off_t
|
||
PyNumber_AsOff_t(PyObject *item, PyObject *err)
|
||
{
|
||
Py_off_t result;
|
||
PyObject *runerr;
|
||
PyObject *value = PyNumber_Index(item);
|
||
if (value == NULL)
|
||
return -1;
|
||
|
||
/* We're done if PyLong_AsSsize_t() returns without error. */
|
||
result = PyLong_AsOff_t(value);
|
||
if (result != -1 || !(runerr = PyErr_Occurred()))
|
||
goto finish;
|
||
|
||
/* Error handling code -- only manage OverflowError differently */
|
||
if (!PyErr_GivenExceptionMatches(runerr, PyExc_OverflowError))
|
||
goto finish;
|
||
|
||
PyErr_Clear();
|
||
/* If no error-handling desired then the default clipping
|
||
is sufficient.
|
||
*/
|
||
if (!err) {
|
||
assert(PyLong_Check(value));
|
||
/* Whether or not it is less than or equal to
|
||
zero is determined by the sign of ob_size
|
||
*/
|
||
if (_PyLong_Sign(value) < 0)
|
||
result = PY_OFF_T_MIN;
|
||
else
|
||
result = PY_OFF_T_MAX;
|
||
}
|
||
else {
|
||
/* Otherwise replace the error with caller's error object. */
|
||
PyErr_Format(err,
|
||
"cannot fit '%.200s' into an offset-sized integer",
|
||
item->ob_type->tp_name);
|
||
}
|
||
|
||
finish:
|
||
Py_DECREF(value);
|
||
return result;
|
||
}
|
||
|
||
|
||
_PyIO_State *
|
||
_PyIO_get_module_state(void)
|
||
{
|
||
PyObject *mod = PyState_FindModule(&_PyIO_Module);
|
||
_PyIO_State *state;
|
||
if (mod == NULL || (state = IO_MOD_STATE(mod)) == NULL) {
|
||
PyErr_SetString(PyExc_RuntimeError,
|
||
"could not find io module state "
|
||
"(interpreter shutdown?)");
|
||
return NULL;
|
||
}
|
||
return state;
|
||
}
|
||
|
||
PyObject *
|
||
_PyIO_get_locale_module(_PyIO_State *state)
|
||
{
|
||
PyObject *mod;
|
||
if (state->locale_module != NULL) {
|
||
assert(PyWeakref_CheckRef(state->locale_module));
|
||
mod = PyWeakref_GET_OBJECT(state->locale_module);
|
||
if (mod != Py_None) {
|
||
Py_INCREF(mod);
|
||
return mod;
|
||
}
|
||
Py_CLEAR(state->locale_module);
|
||
}
|
||
mod = PyImport_ImportModule("_bootlocale");
|
||
if (mod == NULL)
|
||
return NULL;
|
||
state->locale_module = PyWeakref_NewRef(mod, NULL);
|
||
if (state->locale_module == NULL) {
|
||
Py_DECREF(mod);
|
||
return NULL;
|
||
}
|
||
return mod;
|
||
}
|
||
|
||
|
||
static int
|
||
iomodule_traverse(PyObject *mod, visitproc visit, void *arg) {
|
||
_PyIO_State *state = IO_MOD_STATE(mod);
|
||
if (!state->initialized)
|
||
return 0;
|
||
if (state->locale_module != NULL) {
|
||
Py_VISIT(state->locale_module);
|
||
}
|
||
Py_VISIT(state->unsupported_operation);
|
||
return 0;
|
||
}
|
||
|
||
|
||
static int
|
||
iomodule_clear(PyObject *mod) {
|
||
_PyIO_State *state = IO_MOD_STATE(mod);
|
||
if (!state->initialized)
|
||
return 0;
|
||
if (state->locale_module != NULL)
|
||
Py_CLEAR(state->locale_module);
|
||
Py_CLEAR(state->unsupported_operation);
|
||
return 0;
|
||
}
|
||
|
||
static void
|
||
iomodule_free(PyObject *mod) {
|
||
iomodule_clear(mod);
|
||
}
|
||
|
||
|
||
/*
|
||
* Module definition
|
||
*/
|
||
|
||
#include "clinic/_iomodule.c.h"
|
||
|
||
static PyMethodDef module_methods[] = {
|
||
_IO_OPEN_METHODDEF
|
||
_IO_OPEN_CODE_METHODDEF
|
||
{NULL, NULL}
|
||
};
|
||
|
||
struct PyModuleDef _PyIO_Module = {
|
||
PyModuleDef_HEAD_INIT,
|
||
"io",
|
||
module_doc,
|
||
sizeof(_PyIO_State),
|
||
module_methods,
|
||
NULL,
|
||
iomodule_traverse,
|
||
iomodule_clear,
|
||
(freefunc)iomodule_free,
|
||
};
|
||
|
||
PyMODINIT_FUNC
|
||
PyInit__io(void)
|
||
{
|
||
PyObject *m = PyModule_Create(&_PyIO_Module);
|
||
_PyIO_State *state = NULL;
|
||
if (m == NULL)
|
||
return NULL;
|
||
state = IO_MOD_STATE(m);
|
||
state->initialized = 0;
|
||
|
||
#define ADD_TYPE(type, name) \
|
||
if (PyType_Ready(type) < 0) \
|
||
goto fail; \
|
||
Py_INCREF(type); \
|
||
if (PyModule_AddObject(m, name, (PyObject *)type) < 0) { \
|
||
Py_DECREF(type); \
|
||
goto fail; \
|
||
}
|
||
|
||
/* DEFAULT_BUFFER_SIZE */
|
||
if (PyModule_AddIntMacro(m, DEFAULT_BUFFER_SIZE) < 0)
|
||
goto fail;
|
||
|
||
/* UnsupportedOperation inherits from ValueError and OSError */
|
||
state->unsupported_operation = PyObject_CallFunction(
|
||
(PyObject *)&PyType_Type, "s(OO){}",
|
||
"UnsupportedOperation", PyExc_OSError, PyExc_ValueError);
|
||
if (state->unsupported_operation == NULL)
|
||
goto fail;
|
||
Py_INCREF(state->unsupported_operation);
|
||
if (PyModule_AddObject(m, "UnsupportedOperation",
|
||
state->unsupported_operation) < 0)
|
||
goto fail;
|
||
|
||
/* BlockingIOError, for compatibility */
|
||
Py_INCREF(PyExc_BlockingIOError);
|
||
if (PyModule_AddObject(m, "BlockingIOError",
|
||
(PyObject *) PyExc_BlockingIOError) < 0)
|
||
goto fail;
|
||
|
||
/* Concrete base types of the IO ABCs.
|
||
(the ABCs themselves are declared through inheritance in io.py)
|
||
*/
|
||
ADD_TYPE(&PyIOBase_Type, "_IOBase");
|
||
ADD_TYPE(&PyRawIOBase_Type, "_RawIOBase");
|
||
ADD_TYPE(&PyBufferedIOBase_Type, "_BufferedIOBase");
|
||
ADD_TYPE(&PyTextIOBase_Type, "_TextIOBase");
|
||
|
||
/* Implementation of concrete IO objects. */
|
||
/* FileIO */
|
||
PyFileIO_Type.tp_base = &PyRawIOBase_Type;
|
||
ADD_TYPE(&PyFileIO_Type, "FileIO");
|
||
|
||
/* BytesIO */
|
||
PyBytesIO_Type.tp_base = &PyBufferedIOBase_Type;
|
||
ADD_TYPE(&PyBytesIO_Type, "BytesIO");
|
||
if (PyType_Ready(&_PyBytesIOBuffer_Type) < 0)
|
||
goto fail;
|
||
|
||
/* StringIO */
|
||
PyStringIO_Type.tp_base = &PyTextIOBase_Type;
|
||
ADD_TYPE(&PyStringIO_Type, "StringIO");
|
||
|
||
#ifdef MS_WINDOWS
|
||
/* WindowsConsoleIO */
|
||
PyWindowsConsoleIO_Type.tp_base = &PyRawIOBase_Type;
|
||
ADD_TYPE(&PyWindowsConsoleIO_Type, "_WindowsConsoleIO");
|
||
#endif
|
||
|
||
/* BufferedReader */
|
||
PyBufferedReader_Type.tp_base = &PyBufferedIOBase_Type;
|
||
ADD_TYPE(&PyBufferedReader_Type, "BufferedReader");
|
||
|
||
/* BufferedWriter */
|
||
PyBufferedWriter_Type.tp_base = &PyBufferedIOBase_Type;
|
||
ADD_TYPE(&PyBufferedWriter_Type, "BufferedWriter");
|
||
|
||
/* BufferedRWPair */
|
||
PyBufferedRWPair_Type.tp_base = &PyBufferedIOBase_Type;
|
||
ADD_TYPE(&PyBufferedRWPair_Type, "BufferedRWPair");
|
||
|
||
/* BufferedRandom */
|
||
PyBufferedRandom_Type.tp_base = &PyBufferedIOBase_Type;
|
||
ADD_TYPE(&PyBufferedRandom_Type, "BufferedRandom");
|
||
|
||
/* TextIOWrapper */
|
||
PyTextIOWrapper_Type.tp_base = &PyTextIOBase_Type;
|
||
ADD_TYPE(&PyTextIOWrapper_Type, "TextIOWrapper");
|
||
|
||
/* IncrementalNewlineDecoder */
|
||
ADD_TYPE(&PyIncrementalNewlineDecoder_Type, "IncrementalNewlineDecoder");
|
||
|
||
/* Interned strings */
|
||
#define ADD_INTERNED(name) \
|
||
if (!_PyIO_str_ ## name && \
|
||
!(_PyIO_str_ ## name = PyUnicode_InternFromString(# name))) \
|
||
goto fail;
|
||
|
||
ADD_INTERNED(close)
|
||
ADD_INTERNED(closed)
|
||
ADD_INTERNED(decode)
|
||
ADD_INTERNED(encode)
|
||
ADD_INTERNED(fileno)
|
||
ADD_INTERNED(flush)
|
||
ADD_INTERNED(getstate)
|
||
ADD_INTERNED(isatty)
|
||
ADD_INTERNED(newlines)
|
||
ADD_INTERNED(peek)
|
||
ADD_INTERNED(read)
|
||
ADD_INTERNED(read1)
|
||
ADD_INTERNED(readable)
|
||
ADD_INTERNED(readall)
|
||
ADD_INTERNED(readinto)
|
||
ADD_INTERNED(readline)
|
||
ADD_INTERNED(reset)
|
||
ADD_INTERNED(seek)
|
||
ADD_INTERNED(seekable)
|
||
ADD_INTERNED(setstate)
|
||
ADD_INTERNED(tell)
|
||
ADD_INTERNED(truncate)
|
||
ADD_INTERNED(write)
|
||
ADD_INTERNED(writable)
|
||
|
||
if (!_PyIO_str_nl &&
|
||
!(_PyIO_str_nl = PyUnicode_InternFromString("\n")))
|
||
goto fail;
|
||
|
||
if (!_PyIO_empty_str &&
|
||
!(_PyIO_empty_str = PyUnicode_FromStringAndSize(NULL, 0)))
|
||
goto fail;
|
||
if (!_PyIO_empty_bytes &&
|
||
!(_PyIO_empty_bytes = PyBytes_FromStringAndSize(NULL, 0)))
|
||
goto fail;
|
||
|
||
state->initialized = 1;
|
||
|
||
return m;
|
||
|
||
fail:
|
||
Py_XDECREF(state->unsupported_operation);
|
||
Py_DECREF(m);
|
||
return NULL;
|
||
}
|