| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/strings/stringprintf.h" | |
| 6 | |
| 7 #include <errno.h> | |
| 8 | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "base/scoped_clear_errno.h" | |
| 12 #include "base/strings/string_util.h" | |
| 13 #include "base/strings/utf_string_conversions.h" | |
| 14 | |
| 15 namespace base { | |
| 16 | |
| 17 namespace { | |
| 18 | |
| 19 // Overloaded wrappers around vsnprintf and vswprintf. The buf_size parameter | |
| 20 // is the size of the buffer. These return the number of characters in the | |
| 21 // formatted string excluding the NUL terminator. If the buffer is not | |
| 22 // large enough to accommodate the formatted string without truncation, they | |
| 23 // return the number of characters that would be in the fully-formatted string | |
| 24 // (vsnprintf, and vswprintf on Windows), or -1 (vswprintf on POSIX platforms). | |
| 25 inline int vsnprintfT(char* buffer, | |
| 26 size_t buf_size, | |
| 27 const char* format, | |
| 28 va_list argptr) { | |
| 29 return base::vsnprintf(buffer, buf_size, format, argptr); | |
| 30 } | |
| 31 | |
| 32 #if defined(OS_WIN) | |
| 33 inline int vsnprintfT(wchar_t* buffer, | |
| 34 size_t buf_size, | |
| 35 const wchar_t* format, | |
| 36 va_list argptr) { | |
| 37 return base::vswprintf(buffer, buf_size, format, argptr); | |
| 38 } | |
| 39 #endif | |
| 40 | |
| 41 // Templatized backend for StringPrintF/StringAppendF. This does not finalize | |
| 42 // the va_list, the caller is expected to do that. | |
| 43 template <class StringType> | |
| 44 static void StringAppendVT(StringType* dst, | |
| 45 const typename StringType::value_type* format, | |
| 46 va_list ap) { | |
| 47 // First try with a small fixed size buffer. | |
| 48 // This buffer size should be kept in sync with StringUtilTest.GrowBoundary | |
| 49 // and StringUtilTest.StringPrintfBounds. | |
| 50 typename StringType::value_type stack_buf[1024]; | |
| 51 | |
| 52 va_list ap_copy; | |
| 53 va_copy(ap_copy, ap); | |
| 54 | |
| 55 #if !defined(OS_WIN) | |
| 56 ScopedClearErrno clear_errno; | |
| 57 #endif | |
| 58 int result = vsnprintfT(stack_buf, arraysize(stack_buf), format, ap_copy); | |
| 59 va_end(ap_copy); | |
| 60 | |
| 61 if (result >= 0 && result < static_cast<int>(arraysize(stack_buf))) { | |
| 62 // It fit. | |
| 63 dst->append(stack_buf, result); | |
| 64 return; | |
| 65 } | |
| 66 | |
| 67 // Repeatedly increase buffer size until it fits. | |
| 68 int mem_length = arraysize(stack_buf); | |
| 69 while (true) { | |
| 70 if (result < 0) { | |
| 71 #if defined(OS_WIN) | |
| 72 // On Windows, vsnprintfT always returns the number of characters in a | |
| 73 // fully-formatted string, so if we reach this point, something else is | |
| 74 // wrong and no amount of buffer-doubling is going to fix it. | |
| 75 return; | |
| 76 #else | |
| 77 if (errno != 0 && errno != EOVERFLOW) | |
| 78 return; | |
| 79 // Try doubling the buffer size. | |
| 80 mem_length *= 2; | |
| 81 #endif | |
| 82 } else { | |
| 83 // We need exactly "result + 1" characters. | |
| 84 mem_length = result + 1; | |
| 85 } | |
| 86 | |
| 87 if (mem_length > 32 * 1024 * 1024) { | |
| 88 // That should be plenty, don't try anything larger. This protects | |
| 89 // against huge allocations when using vsnprintfT implementations that | |
| 90 // return -1 for reasons other than overflow without setting errno. | |
| 91 DLOG(WARNING) << "Unable to printf the requested string due to size."; | |
| 92 return; | |
| 93 } | |
| 94 | |
| 95 std::vector<typename StringType::value_type> mem_buf(mem_length); | |
| 96 | |
| 97 // NOTE: You can only use a va_list once. Since we're in a while loop, we | |
| 98 // need to make a new copy each time so we don't use up the original. | |
| 99 va_copy(ap_copy, ap); | |
| 100 result = vsnprintfT(&mem_buf[0], mem_length, format, ap_copy); | |
| 101 va_end(ap_copy); | |
| 102 | |
| 103 if ((result >= 0) && (result < mem_length)) { | |
| 104 // It fit. | |
| 105 dst->append(&mem_buf[0], result); | |
| 106 return; | |
| 107 } | |
| 108 } | |
| 109 } | |
| 110 | |
| 111 } // namespace | |
| 112 | |
| 113 std::string StringPrintf(const char* format, ...) { | |
| 114 va_list ap; | |
| 115 va_start(ap, format); | |
| 116 std::string result; | |
| 117 StringAppendV(&result, format, ap); | |
| 118 va_end(ap); | |
| 119 return result; | |
| 120 } | |
| 121 | |
| 122 #if defined(OS_WIN) | |
| 123 std::wstring StringPrintf(const wchar_t* format, ...) { | |
| 124 va_list ap; | |
| 125 va_start(ap, format); | |
| 126 std::wstring result; | |
| 127 StringAppendV(&result, format, ap); | |
| 128 va_end(ap); | |
| 129 return result; | |
| 130 } | |
| 131 #endif | |
| 132 | |
| 133 std::string StringPrintV(const char* format, va_list ap) { | |
| 134 std::string result; | |
| 135 StringAppendV(&result, format, ap); | |
| 136 return result; | |
| 137 } | |
| 138 | |
| 139 const std::string& SStringPrintf(std::string* dst, const char* format, ...) { | |
| 140 va_list ap; | |
| 141 va_start(ap, format); | |
| 142 dst->clear(); | |
| 143 StringAppendV(dst, format, ap); | |
| 144 va_end(ap); | |
| 145 return *dst; | |
| 146 } | |
| 147 | |
| 148 #if defined(OS_WIN) | |
| 149 const std::wstring& SStringPrintf(std::wstring* dst, | |
| 150 const wchar_t* format, ...) { | |
| 151 va_list ap; | |
| 152 va_start(ap, format); | |
| 153 dst->clear(); | |
| 154 StringAppendV(dst, format, ap); | |
| 155 va_end(ap); | |
| 156 return *dst; | |
| 157 } | |
| 158 #endif | |
| 159 | |
| 160 void StringAppendF(std::string* dst, const char* format, ...) { | |
| 161 va_list ap; | |
| 162 va_start(ap, format); | |
| 163 StringAppendV(dst, format, ap); | |
| 164 va_end(ap); | |
| 165 } | |
| 166 | |
| 167 #if defined(OS_WIN) | |
| 168 void StringAppendF(std::wstring* dst, const wchar_t* format, ...) { | |
| 169 va_list ap; | |
| 170 va_start(ap, format); | |
| 171 StringAppendV(dst, format, ap); | |
| 172 va_end(ap); | |
| 173 } | |
| 174 #endif | |
| 175 | |
| 176 void StringAppendV(std::string* dst, const char* format, va_list ap) { | |
| 177 StringAppendVT(dst, format, ap); | |
| 178 } | |
| 179 | |
| 180 #if defined(OS_WIN) | |
| 181 void StringAppendV(std::wstring* dst, const wchar_t* format, va_list ap) { | |
| 182 StringAppendVT(dst, format, ap); | |
| 183 } | |
| 184 #endif | |
| 185 | |
| 186 } // namespace base | |
| OLD | NEW |