netwerk/dns/nsHostRecord.h

   1 /* vim:set ts=4 sw=2 sts=2 et cin: */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #ifndef nsHostRecord_h__
   7 #define nsHostRecord_h__
   8
   9 #include "mozilla/AtomicBitfields.h"
  10 #include "mozilla/DataMutex.h"
  11 #include "mozilla/LinkedList.h"
  12 #include "mozilla/net/HTTPSSVC.h"
  13 #include "nsIDNSService.h"
  14 #include "nsIDNSByTypeRecord.h"
  15 #include "PLDHashTable.h"
  16 #include "nsITRRSkipReason.h"
  17
  18 class nsHostRecord;
  19 class nsHostResolver;
  20
  21 namespace mozilla {
  22 namespace net {
  23 class HostRecordQueue;
  24 class TRR;
  25 class TRRQuery;
  26 }  // namespace net
  27 }  // namespace mozilla
  28
  29 /**
  30  * This class is used to notify listeners when a ResolveHost operation is
  31  * complete. Classes that derive it must implement threadsafe nsISupports
  32  * to be able to use RefPtr with this class.
  33  */
  34 class nsResolveHostCallback
  35     : public mozilla::LinkedListElement<RefPtr<nsResolveHostCallback>>,
  36       public nsISupports {
  37  public:
  38   /**
  39    * OnResolveHostComplete
  40    *
  41    * this function is called to complete a host lookup initiated by
  42    * nsHostResolver::ResolveHost.  it may be invoked recursively from
  43    * ResolveHost or on an unspecified background thread.
  44    *
  45    * NOTE: it is the responsibility of the implementor of this method
  46    * to handle the callback in a thread safe manner.
  47    *
  48    * @param resolver
  49    *        nsHostResolver object associated with this result
  50    * @param record
  51    *        the host record containing the results of the lookup
  52    * @param status
  53    *        if successful, |record| contains non-null results
  54    */
  55   virtual void OnResolveHostComplete(nsHostResolver* resolver,
  56                                      nsHostRecord* record, nsresult status) = 0;
  57   /**
  58    * EqualsAsyncListener
  59    *
  60    * Determines if the listener argument matches the listener member var.
  61    * For subclasses not implementing a member listener, should return false.
  62    * For subclasses having a member listener, the function should check if
  63    * they are the same.  Used for cases where a pointer to an object
  64    * implementing nsResolveHostCallback is unknown, but a pointer to
  65    * the original listener is known.
  66    *
  67    * @param aListener
  68    *        nsIDNSListener object associated with the original request
  69    */
  70   virtual bool EqualsAsyncListener(nsIDNSListener* aListener) = 0;
  71
  72   virtual size_t SizeOfIncludingThis(mozilla::MallocSizeOf) const = 0;
  73
  74  protected:
  75   virtual ~nsResolveHostCallback() = default;
  76 };
  77
  78 struct nsHostKey {
  79   const nsCString host;
  80   const nsCString mTrrServer;
  81   uint16_t type = 0;
  82   nsIDNSService::DNSFlags flags = nsIDNSService::RESOLVE_DEFAULT_FLAGS;
  83   uint16_t af = 0;
  84   bool pb = false;
  85   const nsCString originSuffix;
  86   explicit nsHostKey(const nsACString& host, const nsACString& aTrrServer,
  87                      uint16_t type, nsIDNSService::DNSFlags flags, uint16_t af,
  88                      bool pb, const nsACString& originSuffix);
  89   bool operator==(const nsHostKey& other) const;
  90   size_t SizeOfExcludingThis(mozilla::MallocSizeOf mallocSizeOf) const;
  91   PLDHashNumber Hash() const;
  92 };
  93
  94 /**
  95  * nsHostRecord - ref counted object type stored in host resolver cache.
  96  */
  97 class nsHostRecord : public mozilla::LinkedListElement<RefPtr<nsHostRecord>>,
  98                      public nsHostKey,
  99                      public nsISupports {
 100   using TRRSkippedReason = mozilla::net::TRRSkippedReason;
 101
 102  public:
 103   NS_DECL_THREADSAFE_ISUPPORTS
 104
 105   virtual size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const {
 106     return 0;
 107   }
 108
 109   // Returns the TRR mode encoded by the flags
 110   nsIRequest::TRRMode TRRMode();
 111
 112   // Records the first reason that caused TRR to be skipped or to fail.
 113   void RecordReason(TRRSkippedReason reason) {
 114     if (mTRRSkippedReason == TRRSkippedReason::TRR_UNSET) {
 115       mTRRSkippedReason = reason;
 116     }
 117   }
 118
 119   enum DnsPriority {
 120     DNS_PRIORITY_LOW = nsIDNSService::RESOLVE_PRIORITY_LOW,
 121     DNS_PRIORITY_MEDIUM = nsIDNSService::RESOLVE_PRIORITY_MEDIUM,
 122     DNS_PRIORITY_HIGH,
 123   };
 124
 125  protected:
 126   friend class nsHostResolver;
 127   friend class mozilla::net::HostRecordQueue;
 128   friend class mozilla::net::TRR;
 129   friend class mozilla::net::TRRQuery;
 130
 131   explicit nsHostRecord(const nsHostKey& key);
 132   virtual ~nsHostRecord() = default;
 133
 134   // Mark hostrecord as not usable
 135   void Invalidate();
 136
 137   enum ExpirationStatus {
 138     EXP_VALID,
 139     EXP_GRACE,
 140     EXP_EXPIRED,
 141   };
 142
 143   ExpirationStatus CheckExpiration(const mozilla::TimeStamp& now) const;
 144
 145   // Convenience function for setting the timestamps above (mValidStart,
 146   // mValidEnd, and mGraceStart). valid and grace are durations in seconds.
 147   void SetExpiration(const mozilla::TimeStamp& now, unsigned int valid,
 148                      unsigned int grace);
 149   void CopyExpirationTimesAndFlagsFrom(const nsHostRecord* aFromHostRecord);
 150
 151   // Checks if the record is usable (not expired and has a value)
 152   bool HasUsableResult(const mozilla::TimeStamp& now,
 153                        nsIDNSService::DNSFlags queryFlags =
 154                            nsIDNSService::RESOLVE_DEFAULT_FLAGS) const;
 155
 156   static DnsPriority GetPriority(nsIDNSService::DNSFlags aFlags);
 157
 158   virtual void Cancel();
 159   virtual bool HasUsableResultInternal(
 160       const mozilla::TimeStamp& now,
 161       nsIDNSService::DNSFlags queryFlags) const = 0;
 162   virtual bool RefreshForNegativeResponse() const { return true; }
 163
 164   mozilla::LinkedList<RefPtr<nsResolveHostCallback>> mCallbacks;
 165
 166   bool IsAddrRecord() const {
 167     return type == nsIDNSService::RESOLVE_TYPE_DEFAULT;
 168   }
 169
 170   virtual void Reset() {
 171     mTRRSkippedReason = TRRSkippedReason::TRR_UNSET;
 172     mFirstTRRSkippedReason = TRRSkippedReason::TRR_UNSET;
 173     mTrrAttempts = 0;
 174     mTRRSuccess = false;
 175     mNativeSuccess = false;
 176   }
 177
 178   virtual void OnCompleteLookup() {}
 179
 180   virtual void ResolveComplete() = 0;
 181
 182   // true if pending and on the queue (not yet given to getaddrinfo())
 183   bool onQueue() { return LoadNative() && isInList(); }
 184
 185   // When the record began being valid. Used mainly for bookkeeping.
 186   mozilla::TimeStamp mValidStart;
 187
 188   // When the record is no longer valid (it's time of expiration)
 189   mozilla::TimeStamp mValidEnd;
 190
 191   // When the record enters its grace period. This must be before mValidEnd.
 192   // If a record is in its grace period (and not expired), it will be used
 193   // but a request to refresh it will be made.
 194   mozilla::TimeStamp mGraceStart;
 195
 196   mozilla::TimeDuration mTrrDuration;
 197
 198   mozilla::Atomic<uint32_t, mozilla::Relaxed> mTtl{0};
 199
 200   // The computed TRR mode that is actually used by the request.
 201   // It is set in nsHostResolver::NameLookup and is based on the mode of the
 202   // default resolver and the TRRMode encoded in the flags.
 203   // The mode into account if the TRR service is disabled,
 204   // parental controls are on, domain matches exclusion list, etc.
 205   mozilla::Atomic<nsIRequest::TRRMode> mEffectiveTRRMode{
 206       nsIRequest::TRR_DEFAULT_MODE};
 207
 208   mozilla::Atomic<TRRSkippedReason> mTRRSkippedReason{
 209       TRRSkippedReason::TRR_UNSET};
 210   TRRSkippedReason mFirstTRRSkippedReason = TRRSkippedReason::TRR_UNSET;
 211
 212   mozilla::DataMutex<RefPtr<mozilla::net::TRRQuery>> mTRRQuery;
 213
 214   // counter of outstanding resolving calls
 215   mozilla::Atomic<int32_t> mResolving{0};
 216
 217   // Number of times we've attempted TRR. Reset when we refresh.
 218   // TRR is attempted at most twice - first attempt and retry.
 219   mozilla::Atomic<int32_t> mTrrAttempts{0};
 220
 221   // True if this record is a cache of a failed lookup.  Negative cache
 222   // entries are valid just like any other (though never for more than 60
 223   // seconds), but a use of that negative entry forces an asynchronous refresh.
 224   bool negative = false;
 225
 226   // Explicitly expired
 227   bool mDoomed = false;
 228
 229   // Whether this is resolved by TRR successfully or not.
 230   bool mTRRSuccess = false;
 231
 232   // Whether this is resolved by native resolver successfully or not.
 233   bool mNativeSuccess = false;
 234
 235   // When the lookups of this record started and their durations
 236   mozilla::TimeStamp mNativeStart;
 237   mozilla::TimeDuration mNativeDuration;
 238
 239   // clang-format off
 240   MOZ_ATOMIC_BITFIELDS(mAtomicBitfields, 8, (
 241     // true if this record is being resolved "natively", which means that
 242     // it is either on the pending queue or owned by one of the worker threads.
 243     (uint16_t, Native, 1),
 244     (uint16_t, NativeUsed, 1),
 245     // true if off queue and contributing to mActiveAnyThreadCount
 246     (uint16_t, UsingAnyThread, 1),
 247     (uint16_t, GetTtl, 1),
 248     (uint16_t, ResolveAgain, 1)
 249   ))
 250   // clang-format on
 251 };
 252
 253 // b020e996-f6ab-45e5-9bf5-1da71dd0053a
 254 #define ADDRHOSTRECORD_IID                           \
 255   {                                                  \
 256     0xb020e996, 0xf6ab, 0x45e5, {                    \
 257       0x9b, 0xf5, 0x1d, 0xa7, 0x1d, 0xd0, 0x05, 0x3a \
 258     }                                                \
 259   }
 260
 261 class AddrHostRecord final : public nsHostRecord {
 262   using Mutex = mozilla::Mutex;
 263   using DNSResolverType = mozilla::net::DNSResolverType;
 264
 265  public:
 266   NS_DECLARE_STATIC_IID_ACCESSOR(ADDRHOSTRECORD_IID)
 267   NS_DECL_ISUPPORTS_INHERITED
 268
 269   /* a fully resolved host record has either a non-null |addr_info| or |addr|
 270    * field.  if |addr_info| is null, it implies that the |host| is an IP
 271    * address literal.  in which case, |addr| contains the parsed address.
 272    * otherwise, if |addr_info| is non-null, then it contains one or many
 273    * IP addresses corresponding to the given host name.  if both |addr_info|
 274    * and |addr| are null, then the given host has not yet been fully resolved.
 275    * |af| is the address family of the record we are querying for.
 276    */
 277
 278   /* the lock protects |addr_info| and |addr_info_gencnt| because they
 279    * are mutable and accessed by the resolver worker thread and the
 280    * nsDNSService2 class.  |addr| doesn't change after it has been
 281    * assigned a value.  only the resolver worker thread modifies
 282    * nsHostRecord (and only in nsHostResolver::CompleteLookup);
 283    * the other threads just read it.  therefore the resolver worker
 284    * thread doesn't need to lock when reading |addr_info|.
 285    */
 286   Mutex addr_info_lock MOZ_UNANNOTATED{"AddrHostRecord.addr_info_lock"};
 287   // generation count of |addr_info|
 288   int addr_info_gencnt = 0;
 289   RefPtr<mozilla::net::AddrInfo> addr_info;
 290   mozilla::UniquePtr<mozilla::net::NetAddr> addr;
 291
 292   // hold addr_info_lock when calling the blocklist functions
 293   bool Blocklisted(const mozilla::net::NetAddr* query);
 294   void ResetBlocklist();
 295   void ReportUnusable(const mozilla::net::NetAddr* aAddress);
 296
 297   size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const override;
 298
 299   nsIRequest::TRRMode EffectiveTRRMode() const { return mEffectiveTRRMode; }
 300   nsITRRSkipReason::value TrrSkipReason() const { return mTRRSkippedReason; }
 301
 302   nsresult GetTtl(uint32_t* aResult);
 303
 304  private:
 305   friend class nsHostResolver;
 306   friend class mozilla::net::HostRecordQueue;
 307   friend class mozilla::net::TRR;
 308   friend class mozilla::net::TRRQuery;
 309
 310   explicit AddrHostRecord(const nsHostKey& key);
 311   ~AddrHostRecord();
 312
 313   // Checks if the record is usable (not expired and has a value)
 314   bool HasUsableResultInternal(
 315       const mozilla::TimeStamp& now,
 316       nsIDNSService::DNSFlags queryFlags) const override;
 317
 318   bool RemoveOrRefresh(bool aTrrToo);  // Mark records currently being resolved
 319                                        // as needed to resolve again.
 320
 321   // Saves the skip reason of a first-attempt TRR lookup and clears
 322   // it to prepare for a retry attempt.
 323   void NotifyRetryingTrr();
 324
 325   static DnsPriority GetPriority(nsIDNSService::DNSFlags aFlags);
 326
 327   virtual void Reset() override {
 328     nsHostRecord::Reset();
 329     StoreNativeUsed(false);
 330     mResolverType = DNSResolverType::Native;
 331   }
 332
 333   virtual void OnCompleteLookup() override {
 334     nsHostRecord::OnCompleteLookup();
 335     // This should always be cleared when a request is completed.
 336     StoreNative(false);
 337   }
 338
 339   void ResolveComplete() override;
 340
 341   // TRR was used on this record
 342   mozilla::Atomic<DNSResolverType> mResolverType{DNSResolverType::Native};
 343
 344   // The number of times ReportUnusable() has been called in the record's
 345   // lifetime.
 346   uint32_t mUnusableCount = 0;
 347
 348   // a list of addresses associated with this record that have been reported
 349   // as unusable. the list is kept as a set of strings to make it independent
 350   // of gencnt.
 351   nsTArray<nsCString> mUnusableItems;
 352 };
 353
 354 NS_DEFINE_STATIC_IID_ACCESSOR(AddrHostRecord, ADDRHOSTRECORD_IID)
 355
 356 // 77b786a7-04be-44f2-987c-ab8aa96676e0
 357 #define TYPEHOSTRECORD_IID                           \
 358   {                                                  \
 359     0x77b786a7, 0x04be, 0x44f2, {                    \
 360       0x98, 0x7c, 0xab, 0x8a, 0xa9, 0x66, 0x76, 0xe0 \
 361     }                                                \
 362   }
 363
 364 class TypeHostRecord final : public nsHostRecord,
 365                              public nsIDNSTXTRecord,
 366                              public nsIDNSHTTPSSVCRecord,
 367                              public mozilla::net::DNSHTTPSSVCRecordBase {
 368  public:
 369   NS_DECLARE_STATIC_IID_ACCESSOR(TYPEHOSTRECORD_IID)
 370   NS_DECL_ISUPPORTS_INHERITED
 371   NS_DECL_NSIDNSTXTRECORD
 372   NS_DECL_NSIDNSHTTPSSVCRECORD
 373
 374   size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const override;
 375   uint32_t GetType();
 376   mozilla::net::TypeRecordResultType GetResults();
 377
 378  private:
 379   friend class nsHostResolver;
 380   friend class mozilla::net::TRR;
 381   friend class mozilla::net::TRRQuery;
 382
 383   explicit TypeHostRecord(const nsHostKey& key);
 384   ~TypeHostRecord();
 385
 386   // Checks if the record is usable (not expired and has a value)
 387   bool HasUsableResultInternal(
 388       const mozilla::TimeStamp& now,
 389       nsIDNSService::DNSFlags queryFlags) const override;
 390   bool RefreshForNegativeResponse() const override;
 391
 392   void ResolveComplete() override;
 393
 394   mozilla::net::TypeRecordResultType mResults = AsVariant(mozilla::Nothing());
 395   mozilla::Mutex mResultsLock MOZ_UNANNOTATED{"TypeHostRecord.mResultsLock"};
 396
 397   mozilla::Maybe<nsCString> mOriginHost;
 398   bool mAllRecordsExcluded = false;
 399 };
 400
 401 NS_DEFINE_STATIC_IID_ACCESSOR(TypeHostRecord, TYPEHOSTRECORD_IID)
 402
 403 static inline bool IsHighPriority(nsIDNSService::DNSFlags flags) {
 404   return !(flags & (nsHostRecord::DNS_PRIORITY_LOW |
 405                     nsHostRecord::DNS_PRIORITY_MEDIUM));
 406 }
 407
 408 static inline bool IsMediumPriority(nsIDNSService::DNSFlags flags) {
 409   return flags & nsHostRecord::DNS_PRIORITY_MEDIUM;
 410 }
 411
 412 static inline bool IsLowPriority(nsIDNSService::DNSFlags flags) {
 413   return flags & nsHostRecord::DNS_PRIORITY_LOW;
 414 }
 415
 416 #endif  // nsHostRecord_h__