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
|
// 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 "components/gwp_asan/common/lightweight_detector_state.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace gwp_asan::internal {
static constexpr size_t kMaxMetadata = LightweightDetectorState::kMaxMetadata;
class LightweightDetectorStateTest : public testing::Test {
protected:
void InitializeState(size_t num_metadata, uintptr_t metadata_addr) {
state_.num_metadata = num_metadata;
state_.metadata_addr = metadata_addr;
}
LightweightDetectorState state_;
};
TEST_F(LightweightDetectorStateTest, Valid) {
InitializeState(1, 0x1234);
EXPECT_TRUE(state_.IsValid());
InitializeState(kMaxMetadata, 0x1234);
EXPECT_TRUE(state_.IsValid());
}
TEST_F(LightweightDetectorStateTest, InvalidNumMetadata) {
InitializeState(kMaxMetadata + 1, 0x1234);
EXPECT_FALSE(state_.IsValid());
}
TEST_F(LightweightDetectorStateTest, InvalidAddress) {
InitializeState(kMaxMetadata, 0);
EXPECT_FALSE(state_.IsValid());
}
TEST_F(LightweightDetectorStateTest, EncodeMetadataId) {
static_assert(sizeof(LightweightDetectorState::MetadataId) == 4,
"Update the test to sufficiently cover the MetadataId range.");
LightweightDetectorState::MetadataId id;
for (id = 0; id < 4; ++id) {
EXPECT_EQ(id, LightweightDetectorState::ExtractMetadataId(
LightweightDetectorState::EncodeMetadataId(id)));
}
for (id = 0x100; id < 0x100 + 4; ++id) {
EXPECT_EQ(id, LightweightDetectorState::ExtractMetadataId(
LightweightDetectorState::EncodeMetadataId(id)));
}
for (id = 0x10000; id < 0x10000 + 4; ++id) {
EXPECT_EQ(id, LightweightDetectorState::ExtractMetadataId(
LightweightDetectorState::EncodeMetadataId(id)));
}
for (id = 0x1000000; id < 0x1000000 + 4; ++id) {
EXPECT_EQ(id, LightweightDetectorState::ExtractMetadataId(
LightweightDetectorState::EncodeMetadataId(id)));
}
id = 0xffffffff;
EXPECT_EQ(id, LightweightDetectorState::ExtractMetadataId(
LightweightDetectorState::EncodeMetadataId(id)));
}
} // namespace gwp_asan::internal
|
