netwerk/dns/nsDNSService2.h
author Dragana Damjanovic <dd.mozilla@gmail.com>
Sat, 22 Sep 2018 23:54:11 +0300
changeset 437811 52ee9ddf50602cea9c21dd3855c376f62232a289
parent 437574 8d8dc3f35c3d65b8268c014cf10709c37049bce1
child 448947 6f3709b3878117466168c40affa7bca0b60cf75b
permissions -rw-r--r--
Bug 1473736 - Implement necko part of ESNI r=mak,kmag,mcmanus

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* vim: set sw=4 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 "nsPIDNSService.h"
#include "nsIIDNService.h"
#include "nsIMemoryReporter.h"
#include "nsIObserver.h"
#include "nsHostResolver.h"
#include "nsAutoPtr.h"
#include "nsString.h"
#include "nsTHashtable.h"
#include "nsHashKeys.h"
#include "mozilla/Mutex.h"
#include "mozilla/Attributes.h"
#include "TRRService.h"

class nsAuthSSPI;

class nsDNSService final : public nsPIDNSService
                         , public nsIObserver
                         , public nsIMemoryReporter
{
public:
    NS_DECL_THREADSAFE_ISUPPORTS
    NS_DECL_NSPIDNSSERVICE
    NS_DECL_NSIDNSSERVICE
    NS_DECL_NSIOBSERVER
    NS_DECL_NSIMEMORYREPORTER

    nsDNSService();

    static already_AddRefed<nsIDNSService> GetXPCOMSingleton();

    size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const;

    bool GetOffline() const;

protected:
    friend class nsAuthSSPI;

    nsresult DeprecatedSyncResolve(const nsACString &aHostname,
                                   uint32_t flags,
                                   const mozilla::OriginAttributes &aOriginAttributes,
                                   nsIDNSRecord **result);
private:
    ~nsDNSService();

    nsresult ReadPrefs(const char *name);
    static already_AddRefed<nsDNSService> GetSingleton();

    uint16_t GetAFForLookup(const nsACString &host, uint32_t flags);

    nsresult PreprocessHostname(bool              aLocalDomain,
                                const nsACString &aInput,
                                nsIIDNService    *aIDN,
                                nsACString       &aACE);

    nsresult AsyncResolveInternal(const nsACString        &aHostname,
                                  uint16_t                 type,
                                  uint32_t                 flags,
                                  nsIDNSListener          *aListener,
                                  nsIEventTarget          *target_,
                                  const mozilla::OriginAttributes  &aOriginAttributes,
                                  nsICancelable          **result);

    nsresult CancelAsyncResolveInternal(const nsACString       &aHostname,
                                        uint16_t                aType,
                                        uint32_t                aFlags,
                                        nsIDNSListener         *aListener,
                                        nsresult                aReason,
                                        const mozilla::OriginAttributes &aOriginAttributes);

    nsresult ResolveInternal(const nsACString &aHostname,
                             uint32_t flags,
                             const mozilla::OriginAttributes &aOriginAttributes,
                             nsIDNSRecord **result);

    RefPtr<nsHostResolver>  mResolver;
    nsCOMPtr<nsIIDNService>   mIDN;

    // mLock protects access to mResolver, mLocalDomains and mIPv4OnlyDomains
    mozilla::Mutex            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;
    bool                                      mDisableIPv6;
    bool                                      mDisablePrefetch;
    bool                                      mBlockDotOnion;
    bool                                      mNotifyResolution;
    bool                                      mOfflineLocalhost;
    bool                                      mForceResolveOn;
    uint32_t                                  mProxyType;
    nsTHashtable<nsCStringHashKey>            mLocalDomains;
    RefPtr<mozilla::net::TRRService>          mTrrService;

    uint32_t                                  mResCacheEntries;
    uint32_t                                  mResCacheExpiration;
    uint32_t                                  mResCacheGrace;
    bool                                      mResolverPrefsUpdated;
};

#endif //nsDNSService2_h__