f362aea42d
The idea is to move all pieces related to the actual libc sources to the "src" directory. This allows downstream users to ship and build just the "src" directory. Reviewed By: michaelrj Differential Revision: https://reviews.llvm.org/D112653
58 lines
1.9 KiB
C++
58 lines
1.9 KiB
C++
//===-- Unittests for strncpy ---------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "src/__support/CPP/ArrayRef.h"
|
|
#include "src/string/strncpy.h"
|
|
#include "utils/UnitTest/Test.h"
|
|
#include <stddef.h> // For size_t.
|
|
|
|
class LlvmLibcStrncpyTest : public __llvm_libc::testing::Test {
|
|
public:
|
|
void check_strncpy(__llvm_libc::cpp::MutableArrayRef<char> dst,
|
|
const __llvm_libc::cpp::ArrayRef<char> src, size_t n,
|
|
const __llvm_libc::cpp::ArrayRef<char> expected) {
|
|
// Making sure we don't overflow buffer.
|
|
ASSERT_GE(dst.size(), n);
|
|
// Making sure strncpy returns dst.
|
|
ASSERT_EQ(__llvm_libc::strncpy(dst.data(), src.data(), n), dst.data());
|
|
// Expected must be of the same size as dst.
|
|
ASSERT_EQ(dst.size(), expected.size());
|
|
// Expected and dst are the same.
|
|
for (size_t i = 0; i < expected.size(); ++i)
|
|
ASSERT_EQ(expected[i], dst[i]);
|
|
}
|
|
};
|
|
|
|
TEST_F(LlvmLibcStrncpyTest, Untouched) {
|
|
char dst[] = {'a', 'b'};
|
|
const char src[] = {'x', '\0'};
|
|
const char expected[] = {'a', 'b'};
|
|
check_strncpy(dst, src, 0, expected);
|
|
}
|
|
|
|
TEST_F(LlvmLibcStrncpyTest, CopyOne) {
|
|
char dst[] = {'a', 'b'};
|
|
const char src[] = {'x', 'y'};
|
|
const char expected[] = {'x', 'b'}; // no \0 is appended
|
|
check_strncpy(dst, src, 1, expected);
|
|
}
|
|
|
|
TEST_F(LlvmLibcStrncpyTest, CopyNull) {
|
|
char dst[] = {'a', 'b'};
|
|
const char src[] = {'\0', 'y'};
|
|
const char expected[] = {'\0', 'b'};
|
|
check_strncpy(dst, src, 1, expected);
|
|
}
|
|
|
|
TEST_F(LlvmLibcStrncpyTest, CopyPastSrc) {
|
|
char dst[] = {'a', 'b'};
|
|
const char src[] = {'\0', 'y'};
|
|
const char expected[] = {'\0', '\0'};
|
|
check_strncpy(dst, src, 2, expected);
|
|
}
|