static PyObject *PyRead(PyObject *self, PyObject *args)
{
PyObject *obBuffer = NULL;
PRUint32 n = (PRUint32)-1;
nsIInputStream *pI = GetI(self);
if (pI==NULL)
return NULL;
if (PyArg_ParseTuple(args, "|i", (int *)&n))
// This worked - no args, or just an int.
return DoPyRead_Size(pI, n);
// try our other supported arg format.
PyErr_Clear();
if (!PyArg_ParseTuple(args, "O|i", &obBuffer, (int *)&n)) {
PyErr_Clear();
PyErr_SetString(PyExc_TypeError, "'read()' must be called as (buffer_ob, int_size=-1) or (int_size=-1)");
return NULL;
}
return DoPyRead_Buffer(pI, obBuffer, n);
}
static PyObject *DoPyRead_Buffer(nsIInputStream *pI, PyObject *obBuffer, PRUint32 n)
{
PRUint32 nread;
void *buf;
#ifndef VBOX /* unsafe cast on 64-bit hosts. */
PRUint32 buf_len;
if (PyObject_AsWriteBuffer(obBuffer, &buf, (Py_ssize_t *)&buf_len) != 0) {
#else /* VBOX */
# if PY_VERSION_HEX >= 0x02050000 || defined(PY_SSIZE_T_MIN)
Py_ssize_t buf_len;
# else
int buf_len;
# endif /* VBOX */
if (PyObject_AsWriteBuffer(obBuffer, &buf, &buf_len) != 0) {
#endif
PyErr_Clear();
PyErr_SetString(PyExc_TypeError, "The buffer object does not have a write buffer!");
return NULL;
}
if (n==(PRUint32)-1) {
n = buf_len;
} else {
if (n > buf_len) {
NS_WARNING("Warning: PyIInputStream::read() was passed an integer size greater than the size of the passed buffer! Buffer size used.\n");
n = buf_len;
}
}
nsresult r;
Py_BEGIN_ALLOW_THREADS;
r = pI->Read((char *)buf, n, &nread);
Py_END_ALLOW_THREADS;
if ( NS_FAILED(r) )
return PyXPCOM_BuildPyException(r);
return PyInt_FromLong(nread);
}
static PyObject *DoPyRead_Size(nsIInputStream *pI, PRUint32 n)
{
if (n==(PRUint32)-1) {
nsresult r;
Py_BEGIN_ALLOW_THREADS;
r = pI->Available(&n);
Py_END_ALLOW_THREADS;
if (NS_FAILED(r))
return PyXPCOM_BuildPyException(r);
}
if (n==0) { // mozilla will assert if we alloc zero bytes.
#if PY_MAJOR_VERSION <= 2
return PyBuffer_New(0);
#else
return PyBytes_FromString("");
#endif
}
char *buf = (char *)nsMemory::Alloc(n);
if (buf==NULL) {
PyErr_NoMemory();
return NULL;
}
nsresult r;
PRUint32 nread;
Py_BEGIN_ALLOW_THREADS;
r = pI->Read(buf, n, &nread);
Py_END_ALLOW_THREADS;
PyObject *rc = NULL;
if ( NS_SUCCEEDED(r) ) {
#if PY_MAJOR_VERSION <= 2
rc = PyBuffer_New(nread);
if (rc != NULL) {
void *ob_buf;
#ifndef VBOX /* unsafe cast on 64-bit hosts. */
PRUint32 buf_len;
if (PyObject_AsWriteBuffer(rc, &ob_buf, (Py_ssize_t *)&buf_len) != 0) {
#else /* VBOX */
# if PY_VERSION_HEX >= 0x02050000 || defined(PY_SSIZE_T_MIN)
Py_ssize_t buf_len;
# else
int buf_len;
# endif /* VBOX */
if (PyObject_AsWriteBuffer(rc, &ob_buf, &buf_len) != 0) {
#endif
// should never fail - we just created it!
return NULL;
}
if (buf_len != nread) {
PyErr_SetString(PyExc_RuntimeError, "New buffer isnt the size we create it!");
return NULL;
}
memcpy(ob_buf, buf, nread);
}
#else
rc = PyBytes_FromStringAndSize(buf, nread);
#endif
} else
PyXPCOM_BuildPyException(r);
nsMemory::Free(buf);
return rc;
}
static PyObject *PyRead(PyObject *self, PyObject *args)
{
PyObject *obBuffer = NULL;
PRUint32 n = (PRUint32)-1;
nsIInputStream *pI = GetI(self);
if (pI==NULL)
return NULL;
if (PyArg_ParseTuple(args, "|i", (int *)&n))
// This worked - no args, or just an int.
return DoPyRead_Size(pI, n);
// try our other supported arg format.
PyErr_Clear();
if (!PyArg_ParseTuple(args, "O|i", &obBuffer, (int *)&n)) {
PyErr_Clear();
PyErr_SetString(PyExc_TypeError, "'read()' must be called as (buffer_ob, int_size=-1) or (int_size=-1)");
return NULL;
}
return DoPyRead_Buffer(pI, obBuffer, n);
}
struct PyMethodDef
PyMethods_IInputStream[] =
{
{ "read", PyRead, 1},
// The rest are handled as normal
{NULL}
};
