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 _LIBCPP___LOCALE_DIR_SUPPORT_LINUX_H
10#define _LIBCPP___LOCALE_DIR_SUPPORT_LINUX_H
11
12#include <__config>
13#include <__cstddef/size_t.h>
14#include <__std_mbstate_t.h>
15#include <__utility/forward.h>
16#include <clocale> // std::lconv
17#include <cstdio>
18#include <cstdlib>
19#include <ctype.h>
20#include <langinfo.h>
21#include <stdarg.h>
22#include <string.h>
23#include <time.h>
24#if _LIBCPP_HAS_WIDE_CHARACTERS
25# include <cwchar>
26# include <wctype.h>
27#endif
28
29#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
30# pragma GCC system_header
31#endif
32
33_LIBCPP_BEGIN_NAMESPACE_STD
34namespace __locale {
35
36struct __locale_guard {
37 _LIBCPP_HIDE_FROM_ABI __locale_guard(locale_t& __loc) : __old_loc_(::uselocale(dataset: __loc)) {}
38
39 _LIBCPP_HIDE_FROM_ABI ~__locale_guard() {
40 if (__old_loc_)
41 ::uselocale(dataset: __old_loc_);
42 }
43
44 locale_t __old_loc_;
45
46 __locale_guard(__locale_guard const&) = delete;
47 __locale_guard& operator=(__locale_guard const&) = delete;
48};
49
50//
51// Locale management
52//
53#define _LIBCPP_COLLATE_MASK LC_COLLATE_MASK
54#define _LIBCPP_CTYPE_MASK LC_CTYPE_MASK
55#define _LIBCPP_MONETARY_MASK LC_MONETARY_MASK
56#define _LIBCPP_NUMERIC_MASK LC_NUMERIC_MASK
57#define _LIBCPP_TIME_MASK LC_TIME_MASK
58#define _LIBCPP_MESSAGES_MASK LC_MESSAGES_MASK
59#define _LIBCPP_ALL_MASK LC_ALL_MASK
60#define _LIBCPP_LC_ALL LC_ALL
61
62using __locale_t _LIBCPP_NODEBUG = ::locale_t;
63
64#if defined(_LIBCPP_BUILDING_LIBRARY)
65using __lconv_t _LIBCPP_NODEBUG = std::lconv;
66
67inline _LIBCPP_HIDE_FROM_ABI __locale_t __newlocale(int __category_mask, const char* __locale, __locale_t __base) {
68 return ::newlocale(__category_mask, __locale, __base);
69}
70
71inline _LIBCPP_HIDE_FROM_ABI void __freelocale(__locale_t __loc) { ::freelocale(dataset: __loc); }
72
73inline _LIBCPP_HIDE_FROM_ABI char* __setlocale(int __category, char const* __locale) {
74 return ::setlocale(__category, __locale);
75}
76
77inline _LIBCPP_HIDE_FROM_ABI __lconv_t* __localeconv(__locale_t& __loc) {
78 __locale_guard __current(__loc);
79 return std::localeconv();
80}
81#endif // _LIBCPP_BUILDING_LIBRARY
82
83//
84// Strtonum functions
85//
86inline _LIBCPP_HIDE_FROM_ABI float __strtof(const char* __nptr, char** __endptr, __locale_t __loc) {
87 return ::strtof_l(__nptr, __endptr, __loc);
88}
89
90inline _LIBCPP_HIDE_FROM_ABI double __strtod(const char* __nptr, char** __endptr, __locale_t __loc) {
91 return ::strtod_l(__nptr, __endptr, __loc);
92}
93
94inline _LIBCPP_HIDE_FROM_ABI long double __strtold(const char* __nptr, char** __endptr, __locale_t __loc) {
95 return ::strtold_l(__nptr, __endptr, __loc);
96}
97
98//
99// Character manipulation functions
100//
101#if defined(_LIBCPP_BUILDING_LIBRARY)
102inline _LIBCPP_HIDE_FROM_ABI int __toupper(int __c, __locale_t __loc) { return toupper_l(__c, l: __loc); }
103
104inline _LIBCPP_HIDE_FROM_ABI int __tolower(int __c, __locale_t __loc) { return tolower_l(__c, l: __loc); }
105
106inline _LIBCPP_HIDE_FROM_ABI int __strcoll(const char* __s1, const char* __s2, __locale_t __loc) {
107 return strcoll_l(__s1, __s2, l: __loc);
108}
109
110inline _LIBCPP_HIDE_FROM_ABI size_t __strxfrm(char* __dest, const char* __src, size_t __n, __locale_t __loc) {
111 return strxfrm_l(__dest, __src, __n, l: __loc);
112}
113
114# if _LIBCPP_HAS_WIDE_CHARACTERS
115inline _LIBCPP_HIDE_FROM_ABI int __iswctype(wint_t __c, wctype_t __type, __locale_t __loc) {
116 return iswctype_l(wc: __c, desc: __type, locale: __loc);
117}
118
119inline _LIBCPP_HIDE_FROM_ABI int __iswspace(wint_t __c, __locale_t __loc) { return iswspace_l(wc: __c, locale: __loc); }
120
121inline _LIBCPP_HIDE_FROM_ABI int __iswprint(wint_t __c, __locale_t __loc) { return iswprint_l(wc: __c, locale: __loc); }
122
123inline _LIBCPP_HIDE_FROM_ABI int __iswcntrl(wint_t __c, __locale_t __loc) { return iswcntrl_l(wc: __c, locale: __loc); }
124
125inline _LIBCPP_HIDE_FROM_ABI int __iswupper(wint_t __c, __locale_t __loc) { return iswupper_l(wc: __c, locale: __loc); }
126
127inline _LIBCPP_HIDE_FROM_ABI int __iswlower(wint_t __c, __locale_t __loc) { return iswlower_l(wc: __c, locale: __loc); }
128
129inline _LIBCPP_HIDE_FROM_ABI int __iswalpha(wint_t __c, __locale_t __loc) { return iswalpha_l(wc: __c, locale: __loc); }
130
131inline _LIBCPP_HIDE_FROM_ABI int __iswblank(wint_t __c, __locale_t __loc) { return iswblank_l(wc: __c, locale: __loc); }
132
133inline _LIBCPP_HIDE_FROM_ABI int __iswdigit(wint_t __c, __locale_t __loc) { return iswdigit_l(wc: __c, locale: __loc); }
134
135inline _LIBCPP_HIDE_FROM_ABI int __iswpunct(wint_t __c, __locale_t __loc) { return iswpunct_l(wc: __c, locale: __loc); }
136
137inline _LIBCPP_HIDE_FROM_ABI int __iswxdigit(wint_t __c, __locale_t __loc) { return iswxdigit_l(wc: __c, locale: __loc); }
138
139inline _LIBCPP_HIDE_FROM_ABI wint_t __towupper(wint_t __c, __locale_t __loc) { return towupper_l(wc: __c, locale: __loc); }
140
141inline _LIBCPP_HIDE_FROM_ABI wint_t __towlower(wint_t __c, __locale_t __loc) { return towlower_l(wc: __c, locale: __loc); }
142
143inline _LIBCPP_HIDE_FROM_ABI int __wcscoll(const wchar_t* __ws1, const wchar_t* __ws2, __locale_t __loc) {
144 return wcscoll_l(s1: __ws1, s2: __ws2, __loc);
145}
146
147inline _LIBCPP_HIDE_FROM_ABI size_t __wcsxfrm(wchar_t* __dest, const wchar_t* __src, size_t __n, __locale_t __loc) {
148 return wcsxfrm_l(s1: __dest, s2: __src, __n, __loc);
149}
150# endif // _LIBCPP_HAS_WIDE_CHARACTERS
151
152inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_ATTRIBUTE_FORMAT(__strftime__, 3, 0) size_t
153 __strftime(char* __s, size_t __max, const char* __format, const struct tm* __tm, __locale_t __loc) {
154 return strftime_l(__s, maxsize: __max, __format, tp: __tm, __loc);
155}
156
157//
158// Other functions
159//
160inline _LIBCPP_HIDE_FROM_ABI decltype(MB_CUR_MAX) __mb_len_max(__locale_t __loc) {
161 __locale_guard __current(__loc);
162 return MB_CUR_MAX;
163}
164
165# if _LIBCPP_HAS_WIDE_CHARACTERS
166inline _LIBCPP_HIDE_FROM_ABI wint_t __btowc(int __c, __locale_t __loc) {
167 __locale_guard __current(__loc);
168 return std::btowc(__c);
169}
170
171inline _LIBCPP_HIDE_FROM_ABI int __wctob(wint_t __c, __locale_t __loc) {
172 __locale_guard __current(__loc);
173 return std::wctob(wc: __c);
174}
175
176inline _LIBCPP_HIDE_FROM_ABI size_t
177__wcsnrtombs(char* __dest, const wchar_t** __src, size_t __nwc, size_t __len, mbstate_t* __ps, __locale_t __loc) {
178 __locale_guard __current(__loc);
179 return ::wcsnrtombs(dst: __dest, __src, __nwc, __len, __ps); // non-standard
180}
181
182inline _LIBCPP_HIDE_FROM_ABI size_t __wcrtomb(char* __s, wchar_t __wc, mbstate_t* __ps, __locale_t __loc) {
183 __locale_guard __current(__loc);
184 return std::wcrtomb(__s, __wc, __ps);
185}
186
187inline _LIBCPP_HIDE_FROM_ABI size_t
188__mbsnrtowcs(wchar_t* __dest, const char** __src, size_t __nms, size_t __len, mbstate_t* __ps, __locale_t __loc) {
189 __locale_guard __current(__loc);
190 return ::mbsnrtowcs(dst: __dest, __src, nmc: __nms, __len, __ps); // non-standard
191}
192
193inline _LIBCPP_HIDE_FROM_ABI size_t
194__mbrtowc(wchar_t* __pwc, const char* __s, size_t __n, mbstate_t* __ps, __locale_t __loc) {
195 __locale_guard __current(__loc);
196 return std::mbrtowc(__pwc, __s, __n, p: __ps);
197}
198
199inline _LIBCPP_HIDE_FROM_ABI int __mbtowc(wchar_t* __pwc, const char* __pmb, size_t __max, __locale_t __loc) {
200 __locale_guard __current(__loc);
201 return std::mbtowc(__pwc, s: __pmb, n: __max);
202}
203
204inline _LIBCPP_HIDE_FROM_ABI size_t __mbrlen(const char* __s, size_t __n, mbstate_t* __ps, __locale_t __loc) {
205 __locale_guard __current(__loc);
206 return std::mbrlen(__s, __n, __ps);
207}
208
209inline _LIBCPP_HIDE_FROM_ABI size_t
210__mbsrtowcs(wchar_t* __dest, const char** __src, size_t __len, mbstate_t* __ps, __locale_t __loc) {
211 __locale_guard __current(__loc);
212 return std::mbsrtowcs(dst: __dest, __src, __len, __ps);
213}
214# endif // _LIBCPP_HAS_WIDE_CHARACTERS
215
216inline _LIBCPP_HIDE_FROM_ABI const char* __get_locale_encoding([[maybe_unused]] __locale_t __loc) {
217# if defined(__ANDROID__)
218 return "UTF-8";
219# else
220 return ::nl_langinfo_l(CODESET, l: __loc);
221# endif
222}
223#endif // _LIBCPP_BUILDING_LIBRARY
224
225#ifndef _LIBCPP_COMPILER_GCC // GCC complains that this can't be always_inline due to C-style varargs
226_LIBCPP_HIDE_FROM_ABI
227#endif
228inline _LIBCPP_ATTRIBUTE_FORMAT(__printf__, 4, 5) int __snprintf(
229 char* __s, size_t __n, __locale_t __loc, const char* __format, ...) {
230 va_list __va;
231 va_start(__va, __format);
232 __locale_guard __current(__loc);
233 int __res = std::vsnprintf(__s, maxlen: __n, __format, arg: __va);
234 va_end(__va);
235 return __res;
236}
237
238#ifndef _LIBCPP_COMPILER_GCC // GCC complains that this can't be always_inline due to C-style varargs
239_LIBCPP_HIDE_FROM_ABI
240#endif
241inline _LIBCPP_ATTRIBUTE_FORMAT(__printf__, 3, 4) int __asprintf(
242 char** __s, __locale_t __loc, const char* __format, ...) {
243 va_list __va;
244 va_start(__va, __format);
245 __locale_guard __current(__loc);
246 int __res = ::vasprintf(ptr: __s, f: __format, arg: __va); // non-standard
247 va_end(__va);
248 return __res;
249}
250} // namespace __locale
251_LIBCPP_END_NAMESPACE_STD
252
253#endif // _LIBCPP___LOCALE_DIR_SUPPORT_LINUX_H
254