1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/netwerk/base/src/nsRequestObserverProxy.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,199 @@
1.4 +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include "mozilla/DebugOnly.h"
1.10 +
1.11 +#include "nscore.h"
1.12 +#include "nsRequestObserverProxy.h"
1.13 +#include "nsIRequest.h"
1.14 +#include "nsAutoPtr.h"
1.15 +#include "prlog.h"
1.16 +
1.17 +using namespace mozilla;
1.18 +
1.19 +#if defined(PR_LOGGING)
1.20 +static PRLogModuleInfo *gRequestObserverProxyLog;
1.21 +#endif
1.22 +
1.23 +#undef LOG
1.24 +#define LOG(args) PR_LOG(gRequestObserverProxyLog, PR_LOG_DEBUG, args)
1.25 +
1.26 +//-----------------------------------------------------------------------------
1.27 +// nsARequestObserverEvent internal class...
1.28 +//-----------------------------------------------------------------------------
1.29 +
1.30 +nsARequestObserverEvent::nsARequestObserverEvent(nsIRequest *request)
1.31 + : mRequest(request)
1.32 +{
1.33 + NS_PRECONDITION(mRequest, "null pointer");
1.34 +}
1.35 +
1.36 +//-----------------------------------------------------------------------------
1.37 +// nsOnStartRequestEvent internal class...
1.38 +//-----------------------------------------------------------------------------
1.39 +
1.40 +class nsOnStartRequestEvent : public nsARequestObserverEvent
1.41 +{
1.42 + nsRefPtr<nsRequestObserverProxy> mProxy;
1.43 +public:
1.44 + nsOnStartRequestEvent(nsRequestObserverProxy *proxy,
1.45 + nsIRequest *request)
1.46 + : nsARequestObserverEvent(request)
1.47 + , mProxy(proxy)
1.48 + {
1.49 + NS_PRECONDITION(mProxy, "null pointer");
1.50 + }
1.51 +
1.52 + virtual ~nsOnStartRequestEvent() {}
1.53 +
1.54 + NS_IMETHOD Run()
1.55 + {
1.56 + LOG(("nsOnStartRequestEvent::HandleEvent [req=%x]\n", mRequest.get()));
1.57 +
1.58 + if (!mProxy->mObserver) {
1.59 + NS_NOTREACHED("already handled onStopRequest event (observer is null)");
1.60 + return NS_OK;
1.61 + }
1.62 +
1.63 + LOG(("handle startevent=%p\n", this));
1.64 + nsresult rv = mProxy->mObserver->OnStartRequest(mRequest, mProxy->mContext);
1.65 + if (NS_FAILED(rv)) {
1.66 + LOG(("OnStartRequest failed [rv=%x] canceling request!\n", rv));
1.67 + rv = mRequest->Cancel(rv);
1.68 + NS_ASSERTION(NS_SUCCEEDED(rv), "Cancel failed for request!");
1.69 + }
1.70 +
1.71 + return NS_OK;
1.72 + }
1.73 +};
1.74 +
1.75 +//-----------------------------------------------------------------------------
1.76 +// nsOnStopRequestEvent internal class...
1.77 +//-----------------------------------------------------------------------------
1.78 +
1.79 +class nsOnStopRequestEvent : public nsARequestObserverEvent
1.80 +{
1.81 + nsRefPtr<nsRequestObserverProxy> mProxy;
1.82 +public:
1.83 + nsOnStopRequestEvent(nsRequestObserverProxy *proxy,
1.84 + nsIRequest *request)
1.85 + : nsARequestObserverEvent(request)
1.86 + , mProxy(proxy)
1.87 + {
1.88 + NS_PRECONDITION(mProxy, "null pointer");
1.89 + }
1.90 +
1.91 + virtual ~nsOnStopRequestEvent() {}
1.92 +
1.93 + NS_IMETHOD Run()
1.94 + {
1.95 + LOG(("nsOnStopRequestEvent::HandleEvent [req=%x]\n", mRequest.get()));
1.96 +
1.97 + nsMainThreadPtrHandle<nsIRequestObserver> observer = mProxy->mObserver;
1.98 + if (!observer) {
1.99 + NS_NOTREACHED("already handled onStopRequest event (observer is null)");
1.100 + return NS_OK;
1.101 + }
1.102 + // Do not allow any more events to be handled after OnStopRequest
1.103 + mProxy->mObserver = 0;
1.104 +
1.105 + nsresult status = NS_OK;
1.106 + DebugOnly<nsresult> rv = mRequest->GetStatus(&status);
1.107 + NS_ASSERTION(NS_SUCCEEDED(rv), "GetStatus failed for request!");
1.108 +
1.109 + LOG(("handle stopevent=%p\n", this));
1.110 + (void) observer->OnStopRequest(mRequest, mProxy->mContext, status);
1.111 +
1.112 + return NS_OK;
1.113 + }
1.114 +};
1.115 +
1.116 +//-----------------------------------------------------------------------------
1.117 +// nsRequestObserverProxy::nsISupports implementation...
1.118 +//-----------------------------------------------------------------------------
1.119 +
1.120 +NS_IMPL_ISUPPORTS(nsRequestObserverProxy,
1.121 + nsIRequestObserver,
1.122 + nsIRequestObserverProxy)
1.123 +
1.124 +//-----------------------------------------------------------------------------
1.125 +// nsRequestObserverProxy::nsIRequestObserver implementation...
1.126 +//-----------------------------------------------------------------------------
1.127 +
1.128 +NS_IMETHODIMP
1.129 +nsRequestObserverProxy::OnStartRequest(nsIRequest *request,
1.130 + nsISupports *context)
1.131 +{
1.132 + MOZ_ASSERT(!context || context == mContext);
1.133 + LOG(("nsRequestObserverProxy::OnStartRequest [this=%p req=%x]\n", this, request));
1.134 +
1.135 + nsOnStartRequestEvent *ev =
1.136 + new nsOnStartRequestEvent(this, request);
1.137 + if (!ev)
1.138 + return NS_ERROR_OUT_OF_MEMORY;
1.139 +
1.140 + LOG(("post startevent=%p\n", ev));
1.141 + nsresult rv = FireEvent(ev);
1.142 + if (NS_FAILED(rv))
1.143 + delete ev;
1.144 + return rv;
1.145 +}
1.146 +
1.147 +NS_IMETHODIMP
1.148 +nsRequestObserverProxy::OnStopRequest(nsIRequest *request,
1.149 + nsISupports *context,
1.150 + nsresult status)
1.151 +{
1.152 + MOZ_ASSERT(!context || context == mContext);
1.153 + LOG(("nsRequestObserverProxy: OnStopRequest [this=%p req=%x status=%x]\n",
1.154 + this, request, status));
1.155 +
1.156 + // The status argument is ignored because, by the time the OnStopRequestEvent
1.157 + // is actually processed, the status of the request may have changed :-(
1.158 + // To make sure that an accurate status code is always used, GetStatus() is
1.159 + // called when the OnStopRequestEvent is actually processed (see above).
1.160 +
1.161 + nsOnStopRequestEvent *ev =
1.162 + new nsOnStopRequestEvent(this, request);
1.163 + if (!ev)
1.164 + return NS_ERROR_OUT_OF_MEMORY;
1.165 +
1.166 + LOG(("post stopevent=%p\n", ev));
1.167 + nsresult rv = FireEvent(ev);
1.168 + if (NS_FAILED(rv))
1.169 + delete ev;
1.170 + return rv;
1.171 +}
1.172 +
1.173 +//-----------------------------------------------------------------------------
1.174 +// nsRequestObserverProxy::nsIRequestObserverProxy implementation...
1.175 +//-----------------------------------------------------------------------------
1.176 +
1.177 +NS_IMETHODIMP
1.178 +nsRequestObserverProxy::Init(nsIRequestObserver *observer, nsISupports *context)
1.179 +{
1.180 + NS_ENSURE_ARG_POINTER(observer);
1.181 +
1.182 +#if defined(PR_LOGGING)
1.183 + if (!gRequestObserverProxyLog)
1.184 + gRequestObserverProxyLog = PR_NewLogModule("nsRequestObserverProxy");
1.185 +#endif
1.186 +
1.187 + mObserver = new nsMainThreadPtrHolder<nsIRequestObserver>(observer);
1.188 + mContext = new nsMainThreadPtrHolder<nsISupports>(context);
1.189 +
1.190 + return NS_OK;
1.191 +}
1.192 +
1.193 +//-----------------------------------------------------------------------------
1.194 +// nsRequestObserverProxy implementation...
1.195 +//-----------------------------------------------------------------------------
1.196 +
1.197 +nsresult
1.198 +nsRequestObserverProxy::FireEvent(nsARequestObserverEvent *event)
1.199 +{
1.200 + nsCOMPtr<nsIEventTarget> mainThread(do_GetMainThread());
1.201 + return mainThread->Dispatch(event, NS_DISPATCH_NORMAL);
1.202 +}
