gecko-dev/network/module/nsNetStream.h

204 lines
4.7 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
*
* 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.
*/
#ifndef net_strm_h___
#define net_strm_h___
#include "nspr.h"
#include "nsIURL.h"
#include "nsIInputStream.h"
#include "nsIOutputStream.h"
#include "nsIStreamListener.h"
#include "nsIConnectionInfo.h"
/* Forward declaration... */
class nsNetlibStream;
/* Enumeration of the various connection states... */
typedef enum {
nsConnectionActive = 0,
nsConnectionSucceeded,
nsConnectionAborted,
nsConnectionFailed,
nsConnectionMax
} nsConnectionStatus;
class nsConnectionInfo : public nsIConnectionInfo
{
public:
NS_DECL_ISUPPORTS
nsConnectionInfo(nsIURL *aURL,
nsNetlibStream *aStream,
nsIStreamListener *aNotify);
NS_IMETHOD GetURL(nsIURL **aURL);
NS_IMETHOD GetInputStream(nsIInputStream **aStream);
NS_IMETHOD GetOutputStream(nsIOutputStream **aStream);
NS_IMETHOD GetConsumer(nsIStreamListener **aConsumer);
protected:
virtual ~nsConnectionInfo();
public:
nsIURL *pURL;
nsNetlibStream *pNetStream;
nsIStreamListener *pConsumer;
nsConnectionStatus mStatus;
PRThread* mRequestingThread;
};
class nsNetlibStream : public nsIInputStream,
public nsIOutputStream
{
public:
NS_DECL_ISUPPORTS
nsNetlibStream(void);
NS_IMETHOD GetAvailableSpace(PRUint32 *aCount) = 0;
/* From nsIBaseStream interface */
NS_IMETHOD Close(void);
protected:
virtual ~nsNetlibStream();
inline void LockStream (void) { if (m_Lock) PR_EnterMonitor(m_Lock); }
inline void UnlockStream(void) { if (m_Lock) PR_ExitMonitor (m_Lock); }
protected:
PRBool m_bIsClosed;
PRMonitor* m_Lock;
};
/*
* Variable size, buffered stream...
*/
class nsBufferedStream : public nsNetlibStream {
public:
nsBufferedStream(void);
NS_IMETHOD GetAvailableSpace(PRUint32 *aCount);
/* nsIInputStream interface */
NS_IMETHOD GetLength(PRUint32 *aLength);
NS_IMETHOD Read(char *aBuf,
PRUint32 aCount,
PRUint32 *aReadCount);
/* nsIOutputStream interface */
NS_IMETHOD Write(const char *aBuf,
PRUint32 aLen,
PRUint32 *aWriteCount);
protected:
virtual ~nsBufferedStream();
private:
char *m_Buffer;
PRUint32 m_BufferLength;
PRUint32 m_DataLength;
PRUint32 m_ReadOffset;
PRUint32 m_WriteOffset;
};
/*
* Fixed size stream...
*/
class nsAsyncStream : public nsNetlibStream {
public:
nsAsyncStream(PRUint32 buffer_size);
NS_IMETHOD GetAvailableSpace(PRUint32 *aCount);
/* nsIInputStream interface */
NS_IMETHOD GetLength(PRUint32 *aLength);
NS_IMETHOD Read(char *aBuf,
PRUint32 aCount,
PRUint32 *aReadLength);
/* nsIOutputStream interface */
NS_IMETHOD Write(const char *aBuf,
PRUint32 aLen,
PRUint32 *aWriteLength);
protected:
virtual ~nsAsyncStream();
private:
char *m_Buffer;
PRUint32 m_BufferLength;
PRUint32 m_DataLength;
PRUint32 m_ReadOffset;
PRUint32 m_WriteOffset;
};
/*
* Fixed size, blocking stream...
*/
class nsBlockingStream : public nsNetlibStream {
public:
nsBlockingStream(void);
NS_IMETHOD GetAvailableSpace(PRUint32 *aCount);
/* nsIBaseStream interface */
NS_IMETHOD Close(void);
/* nsIInputStream interface */
NS_IMETHOD GetLength(PRUint32 *aLength);
NS_IMETHOD Read(char *aBuf,
PRUint32 aCount,
PRUint32 *aReadLength);
/* nsIOutputStream interface */
NS_IMETHOD Write(const char *aBuf,
PRUint32 aLen,
PRUint32 *aWriteLength);
protected:
virtual ~nsBlockingStream();
PRInt32 ReadBuffer(char *aBuf, PRUint32 aCount);
private:
char *m_Buffer;
PRUint32 m_BufferLength;
PRUint32 m_DataLength;
PRUint32 m_ReadOffset;
PRUint32 m_WriteOffset;
};
#endif /* net_strm_h___ */