chrome/browser/data_use_measurement/chrome_data_use_measurement.cc

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/data_use_measurement/chrome_data_use_measurement.h"

#include "base/metrics/histogram_functions.h"
#include "base/metrics/histogram_macros.h"
#include "base/no_destructor.h"
#include "base/trace_event/trace_event.h"
#include "content/public/browser/browser_thread.h"

using content::BrowserThread;

// static
ChromeDataUseMeasurement& ChromeDataUseMeasurement::GetInstance() {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI) ||
         !BrowserThread::IsThreadInitialized(BrowserThread::UI));

  static base::NoDestructor<ChromeDataUseMeasurement>
      s_chrome_data_use_measurement;
  return *s_chrome_data_use_measurement;
}

ChromeDataUseMeasurement::ChromeDataUseMeasurement() = default;

ChromeDataUseMeasurement::~ChromeDataUseMeasurement() = default;

void ChromeDataUseMeasurement::ReportNetworkServiceDataUse(
    int32_t network_traffic_annotation_id_hash,
    int64_t recv_bytes,
    int64_t sent_bytes) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  // Negative byte numbers is not a critical problem (i.e., should have no
  // security implications) but is not expected. TODO(rajendrant): remove these
  // DCHECKs or consider using uint in Mojo instead.
  DCHECK_GE(recv_bytes, 0);
  DCHECK_GE(sent_bytes, 0);

  ReportDataUsage(TrafficDirection::kUpstream, sent_bytes);
  ReportDataUsage(TrafficDirection::kDownstream, recv_bytes);
}

void ChromeDataUseMeasurement::ReportDataUsage(TrafficDirection dir,
                                               int64_t message_size_bytes) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  TRACE_EVENT0("browser", "ChromeDataUseMeasurement::ReportDataUsage");

  if (message_size_bytes <= 0)
    return;

  if (dir == TrafficDirection::kDownstream) {
    base::UmaHistogramCustomCounts("DataUse.BytesReceived3.Delegate",
                                   message_size_bytes, 50, 10 * 1000 * 1000,
                                   50);
  }

  if (dir == TrafficDirection::kUpstream)
    UMA_HISTOGRAM_COUNTS_1M("DataUse.BytesSent3.Delegate", message_size_bytes);
}