1*9e9803bfSMichael Jones //===-- Unittests for strncat ---------------------------------------------===//
2*9e9803bfSMichael Jones //
3*9e9803bfSMichael Jones // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*9e9803bfSMichael Jones // See https://llvm.org/LICENSE.txt for license information.
5*9e9803bfSMichael Jones // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*9e9803bfSMichael Jones //
7*9e9803bfSMichael Jones //===----------------------------------------------------------------------===//
8*9e9803bfSMichael Jones
9*9e9803bfSMichael Jones #include "src/string/strncat.h"
10*9e9803bfSMichael Jones #include "utils/UnitTest/Test.h"
11*9e9803bfSMichael Jones
TEST(LlvmLibcStrNCatTest,EmptyDest)12*9e9803bfSMichael Jones TEST(LlvmLibcStrNCatTest, EmptyDest) {
13*9e9803bfSMichael Jones const char *abc = "abc";
14*9e9803bfSMichael Jones char dest[4];
15*9e9803bfSMichael Jones
16*9e9803bfSMichael Jones dest[0] = '\0';
17*9e9803bfSMichael Jones
18*9e9803bfSMichael Jones // Start by copying nothing
19*9e9803bfSMichael Jones char *result = __llvm_libc::strncat(dest, abc, 0);
20*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
21*9e9803bfSMichael Jones ASSERT_EQ(dest[0], '\0');
22*9e9803bfSMichael Jones
23*9e9803bfSMichael Jones // Then copy part of it.
24*9e9803bfSMichael Jones result = __llvm_libc::strncat(dest, abc, 1);
25*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
26*9e9803bfSMichael Jones ASSERT_STREQ(dest, "a");
27*9e9803bfSMichael Jones
28*9e9803bfSMichael Jones // Reset for the last test.
29*9e9803bfSMichael Jones dest[0] = '\0';
30*9e9803bfSMichael Jones
31*9e9803bfSMichael Jones // Then copy all of it.
32*9e9803bfSMichael Jones result = __llvm_libc::strncat(dest, abc, 3);
33*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
34*9e9803bfSMichael Jones ASSERT_STREQ(dest, result);
35*9e9803bfSMichael Jones ASSERT_STREQ(dest, abc);
36*9e9803bfSMichael Jones }
37*9e9803bfSMichael Jones
TEST(LlvmLibcStrNCatTest,NonEmptyDest)38*9e9803bfSMichael Jones TEST(LlvmLibcStrNCatTest, NonEmptyDest) {
39*9e9803bfSMichael Jones const char *abc = "abc";
40*9e9803bfSMichael Jones char dest[7];
41*9e9803bfSMichael Jones
42*9e9803bfSMichael Jones dest[0] = 'x';
43*9e9803bfSMichael Jones dest[1] = 'y';
44*9e9803bfSMichael Jones dest[2] = 'z';
45*9e9803bfSMichael Jones dest[3] = '\0';
46*9e9803bfSMichael Jones
47*9e9803bfSMichael Jones // Copy only part of the string onto the end
48*9e9803bfSMichael Jones char *result = __llvm_libc::strncat(dest, abc, 1);
49*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
50*9e9803bfSMichael Jones ASSERT_STREQ(dest, "xyza");
51*9e9803bfSMichael Jones
52*9e9803bfSMichael Jones // Copy a bit more, but without resetting.
53*9e9803bfSMichael Jones result = __llvm_libc::strncat(dest, abc, 2);
54*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
55*9e9803bfSMichael Jones ASSERT_STREQ(dest, "xyzaab");
56*9e9803bfSMichael Jones
57*9e9803bfSMichael Jones // Set just the end marker, to make sure it overwrites properly.
58*9e9803bfSMichael Jones dest[3] = '\0';
59*9e9803bfSMichael Jones
60*9e9803bfSMichael Jones result = __llvm_libc::strncat(dest, abc, 3);
61*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
62*9e9803bfSMichael Jones ASSERT_STREQ(dest, "xyzabc");
63*9e9803bfSMichael Jones
64*9e9803bfSMichael Jones // Check that copying still works when count > src length
65*9e9803bfSMichael Jones dest[0] = '\0';
66*9e9803bfSMichael Jones // And that it doesn't write beyond what is necessary.
67*9e9803bfSMichael Jones dest[4] = 'Z';
68*9e9803bfSMichael Jones result = __llvm_libc::strncat(dest, abc, 4);
69*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
70*9e9803bfSMichael Jones ASSERT_STREQ(dest, "abc");
71*9e9803bfSMichael Jones ASSERT_EQ(dest[4], 'Z');
72*9e9803bfSMichael Jones
73*9e9803bfSMichael Jones result = __llvm_libc::strncat(dest, abc, 5);
74*9e9803bfSMichael Jones ASSERT_EQ(dest, result);
75*9e9803bfSMichael Jones ASSERT_STREQ(dest, "abcabc");
76*9e9803bfSMichael Jones }
77