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
|
/* 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 nsStreamListenerWrapper_h__
#define nsStreamListenerWrapper_h__
#include "nsCOMPtr.h"
#include "nsIRequest.h"
#include "nsIStreamListener.h"
#include "nsIThreadRetargetableStreamListener.h"
#include "nsIMultiPartChannel.h"
namespace mozilla {
namespace net {
// Wrapper class to make replacement of nsHttpChannel's listener
// from JavaScript possible. It is workaround for bug 433711 and 682305.
class nsStreamListenerWrapper final
: public nsIMultiPartChannelListener,
public nsIThreadRetargetableStreamListener {
public:
explicit nsStreamListenerWrapper(nsIStreamListener* listener)
: mListener(listener) {
MOZ_ASSERT(mListener, "no stream listener specified");
}
NS_DECL_THREADSAFE_ISUPPORTS
NS_FORWARD_SAFE_NSISTREAMLISTENER(mListener)
NS_DECL_NSIMULTIPARTCHANNELLISTENER
NS_DECL_NSITHREADRETARGETABLESTREAMLISTENER
// Don't use NS_FORWARD_NSIREQUESTOBSERVER(mListener->) here, because we need
// to release mListener in OnStopRequest, and IDL-generated function doesn't.
NS_IMETHOD OnStartRequest(nsIRequest* aRequest) override {
// OnStartRequest can come after OnStopRequest in certain cases (multipart
// listeners)
nsCOMPtr<nsIMultiPartChannel> multiPartChannel =
do_QueryInterface(aRequest);
if (multiPartChannel) {
mIsMulti = true;
}
return mListener->OnStartRequest(aRequest);
}
NS_IMETHOD OnStopRequest(nsIRequest* aRequest,
nsresult aStatusCode) override {
nsresult rv = mListener->OnStopRequest(aRequest, aStatusCode);
if (!mIsMulti) {
// Multipart channels can call OnStartRequest again
mListener = nullptr;
}
return rv;
}
private:
bool mIsMulti{false};
~nsStreamListenerWrapper() = default;
nsCOMPtr<nsIStreamListener> mListener;
};
} // namespace net
} // namespace mozilla
#endif // nsStreamListenerWrapper_h__
|
