1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
/* 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 MOZ_GUARDED_BY(mLock);
mozilla::Mutex mLock{"nsAsyncStreamCopier.mLock"};
nsAsyncCopyMode mMode{NS_ASYNCCOPY_VIA_READSEGMENTS};
uint32_t mChunkSize; // only modified in Init
nsresult mStatus MOZ_GUARDED_BY(mLock){NS_OK};
bool mIsPending MOZ_GUARDED_BY(mLock){false};
bool mCloseSource MOZ_GUARDED_BY(mLock){false};
bool mCloseSink MOZ_GUARDED_BY(mLock){false};
bool mShouldSniffBuffering{false}; // only modified in Init
friend class ProceedWithAsyncCopy;
friend class AsyncApplyBufferingPolicyEvent;
};
#endif // !nsAsyncStreamCopier_h__
|
