1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/netwerk/base/src/nsBufferedStreams.h Wed Dec 31 06:09:35 2014 +0100 1.3 @@ -0,0 +1,119 @@ 1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ 1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public 1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this 1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ 1.8 + 1.9 +#ifndef nsBufferedStreams_h__ 1.10 +#define nsBufferedStreams_h__ 1.11 + 1.12 +#include "nsIBufferedStreams.h" 1.13 +#include "nsIInputStream.h" 1.14 +#include "nsIOutputStream.h" 1.15 +#include "nsISafeOutputStream.h" 1.16 +#include "nsISeekableStream.h" 1.17 +#include "nsIStreamBufferAccess.h" 1.18 +#include "nsCOMPtr.h" 1.19 +#include "nsIIPCSerializableInputStream.h" 1.20 + 1.21 +//////////////////////////////////////////////////////////////////////////////// 1.22 + 1.23 +class nsBufferedStream : public nsISeekableStream 1.24 +{ 1.25 +public: 1.26 + NS_DECL_THREADSAFE_ISUPPORTS 1.27 + NS_DECL_NSISEEKABLESTREAM 1.28 + 1.29 + nsBufferedStream(); 1.30 + virtual ~nsBufferedStream(); 1.31 + 1.32 + nsresult Close(); 1.33 + 1.34 +protected: 1.35 + nsresult Init(nsISupports* stream, uint32_t bufferSize); 1.36 + NS_IMETHOD Fill() = 0; 1.37 + NS_IMETHOD Flush() = 0; 1.38 + 1.39 + uint32_t mBufferSize; 1.40 + char* mBuffer; 1.41 + 1.42 + // mBufferStartOffset is the offset relative to the start of mStream. 1.43 + int64_t mBufferStartOffset; 1.44 + 1.45 + // mCursor is the read cursor for input streams, or write cursor for 1.46 + // output streams, and is relative to mBufferStartOffset. 1.47 + uint32_t mCursor; 1.48 + 1.49 + // mFillPoint is the amount available in the buffer for input streams, 1.50 + // or the high watermark of bytes written into the buffer, and therefore 1.51 + // is relative to mBufferStartOffset. 1.52 + uint32_t mFillPoint; 1.53 + 1.54 + nsISupports* mStream; // cast to appropriate subclass 1.55 + 1.56 + bool mBufferDisabled; 1.57 + bool mEOF; // True if mStream is at EOF 1.58 + uint8_t mGetBufferCount; 1.59 +}; 1.60 + 1.61 +//////////////////////////////////////////////////////////////////////////////// 1.62 + 1.63 +class nsBufferedInputStream : public nsBufferedStream, 1.64 + public nsIBufferedInputStream, 1.65 + public nsIStreamBufferAccess, 1.66 + public nsIIPCSerializableInputStream 1.67 +{ 1.68 +public: 1.69 + NS_DECL_ISUPPORTS_INHERITED 1.70 + NS_DECL_NSIINPUTSTREAM 1.71 + NS_DECL_NSIBUFFEREDINPUTSTREAM 1.72 + NS_DECL_NSISTREAMBUFFERACCESS 1.73 + NS_DECL_NSIIPCSERIALIZABLEINPUTSTREAM 1.74 + 1.75 + nsBufferedInputStream() : nsBufferedStream() {} 1.76 + virtual ~nsBufferedInputStream() {} 1.77 + 1.78 + static nsresult 1.79 + Create(nsISupports *aOuter, REFNSIID aIID, void **aResult); 1.80 + 1.81 + nsIInputStream* Source() { 1.82 + return (nsIInputStream*)mStream; 1.83 + } 1.84 + 1.85 +protected: 1.86 + NS_IMETHOD Fill(); 1.87 + NS_IMETHOD Flush() { return NS_OK; } // no-op for input streams 1.88 +}; 1.89 + 1.90 +//////////////////////////////////////////////////////////////////////////////// 1.91 + 1.92 +class nsBufferedOutputStream : public nsBufferedStream, 1.93 + public nsISafeOutputStream, 1.94 + public nsIBufferedOutputStream, 1.95 + public nsIStreamBufferAccess 1.96 +{ 1.97 +public: 1.98 + NS_DECL_ISUPPORTS_INHERITED 1.99 + NS_DECL_NSIOUTPUTSTREAM 1.100 + NS_DECL_NSISAFEOUTPUTSTREAM 1.101 + NS_DECL_NSIBUFFEREDOUTPUTSTREAM 1.102 + NS_DECL_NSISTREAMBUFFERACCESS 1.103 + 1.104 + nsBufferedOutputStream() : nsBufferedStream() {} 1.105 + virtual ~nsBufferedOutputStream() { nsBufferedOutputStream::Close(); } 1.106 + 1.107 + static nsresult 1.108 + Create(nsISupports *aOuter, REFNSIID aIID, void **aResult); 1.109 + 1.110 + nsIOutputStream* Sink() { 1.111 + return (nsIOutputStream*)mStream; 1.112 + } 1.113 + 1.114 +protected: 1.115 + NS_IMETHOD Fill() { return NS_OK; } // no-op for input streams 1.116 + 1.117 + nsCOMPtr<nsISafeOutputStream> mSafeStream; // QI'd from mStream 1.118 +}; 1.119 + 1.120 +//////////////////////////////////////////////////////////////////////////////// 1.121 + 1.122 +#endif // nsBufferedStreams_h__