зеркало из https://github.com/mozilla/gecko-dev.git
77 строки
2.4 KiB
C++
77 строки
2.4 KiB
C++
/* 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 final : 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) override;
|
|
|
|
nsAsyncStreamCopier();
|
|
|
|
//-------------------------------------------------------------------------
|
|
// these methods may be called on any thread
|
|
|
|
bool IsComplete(nsresult* status = nullptr);
|
|
void Complete(nsresult status);
|
|
|
|
private:
|
|
virtual ~nsAsyncStreamCopier();
|
|
|
|
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<nsIInputStream> mSource;
|
|
nsCOMPtr<nsIOutputStream> mSink;
|
|
|
|
nsCOMPtr<nsIRequestObserver> mObserver;
|
|
|
|
nsCOMPtr<nsIEventTarget> mTarget;
|
|
|
|
nsCOMPtr<nsISupports> mCopierCtx GUARDED_BY(mLock);
|
|
|
|
mozilla::Mutex mLock{"nsAsyncStreamCopier.mLock"};
|
|
|
|
nsAsyncCopyMode mMode{NS_ASYNCCOPY_VIA_READSEGMENTS};
|
|
uint32_t mChunkSize; // only modified in Init
|
|
nsresult mStatus GUARDED_BY(mLock){NS_OK};
|
|
bool mIsPending GUARDED_BY(mLock){false};
|
|
bool mCloseSource GUARDED_BY(mLock){false};
|
|
bool mCloseSink GUARDED_BY(mLock){false};
|
|
bool mShouldSniffBuffering{false}; // only modified in Init
|
|
|
|
friend class ProceedWithAsyncCopy;
|
|
friend class AsyncApplyBufferingPolicyEvent;
|
|
};
|
|
|
|
#endif // !nsAsyncStreamCopier_h__
|