105 lines
3.8 KiB
C++
105 lines
3.8 KiB
C++
|
// Copyright 2014 The Crashpad Authors. All rights reserved.
|
|||
|
//
|
|||
|
// 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 <string>
|
|||
|
|
|||
|
#include "base/cxx17_backports.h"
|
|||
|
#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 < base::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(),
|
|||
|
base::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, base::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,
|
|||
|
base::size(destination.lead_guard)) == 0);
|
|||
|
size_t expected_untouched_length =
|
|||
|
base::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,
|
|||
|
base::size(destination.trail_guard)) == 0);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
} // namespace
|
|||
|
} // namespace test
|
|||
|
} // namespace crashpad
|