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
|
// Copyright 2023 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/interest_group/auction_shared_storage_host.h"
#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "content/browser/shared_storage/shared_storage_lock_manager.h"
#include "content/browser/shared_storage/shared_storage_runtime_manager.h"
#include "content/browser/storage_partition_impl.h"
#include "services/network/public/mojom/shared_storage.mojom.h"
#include "third_party/blink/public/common/shared_storage/shared_storage_utils.h"
namespace content {
namespace {
using AccessScope = blink::SharedStorageAccessScope;
blink::mojom::WebFeature ToWebFeature(
auction_worklet::mojom::AuctionWorkletFunction auction_worklet_function) {
switch (auction_worklet_function) {
case auction_worklet::mojom::AuctionWorkletFunction::kBidderGenerateBid:
return blink::mojom::WebFeature::kSharedStorageWriteFromBidderGenerateBid;
case auction_worklet::mojom::AuctionWorkletFunction::kBidderReportWin:
return blink::mojom::WebFeature::kSharedStorageWriteFromBidderReportWin;
case auction_worklet::mojom::AuctionWorkletFunction::kSellerScoreAd:
return blink::mojom::WebFeature::kSharedStorageWriteFromSellerScoreAd;
case auction_worklet::mojom::AuctionWorkletFunction::kSellerReportResult:
return blink::mojom::WebFeature::
kSharedStorageWriteFromSellerReportResult;
}
NOTREACHED();
}
} // namespace
struct AuctionSharedStorageHost::ReceiverContext {
// `auction_runner_rfh` is the frame associated with the
// `AdAuctionServiceImpl` that owns `this`. Thus, `auction_runner_rfh` must
// outlive `this`.
raw_ptr<RenderFrameHostImpl> auction_runner_rfh;
url::Origin worklet_origin;
};
AuctionSharedStorageHost::AuctionSharedStorageHost(
StoragePartitionImpl* storage_partition)
: storage_partition_(storage_partition) {}
AuctionSharedStorageHost::~AuctionSharedStorageHost() = default;
void AuctionSharedStorageHost::BindNewReceiver(
RenderFrameHostImpl* auction_runner_rfh,
const url::Origin& worklet_origin,
mojo::PendingReceiver<auction_worklet::mojom::AuctionSharedStorageHost>
receiver) {
receiver_set_.Add(this, std::move(receiver),
ReceiverContext{.auction_runner_rfh = auction_runner_rfh,
.worklet_origin = worklet_origin});
}
void AuctionSharedStorageHost::SharedStorageUpdate(
network::mojom::SharedStorageModifierMethodWithOptionsPtr
method_with_options,
auction_worklet::mojom::AuctionWorkletFunction
source_auction_worklet_function) {
GlobalRenderFrameHostId main_frame_id =
receiver_set_.current_context()
.auction_runner_rfh->GetOutermostMainFrame()
->GetGlobalId();
storage_partition_->GetSharedStorageRuntimeManager()
->lock_manager()
.SharedStorageUpdate(
std::move(method_with_options),
receiver_set_.current_context().worklet_origin,
AccessScope::kProtectedAudienceWorklet, main_frame_id,
/*worklet_devtools_token=*/base::UnguessableToken::Null(),
base::DoNothing());
GetContentClient()->browser()->LogWebFeatureForCurrentPage(
receiver_set_.current_context().auction_runner_rfh,
ToWebFeature(source_auction_worklet_function));
}
void AuctionSharedStorageHost::SharedStorageBatchUpdate(
std::vector<network::mojom::SharedStorageModifierMethodWithOptionsPtr>
methods_with_options,
const std::optional<std::string>& with_lock,
auction_worklet::mojom::AuctionWorkletFunction
source_auction_worklet_function) {
GlobalRenderFrameHostId main_frame_id =
receiver_set_.current_context()
.auction_runner_rfh->GetOutermostMainFrame()
->GetGlobalId();
storage_partition_->GetSharedStorageRuntimeManager()
->lock_manager()
.SharedStorageBatchUpdate(
std::move(methods_with_options), with_lock,
receiver_set_.current_context().worklet_origin,
AccessScope::kProtectedAudienceWorklet, main_frame_id,
/*worklet_devtools_token=*/base::UnguessableToken::Null(),
base::DoNothing());
GetContentClient()->browser()->LogWebFeatureForCurrentPage(
receiver_set_.current_context().auction_runner_rfh,
ToWebFeature(source_auction_worklet_function));
}
} // namespace content
|
