1 /*
2  * Copyright (C) 2008 The Android Open Source Project
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *  * Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  *  * Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in
12  *    the documentation and/or other materials provided with the
13  *    distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28 
29 #include <errno.h>
30 #include <locale.h>
31 #include <pthread.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <strings.h>
35 #include <time.h>
36 #include <wchar.h>
37 
38 #include "platform/bionic/macros.h"
39 
40 #if defined(__BIONIC_BUILD_FOR_ANDROID_SUPPORT)
41 #define USE_TLS_SLOT 0
42 #else
43 #define USE_TLS_SLOT 1
44 #endif
45 
46 #if USE_TLS_SLOT
47 #include "bionic/pthread_internal.h"
48 #endif
49 
50 // We only support two locales, the "C" locale (also known as "POSIX"),
51 // and the "C.UTF-8" locale (also known as "en_US.UTF-8").
52 
53 static bool __bionic_current_locale_is_utf8 = true;
54 
55 struct __locale_t {
56   size_t mb_cur_max;
57 };
58 
59 // Avoid using new/delete in this file, because a user may have overridden
60 // new/delete, and we want to avoid making extraneous calls to them. This isn't
61 // an issue for libc.so in the platform, but this file is also compiled into the
62 // NDK's libandroid_support.a, and there are libc++ tests that count the number
63 // of calls to new/delete.
64 #pragma clang poison new delete
65 
__alloc_locale(size_t mb_cur_max)66 static inline locale_t __alloc_locale(size_t mb_cur_max) {
67   auto result = static_cast<__locale_t*>(malloc(sizeof(__locale_t)));
68   if (result == nullptr) return nullptr;
69   result->mb_cur_max = mb_cur_max;
70   return result;
71 }
72 
get_locale_mb_cur_max(locale_t l)73 static inline size_t get_locale_mb_cur_max(locale_t l) {
74   if (l == LC_GLOBAL_LOCALE) {
75     return __bionic_current_locale_is_utf8 ? 4 : 1;
76   } else {
77     return l->mb_cur_max;
78   }
79 }
80 
__ctype_get_mb_cur_max()81 size_t __ctype_get_mb_cur_max() {
82   return get_locale_mb_cur_max(uselocale(nullptr));
83 }
84 
85 #if !USE_TLS_SLOT
86 static thread_local locale_t g_current_locale;
87 #endif
88 
89 static pthread_once_t g_locale_once = PTHREAD_ONCE_INIT;
90 static lconv g_locale;
91 
__locale_init()92 static void __locale_init() {
93   g_locale.decimal_point = const_cast<char*>(".");
94 
95   char* not_available = const_cast<char*>("");
96   g_locale.thousands_sep = not_available;
97   g_locale.grouping = not_available;
98   g_locale.int_curr_symbol = not_available;
99   g_locale.currency_symbol = not_available;
100   g_locale.mon_decimal_point = not_available;
101   g_locale.mon_thousands_sep = not_available;
102   g_locale.mon_grouping = not_available;
103   g_locale.positive_sign = not_available;
104   g_locale.negative_sign = not_available;
105 
106   g_locale.int_frac_digits = CHAR_MAX;
107   g_locale.frac_digits = CHAR_MAX;
108   g_locale.p_cs_precedes = CHAR_MAX;
109   g_locale.p_sep_by_space = CHAR_MAX;
110   g_locale.n_cs_precedes = CHAR_MAX;
111   g_locale.n_sep_by_space = CHAR_MAX;
112   g_locale.p_sign_posn = CHAR_MAX;
113   g_locale.n_sign_posn = CHAR_MAX;
114   g_locale.int_p_cs_precedes = CHAR_MAX;
115   g_locale.int_p_sep_by_space = CHAR_MAX;
116   g_locale.int_n_cs_precedes = CHAR_MAX;
117   g_locale.int_n_sep_by_space = CHAR_MAX;
118   g_locale.int_p_sign_posn = CHAR_MAX;
119   g_locale.int_n_sign_posn = CHAR_MAX;
120 }
121 
__is_supported_locale(const char * locale_name)122 static bool __is_supported_locale(const char* locale_name) {
123   return (strcmp(locale_name, "") == 0 ||
124           strcmp(locale_name, "C") == 0 ||
125           strcmp(locale_name, "C.UTF-8") == 0 ||
126           strcmp(locale_name, "en_US.UTF-8") == 0 ||
127           strcmp(locale_name, "POSIX") == 0);
128 }
129 
__is_utf8_locale(const char * locale_name)130 static bool __is_utf8_locale(const char* locale_name) {
131   return (*locale_name == '\0' || strstr(locale_name, "UTF-8"));
132 }
133 
localeconv()134 lconv* localeconv() {
135   pthread_once(&g_locale_once, __locale_init);
136   return &g_locale;
137 }
138 
duplocale(locale_t l)139 locale_t duplocale(locale_t l) {
140   return __alloc_locale(get_locale_mb_cur_max(l));
141 }
142 
freelocale(locale_t l)143 void freelocale(locale_t l) {
144   free(l);
145 }
146 
newlocale(int category_mask,const char * locale_name,locale_t)147 locale_t newlocale(int category_mask, const char* locale_name, locale_t /*base*/) {
148   // Are 'category_mask' and 'locale_name' valid?
149   if ((category_mask & ~LC_ALL_MASK) != 0 || locale_name == nullptr) {
150     errno = EINVAL;
151     return nullptr;
152   }
153 
154   if (!__is_supported_locale(locale_name)) {
155     errno = ENOENT;
156     return nullptr;
157   }
158 
159   return __alloc_locale(__is_utf8_locale(locale_name) ? 4 : 1);
160 }
161 
setlocale(int category,const char * locale_name)162 char* setlocale(int category, const char* locale_name) {
163   // Is 'category' valid?
164   if (category < LC_CTYPE || category > LC_IDENTIFICATION) {
165     errno = EINVAL;
166     return nullptr;
167   }
168 
169   // Caller wants to set the locale rather than just query?
170   if (locale_name != nullptr) {
171     if (!__is_supported_locale(locale_name)) {
172       // We don't support this locale.
173       errno = ENOENT;
174       return nullptr;
175     }
176     __bionic_current_locale_is_utf8 = __is_utf8_locale(locale_name);
177   }
178 
179   return const_cast<char*>(__bionic_current_locale_is_utf8 ? "C.UTF-8" : "C");
180 }
181 
get_current_locale_ptr()182 static locale_t* get_current_locale_ptr() {
183 #if USE_TLS_SLOT
184   return &__get_bionic_tls().locale;
185 #else
186   return &g_current_locale;
187 #endif
188 }
189 
uselocale(locale_t new_locale)190 locale_t uselocale(locale_t new_locale) {
191   locale_t old_locale = *get_current_locale_ptr();
192 
193   // If this is the first call to uselocale(3) on this thread, we return LC_GLOBAL_LOCALE.
194   if (old_locale == nullptr) {
195     old_locale = LC_GLOBAL_LOCALE;
196   }
197 
198   if (new_locale != nullptr) {
199     *get_current_locale_ptr() = new_locale;
200   }
201 
202   return old_locale;
203 }
204