Quelle  TRRServiceBase.h

/* 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 TRRServiceBase_h_
#define TRRServiceBase_h_

#include "mozilla/Atomics.h"
#include "mozilla/DataMutex.h"
#include "mozilla/net/rust_helper.h"
#include "nsString.h"
#include "nsIDNSService.h"
#include "nsIProtocolProxyService2.h"
#include "nsTHashMap.h"

class nsICancelable;
class nsIProxyInfo;

namespace mozilla {
namespace net {

class nsHttpConnectionInfo;

static const char kRolloutURIPref[] = "doh-rollout.uri";
static const char kRolloutModePref[] = "doh-rollout.mode";

class TRRServiceBase : public nsIProxyConfigChangedCallback {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS

  TRRServiceBase();
  nsIDNSService::ResolverMode Mode() { return mMode; }
  virtual void GetURI(nsACString& result) = 0;
  already_AddRefed<nsHttpConnectionInfo> TRRConnectionInfo();
  // Called to initialize the connection info. Once the connection info is
  // created first time, mTRRConnectionInfoInited will be set to true.
  // When aForceReinit is true, we always create the conncetion info again.
  virtual void InitTRRConnectionInfo(bool aForceReinit = false);
  bool TRRConnectionInfoInited() const { return mTRRConnectionInfoInited; }

  void SetHttp3FirstForServer(const nsACString& aServer, bool aEnabled);
  bool GetHttp3FirstForServer(const nsACString& aServer);

 protected:
  virtual ~TRRServiceBase();

  virtual bool MaybeSetPrivateURI(const nsACString& aURI) = 0;
  void ProcessURITemplate(nsACString& aURI);
  // Checks the network.trr.uri or the doh-rollout.uri prefs and sets the URI
  // in order of preference:
  // 1. The value of network.trr.uri if it is not the default one, meaning
  //    is was set by an explicit user action
  // 2. The value of doh-rollout.uri if it exists
  //    this is set by the rollout addon
  // 3. The default value of network.trr.uri
  void CheckURIPrefs();

  void OnTRRModeChange();
  void OnTRRURIChange();

  virtual void ReadEtcHostsFile() = 0;
  // Called to create a connection info that will be used by TRRServiceChannel.
  // Note that when this function is called, mDefaultTRRConnectionInfo will be
  // set to null to invalidate the connection info.
  // When the connection info is created, SetDefaultTRRConnectionInfo() is
  // called to set the result to mDefaultTRRConnectionInfo.
  // Note that this method does nothing when mTRRConnectionInfoInited is false.
  // We want to starting updating the connection info after it's create first
  // time.
  void AsyncCreateTRRConnectionInfo(const nsACString& aURI);
  void AsyncCreateTRRConnectionInfoInternal(const nsACString& aURI);
  virtual void SetDefaultTRRConnectionInfo(nsHttpConnectionInfo* aConnInfo);
  void RegisterProxyChangeListener();
  void UnregisterProxyChangeListener();

  already_AddRefed<nsHttpConnectionInfo> CreateConnInfoHelper(
      nsIURI* aURI, nsIProxyInfo* aProxyInfo);

  nsCString mPrivateURI;  // protected by mMutex
  // Pref caches should only be used on the main thread.
  nsCString mURIPref;
  nsCString mRolloutURIPref;
  nsCString mDefaultURIPref;
  nsCString mOHTTPURIPref;

  Atomic<nsIDNSService::ResolverMode, Relaxed> mMode{
      nsIDNSService::MODE_NATIVEONLY};
  Atomic<bool, Relaxed> mURISetByDetection{false};
  Atomic<bool, Relaxed> mTRRConnectionInfoInited{false};
  DataMutex<RefPtr<nsHttpConnectionInfo>> mDefaultTRRConnectionInfo;
  bool mNativeHTTPSQueryEnabled{false};

  Mutex mLock{"TRRService"};
  nsTHashMap<nsCString, bool> mHttp3FirstServers MOZ_GUARDED_BY(mLock);
};

}  // namespace net
}  // namespace mozilla

#endif  // TRRServiceBase_h_