1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103
  104
  105
  106
  107
  108
  109
  110
  111
  112
  113
  114
  115
  116
  117
  118
  119
  120
  121
  122


content / browser / loader / keep_alive_attribution_request_helper.cc [blame]

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

#include "content/browser/loader/keep_alive_attribution_request_helper.h"

#include <stdint.h>

#include <memory>
#include <optional>
#include <string>
#include <utility>

#include "base/atomic_sequence_num.h"
#include "base/check.h"
#include "base/feature_list.h"
#include "base/memory/ptr_util.h"
#include "base/memory/weak_ptr.h"
#include "base/unguessable_token.h"
#include "components/attribution_reporting/eligibility.h"
#include "components/attribution_reporting/features.h"
#include "components/attribution_reporting/registration_eligibility.mojom-forward.h"
#include "components/attribution_reporting/suitable_origin.h"
#include "content/browser/attribution_reporting/attribution_background_registrations_id.h"
#include "content/browser/attribution_reporting/attribution_data_host_manager.h"
#include "content/browser/attribution_reporting/attribution_suitable_context.h"
#include "content/public/browser/global_routing_id.h"
#include "net/http/http_response_headers.h"
#include "services/network/public/mojom/attribution.mojom-forward.h"
#include "third_party/blink/public/common/features.h"
#include "third_party/blink/public/common/tokens/tokens.h"
#include "url/gurl.h"

namespace content {

namespace {

using ::attribution_reporting::mojom::RegistrationEligibility;

}  // namespace

// static
std::unique_ptr<KeepAliveAttributionRequestHelper>
KeepAliveAttributionRequestHelper::CreateIfNeeded(
    network::mojom::AttributionReportingEligibility eligibility,
    const GURL& request_url,
    const std::optional<base::UnguessableToken>& attribution_src_token,
    const std::optional<std::string>& devtools_request_id,
    const AttributionSuitableContext& context) {
  if (!base::FeatureList::IsEnabled(
          blink::features::kAttributionReportingInBrowserMigration)) {
    return nullptr;
  }

  std::optional<RegistrationEligibility> registration_eligibility =
      attribution_reporting::GetRegistrationEligibility(eligibility);
  if (!registration_eligibility.has_value()) {
    return nullptr;
  }

  AttributionDataHostManager* data_host_manager = context.data_host_manager();
  if (!data_host_manager) {
    return nullptr;
  }

  std::optional<blink::AttributionSrcToken> token;
  if (attribution_src_token.has_value()) {
    token = blink::AttributionSrcToken(attribution_src_token.value());
  }

  static base::AtomicSequenceNumber unique_id_counter;
  BackgroundRegistrationsId id(unique_id_counter.GetNext());

  data_host_manager->NotifyBackgroundRegistrationStarted(
      id, context, *registration_eligibility, std::move(token),
      devtools_request_id);
  return base::WrapUnique(
      new KeepAliveAttributionRequestHelper(id, data_host_manager,
                                            /*reporting_url=*/request_url));
}

KeepAliveAttributionRequestHelper::KeepAliveAttributionRequestHelper(
    BackgroundRegistrationsId id,
    AttributionDataHostManager* attribution_data_host_manager,
    const GURL& reporting_url)
    : id_(id), reporting_url_(reporting_url) {
  CHECK(attribution_data_host_manager);
  attribution_data_host_manager_ = attribution_data_host_manager->AsWeakPtr();
}

void KeepAliveAttributionRequestHelper::OnReceiveRedirect(
    const net::HttpResponseHeaders* headers,
    const GURL& redirect_url) {
  if (!attribution_data_host_manager_) {
    return;
  }

  attribution_data_host_manager_->NotifyBackgroundRegistrationData(
      id_, headers, reporting_url_);
  reporting_url_ = redirect_url;
}

void KeepAliveAttributionRequestHelper::OnReceiveResponse(
    const net::HttpResponseHeaders* headers) {
  if (!attribution_data_host_manager_) {
    return;
  }

  attribution_data_host_manager_->NotifyBackgroundRegistrationData(
      id_, headers, reporting_url_);
  attribution_data_host_manager_->NotifyBackgroundRegistrationCompleted(id_);
  attribution_data_host_manager_.reset();
}

KeepAliveAttributionRequestHelper::~KeepAliveAttributionRequestHelper() {
  if (!attribution_data_host_manager_) {
    return;
  }
  attribution_data_host_manager_->NotifyBackgroundRegistrationCompleted(id_);
}

}  // namespace content