docshell/base/ChildProcessChannelListener.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/dom/ChildProcessChannelListener.h"
   8
   9 #include "mozilla/ipc/Endpoint.h"
  10 #include "nsDocShellLoadState.h"
  11
  12 namespace mozilla {
  13 namespace dom {
  14
  15 static StaticRefPtr<ChildProcessChannelListener> sCPCLSingleton;
  16
  17 void ChildProcessChannelListener::RegisterCallback(uint64_t aIdentifier,
  18                                                    Callback&& aCallback) {
  19   if (auto args = mChannelArgs.Extract(aIdentifier)) {
  20     nsresult rv =
  21         aCallback(args->mLoadState, std::move(args->mStreamFilterEndpoints),
  22                   args->mTiming);
  23     args->mResolver(rv);
  24   } else {
  25     mCallbacks.InsertOrUpdate(aIdentifier, std::move(aCallback));
  26   }
  27 }
  28
  29 void ChildProcessChannelListener::OnChannelReady(
  30     nsDocShellLoadState* aLoadState, uint64_t aIdentifier,
  31     nsTArray<Endpoint>&& aStreamFilterEndpoints, nsDOMNavigationTiming* aTiming,
  32     Resolver&& aResolver) {
  33   if (auto callback = mCallbacks.Extract(aIdentifier)) {
  34     nsresult rv =
  35         (*callback)(aLoadState, std::move(aStreamFilterEndpoints), aTiming);
  36     aResolver(rv);
  37   } else {
  38     mChannelArgs.InsertOrUpdate(
  39         aIdentifier, CallbackArgs{aLoadState, std::move(aStreamFilterEndpoints),
  40                                   aTiming, std::move(aResolver)});
  41   }
  42 }
  43
  44 ChildProcessChannelListener::~ChildProcessChannelListener() {
  45   for (const auto& args : mChannelArgs.Values()) {
  46     args.mResolver(NS_ERROR_FAILURE);
  47   }
  48 }
  49
  50 already_AddRefed<ChildProcessChannelListener>
  51 ChildProcessChannelListener::GetSingleton() {
  52   if (!sCPCLSingleton) {
  53     sCPCLSingleton = new ChildProcessChannelListener();
  54     ClearOnShutdown(&sCPCLSingleton);
  55   }
  56   RefPtr<ChildProcessChannelListener> cpcl = sCPCLSingleton;
  57   return cpcl.forget();
  58 }
  59
  60 }  // namespace dom
  61 }  // namespace mozilla