mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-23 12:51:06 +00:00
139 lines
4.5 KiB
C++
139 lines
4.5 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set sw=2 ts=8 et tw=80 : */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef nsDNSService2_h__
|
|
#define nsDNSService2_h__
|
|
|
|
#include "DNSServiceBase.h"
|
|
#include "nsClassHashtable.h"
|
|
#include "nsPIDNSService.h"
|
|
#include "nsIMemoryReporter.h"
|
|
#include "nsIObserver.h"
|
|
#include "nsHostResolver.h"
|
|
#include "nsString.h"
|
|
#include "nsTHashSet.h"
|
|
#include "nsHashKeys.h"
|
|
#include "mozilla/Atomics.h"
|
|
#include "mozilla/Mutex.h"
|
|
#include "mozilla/Attributes.h"
|
|
#include "TRRService.h"
|
|
|
|
class nsAuthSSPI;
|
|
|
|
class DNSServiceWrapper final : public nsPIDNSService {
|
|
public:
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
NS_FORWARD_NSPIDNSSERVICE(PIDNSService()->)
|
|
NS_FORWARD_NSIDNSSERVICE(DNSService()->)
|
|
|
|
DNSServiceWrapper() = default;
|
|
|
|
static already_AddRefed<nsIDNSService> GetSingleton();
|
|
static void SwitchToBackupDNSService();
|
|
|
|
private:
|
|
~DNSServiceWrapper() = default;
|
|
nsIDNSService* DNSService();
|
|
nsPIDNSService* PIDNSService();
|
|
|
|
mozilla::Mutex mLock{"DNSServiceWrapper.mLock"};
|
|
nsCOMPtr<nsIDNSService> mDNSServiceInUse;
|
|
nsCOMPtr<nsIDNSService> mBackupDNSService;
|
|
};
|
|
|
|
class nsDNSService final : public mozilla::net::DNSServiceBase,
|
|
public nsPIDNSService,
|
|
public nsIMemoryReporter {
|
|
public:
|
|
NS_DECL_ISUPPORTS_INHERITED
|
|
NS_DECL_NSPIDNSSERVICE
|
|
NS_DECL_NSIDNSSERVICE
|
|
NS_DECL_NSIOBSERVER
|
|
NS_DECL_NSIMEMORYREPORTER
|
|
|
|
nsDNSService() = default;
|
|
|
|
static already_AddRefed<nsIDNSService> GetXPCOMSingleton();
|
|
|
|
size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const;
|
|
|
|
bool GetOffline() const;
|
|
|
|
protected:
|
|
friend class nsAuthSSPI;
|
|
friend class DNSServiceWrapper;
|
|
|
|
nsresult DeprecatedSyncResolve(
|
|
const nsACString& aHostname, nsIDNSService::DNSFlags flags,
|
|
const mozilla::OriginAttributes& aOriginAttributes,
|
|
nsIDNSRecord** result);
|
|
|
|
private:
|
|
~nsDNSService() = default;
|
|
|
|
void ReadPrefs(const char* name) override;
|
|
static already_AddRefed<nsDNSService> GetSingleton();
|
|
|
|
uint16_t GetAFForLookup(const nsACString& host,
|
|
nsIDNSService::DNSFlags flags);
|
|
|
|
nsresult PreprocessHostname(bool aLocalDomain, const nsACString& aInput,
|
|
nsACString& aACE);
|
|
|
|
bool IsLocalDomain(const nsACString& aHostname) const;
|
|
|
|
nsresult AsyncResolveInternal(
|
|
const nsACString& aHostname, uint16_t type, nsIDNSService::DNSFlags flags,
|
|
nsIDNSAdditionalInfo* aInfo, nsIDNSListener* aListener,
|
|
nsIEventTarget* target_,
|
|
const mozilla::OriginAttributes& aOriginAttributes,
|
|
nsICancelable** result);
|
|
|
|
nsresult CancelAsyncResolveInternal(
|
|
const nsACString& aHostname, uint16_t aType,
|
|
nsIDNSService::DNSFlags aFlags, nsIDNSAdditionalInfo* aInfo,
|
|
nsIDNSListener* aListener, nsresult aReason,
|
|
const mozilla::OriginAttributes& aOriginAttributes);
|
|
|
|
nsresult ResolveInternal(const nsACString& aHostname,
|
|
nsIDNSService::DNSFlags flags,
|
|
const mozilla::OriginAttributes& aOriginAttributes,
|
|
nsIDNSRecord** result);
|
|
|
|
// Locks the mutex and returns an addreffed resolver. May return null.
|
|
already_AddRefed<nsHostResolver> GetResolverLocked();
|
|
|
|
RefPtr<nsHostResolver> mResolver;
|
|
|
|
// mLock protects access to mResolver, mLocalDomains, mIPv4OnlyDomains,
|
|
// mFailedSVCDomainNames, and mMockHTTPSRRDomain.
|
|
mozilla::Mutex mLock MOZ_UNANNOTATED{"nsDNSServer.mLock"};
|
|
|
|
// mIPv4OnlyDomains is a comma-separated list of domains for which only
|
|
// IPv4 DNS lookups are performed. This allows the user to disable IPv6 on
|
|
// a per-domain basis and work around broken DNS servers. See bug 68796.
|
|
nsCString mIPv4OnlyDomains;
|
|
nsCString mForceResolve;
|
|
nsCString mMockHTTPSRRDomain;
|
|
mozilla::Atomic<bool, mozilla::Relaxed> mHasMockHTTPSRRDomainSet{false};
|
|
bool mBlockDotOnion = false;
|
|
bool mNotifyResolution = false;
|
|
bool mOfflineLocalhost = false;
|
|
bool mForceResolveOn = false;
|
|
nsTHashSet<nsCString> mLocalDomains;
|
|
RefPtr<mozilla::net::TRRService> mTrrService;
|
|
|
|
uint32_t mResCacheEntries = 0;
|
|
uint32_t mResCacheExpiration = 0;
|
|
uint32_t mResCacheGrace = 0;
|
|
bool mResolverPrefsUpdated = false;
|
|
nsClassHashtable<nsCStringHashKey, nsTArray<nsCString>> mFailedSVCDomainNames;
|
|
};
|
|
|
|
already_AddRefed<nsIDNSService> GetOrInitDNSService();
|
|
|
|
#endif // nsDNSService2_h__
|