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 #include <__assert>
10 #include <__support/ibm/xlocale.h>
11 #include <sstream>
12 #include <vector>
13 
14 #ifdef __cplusplus
15 extern "C" {
16 #endif // __cplusplus
17 
18 locale_t newlocale(int category_mask, const char* locale, locale_t base) {
19   // Maintain current locale name(s) to restore later.
20   std::string current_loc_name(setlocale(LC_ALL, 0));
21 
22   // Check for errors.
23   if (category_mask == LC_ALL_MASK && setlocale(LC_ALL, locale) == NULL) {
24     errno = EINVAL;
25     return (locale_t)0;
26   } else {
27     for (int _Cat = 0; _Cat <= _LC_MAX; ++_Cat) {
28       if ((_CATMASK(_Cat) & category_mask) != 0 && setlocale(_Cat, locale) == NULL) {
29         setlocale(LC_ALL, current_loc_name.c_str());
30         errno = EINVAL;
31         return (locale_t)0;
32       }
33     }
34   }
35 
36   // Create new locale.
37   locale_t newloc = new locale_struct();
38 
39   if (base) {
40     if (category_mask != LC_ALL_MASK) {
41       // Copy base when it will not be overwritten.
42       memcpy(newloc, base, sizeof (locale_struct));
43       newloc->category_mask = category_mask | base->category_mask;
44     }
45     delete base;
46   } else {
47     newloc->category_mask = category_mask;
48   }
49 
50   if (category_mask & LC_COLLATE_MASK)
51     newloc->lc_collate = locale;
52   if (category_mask & LC_CTYPE_MASK)
53     newloc->lc_ctype = locale;
54   if (category_mask & LC_MONETARY_MASK)
55     newloc->lc_monetary = locale;
56   if (category_mask & LC_NUMERIC_MASK)
57     newloc->lc_numeric = locale;
58   if (category_mask & LC_TIME_MASK)
59     newloc->lc_time = locale;
60   if (category_mask & LC_MESSAGES_MASK)
61     newloc->lc_messages = locale;
62 
63   // Restore current locale.
64   setlocale(LC_ALL, current_loc_name.c_str());
65   return (locale_t)newloc;
66 }
67 
68 void freelocale(locale_t locobj) {
69   delete locobj;
70 }
71 
72 locale_t uselocale(locale_t newloc) {
73   // Maintain current locale name(s).
74   std::string current_loc_name(setlocale(LC_ALL, 0));
75 
76   if (newloc) {
77     // Set locales and check for errors.
78     bool is_error =
79       (newloc->category_mask & LC_COLLATE_MASK &&
80         setlocale(LC_COLLATE, newloc->lc_collate.c_str()) == NULL) ||
81       (newloc->category_mask & LC_CTYPE_MASK &&
82         setlocale(LC_CTYPE, newloc->lc_ctype.c_str()) == NULL) ||
83       (newloc->category_mask & LC_MONETARY_MASK &&
84         setlocale(LC_MONETARY, newloc->lc_monetary.c_str()) == NULL) ||
85       (newloc->category_mask & LC_NUMERIC_MASK &&
86         setlocale(LC_NUMERIC, newloc->lc_numeric.c_str()) == NULL) ||
87       (newloc->category_mask & LC_TIME_MASK &&
88         setlocale(LC_TIME, newloc->lc_time.c_str()) == NULL) ||
89       (newloc->category_mask & LC_MESSAGES_MASK &&
90         setlocale(LC_MESSAGES, newloc->lc_messages.c_str()) == NULL);
91 
92     if (is_error) {
93       setlocale(LC_ALL, current_loc_name.c_str());
94       errno = EINVAL;
95       return (locale_t)0;
96     }
97   }
98 
99   // Construct and return previous locale.
100   locale_t previous_loc = new locale_struct();
101 
102   // current_loc_name might be a comma-separated locale name list.
103   if (current_loc_name.find(',') != std::string::npos) {
104     // Tokenize locale name list.
105     const char delimiter = ',';
106     std::vector<std::string> tokenized;
107     std::stringstream ss(current_loc_name);
108     std::string s;
109 
110     while (std::getline(ss, s, delimiter)) {
111         tokenized.push_back(s);
112     }
113 
114     _LIBCPP_ASSERT_UNCATEGORIZED(tokenized.size() >= _NCAT, "locale-name list is too short");
115 
116     previous_loc->lc_collate = tokenized[LC_COLLATE];
117     previous_loc->lc_ctype = tokenized[LC_CTYPE];
118     previous_loc->lc_monetary = tokenized[LC_MONETARY];
119     previous_loc->lc_numeric = tokenized[LC_NUMERIC];
120     previous_loc->lc_time = tokenized[LC_TIME];
121     // Skip LC_TOD.
122     previous_loc->lc_messages = tokenized[LC_MESSAGES];
123   } else {
124     previous_loc->lc_collate = current_loc_name;
125     previous_loc->lc_ctype = current_loc_name;
126     previous_loc->lc_monetary = current_loc_name;
127     previous_loc->lc_numeric = current_loc_name;
128     previous_loc->lc_time = current_loc_name;
129     previous_loc->lc_messages = current_loc_name;
130   }
131 
132   previous_loc->category_mask = LC_ALL_MASK;
133   return previous_loc;
134 }
135 
136 #ifdef __cplusplus
137 }
138 #endif // __cplusplus
139