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
77
78
79
80
81
82
83
|
/* 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;
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__
|
