gecko-dev/network/module/nsNetStreamLoader.cpp

221 строка
5.7 KiB
C++

/* -*- 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(nsIURI* aURL,
nsStreamCompleteFunc aFunc,
void* aRef);
virtual ~nsUnicharStreamLoader();
NS_DECL_ISUPPORTS
NS_IMETHOD GetNumCharsRead(PRInt32* aNumBytes);
NS_IMETHOD OnStartRequest(nsIURI* aURL, const char *aContentType);
NS_IMETHOD OnProgress(nsIURI* aURL, PRUint32 aProgress, PRUint32 aProgressMax);
NS_IMETHOD OnStatus(nsIURI* aURL, const PRUnichar* aMsg);
NS_IMETHOD OnStopRequest(nsIURI* aURL, nsresult aStatus, const PRUnichar* aMsg);
NS_IMETHOD GetBindInfo(nsIURI* aURL, nsStreamBindingInfo* aInfo);
NS_IMETHOD OnDataAvailable(nsIURI* aURL, nsIInputStream *aIStream,
PRUint32 aLength);
protected:
nsStreamCompleteFunc mFunc;
void* mRef;
nsString* mData;
};
nsUnicharStreamLoader::nsUnicharStreamLoader(nsIURI* aURL,
nsStreamCompleteFunc aFunc,
void* aRef)
{
NS_INIT_REFCNT();
mFunc = aFunc;
mRef = aRef;
mData = new nsString();
// XXX This is vile vile vile!!!
nsresult rv;
if (aURL) {
rv = NS_OpenURL(aURL, this);
if ((NS_OK != rv) && (nsnull != mFunc)) {
// Thou shalt not call out of scope whilst ones refcnt is zero
mRefCnt = 999;
(*mFunc)(this, *mData, mRef, rv);
mRefCnt = 0;
}
}
}
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)) {
nsIStreamListener* tmp = this;
*aInstancePtr = (void*) tmp;
AddRef();
return NS_OK;
}
if (aIID.Equals(kIUnicharStreamLoaderIID)) {
nsIUnicharStreamLoader* tmp = this;
*aInstancePtr = (void*) tmp;
AddRef();
return NS_OK;
}
if (aIID.Equals(kISupportsIID)) {
nsIUnicharStreamLoader* tmp = this;
nsISupports* tmp2 = tmp;
*aInstancePtr = (void*) tmp2;
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::OnStartRequest(nsIURI* aURL,
const char *aContentType)
{
return NS_OK;
}
NS_IMETHODIMP
nsUnicharStreamLoader::OnProgress(nsIURI* aURL,
PRUint32 aProgress,
PRUint32 aProgressMax)
{
return NS_OK;
}
NS_IMETHODIMP
nsUnicharStreamLoader::OnStatus(nsIURI* aURL, const PRUnichar* aMsg)
{
return NS_OK;
}
NS_IMETHODIMP
nsUnicharStreamLoader::OnStopRequest(nsIURI* aURL,
nsresult aStatus,
const PRUnichar* aMsg)
{
(*mFunc)(this, *mData, mRef, aStatus);
return NS_OK;
}
NS_IMETHODIMP
nsUnicharStreamLoader::GetBindInfo(nsIURI* aURL,
nsStreamBindingInfo* aInfo)
{
return NS_OK;
}
#define BUF_SIZE 1024
NS_IMETHODIMP
nsUnicharStreamLoader::OnDataAvailable(nsIURI* 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, lenRead, &lenRead);
if (NS_OK != rv) {
return rv;
}
mData->Append(buffer, lenRead);
len -= lenRead;
}
return rv;
}
extern NS_NET nsresult
NS_NewUnicharStreamLoader(nsIUnicharStreamLoader** aInstancePtrResult,
nsIURI* 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);
}