| #include "Python.h" | |
| #include "structmember.h" /* for offsetof() */ | |
| #include "_iomodule.h" | |
| typedef struct { | |
| PyObject_HEAD | |
| char *buf; | |
| Py_ssize_t pos; | |
| Py_ssize_t string_size; | |
| size_t buf_size; | |
| PyObject *dict; | |
| PyObject *weakreflist; | |
| } bytesio; | |
| #define CHECK_CLOSED(self) \ | |
| if ((self)->buf == NULL) { \ | |
| PyErr_SetString(PyExc_ValueError, \ | |
| "I/O operation on closed file."); \ | |
| return NULL; \ | |
| } | |
| /* Internal routine to get a line from the buffer of a BytesIO | |
| object. Returns the length between the current position to the | |
| next newline character. */ | |
| static Py_ssize_t | |
| get_line(bytesio *self, char **output) | |
| { | |
| char *n; | |
| const char *str_end; | |
| Py_ssize_t len; | |
| assert(self->buf != NULL); | |
| /* Move to the end of the line, up to the end of the string, s. */ | |
| str_end = self->buf + self->string_size; | |
| for (n = self->buf + self->pos; | |
| n < str_end && *n != '\n'; | |
| n++); | |
| /* Skip the newline character */ | |
| if (n < str_end) | |
| n++; | |
| /* Get the length from the current position to the end of the line. */ | |
| len = n - (self->buf + self->pos); | |
| *output = self->buf + self->pos; | |
| assert(len >= 0); | |
| assert(self->pos < PY_SSIZE_T_MAX - len); | |
| self->pos += len; | |
| return len; | |
| } | |
| /* Internal routine for changing the size of the buffer of BytesIO objects. | |
| The caller should ensure that the 'size' argument is non-negative. Returns | |
| 0 on success, -1 otherwise. */ | |
| static int | |
| resize_buffer(bytesio *self, size_t size) | |
| { | |
| /* Here, unsigned types are used to avoid dealing with signed integer | |
| overflow, which is undefined in C. */ | |
| size_t alloc = self->buf_size; | |
| char *new_buf = NULL; | |
| assert(self->buf != NULL); | |
| /* For simplicity, stay in the range of the signed type. Anyway, Python | |
| doesn't allow strings to be longer than this. */ | |
| if (size > PY_SSIZE_T_MAX) | |
| goto overflow; | |
| if (size < alloc / 2) { | |
| /* Major downsize; resize down to exact size. */ | |
| alloc = size + 1; | |
| } | |
| else if (size < alloc) { | |
| /* Within allocated size; quick exit */ | |
| return 0; | |
| } | |
| else if (size <= alloc * 1.125) { | |
| /* Moderate upsize; overallocate similar to list_resize() */ | |
| alloc = size + (size >> 3) + (size < 9 ? 3 : 6); | |
| } | |
| else { | |
| /* Major upsize; resize up to exact size */ | |
| alloc = size + 1; | |
| } | |
| if (alloc > ((size_t)-1) / sizeof(char)) | |
| goto overflow; | |
| new_buf = (char *)PyMem_Realloc(self->buf, alloc * sizeof(char)); | |
| if (new_buf == NULL) { | |
| PyErr_NoMemory(); | |
| return -1; | |
| } | |
| self->buf_size = alloc; | |
| self->buf = new_buf; | |
| return 0; | |
| overflow: | |
| PyErr_SetString(PyExc_OverflowError, | |
| "new buffer size too large"); | |
| return -1; | |
| } | |
| /* Internal routine for writing a string of bytes to the buffer of a BytesIO | |
| object. Returns the number of bytes wrote, or -1 on error. */ | |
| static Py_ssize_t | |
| write_bytes(bytesio *self, const char *bytes, Py_ssize_t len) | |
| { | |
| assert(self->buf != NULL); | |
| assert(self->pos >= 0); | |
| assert(len >= 0); | |
| if ((size_t)self->pos + len > self->buf_size) { | |
| if (resize_buffer(self, (size_t)self->pos + len) < 0) | |
| return -1; | |
| } | |
| if (self->pos > self->string_size) { | |
| /* In case of overseek, pad with null bytes the buffer region between | |
| the end of stream and the current position. | |
| 0 lo string_size hi | |
| | |<---used--->|<----------available----------->| | |
| | | <--to pad-->|<---to write---> | | |
| 0 buf position | |
| */ | |
| memset(self->buf + self->string_size, '\0', | |
| (self->pos - self->string_size) * sizeof(char)); | |
| } | |
| /* Copy the data to the internal buffer, overwriting some of the existing | |
| data if self->pos < self->string_size. */ | |
| memcpy(self->buf + self->pos, bytes, len); | |
| self->pos += len; | |
| /* Set the new length of the internal string if it has changed. */ | |
| if (self->string_size < self->pos) { | |
| self->string_size = self->pos; | |
| } | |
| return len; | |
| } | |
| static PyObject * | |
| bytesio_get_closed(bytesio *self) | |
| { | |
| if (self->buf == NULL) { | |
| Py_RETURN_TRUE; | |
| } | |
| else { | |
| Py_RETURN_FALSE; | |
| } | |
| } | |
| /* Generic getter for the writable, readable and seekable properties */ | |
| static PyObject * | |
| return_true(bytesio *self) | |
| { | |
| Py_RETURN_TRUE; | |
| } | |
| PyDoc_STRVAR(flush_doc, | |
| "flush() -> None. Does nothing."); | |
| static PyObject * | |
| bytesio_flush(bytesio *self) | |
| { | |
| CHECK_CLOSED(self); | |
| Py_RETURN_NONE; | |
| } | |
| PyDoc_STRVAR(getval_doc, | |
| "getvalue() -> bytes.\n" | |
| "\n" | |
| "Retrieve the entire contents of the BytesIO object."); | |
| static PyObject * | |
| bytesio_getvalue(bytesio *self) | |
| { | |
| CHECK_CLOSED(self); | |
| return PyBytes_FromStringAndSize(self->buf, self->string_size); | |
| } | |
| PyDoc_STRVAR(isatty_doc, | |
| "isatty() -> False.\n" | |
| "\n" | |
| "Always returns False since BytesIO objects are not connected\n" | |
| "to a tty-like device."); | |
| static PyObject * | |
| bytesio_isatty(bytesio *self) | |
| { | |
| CHECK_CLOSED(self); | |
| Py_RETURN_FALSE; | |
| } | |
| PyDoc_STRVAR(tell_doc, | |
| "tell() -> current file position, an integer\n"); | |
| static PyObject * | |
| bytesio_tell(bytesio *self) | |
| { | |
| CHECK_CLOSED(self); | |
| return PyLong_FromSsize_t(self->pos); | |
| } | |
| PyDoc_STRVAR(read_doc, | |
| "read([size]) -> read at most size bytes, returned as a string.\n" | |
| "\n" | |
| "If the size argument is negative, read until EOF is reached.\n" | |
| "Return an empty string at EOF."); | |
| static PyObject * | |
| bytesio_read(bytesio *self, PyObject *args) | |
| { | |
| Py_ssize_t size, n; | |
| char *output; | |
| PyObject *arg = Py_None; | |
| CHECK_CLOSED(self); | |
| if (!PyArg_ParseTuple(args, "|O:read", &arg)) | |
| return NULL; | |
| if (PyNumber_Check(arg)) { | |
| size = PyNumber_AsSsize_t(arg, PyExc_OverflowError); | |
| if (size == -1 && PyErr_Occurred()) | |
| return NULL; | |
| } | |
| else if (arg == Py_None) { | |
| /* Read until EOF is reached, by default. */ | |
| size = -1; | |
| } | |
| else { | |
| PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'", | |
| Py_TYPE(arg)->tp_name); | |
| return NULL; | |
| } | |
| /* adjust invalid sizes */ | |
| n = self->string_size - self->pos; | |
| if (size < 0 || size > n) { | |
| size = n; | |
| if (size < 0) | |
| size = 0; | |
| } | |
| assert(self->buf != NULL); | |
| output = self->buf + self->pos; | |
| self->pos += size; | |
| return PyBytes_FromStringAndSize(output, size); | |
| } | |
| PyDoc_STRVAR(read1_doc, | |
| "read1(size) -> read at most size bytes, returned as a string.\n" | |
| "\n" | |
| "If the size argument is negative or omitted, read until EOF is reached.\n" | |
| "Return an empty string at EOF."); | |
| static PyObject * | |
| bytesio_read1(bytesio *self, PyObject *n) | |
| { | |
| PyObject *arg, *res; | |
| arg = PyTuple_Pack(1, n); | |
| if (arg == NULL) | |
| return NULL; | |
| res = bytesio_read(self, arg); | |
| Py_DECREF(arg); | |
| return res; | |
| } | |
| PyDoc_STRVAR(readline_doc, | |
| "readline([size]) -> next line from the file, as a string.\n" | |
| "\n" | |
| "Retain newline. A non-negative size argument limits the maximum\n" | |
| "number of bytes to return (an incomplete line may be returned then).\n" | |
| "Return an empty string at EOF.\n"); | |
| static PyObject * | |
| bytesio_readline(bytesio *self, PyObject *args) | |
| { | |
| Py_ssize_t size, n; | |
| char *output; | |
| PyObject *arg = Py_None; | |
| CHECK_CLOSED(self); | |
| if (!PyArg_ParseTuple(args, "|O:readline", &arg)) | |
| return NULL; | |
| if (PyNumber_Check(arg)) { | |
| size = PyNumber_AsSsize_t(arg, PyExc_OverflowError); | |
| if (size == -1 && PyErr_Occurred()) | |
| return NULL; | |
| } | |
| else if (arg == Py_None) { | |
| /* No size limit, by default. */ | |
| size = -1; | |
| } | |
| else { | |
| PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'", | |
| Py_TYPE(arg)->tp_name); | |
| return NULL; | |
| } | |
| n = get_line(self, &output); | |
| if (size >= 0 && size < n) { | |
| size = n - size; | |
| n -= size; | |
| self->pos -= size; | |
| } | |
| return PyBytes_FromStringAndSize(output, n); | |
| } | |
| PyDoc_STRVAR(readlines_doc, | |
| "readlines([size]) -> list of strings, each a line from the file.\n" | |
| "\n" | |
| "Call readline() repeatedly and return a list of the lines so read.\n" | |
| "The optional size argument, if given, is an approximate bound on the\n" | |
| "total number of bytes in the lines returned.\n"); | |
| static PyObject * | |
| bytesio_readlines(bytesio *self, PyObject *args) | |
| { | |
| Py_ssize_t maxsize, size, n; | |
| PyObject *result, *line; | |
| char *output; | |
| PyObject *arg = Py_None; | |
| CHECK_CLOSED(self); | |
| if (!PyArg_ParseTuple(args, "|O:readlines", &arg)) | |
| return NULL; | |
| if (PyNumber_Check(arg)) { | |
| maxsize = PyNumber_AsSsize_t(arg, PyExc_OverflowError); | |
| if (maxsize == -1 && PyErr_Occurred()) | |
| return NULL; | |
| } | |
| else if (arg == Py_None) { | |
| /* No size limit, by default. */ | |
| maxsize = -1; | |
| } | |
| else { | |
| PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'", | |
| Py_TYPE(arg)->tp_name); | |
| return NULL; | |
| } | |
| size = 0; | |
| result = PyList_New(0); | |
| if (!result) | |
| return NULL; | |
| while ((n = get_line(self, &output)) != 0) { | |
| line = PyBytes_FromStringAndSize(output, n); | |
| if (!line) | |
| goto on_error; | |
| if (PyList_Append(result, line) == -1) { | |
| Py_DECREF(line); | |
| goto on_error; | |
| } | |
| Py_DECREF(line); | |
| size += n; | |
| if (maxsize > 0 && size >= maxsize) | |
| break; | |
| } | |
| return result; | |
| on_error: | |
| Py_DECREF(result); | |
| return NULL; | |
| } | |
| PyDoc_STRVAR(readinto_doc, | |
| "readinto(bytearray) -> int. Read up to len(b) bytes into b.\n" | |
| "\n" | |
| "Returns number of bytes read (0 for EOF), or None if the object\n" | |
| "is set not to block as has no data to read."); | |
| static PyObject * | |
| bytesio_readinto(bytesio *self, PyObject *args) | |
| { | |
| Py_buffer buf; | |
| Py_ssize_t len, n; | |
| CHECK_CLOSED(self); | |
| if (!PyArg_ParseTuple(args, "w*", &buf)) | |
| return NULL; | |
| len = buf.len; | |
| /* adjust invalid sizes */ | |
| n = self->string_size - self->pos; | |
| if (len > n) { | |
| len = n; | |
| if (len < 0) | |
| len = 0; | |
| } | |
| memcpy(buf.buf, self->buf + self->pos, len); | |
| assert(self->pos + len < PY_SSIZE_T_MAX); | |
| assert(len >= 0); | |
| self->pos += len; | |
| PyBuffer_Release(&buf); | |
| return PyLong_FromSsize_t(len); | |
| } | |
| PyDoc_STRVAR(truncate_doc, | |
| "truncate([size]) -> int. Truncate the file to at most size bytes.\n" | |
| "\n" | |
| "Size defaults to the current file position, as returned by tell().\n" | |
| "The current file position is unchanged. Returns the new size.\n"); | |
| static PyObject * | |
| bytesio_truncate(bytesio *self, PyObject *args) | |
| { | |
| Py_ssize_t size; | |
| PyObject *arg = Py_None; | |
| CHECK_CLOSED(self); | |
| if (!PyArg_ParseTuple(args, "|O:truncate", &arg)) | |
| return NULL; | |
| if (PyNumber_Check(arg)) { | |
| size = PyNumber_AsSsize_t(arg, PyExc_OverflowError); | |
| if (size == -1 && PyErr_Occurred()) | |
| return NULL; | |
| } | |
| else if (arg == Py_None) { | |
| /* Truncate to current position if no argument is passed. */ | |
| size = self->pos; | |
| } | |
| else { | |
| PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'", | |
| Py_TYPE(arg)->tp_name); | |
| return NULL; | |
| } | |
| if (size < 0) { | |
| PyErr_Format(PyExc_ValueError, | |
| "negative size value %zd", size); | |
| return NULL; | |
| } | |
| if (size < self->string_size) { | |
| self->string_size = size; | |
| if (resize_buffer(self, size) < 0) | |
| return NULL; | |
| } | |
| return PyLong_FromSsize_t(size); | |
| } | |
| static PyObject * | |
| bytesio_iternext(bytesio *self) | |
| { | |
| char *next; | |
| Py_ssize_t n; | |
| CHECK_CLOSED(self); | |
| n = get_line(self, &next); | |
| if (!next || n == 0) | |
| return NULL; | |
| return PyBytes_FromStringAndSize(next, n); | |
| } | |
| PyDoc_STRVAR(seek_doc, | |
| "seek(pos, whence=0) -> int. Change stream position.\n" | |
| "\n" | |
| "Seek to byte offset pos relative to position indicated by whence:\n" | |
| " 0 Start of stream (the default). pos should be >= 0;\n" | |
| " 1 Current position - pos may be negative;\n" | |
| " 2 End of stream - pos usually negative.\n" | |
| "Returns the new absolute position."); | |
| static PyObject * | |
| bytesio_seek(bytesio *self, PyObject *args) | |
| { | |
| PyObject *posobj; | |
| Py_ssize_t pos; | |
| int mode = 0; | |
| CHECK_CLOSED(self); | |
| if (!PyArg_ParseTuple(args, "O|i:seek", &posobj, &mode)) | |
| return NULL; | |
| pos = PyNumber_AsSsize_t(posobj, PyExc_OverflowError); | |
| if (pos == -1 && PyErr_Occurred()) | |
| return NULL; | |
| if (pos < 0 && mode == 0) { | |
| PyErr_Format(PyExc_ValueError, | |
| "negative seek value %zd", pos); | |
| return NULL; | |
| } | |
| /* mode 0: offset relative to beginning of the string. | |
| mode 1: offset relative to current position. | |
| mode 2: offset relative the end of the string. */ | |
| if (mode == 1) { | |
| if (pos > PY_SSIZE_T_MAX - self->pos) { | |
| PyErr_SetString(PyExc_OverflowError, | |
| "new position too large"); | |
| return NULL; | |
| } | |
| pos += self->pos; | |
| } | |
| else if (mode == 2) { | |
| if (pos > PY_SSIZE_T_MAX - self->string_size) { | |
| PyErr_SetString(PyExc_OverflowError, | |
| "new position too large"); | |
| return NULL; | |
| } | |
| pos += self->string_size; | |
| } | |
| else if (mode != 0) { | |
| PyErr_Format(PyExc_ValueError, | |
| "invalid whence (%i, should be 0, 1 or 2)", mode); | |
| return NULL; | |
| } | |
| if (pos < 0) | |
| pos = 0; | |
| self->pos = pos; | |
| return PyLong_FromSsize_t(self->pos); | |
| } | |
| PyDoc_STRVAR(write_doc, | |
| "write(bytes) -> int. Write bytes to file.\n" | |
| "\n" | |
| "Return the number of bytes written."); | |
| static PyObject * | |
| bytesio_write(bytesio *self, PyObject *obj) | |
| { | |
| Py_ssize_t n = 0; | |
| Py_buffer buf; | |
| PyObject *result = NULL; | |
| CHECK_CLOSED(self); | |
| if (PyObject_GetBuffer(obj, &buf, PyBUF_CONTIG_RO) < 0) | |
| return NULL; | |
| if (buf.len != 0) | |
| n = write_bytes(self, buf.buf, buf.len); | |
| if (n >= 0) | |
| result = PyLong_FromSsize_t(n); | |
| PyBuffer_Release(&buf); | |
| return result; | |
| } | |
| PyDoc_STRVAR(writelines_doc, | |
| "writelines(sequence_of_strings) -> None. Write strings to the file.\n" | |
| "\n" | |
| "Note that newlines are not added. The sequence can be any iterable\n" | |
| "object producing strings. This is equivalent to calling write() for\n" | |
| "each string."); | |
| static PyObject * | |
| bytesio_writelines(bytesio *self, PyObject *v) | |
| { | |
| PyObject *it, *item; | |
| PyObject *ret; | |
| CHECK_CLOSED(self); | |
| it = PyObject_GetIter(v); | |
| if (it == NULL) | |
| return NULL; | |
| while ((item = PyIter_Next(it)) != NULL) { | |
| ret = bytesio_write(self, item); | |
| Py_DECREF(item); | |
| if (ret == NULL) { | |
| Py_DECREF(it); | |
| return NULL; | |
| } | |
| Py_DECREF(ret); | |
| } | |
| Py_DECREF(it); | |
| /* See if PyIter_Next failed */ | |
| if (PyErr_Occurred()) | |
| return NULL; | |
| Py_RETURN_NONE; | |
| } | |
| PyDoc_STRVAR(close_doc, | |
| "close() -> None. Disable all I/O operations."); | |
| static PyObject * | |
| bytesio_close(bytesio *self) | |
| { | |
| if (self->buf != NULL) { | |
| PyMem_Free(self->buf); | |
| self->buf = NULL; | |
| } | |
| Py_RETURN_NONE; | |
| } | |
| /* Pickling support. | |
| Note that only pickle protocol 2 and onward are supported since we use | |
| extended __reduce__ API of PEP 307 to make BytesIO instances picklable. | |
| Providing support for protocol < 2 would require the __reduce_ex__ method | |
| which is notably long-winded when defined properly. | |
| For BytesIO, the implementation would similar to one coded for | |
| object.__reduce_ex__, but slightly less general. To be more specific, we | |
| could call bytesio_getstate directly and avoid checking for the presence of | |
| a fallback __reduce__ method. However, we would still need a __newobj__ | |
| function to use the efficient instance representation of PEP 307. | |
| */ | |
| static PyObject * | |
| bytesio_getstate(bytesio *self) | |
| { | |
| PyObject *initvalue = bytesio_getvalue(self); | |
| PyObject *dict; | |
| PyObject *state; | |
| if (initvalue == NULL) | |
| return NULL; | |
| if (self->dict == NULL) { | |
| Py_INCREF(Py_None); | |
| dict = Py_None; | |
| } | |
| else { | |
| dict = PyDict_Copy(self->dict); | |
| if (dict == NULL) | |
| return NULL; | |
| } | |
| state = Py_BuildValue("(OnN)", initvalue, self->pos, dict); | |
| Py_DECREF(initvalue); | |
| return state; | |
| } | |
| static PyObject * | |
| bytesio_setstate(bytesio *self, PyObject *state) | |
| { | |
| PyObject *result; | |
| PyObject *position_obj; | |
| PyObject *dict; | |
| Py_ssize_t pos; | |
| assert(state != NULL); | |
| /* We allow the state tuple to be longer than 3, because we may need | |
| someday to extend the object's state without breaking | |
| backward-compatibility. */ | |
| if (!PyTuple_Check(state) || Py_SIZE(state) < 3) { | |
| PyErr_Format(PyExc_TypeError, | |
| "%.200s.__setstate__ argument should be 3-tuple, got %.200s", | |
| Py_TYPE(self)->tp_name, Py_TYPE(state)->tp_name); | |
| return NULL; | |
| } | |
| /* Reset the object to its default state. This is only needed to handle | |
| the case of repeated calls to __setstate__. */ | |
| self->string_size = 0; | |
| self->pos = 0; | |
| /* Set the value of the internal buffer. If state[0] does not support the | |
| buffer protocol, bytesio_write will raise the appropriate TypeError. */ | |
| result = bytesio_write(self, PyTuple_GET_ITEM(state, 0)); | |
| if (result == NULL) | |
| return NULL; | |
| Py_DECREF(result); | |
| /* Set carefully the position value. Alternatively, we could use the seek | |
| method instead of modifying self->pos directly to better protect the | |
| object internal state against errneous (or malicious) inputs. */ | |
| position_obj = PyTuple_GET_ITEM(state, 1); | |
| if (!PyIndex_Check(position_obj)) { | |
| PyErr_Format(PyExc_TypeError, | |
| "second item of state must be an integer, not %.200s", | |
| Py_TYPE(position_obj)->tp_name); | |
| return NULL; | |
| } | |
| pos = PyNumber_AsSsize_t(position_obj, PyExc_OverflowError); | |
| if (pos == -1 && PyErr_Occurred()) | |
| return NULL; | |
| if (pos < 0) { | |
| PyErr_SetString(PyExc_ValueError, | |
| "position value cannot be negative"); | |
| return NULL; | |
| } | |
| self->pos = pos; | |
| /* Set the dictionary of the instance variables. */ | |
| dict = PyTuple_GET_ITEM(state, 2); | |
| if (dict != Py_None) { | |
| if (!PyDict_Check(dict)) { | |
| PyErr_Format(PyExc_TypeError, | |
| "third item of state should be a dict, got a %.200s", | |
| Py_TYPE(dict)->tp_name); | |
| return NULL; | |
| } | |
| if (self->dict) { | |
| /* Alternatively, we could replace the internal dictionary | |
| completely. However, it seems more practical to just update it. */ | |
| if (PyDict_Update(self->dict, dict) < 0) | |
| return NULL; | |
| } | |
| else { | |
| Py_INCREF(dict); | |
| self->dict = dict; | |
| } | |
| } | |
| Py_RETURN_NONE; | |
| } | |
| static void | |
| bytesio_dealloc(bytesio *self) | |
| { | |
| _PyObject_GC_UNTRACK(self); | |
| if (self->buf != NULL) { | |
| PyMem_Free(self->buf); | |
| self->buf = NULL; | |
| } | |
| Py_CLEAR(self->dict); | |
| if (self->weakreflist != NULL) | |
| PyObject_ClearWeakRefs((PyObject *) self); | |
| Py_TYPE(self)->tp_free(self); | |
| } | |
| static PyObject * | |
| bytesio_new(PyTypeObject *type, PyObject *args, PyObject *kwds) | |
| { | |
| bytesio *self; | |
| assert(type != NULL && type->tp_alloc != NULL); | |
| self = (bytesio *)type->tp_alloc(type, 0); | |
| if (self == NULL) | |
| return NULL; | |
| /* tp_alloc initializes all the fields to zero. So we don't have to | |
| initialize them here. */ | |
| self->buf = (char *)PyMem_Malloc(0); | |
| if (self->buf == NULL) { | |
| Py_DECREF(self); | |
| return PyErr_NoMemory(); | |
| } | |
| return (PyObject *)self; | |
| } | |
| static int | |
| bytesio_init(bytesio *self, PyObject *args, PyObject *kwds) | |
| { | |
| char *kwlist[] = {"initial_bytes", NULL}; | |
| PyObject *initvalue = NULL; | |
| if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:BytesIO", kwlist, | |
| &initvalue)) | |
| return -1; | |
| /* In case, __init__ is called multiple times. */ | |
| self->string_size = 0; | |
| self->pos = 0; | |
| if (initvalue && initvalue != Py_None) { | |
| PyObject *res; | |
| res = bytesio_write(self, initvalue); | |
| if (res == NULL) | |
| return -1; | |
| Py_DECREF(res); | |
| self->pos = 0; | |
| } | |
| return 0; | |
| } | |
| static int | |
| bytesio_traverse(bytesio *self, visitproc visit, void *arg) | |
| { | |
| Py_VISIT(self->dict); | |
| return 0; | |
| } | |
| static int | |
| bytesio_clear(bytesio *self) | |
| { | |
| Py_CLEAR(self->dict); | |
| return 0; | |
| } | |
| static PyGetSetDef bytesio_getsetlist[] = { | |
| {"closed", (getter)bytesio_get_closed, NULL, | |
| "True if the file is closed."}, | |
| {NULL}, /* sentinel */ | |
| }; | |
| static struct PyMethodDef bytesio_methods[] = { | |
| {"readable", (PyCFunction)return_true, METH_NOARGS, NULL}, | |
| {"seekable", (PyCFunction)return_true, METH_NOARGS, NULL}, | |
| {"writable", (PyCFunction)return_true, METH_NOARGS, NULL}, | |
| {"close", (PyCFunction)bytesio_close, METH_NOARGS, close_doc}, | |
| {"flush", (PyCFunction)bytesio_flush, METH_NOARGS, flush_doc}, | |
| {"isatty", (PyCFunction)bytesio_isatty, METH_NOARGS, isatty_doc}, | |
| {"tell", (PyCFunction)bytesio_tell, METH_NOARGS, tell_doc}, | |
| {"write", (PyCFunction)bytesio_write, METH_O, write_doc}, | |
| {"writelines", (PyCFunction)bytesio_writelines, METH_O, writelines_doc}, | |
| {"read1", (PyCFunction)bytesio_read1, METH_O, read1_doc}, | |
| {"readinto", (PyCFunction)bytesio_readinto, METH_VARARGS, readinto_doc}, | |
| {"readline", (PyCFunction)bytesio_readline, METH_VARARGS, readline_doc}, | |
| {"readlines", (PyCFunction)bytesio_readlines, METH_VARARGS, readlines_doc}, | |
| {"read", (PyCFunction)bytesio_read, METH_VARARGS, read_doc}, | |
| {"getvalue", (PyCFunction)bytesio_getvalue, METH_NOARGS, getval_doc}, | |
| {"seek", (PyCFunction)bytesio_seek, METH_VARARGS, seek_doc}, | |
| {"truncate", (PyCFunction)bytesio_truncate, METH_VARARGS, truncate_doc}, | |
| {"__getstate__", (PyCFunction)bytesio_getstate, METH_NOARGS, NULL}, | |
| {"__setstate__", (PyCFunction)bytesio_setstate, METH_O, NULL}, | |
| {NULL, NULL} /* sentinel */ | |
| }; | |
| PyDoc_STRVAR(bytesio_doc, | |
| "BytesIO([buffer]) -> object\n" | |
| "\n" | |
| "Create a buffered I/O implementation using an in-memory bytes\n" | |
| "buffer, ready for reading and writing."); | |
| PyTypeObject PyBytesIO_Type = { | |
| PyVarObject_HEAD_INIT(NULL, 0) | |
| "_io.BytesIO", /*tp_name*/ | |
| sizeof(bytesio), /*tp_basicsize*/ | |
| 0, /*tp_itemsize*/ | |
| (destructor)bytesio_dealloc, /*tp_dealloc*/ | |
| 0, /*tp_print*/ | |
| 0, /*tp_getattr*/ | |
| 0, /*tp_setattr*/ | |
| 0, /*tp_reserved*/ | |
| 0, /*tp_repr*/ | |
| 0, /*tp_as_number*/ | |
| 0, /*tp_as_sequence*/ | |
| 0, /*tp_as_mapping*/ | |
| 0, /*tp_hash*/ | |
| 0, /*tp_call*/ | |
| 0, /*tp_str*/ | |
| 0, /*tp_getattro*/ | |
| 0, /*tp_setattro*/ | |
| 0, /*tp_as_buffer*/ | |
| Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | | |
| Py_TPFLAGS_HAVE_GC, /*tp_flags*/ | |
| bytesio_doc, /*tp_doc*/ | |
| (traverseproc)bytesio_traverse, /*tp_traverse*/ | |
| (inquiry)bytesio_clear, /*tp_clear*/ | |
| 0, /*tp_richcompare*/ | |
| offsetof(bytesio, weakreflist), /*tp_weaklistoffset*/ | |
| PyObject_SelfIter, /*tp_iter*/ | |
| (iternextfunc)bytesio_iternext, /*tp_iternext*/ | |
| bytesio_methods, /*tp_methods*/ | |
| 0, /*tp_members*/ | |
| bytesio_getsetlist, /*tp_getset*/ | |
| 0, /*tp_base*/ | |
| 0, /*tp_dict*/ | |
| 0, /*tp_descr_get*/ | |
| 0, /*tp_descr_set*/ | |
| offsetof(bytesio, dict), /*tp_dictoffset*/ | |
| (initproc)bytesio_init, /*tp_init*/ | |
| 0, /*tp_alloc*/ | |
| bytesio_new, /*tp_new*/ | |
| }; |