1 | //===-- Unittests for memmove ---------------------------------------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "hdr/signal_macros.h" |
10 | #include "src/__support/macros/config.h" |
11 | #include "src/string/memmove.h" |
12 | |
13 | #include "memory_utils/memory_check_utils.h" |
14 | #include "src/__support/CPP/span.h" |
15 | #include "test/UnitTest/MemoryMatcher.h" |
16 | #include "test/UnitTest/Test.h" |
17 | |
18 | using LIBC_NAMESPACE::cpp::array; |
19 | using LIBC_NAMESPACE::cpp::span; |
20 | |
21 | namespace LIBC_NAMESPACE_DECL { |
22 | |
23 | TEST(LlvmLibcMemmoveTest, MoveZeroByte) { |
24 | char Buffer[] = {'a', 'b', 'y', 'z'}; |
25 | const char Expected[] = {'a', 'b', 'y', 'z'}; |
26 | void *const Dst = Buffer; |
27 | void *const Ret = LIBC_NAMESPACE::memmove(Dst, Buffer + 2, 0); |
28 | EXPECT_EQ(Ret, Dst); |
29 | ASSERT_MEM_EQ(Buffer, testing::MemoryView(Expected)); |
30 | } |
31 | |
32 | TEST(LlvmLibcMemmoveTest, DstAndSrcPointToSameAddress) { |
33 | char Buffer[] = {'a', 'b'}; |
34 | const char Expected[] = {'a', 'b'}; |
35 | void *const Dst = Buffer; |
36 | void *const Ret = LIBC_NAMESPACE::memmove(Dst, Buffer, 1); |
37 | EXPECT_EQ(Ret, Dst); |
38 | ASSERT_MEM_EQ(Buffer, testing::MemoryView(Expected)); |
39 | } |
40 | |
41 | TEST(LlvmLibcMemmoveTest, DstStartsBeforeSrc) { |
42 | // Set boundary at beginning and end for not overstepping when |
43 | // copy forward or backward. |
44 | char Buffer[] = {'z', 'a', 'b', 'c', 'z'}; |
45 | const char Expected[] = {'z', 'b', 'c', 'c', 'z'}; |
46 | void *const Dst = Buffer + 1; |
47 | void *const Ret = LIBC_NAMESPACE::memmove(Dst, Buffer + 2, 2); |
48 | EXPECT_EQ(Ret, Dst); |
49 | ASSERT_MEM_EQ(Buffer, testing::MemoryView(Expected)); |
50 | } |
51 | |
52 | TEST(LlvmLibcMemmoveTest, DstStartsAfterSrc) { |
53 | char Buffer[] = {'z', 'a', 'b', 'c', 'z'}; |
54 | const char Expected[] = {'z', 'a', 'a', 'b', 'z'}; |
55 | void *const Dst = Buffer + 2; |
56 | void *const Ret = LIBC_NAMESPACE::memmove(Dst, Buffer + 1, 2); |
57 | EXPECT_EQ(Ret, Dst); |
58 | ASSERT_MEM_EQ(Buffer, testing::MemoryView(Expected)); |
59 | } |
60 | |
61 | // e.g. `Dst` follow `src`. |
62 | // str: [abcdefghij] |
63 | // [__src_____] |
64 | // [_____Dst__] |
65 | TEST(LlvmLibcMemmoveTest, SrcFollowDst) { |
66 | char Buffer[] = {'z', 'a', 'b', 'z'}; |
67 | const char Expected[] = {'z', 'b', 'b', 'z'}; |
68 | void *const Dst = Buffer + 1; |
69 | void *const Ret = LIBC_NAMESPACE::memmove(Dst, Buffer + 2, 1); |
70 | EXPECT_EQ(Ret, Dst); |
71 | ASSERT_MEM_EQ(Buffer, testing::MemoryView(Expected)); |
72 | } |
73 | |
74 | TEST(LlvmLibcMemmoveTest, DstFollowSrc) { |
75 | char Buffer[] = {'z', 'a', 'b', 'z'}; |
76 | const char Expected[] = {'z', 'a', 'a', 'z'}; |
77 | void *const Dst = Buffer + 2; |
78 | void *const Ret = LIBC_NAMESPACE::memmove(Dst, Buffer + 1, 1); |
79 | EXPECT_EQ(Ret, Dst); |
80 | ASSERT_MEM_EQ(Buffer, testing::MemoryView(Expected)); |
81 | } |
82 | |
83 | // Adapt CheckMemmove signature to op implementation signatures. |
84 | static inline void Adaptor(cpp::span<char> dst, cpp::span<char> src, |
85 | size_t size) { |
86 | LIBC_NAMESPACE::memmove(dst.begin(), src.begin(), size); |
87 | } |
88 | |
89 | TEST(LlvmLibcMemmoveTest, SizeSweep) { |
90 | static constexpr int kMaxSize = 400; |
91 | static constexpr int kDenseOverlap = 15; |
92 | using LargeBuffer = array<char, 2 * kMaxSize + 1>; |
93 | LargeBuffer Buffer; |
94 | Randomize(Buffer); |
95 | for (int Size = 0; Size < kMaxSize; ++Size) |
96 | for (int Overlap = -1; Overlap < Size;) { |
97 | ASSERT_TRUE( |
98 | CheckMemmove<Adaptor>(Buffer, static_cast<size_t>(Size), Overlap)); |
99 | // Prevent quadratic behavior by skipping offset above kDenseOverlap. |
100 | if (Overlap > kDenseOverlap) |
101 | Overlap *= 2; |
102 | else |
103 | ++Overlap; |
104 | } |
105 | } |
106 | |
107 | #if defined(LIBC_ADD_NULL_CHECKS) && !defined(LIBC_HAS_SANITIZER) |
108 | |
109 | TEST(LlvmLibcMemmoveTest, CrashOnNullPtr) { |
110 | ASSERT_DEATH([]() { LIBC_NAMESPACE::memmove(nullptr, nullptr, 2); }, |
111 | WITH_SIGNAL(-1)); |
112 | } |
113 | |
114 | #endif // LIBC_TARGET_OS_IS_LINUX |
115 | |
116 | } // namespace LIBC_NAMESPACE_DECL |
117 | |