locale_win32.cpp revision cac9e30dbedef0887a35c42dbd96d8974660a11a
1// -*- C++ -*-
2//===-------------------- support/win32/locale_win32.cpp ------------------===//
3//
4//                     The LLVM Compiler Infrastructure
5//
6// This file is dual licensed under the MIT and the University of Illinois Open
7// Source Licenses. See LICENSE.TXT for details.
8//
9//===----------------------------------------------------------------------===//
10
11#include <locale>
12#include <cstdarg> // va_start, va_end
13#include <memory>
14#include <type_traits>
15
16typedef _VSTD::remove_pointer<locale_t>::type __locale_struct;
17typedef _VSTD::unique_ptr<__locale_struct, decltype(&uselocale)> __locale_raii;
18
19// FIXME: base currently unused. Needs manual work to construct the new locale
20locale_t newlocale( int mask, const char * locale, locale_t /*base*/ )
21{
22    return _create_locale( mask, locale );
23}
24locale_t uselocale( locale_t newloc )
25{
26    locale_t old_locale = _get_current_locale();
27    if ( newloc == NULL )
28        return old_locale;
29    // uselocale sets the thread's locale by definition, so unconditionally use thread-local locale
30    _configthreadlocale( _ENABLE_PER_THREAD_LOCALE );
31    // uselocale sets all categories
32    setlocale( LC_ALL, newloc->locinfo->lc_category[LC_ALL].locale );
33    // uselocale returns the old locale_t
34    return old_locale;
35}
36lconv *localeconv_l( locale_t loc )
37{
38    __locale_raii __current( uselocale(loc), uselocale );
39    return localeconv();
40}
41size_t mbrlen_l( const char *__restrict s, size_t n,
42                 mbstate_t *__restrict ps, locale_t loc )
43{
44    __locale_raii __current( uselocale(loc), uselocale );
45    return mbrlen( s, n, ps );
46}
47size_t mbsrtowcs_l( wchar_t *__restrict dst, const char **__restrict src,
48                    size_t len, mbstate_t *__restrict ps, locale_t loc )
49{
50    __locale_raii __current( uselocale(loc), uselocale );
51    return mbsrtowcs( dst, src, len, ps );
52}
53size_t wcrtomb_l( char *__restrict s, wchar_t wc, mbstate_t *__restrict ps,
54                  locale_t loc )
55{
56    __locale_raii __current( uselocale(loc), uselocale );
57    return wcrtomb( s, wc, ps );
58}
59size_t mbrtowc_l( wchar_t *__restrict pwc, const char *__restrict s,
60                  size_t n, mbstate_t *__restrict ps, locale_t loc )
61{
62    __locale_raii __current( uselocale(loc), uselocale );
63    return mbrtowc( pwc, s, n, ps );
64}
65size_t mbsnrtowcs_l( wchar_t *__restrict dst, const char **__restrict src,
66                     size_t nms, size_t len, mbstate_t *__restrict ps, locale_t loc )
67{
68    __locale_raii __current( uselocale(loc), uselocale );
69    return mbsnrtowcs( dst, src, nms, len, ps );
70}
71size_t wcsnrtombs_l( char *__restrict dst, const wchar_t **__restrict src,
72                     size_t nwc, size_t len, mbstate_t *__restrict ps, locale_t loc )
73{
74    __locale_raii __current( uselocale(loc), uselocale );
75    return wcsnrtombs( dst, src, nwc, len, ps );
76}
77wint_t btowc_l( int c, locale_t loc )
78{
79    __locale_raii __current( uselocale(loc), uselocale );
80    return btowc( c );
81}
82int wctob_l( wint_t c, locale_t loc )
83{
84    __locale_raii __current( uselocale(loc), uselocale );
85    return wctob( c );
86}
87
88int snprintf_l(char *ret, size_t n, locale_t loc, const char *format, ...)
89{
90    __locale_raii __current( uselocale(loc), uselocale );
91    va_list ap;
92    va_start( ap, format );
93    int result = vsnprintf( ret, n, format, ap );
94    va_end(ap);
95    return result;
96}
97
98int asprintf_l( char **ret, locale_t loc, const char *format, ... )
99{
100    va_list ap;
101    va_start( ap, format );
102    int result = vasprintf_l( ret, loc, format, ap );
103    va_end(ap);
104    return result;
105}
106int vasprintf_l( char **ret, locale_t loc, const char *format, va_list ap )
107{
108    __locale_raii __current( uselocale(loc), uselocale );
109    return vasprintf( ret, format, ap );
110}
111