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 #ifndef FILESYSTEM_FORMAT_STRING_H 10 #define FILESYSTEM_FORMAT_STRING_H 11 12 #include <__assert> 13 #include <__config> 14 #include <array> 15 #include <cstdarg> 16 #include <cstddef> 17 #include <cstdio> 18 #include <string> 19 20 #if defined(_LIBCPP_WIN32API) 21 # define PATHSTR(x) (L##x) 22 # define PATH_CSTR_FMT "\"%ls\"" 23 #else 24 # define PATHSTR(x) (x) 25 # define PATH_CSTR_FMT "\"%s\"" 26 #endif 27 28 _LIBCPP_BEGIN_NAMESPACE_FILESYSTEM 29 30 namespace detail { 31 32 inline _LIBCPP_ATTRIBUTE_FORMAT(__printf__, 1, 0) string vformat_string(const char* msg, va_list ap) { 33 array<char, 256> buf; 34 35 va_list apcopy; 36 va_copy(apcopy, ap); 37 int ret = ::vsnprintf(buf.data(), buf.size(), msg, apcopy); 38 va_end(apcopy); 39 40 string result; 41 if (static_cast<size_t>(ret) < buf.size()) { 42 result.assign(buf.data(), static_cast<size_t>(ret)); 43 } else { 44 // we did not provide a long enough buffer on our first attempt. The 45 // return value is the number of bytes (excluding the null byte) that are 46 // needed for formatting. 47 size_t size_with_null = static_cast<size_t>(ret) + 1; 48 result.__resize_default_init(size_with_null - 1); 49 ret = ::vsnprintf(&result[0], size_with_null, msg, ap); 50 _LIBCPP_ASSERT_UNCATEGORIZED(static_cast<size_t>(ret) == (size_with_null - 1), "TODO"); 51 } 52 return result; 53 } 54 55 inline _LIBCPP_ATTRIBUTE_FORMAT(__printf__, 1, 2) string format_string(const char* msg, ...) { 56 string ret; 57 va_list ap; 58 va_start(ap, msg); 59 #ifndef _LIBCPP_HAS_NO_EXCEPTIONS 60 try { 61 #endif // _LIBCPP_HAS_NO_EXCEPTIONS 62 ret = detail::vformat_string(msg, ap); 63 #ifndef _LIBCPP_HAS_NO_EXCEPTIONS 64 } catch (...) { 65 va_end(ap); 66 throw; 67 } 68 #endif // _LIBCPP_HAS_NO_EXCEPTIONS 69 va_end(ap); 70 return ret; 71 } 72 73 } // end namespace detail 74 75 _LIBCPP_END_NAMESPACE_FILESYSTEM 76 77 #endif // FILESYSTEM_FORMAT_STRING_H 78