| Avi Drissman | e4622aa | 2022-09-08 20:36:06 | [diff] [blame] | 1 | // Copyright 2013 The Chromium Authors |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| danakj | 51d26a40 | 2024-04-25 14:23:56 | [diff] [blame] | 5 | #ifdef UNSAFE_BUFFERS_BUILD |
| 6 | // TODO(crbug.com/40284755): Remove this and spanify to fix the errors. |
| 7 | #pragma allow_unsafe_buffers |
| 8 | #endif |
| 9 | |
| [email protected] | 8ad97ad | 2013-06-08 06:05:47 | [diff] [blame] | 10 | #include "base/strings/stringprintf.h" |
| 11 | |
| [email protected] | 34d58a2 | 2013-05-02 23:06:26 | [diff] [blame] | 12 | #include <errno.h> |
| avi | 84f37e1 | 2015-12-25 09:31:42 | [diff] [blame] | 13 | #include <stddef.h> |
| [email protected] | 34d58a2 | 2013-05-02 23:06:26 | [diff] [blame] | 14 | |
| avi | 84f37e1 | 2015-12-25 09:31:42 | [diff] [blame] | 15 | #include "build/build_config.h" |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 16 | #include "testing/gtest/include/gtest/gtest.h" |
| 17 | |
| 18 | namespace base { |
| 19 | |
| 20 | namespace { |
| 21 | |
| 22 | // A helper for the StringAppendV test that follows. |
| 23 | // |
| 24 | // Just forwards its args to StringAppendV. |
| Jan Wilken Dörrie | 8ef22c7 | 2019-09-20 05:48:19 | [diff] [blame] | 25 | template <class CharT> |
| 26 | static void StringAppendVTestHelper(std::basic_string<CharT>* out, |
| 27 | const CharT* format, |
| 28 | ...) { |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 29 | va_list ap; |
| 30 | va_start(ap, format); |
| 31 | StringAppendV(out, format, ap); |
| 32 | va_end(ap); |
| 33 | } |
| 34 | |
| 35 | } // namespace |
| 36 | |
| 37 | TEST(StringPrintfTest, StringPrintfEmpty) { |
| 38 | EXPECT_EQ("", StringPrintf("%s", "")); |
| 39 | } |
| 40 | |
| 41 | TEST(StringPrintfTest, StringPrintfMisc) { |
| 42 | EXPECT_EQ("123hello w", StringPrintf("%3d%2s %1c", 123, "hello", 'w')); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 43 | } |
| 44 | |
| 45 | TEST(StringPrintfTest, StringAppendfEmptyString) { |
| 46 | std::string value("Hello"); |
| 47 | StringAppendF(&value, "%s", ""); |
| 48 | EXPECT_EQ("Hello", value); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 49 | } |
| 50 | |
| 51 | TEST(StringPrintfTest, StringAppendfString) { |
| 52 | std::string value("Hello"); |
| 53 | StringAppendF(&value, " %s", "World"); |
| 54 | EXPECT_EQ("Hello World", value); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 55 | } |
| 56 | |
| 57 | TEST(StringPrintfTest, StringAppendfInt) { |
| 58 | std::string value("Hello"); |
| 59 | StringAppendF(&value, " %d", 123); |
| 60 | EXPECT_EQ("Hello 123", value); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 61 | } |
| 62 | |
| 63 | // Make sure that lengths exactly around the initial buffer size are handled |
| 64 | // correctly. |
| 65 | TEST(StringPrintfTest, StringPrintfBounds) { |
| 66 | const int kSrcLen = 1026; |
| 67 | char src[kSrcLen]; |
| Jan Wilken Dörrie | 8ef22c7 | 2019-09-20 05:48:19 | [diff] [blame] | 68 | std::fill_n(src, kSrcLen, 'A'); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 69 | |
| 70 | wchar_t srcw[kSrcLen]; |
| Jan Wilken Dörrie | 8ef22c7 | 2019-09-20 05:48:19 | [diff] [blame] | 71 | std::fill_n(srcw, kSrcLen, 'A'); |
| 72 | |
| 73 | char16_t src16[kSrcLen]; |
| 74 | std::fill_n(src16, kSrcLen, 'A'); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 75 | |
| 76 | for (int i = 1; i < 3; i++) { |
| 77 | src[kSrcLen - i] = 0; |
| 78 | std::string out; |
| Peter Kasting | 638bd69 | 2023-08-15 23:56:58 | [diff] [blame] | 79 | EXPECT_EQ(src, StringPrintf("%s", src)); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 80 | } |
| 81 | } |
| 82 | |
| 83 | // Test very large sprintfs that will cause the buffer to grow. |
| 84 | TEST(StringPrintfTest, Grow) { |
| 85 | char src[1026]; |
| Peter Kasting | 134ef9af | 2024-12-28 02:30:09 | [diff] [blame] | 86 | for (auto& i : src) { |
| jdoerrie | 6c622935 | 2018-10-22 15:55:43 | [diff] [blame] | 87 | i = 'A'; |
| Peter Kasting | 134ef9af | 2024-12-28 02:30:09 | [diff] [blame] | 88 | } |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 89 | src[1025] = 0; |
| 90 | |
| thestig | 073d514d | 2014-10-21 03:11:21 | [diff] [blame] | 91 | const char fmt[] = "%sB%sB%sB%sB%sB%sB%s"; |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 92 | |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 93 | const int kRefSize = 320000; |
| 94 | char* ref = new char[kRefSize]; |
| Xiaohan Wang | 6700dcf1 | 2022-01-15 14:47:00 | [diff] [blame] | 95 | #if BUILDFLAG(IS_WIN) |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 96 | sprintf_s(ref, kRefSize, fmt, src, src, src, src, src, src, src); |
| Xiaohan Wang | 6700dcf1 | 2022-01-15 14:47:00 | [diff] [blame] | 97 | #elif BUILDFLAG(IS_POSIX) || BUILDFLAG(IS_FUCHSIA) |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 98 | snprintf(ref, kRefSize, fmt, src, src, src, src, src, src, src); |
| 99 | #endif |
| 100 | |
| Peter Kasting | 638bd69 | 2023-08-15 23:56:58 | [diff] [blame] | 101 | EXPECT_EQ(ref, StringPrintf(fmt, src, src, src, src, src, src, src)); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 102 | delete[] ref; |
| 103 | } |
| 104 | |
| 105 | TEST(StringPrintfTest, StringAppendV) { |
| 106 | std::string out; |
| 107 | StringAppendVTestHelper(&out, "%d foo %s", 1, "bar"); |
| 108 | EXPECT_EQ("1 foo bar", out); |
| 109 | } |
| 110 | |
| 111 | // Test the boundary condition for the size of the string_util's |
| 112 | // internal buffer. |
| 113 | TEST(StringPrintfTest, GrowBoundary) { |
| thestig | 073d514d | 2014-10-21 03:11:21 | [diff] [blame] | 114 | const int kStringUtilBufLen = 1024; |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 115 | // Our buffer should be one larger than the size of StringAppendVT's stack |
| 116 | // buffer. |
| thestig | 073d514d | 2014-10-21 03:11:21 | [diff] [blame] | 117 | // And need extra one for NULL-terminator. |
| 118 | const int kBufLen = kStringUtilBufLen + 1 + 1; |
| 119 | char src[kBufLen]; |
| Peter Kasting | 134ef9af | 2024-12-28 02:30:09 | [diff] [blame] | 120 | for (int i = 0; i < kBufLen - 1; ++i) { |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 121 | src[i] = 'a'; |
| Peter Kasting | 134ef9af | 2024-12-28 02:30:09 | [diff] [blame] | 122 | } |
| thestig | 073d514d | 2014-10-21 03:11:21 | [diff] [blame] | 123 | src[kBufLen - 1] = 0; |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 124 | |
| Peter Kasting | 638bd69 | 2023-08-15 23:56:58 | [diff] [blame] | 125 | EXPECT_EQ(src, StringPrintf("%s", src)); |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 126 | } |
| 127 | |
| [email protected] | 34d58a2 | 2013-05-02 23:06:26 | [diff] [blame] | 128 | // Test that StringPrintf and StringAppendV do not change errno. |
| 129 | TEST(StringPrintfTest, StringPrintfErrno) { |
| 130 | errno = 1; |
| 131 | EXPECT_EQ("", StringPrintf("%s", "")); |
| 132 | EXPECT_EQ(1, errno); |
| 133 | std::string out; |
| 134 | StringAppendVTestHelper(&out, "%d foo %s", 1, "bar"); |
| 135 | EXPECT_EQ(1, errno); |
| 136 | } |
| 137 | |
| [email protected] | e6811ed5 | 2010-08-17 03:45:37 | [diff] [blame] | 138 | } // namespace base |