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