| 1 | //===----------------------------------------------------------------------===// |
| 2 | // |
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | |
| 9 | // <string> |
| 10 | |
| 11 | // void push_back(charT c) // constexpr since C++20 |
| 12 | |
| 13 | #include <string> |
| 14 | #include <algorithm> |
| 15 | #include <cassert> |
| 16 | |
| 17 | #include "test_macros.h" |
| 18 | #include "min_allocator.h" |
| 19 | #include "asan_testing.h" |
| 20 | |
| 21 | struct VeryLarge { |
| 22 | long long a; |
| 23 | char b; |
| 24 | }; |
| 25 | |
| 26 | template <> |
| 27 | struct std::char_traits<VeryLarge> { |
| 28 | using char_type = VeryLarge; |
| 29 | using int_type = int; |
| 30 | using off_type = streamoff; |
| 31 | using pos_type = streampos; |
| 32 | using state_type = mbstate_t; |
| 33 | |
| 34 | static TEST_CONSTEXPR_CXX20 void assign(char_type& c1, const char_type& c2) { c1 = c2; } |
| 35 | static bool eq(char_type c1, char_type c2); |
| 36 | static bool lt(char_type c1, char_type c2); |
| 37 | |
| 38 | static int compare(const char_type* s1, const char_type* s2, std::size_t n); |
| 39 | static std::size_t length(const char_type* s); |
| 40 | static const char_type* find(const char_type* s, std::size_t n, const char_type& a); |
| 41 | static char_type* move(char_type* s1, const char_type* s2, std::size_t n); |
| 42 | static TEST_CONSTEXPR_CXX20 char_type* copy(char_type* s1, const char_type* s2, std::size_t n) { |
| 43 | std::copy_n(s2, n, s1); |
| 44 | return s1; |
| 45 | } |
| 46 | static TEST_CONSTEXPR_CXX20 char_type* assign(char_type* s, std::size_t n, char_type a) { |
| 47 | std::fill_n(s, n, a); |
| 48 | return s; |
| 49 | } |
| 50 | |
| 51 | static int_type not_eof(int_type c); |
| 52 | static char_type to_char_type(int_type c); |
| 53 | static int_type to_int_type(char_type c); |
| 54 | static bool eq_int_type(int_type c1, int_type c2); |
| 55 | static int_type eof(); |
| 56 | }; |
| 57 | |
| 58 | template <class S> |
| 59 | TEST_CONSTEXPR_CXX20 void test(S s, typename S::value_type c, S expected) { |
| 60 | s.push_back(c); |
| 61 | LIBCPP_ASSERT(s.__invariants()); |
| 62 | assert(s == expected); |
| 63 | LIBCPP_ASSERT(is_string_asan_correct(s)); |
| 64 | } |
| 65 | |
| 66 | template <class S> |
| 67 | TEST_CONSTEXPR_CXX20 void test_string() { |
| 68 | test(S(), 'a', S(1, 'a')); |
| 69 | test(S("12345" ), 'a', S("12345a" )); |
| 70 | test(S("12345678901234567890" ), 'a', S("12345678901234567890a" )); |
| 71 | test(S("123abcabcdefghabcdefgh" ), 'a', S("123abcabcdefghabcdefgha" )); |
| 72 | test(S("123abcabcdefghabcdefgha" ), 'b', S("123abcabcdefghabcdefghab" )); |
| 73 | test(S("123abcabcdefghabcdefghab" ), 'c', S("123abcabcdefghabcdefghabc" )); |
| 74 | test(S("123abcabcdefghabcdefghabc" ), 'd', S("123abcabcdefghabcdefghabcd" )); |
| 75 | } |
| 76 | |
| 77 | TEST_CONSTEXPR_CXX20 bool test() { |
| 78 | test_string<std::string>(); |
| 79 | #if TEST_STD_VER >= 11 |
| 80 | test_string<std::basic_string<char, std::char_traits<char>, min_allocator<char> > >(); |
| 81 | test_string<std::basic_string<char, std::char_traits<char>, safe_allocator<char> > >(); |
| 82 | #endif |
| 83 | { |
| 84 | // https://llvm.org/PR31454 |
| 85 | std::basic_string<VeryLarge> s; |
| 86 | VeryLarge vl = {}; |
| 87 | s.push_back(c: vl); |
| 88 | s.push_back(c: vl); |
| 89 | s.push_back(c: vl); |
| 90 | } |
| 91 | |
| 92 | return true; |
| 93 | } |
| 94 | |
| 95 | int main(int, char**) { |
| 96 | test(); |
| 97 | #if TEST_STD_VER > 17 |
| 98 | static_assert(test()); |
| 99 | #endif |
| 100 | |
| 101 | return 0; |
| 102 | } |
| 103 | |