1 files changed, 112 insertions, 0 deletions
diff --git a/chromium/net/dns/httpssvc_metrics.h b/chromium/net/dns/httpssvc_metrics.h
new file mode 100644
index 00000000000..76e4795cafc
--- /dev/null
+++ b/chromium/net/dns/httpssvc_metrics.h
@@ -0,0 +1,112 @@
+// Copyright 2020 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef NET_DNS_HTTPSSVC_METRICS_H_
+#define NET_DNS_HTTPSSVC_METRICS_H_
+
+#include <string>
+#include <vector>
+
+#include "base/containers/flat_set.h"
+#include "base/optional.h"
+#include "base/strings/string_piece_forward.h"
+#include "base/time/time.h"
+#include "net/base/net_export.h"
+
+namespace net {
+
+// These values are persisted to logs. Entries should not be renumbered and
+// numeric values should never be reused. (See HttpssvcDnsRcode in
+// tools/metrics/histograms/enums.xml.)
+enum HttpssvcDnsRcode {
+  kTimedOut = 0,
+  kUnrecognizedRcode,
+  kMissingDnsResponse,
+  kNoError,
+  kFormErr,
+  kServFail,
+  kNxDomain,
+  kNotImp,
+  kRefused,
+  kMaxValue = kRefused,
+};
+
+// Helper that classifies domains as experimental, control, or other. Queries
+// feature params and caches result to avoid repeated parsing.
+class NET_EXPORT_PRIVATE HttpssvcExperimentDomainCache {
+ public:
+  HttpssvcExperimentDomainCache();
+  ~HttpssvcExperimentDomainCache();
+  bool IsExperimental(base::StringPiece domain);
+  bool IsControl(base::StringPiece domain);
+
+ private:
+  bool ListContainsDomain(
+      const std::string& domain_list,
+      base::StringPiece domain,
+      base::Optional<base::flat_set<std::string>>& in_out_cached_list);
+
+  base::Optional<base::flat_set<std::string>> experimental_list_;
+  base::Optional<base::flat_set<std::string>> control_list_;
+};
+
+// Translate an RCODE value to the |HttpssvcDnsRcode| enum, which is used for
+// HTTPSSVC experimentation. The goal is to keep these values in a small,
+// contiguous range in order to satisfy the UMA enumeration function's
+// requirements. This function never returns |kTimedOut| |kUnrecognizedRcode|,
+// or |kMissingDnsResponse|.
+enum HttpssvcDnsRcode TranslateDnsRcodeForHttpssvcExperiment(uint8_t rcode);
+
+// Tool for aggregating HTTPSSVC and INTEGRITY metrics. Accumulates metrics via
+// the Save* methods. Records metrics to UMA on destruction.
+class NET_EXPORT_PRIVATE HttpssvcMetrics {
+ public:
+  explicit HttpssvcMetrics(bool expect_intact);
+  ~HttpssvcMetrics();
+  HttpssvcMetrics(HttpssvcMetrics&) = delete;
+  HttpssvcMetrics(HttpssvcMetrics&&) = delete;
+
+  // May be called many times.
+  void SaveForNonIntegrity(base::Optional<std::string> doh_provider_id,
+                           base::TimeDelta resolve_time,
+                           enum HttpssvcDnsRcode rcode);
+
+  // Save the fact that the non-integrity queries failed. Prevents metrics from
+  // being recorded.
+  void SaveNonIntegrityFailure();
+
+  // Must only be called once.
+  void SaveForIntegrity(base::Optional<std::string> doh_provider_id,
+                        enum HttpssvcDnsRcode rcode,
+                        const std::vector<bool>& condensed_records,
+                        base::TimeDelta integrity_resolve_time);
+
+ private:
+  std::string BuildMetricName(base::StringPiece leaf_name) const;
+
+  // Records all the aggregated metrics to UMA.
+  void RecordIntegrityMetrics();
+  void RecordIntegrityCommonMetrics();
+  void RecordIntegrityExpectIntactMetrics();
+  void RecordIntegrityExpectNoerrorMetrics();
+
+  void set_doh_provider_id(base::Optional<std::string> doh_provider_id);
+
+  // RecordIntegrityMetrics() will do nothing when |disqualified_| is true.
+  bool disqualified_ = false;
+  const bool expect_intact_;
+  bool in_progress_ = true;
+  base::Optional<std::string> doh_provider_id_;
+  base::Optional<enum HttpssvcDnsRcode> rcode_integrity_;
+  size_t num_integrity_records_ = 0;
+  base::Optional<bool> is_integrity_intact_;
+  // We never make multiple INTEGRITY queries per DnsTask, so we only need
+  // one TimeDelta for the INTEGRITY query.
+  base::Optional<base::TimeDelta> integrity_resolve_time_;
+  std::vector<base::TimeDelta> non_integrity_resolve_times_;
+};
+
+}  // namespace net
+
+#endif  // NET_DNS_HTTPSSVC_METRICS_H_