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
		
			
				
	
	
		
			883 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			883 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
#define PY_SSIZE_T_CLEAN
 | 
						|
#include "Python.h"
 | 
						|
#include "structmember.h"
 | 
						|
#include "_iomodule.h"
 | 
						|
 | 
						|
/* Implementation note: the buffer is always at least one character longer
 | 
						|
   than the enclosed string, for proper functioning of _PyIO_find_line_ending.
 | 
						|
*/
 | 
						|
 | 
						|
typedef struct {
 | 
						|
    PyObject_HEAD
 | 
						|
    Py_UNICODE *buf;
 | 
						|
    Py_ssize_t pos;
 | 
						|
    Py_ssize_t string_size;
 | 
						|
    size_t buf_size;
 | 
						|
 | 
						|
    char ok; /* initialized? */
 | 
						|
    char closed;
 | 
						|
    char readuniversal;
 | 
						|
    char readtranslate;
 | 
						|
    PyObject *decoder;
 | 
						|
    PyObject *readnl;
 | 
						|
    PyObject *writenl;
 | 
						|
    
 | 
						|
    PyObject *dict;
 | 
						|
    PyObject *weakreflist;
 | 
						|
} stringio;
 | 
						|
 | 
						|
#define CHECK_INITIALIZED(self) \
 | 
						|
    if (self->ok <= 0) { \
 | 
						|
        PyErr_SetString(PyExc_ValueError, \
 | 
						|
            "I/O operation on uninitialized object"); \
 | 
						|
        return NULL; \
 | 
						|
    }
 | 
						|
 | 
						|
#define CHECK_CLOSED(self) \
 | 
						|
    if (self->closed) { \
 | 
						|
        PyErr_SetString(PyExc_ValueError, \
 | 
						|
            "I/O operation on closed file"); \
 | 
						|
        return NULL; \
 | 
						|
    }
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_doc,
 | 
						|
    "Text I/O implementation using an in-memory buffer.\n"
 | 
						|
    "\n"
 | 
						|
    "The initial_value argument sets the value of object.  The newline\n"
 | 
						|
    "argument is like the one of TextIOWrapper's constructor.");
 | 
						|
 | 
						|
 | 
						|
/* Internal routine for changing the size, in terms of characters, of the
 | 
						|
   buffer of StringIO objects.  The caller should ensure that the 'size'
 | 
						|
   argument is non-negative.  Returns 0 on success, -1 otherwise. */
 | 
						|
static int
 | 
						|
resize_buffer(stringio *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;
 | 
						|
    Py_UNICODE *new_buf = NULL;
 | 
						|
 | 
						|
    assert(self->buf != NULL);
 | 
						|
 | 
						|
    /* Reserve one more char for line ending detection. */
 | 
						|
    size = size + 1;
 | 
						|
    /* 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(Py_UNICODE))
 | 
						|
        goto overflow;
 | 
						|
    new_buf = (Py_UNICODE *)PyMem_Realloc(self->buf,
 | 
						|
                                          alloc * sizeof(Py_UNICODE));
 | 
						|
    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 whole PyUnicode object to the buffer of a
 | 
						|
   StringIO object. Returns 0 on success, or -1 on error. */
 | 
						|
static Py_ssize_t
 | 
						|
write_str(stringio *self, PyObject *obj)
 | 
						|
{
 | 
						|
    Py_UNICODE *str;
 | 
						|
    Py_ssize_t len;
 | 
						|
    PyObject *decoded = NULL;
 | 
						|
    assert(self->buf != NULL);
 | 
						|
    assert(self->pos >= 0);
 | 
						|
 | 
						|
    if (self->decoder != NULL) {
 | 
						|
        decoded = _PyIncrementalNewlineDecoder_decode(
 | 
						|
            self->decoder, obj, 1 /* always final */);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        decoded = obj;
 | 
						|
        Py_INCREF(decoded);
 | 
						|
    }
 | 
						|
    if (self->writenl) {
 | 
						|
        PyObject *translated = PyUnicode_Replace(
 | 
						|
            decoded, _PyIO_str_nl, self->writenl, -1);
 | 
						|
        Py_DECREF(decoded);
 | 
						|
        decoded = translated;
 | 
						|
    }
 | 
						|
    if (decoded == NULL)
 | 
						|
        return -1;
 | 
						|
 | 
						|
    assert(PyUnicode_Check(decoded));
 | 
						|
    str = PyUnicode_AS_UNICODE(decoded);
 | 
						|
    len = PyUnicode_GET_SIZE(decoded);
 | 
						|
 | 
						|
    assert(len >= 0);
 | 
						|
 | 
						|
    /* This overflow check is not strictly necessary. However, it avoids us to
 | 
						|
       deal with funky things like comparing an unsigned and a signed
 | 
						|
       integer. */
 | 
						|
    if (self->pos > PY_SSIZE_T_MAX - len) {
 | 
						|
        PyErr_SetString(PyExc_OverflowError,
 | 
						|
                        "new position too large");
 | 
						|
        goto fail;
 | 
						|
    }
 | 
						|
    if (self->pos + len > self->string_size) {
 | 
						|
        if (resize_buffer(self, self->pos + len) < 0)
 | 
						|
            goto fail;
 | 
						|
    }
 | 
						|
 | 
						|
    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(Py_UNICODE));
 | 
						|
    }
 | 
						|
 | 
						|
    /* Copy the data to the internal buffer, overwriting some of the
 | 
						|
       existing data if self->pos < self->string_size. */
 | 
						|
    memcpy(self->buf + self->pos, str, len * sizeof(Py_UNICODE));
 | 
						|
    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;
 | 
						|
    }
 | 
						|
 | 
						|
    Py_DECREF(decoded);
 | 
						|
    return 0;
 | 
						|
 | 
						|
fail:
 | 
						|
    Py_XDECREF(decoded);
 | 
						|
    return -1;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_getvalue_doc,
 | 
						|
    "Retrieve the entire contents of the object.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_getvalue(stringio *self)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
    return PyUnicode_FromUnicode(self->buf, self->string_size);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_tell_doc,
 | 
						|
    "Tell the current file position.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_tell(stringio *self)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
    return PyLong_FromSsize_t(self->pos);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_read_doc,
 | 
						|
    "Read at most n characters, returned as a string.\n"
 | 
						|
    "\n"
 | 
						|
    "If the argument is negative or omitted, read until EOF\n"
 | 
						|
    "is reached. Return an empty string at EOF.\n");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_read(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t size, n;
 | 
						|
    Py_UNICODE *output;
 | 
						|
    PyObject *arg = Py_None;
 | 
						|
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    if (!PyArg_ParseTuple(args, "|O:read", &arg))
 | 
						|
        return NULL;
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
 | 
						|
    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;
 | 
						|
    }
 | 
						|
 | 
						|
    output = self->buf + self->pos;
 | 
						|
    self->pos += size;
 | 
						|
    return PyUnicode_FromUnicode(output, size);
 | 
						|
}
 | 
						|
 | 
						|
/* Internal helper, used by stringio_readline and stringio_iternext */
 | 
						|
static PyObject *
 | 
						|
_stringio_readline(stringio *self, Py_ssize_t limit)
 | 
						|
{
 | 
						|
    Py_UNICODE *start, *end, old_char;
 | 
						|
    Py_ssize_t len, consumed;
 | 
						|
 | 
						|
    /* In case of overseek, return the empty string */
 | 
						|
    if (self->pos >= self->string_size)
 | 
						|
        return PyUnicode_FromString("");
 | 
						|
 | 
						|
    start = self->buf + self->pos;
 | 
						|
    if (limit < 0 || limit > self->string_size - self->pos)
 | 
						|
        limit = self->string_size - self->pos;
 | 
						|
 | 
						|
    end = start + limit;
 | 
						|
    old_char = *end;
 | 
						|
    *end = '\0';
 | 
						|
    len = _PyIO_find_line_ending(
 | 
						|
        self->readtranslate, self->readuniversal, self->readnl,
 | 
						|
        start, end, &consumed);
 | 
						|
    *end = old_char;
 | 
						|
    /* If we haven't found any line ending, we just return everything
 | 
						|
       (`consumed` is ignored). */
 | 
						|
    if (len < 0)
 | 
						|
        len = limit;
 | 
						|
    self->pos += len;
 | 
						|
    return PyUnicode_FromUnicode(start, len);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_readline_doc,
 | 
						|
    "Read until newline or EOF.\n"
 | 
						|
    "\n"
 | 
						|
    "Returns an empty string if EOF is hit immediately.\n");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_readline(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    PyObject *arg = Py_None;
 | 
						|
    Py_ssize_t limit = -1;
 | 
						|
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    if (!PyArg_ParseTuple(args, "|O:readline", &arg))
 | 
						|
        return NULL;
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
 | 
						|
    if (PyNumber_Check(arg)) {
 | 
						|
        limit = PyNumber_AsSsize_t(arg, PyExc_OverflowError);
 | 
						|
        if (limit == -1 && PyErr_Occurred())
 | 
						|
            return NULL;
 | 
						|
    }
 | 
						|
    else if (arg != Py_None) {
 | 
						|
        PyErr_Format(PyExc_TypeError, "integer argument expected, got '%s'",
 | 
						|
                     Py_TYPE(arg)->tp_name);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return _stringio_readline(self, limit);
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_iternext(stringio *self)
 | 
						|
{
 | 
						|
    PyObject *line;
 | 
						|
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
 | 
						|
    if (Py_TYPE(self) == &PyStringIO_Type) {
 | 
						|
        /* Skip method call overhead for speed */
 | 
						|
        line = _stringio_readline(self, -1);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        /* XXX is subclassing StringIO really supported? */
 | 
						|
        line = PyObject_CallMethodObjArgs((PyObject *)self,
 | 
						|
                                           _PyIO_str_readline, NULL);
 | 
						|
        if (line && !PyUnicode_Check(line)) {
 | 
						|
            PyErr_Format(PyExc_IOError,
 | 
						|
                         "readline() should have returned an str object, "
 | 
						|
                         "not '%.200s'", Py_TYPE(line)->tp_name);
 | 
						|
            Py_DECREF(line);
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    if (line == NULL)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (PyUnicode_GET_SIZE(line) == 0) {
 | 
						|
        /* Reached EOF */
 | 
						|
        Py_DECREF(line);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    return line;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_truncate_doc,
 | 
						|
    "Truncate size to pos.\n"
 | 
						|
    "\n"
 | 
						|
    "The pos argument defaults to the current file position, as\n"
 | 
						|
    "returned by tell().  The current file position is unchanged.\n"
 | 
						|
    "Returns the new absolute position.\n");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_truncate(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t size;
 | 
						|
    PyObject *arg = Py_None;
 | 
						|
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    if (!PyArg_ParseTuple(args, "|O:truncate", &arg))
 | 
						|
        return NULL;
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
 | 
						|
    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) {
 | 
						|
        if (resize_buffer(self, size) < 0)
 | 
						|
            return NULL;
 | 
						|
        self->string_size = size;
 | 
						|
    }
 | 
						|
 | 
						|
    return PyLong_FromSsize_t(size);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_seek_doc,
 | 
						|
    "Change stream position.\n"
 | 
						|
    "\n"
 | 
						|
    "Seek to character offset pos relative to position indicated by whence:\n"
 | 
						|
    "    0  Start of stream (the default).  pos should be >= 0;\n"
 | 
						|
    "    1  Current position - pos must be 0;\n"
 | 
						|
    "    2  End of stream - pos must be 0.\n"
 | 
						|
    "Returns the new absolute position.\n");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_seek(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    PyObject *posobj;
 | 
						|
    Py_ssize_t pos;
 | 
						|
    int mode = 0;
 | 
						|
 | 
						|
    CHECK_INITIALIZED(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;
 | 
						|
    
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
 | 
						|
    if (mode != 0 && mode != 1 && mode != 2) {
 | 
						|
        PyErr_Format(PyExc_ValueError,
 | 
						|
                     "Invalid whence (%i, should be 0, 1 or 2)", mode);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    else if (pos < 0 && mode == 0) {
 | 
						|
        PyErr_Format(PyExc_ValueError,
 | 
						|
                     "Negative seek position %zd", pos);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    else if (mode != 0 && pos != 0) {
 | 
						|
        PyErr_SetString(PyExc_IOError,
 | 
						|
                        "Can't do nonzero cur-relative seeks");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    /* mode 0: offset relative to beginning of the string.
 | 
						|
       mode 1: no change to current position.
 | 
						|
       mode 2: change position to end of file. */
 | 
						|
    if (mode == 1) {
 | 
						|
        pos = self->pos;
 | 
						|
    }
 | 
						|
    else if (mode == 2) {
 | 
						|
        pos = self->string_size;
 | 
						|
    }
 | 
						|
 | 
						|
    self->pos = pos;
 | 
						|
 | 
						|
    return PyLong_FromSsize_t(self->pos);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_write_doc,
 | 
						|
    "Write string to file.\n"
 | 
						|
    "\n"
 | 
						|
    "Returns the number of characters written, which is always equal to\n"
 | 
						|
    "the length of the string.\n");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_write(stringio *self, PyObject *obj)
 | 
						|
{
 | 
						|
    Py_ssize_t size;
 | 
						|
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    if (!PyUnicode_Check(obj)) {
 | 
						|
        PyErr_Format(PyExc_TypeError, "string argument expected, got '%s'",
 | 
						|
                     Py_TYPE(obj)->tp_name);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
    size = PyUnicode_GET_SIZE(obj);
 | 
						|
 | 
						|
    if (size > 0 && write_str(self, obj) < 0)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    return PyLong_FromSsize_t(size);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(stringio_close_doc,
 | 
						|
    "Close the IO object. Attempting any further operation after the\n"
 | 
						|
    "object is closed will raise a ValueError.\n"
 | 
						|
    "\n"
 | 
						|
    "This method has no effect if the file is already closed.\n");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_close(stringio *self)
 | 
						|
{
 | 
						|
    self->closed = 1;
 | 
						|
    /* Free up some memory */
 | 
						|
    if (resize_buffer(self, 0) < 0)
 | 
						|
        return NULL;
 | 
						|
    Py_CLEAR(self->readnl);
 | 
						|
    Py_CLEAR(self->writenl);
 | 
						|
    Py_CLEAR(self->decoder);
 | 
						|
    Py_RETURN_NONE;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
stringio_traverse(stringio *self, visitproc visit, void *arg)
 | 
						|
{
 | 
						|
    Py_VISIT(self->dict);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
stringio_clear(stringio *self)
 | 
						|
{
 | 
						|
    Py_CLEAR(self->dict);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
stringio_dealloc(stringio *self)
 | 
						|
{
 | 
						|
    _PyObject_GC_UNTRACK(self);
 | 
						|
    self->ok = 0;
 | 
						|
    if (self->buf) {
 | 
						|
        PyMem_Free(self->buf);
 | 
						|
        self->buf = NULL;
 | 
						|
    }
 | 
						|
    Py_CLEAR(self->readnl);
 | 
						|
    Py_CLEAR(self->writenl);
 | 
						|
    Py_CLEAR(self->decoder);
 | 
						|
    Py_CLEAR(self->dict);
 | 
						|
    if (self->weakreflist != NULL)
 | 
						|
        PyObject_ClearWeakRefs((PyObject *) self);
 | 
						|
    Py_TYPE(self)->tp_free(self);
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    stringio *self;
 | 
						|
 | 
						|
    assert(type != NULL && type->tp_alloc != NULL);
 | 
						|
    self = (stringio *)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 = (Py_UNICODE *)PyMem_Malloc(0);
 | 
						|
    if (self->buf == NULL) {
 | 
						|
        Py_DECREF(self);
 | 
						|
        return PyErr_NoMemory();
 | 
						|
    }
 | 
						|
 | 
						|
    return (PyObject *)self;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
stringio_init(stringio *self, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    char *kwlist[] = {"initial_value", "newline", NULL};
 | 
						|
    PyObject *value = NULL;
 | 
						|
    char *newline = "\n";
 | 
						|
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oz:__init__", kwlist,
 | 
						|
                                     &value, &newline))
 | 
						|
        return -1;
 | 
						|
 | 
						|
    if (newline && newline[0] != '\0'
 | 
						|
        && !(newline[0] == '\n' && newline[1] == '\0')
 | 
						|
        && !(newline[0] == '\r' && newline[1] == '\0')
 | 
						|
        && !(newline[0] == '\r' && newline[1] == '\n' && newline[2] == '\0')) {
 | 
						|
        PyErr_Format(PyExc_ValueError,
 | 
						|
                     "illegal newline value: %s", newline);
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    if (value && value != Py_None && !PyUnicode_Check(value)) {
 | 
						|
        PyErr_Format(PyExc_TypeError,
 | 
						|
                     "initial_value must be unicode or None, not %.200s",
 | 
						|
                     Py_TYPE(value)->tp_name);
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
 | 
						|
    self->ok = 0;
 | 
						|
 | 
						|
    Py_CLEAR(self->readnl);
 | 
						|
    Py_CLEAR(self->writenl);
 | 
						|
    Py_CLEAR(self->decoder);
 | 
						|
 | 
						|
    if (newline) {
 | 
						|
        self->readnl = PyString_FromString(newline);
 | 
						|
        if (self->readnl == NULL)
 | 
						|
            return -1;
 | 
						|
    }
 | 
						|
    self->readuniversal = (newline == NULL || newline[0] == '\0');
 | 
						|
    self->readtranslate = (newline == NULL);
 | 
						|
    /* If newline == "", we don't translate anything.
 | 
						|
       If newline == "\n" or newline == None, we translate to "\n", which is
 | 
						|
       a no-op.
 | 
						|
       (for newline == None, TextIOWrapper translates to os.sepline, but it
 | 
						|
       is pointless for StringIO)
 | 
						|
    */
 | 
						|
    if (newline != NULL && newline[0] == '\r') {
 | 
						|
        self->writenl = PyUnicode_FromString(newline);
 | 
						|
    }
 | 
						|
 | 
						|
    if (self->readuniversal) {
 | 
						|
        self->decoder = PyObject_CallFunction(
 | 
						|
            (PyObject *)&PyIncrementalNewlineDecoder_Type,
 | 
						|
            "Oi", Py_None, (int) self->readtranslate);
 | 
						|
        if (self->decoder == NULL)
 | 
						|
            return -1;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Now everything is set up, resize buffer to size of initial value,
 | 
						|
       and copy it */
 | 
						|
    self->string_size = 0;
 | 
						|
    if (value && value != Py_None) {
 | 
						|
        Py_ssize_t len = PyUnicode_GetSize(value);
 | 
						|
        /* This is a heuristic, for newline translation might change
 | 
						|
           the string length. */
 | 
						|
        if (resize_buffer(self, len) < 0)
 | 
						|
            return -1;
 | 
						|
        self->pos = 0;
 | 
						|
        if (write_str(self, value) < 0)
 | 
						|
            return -1;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        if (resize_buffer(self, 0) < 0)
 | 
						|
            return -1;
 | 
						|
    }
 | 
						|
    self->pos = 0;
 | 
						|
 | 
						|
    self->closed = 0;
 | 
						|
    self->ok = 1;
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
/* Properties and pseudo-properties */
 | 
						|
static PyObject *
 | 
						|
stringio_seekable(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    Py_RETURN_TRUE;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_readable(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    Py_RETURN_TRUE;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_writable(stringio *self, PyObject *args)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    Py_RETURN_TRUE;
 | 
						|
}
 | 
						|
 | 
						|
/* Pickling support.
 | 
						|
 | 
						|
   The implementation of __getstate__ is similar to the one for BytesIO,
 | 
						|
   except that we also save the newline parameter. For __setstate__ and unlike
 | 
						|
   BytesIO, we call __init__ to restore the object's state. Doing so allows us
 | 
						|
   to avoid decoding the complex newline state while keeping the object
 | 
						|
   representation compact.
 | 
						|
 | 
						|
   See comment in bytesio.c regarding why only pickle protocols and onward are
 | 
						|
   supported.
 | 
						|
*/
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_getstate(stringio *self)
 | 
						|
{
 | 
						|
    PyObject *initvalue = stringio_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("(OOnN)", initvalue,
 | 
						|
                          self->readnl ? self->readnl : Py_None,
 | 
						|
                          self->pos, dict);
 | 
						|
    Py_DECREF(initvalue);
 | 
						|
    return state;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_setstate(stringio *self, PyObject *state)
 | 
						|
{
 | 
						|
    PyObject *initarg;
 | 
						|
    PyObject *position_obj;
 | 
						|
    PyObject *dict;
 | 
						|
    Py_ssize_t pos;
 | 
						|
 | 
						|
    assert(state != NULL);
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
 | 
						|
    /* We allow the state tuple to be longer than 4, because we may need
 | 
						|
       someday to extend the object's state without breaking
 | 
						|
       backward-compatibility. */
 | 
						|
    if (!PyTuple_Check(state) || Py_SIZE(state) < 4) {
 | 
						|
        PyErr_Format(PyExc_TypeError,
 | 
						|
                     "%.200s.__setstate__ argument should be 4-tuple, got %.200s",
 | 
						|
                     Py_TYPE(self)->tp_name, Py_TYPE(state)->tp_name);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Initialize the object's state. */
 | 
						|
    initarg = PyTuple_GetSlice(state, 0, 2);
 | 
						|
    if (initarg == NULL)
 | 
						|
        return NULL;
 | 
						|
    if (stringio_init(self, initarg, NULL) < 0) {
 | 
						|
        Py_DECREF(initarg);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    Py_DECREF(initarg);
 | 
						|
 | 
						|
    /* Restore the buffer state. Even if __init__ did initialize the buffer,
 | 
						|
       we have to initialize it again since __init__ may translates the
 | 
						|
       newlines in the inital_value string. We clearly do not want that
 | 
						|
       because the string value in the state tuple has already been translated
 | 
						|
       once by __init__. So we do not take any chance and replace object's
 | 
						|
       buffer completely. */
 | 
						|
    {
 | 
						|
        Py_UNICODE *buf = PyUnicode_AS_UNICODE(PyTuple_GET_ITEM(state, 0));
 | 
						|
        Py_ssize_t bufsize = PyUnicode_GET_SIZE(PyTuple_GET_ITEM(state, 0));
 | 
						|
        if (resize_buffer(self, bufsize) < 0)
 | 
						|
            return NULL;
 | 
						|
        memcpy(self->buf, buf, bufsize * sizeof(Py_UNICODE));
 | 
						|
        self->string_size = bufsize;
 | 
						|
    }
 | 
						|
 | 
						|
    /* 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, 2);
 | 
						|
    if (!PyIndex_Check(position_obj)) {
 | 
						|
        PyErr_Format(PyExc_TypeError,
 | 
						|
                     "third item of state must be an integer, got %.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, 3);
 | 
						|
    if (dict != Py_None) {
 | 
						|
        if (!PyDict_Check(dict)) {
 | 
						|
            PyErr_Format(PyExc_TypeError,
 | 
						|
                         "fourth 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 PyObject *
 | 
						|
stringio_closed(stringio *self, void *context)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    return PyBool_FromLong(self->closed);
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_line_buffering(stringio *self, void *context)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
    Py_RETURN_FALSE;
 | 
						|
}
 | 
						|
 | 
						|
static PyObject *
 | 
						|
stringio_newlines(stringio *self, void *context)
 | 
						|
{
 | 
						|
    CHECK_INITIALIZED(self);
 | 
						|
    CHECK_CLOSED(self);
 | 
						|
    if (self->decoder == NULL)
 | 
						|
        Py_RETURN_NONE;
 | 
						|
    return PyObject_GetAttr(self->decoder, _PyIO_str_newlines);
 | 
						|
}
 | 
						|
 | 
						|
static struct PyMethodDef stringio_methods[] = {
 | 
						|
    {"close",    (PyCFunction)stringio_close,    METH_NOARGS,  stringio_close_doc},
 | 
						|
    {"getvalue", (PyCFunction)stringio_getvalue, METH_NOARGS,  stringio_getvalue_doc},
 | 
						|
    {"read",     (PyCFunction)stringio_read,     METH_VARARGS, stringio_read_doc},
 | 
						|
    {"readline", (PyCFunction)stringio_readline, METH_VARARGS, stringio_readline_doc},
 | 
						|
    {"tell",     (PyCFunction)stringio_tell,     METH_NOARGS,  stringio_tell_doc},
 | 
						|
    {"truncate", (PyCFunction)stringio_truncate, METH_VARARGS, stringio_truncate_doc},
 | 
						|
    {"seek",     (PyCFunction)stringio_seek,     METH_VARARGS, stringio_seek_doc},
 | 
						|
    {"write",    (PyCFunction)stringio_write,    METH_O,       stringio_write_doc},
 | 
						|
 | 
						|
    {"seekable", (PyCFunction)stringio_seekable, METH_NOARGS},
 | 
						|
    {"readable", (PyCFunction)stringio_readable, METH_NOARGS},
 | 
						|
    {"writable", (PyCFunction)stringio_writable, METH_NOARGS},
 | 
						|
 | 
						|
    {"__getstate__", (PyCFunction)stringio_getstate, METH_NOARGS},
 | 
						|
    {"__setstate__", (PyCFunction)stringio_setstate, METH_O},
 | 
						|
    {NULL, NULL}        /* sentinel */
 | 
						|
};
 | 
						|
 | 
						|
static PyGetSetDef stringio_getset[] = {
 | 
						|
    {"closed",         (getter)stringio_closed,         NULL, NULL},
 | 
						|
    {"newlines",       (getter)stringio_newlines,       NULL, NULL},
 | 
						|
    /*  (following comments straight off of the original Python wrapper:)
 | 
						|
        XXX Cruft to support the TextIOWrapper API. This would only
 | 
						|
        be meaningful if StringIO supported the buffer attribute.
 | 
						|
        Hopefully, a better solution, than adding these pseudo-attributes,
 | 
						|
        will be found.
 | 
						|
    */
 | 
						|
    {"line_buffering", (getter)stringio_line_buffering, NULL, NULL},
 | 
						|
    {NULL}
 | 
						|
};
 | 
						|
 | 
						|
PyTypeObject PyStringIO_Type = {
 | 
						|
    PyVarObject_HEAD_INIT(NULL, 0)
 | 
						|
    "_io.StringIO",                            /*tp_name*/
 | 
						|
    sizeof(stringio),                    /*tp_basicsize*/
 | 
						|
    0,                                         /*tp_itemsize*/
 | 
						|
    (destructor)stringio_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*/
 | 
						|
    stringio_doc,                              /*tp_doc*/
 | 
						|
    (traverseproc)stringio_traverse,           /*tp_traverse*/
 | 
						|
    (inquiry)stringio_clear,                   /*tp_clear*/
 | 
						|
    0,                                         /*tp_richcompare*/
 | 
						|
    offsetof(stringio, weakreflist),            /*tp_weaklistoffset*/
 | 
						|
    0,                                         /*tp_iter*/
 | 
						|
    (iternextfunc)stringio_iternext,           /*tp_iternext*/
 | 
						|
    stringio_methods,                          /*tp_methods*/
 | 
						|
    0,                                         /*tp_members*/
 | 
						|
    stringio_getset,                           /*tp_getset*/
 | 
						|
    0,                                         /*tp_base*/
 | 
						|
    0,                                         /*tp_dict*/
 | 
						|
    0,                                         /*tp_descr_get*/
 | 
						|
    0,                                         /*tp_descr_set*/
 | 
						|
    offsetof(stringio, dict),                  /*tp_dictoffset*/
 | 
						|
    (initproc)stringio_init,                   /*tp_init*/
 | 
						|
    0,                                         /*tp_alloc*/
 | 
						|
    stringio_new,                              /*tp_new*/
 | 
						|
};
 |