netwerk/dns/DNSServiceBase.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 "DNSServiceBase.h"
   8
   9 #include "DNS.h"
  10 #include "mozilla/Preferences.h"
  11 #include "mozilla/StaticPrefs_network.h"
  12 #include "nsIDNSService.h"
  13 #include "nsIProtocolProxyService2.h"
  14 #include "nsIPrefBranch.h"
  15
  16 namespace mozilla::net {
  17
  18 static const char kPrefProxyType[] = "network.proxy.type";
  19 static const char kPrefDisablePrefetch[] = "network.dns.disablePrefetch";
  20 static const char kPrefNetworkProxySOCKS[] = "network.proxy.socks";
  21
  22 NS_IMPL_ISUPPORTS(DNSServiceBase, nsIObserver)
  23
  24 void DNSServiceBase::AddPrefObserver(nsIPrefBranch* aPrefs) {
  25   aPrefs->AddObserver(kPrefProxyType, this, false);
  26   aPrefs->AddObserver(kPrefDisablePrefetch, this, false);
  27   // Monitor these to see if there is a change in proxy configuration
  28   aPrefs->AddObserver(kPrefNetworkProxySOCKS, this, false);
  29 }
  30
  31 void DNSServiceBase::ReadPrefs(const char* aName) {
  32   if (!aName || !strcmp(aName, kPrefNetworkProxySOCKS)) {
  33     nsAutoCString socks;
  34     if (NS_SUCCEEDED(Preferences::GetCString(kPrefNetworkProxySOCKS, socks))) {
  35       mHasSocksProxy = !socks.IsEmpty();
  36     }
  37   }
  38   if (!aName || !strcmp(aName, kPrefDisablePrefetch) ||
  39       !strcmp(aName, kPrefProxyType)) {
  40     mDisablePrefetch = Preferences::GetBool(kPrefDisablePrefetch, false) ||
  41                        (StaticPrefs::network_proxy_type() ==
  42                         nsIProtocolProxyService::PROXYCONFIG_MANUAL);
  43   }
  44 }
  45
  46 bool DNSServiceBase::DNSForbiddenByActiveProxy(const nsACString& aHostname,
  47                                                uint32_t aFlags) {
  48   if (aFlags & nsIDNSService::RESOLVE_IGNORE_SOCKS_DNS) {
  49     return false;
  50   }
  51
  52   // We should avoid doing DNS when a proxy is in use.
  53   if (StaticPrefs::network_proxy_type() ==
  54           nsIProtocolProxyService::PROXYCONFIG_MANUAL &&
  55       mHasSocksProxy && StaticPrefs::network_proxy_socks_remote_dns()) {
  56     // Allow IP lookups through, but nothing else.
  57     if (!HostIsIPLiteral(aHostname)) {
  58       return true;
  59     }
  60   }
  61   return false;
  62 }
  63
  64 }  // namespace mozilla::net