summaryrefslogtreecommitdiffstats
path: root/toolkit/components/url-classifier/nsUrlClassifierProxies.h
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
commit26a029d407be480d791972afb5975cf62c9360a6 (patch)
treef435a8308119effd964b339f76abb83a57c29483 /toolkit/components/url-classifier/nsUrlClassifierProxies.h
parentInitial commit. (diff)
downloadfirefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz
firefox-26a029d407be480d791972afb5975cf62c9360a6.zip
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'toolkit/components/url-classifier/nsUrlClassifierProxies.h')
-rw-r--r--toolkit/components/url-classifier/nsUrlClassifierProxies.h359
1 files changed, 359 insertions, 0 deletions
diff --git a/toolkit/components/url-classifier/nsUrlClassifierProxies.h b/toolkit/components/url-classifier/nsUrlClassifierProxies.h
new file mode 100644
index 0000000000..5739d7bf21
--- /dev/null
+++ b/toolkit/components/url-classifier/nsUrlClassifierProxies.h
@@ -0,0 +1,359 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* 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 nsUrlClassifierProxies_h
+#define nsUrlClassifierProxies_h
+
+#include "nsIUrlClassifierDBService.h"
+#include "nsUrlClassifierDBService.h"
+#include "nsProxyRelease.h"
+#include "nsThreadUtils.h"
+#include "mozilla/Attributes.h"
+#include "nsIPrincipal.h"
+#include "LookupCache.h"
+
+/**
+ * Thread proxy from the main thread to the worker thread.
+ */
+class UrlClassifierDBServiceWorkerProxy final
+ : public nsIUrlClassifierDBService {
+ public:
+ explicit UrlClassifierDBServiceWorkerProxy(
+ nsUrlClassifierDBServiceWorker* aTarget)
+ : mTarget(aTarget) {}
+
+ NS_DECL_THREADSAFE_ISUPPORTS
+ NS_DECL_NSIURLCLASSIFIERDBSERVICE
+
+ class LookupRunnable : public mozilla::Runnable {
+ public:
+ LookupRunnable(nsUrlClassifierDBServiceWorker* aTarget,
+ nsIPrincipal* aPrincipal, const nsACString& aTables,
+ nsIUrlClassifierCallback* aCB)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::LookupRunnable"),
+ mTarget(aTarget),
+ mPrincipal(aPrincipal),
+ mLookupTables(aTables),
+ mCB(aCB) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const nsCOMPtr<nsIPrincipal> mPrincipal;
+ const nsCString mLookupTables;
+ const nsCOMPtr<nsIUrlClassifierCallback> mCB;
+ };
+
+ class GetTablesRunnable : public mozilla::Runnable {
+ public:
+ GetTablesRunnable(nsUrlClassifierDBServiceWorker* aTarget,
+ nsIUrlClassifierCallback* aCB)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::GetTablesRunnable"),
+ mTarget(aTarget),
+ mCB(aCB) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const nsCOMPtr<nsIUrlClassifierCallback> mCB;
+ };
+
+ class BeginUpdateRunnable : public mozilla::Runnable {
+ public:
+ BeginUpdateRunnable(nsUrlClassifierDBServiceWorker* aTarget,
+ nsIUrlClassifierUpdateObserver* aUpdater,
+ const nsACString& aTables)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::BeginUpdateRunnable"),
+ mTarget(aTarget),
+ mUpdater(aUpdater),
+ mTables(aTables) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const nsCOMPtr<nsIUrlClassifierUpdateObserver> mUpdater;
+ const nsCString mTables;
+ };
+
+ class BeginStreamRunnable : public mozilla::Runnable {
+ public:
+ BeginStreamRunnable(nsUrlClassifierDBServiceWorker* aTarget,
+ const nsACString& aTable)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::BeginStreamRunnable"),
+ mTarget(aTarget),
+ mTable(aTable) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const nsCString mTable;
+ };
+
+ class UpdateStreamRunnable : public mozilla::Runnable {
+ public:
+ UpdateStreamRunnable(nsUrlClassifierDBServiceWorker* aTarget,
+ const nsACString& aUpdateChunk)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::UpdateStreamRunnable"),
+ mTarget(aTarget),
+ mUpdateChunk(aUpdateChunk) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const nsCString mUpdateChunk;
+ };
+
+ class CacheCompletionsRunnable : public mozilla::Runnable {
+ public:
+ CacheCompletionsRunnable(
+ nsUrlClassifierDBServiceWorker* aTarget,
+ const mozilla::safebrowsing::ConstCacheResultArray& aEntries)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::CacheCompletionsRunnable"),
+ mTarget(aTarget),
+ mEntries(aEntries.Clone()) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const mozilla::safebrowsing::ConstCacheResultArray mEntries;
+ };
+
+ class ClearLastResultsRunnable : public mozilla::Runnable {
+ public:
+ explicit ClearLastResultsRunnable(nsUrlClassifierDBServiceWorker* aTarget)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::ClearLastResultsRunnable"),
+ mTarget(aTarget) {}
+
+ NS_DECL_NSIRUNNABLE
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ };
+
+ class GetCacheInfoRunnable : public mozilla::Runnable {
+ public:
+ explicit GetCacheInfoRunnable(nsUrlClassifierDBServiceWorker* aTarget,
+ const nsACString& aTable,
+ nsIUrlClassifierGetCacheCallback* aCallback)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::GetCacheInfoRunnable"),
+ mTarget(aTarget),
+ mTable(aTable),
+ mCache(nullptr),
+ mCallback(new nsMainThreadPtrHolder<nsIUrlClassifierGetCacheCallback>(
+ "nsIUrlClassifierGetCacheCallback", aCallback)) {}
+
+ NS_DECL_NSIRUNNABLE
+ private:
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+ const nsCString mTable;
+ nsCOMPtr<nsIUrlClassifierCacheInfo> mCache;
+ const nsMainThreadPtrHandle<nsIUrlClassifierGetCacheCallback> mCallback;
+ };
+
+ class GetCacheInfoCallbackRunnable : public mozilla::Runnable {
+ public:
+ explicit GetCacheInfoCallbackRunnable(
+ nsIUrlClassifierCacheInfo* aCache,
+ const nsMainThreadPtrHandle<nsIUrlClassifierGetCacheCallback>&
+ aCallback)
+ : mozilla::Runnable(
+ "UrlClassifierDBServiceWorkerProxy::"
+ "GetCacheInfoCallbackRunnable"),
+ mCache(aCache),
+ mCallback(aCallback) {}
+
+ NS_DECL_NSIRUNNABLE
+ private:
+ nsCOMPtr<nsIUrlClassifierCacheInfo> mCache;
+ const nsMainThreadPtrHandle<nsIUrlClassifierGetCacheCallback> mCallback;
+ };
+
+ public:
+ nsresult OpenDb() const;
+ nsresult CloseDb() const;
+ nsresult PreShutdown() const;
+
+ nsresult CacheCompletions(
+ const mozilla::safebrowsing::ConstCacheResultArray& aEntries) const;
+
+ nsresult GetCacheInfo(const nsACString& aTable,
+ nsIUrlClassifierGetCacheCallback* aCallback) const;
+
+ private:
+ ~UrlClassifierDBServiceWorkerProxy() = default;
+
+ const RefPtr<nsUrlClassifierDBServiceWorker> mTarget;
+};
+
+// The remaining classes here are all proxies to the main thread
+
+class UrlClassifierLookupCallbackProxy final
+ : public nsIUrlClassifierLookupCallback {
+ public:
+ explicit UrlClassifierLookupCallbackProxy(
+ nsIUrlClassifierLookupCallback* aTarget)
+ : mTarget(new nsMainThreadPtrHolder<nsIUrlClassifierLookupCallback>(
+ "UrlClassifierLookupCallbackProxy::mTarget", aTarget)) {}
+
+ NS_DECL_THREADSAFE_ISUPPORTS
+ NS_DECL_NSIURLCLASSIFIERLOOKUPCALLBACK
+
+ class LookupCompleteRunnable : public mozilla::Runnable {
+ public:
+ LookupCompleteRunnable(
+ const nsMainThreadPtrHandle<nsIUrlClassifierLookupCallback>& aTarget,
+ mozilla::UniquePtr<mozilla::safebrowsing::LookupResultArray> aResults)
+ : mozilla::Runnable(
+ "UrlClassifierLookupCallbackProxy::LookupCompleteRunnable"),
+ mTarget(aTarget),
+ mResults(std::move(aResults)) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const nsMainThreadPtrHandle<nsIUrlClassifierLookupCallback> mTarget;
+ mozilla::UniquePtr<mozilla::safebrowsing::LookupResultArray> mResults;
+ };
+
+ private:
+ ~UrlClassifierLookupCallbackProxy() = default;
+
+ const nsMainThreadPtrHandle<nsIUrlClassifierLookupCallback> mTarget;
+};
+
+class UrlClassifierCallbackProxy final : public nsIUrlClassifierCallback {
+ public:
+ explicit UrlClassifierCallbackProxy(nsIUrlClassifierCallback* aTarget)
+ : mTarget(new nsMainThreadPtrHolder<nsIUrlClassifierCallback>(
+ "UrlClassifierCallbackProxy::mTarget", aTarget)) {}
+
+ NS_DECL_THREADSAFE_ISUPPORTS
+ NS_DECL_NSIURLCLASSIFIERCALLBACK
+
+ class HandleEventRunnable : public mozilla::Runnable {
+ public:
+ HandleEventRunnable(
+ const nsMainThreadPtrHandle<nsIUrlClassifierCallback>& aTarget,
+ const nsACString& aValue)
+ : mozilla::Runnable("UrlClassifierCallbackProxy::HandleEventRunnable"),
+ mTarget(aTarget),
+ mValue(aValue) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const nsMainThreadPtrHandle<nsIUrlClassifierCallback> mTarget;
+ const nsCString mValue;
+ };
+
+ private:
+ ~UrlClassifierCallbackProxy() = default;
+
+ const nsMainThreadPtrHandle<nsIUrlClassifierCallback> mTarget;
+};
+
+class UrlClassifierUpdateObserverProxy final
+ : public nsIUrlClassifierUpdateObserver {
+ public:
+ explicit UrlClassifierUpdateObserverProxy(
+ nsIUrlClassifierUpdateObserver* aTarget)
+ : mTarget(new nsMainThreadPtrHolder<nsIUrlClassifierUpdateObserver>(
+ "UrlClassifierUpdateObserverProxy::mTarget", aTarget)) {}
+
+ NS_DECL_THREADSAFE_ISUPPORTS
+ NS_DECL_NSIURLCLASSIFIERUPDATEOBSERVER
+
+ class UpdateUrlRequestedRunnable : public mozilla::Runnable {
+ public:
+ UpdateUrlRequestedRunnable(
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
+ const nsACString& aURL, const nsACString& aTable)
+ : mozilla::Runnable(
+ "UrlClassifierUpdateObserverProxy::UpdateUrlRequestedRunnable"),
+ mTarget(aTarget),
+ mURL(aURL),
+ mTable(aTable) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
+ const nsCString mURL;
+ const nsCString mTable;
+ };
+
+ class StreamFinishedRunnable : public mozilla::Runnable {
+ public:
+ StreamFinishedRunnable(
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
+ nsresult aStatus, uint32_t aDelay)
+ : mozilla::Runnable(
+ "UrlClassifierUpdateObserverProxy::StreamFinishedRunnable"),
+ mTarget(aTarget),
+ mStatus(aStatus),
+ mDelay(aDelay) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
+ const nsresult mStatus;
+ const uint32_t mDelay;
+ };
+
+ class UpdateErrorRunnable : public mozilla::Runnable {
+ public:
+ UpdateErrorRunnable(
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
+ nsresult aError)
+ : mozilla::Runnable(
+ "UrlClassifierUpdateObserverProxy::UpdateErrorRunnable"),
+ mTarget(aTarget),
+ mError(aError) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
+ const nsresult mError;
+ };
+
+ class UpdateSuccessRunnable : public mozilla::Runnable {
+ public:
+ UpdateSuccessRunnable(
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver>& aTarget,
+ uint32_t aRequestedTimeout)
+ : mozilla::Runnable(
+ "UrlClassifierUpdateObserverProxy::UpdateSuccessRunnable"),
+ mTarget(aTarget),
+ mRequestedTimeout(aRequestedTimeout) {}
+
+ NS_DECL_NSIRUNNABLE
+
+ private:
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
+ const uint32_t mRequestedTimeout;
+ };
+
+ private:
+ ~UrlClassifierUpdateObserverProxy() = default;
+
+ const nsMainThreadPtrHandle<nsIUrlClassifierUpdateObserver> mTarget;
+};
+
+#endif // nsUrlClassifierProxies_h