The Tor Browser / file comparison
comparison: netwerk/base/src/nsRequestObserverProxy.cpp
netwerk/base/src/nsRequestObserverProxy.cpp
- branch
- TOR_BUG_9701
- changeset 15
- b8a032363ba2
equal
deleted
inserted
replaced
| -1:000000000000 | 0:43271a738055 |
|---|---|
| 1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ | |
| 2 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 3 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 5 | |
| 6 #include "mozilla/DebugOnly.h" | |
| 7 | |
| 8 #include "nscore.h" | |
| 9 #include "nsRequestObserverProxy.h" | |
| 10 #include "nsIRequest.h" | |
| 11 #include "nsAutoPtr.h" | |
| 12 #include "prlog.h" | |
| 13 | |
| 14 using namespace mozilla; | |
| 15 | |
| 16 #if defined(PR_LOGGING) | |
| 17 static PRLogModuleInfo *gRequestObserverProxyLog; | |
| 18 #endif | |
| 19 | |
| 20 #undef LOG | |
| 21 #define LOG(args) PR_LOG(gRequestObserverProxyLog, PR_LOG_DEBUG, args) | |
| 22 | |
| 23 //----------------------------------------------------------------------------- | |
| 24 // nsARequestObserverEvent internal class... | |
| 25 //----------------------------------------------------------------------------- | |
| 26 | |
| 27 nsARequestObserverEvent::nsARequestObserverEvent(nsIRequest *request) | |
| 28 : mRequest(request) | |
| 29 { | |
| 30 NS_PRECONDITION(mRequest, "null pointer"); | |
| 31 } | |
| 32 | |
| 33 //----------------------------------------------------------------------------- | |
| 34 // nsOnStartRequestEvent internal class... | |
| 35 //----------------------------------------------------------------------------- | |
| 36 | |
| 37 class nsOnStartRequestEvent : public nsARequestObserverEvent | |
| 38 { | |
| 39 nsRefPtr<nsRequestObserverProxy> mProxy; | |
| 40 public: | |
| 41 nsOnStartRequestEvent(nsRequestObserverProxy *proxy, | |
| 42 nsIRequest *request) | |
| 43 : nsARequestObserverEvent(request) | |
| 44 , mProxy(proxy) | |
| 45 { | |
| 46 NS_PRECONDITION(mProxy, "null pointer"); | |
| 47 } | |
| 48 | |
| 49 virtual ~nsOnStartRequestEvent() {} | |
| 50 | |
| 51 NS_IMETHOD Run() | |
| 52 { | |
| 53 LOG(("nsOnStartRequestEvent::HandleEvent [req=%x]\n", mRequest.get())); | |
| 54 | |
| 55 if (!mProxy->mObserver) { | |
| 56 NS_NOTREACHED("already handled onStopRequest event (observer is null)"); | |
| 57 return NS_OK; | |
| 58 } | |
| 59 | |
| 60 LOG(("handle startevent=%p\n", this)); | |
| 61 nsresult rv = mProxy->mObserver->OnStartRequest(mRequest, mProxy->mContext); | |
| 62 if (NS_FAILED(rv)) { | |
| 63 LOG(("OnStartRequest failed [rv=%x] canceling request!\n", rv)); | |
| 64 rv = mRequest->Cancel(rv); | |
| 65 NS_ASSERTION(NS_SUCCEEDED(rv), "Cancel failed for request!"); | |
| 66 } | |
| 67 | |
| 68 return NS_OK; | |
| 69 } | |
| 70 }; | |
| 71 | |
| 72 //----------------------------------------------------------------------------- | |
| 73 // nsOnStopRequestEvent internal class... | |
| 74 //----------------------------------------------------------------------------- | |
| 75 | |
| 76 class nsOnStopRequestEvent : public nsARequestObserverEvent | |
| 77 { | |
| 78 nsRefPtr<nsRequestObserverProxy> mProxy; | |
| 79 public: | |
| 80 nsOnStopRequestEvent(nsRequestObserverProxy *proxy, | |
| 81 nsIRequest *request) | |
| 82 : nsARequestObserverEvent(request) | |
| 83 , mProxy(proxy) | |
| 84 { | |
| 85 NS_PRECONDITION(mProxy, "null pointer"); | |
| 86 } | |
| 87 | |
| 88 virtual ~nsOnStopRequestEvent() {} | |
| 89 | |
| 90 NS_IMETHOD Run() | |
| 91 { | |
| 92 LOG(("nsOnStopRequestEvent::HandleEvent [req=%x]\n", mRequest.get())); | |
| 93 | |
| 94 nsMainThreadPtrHandle<nsIRequestObserver> observer = mProxy->mObserver; | |
| 95 if (!observer) { | |
| 96 NS_NOTREACHED("already handled onStopRequest event (observer is null)"); | |
| 97 return NS_OK; | |
| 98 } | |
| 99 // Do not allow any more events to be handled after OnStopRequest | |
| 100 mProxy->mObserver = 0; | |
| 101 | |
| 102 nsresult status = NS_OK; | |
| 103 DebugOnly<nsresult> rv = mRequest->GetStatus(&status); | |
| 104 NS_ASSERTION(NS_SUCCEEDED(rv), "GetStatus failed for request!"); | |
| 105 | |
| 106 LOG(("handle stopevent=%p\n", this)); | |
| 107 (void) observer->OnStopRequest(mRequest, mProxy->mContext, status); | |
| 108 | |
| 109 return NS_OK; | |
| 110 } | |
| 111 }; | |
| 112 | |
| 113 //----------------------------------------------------------------------------- | |
| 114 // nsRequestObserverProxy::nsISupports implementation... | |
| 115 //----------------------------------------------------------------------------- | |
| 116 | |
| 117 NS_IMPL_ISUPPORTS(nsRequestObserverProxy, | |
| 118 nsIRequestObserver, | |
| 119 nsIRequestObserverProxy) | |
| 120 | |
| 121 //----------------------------------------------------------------------------- | |
| 122 // nsRequestObserverProxy::nsIRequestObserver implementation... | |
| 123 //----------------------------------------------------------------------------- | |
| 124 | |
| 125 NS_IMETHODIMP | |
| 126 nsRequestObserverProxy::OnStartRequest(nsIRequest *request, | |
| 127 nsISupports *context) | |
| 128 { | |
| 129 MOZ_ASSERT(!context || context == mContext); | |
| 130 LOG(("nsRequestObserverProxy::OnStartRequest [this=%p req=%x]\n", this, request)); | |
| 131 | |
| 132 nsOnStartRequestEvent *ev = | |
| 133 new nsOnStartRequestEvent(this, request); | |
| 134 if (!ev) | |
| 135 return NS_ERROR_OUT_OF_MEMORY; | |
| 136 | |
| 137 LOG(("post startevent=%p\n", ev)); | |
| 138 nsresult rv = FireEvent(ev); | |
| 139 if (NS_FAILED(rv)) | |
| 140 delete ev; | |
| 141 return rv; | |
| 142 } | |
| 143 | |
| 144 NS_IMETHODIMP | |
| 145 nsRequestObserverProxy::OnStopRequest(nsIRequest *request, | |
| 146 nsISupports *context, | |
| 147 nsresult status) | |
| 148 { | |
| 149 MOZ_ASSERT(!context || context == mContext); | |
| 150 LOG(("nsRequestObserverProxy: OnStopRequest [this=%p req=%x status=%x]\n", | |
| 151 this, request, status)); | |
| 152 | |
| 153 // The status argument is ignored because, by the time the OnStopRequestEvent | |
| 154 // is actually processed, the status of the request may have changed :-( | |
| 155 // To make sure that an accurate status code is always used, GetStatus() is | |
| 156 // called when the OnStopRequestEvent is actually processed (see above). | |
| 157 | |
| 158 nsOnStopRequestEvent *ev = | |
| 159 new nsOnStopRequestEvent(this, request); | |
| 160 if (!ev) | |
| 161 return NS_ERROR_OUT_OF_MEMORY; | |
| 162 | |
| 163 LOG(("post stopevent=%p\n", ev)); | |
| 164 nsresult rv = FireEvent(ev); | |
| 165 if (NS_FAILED(rv)) | |
| 166 delete ev; | |
| 167 return rv; | |
| 168 } | |
| 169 | |
| 170 //----------------------------------------------------------------------------- | |
| 171 // nsRequestObserverProxy::nsIRequestObserverProxy implementation... | |
| 172 //----------------------------------------------------------------------------- | |
| 173 | |
| 174 NS_IMETHODIMP | |
| 175 nsRequestObserverProxy::Init(nsIRequestObserver *observer, nsISupports *context) | |
| 176 { | |
| 177 NS_ENSURE_ARG_POINTER(observer); | |
| 178 | |
| 179 #if defined(PR_LOGGING) | |
| 180 if (!gRequestObserverProxyLog) | |
| 181 gRequestObserverProxyLog = PR_NewLogModule("nsRequestObserverProxy"); | |
| 182 #endif | |
| 183 | |
| 184 mObserver = new nsMainThreadPtrHolder<nsIRequestObserver>(observer); | |
| 185 mContext = new nsMainThreadPtrHolder<nsISupports>(context); | |
| 186 | |
| 187 return NS_OK; | |
| 188 } | |
| 189 | |
| 190 //----------------------------------------------------------------------------- | |
| 191 // nsRequestObserverProxy implementation... | |
| 192 //----------------------------------------------------------------------------- | |
| 193 | |
| 194 nsresult | |
| 195 nsRequestObserverProxy::FireEvent(nsARequestObserverEvent *event) | |
| 196 { | |
| 197 nsCOMPtr<nsIEventTarget> mainThread(do_GetMainThread()); | |
| 198 return mainThread->Dispatch(event, NS_DISPATCH_NORMAL); | |
| 199 } |
