First checkin of the Python XPCOM bindings.

git-svn-id: svn://10.0.0.236/trunk@87331 18797224-902f-48f8-a5cc-f745e15eee43

mozilla/extensions/python/xpcom/src/PyIInputStream.cpp

@@ -0,0 +1,127 @@
+/* Copyright (c) 2000-2001 ActiveState Tool Corporation.
+   See the file LICENSE.txt for licensing information. */
+
+//
+// This code is part of the XPCOM extensions for Python.
+//
+// Written September 2000 by Mark Hammond.
+//
+// Based heavily on the Python COM support, which is
+// (c) Mark Hammond and Greg Stein.
+//
+// (c) 2000, ActiveState corp.
+
+#include "PyXPCOM_std.h"
+#include "nsIInputStream.h"
+
+// Prevents us needing to use an nsIScriptableInputStream
+// (and even that can't read binary data!!!)
+
+static nsIInputStream *GetI(PyObject *self) {
+	nsIID iid = NS_GET_IID(nsIInputStream);
+
+	if (!Py_nsISupports::Check(self, iid)) {
+		PyErr_SetString(PyExc_TypeError, "This object is not the correct interface");
+		return NULL;
+	}
+	return (nsIInputStream *)Py_nsISupports::GetI(self);
+}
+
+static PyObject *DoPyRead_Buffer(nsIInputStream *pI, PyObject *obBuffer, PRUint32 n)
+{
+	PRUint32 nread;
+	void *buf;
+	PRUint32 buf_len;
+	if (PyObject_AsWriteBuffer(obBuffer, &buf, (int *)&buf_len) != 0) {
+		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==-1) {
+		nsresult r;
+		Py_BEGIN_ALLOW_THREADS;
+		r = pI->Available(&n);
+		Py_END_ALLOW_THREADS;
+		if (NS_FAILED(r))
+			return PyXPCOM_BuildPyException(r);
+	}
+	char *buf = (char *)nsAllocator::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) ) {
+		rc = PyBuffer_New(nread);
+		if (rc != NULL) {
+			void *ob_buf;
+			PRUint32 buf_len;
+			if (PyObject_AsWriteBuffer(rc, &ob_buf, (int *)&buf_len) != 0) {
+				// 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
+		PyXPCOM_BuildPyException(r);
+	nsAllocator::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}
+};