netwerk/dns/TRRServiceParent.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set sw=2 ts=8 et 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/net/TRRServiceParent.h"
   8
   9 #include "mozilla/ipc/FileDescriptor.h"
  10 #include "mozilla/net/SocketProcessParent.h"
  11 #include "mozilla/psm/PSMIPCTypes.h"
  12 #include "mozilla/Preferences.h"
  13 #include "mozilla/Unused.h"
  14 #include "nsHttpConnectionInfo.h"
  15 #include "nsICaptivePortalService.h"
  16 #include "nsIParentalControlsService.h"
  17 #include "nsINetworkLinkService.h"
  18 #include "nsIObserverService.h"
  19 #include "nsIOService.h"
  20 #include "nsNetCID.h"
  21 #include "TRRService.h"
  22
  23 #include "DNSLogging.h"
  24
  25 namespace mozilla {
  26 namespace net {
  27
  28 static Atomic<TRRServiceParent*> sTRRServiceParentPtr;
  29
  30 static const char* gTRRUriCallbackPrefs[] = {
  31     "network.trr.uri",  "network.trr.default_provider_uri",
  32     "network.trr.mode", kRolloutURIPref,
  33     kRolloutModePref,   nullptr,
  34 };
  35
  36 NS_IMPL_ISUPPORTS_INHERITED(TRRServiceParent, TRRServiceBase, nsIObserver,
  37                             nsISupportsWeakReference)
  38
  39 TRRServiceParent::~TRRServiceParent() = default;
  40
  41 void TRRServiceParent::Init() {
  42   MOZ_ASSERT(gIOService);
  43
  44   if (!gIOService->SocketProcessReady()) {
  45     RefPtr<TRRServiceParent> self = this;
  46     gIOService->CallOrWaitForSocketProcess([self]() { self->Init(); });
  47     return;
  48   }
  49
  50   RefPtr<SocketProcessParent> socketParent =
  51       SocketProcessParent::GetSingleton();
  52   if (!socketParent) {
  53     return;
  54   }
  55
  56   nsCOMPtr<nsIObserverService> obs =
  57       static_cast<nsIObserverService*>(gIOService);
  58   TRRService::AddObserver(this, obs);
  59
  60   bool captiveIsPassed = TRRService::CheckCaptivePortalIsPassed();
  61   bool parentalControlEnabled =
  62       TRRService::GetParentalControlsEnabledInternal();
  63
  64   nsCOMPtr<nsINetworkLinkService> nls =
  65       do_GetService(NS_NETWORK_LINK_SERVICE_CONTRACTID);
  66   nsTArray<nsCString> suffixList;
  67   if (nls) {
  68     nls->GetDnsSuffixList(suffixList);
  69   }
  70
  71   Preferences::RegisterPrefixCallbacks(TRRServiceParent::PrefsChanged,
  72                                        gTRRUriCallbackPrefs, this);
  73   prefsChanged(nullptr);
  74
  75   if (socketParent->SendPTRRServiceConstructor(
  76           this, captiveIsPassed, parentalControlEnabled, suffixList)) {
  77     sTRRServiceParentPtr = this;
  78   }
  79 }
  80
  81 NS_IMETHODIMP
  82 TRRServiceParent::Observe(nsISupports* aSubject, const char* aTopic,
  83                           const char16_t* aData) {
  84   if (!strcmp(aTopic, NS_DNS_SUFFIX_LIST_UPDATED_TOPIC) ||
  85       !strcmp(aTopic, NS_NETWORK_LINK_TOPIC)) {
  86     nsCOMPtr<nsINetworkLinkService> link = do_QueryInterface(aSubject);
  87     // The network link service notification normally passes itself as the
  88     // subject, but some unit tests will sometimes pass a null subject.
  89     if (link) {
  90       nsTArray<nsCString> suffixList;
  91       link->GetDnsSuffixList(suffixList);
  92       Unused << SendUpdatePlatformDNSInformation(suffixList);
  93     }
  94
  95     if (!strcmp(aTopic, NS_NETWORK_LINK_TOPIC) && mURISetByDetection) {
  96       CheckURIPrefs();
  97     }
  98   }
  99
 100   return NS_OK;
 101 }
 102
 103 mozilla::ipc::IPCResult
 104 TRRServiceParent::RecvNotifyNetworkConnectivityServiceObservers(
 105     const nsCString& aTopic) {
 106   nsCOMPtr<nsIObserverService> obs = mozilla::services::GetObserverService();
 107   if (obs) {
 108     obs->NotifyObservers(nullptr, aTopic.get(), nullptr);
 109   }
 110   return IPC_OK();
 111 }
 112
 113 bool TRRServiceParent::MaybeSetPrivateURI(const nsACString& aURI) {
 114   nsAutoCString newURI(aURI);
 115   ProcessURITemplate(newURI);
 116
 117   if (mPrivateURI.Equals(newURI)) {
 118     return false;
 119   }
 120
 121   mPrivateURI = newURI;
 122   AsyncCreateTRRConnectionInfo(mPrivateURI);
 123
 124   nsCOMPtr<nsIObserverService> obs = mozilla::services::GetObserverService();
 125   if (obs) {
 126     obs->NotifyObservers(nullptr, NS_NETWORK_TRR_URI_CHANGED_TOPIC, nullptr);
 127   }
 128   return true;
 129 }
 130
 131 void TRRServiceParent::SetDetectedTrrURI(const nsACString& aURI) {
 132   if (!mURIPref.IsEmpty()) {
 133     return;
 134   }
 135
 136   mURISetByDetection = MaybeSetPrivateURI(aURI);
 137   gIOService->CallOrWaitForSocketProcess(
 138       [self = RefPtr{this}, uri = nsAutoCString(aURI)]() {
 139         Unused << self->SendSetDetectedTrrURI(uri);
 140       });
 141 }
 142
 143 void TRRServiceParent::GetURI(nsACString& aURI) {
 144   // We don't need a lock here, since mPrivateURI is only touched on main
 145   // thread.
 146   MOZ_ASSERT(NS_IsMainThread());
 147   aURI = mPrivateURI;
 148 }
 149
 150 void TRRServiceParent::ReloadParentalControlsEnabled() {
 151   bool enabled = TRRService::ReloadParentalControlsEnabled();
 152   RefPtr<TRRServiceParent> self = this;
 153   gIOService->CallOrWaitForSocketProcess([self, enabled]() {
 154     Unused << self->SendUpdateParentalControlEnabled(enabled);
 155   });
 156 }
 157
 158 // static
 159 void TRRServiceParent::PrefsChanged(const char* aName, void* aSelf) {
 160   static_cast<TRRServiceParent*>(aSelf)->prefsChanged(aName);
 161 }
 162
 163 void TRRServiceParent::prefsChanged(const char* aName) {
 164   if (!aName || !strcmp(aName, "network.trr.uri") ||
 165       !strcmp(aName, "network.trr.default_provider_uri") ||
 166       !strcmp(aName, kRolloutURIPref) ||
 167       !strcmp(aName, "network.trr.ohttp.uri")) {
 168     OnTRRURIChange();
 169   }
 170
 171   if (!aName || !strcmp(aName, "network.trr.mode") ||
 172       !strcmp(aName, kRolloutModePref)) {
 173     OnTRRModeChange();
 174   }
 175 }
 176
 177 void TRRServiceParent::ActorDestroy(ActorDestroyReason why) {
 178   sTRRServiceParentPtr = nullptr;
 179   Preferences::UnregisterPrefixCallbacks(TRRServiceParent::PrefsChanged,
 180                                          gTRRUriCallbackPrefs, this);
 181 }
 182
 183 NS_IMETHODIMP TRRServiceParent::OnProxyConfigChanged() {
 184   LOG(("TRRServiceParent::OnProxyConfigChanged"));
 185
 186   AsyncCreateTRRConnectionInfo(mPrivateURI);
 187   return NS_OK;
 188 }
 189
 190 void TRRServiceParent::SetDefaultTRRConnectionInfo(
 191     nsHttpConnectionInfo* aConnInfo) {
 192   TRRServiceBase::SetDefaultTRRConnectionInfo(aConnInfo);
 193
 194   if (!CanSend()) {
 195     return;
 196   }
 197
 198   if (!aConnInfo) {
 199     Unused << SendSetDefaultTRRConnectionInfo(Nothing());
 200     return;
 201   }
 202
 203   HttpConnectionInfoCloneArgs infoArgs;
 204   nsHttpConnectionInfo::SerializeHttpConnectionInfo(aConnInfo, infoArgs);
 205   Unused << SendSetDefaultTRRConnectionInfo(Some(infoArgs));
 206 }
 207
 208 mozilla::ipc::IPCResult TRRServiceParent::RecvInitTRRConnectionInfo() {
 209   InitTRRConnectionInfo();
 210   return IPC_OK();
 211 }
 212
 213 mozilla::ipc::IPCResult TRRServiceParent::RecvSetConfirmationState(
 214     uint32_t aNewState) {
 215   mConfirmationState = aNewState;
 216   return IPC_OK();
 217 }
 218
 219 void TRRServiceParent::ReadEtcHostsFile() {
 220   if (!sTRRServiceParentPtr) {
 221     return;
 222   }
 223
 224   DoReadEtcHostsFile([](const nsTArray<nsCString>* aArray) -> bool {
 225     RefPtr<TRRServiceParent> service(sTRRServiceParentPtr);
 226     if (service && aArray) {
 227       nsTArray<nsCString> hosts(aArray->Clone());
 228       NS_DispatchToMainThread(NS_NewRunnableFunction(
 229           "TRRServiceParent::ReadEtcHostsFile",
 230           [service, hosts = std::move(hosts)]() mutable {
 231             if (service->CanSend()) {
 232               Unused << service->SendUpdateEtcHosts(hosts);
 233             }
 234           }));
 235     }
 236     return !!service;
 237   });
 238 }
 239
 240 }  // namespace net
 241 }  // namespace mozilla