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
|
// Copyright 2025 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/ui/unowned_user_data/unowned_user_data_host.h"
#include <ostream>
#include "base/check.h"
UnownedUserDataHost::UnownedUserDataHost() = default;
UnownedUserDataHost::~UnownedUserDataHost() {
// All UnownedUserData should be removed before the host is destroyed;
// otherwise, there could be a UAF when they try to remove themselves as they
// are destroyed.
// If any remain, print out the first entry's key. There should never be any,
// so this should be sufficient to help folks debug.
CHECK(map_.empty()) << "All UnownedUserData must be removed before the "
<< "corresponding UnownedUserDataHost is destroyed. "
<< "First remaining key: " << map_.begin()->first;
}
void UnownedUserDataHost::MarkKeyForTesting(const char* key) {
testing_keys_.insert(key);
}
void UnownedUserDataHost::Set(
base::PassKey<internal::ScopedUnownedUserDataBase> pass_key,
const char* key,
void* data) {
CHECK(data) << "Assigning bad data for key: " << key;
bool inserted = map_.insert_or_assign(key, data).second;
// Ensure a new value was inserted into the map unless the key was explicitly
// marked as being used for testing (in which case, we allow it to be
// overwritten).
CHECK(inserted || testing_keys_.contains(key))
<< "Attempted to reinsert data for key: " << key;
}
void UnownedUserDataHost::Erase(
base::PassKey<internal::ScopedUnownedUserDataBase> pass_key,
const char* key) {
bool erased = map_.erase(key);
// The value should have been erased unless the key was marked as being used
// in testing. In that case, the previous testing instance may have erased the
// entry in the map, and we don't expect a second erasure.
CHECK(erased || testing_keys_.contains(key))
<< "Erasing invalid data for key: " << key;
}
void* UnownedUserDataHost::Get(
base::PassKey<internal::ScopedUnownedUserDataBase>,
const char* key) {
auto iter = map_.find(key);
return iter == map_.end() ? nullptr : iter->second;
}
|
