diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /dom/system/gonk/NetworkWorker.cpp | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip |
Add m-esr52 at 52.6.0
Diffstat (limited to 'dom/system/gonk/NetworkWorker.cpp')
-rw-r--r-- | dom/system/gonk/NetworkWorker.cpp | 271 |
1 files changed, 271 insertions, 0 deletions
diff --git a/dom/system/gonk/NetworkWorker.cpp b/dom/system/gonk/NetworkWorker.cpp new file mode 100644 index 000000000..caf07f375 --- /dev/null +++ b/dom/system/gonk/NetworkWorker.cpp @@ -0,0 +1,271 @@ +/* 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/. */ + +#include "NetworkWorker.h" +#include "NetworkUtils.h" +#include <nsThreadUtils.h> +#include "mozilla/ModuleUtils.h" +#include "mozilla/ClearOnShutdown.h" +#include "mozilla/dom/ScriptSettings.h" +#include "mozilla/dom/ToJSValue.h" +#include "nsAutoPtr.h" +#include "nsXULAppAPI.h" + +#define NS_NETWORKWORKER_CID \ + { 0x6df093e1, 0x8127, 0x4fa7, {0x90, 0x13, 0xa3, 0xaa, 0xa7, 0x79, 0xbb, 0xdd} } + +using namespace mozilla; +using namespace mozilla::dom; +using namespace mozilla::ipc; + +namespace mozilla { + +nsCOMPtr<nsIThread> gWorkerThread; + +// The singleton network worker, to be used on the main thread. +StaticRefPtr<NetworkWorker> gNetworkWorker; + +// The singleton networkutils class, that can be used on any thread. +static nsAutoPtr<NetworkUtils> gNetworkUtils; + +// Runnable used dispatch command result on the main thread. +class NetworkResultDispatcher : public Runnable +{ +public: + NetworkResultDispatcher(const NetworkResultOptions& aResult) + : mResult(aResult) + { + MOZ_ASSERT(!NS_IsMainThread()); + } + + NS_IMETHOD Run() override + { + MOZ_ASSERT(NS_IsMainThread()); + + if (gNetworkWorker) { + gNetworkWorker->DispatchNetworkResult(mResult); + } + return NS_OK; + } +private: + NetworkResultOptions mResult; +}; + +// Runnable used dispatch netd command on the worker thread. +class NetworkCommandDispatcher : public Runnable +{ +public: + NetworkCommandDispatcher(const NetworkParams& aParams) + : mParams(aParams) + { + MOZ_ASSERT(NS_IsMainThread()); + } + + NS_IMETHOD Run() override + { + MOZ_ASSERT(!NS_IsMainThread()); + + if (gNetworkUtils) { + gNetworkUtils->ExecuteCommand(mParams); + } + return NS_OK; + } +private: + NetworkParams mParams; +}; + +// Runnable used dispatch netd result on the worker thread. +class NetdEventRunnable : public Runnable +{ +public: + NetdEventRunnable(NetdCommand* aCommand) + : mCommand(aCommand) + { + MOZ_ASSERT(!NS_IsMainThread()); + } + + NS_IMETHOD Run() override + { + MOZ_ASSERT(!NS_IsMainThread()); + + if (gNetworkUtils) { + gNetworkUtils->onNetdMessage(mCommand); + } + return NS_OK; + } + +private: + nsAutoPtr<NetdCommand> mCommand; +}; + +class NetdMessageConsumer : public NetdConsumer +{ +public: + NetdMessageConsumer() + { + MOZ_ASSERT(NS_IsMainThread()); + } + + void MessageReceived(NetdCommand* aCommand) + { + MOZ_ASSERT(!NS_IsMainThread()); + + nsCOMPtr<nsIRunnable> runnable = new NetdEventRunnable(aCommand); + if (gWorkerThread) { + gWorkerThread->Dispatch(runnable, nsIEventTarget::DISPATCH_NORMAL); + } + } +}; + +NS_IMPL_ISUPPORTS(NetworkWorker, nsINetworkWorker) + +NetworkWorker::NetworkWorker() +{ + MOZ_ASSERT(NS_IsMainThread()); + MOZ_ASSERT(!gNetworkWorker); +} + +NetworkWorker::~NetworkWorker() +{ + MOZ_ASSERT(!gNetworkWorker); + MOZ_ASSERT(!mListener); +} + +already_AddRefed<NetworkWorker> +NetworkWorker::FactoryCreate() +{ + if (!XRE_IsParentProcess()) { + return nullptr; + } + + MOZ_ASSERT(NS_IsMainThread()); + + if (!gNetworkWorker) { + gNetworkWorker = new NetworkWorker(); + ClearOnShutdown(&gNetworkWorker); + + gNetworkUtils = new NetworkUtils(NetworkWorker::NotifyResult); + ClearOnShutdown(&gNetworkUtils); + } + + RefPtr<NetworkWorker> worker = gNetworkWorker.get(); + return worker.forget(); +} + +NS_IMETHODIMP +NetworkWorker::Start(nsINetworkEventListener* aListener) +{ + MOZ_ASSERT(NS_IsMainThread()); + MOZ_ASSERT(aListener); + + if (mListener) { + return NS_OK; + } + + nsresult rv; + + rv = NS_NewNamedThread("NetworkWorker", getter_AddRefs(gWorkerThread)); + if (NS_FAILED(rv)) { + NS_WARNING("Can't create network control thread"); + return NS_ERROR_FAILURE; + } + + StartNetd(new NetdMessageConsumer()); + mListener = aListener; + + return NS_OK; +} + +NS_IMETHODIMP +NetworkWorker::Shutdown() +{ + MOZ_ASSERT(NS_IsMainThread()); + + if (!mListener) { + return NS_OK; + } + + StopNetd(); + + gWorkerThread->Shutdown(); + gWorkerThread = nullptr; + + mListener = nullptr; + return NS_OK; +} + +// Receive command from main thread (NetworkService.js). +NS_IMETHODIMP +NetworkWorker::PostMessage(JS::Handle<JS::Value> aOptions, JSContext* aCx) +{ + MOZ_ASSERT(NS_IsMainThread()); + + NetworkCommandOptions options; + if (!options.Init(aCx, aOptions)) { + NS_WARNING("Bad dictionary passed to NetworkWorker::SendCommand"); + return NS_ERROR_FAILURE; + } + + // Dispatch the command to the control thread. + NetworkParams NetworkParams(options); + nsCOMPtr<nsIRunnable> runnable = new NetworkCommandDispatcher(NetworkParams); + if (gWorkerThread) { + gWorkerThread->Dispatch(runnable, nsIEventTarget::DISPATCH_NORMAL); + } + return NS_OK; +} + +void +NetworkWorker::DispatchNetworkResult(const NetworkResultOptions& aOptions) +{ + MOZ_ASSERT(NS_IsMainThread()); + + mozilla::AutoSafeJSContext cx; + JS::RootedValue val(cx); + + if (!ToJSValue(cx, aOptions, &val)) { + return; + } + + // Call the listener with a JS value. + if (mListener) { + mListener->OnEvent(val); + } +} + +// Callback function from network worker thread to update result on main thread. +void +NetworkWorker::NotifyResult(NetworkResultOptions& aResult) +{ + MOZ_ASSERT(!NS_IsMainThread()); + + nsCOMPtr<nsIRunnable> runnable = new NetworkResultDispatcher(aResult); + NS_DispatchToMainThread(runnable); +} + +NS_GENERIC_FACTORY_SINGLETON_CONSTRUCTOR(NetworkWorker, + NetworkWorker::FactoryCreate) + +NS_DEFINE_NAMED_CID(NS_NETWORKWORKER_CID); + +static const mozilla::Module::CIDEntry kNetworkWorkerCIDs[] = { + { &kNS_NETWORKWORKER_CID, false, nullptr, NetworkWorkerConstructor }, + { nullptr } +}; + +static const mozilla::Module::ContractIDEntry kNetworkWorkerContracts[] = { + { "@mozilla.org/network/worker;1", &kNS_NETWORKWORKER_CID }, + { nullptr } +}; + +static const mozilla::Module kNetworkWorkerModule = { + mozilla::Module::kVersion, + kNetworkWorkerCIDs, + kNetworkWorkerContracts, + nullptr +}; + +} // namespace mozilla + +NSMODULE_DEFN(NetworkWorkerModule) = &kNetworkWorkerModule; |