Added stream loader

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

mozilla/network/module/nsNetStreamLoader.cpp

@@ -0,0 +1,212 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+ *
+ * The contents of this file are subject to the Netscape Public License
+ * Version 1.0 (the "NPL"); you may not use this file except in
+ * compliance with the NPL.  You may obtain a copy of the NPL at
+ * http://www.mozilla.org/NPL/
+ *
+ * Software distributed under the NPL is distributed on an "AS IS" basis,
+ * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
+ * for the specific language governing rights and limitations under the
+ * NPL.
+ *
+ * The Initial Developer of this code under the NPL is Netscape
+ * Communications Corporation.  Portions created by Netscape are
+ * Copyright (C) 1998 Netscape Communications Corporation.  All Rights
+ * Reserved.
+ */
+
+#include "nsIUnicharStreamLoader.h"
+#include "nsIStreamListener.h"
+#include "nsIInputStream.h"
+#include "nsIURL.h"
+
+static NS_DEFINE_IID(kISupportsIID,     NS_ISUPPORTS_IID);
+static NS_DEFINE_IID(kIStreamListenerIID,  NS_ISTREAMLISTENER_IID);
+static NS_DEFINE_IID(kIUnicharStreamLoaderIID,  NS_IUNICHARSTREAMLOADER_IID);
+
+class nsUnicharStreamLoader : public nsIUnicharStreamLoader,
+                              public nsIStreamListener
+{
+public:
+  nsUnicharStreamLoader(nsIURL* aURL,
+                        nsStreamCompleteFunc aFunc,
+                        void* aRef);
+  ~nsUnicharStreamLoader();
+
+  NS_DECL_ISUPPORTS
+  
+  NS_IMETHOD GetNumCharsRead(PRInt32* aNumBytes);
+
+  NS_IMETHOD OnStartBinding(nsIURL* aURL, const char *aContentType);
+  NS_IMETHOD OnProgress(nsIURL* aURL, PRUint32 aProgress, PRUint32 aProgressMax);
+  NS_IMETHOD OnStatus(nsIURL* aURL, const PRUnichar* aMsg);
+  NS_IMETHOD OnStopBinding(nsIURL* aURL, nsresult aStatus, const PRUnichar* aMsg);
+  NS_IMETHOD GetBindInfo(nsIURL* aURL, nsStreamBindingInfo* aInfo);
+  NS_IMETHOD OnDataAvailable(nsIURL* aURL, nsIInputStream *aIStream, 
+                             PRUint32 aLength);
+
+protected:
+  nsStreamCompleteFunc mFunc;
+  void* mRef;
+  nsString* mData;
+};
+
+
+nsUnicharStreamLoader::nsUnicharStreamLoader(nsIURL* aURL,
+                                             nsStreamCompleteFunc aFunc,
+                                             void* aRef)
+{
+  NS_INIT_REFCNT();
+  mFunc = aFunc;
+  mRef = aRef;
+  mData = new nsString();
+
+  nsresult rv;
+  if (aURL) {
+    rv = NS_OpenURL(aURL, this);
+    if ((NS_OK != rv) && (nsnull != mFunc)) {
+      (*mFunc)(this, *mData, mRef, rv);
+    }
+  }
+}
+
+nsUnicharStreamLoader::~nsUnicharStreamLoader()
+{
+  if (nsnull != mData) {
+    delete mData;
+  }
+}
+
+NS_IMPL_ADDREF(nsUnicharStreamLoader)
+NS_IMPL_RELEASE(nsUnicharStreamLoader)
+
+nsresult 
+nsUnicharStreamLoader::QueryInterface(const nsIID &aIID, void** aInstancePtr)
+{
+  if (NULL == aInstancePtr) {
+    return NS_ERROR_NULL_POINTER;
+  }
+  if (aIID.Equals(kIStreamListenerIID)) {
+    *aInstancePtr = (void*) ((nsIStreamListener*)this);
+    AddRef();
+    return NS_OK;
+  }
+  if (aIID.Equals(kIUnicharStreamLoaderIID)) {
+    *aInstancePtr = (void*) ((nsIUnicharStreamLoader*)this);
+    AddRef();
+    return NS_OK;
+  }
+  if (aIID.Equals(kISupportsIID)) {
+    *aInstancePtr = ((nsISupports *)(nsIUnicharStreamLoader *)this);
+    AddRef();
+    return NS_OK;
+  }
+
+  return NS_NOINTERFACE;
+}
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::GetNumCharsRead(PRInt32* aNumBytes)
+{
+  if (nsnull != mData) {
+    *aNumBytes = mData->Length();
+  }
+  else {
+    *aNumBytes = 0;
+  }
+
+  return NS_OK;
+}
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::OnStartBinding(nsIURL* aURL, 
+                                      const char *aContentType)
+{
+  return NS_OK;
+}
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::OnProgress(nsIURL* aURL, 
+                                  PRUint32 aProgress, 
+                                  PRUint32 aProgressMax)
+{
+  return NS_OK;
+}
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::OnStatus(nsIURL* aURL, const PRUnichar* aMsg)
+{
+  return NS_OK;
+}
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::OnStopBinding(nsIURL* aURL, 
+                                     nsresult aStatus, 
+                                     const PRUnichar* aMsg)
+{
+  (*mFunc)(this, *mData, mRef, aStatus);
+
+  return NS_OK;
+}
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::GetBindInfo(nsIURL* aURL,
+                                   nsStreamBindingInfo* aInfo)
+{
+  return NS_OK;
+}
+
+#define BUF_SIZE 1024
+
+NS_IMETHODIMP 
+nsUnicharStreamLoader::OnDataAvailable(nsIURL* aURL, 
+                                       nsIInputStream *aIStream, 
+                                       PRUint32 aLength)
+{
+  nsresult rv = NS_OK;
+  char buffer[BUF_SIZE];
+  PRUint32 len, lenRead;
+  
+  aIStream->GetLength(&len);
+
+  while (len > 0) {
+    if (len < BUF_SIZE) {
+      lenRead = len;
+    }
+    else {
+      lenRead = BUF_SIZE;
+    }
+
+    rv = aIStream->Read(buffer, 0, lenRead, &lenRead);
+    if (NS_OK != rv) {
+      return rv;
+    }
+
+    mData->Append(buffer, lenRead);
+    len -= lenRead;
+  }
+
+  return rv;
+}
+
+extern NS_NET nsresult 
+NS_NewUnicharStreamLoader(nsIUnicharStreamLoader** aInstancePtrResult,
+                          nsIURL* aURL,
+                          nsStreamCompleteFunc aFunc,
+                          void* aRef)
+{
+  NS_PRECONDITION(nsnull != aInstancePtrResult, "null ptr");
+  if (nsnull == aInstancePtrResult) {
+    return NS_ERROR_NULL_POINTER;
+  }
+
+  nsUnicharStreamLoader* it = new nsUnicharStreamLoader(aURL,
+                                                        aFunc,
+                                                        aRef);
+  if (nsnull == it) {
+    return NS_ERROR_OUT_OF_MEMORY;
+  }
+  return it->QueryInterface(kIUnicharStreamLoaderIID, 
+                            (void **) aInstancePtrResult);
+}