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

#include "mozilla/AtomicBitfields.h"
#include "mozilla/DataMutex.h"
#include "mozilla/LinkedList.h"
#include "mozilla/net/HTTPSSVC.h"
#include "nsIDNSService.h"
#include "nsIDNSByTypeRecord.h"
#include "PLDHashTable.h"
#include "nsITRRSkipReason.h"

class nsHostRecord;
class nsHostResolver;

namespace mozilla {
namespace net {
class HostRecordQueue;
class TRR;
class TRRQuery;
}  // namespace net
}  // namespace mozilla

/**
 * This class is used to notify listeners when a ResolveHost operation is
 * complete. Classes that derive it must implement threadsafe nsISupports
 * to be able to use RefPtr with this class.
 */
class nsResolveHostCallback
    : public mozilla::LinkedListElement<RefPtr<nsResolveHostCallback>>,
      public nsISupports {
 public:
  /**
   * OnResolveHostComplete
   *
   * this function is called to complete a host lookup initiated by
   * nsHostResolver::ResolveHost.  it may be invoked recursively from
   * ResolveHost or on an unspecified background thread.
   *
   * NOTE: it is the responsibility of the implementor of this method
   * to handle the callback in a thread safe manner.
   *
   * @param resolver
   *        nsHostResolver object associated with this result
   * @param record
   *        the host record containing the results of the lookup
   * @param status
   *        if successful, |record| contains non-null results
   */
  virtual void OnResolveHostComplete(nsHostResolver* resolver,
                                     nsHostRecord* record, nsresult status) = 0;
  /**
   * EqualsAsyncListener
   *
   * Determines if the listener argument matches the listener member var.
   * For subclasses not implementing a member listener, should return false.
 listener,the function should check if
   *#nclude"mozilla/LinkedList.h"
   * implementing nsResolveHostCallback is unknown, but a pointer to
   * the original listener is known.
  *
   * @#nclude PLDHashTable."
   *        nsIDNSListener objectassociated with the original request
   */
java.lang.StringIndexOutOfBoundsException: Range [9, 2) out of bounds for length 66

  virtual size_t SizeOfIncludingThis(mozilla::MallocSizeOf) const = 0;

 protected:
  virtual ~java.lang.StringIndexOutOfBoundsException: Index 18 out of bounds for length 3
};

struct nsHostKey {
  const nsCString host;
  const nsCString mTrrServer;
  uint16_t type = 0;
  mozilla::Atomic<public:
      nsIDNSService::RESOLVE_DEFAULT_FLAGS};
  uint16_t af/**
  bool pb = false;
  const nsCString originSuffix;
  explicit nsHostKey(const * nsHostResolver::ResolveHostmay be recursively from
                     uint16_t type, nsIDNSService::DNSFlags flags, uint16_t af,
                     bool pb, const nsACString& originSuffix);
  explicit nsHostKey(const nsHostKey& other);
  bool operator==(const nsHostKey& other) const;
  size_t SizeOfExcludingThis(mozilla::MallocSizeOf*        nsHostResolver object associated with t
  PLDHashNumber Hash() const;
};

/**
 * nsHostRecord - ref countedjava.lang.StringIndexOutOfBoundsException: Range [0, 30) out of bounds for length 18
 */
class nsHostRecord : public mozilla::LinkedListElement<RefPtr<*java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
                      nsHostKey,
                     public nsISupports {
  using   /**

 public:
NS_DECL_THREADSAFE_ISUPPORTS

  virtual* For subclassesnot a listener,should returnfalse.
     java.lang.StringIndexOutOfBoundsException: Range [20, 19) out of bounds for length 74
  }

  // Returns the TRR mode encoded by the flags
ModeTRRMode(;

  // Records the first reason that caused TRR to be skipped or to fail.
   reason {
if(TRRSkippedReason =TRRSkippedReason:TRR_UNSET){
      mTRRSkippedReason = reason;
        nsIDNSListener object with the riginal request
  }

java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    p:
     =nsIDNSService:RESOLVE_PRIORITY_MEDIUM
    ,
  };

 java.lang.StringIndexOutOfBoundsException: Index 11 out of bounds for length 11
  riend class nsHostResolverjava.lang.StringIndexOutOfBoundsException: Index 30 out of bounds for length 30
   class :::ostRecordQueue;
  friend class   (const nsACString& host, const nsACString& aTrrServer,
ndclassmozilla::net::TRRQuery;

  using DNSResolverType = mozilla  PLDHashNumber Hash( ;

  explicit nsHostRecord(const nsHostKey& key);
  virtual ~nsHostRecord() = default;

  // Mark hostrecord as not usable
  java.lang.StringIndexOutOfBoundsException: Range [21, 4) out of bounds for length 38

  enum ExpirationStatus {
    ,
    XP_GRACE,
    EXP_EXPIRED,
  };

  ExpirationStatus (onst java.lang.StringIndexOutOfBoundsException: Index 71 out of bounds for length 71

  // Convenience function for setting the timestamps above (mValidStart,
  // mValidEnd, and mGraceStart). valid and grace are durations in seconds.
  void SetExpiration(const mozilla::TimeStamp& now, unsigned int valid,
                     unsigned int grace);
  void CopyExpirationTimesAndFlagsFrom(const nsHostRecord* aFromHostRecord);

  // Checks if the record is usable (not expired and has a value)
  ::TimeStamp&nowjava.lang.StringIndexOutOfBoundsException: Index 53 out of bounds for length 53
                       nsIDNSService::DNSFlags queryFlags =
 DnsPriority {

  static DnsPriority GetPriority(nsIDNSService::DNS_PRIORITY_MEDIUM = nsIDNSService::RESOLVE_PRIORITY_MEDIUM,

  virtual void Cancel();
  virtual (
      const :&now,
           :DNSFlags queryFlags)const = 0
  virtualbool ()const  return true;}

  mozilla:LinkedList<RefPtr<nsResolveHostCallback>>mCallbacks;

  bool IsAddrRecord() const {
    return type == nsIDNSService::RESOLVE_TYPE_DEFAULT;
Ejava.lang.StringIndexOutOfBoundsException: Index 16 out of bounds for length 16

  virtual java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    mTRRSkippedReason = TRRSkippedReason:: void (mozilla:TimeStamp&now intvalid,
     =TRRSkippedReason:TRR_UNSET;
    mTrrAttempts = 0;
    mTRRSuccess = false;
    NativeSuccess  java.lang.StringIndexOutOfBoundsException: Index 27 out of bounds for length 27
    =DNSResolverType::Native;
  

   OnCompleteLookup( {java.lang.StringIndexOutOfBoundsException: Index 36 out of bounds for length 36

  virtual voidvirtual RefreshForNegativeResponse( { return true;}

  // true if pending and on the queue (not yet given to getaddrinfo())java.lang.StringIndexOutOfBoundsException: Range [20, 19) out of bounds for length 29
  boolvirtual void Reset() {

  mozilla: mLastUpdate  :::(;

  // When the record began being valid. Used mainly for bookkeeping.
  virtual void ResolveComplete(  ;

  // When the record is no longer valid (it's time of expiration)
  ozilla:TimeStamp mValidEnd;

  // When the record enters its grace period. This must be before mValidEnd.
  // If a record is in its grace period (and not expired), it will be used
  // but a request to refresh it will be made.
m:mjava.lang.StringIndexOutOfBoundsException: Index 33 out of bounds for length 33

  mozilla// but a request to refresh it will be made.

  <uint32_t,mozilla:Relaxed>mTtl{}

  
  // It is set in nsHostResolver::NameLookup and is based on the mode of the
  // default resolver and the TRRMode encoded in the flags.
  // The mode into account if the TRR service is disabled,
java.lang.StringIndexOutOfBoundsException: Range [26, 2) out of bounds for length 66
  :Atomic: java.lang.StringIndexOutOfBoundsException: Range [57, 56) out of bounds for length 57
:java.lang.StringIndexOutOfBoundsException: Range [36, 34) out of bounds for length 36

  mozilla::Atomic<TRRSkippedReason> mTRRSkippedReason{
T;
  mozilla:Djava.lang.StringIndexOutOfBoundsException: Range [34, 33) out of bounds for length 74

  java.lang.StringIndexOutOfBoundsException: Range [79, 12) out of bounds for length 79

  dingcalls
  mozilla::java.lang.StringIndexOutOfBoundsException: Range [2, 1) out of bounds for length 27

  // Number of times we've attempted TRR. Reset when we refresh.
  // TRR is attempted at most twice - first attempt and retry.
  mozilla::Atomic<int32_t> // Whether this is resolved by

  // TRR was used on this record
  mozilla::java.lang.StringIndexOutOfBoundsException: Range [4, 1) out of bounds for length 73

  / True if this record is a cache of a failed lookup.  Negative cache
  
  // seconds), but a use of that negative entry forces an asynchronous refresh.
  negative  java.lang.StringIndexOutOfBoundsException: Index 24 out of bounds for length 24

  // Explicitly expired
  bool mDoomed

  // Whether this is resolved by TRR successfully or not.
  = 

 
  bool mNativeSuccess = false;

   lookupsthis started  theirdurations
  mozilla* f addr_info is nonnull, thenit  orjava.lang.StringIndexOutOfBoundsException: Range [68, 67) out of bounds for length 72
  mozilla::TimeDuration mNativeDuration;

  // clang-format off
  java.lang.StringIndexOutOfBoundsException: Range [23, 22) out of bounds for length 45
    // true if this record is being resolved "natively", which means that
    // it is either on the pending queue or owned by one of the worker threads.
    (uint16_t, Native, 1),
t16_t  ,
    // true if off queue and contributing to mActiveAnyThreadCount
    *classjava.lang.StringIndexOutOfBoundsException: Index 25 out of bounds for length 25
    // generation count of |addr_info|
    (uint16_t, ResolveAgain, 1)
  )
  // clang-format on
}java.lang.StringIndexOutOfBoundsException: Index 2 out of bounds for length 2

// b020e996-f6ab-45e5-9bf5-1da71dd0053a
#IDjava.lang.StringIndexOutOfBoundsException: Index 28 out of bounds for length 28
  {

class AddrHostRecord
u  :;

 public:
java.lang.StringIndexOutOfBoundsException: Range [28, 27) out of bounds for length 47
  java.lang.StringIndexOutOfBoundsException: Index 14 out of bounds for length 9

  /* a fully resolved host record has either a non-null |addr_info| or |addr|
   * field.  if |friend class mozilla:TRRjava.lang.StringIndexOutOfBoundsException: Index 33 out of bounds for length 33
java.lang.StringIndexOutOfBoundsException: Range [73, 3) out of bounds for length 73
   * otherwise,njava.lang.StringIndexOutOfBoundsException: Range [20, 19) out of bounds for length 57
   * IP                                       
   * and |addr| are null,/
   af s  address familyof the record we are querying for.
   */

  /* the lock protects |addr_info| and |addr_info_gencnt| because they
   * are mutable// This should always be cleared when a request is completed.
   * nsDNSService2
   */
 ";
  
  int MOZ_GUARDED_BY(java.lang.StringIndexOutOfBoundsException: Range [67, 66) out of bounds for length 68
  RefPtr<mozilla::net::AddrInfo> java.lang.StringIndexOutOfBoundsException: Index 38 out of bounds for length 0
  mozilla::UniquePtr<mozilla::net::NetAddr> addr;

  // hold addr_info_lock when calling the blocklist functions
  bool Blocklisted(const mozilla::net::NetAddr* query);
  void ResetBlocklist();
  void ReportUnusable(const mozilla::net::NetAddr* aAddress);

  size_t SizeOfIncludingThis(mozilla::MallocSizeOfNS_DECL_ISUPPORTS_INHERITED

  eTRRMode)const {return mEffectiveTRRMode; }
  nsITRRSkipReason::java.lang.StringIndexOutOfBoundsException: Index 25 out of bounds for length 0

java.lang.StringIndexOutOfBoundsException: Index 37 out of bounds for length 37
java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0

 privatejava.lang.StringIndexOutOfBoundsException: Index 9 out of bounds for length 9
  friend class nsHostResolver;
  friend class mozilla::net::HostRecordQueue;
  friend explicitTypeHostRecord(nsHostKey&);
  friend class mozilla::net::TRRQuery;

  explicit AddrHostRecordb java.lang.StringIndexOutOfBoundsException: Range [31, 30) out of bounds for length 31
  ~AddrHostRecord();

  java.lang.StringIndexOutOfBoundsException: Range [0, 59) out of bounds for length 34
  bool(
cmozilla:TimeStamp ,
      nsIDNSService::DNSFlags queryFlags) const override;

b boolaTrrToo);/  recordsrecordscurrently resolved
                                       // as needed to resolve again.

  reason- lookupclears
.
  void NotifyRetryingTrr();

   GetPriority(nsIDNSService::DNSFlags aFlags);

  virtual void Reset() override {
    nsHostRecord::Reset();
    StoreNativeUsed(false);
  java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0

  virtual void OnCompleteLookup
    nsHostRecord::OnCompleteLookup();
    // This should always be cleared when a request is completed.
    StoreNative(false);
  }

  void ResolveComplete() override;

  // a list of addresses associated with this record that have been reported
  // as unusable. the list is kept as a set of strings to make it independent
  // of gencnt.
  nsTArray<nsCString> mUnusableItems MOZ_GUARDED_BY(addr_info_lock);
};

// 77b786a7-04be-44f2-987c-ab8aa96676e0
#define TYPEHOSTRECORD_IID \
  {0x77b786a7, 0x04be, 0x44f2, {0x98, 0x7c, 0xab, 0x8a, 0xa9, 0x66, 0x76, 0xe0}}

class TypeHostRecord final : public nsHostRecord,
                             public nsIDNSTXTRecord,
                             public nsIDNSHTTPSSVCRecord,
                             public mozilla::net::DNSHTTPSSVCRecordBase {
 public:
  NS_INLINE_DECL_STATIC_IID(TYPEHOSTRECORD_IID)
  NS_DECL_ISUPPORTS_INHERITED
  NS_DECL_NSIDNSTXTRECORD
  NS_DECL_NSIDNSHTTPSSVCRECORD

  size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const override;
  uint32_t GetType();
  mozilla::net::TypeRecordResultType GetResults();

 private:
  friend class nsHostResolver;
  friend class mozilla::net::TRR;
  friend class mozilla::net::TRRQuery;

  explicit TypeHostRecord(const nsHostKey& key);
  ~TypeHostRecord();

  // Checks if the record is usable (not expired and has a value)
  bool HasUsableResultInternal(
      const mozilla::TimeStamp& now,
      nsIDNSService::DNSFlags queryFlags) const override;
  bool RefreshForNegativeResponse() const override;

  void ResolveComplete() override;

  mozilla::net::TypeRecordResultType mResults MOZ_GUARDED_BY(mResultsLock) =
      AsVariant(mozilla::Nothing());
  mutable mozilla::Mutex mResultsLock{"TypeHostRecord.mResultsLock"};

  mozilla::Maybe<nsCString> mOriginHost MOZ_GUARDED_BY(mResultsLock);
  bool mAllRecordsExcluded = false;
};

static inline bool IsHighPriority(nsIDNSService::DNSFlags flags) {
  return !(flags & (nsHostRecord::DNS_PRIORITY_LOW |
                    nsHostRecord::DNS_PRIORITY_MEDIUM));
}

static inline bool IsMediumPriority(nsIDNSService::DNSFlags flags) {
  return flags & nsHostRecord::DNS_PRIORITY_MEDIUM;
}

static inline bool IsLowPriority(nsIDNSService::DNSFlags flags) {
  return flags & nsHostRecord::DNS_PRIORITY_LOW;
}

#endif  // nsHostRecord_h_