diff -r 000000000000 -r 6474c204b198 netwerk/base/src/nsAsyncStreamCopier.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/netwerk/base/src/nsAsyncStreamCopier.h Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,81 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#ifndef nsAsyncStreamCopier_h__ +#define nsAsyncStreamCopier_h__ + +#include "nsIAsyncStreamCopier.h" +#include "nsIAsyncStreamCopier2.h" +#include "mozilla/Mutex.h" +#include "nsStreamUtils.h" +#include "nsCOMPtr.h" + +class nsIRequestObserver; + +//----------------------------------------------------------------------------- + +class nsAsyncStreamCopier : public nsIAsyncStreamCopier, nsIAsyncStreamCopier2 +{ +public: + NS_DECL_THREADSAFE_ISUPPORTS + NS_DECL_NSIREQUEST + NS_DECL_NSIASYNCSTREAMCOPIER + + // nsIAsyncStreamCopier2 + // We declare it by hand instead of NS_DECL_NSIASYNCSTREAMCOPIER2 + // as nsIAsyncStreamCopier2 duplicates methods of nsIAsyncStreamCopier + NS_IMETHOD Init(nsIInputStream *aSource, + nsIOutputStream *aSink, + nsIEventTarget *aTarget, + uint32_t aChunkSize, + bool aCloseSource, + bool aCloseSink); + + nsAsyncStreamCopier(); + virtual ~nsAsyncStreamCopier(); + + //------------------------------------------------------------------------- + // these methods may be called on any thread + + bool IsComplete(nsresult *status = nullptr); + void Complete(nsresult status); + +private: + nsresult InitInternal(nsIInputStream *source, + nsIOutputStream *sink, + nsIEventTarget *target, + uint32_t chunkSize, + bool closeSource, + bool closeSink); + + static void OnAsyncCopyComplete(void *, nsresult); + + void AsyncCopyInternal(); + nsresult ApplyBufferingPolicy(); + nsIRequest* AsRequest(); + + nsCOMPtr mSource; + nsCOMPtr mSink; + + nsCOMPtr mObserver; + + nsCOMPtr mTarget; + + nsCOMPtr mCopierCtx; + + mozilla::Mutex mLock; + + nsAsyncCopyMode mMode; + uint32_t mChunkSize; + nsresult mStatus; + bool mIsPending; + bool mCloseSource; + bool mCloseSink; + bool mShouldSniffBuffering; + + friend class ProceedWithAsyncCopy; + friend class AsyncApplyBufferingPolicyEvent; +}; + +#endif // !nsAsyncStreamCopier_h__