1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/toolkit/components/url-classifier/nsUrlClassifierProxies.h Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,303 @@
1.4 +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
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 +#ifndef nsUrlClassifierProxies_h
1.10 +#define nsUrlClassifierProxies_h
1.11 +
1.12 +#include "nsIUrlClassifierDBService.h"
1.13 +#include "nsProxyRelease.h"
1.14 +#include "nsThreadUtils.h"
1.15 +#include "mozilla/Attributes.h"
1.16 +#include "nsIPrincipal.h"
1.17 +#include "LookupCache.h"
1.18 +
1.19 +
1.20 +/**
1.21 + * Thread proxy from the main thread to the worker thread.
1.22 + */
1.23 +class UrlClassifierDBServiceWorkerProxy MOZ_FINAL :
1.24 + public nsIUrlClassifierDBServiceWorker
1.25 +{
1.26 +public:
1.27 + UrlClassifierDBServiceWorkerProxy(nsIUrlClassifierDBServiceWorker* aTarget)
1.28 + : mTarget(aTarget)
1.29 + { }
1.30 +
1.31 + NS_DECL_THREADSAFE_ISUPPORTS
1.32 + NS_DECL_NSIURLCLASSIFIERDBSERVICE
1.33 + NS_DECL_NSIURLCLASSIFIERDBSERVICEWORKER
1.34 +
1.35 + class LookupRunnable : public nsRunnable
1.36 + {
1.37 + public:
1.38 + LookupRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.39 + nsIPrincipal* aPrincipal,
1.40 + const nsACString& aTables,
1.41 + nsIUrlClassifierCallback* aCB)
1.42 + : mTarget(aTarget)
1.43 + , mPrincipal(aPrincipal)
1.44 + , mLookupTables(aTables)
1.45 + , mCB(aCB)
1.46 + { }
1.47 +
1.48 + NS_DECL_NSIRUNNABLE
1.49 +
1.50 + private:
1.51 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.52 + nsCOMPtr<nsIPrincipal> mPrincipal;
1.53 + nsCString mLookupTables;
1.54 + nsCOMPtr<nsIUrlClassifierCallback> mCB;
1.55 + };
1.56 +
1.57 + class GetTablesRunnable : public nsRunnable
1.58 + {
1.59 + public:
1.60 + GetTablesRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.61 + nsIUrlClassifierCallback* aCB)
1.62 + : mTarget(aTarget)
1.63 + , mCB(aCB)
1.64 + { }
1.65 +
1.66 + NS_DECL_NSIRUNNABLE
1.67 +
1.68 + private:
1.69 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.70 + nsCOMPtr<nsIUrlClassifierCallback> mCB;
1.71 + };
1.72 +
1.73 + class BeginUpdateRunnable : public nsRunnable
1.74 + {
1.75 + public:
1.76 + BeginUpdateRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.77 + nsIUrlClassifierUpdateObserver* aUpdater,
1.78 + const nsACString& aTables)
1.79 + : mTarget(aTarget)
1.80 + , mUpdater(aUpdater)
1.81 + , mTables(aTables)
1.82 + { }
1.83 +
1.84 + NS_DECL_NSIRUNNABLE
1.85 +
1.86 + private:
1.87 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.88 + nsCOMPtr<nsIUrlClassifierUpdateObserver> mUpdater;
1.89 + nsCString mTables;
1.90 + };
1.91 +
1.92 + class BeginStreamRunnable : public nsRunnable
1.93 + {
1.94 + public:
1.95 + BeginStreamRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.96 + const nsACString& aTable)
1.97 + : mTarget(aTarget)
1.98 + , mTable(aTable)
1.99 + { }
1.100 +
1.101 + NS_DECL_NSIRUNNABLE
1.102 +
1.103 + private:
1.104 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.105 + nsCString mTable;
1.106 + };
1.107 +
1.108 + class UpdateStreamRunnable : public nsRunnable
1.109 + {
1.110 + public:
1.111 + UpdateStreamRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.112 + const nsACString& aUpdateChunk)
1.113 + : mTarget(aTarget)
1.114 + , mUpdateChunk(aUpdateChunk)
1.115 + { }
1.116 +
1.117 + NS_DECL_NSIRUNNABLE
1.118 +
1.119 + private:
1.120 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.121 + nsCString mUpdateChunk;
1.122 + };
1.123 +
1.124 + class CacheCompletionsRunnable : public nsRunnable
1.125 + {
1.126 + public:
1.127 + CacheCompletionsRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.128 + mozilla::safebrowsing::CacheResultArray *aEntries)
1.129 + : mTarget(aTarget)
1.130 + , mEntries(aEntries)
1.131 + { }
1.132 +
1.133 + NS_DECL_NSIRUNNABLE
1.134 +
1.135 + private:
1.136 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.137 + mozilla::safebrowsing::CacheResultArray *mEntries;
1.138 + };
1.139 +
1.140 + class CacheMissesRunnable : public nsRunnable
1.141 + {
1.142 + public:
1.143 + CacheMissesRunnable(nsIUrlClassifierDBServiceWorker* aTarget,
1.144 + mozilla::safebrowsing::PrefixArray *aEntries)
1.145 + : mTarget(aTarget)
1.146 + , mEntries(aEntries)
1.147 + { }
1.148 +
1.149 + NS_DECL_NSIRUNNABLE
1.150 +
1.151 + private:
1.152 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.153 + mozilla::safebrowsing::PrefixArray *mEntries;
1.154 + };
1.155 +
1.156 +private:
1.157 + nsCOMPtr<nsIUrlClassifierDBServiceWorker> mTarget;
1.158 +};
1.159 +
1.160 +// The remaining classes here are all proxies to the main thread
1.161 +
1.162 +class UrlClassifierLookupCallbackProxy MOZ_FINAL :
1.163 + public nsIUrlClassifierLookupCallback
1.164 +{
1.165 +public:
1.166 + UrlClassifierLookupCallbackProxy(nsIUrlClassifierLookupCallback* aTarget)
1.167 + : mTarget(new nsMainThreadPtrHolder<nsIUrlClassifierLookupCallback>(aTarget))
1.168 + { }
1.169 +
1.170 + NS_DECL_THREADSAFE_ISUPPORTS
1.171 + NS_DECL_NSIURLCLASSIFIERLOOKUPCALLBACK
1.172 +
1.173 + class LookupCompleteRunnable : public nsRunnable
1.174 + {
1.175 + public:
1.176 + LookupCompleteRunnable(const nsMainThreadPtrHandle<nsIUrlClassifierLookupCallback>& aTarget,
1.177 + mozilla::safebrowsing::LookupResultArray *aResults)
1.178 + : mTarget(aTarget)
1.179 + , mResults(aResults)
1.180 + { }
1.181 +
1.182 + NS_DECL_NSIRUNNABLE
1.183 +
1.184 + private:
1.185 + nsMainThreadPtrHandle<nsIUrlClassifierLookupCallback> mTarget;
1.186 + mozilla::safebrowsing::LookupResultArray * mResults;
1.187 + };
1.188 +
1.189 +private:
1.190 + nsMainThreadPtrHandle<nsIUrlClassifierLookupCallback> mTarget;
1.191 +};
1.192 +
1.193 +class UrlClassifierCallbackProxy MOZ_FINAL : public nsIUrlClassifierCallback
1.194 +{
1.195 +public:
1.196 + UrlClassifierCallbackProxy(nsIUrlClassifierCallback* aTarget)
1.197 + : mTarget(new nsMainThreadPtrHolder<nsIUrlClassifierCallback>(aTarget))
1.198 + { }
1.199 +
1.200 + NS_DECL_THREADSAFE_ISUPPORTS
1.201 + NS_DECL_NSIURLCLASSIFIERCALLBACK
1.202 +
1.203 + class HandleEventRunnable : public nsRunnable
1.204 + {
1.205 + public:
1.206 + HandleEventRunnable(const nsMainThreadPtrHandle<nsIUrlClassifierCallback>& aTarget,
1.207 + const nsACString& aValue)
1.208 + : mTarget(aTarget)
1.209 + , mValue(aValue)
1.210 + { }
1.211 +
1.212 + NS_DECL_NSIRUNNABLE
1.213 +
1.214 + private:
1.215 + nsMainThreadPtrHandle<nsIUrlClassifierCallback> mTarget;
1.216 + nsCString mValue;
1.217 + };
1.218 +
1.219 +private:
1.220 + nsMainThreadPtrHandle<nsIUrlClassifierCallback> mTarget;
1.221 +};
1.222 +
1.223 +class UrlClassifierUpdateObserverProxy MOZ_FINAL :
1.224 + public nsIUrlClassifierUpdateObserver
1.225 +{
1.226 +public:
1.227 + UrlClassifierUpdateObserverProxy(nsIUrlClassifierUpdateObserver* aTarget)
1.228 + : mTarget(new nsMainThreadPtrHolder<nsIUrlClassifierUpdateObserver>(aTarget))
1.229 + { }
1.230 +
1.231 + NS_DECL_THREADSAFE_ISUPPORTS
1.232 + NS_DECL_NSIURLCLASSIFIERUPDATEOBSERVER
1.233 +
1.234 + class UpdateUrlRequestedRunnable : public nsRunnable
1.235 + {
1.236 + public:
1.237 + UpdateUrlRequestedRunnable(const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
1.238 + const nsACString& aURL,
1.239 + const nsACString& aTable)
1.240 + : mTarget(aTarget)
1.241 + , mURL(aURL)
1.242 + , mTable(aTable)
1.243 + { }
1.244 +
1.245 + NS_DECL_NSIRUNNABLE
1.246 +
1.247 + private:
1.248 + nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
1.249 + nsCString mURL, mTable;
1.250 + };
1.251 +
1.252 + class StreamFinishedRunnable : public nsRunnable
1.253 + {
1.254 + public:
1.255 + StreamFinishedRunnable(const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
1.256 + nsresult aStatus, uint32_t aDelay)
1.257 + : mTarget(aTarget)
1.258 + , mStatus(aStatus)
1.259 + , mDelay(aDelay)
1.260 + { }
1.261 +
1.262 + NS_DECL_NSIRUNNABLE
1.263 +
1.264 + private:
1.265 + nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
1.266 + nsresult mStatus;
1.267 + uint32_t mDelay;
1.268 + };
1.269 +
1.270 + class UpdateErrorRunnable : public nsRunnable
1.271 + {
1.272 + public:
1.273 + UpdateErrorRunnable(const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
1.274 + nsresult aError)
1.275 + : mTarget(aTarget)
1.276 + , mError(aError)
1.277 + { }
1.278 +
1.279 + NS_DECL_NSIRUNNABLE
1.280 +
1.281 + private:
1.282 + nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
1.283 + nsresult mError;
1.284 + };
1.285 +
1.286 + class UpdateSuccessRunnable : public nsRunnable
1.287 + {
1.288 + public:
1.289 + UpdateSuccessRunnable(const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
1.290 + uint32_t aRequestedTimeout)
1.291 + : mTarget(aTarget)
1.292 + , mRequestedTimeout(aRequestedTimeout)
1.293 + { }
1.294 +
1.295 + NS_DECL_NSIRUNNABLE
1.296 +
1.297 + private:
1.298 + nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
1.299 + uint32_t mRequestedTimeout;
1.300 + };
1.301 +
1.302 +private:
1.303 + nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
1.304 +};
1.305 +
1.306 +#endif // nsUrlClassifierProxies_h
