19e9803bfSMichael Jones //===-- Unittests for strncat ---------------------------------------------===//
29e9803bfSMichael Jones //
39e9803bfSMichael Jones // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
49e9803bfSMichael Jones // See https://llvm.org/LICENSE.txt for license information.
59e9803bfSMichael Jones // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
69e9803bfSMichael Jones //
79e9803bfSMichael Jones //===----------------------------------------------------------------------===//
89e9803bfSMichael Jones
99e9803bfSMichael Jones #include "src/string/strncat.h"
10af1315c2SSiva Chandra Reddy #include "test/UnitTest/Test.h"
119e9803bfSMichael Jones
TEST(LlvmLibcStrNCatTest,EmptyDest)129e9803bfSMichael Jones TEST(LlvmLibcStrNCatTest, EmptyDest) {
139e9803bfSMichael Jones const char *abc = "abc";
149e9803bfSMichael Jones char dest[4];
159e9803bfSMichael Jones
169e9803bfSMichael Jones dest[0] = '\0';
179e9803bfSMichael Jones
189e9803bfSMichael Jones // Start by copying nothing
19*b6bc9d72SGuillaume Chatelet char *result = LIBC_NAMESPACE::strncat(dest, abc, 0);
209e9803bfSMichael Jones ASSERT_EQ(dest, result);
219e9803bfSMichael Jones ASSERT_EQ(dest[0], '\0');
229e9803bfSMichael Jones
239e9803bfSMichael Jones // Then copy part of it.
24*b6bc9d72SGuillaume Chatelet result = LIBC_NAMESPACE::strncat(dest, abc, 1);
259e9803bfSMichael Jones ASSERT_EQ(dest, result);
269e9803bfSMichael Jones ASSERT_STREQ(dest, "a");
279e9803bfSMichael Jones
289e9803bfSMichael Jones // Reset for the last test.
299e9803bfSMichael Jones dest[0] = '\0';
309e9803bfSMichael Jones
319e9803bfSMichael Jones // Then copy all of it.
32*b6bc9d72SGuillaume Chatelet result = LIBC_NAMESPACE::strncat(dest, abc, 3);
339e9803bfSMichael Jones ASSERT_EQ(dest, result);
349e9803bfSMichael Jones ASSERT_STREQ(dest, result);
359e9803bfSMichael Jones ASSERT_STREQ(dest, abc);
369e9803bfSMichael Jones }
379e9803bfSMichael Jones
TEST(LlvmLibcStrNCatTest,NonEmptyDest)389e9803bfSMichael Jones TEST(LlvmLibcStrNCatTest, NonEmptyDest) {
399e9803bfSMichael Jones const char *abc = "abc";
409e9803bfSMichael Jones char dest[7];
419e9803bfSMichael Jones
429e9803bfSMichael Jones dest[0] = 'x';
439e9803bfSMichael Jones dest[1] = 'y';
449e9803bfSMichael Jones dest[2] = 'z';
459e9803bfSMichael Jones dest[3] = '\0';
469e9803bfSMichael Jones
479e9803bfSMichael Jones // Copy only part of the string onto the end
48*b6bc9d72SGuillaume Chatelet char *result = LIBC_NAMESPACE::strncat(dest, abc, 1);
499e9803bfSMichael Jones ASSERT_EQ(dest, result);
509e9803bfSMichael Jones ASSERT_STREQ(dest, "xyza");
519e9803bfSMichael Jones
529e9803bfSMichael Jones // Copy a bit more, but without resetting.
53*b6bc9d72SGuillaume Chatelet result = LIBC_NAMESPACE::strncat(dest, abc, 2);
549e9803bfSMichael Jones ASSERT_EQ(dest, result);
559e9803bfSMichael Jones ASSERT_STREQ(dest, "xyzaab");
569e9803bfSMichael Jones
579e9803bfSMichael Jones // Set just the end marker, to make sure it overwrites properly.
589e9803bfSMichael Jones dest[3] = '\0';
599e9803bfSMichael Jones
60*b6bc9d72SGuillaume Chatelet result = LIBC_NAMESPACE::strncat(dest, abc, 3);
619e9803bfSMichael Jones ASSERT_EQ(dest, result);
629e9803bfSMichael Jones ASSERT_STREQ(dest, "xyzabc");
639e9803bfSMichael Jones
649e9803bfSMichael Jones // Check that copying still works when count > src length
659e9803bfSMichael Jones dest[0] = '\0';
669e9803bfSMichael Jones // And that it doesn't write beyond what is necessary.
679e9803bfSMichael Jones dest[4] = 'Z';
68*b6bc9d72SGuillaume Chatelet result = LIBC_NAMESPACE::strncat(dest, abc, 4);
699e9803bfSMichael Jones ASSERT_EQ(dest, result);
709e9803bfSMichael Jones ASSERT_STREQ(dest, "abc");
719e9803bfSMichael Jones ASSERT_EQ(dest[4], 'Z');
729e9803bfSMichael Jones
73*b6bc9d72SGuillaume Chatelet result = LIBC_NAMESPACE::strncat(dest, abc, 5);
749e9803bfSMichael Jones ASSERT_EQ(dest, result);
759e9803bfSMichael Jones ASSERT_STREQ(dest, "abcabc");
769e9803bfSMichael Jones }
77