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