cpython-lru_cache-weakref: Modules/

annotate Modules/_io/bytesio.c @ 80661:94de77bd0b4b 2.7

Issue #16194: document imp.load_dynamic problems when read several extension modules from the same binary object file.

author	Andrew Svetlov <andrew.svetlov@gmail.com>
date	Thu, 29 Nov 2012 19:51:28 +0200
parents	d0ab34d4c733
children

rev	line source
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	1 #include "Python.h"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	2 #include "structmember.h" /* for offsetof() */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	3 #include "_iomodule.h"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	4
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	5 typedef struct {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	6 PyObject_HEAD
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	7 char *buf;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	8 Py_ssize_t pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	9 Py_ssize_t string_size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	10 size_t buf_size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	11 PyObject *dict;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	12 PyObject *weakreflist;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	13 } bytesio;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	14
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	15 #define CHECK_CLOSED(self) \
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	16 if ((self)->buf == NULL) { \
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	17 PyErr_SetString(PyExc_ValueError, \
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	18 "I/O operation on closed file."); \
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	19 return NULL; \
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	20 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	21
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	22 /* Internal routine to get a line from the buffer of a BytesIO
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	23 object. Returns the length between the current position to the
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	24 next newline character. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	25 static Py_ssize_t
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	26 get_line(bytesio self, char *output)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	27 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	28 char *n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	29 const char *str_end;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	30 Py_ssize_t len;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	31
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	32 assert(self->buf != NULL);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	33
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	34 /* Move to the end of the line, up to the end of the string, s. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	35 str_end = self->buf + self->string_size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	36 for (n = self->buf + self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	37 n < str_end && *n != '\n';
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	38 n++);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	39
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	40 /* Skip the newline character */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	41 if (n < str_end)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	42 n++;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	43
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	44 /* Get the length from the current position to the end of the line. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	45 len = n - (self->buf + self->pos);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	46 *output = self->buf + self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	47
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	48 assert(len >= 0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	49 assert(self->pos < PY_SSIZE_T_MAX - len);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	50 self->pos += len;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	51
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	52 return len;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	53 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	54
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	55 /* Internal routine for changing the size of the buffer of BytesIO objects.
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	56 The caller should ensure that the 'size' argument is non-negative. Returns
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	57 0 on success, -1 otherwise. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	58 static int
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	59 resize_buffer(bytesio *self, size_t size)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	60 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	61 /* Here, unsigned types are used to avoid dealing with signed integer
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	62 overflow, which is undefined in C. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	63 size_t alloc = self->buf_size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	64 char *new_buf = NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	65
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	66 assert(self->buf != NULL);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	67
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	68 /* For simplicity, stay in the range of the signed type. Anyway, Python
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	69 doesn't allow strings to be longer than this. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	70 if (size > PY_SSIZE_T_MAX)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	71 goto overflow;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	72
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	73 if (size < alloc / 2) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	74 /* Major downsize; resize down to exact size. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	75 alloc = size + 1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	76 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	77 else if (size < alloc) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	78 /* Within allocated size; quick exit */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	79 return 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	80 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	81 else if (size <= alloc * 1.125) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	82 /* Moderate upsize; overallocate similar to list_resize() */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	83 alloc = size + (size >> 3) + (size < 9 ? 3 : 6);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	84 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	85 else {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	86 /* Major upsize; resize up to exact size */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	87 alloc = size + 1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	88 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	89
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	90 if (alloc > ((size_t)-1) / sizeof(char))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	91 goto overflow;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	92 new_buf = (char )PyMem_Realloc(self->buf, alloc sizeof(char));
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	93 if (new_buf == NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	94 PyErr_NoMemory();
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	95 return -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	96 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	97 self->buf_size = alloc;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	98 self->buf = new_buf;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	99
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	100 return 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	101
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	102 overflow:
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	103 PyErr_SetString(PyExc_OverflowError,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	104 "new buffer size too large");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	105 return -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	106 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	107
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	108 /* Internal routine for writing a string of bytes to the buffer of a BytesIO
78854 d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	109 object. Returns the number of bytes written, or -1 on error. */
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	110 static Py_ssize_t
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	111 write_bytes(bytesio self, const char bytes, Py_ssize_t len)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	112 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	113 assert(self->buf != NULL);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	114 assert(self->pos >= 0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	115 assert(len >= 0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	116
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	117 if ((size_t)self->pos + len > self->buf_size) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	118 if (resize_buffer(self, (size_t)self->pos + len) < 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	119 return -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	120 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	121
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	122 if (self->pos > self->string_size) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	123 /* In case of overseek, pad with null bytes the buffer region between
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	124 the end of stream and the current position.
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	125
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	126 0 lo string_size hi
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	127 \| \|<---used--->\|<----------available----------->\|
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	128 \| \| <--to pad-->\|<---to write---> \|
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	129 0 buf position
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	130 */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	131 memset(self->buf + self->string_size, '\0',
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	132 (self->pos - self->string_size) * sizeof(char));
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	133 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	134
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	135 /* Copy the data to the internal buffer, overwriting some of the existing
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	136 data if self->pos < self->string_size. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	137 memcpy(self->buf + self->pos, bytes, len);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	138 self->pos += len;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	139
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	140 /* Set the new length of the internal string if it has changed. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	141 if (self->string_size < self->pos) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	142 self->string_size = self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	143 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	144
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	145 return len;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	146 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	147
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	148 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	149 bytesio_get_closed(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	150 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	151 if (self->buf == NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	152 Py_RETURN_TRUE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	153 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	154 else {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	155 Py_RETURN_FALSE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	156 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	157 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	158
78854 d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	159 PyDoc_STRVAR(readable_doc,
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	160 "readable() -> bool. Returns True if the IO object can be read.");
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	161
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	162 PyDoc_STRVAR(writable_doc,
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	163 "writable() -> bool. Returns True if the IO object can be written.");
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	164
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	165 PyDoc_STRVAR(seekable_doc,
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	166 "seekable() -> bool. Returns True if the IO object can be seeked.");
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	167
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	168 /* Generic getter for the writable, readable and seekable properties */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	169 static PyObject *
78854 d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	170 return_not_closed(bytesio *self)
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	171 {
78854 d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	172 CHECK_CLOSED(self);
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	173 Py_RETURN_TRUE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	174 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	175
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	176 PyDoc_STRVAR(flush_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	177 "flush() -> None. Does nothing.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	178
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	179 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	180 bytesio_flush(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	181 {
61180 9b438bb06d4d Issue #7865: The close() method of :mod:`io` objects should not swallow Antoine Pitrou <solipsis@pitrou.net> parents: 58624 diff changeset	182 CHECK_CLOSED(self);
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	183 Py_RETURN_NONE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	184 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	185
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	186 PyDoc_STRVAR(getval_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	187 "getvalue() -> bytes.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	188 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	189 "Retrieve the entire contents of the BytesIO object.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	190
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	191 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	192 bytesio_getvalue(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	193 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	194 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	195 return PyBytes_FromStringAndSize(self->buf, self->string_size);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	196 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	197
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	198 PyDoc_STRVAR(isatty_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	199 "isatty() -> False.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	200 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	201 "Always returns False since BytesIO objects are not connected\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	202 "to a tty-like device.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	203
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	204 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	205 bytesio_isatty(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	206 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	207 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	208 Py_RETURN_FALSE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	209 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	210
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	211 PyDoc_STRVAR(tell_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	212 "tell() -> current file position, an integer\n");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	213
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	214 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	215 bytesio_tell(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	216 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	217 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	218 return PyLong_FromSsize_t(self->pos);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	219 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	220
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	221 PyDoc_STRVAR(read_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	222 "read([size]) -> read at most size bytes, returned as a string.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	223 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	224 "If the size argument is negative, read until EOF is reached.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	225 "Return an empty string at EOF.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	226
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	227 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	228 bytesio_read(bytesio self, PyObject args)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	229 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	230 Py_ssize_t size, n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	231 char *output;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	232 PyObject *arg = Py_None;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	233
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	234 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	235
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	236 if (!PyArg_ParseTuple(args, "\|O:read", &arg))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	237 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	238
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	239 if (PyNumber_Check(arg)) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	240 size = PyNumber_AsSsize_t(arg, PyExc_OverflowError);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	241 if (size == -1 && PyErr_Occurred())
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	242 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	243 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	244 else if (arg == Py_None) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	245 /* Read until EOF is reached, by default. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	246 size = -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	247 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	248 else {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	249 PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'",
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	250 Py_TYPE(arg)->tp_name);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	251 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	252 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	253
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	254 /* adjust invalid sizes */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	255 n = self->string_size - self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	256 if (size < 0 \|\| size > n) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	257 size = n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	258 if (size < 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	259 size = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	260 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	261
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	262 assert(self->buf != NULL);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	263 output = self->buf + self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	264 self->pos += size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	265
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	266 return PyBytes_FromStringAndSize(output, size);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	267 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	268
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	269
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	270 PyDoc_STRVAR(read1_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	271 "read1(size) -> read at most size bytes, returned as a string.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	272 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	273 "If the size argument is negative or omitted, read until EOF is reached.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	274 "Return an empty string at EOF.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	275
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	276 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	277 bytesio_read1(bytesio self, PyObject n)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	278 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	279 PyObject arg, res;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	280
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	281 arg = PyTuple_Pack(1, n);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	282 if (arg == NULL)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	283 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	284 res = bytesio_read(self, arg);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	285 Py_DECREF(arg);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	286 return res;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	287 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	288
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	289 PyDoc_STRVAR(readline_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	290 "readline([size]) -> next line from the file, as a string.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	291 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	292 "Retain newline. A non-negative size argument limits the maximum\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	293 "number of bytes to return (an incomplete line may be returned then).\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	294 "Return an empty string at EOF.\n");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	295
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	296 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	297 bytesio_readline(bytesio self, PyObject args)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	298 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	299 Py_ssize_t size, n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	300 char *output;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	301 PyObject *arg = Py_None;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	302
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	303 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	304
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	305 if (!PyArg_ParseTuple(args, "\|O:readline", &arg))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	306 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	307
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	308 if (PyNumber_Check(arg)) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	309 size = PyNumber_AsSsize_t(arg, PyExc_OverflowError);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	310 if (size == -1 && PyErr_Occurred())
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	311 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	312 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	313 else if (arg == Py_None) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	314 /* No size limit, by default. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	315 size = -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	316 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	317 else {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	318 PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'",
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	319 Py_TYPE(arg)->tp_name);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	320 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	321 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	322
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	323 n = get_line(self, &output);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	324
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	325 if (size >= 0 && size < n) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	326 size = n - size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	327 n -= size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	328 self->pos -= size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	329 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	330
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	331 return PyBytes_FromStringAndSize(output, n);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	332 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	333
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	334 PyDoc_STRVAR(readlines_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	335 "readlines([size]) -> list of strings, each a line from the file.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	336 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	337 "Call readline() repeatedly and return a list of the lines so read.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	338 "The optional size argument, if given, is an approximate bound on the\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	339 "total number of bytes in the lines returned.\n");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	340
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	341 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	342 bytesio_readlines(bytesio self, PyObject args)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	343 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	344 Py_ssize_t maxsize, size, n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	345 PyObject result, line;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	346 char *output;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	347 PyObject *arg = Py_None;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	348
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	349 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	350
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	351 if (!PyArg_ParseTuple(args, "\|O:readlines", &arg))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	352 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	353
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	354 if (PyNumber_Check(arg)) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	355 maxsize = PyNumber_AsSsize_t(arg, PyExc_OverflowError);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	356 if (maxsize == -1 && PyErr_Occurred())
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	357 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	358 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	359 else if (arg == Py_None) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	360 /* No size limit, by default. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	361 maxsize = -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	362 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	363 else {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	364 PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'",
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	365 Py_TYPE(arg)->tp_name);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	366 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	367 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	368
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	369 size = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	370 result = PyList_New(0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	371 if (!result)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	372 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	373
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	374 while ((n = get_line(self, &output)) != 0) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	375 line = PyBytes_FromStringAndSize(output, n);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	376 if (!line)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	377 goto on_error;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	378 if (PyList_Append(result, line) == -1) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	379 Py_DECREF(line);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	380 goto on_error;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	381 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	382 Py_DECREF(line);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	383 size += n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	384 if (maxsize > 0 && size >= maxsize)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	385 break;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	386 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	387 return result;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	388
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	389 on_error:
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	390 Py_DECREF(result);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	391 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	392 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	393
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	394 PyDoc_STRVAR(readinto_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	395 "readinto(bytearray) -> int. Read up to len(b) bytes into b.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	396 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	397 "Returns number of bytes read (0 for EOF), or None if the object\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	398 "is set not to block as has no data to read.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	399
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	400 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	401 bytesio_readinto(bytesio self, PyObject args)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	402 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	403 Py_buffer buf;
66303 c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	404 Py_ssize_t len, n;
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	405
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	406 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	407
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	408 if (!PyArg_ParseTuple(args, "w*", &buf))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	409 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	410
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	411 len = buf.len;
66303 c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	412 /* adjust invalid sizes */
c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	413 n = self->string_size - self->pos;
c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	414 if (len > n) {
c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	415 len = n;
c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	416 if (len < 0)
c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	417 len = 0;
c404c2a346a5 Merged revisions 86587 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 64384 diff changeset	418 }
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	419
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	420 memcpy(buf.buf, self->buf + self->pos, len);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	421 assert(self->pos + len < PY_SSIZE_T_MAX);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	422 assert(len >= 0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	423 self->pos += len;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	424
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	425 PyBuffer_Release(&buf);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	426 return PyLong_FromSsize_t(len);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	427 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	428
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	429 PyDoc_STRVAR(truncate_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	430 "truncate([size]) -> int. Truncate the file to at most size bytes.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	431 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	432 "Size defaults to the current file position, as returned by tell().\n"
58624 77130f2fe944 - Issue #6939: Fix file I/O objects in the `io` module to keep the original Antoine Pitrou <solipsis@pitrou.net> parents: 56704 diff changeset	433 "The current file position is unchanged. Returns the new size.\n");
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	434
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	435 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	436 bytesio_truncate(bytesio self, PyObject args)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	437 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	438 Py_ssize_t size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	439 PyObject *arg = Py_None;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	440
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	441 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	442
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	443 if (!PyArg_ParseTuple(args, "\|O:truncate", &arg))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	444 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	445
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	446 if (PyNumber_Check(arg)) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	447 size = PyNumber_AsSsize_t(arg, PyExc_OverflowError);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	448 if (size == -1 && PyErr_Occurred())
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	449 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	450 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	451 else if (arg == Py_None) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	452 /* Truncate to current position if no argument is passed. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	453 size = self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	454 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	455 else {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	456 PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'",
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	457 Py_TYPE(arg)->tp_name);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	458 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	459 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	460
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	461 if (size < 0) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	462 PyErr_Format(PyExc_ValueError,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	463 "negative size value %zd", size);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	464 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	465 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	466
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	467 if (size < self->string_size) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	468 self->string_size = size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	469 if (resize_buffer(self, size) < 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	470 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	471 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	472
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	473 return PyLong_FromSsize_t(size);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	474 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	475
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	476 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	477 bytesio_iternext(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	478 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	479 char *next;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	480 Py_ssize_t n;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	481
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	482 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	483
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	484 n = get_line(self, &next);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	485
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	486 if (!next \|\| n == 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	487 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	488
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	489 return PyBytes_FromStringAndSize(next, n);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	490 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	491
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	492 PyDoc_STRVAR(seek_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	493 "seek(pos, whence=0) -> int. Change stream position.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	494 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	495 "Seek to byte offset pos relative to position indicated by whence:\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	496 " 0 Start of stream (the default). pos should be >= 0;\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	497 " 1 Current position - pos may be negative;\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	498 " 2 End of stream - pos usually negative.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	499 "Returns the new absolute position.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	500
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	501 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	502 bytesio_seek(bytesio self, PyObject args)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	503 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	504 PyObject *posobj;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	505 Py_ssize_t pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	506 int mode = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	507
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	508 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	509
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	510 if (!PyArg_ParseTuple(args, "O\|i:seek", &posobj, &mode))
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	511 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	512
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	513 pos = PyNumber_AsSsize_t(posobj, PyExc_OverflowError);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	514 if (pos == -1 && PyErr_Occurred())
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	515 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	516
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	517 if (pos < 0 && mode == 0) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	518 PyErr_Format(PyExc_ValueError,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	519 "negative seek value %zd", pos);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	520 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	521 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	522
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	523 /* mode 0: offset relative to beginning of the string.
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	524 mode 1: offset relative to current position.
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	525 mode 2: offset relative the end of the string. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	526 if (mode == 1) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	527 if (pos > PY_SSIZE_T_MAX - self->pos) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	528 PyErr_SetString(PyExc_OverflowError,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	529 "new position too large");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	530 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	531 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	532 pos += self->pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	533 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	534 else if (mode == 2) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	535 if (pos > PY_SSIZE_T_MAX - self->string_size) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	536 PyErr_SetString(PyExc_OverflowError,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	537 "new position too large");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	538 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	539 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	540 pos += self->string_size;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	541 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	542 else if (mode != 0) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	543 PyErr_Format(PyExc_ValueError,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	544 "invalid whence (%i, should be 0, 1 or 2)", mode);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	545 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	546 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	547
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	548 if (pos < 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	549 pos = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	550 self->pos = pos;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	551
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	552 return PyLong_FromSsize_t(self->pos);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	553 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	554
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	555 PyDoc_STRVAR(write_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	556 "write(bytes) -> int. Write bytes to file.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	557 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	558 "Return the number of bytes written.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	559
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	560 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	561 bytesio_write(bytesio self, PyObject obj)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	562 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	563 Py_ssize_t n = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	564 Py_buffer buf;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	565 PyObject *result = NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	566
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	567 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	568
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	569 if (PyObject_GetBuffer(obj, &buf, PyBUF_CONTIG_RO) < 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	570 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	571
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	572 if (buf.len != 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	573 n = write_bytes(self, buf.buf, buf.len);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	574 if (n >= 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	575 result = PyLong_FromSsize_t(n);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	576
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	577 PyBuffer_Release(&buf);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	578 return result;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	579 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	580
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	581 PyDoc_STRVAR(writelines_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	582 "writelines(sequence_of_strings) -> None. Write strings to the file.\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	583 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	584 "Note that newlines are not added. The sequence can be any iterable\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	585 "object producing strings. This is equivalent to calling write() for\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	586 "each string.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	587
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	588 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	589 bytesio_writelines(bytesio self, PyObject v)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	590 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	591 PyObject it, item;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	592 PyObject *ret;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	593
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	594 CHECK_CLOSED(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	595
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	596 it = PyObject_GetIter(v);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	597 if (it == NULL)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	598 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	599
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	600 while ((item = PyIter_Next(it)) != NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	601 ret = bytesio_write(self, item);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	602 Py_DECREF(item);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	603 if (ret == NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	604 Py_DECREF(it);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	605 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	606 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	607 Py_DECREF(ret);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	608 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	609 Py_DECREF(it);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	610
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	611 /* See if PyIter_Next failed */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	612 if (PyErr_Occurred())
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	613 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	614
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	615 Py_RETURN_NONE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	616 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	617
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	618 PyDoc_STRVAR(close_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	619 "close() -> None. Disable all I/O operations.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	620
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	621 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	622 bytesio_close(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	623 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	624 if (self->buf != NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	625 PyMem_Free(self->buf);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	626 self->buf = NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	627 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	628 Py_RETURN_NONE;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	629 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	630
56703 264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	631 /* Pickling support.
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	632
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	633 Note that only pickle protocol 2 and onward are supported since we use
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	634 extended __reduce__ API of PEP 307 to make BytesIO instances picklable.
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	635
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	636 Providing support for protocol < 2 would require the __reduce_ex__ method
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	637 which is notably long-winded when defined properly.
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	638
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	639 For BytesIO, the implementation would similar to one coded for
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	640 object.__reduce_ex__, but slightly less general. To be more specific, we
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	641 could call bytesio_getstate directly and avoid checking for the presence of
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	642 a fallback __reduce__ method. However, we would still need a __newobj__
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	643 function to use the efficient instance representation of PEP 307.
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	644 */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	645
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	646 static PyObject *
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	647 bytesio_getstate(bytesio *self)
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	648 {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	649 PyObject *initvalue = bytesio_getvalue(self);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	650 PyObject *dict;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	651 PyObject *state;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	652
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	653 if (initvalue == NULL)
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	654 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	655 if (self->dict == NULL) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	656 Py_INCREF(Py_None);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	657 dict = Py_None;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	658 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	659 else {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	660 dict = PyDict_Copy(self->dict);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	661 if (dict == NULL)
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	662 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	663 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	664
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	665 state = Py_BuildValue("(OnN)", initvalue, self->pos, dict);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	666 Py_DECREF(initvalue);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	667 return state;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	668 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	669
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	670 static PyObject *
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	671 bytesio_setstate(bytesio self, PyObject state)
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	672 {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	673 PyObject *result;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	674 PyObject *position_obj;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	675 PyObject *dict;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	676 Py_ssize_t pos;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	677
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	678 assert(state != NULL);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	679
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	680 /* We allow the state tuple to be longer than 3, because we may need
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	681 someday to extend the object's state without breaking
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	682 backward-compatibility. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	683 if (!PyTuple_Check(state) \|\| Py_SIZE(state) < 3) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	684 PyErr_Format(PyExc_TypeError,
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	685 "%.200s.__setstate__ argument should be 3-tuple, got %.200s",
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	686 Py_TYPE(self)->tp_name, Py_TYPE(state)->tp_name);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	687 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	688 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	689 /* Reset the object to its default state. This is only needed to handle
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	690 the case of repeated calls to __setstate__. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	691 self->string_size = 0;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	692 self->pos = 0;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	693
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	694 /* Set the value of the internal buffer. If state[0] does not support the
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	695 buffer protocol, bytesio_write will raise the appropriate TypeError. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	696 result = bytesio_write(self, PyTuple_GET_ITEM(state, 0));
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	697 if (result == NULL)
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	698 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	699 Py_DECREF(result);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	700
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	701 /* Set carefully the position value. Alternatively, we could use the seek
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	702 method instead of modifying self->pos directly to better protect the
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	703 object internal state against errneous (or malicious) inputs. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	704 position_obj = PyTuple_GET_ITEM(state, 1);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	705 if (!PyIndex_Check(position_obj)) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	706 PyErr_Format(PyExc_TypeError,
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	707 "second item of state must be an integer, not %.200s",
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	708 Py_TYPE(position_obj)->tp_name);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	709 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	710 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	711 pos = PyNumber_AsSsize_t(position_obj, PyExc_OverflowError);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	712 if (pos == -1 && PyErr_Occurred())
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	713 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	714 if (pos < 0) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	715 PyErr_SetString(PyExc_ValueError,
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	716 "position value cannot be negative");
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	717 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	718 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	719 self->pos = pos;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	720
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	721 /* Set the dictionary of the instance variables. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	722 dict = PyTuple_GET_ITEM(state, 2);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	723 if (dict != Py_None) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	724 if (!PyDict_Check(dict)) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	725 PyErr_Format(PyExc_TypeError,
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	726 "third item of state should be a dict, got a %.200s",
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	727 Py_TYPE(dict)->tp_name);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	728 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	729 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	730 if (self->dict) {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	731 /* Alternatively, we could replace the internal dictionary
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	732 completely. However, it seems more practical to just update it. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	733 if (PyDict_Update(self->dict, dict) < 0)
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	734 return NULL;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	735 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	736 else {
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	737 Py_INCREF(dict);
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	738 self->dict = dict;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	739 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	740 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	741
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	742 Py_RETURN_NONE;
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	743 }
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	744
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	745 static void
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	746 bytesio_dealloc(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	747 {
56699 2f307c7adb8f Manual py3k backport: [svn r74155] Issue #6242: Fix deallocator of io.StringIO and io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 54717 diff changeset	748 _PyObject_GC_UNTRACK(self);
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	749 if (self->buf != NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	750 PyMem_Free(self->buf);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	751 self->buf = NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	752 }
56699 2f307c7adb8f Manual py3k backport: [svn r74155] Issue #6242: Fix deallocator of io.StringIO and io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 54717 diff changeset	753 Py_CLEAR(self->dict);
2f307c7adb8f Manual py3k backport: [svn r74155] Issue #6242: Fix deallocator of io.StringIO and io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 54717 diff changeset	754 if (self->weakreflist != NULL)
2f307c7adb8f Manual py3k backport: [svn r74155] Issue #6242: Fix deallocator of io.StringIO and io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 54717 diff changeset	755 PyObject_ClearWeakRefs((PyObject *) self);
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	756 Py_TYPE(self)->tp_free(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	757 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	758
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	759 static PyObject *
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	760 bytesio_new(PyTypeObject type, PyObject args, PyObject *kwds)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	761 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	762 bytesio *self;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	763
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	764 assert(type != NULL && type->tp_alloc != NULL);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	765 self = (bytesio *)type->tp_alloc(type, 0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	766 if (self == NULL)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	767 return NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	768
56703 264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	769 /* tp_alloc initializes all the fields to zero. So we don't have to
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	770 initialize them here. */
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	771
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	772 self->buf = (char *)PyMem_Malloc(0);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	773 if (self->buf == NULL) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	774 Py_DECREF(self);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	775 return PyErr_NoMemory();
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	776 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	777
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	778 return (PyObject *)self;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	779 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	780
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	781 static int
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	782 bytesio_init(bytesio self, PyObject args, PyObject *kwds)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	783 {
56704 ae02b168162c Manual py3k backport: [svn r74316] Issue #5449: Fix io.BytesIO to not accept arbitrary keywords Antoine Pitrou <solipsis@pitrou.net> parents: 56703 diff changeset	784 char *kwlist[] = {"initial_bytes", NULL};
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	785 PyObject *initvalue = NULL;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	786
56704 ae02b168162c Manual py3k backport: [svn r74316] Issue #5449: Fix io.BytesIO to not accept arbitrary keywords Antoine Pitrou <solipsis@pitrou.net> parents: 56703 diff changeset	787 if (!PyArg_ParseTupleAndKeywords(args, kwds, "\|O:BytesIO", kwlist,
ae02b168162c Manual py3k backport: [svn r74316] Issue #5449: Fix io.BytesIO to not accept arbitrary keywords Antoine Pitrou <solipsis@pitrou.net> parents: 56703 diff changeset	788 &initvalue))
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	789 return -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	790
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	791 /* In case, __init__ is called multiple times. */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	792 self->string_size = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	793 self->pos = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	794
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	795 if (initvalue && initvalue != Py_None) {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	796 PyObject *res;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	797 res = bytesio_write(self, initvalue);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	798 if (res == NULL)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	799 return -1;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	800 Py_DECREF(res);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	801 self->pos = 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	802 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	803
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	804 return 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	805 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	806
78323 e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	807 static PyObject *
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	808 bytesio_sizeof(bytesio self, void unused)
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	809 {
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	810 Py_ssize_t res;
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	811
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	812 res = sizeof(bytesio);
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	813 if (self->buf)
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	814 res += self->buf_size;
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	815 return PyLong_FromSsize_t(res);
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	816 }
e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	817
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	818 static int
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	819 bytesio_traverse(bytesio self, visitproc visit, void arg)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	820 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	821 Py_VISIT(self->dict);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	822 return 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	823 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	824
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	825 static int
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	826 bytesio_clear(bytesio *self)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	827 {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	828 Py_CLEAR(self->dict);
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	829 return 0;
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	830 }
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	831
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	832
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	833 static PyGetSetDef bytesio_getsetlist[] = {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	834 {"closed", (getter)bytesio_get_closed, NULL,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	835 "True if the file is closed."},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	836 {NULL}, /* sentinel */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	837 };
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	838
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	839 static struct PyMethodDef bytesio_methods[] = {
78854 d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	840 {"readable", (PyCFunction)return_not_closed, METH_NOARGS, readable_doc},
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	841 {"seekable", (PyCFunction)return_not_closed, METH_NOARGS, seekable_doc},
d0ab34d4c733 Issue #15841: The readable(), writable() and seekable() methods of io.BytesIO Antoine Pitrou <solipsis@pitrou.net> parents: 78323 diff changeset	842 {"writable", (PyCFunction)return_not_closed, METH_NOARGS, writable_doc},
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	843 {"close", (PyCFunction)bytesio_close, METH_NOARGS, close_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	844 {"flush", (PyCFunction)bytesio_flush, METH_NOARGS, flush_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	845 {"isatty", (PyCFunction)bytesio_isatty, METH_NOARGS, isatty_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	846 {"tell", (PyCFunction)bytesio_tell, METH_NOARGS, tell_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	847 {"write", (PyCFunction)bytesio_write, METH_O, write_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	848 {"writelines", (PyCFunction)bytesio_writelines, METH_O, writelines_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	849 {"read1", (PyCFunction)bytesio_read1, METH_O, read1_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	850 {"readinto", (PyCFunction)bytesio_readinto, METH_VARARGS, readinto_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	851 {"readline", (PyCFunction)bytesio_readline, METH_VARARGS, readline_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	852 {"readlines", (PyCFunction)bytesio_readlines, METH_VARARGS, readlines_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	853 {"read", (PyCFunction)bytesio_read, METH_VARARGS, read_doc},
64384 4ad7efd909bc Merged revisions 84438 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 61180 diff changeset	854 {"getvalue", (PyCFunction)bytesio_getvalue, METH_NOARGS, getval_doc},
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	855 {"seek", (PyCFunction)bytesio_seek, METH_VARARGS, seek_doc},
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	856 {"truncate", (PyCFunction)bytesio_truncate, METH_VARARGS, truncate_doc},
56703 264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	857 {"__getstate__", (PyCFunction)bytesio_getstate, METH_NOARGS, NULL},
264609fbda48 Manual py3k backport: [svn r74158] Issue #6218: Make io.BytesIO and io.StringIO picklable. Antoine Pitrou <solipsis@pitrou.net> parents: 56699 diff changeset	858 {"__setstate__", (PyCFunction)bytesio_setstate, METH_O, NULL},
78323 e8851809ae83 Issue #15489: Add a __sizeof__ implementation for BytesIO objects. Antoine Pitrou <solipsis@pitrou.net> parents: 66303 diff changeset	859 {"__sizeof__", (PyCFunction)bytesio_sizeof, METH_NOARGS, NULL},
54717 760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	860 {NULL, NULL} /* sentinel */
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	861 };
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	862
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	863 PyDoc_STRVAR(bytesio_doc,
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	864 "BytesIO([buffer]) -> object\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	865 "\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	866 "Create a buffered I/O implementation using an in-memory bytes\n"
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	867 "buffer, ready for reading and writing.");
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	868
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	869 PyTypeObject PyBytesIO_Type = {
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	870 PyVarObject_HEAD_INIT(NULL, 0)
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	871 "_io.BytesIO", /tp_name/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	872 sizeof(bytesio), /tp_basicsize/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	873 0, /tp_itemsize/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	874 (destructor)bytesio_dealloc, /tp_dealloc/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	875 0, /tp_print/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	876 0, /tp_getattr/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	877 0, /tp_setattr/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	878 0, /tp_reserved/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	879 0, /tp_repr/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	880 0, /tp_as_number/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	881 0, /tp_as_sequence/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	882 0, /tp_as_mapping/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	883 0, /tp_hash/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	884 0, /tp_call/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	885 0, /tp_str/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	886 0, /tp_getattro/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	887 0, /tp_setattro/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	888 0, /tp_as_buffer/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	889 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_BASETYPE \|
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	890 Py_TPFLAGS_HAVE_GC, /tp_flags/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	891 bytesio_doc, /tp_doc/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	892 (traverseproc)bytesio_traverse, /tp_traverse/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	893 (inquiry)bytesio_clear, /tp_clear/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	894 0, /tp_richcompare/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	895 offsetof(bytesio, weakreflist), /tp_weaklistoffset/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	896 PyObject_SelfIter, /tp_iter/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	897 (iternextfunc)bytesio_iternext, /tp_iternext/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	898 bytesio_methods, /tp_methods/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	899 0, /tp_members/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	900 bytesio_getsetlist, /tp_getset/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	901 0, /tp_base/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	902 0, /tp_dict/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	903 0, /tp_descr_get/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	904 0, /tp_descr_set/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	905 offsetof(bytesio, dict), /tp_dictoffset/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	906 (initproc)bytesio_init, /tp_init/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	907 0, /tp_alloc/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	908 bytesio_new, /tp_new/
760a710eb6c1 Issue #6215: backport the 3.1 io lib Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	909 };

Mercurial > cpython-lru_cache-weakref

annotate Modules/_io/bytesio.c @ 80661:94de77bd0b4b 2.7