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