105 lines
3.8 KiB
C++
105 lines
3.8 KiB
C++
// Copyright 2014 The Crashpad Authors
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
|
||
#include "util/stdlib/strlcpy.h"
|
||
|
||
#include <string.h>
|
||
#include <wchar.h>
|
||
|
||
#include <algorithm>
|
||
#include <iterator>
|
||
#include <string>
|
||
|
||
#include "base/format_macros.h"
|
||
#include "base/strings/stringprintf.h"
|
||
#include "gtest/gtest.h"
|
||
|
||
namespace crashpad {
|
||
namespace test {
|
||
namespace {
|
||
|
||
size_t C16Len(const char16_t* s) {
|
||
return std::char_traits<char16_t>::length(s);
|
||
}
|
||
|
||
int C16Memcmp(const char16_t* s1, const char16_t* s2, size_t n) {
|
||
return std::char_traits<char16_t>::compare(s1, s2, n);
|
||
}
|
||
|
||
TEST(strlcpy, c16lcpy) {
|
||
// Use a destination buffer that’s larger than the length passed to c16lcpy.
|
||
// The unused portion is a guard area that must not be written to.
|
||
struct TestBuffer {
|
||
char16_t lead_guard[64];
|
||
char16_t data[128];
|
||
char16_t trail_guard[64];
|
||
};
|
||
TestBuffer expected_untouched;
|
||
memset(&expected_untouched, 0xa5, sizeof(expected_untouched));
|
||
|
||
// Test with M, é, Ā, ő, and Ḙ. This is a mix of characters that have zero and
|
||
// nonzero low and high bytes.
|
||
static constexpr char16_t test_characters[] = {
|
||
0x4d, 0xe9, 0x100, 0x151, 0x1e18};
|
||
|
||
for (size_t index = 0; index < std::size(test_characters); ++index) {
|
||
char16_t test_character = test_characters[index];
|
||
SCOPED_TRACE(base::StringPrintf(
|
||
"character index %" PRIuS ", character 0x%x", index, test_character));
|
||
for (size_t length = 0; length < 256; ++length) {
|
||
SCOPED_TRACE(
|
||
base::StringPrintf("index %" PRIuS, length));
|
||
std::u16string test_string(length, test_character);
|
||
|
||
TestBuffer destination;
|
||
memset(&destination, 0xa5, sizeof(destination));
|
||
|
||
EXPECT_EQ(c16lcpy(destination.data,
|
||
test_string.c_str(),
|
||
std::size(destination.data)),
|
||
length);
|
||
|
||
// Make sure that the destination buffer is NUL-terminated, and that as
|
||
// much of the test string was copied as could fit.
|
||
size_t expected_destination_length =
|
||
std::min(length, std::size(destination.data) - 1);
|
||
|
||
EXPECT_EQ(destination.data[expected_destination_length], '\0');
|
||
EXPECT_EQ(C16Len(destination.data), expected_destination_length);
|
||
EXPECT_TRUE(C16Memcmp(test_string.c_str(),
|
||
destination.data,
|
||
expected_destination_length) == 0);
|
||
|
||
// Make sure that the portion of the destination buffer that was not used
|
||
// was not touched. This includes the guard areas and the unused portion
|
||
// of the buffer passed to c16lcpy.
|
||
EXPECT_TRUE(C16Memcmp(expected_untouched.lead_guard,
|
||
destination.lead_guard,
|
||
std::size(destination.lead_guard)) == 0);
|
||
size_t expected_untouched_length =
|
||
std::size(destination.data) - expected_destination_length - 1;
|
||
EXPECT_TRUE(C16Memcmp(expected_untouched.data,
|
||
&destination.data[expected_destination_length + 1],
|
||
expected_untouched_length) == 0);
|
||
EXPECT_TRUE(C16Memcmp(expected_untouched.trail_guard,
|
||
destination.trail_guard,
|
||
std::size(destination.trail_guard)) == 0);
|
||
}
|
||
}
|
||
}
|
||
|
||
} // namespace
|
||
} // namespace test
|
||
} // namespace crashpad
|