Correct BSD License clause numbering from 1-2-4 to 1-2-3.
[dragonfly.git] / lib / libc / locale / setlocale.c
1 /*
2  * Copyright (c) 1996 - 2002 FreeBSD Project
3  * Copyright (c) 1991, 1993
4  *      The Regents of the University of California.  All rights reserved.
5  *
6  * This code is derived from software contributed to Berkeley by
7  * Paul Borman at Krystal Technologies.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#)setlocale.c      8.1 (Berkeley) 7/4/93
34  * $FreeBSD: head/lib/libc/locale/setlocale.c 228921 2011-12-27 23:28:01Z jilles $
35  */
36
37
38 #include <sys/types.h>
39 #include <sys/stat.h>
40 #include <errno.h>
41 #include <limits.h>
42 #include <locale.h>
43 #include <paths.h>      /* for _PATH_LOCALE */
44 #include <stdlib.h>
45 #include <string.h>
46 #include <unistd.h>
47 #include "collate.h"
48 #include "lmonetary.h"  /* for __monetary_load_locale() */
49 #include "lnumeric.h"   /* for __numeric_load_locale() */
50 #include "lmessages.h"  /* for __messages_load_locale() */
51 #include "setlocale.h"
52 #include "ldpart.h"
53 #include "../stdtime/timelocal.h" /* for __time_load_locale() */
54
55 /*
56  * Category names for getenv()
57  */
58 static const char categories[_LC_LAST][12] = {
59     "LC_ALL",
60     "LC_COLLATE",
61     "LC_CTYPE",
62     "LC_MONETARY",
63     "LC_NUMERIC",
64     "LC_TIME",
65     "LC_MESSAGES",
66 };
67
68 /*
69  * Current locales for each category
70  */
71 static char current_categories[_LC_LAST][ENCODING_LEN + 1] = {
72     "C",
73     "C",
74     "C",
75     "C",
76     "C",
77     "C",
78     "C",
79 };
80
81 /*
82  * Path to locale storage directory
83  */
84 char    *_PathLocale;
85
86 /*
87  * The locales we are going to try and load
88  */
89 static char new_categories[_LC_LAST][ENCODING_LEN + 1];
90 static char saved_categories[_LC_LAST][ENCODING_LEN + 1];
91
92 static char current_locale_string[_LC_LAST * (ENCODING_LEN + 1/*"/"*/ + 1)];
93
94 static char     *currentlocale(void);
95 static char     *loadlocale(int);
96 const char *__get_locale_env(int);
97
98 char *
99 setlocale(int category, const char *locale)
100 {
101         int i, j, len, saverr;
102         const char *env, *r;
103
104         if (category < LC_ALL || category >= _LC_LAST) {
105                 errno = EINVAL;
106                 return (NULL);
107         }
108
109         if (locale == NULL)
110                 return (category != LC_ALL ?
111                     current_categories[category] : currentlocale());
112
113         /*
114          * Default to the current locale for everything.
115          */
116         for (i = 1; i < _LC_LAST; ++i)
117                 (void)strcpy(new_categories[i], current_categories[i]);
118
119         /*
120          * Now go fill up new_categories from the locale argument
121          */
122         if (!*locale) {
123                 if (category == LC_ALL) {
124                         for (i = 1; i < _LC_LAST; ++i) {
125                                 env = __get_locale_env(i);
126                                 if (strlen(env) > ENCODING_LEN) {
127                                         errno = EINVAL;
128                                         return (NULL);
129                                 }
130                                 (void)strcpy(new_categories[i], env);
131                         }
132                 } else {
133                         env = __get_locale_env(category);
134                         if (strlen(env) > ENCODING_LEN) {
135                                 errno = EINVAL;
136                                 return (NULL);
137                         }
138                         (void)strcpy(new_categories[category], env);
139                 }
140         } else if (category != LC_ALL) {
141                 if (strlen(locale) > ENCODING_LEN) {
142                         errno = EINVAL;
143                         return (NULL);
144                 }
145                 (void)strcpy(new_categories[category], locale);
146         } else {
147                 if ((r = strchr(locale, '/')) == NULL) {
148                         if (strlen(locale) > ENCODING_LEN) {
149                                 errno = EINVAL;
150                                 return (NULL);
151                         }
152                         for (i = 1; i < _LC_LAST; ++i)
153                                 (void)strcpy(new_categories[i], locale);
154                 } else {
155                         for (i = 1; r[1] == '/'; ++r)
156                                 ;
157                         if (!r[1]) {
158                                 errno = EINVAL;
159                                 return (NULL);  /* Hmm, just slashes... */
160                         }
161                         do {
162                                 if (i == _LC_LAST)
163                                         break;  /* Too many slashes... */
164                                 if ((len = r - locale) > ENCODING_LEN) {
165                                         errno = EINVAL;
166                                         return (NULL);
167                                 }
168                                 (void)strlcpy(new_categories[i], locale,
169                                               len + 1);
170                                 i++;
171                                 while (*r == '/')
172                                         r++;
173                                 locale = r;
174                                 while (*r && *r != '/')
175                                         r++;
176                         } while (*locale);
177                         while (i < _LC_LAST) {
178                                 (void)strcpy(new_categories[i],
179                                              new_categories[i-1]);
180                                 i++;
181                         }
182                 }
183         }
184
185         if (category != LC_ALL)
186                 return (loadlocale(category));
187
188         for (i = 1; i < _LC_LAST; ++i) {
189                 (void)strcpy(saved_categories[i], current_categories[i]);
190                 if (loadlocale(i) == NULL) {
191                         saverr = errno;
192                         for (j = 1; j < i; j++) {
193                                 (void)strcpy(new_categories[j],
194                                              saved_categories[j]);
195                                 if (loadlocale(j) == NULL) {
196                                         (void)strcpy(new_categories[j], "C");
197                                         (void)loadlocale(j);
198                                 }
199                         }
200                         errno = saverr;
201                         return (NULL);
202                 }
203         }
204         return (currentlocale());
205 }
206
207 static char *
208 currentlocale(void)
209 {
210         int i;
211
212         (void)strcpy(current_locale_string, current_categories[1]);
213
214         for (i = 2; i < _LC_LAST; ++i)
215                 if (strcmp(current_categories[1], current_categories[i])) {
216                         for (i = 2; i < _LC_LAST; ++i) {
217                                 (void)strcat(current_locale_string, "/");
218                                 (void)strcat(current_locale_string,
219                                              current_categories[i]);
220                         }
221                         break;
222                 }
223         return (current_locale_string);
224 }
225
226 static char *
227 loadlocale(int category)
228 {
229         char *new = new_categories[category];
230         char *old = current_categories[category];
231         int (*func)(const char *);
232         int saved_errno;
233
234         if ((new[0] == '.' &&
235              (new[1] == '\0' || (new[1] == '.' && new[2] == '\0'))) ||
236             strchr(new, '/') != NULL) {
237                 errno = EINVAL;
238                 return (NULL);
239         }
240
241         saved_errno = errno;
242         errno = __detect_path_locale();
243         if (errno != 0)
244                 return (NULL);
245         errno = saved_errno;
246
247         switch (category) {
248         case LC_CTYPE:
249                 func = __wrap_setrunelocale;
250                 break;
251         case LC_COLLATE:
252                 func = __collate_load_tables;
253                 break;
254         case LC_TIME:
255                 func = __time_load_locale;
256                 break;
257         case LC_NUMERIC:
258                 func = __numeric_load_locale;
259                 break;
260         case LC_MONETARY:
261                 func = __monetary_load_locale;
262                 break;
263         case LC_MESSAGES:
264                 func = __messages_load_locale;
265                 break;
266         default:
267                 errno = EINVAL;
268                 return (NULL);
269         }
270
271         if (strcmp(new, old) == 0)
272                 return (old);
273
274         if (func(new) != _LDP_ERROR) {
275                 (void)strcpy(old, new);
276                 (void)strcpy(__xlocale_global_locale.components[category-1]->locale, new);
277                 return (old);
278         }
279
280         return (NULL);
281 }
282
283 const char *
284 __get_locale_env(int category)
285 {
286         const char *env;
287
288         /* 1. check LC_ALL. */
289         env = getenv(categories[0]);
290
291         /* 2. check LC_* */
292         if (env == NULL || !*env)
293                 env = getenv(categories[category]);
294
295         /* 3. check LANG */
296         if (env == NULL || !*env)
297                 env = getenv("LANG");
298
299         /* 4. if none is set, fall to "C" */
300         if (env == NULL || !*env)
301                 env = "C";
302
303         return (env);
304 }
305
306 /*
307  * Detect locale storage location and store its value to _PathLocale variable
308  */
309 int
310 __detect_path_locale(void)
311 {
312         if (_PathLocale == NULL) {
313                 char *p = getenv("PATH_LOCALE");
314
315                 if (p != NULL && !issetugid()) {
316                         if (strlen(p) + 1/*"/"*/ + ENCODING_LEN +
317                             1/*"/"*/ + CATEGORY_LEN >= PATH_MAX)
318                                 return (ENAMETOOLONG);
319                         _PathLocale = strdup(p);
320                         if (_PathLocale == NULL)
321                                 return (errno == 0 ? ENOMEM : errno);
322                 } else
323                         _PathLocale = _PATH_LOCALE;
324         }
325         return (0);
326 }