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
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/390223051): Remove C-library calls to fix the errors.
#pragma allow_unsafe_libc_calls
#endif
#include "base/memory/aligned_memory.h"
#include <stdint.h>
#include <string.h>
#include <memory>
#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace base {
TEST(AlignedMemoryTest, AlignedUninit) {
{
base::AlignedHeapArray<char> h = AlignedUninit<char>(8, 32);
EXPECT_EQ(h.size(), 8u);
EXPECT_TRUE(IsAligned(h.data(), 32));
}
{
base::AlignedHeapArray<int16_t> h = AlignedUninit<int16_t>(8, 32);
EXPECT_EQ(h.size(), 8u);
EXPECT_TRUE(IsAligned(h.data(), 32));
}
}
TEST(AlignedMemoryTest, AlignedUninitCharArray) {
auto [h, s] = AlignedUninitCharArray<int16_t>(8, 32);
static_assert(std::same_as<base::AlignedHeapArray<char>, decltype(h)>);
static_assert(std::same_as<base::span<int16_t>, decltype(s)>);
EXPECT_EQ(h.size(), 8u * sizeof(int16_t));
EXPECT_TRUE(IsAligned(h.data(), 32));
EXPECT_EQ(s.size(), 8u);
EXPECT_TRUE(IsAligned(s.data(), 32));
}
TEST(AlignedMemoryTest, DynamicAllocation) {
void* p = AlignedAlloc(8, 8);
ASSERT_TRUE(p);
EXPECT_TRUE(IsAligned(p, 8));
memset(p, 0, 8); // Fill to check allocated size under ASAN.
AlignedFree(p);
p = AlignedAlloc(8, 16);
ASSERT_TRUE(p);
EXPECT_TRUE(IsAligned(p, 16));
memset(p, 0, 8); // Fill to check allocated size under ASAN.
AlignedFree(p);
p = AlignedAlloc(8, 256);
ASSERT_TRUE(p);
EXPECT_TRUE(IsAligned(p, 256));
memset(p, 0, 8); // Fill to check allocated size under ASAN.
AlignedFree(p);
p = AlignedAlloc(8, 4096);
ASSERT_TRUE(p);
EXPECT_TRUE(IsAligned(p, 4096));
memset(p, 0, 8); // Fill to check allocated size under ASAN.
AlignedFree(p);
}
TEST(AlignedMemoryTest, ScopedDynamicAllocation) {
std::unique_ptr<float, AlignedFreeDeleter> p(
static_cast<float*>(AlignedAlloc(8, 8)));
EXPECT_TRUE(p.get());
EXPECT_TRUE(IsAligned(p.get(), 8));
// Make sure IsAligned() can check const pointers as well.
const float* const_p = p.get();
EXPECT_TRUE(IsAligned(const_p, 8));
}
TEST(AlignedMemoryTest, IsAligned) {
// Check alignment around powers of two.
for (int i = 0; i < 64; ++i) {
const uint64_t n = static_cast<uint64_t>(1) << i;
// Walk back down all lower powers of two checking alignment.
for (int j = i - 1; j >= 0; --j) {
// n is aligned on all powers of two less than or equal to 2^i.
EXPECT_TRUE(IsAligned(n, n >> j))
<< "Expected " << n << " to be " << (n >> j) << " aligned";
// Also, n - 1 should not be aligned on ANY lower power of two except 1
// (but since we're starting from i - 1 we don't test that case here.
EXPECT_FALSE(IsAligned(n - 1, n >> j))
<< "Expected " << (n - 1) << " to NOT be " << (n >> j) << " aligned";
}
}
// And a few hard coded smoke tests for completeness:
EXPECT_TRUE(IsAligned(4, 2));
EXPECT_TRUE(IsAligned(8, 4));
EXPECT_TRUE(IsAligned(8, 2));
EXPECT_TRUE(IsAligned(0x1000, 4 << 10));
EXPECT_TRUE(IsAligned(0x2000, 8 << 10));
EXPECT_TRUE(IsAligned(1, 1));
EXPECT_TRUE(IsAligned(7, 1));
EXPECT_TRUE(IsAligned(reinterpret_cast<void*>(0x1000), 4 << 10));
EXPECT_TRUE(IsAligned(reinterpret_cast<int*>(0x1000), 4 << 10));
EXPECT_FALSE(IsAligned(3, 2));
EXPECT_FALSE(IsAligned(7, 4));
EXPECT_FALSE(IsAligned(7, 2));
EXPECT_FALSE(IsAligned(0x1001, 4 << 10));
EXPECT_FALSE(IsAligned(0x999, 8 << 10));
EXPECT_FALSE(IsAligned(7, 8));
}
} // namespace base
|
