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
|
// 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 "chrome/browser/hid/web_view_chooser_context.h"
#include "base/containers/map_util.h"
#include "base/feature_list.h"
#include "chrome/browser/hid/hid_chooser_context.h"
#include "extensions/common/extension_features.h"
WebViewChooserContext::WebViewChooserContext(HidChooserContext* chooser_context)
: chooser_context_(chooser_context) {
permission_observation_.Observe(chooser_context_);
}
WebViewChooserContext::~WebViewChooserContext() = default;
void WebViewChooserContext::GrantDevicePermission(
const url::Origin& origin,
const url::Origin& embedding_origin,
const device::mojom::HidDeviceInfo& device) {
CHECK(base::FeatureList::IsEnabled(
extensions_features::kEnableWebHidInWebView));
CHECK(chooser_context_->HasDevicePermission(embedding_origin, device));
device_access_[embedding_origin][device.guid].insert(origin);
chooser_context_->PermissionForWebViewChanged();
}
bool WebViewChooserContext::HasDevicePermission(
const url::Origin& origin,
const url::Origin& embedding_origin,
const device::mojom::HidDeviceInfo& device) const {
auto* origins_per_device = base::FindOrNull(device_access_, embedding_origin);
if (!origins_per_device) {
return false;
}
auto* origins = base::FindOrNull(*origins_per_device, device.guid);
if (!origins) {
return false;
}
return origins->contains(origin);
}
void WebViewChooserContext::RevokeDevicePermission(
const url::Origin& origin,
const url::Origin& embedding_origin,
const device::mojom::HidDeviceInfo& device) {
auto it = device_access_.find(embedding_origin);
if (it == device_access_.end()) {
return;
}
auto device_it = it->second.find(device.guid);
if (device_it == it->second.end()) {
return;
}
bool revoked_permission = device_it->second.erase(origin) > 0;
if (device_it->second.empty()) {
it->second.erase(device_it);
}
if (it->second.empty()) {
device_access_.erase(it);
}
if (revoked_permission) {
chooser_context_->PermissionForWebViewRevoked(origin);
}
}
void WebViewChooserContext::OnPermissionRevoked(const url::Origin& origin) {
auto* origins_per_device = base::FindOrNull(device_access_, origin);
if (!origins_per_device) {
return;
}
// If permission was revoked for the embedding origin, permission for web view
// must be revoked as well.
for (auto it = origins_per_device->begin();
it != origins_per_device->end();) {
auto* device = chooser_context_->GetDeviceInfo(it->first);
if (!device || chooser_context_->HasDevicePermission(origin, *device)) {
++it;
continue;
}
std::set<url::Origin> revoked_guest_origins;
revoked_guest_origins.swap(it->second);
for (const auto& guest_origin : revoked_guest_origins) {
chooser_context_->PermissionForWebViewRevoked(guest_origin);
}
it = origins_per_device->erase(it);
}
}
void WebViewChooserContext::OnHidChooserContextShutdown() {
permission_observation_.Reset();
}
|
