netwerk/socket/nsSocketProviderService.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "nsString.h"
   7 #include "nsISocketProvider.h"
   8 #include "nsError.h"
   9 #include "nsNSSComponent.h"
  10 #include "nsSOCKSSocketProvider.h"
  11 #include "nsSocketProviderService.h"
  12 #include "nsSSLSocketProvider.h"
  13 #include "nsTLSSocketProvider.h"
  14 #include "nsUDPSocketProvider.h"
  15 #include "mozilla/ClearOnShutdown.h"
  16 #include "nsThreadUtils.h"
  17 #include "nsCRT.h"
  18
  19 mozilla::StaticRefPtr<nsSocketProviderService>
  20     nsSocketProviderService::gSingleton;
  21
  22 ////////////////////////////////////////////////////////////////////////////////
  23
  24 already_AddRefed<nsISocketProviderService>
  25 nsSocketProviderService::GetOrCreate() {
  26   RefPtr<nsSocketProviderService> inst;
  27   if (gSingleton) {
  28     inst = gSingleton;
  29   } else {
  30     inst = new nsSocketProviderService();
  31     gSingleton = inst;
  32     if (NS_IsMainThread()) {
  33       mozilla::ClearOnShutdown(&gSingleton);
  34     } else {
  35       NS_DispatchToMainThread(NS_NewRunnableFunction(
  36           "net::nsSocketProviderService::GetOrCreate",
  37           []() -> void { mozilla::ClearOnShutdown(&gSingleton); }));
  38     }
  39   }
  40   return inst.forget();
  41 }
  42
  43 NS_IMPL_ISUPPORTS(nsSocketProviderService, nsISocketProviderService)
  44
  45 ////////////////////////////////////////////////////////////////////////////////
  46
  47 NS_IMETHODIMP
  48 nsSocketProviderService::GetSocketProvider(const char* type,
  49                                            nsISocketProvider** result) {
  50   nsCOMPtr<nsISocketProvider> inst;
  51   if (!nsCRT::strcmp(type, "ssl") &&
  52       (XRE_IsParentProcess() || XRE_IsSocketProcess()) &&
  53       EnsureNSSInitializedChromeOrContent()) {
  54     inst = new nsSSLSocketProvider();
  55   } else if (!nsCRT::strcmp(type, "starttls") &&
  56              (XRE_IsParentProcess() || XRE_IsSocketProcess()) &&
  57              EnsureNSSInitializedChromeOrContent()) {
  58     inst = new nsTLSSocketProvider();
  59   } else if (!nsCRT::strcmp(type, "socks")) {
  60     inst = new nsSOCKSSocketProvider(NS_SOCKS_VERSION_5);
  61   } else if (!nsCRT::strcmp(type, "socks4")) {
  62     inst = new nsSOCKSSocketProvider(NS_SOCKS_VERSION_4);
  63   } else if (!nsCRT::strcmp(type, "udp")) {
  64     inst = new nsUDPSocketProvider();
  65   } else {
  66     return NS_ERROR_UNKNOWN_SOCKET_TYPE;
  67   }
  68   inst.forget(result);
  69   return NS_OK;
  70 }
  71
  72 ////////////////////////////////////////////////////////////////////////////////