update from main archive 960927
[kopensolaris-gnu/glibc.git] / locale / setlocale.c
1 /* Copyright (C) 1991, 1992, 1995, 1996 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Library General Public License as
6 published by the Free Software Foundation; either version 2 of the
7 License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 Library General Public License for more details.
13
14 You should have received a copy of the GNU Library General Public
15 License along with the GNU C Library; see the file COPYING.LIB.  If
16 not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17 Boston, MA 02111-1307, USA.  */
18
19 #include <alloca.h>
20 #include <argz.h>
21 #include <errno.h>
22 #include <libc-lock.h>
23 #include <locale.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <unistd.h>
27
28 #include "localeinfo.h"
29
30 /* For each category declare two external variables (with weak references):
31      extern const struct locale_data *_nl_current_CATEGORY;
32    This points to the current locale's in-core data for CATEGORY.
33      extern const struct locale_data _nl_C_CATEGORY;
34    This contains the built-in "C"/"POSIX" locale's data for CATEGORY.
35    Both are weak references; if &_nl_current_CATEGORY is zero,
36    then nothing is using the locale data.  */
37 #define DEFINE_CATEGORY(category, category_name, items, a, b, c, d) \
38 extern const struct locale_data *_nl_current_##category;                      \
39 extern const struct locale_data _nl_C_##category;                             \
40 weak_extern (_nl_current_##category) weak_extern (_nl_C_##category)
41 #include "categories.def"
42 #undef  DEFINE_CATEGORY
43
44 /* Array indexed by category of pointers to _nl_current_CATEGORY slots.
45    Elements are zero for categories whose data is never used.  */
46 static const struct locale_data * *const _nl_current[] =
47   {
48 #define DEFINE_CATEGORY(category, category_name, items, a, b, c, d) \
49     [category] = &_nl_current_##category,
50 #include "categories.def"
51 #undef  DEFINE_CATEGORY
52   };
53
54 /* Array indexed by category of pointers to _nl_C_CATEGORY slots.
55    Elements are zero for categories whose data is never used.  */
56 const struct locale_data *const _nl_C[] =
57   {
58 #define DEFINE_CATEGORY(category, category_name, items, a, b, c, d) \
59     [category] = &_nl_C_##category,
60 #include "categories.def"
61 #undef  DEFINE_CATEGORY
62   };
63
64
65 /* Define an array of category names (also the environment variable names),
66    indexed by integral category.  */
67 const char *const _nl_category_names[] =
68   {
69 #define DEFINE_CATEGORY(category, category_name, items, a, b, c, d) \
70     [category] = category_name,
71 #include "categories.def"
72 #undef  DEFINE_CATEGORY
73     [LC_ALL] = "LC_ALL"
74   };
75 /* An array of their lengths, for convenience.  */
76 const size_t _nl_category_name_sizes[] =
77   {
78 #define DEFINE_CATEGORY(category, category_name, items, a, b, c, d) \
79     [category] = sizeof (category_name) - 1,
80 #include "categories.def"
81 #undef  DEFINE_CATEGORY
82     [LC_ALL] = sizeof ("LC_ALL") - 1
83   };
84
85
86 /* Declare the postload functions used below.  */
87 #undef  NO_POSTLOAD
88 #define NO_POSTLOAD _nl_postload_ctype /* Harmless thing known to exist.  */
89 #define DEFINE_CATEGORY(category, category_name, items, postload, b, c, d) \
90 extern void postload (void);
91 #include "categories.def"
92 #undef  DEFINE_CATEGORY
93 #undef  NO_POSTLOAD
94
95 /* Define an array indexed by category of postload functions to call after
96    loading and installing that category's data.  */
97 static void (*const _nl_category_postload[]) (void) =
98   {
99 #define DEFINE_CATEGORY(category, category_name, items, postload, b, c, d) \
100     [category] = postload,
101 #include "categories.def"
102 #undef  DEFINE_CATEGORY
103   };
104
105
106 /* Name of current locale for each individual category.
107    Each is malloc'd unless it is nl_C_name.  */
108 static const char *_nl_current_names[] =
109   {
110 #define DEFINE_CATEGORY(category, category_name, items, a, b, c, d) \
111     [category] = _nl_C_name,
112 #include "categories.def"
113 #undef  DEFINE_CATEGORY
114     [LC_ALL] = _nl_C_name               /* For LC_ALL.  */
115   };
116
117
118 /* Lock for protecting global data.  */
119 __libc_lock_define_initialized (, __libc_setlocale_lock)
120
121
122 /* Use this when we come along an error.  */
123 #define ERROR_RETURN                                                          \
124   do {                                                                        \
125     __set_errno (EINVAL);                                                     \
126     return NULL;                                                              \
127   } while (0)
128
129
130 static inline char *
131 clever_copy (const char *string)
132 {
133   size_t len;
134   char *new;
135
136   if (strcmp (string, "C") == 0 || strcmp (string, "POSIX") == 0)
137     /* This return is dangerous because the returned string might be
138        placed in read-only memory.  But everything should be set up to
139        handle this case.  */
140     return (char *) _nl_C_name;
141
142   len = strlen (string) + 1;
143   new = (char *) malloc (len);
144   return new != NULL ? memcpy (new, string, len) : NULL;
145 }
146
147
148 /* Construct a new composite name.  */
149 static inline char *
150 new_composite_name (int category, char *newnames[LC_ALL])
151 {
152   size_t last_len;
153   size_t cumlen = 0;
154   int i;
155   char *new, *p;
156   int same = 1;
157
158   for (i = 0; i < LC_ALL; ++i)
159     {
160       char *name = (category == LC_ALL ? newnames[i] :
161                     category == i ? newnames[0] :
162                     (char *) _nl_current_names[i]);
163       last_len = strlen (name);
164       cumlen += _nl_category_name_sizes[i] + 1 + last_len + 1;
165       if (i > 0 && same && strcmp (name, newnames[0]) != 0)
166         same = 0;
167     }
168
169   if (same)
170     {
171       /* All the categories use the same name.  */
172       if (strcmp (newnames[0], "C") == 0 || strcmp (newnames[0], "POSIX") == 0)
173         return (char *) _nl_C_name;
174
175       new = malloc (last_len + 1);
176
177       return new == NULL ? NULL : memcpy (new, newnames[0], last_len + 1);
178     }
179
180   new = malloc (cumlen);
181   if (new == NULL)
182     return NULL;
183   p = new;
184   for (i = 0; i < LC_ALL; ++i)
185     {
186       /* Add "CATEGORY=NAME;" to the string.  */
187       char *name = (category == LC_ALL ? newnames[i] :
188                     category == i ? newnames[0] :
189                     (char *) _nl_current_names[i]);
190       p = __stpcpy (p, _nl_category_names[i]);
191       *p++ = '=';
192       p = __stpcpy (p, name);
193       *p++ = ';';
194     }
195   p[-1] = '\0';         /* Clobber the last ';'.  */
196   return new;
197 }
198
199
200 /* Put NAME in _nl_current_names.  */
201 static inline void
202 setname (int category, const char *name)
203 {
204   if (_nl_current[category] == NULL
205       && _nl_current_names[category] != _nl_C_name)
206     free ((void *) _nl_current_names[category]);
207
208   _nl_current_names[category] = name;
209 }
210
211
212 /* Put DATA in *_nl_current[CATEGORY].  */
213 static inline void
214 setdata (int category, const struct locale_data *data)
215 {
216   if (_nl_current[category] != NULL)
217     {
218       *_nl_current[category] = data;
219       if (_nl_category_postload[category])
220         (*_nl_category_postload[category]) ();
221     }
222 }
223
224
225 char *
226 setlocale (int category, const char *locale)
227 {
228   char *locale_path;
229   size_t locale_path_len;
230   const char *locpath_var;
231   char *composite;
232
233   /* Sanity check for CATEGORY argument.  */
234   if (category < 0 || category > LC_ALL)
235     ERROR_RETURN;
236
237   /* Does user want name of current locale?  */
238   if (locale == NULL)
239     return (char *) _nl_current_names[category];
240
241   if (strcmp (locale, _nl_current_names[category]) == 0)
242     /* Changing to the same thing.  */
243     return (char *) _nl_current_names[category];
244
245   /* We perhaps really have to load some data.  So we determine the
246      path in which to look for the data now.  The environment variable
247      `LOCPATH' must only be used when the binary has no SUID or SGID
248      bit set.  */
249   locale_path = NULL;
250   locale_path_len = 0;
251
252   locpath_var = __secure_getenv ("LOCPATH");
253   if (locpath_var != NULL && locpath_var[0] != '\0')
254     if (__argz_create_sep (locpath_var, ':',
255                            &locale_path, &locale_path_len) != 0)
256       return NULL;
257
258   if (__argz_append (&locale_path, &locale_path_len,
259                      LOCALE_PATH, sizeof (LOCALE_PATH)) != 0)
260     return NULL;
261
262   if (category == LC_ALL)
263     {
264       /* The user wants to set all categories.  The desired locales
265          for the individual categories can be selected by using a
266          composite locale name.  This is a semi-colon separated list
267          of entries of the form `CATEGORY=VALUE'.  */
268       char *newnames[LC_ALL];
269       const struct locale_data *newdata[LC_ALL];
270
271       /* Set all name pointers to the argument name.  */
272       for (category = 0; category < LC_ALL; ++category)
273         newnames[category] = (char *) locale;
274
275       if (strchr (locale, ';') != NULL)
276         {
277           /* This is a composite name.  Make a copy and split it up.  */
278           char *np = strdupa (locale);
279           char *cp;
280           int cnt;
281
282           while ((cp = strchr (np, '=')) != NULL)
283             {
284               for (cnt = 0; cnt < LC_ALL; ++cnt)
285                 if ((size_t) (cp - np) == _nl_category_name_sizes[cnt]
286                     && memcmp (np, _nl_category_names[cnt], cp - np) == 0)
287                   break;
288
289               if (cnt == LC_ALL)
290                 /* Bogus category name.  */
291                 ERROR_RETURN;
292
293               /* Found the category this clause sets.  */
294               newnames[cnt] = ++cp;
295               cp = strchr (cp, ';');
296               if (cp != NULL)
297                 {
298                   /* Examine the next clause.  */
299                   *cp = '\0';
300                   np = cp + 1;
301                 }
302               else
303                 /* This was the last clause.  We are done.  */
304                 break;
305             }
306
307           for (cnt = 0; cnt < LC_ALL; ++cnt)
308             if (newnames[cnt] == locale)
309               /* The composite name did not specify all categories.  */
310               ERROR_RETURN;
311         }
312
313       /* Protect global data.  */
314       __libc_lock_lock (__libc_setlocale_lock);
315
316       /* Load the new data for each category.  */
317       while (category-- > 0)
318         /* Only actually load the data if anything will use it.  */
319         if (_nl_current[category] != NULL)
320           {
321             newdata[category] = _nl_find_locale (locale_path, locale_path_len,
322                                                  category,
323                                                  &newnames[category]);
324
325             if (newdata[category] == NULL)
326               break;
327           }
328         else
329           {
330             /* The data is never used; just change the name.  */
331             newnames[category] = clever_copy (newnames[category]);
332             if (newnames[category] == NULL)
333               break;
334           }
335
336       /* Create new composite name.  */
337       if (category >= 0
338           || (composite = new_composite_name (LC_ALL, newnames)) == NULL)
339         /* Loading this part of the locale failed.  Abort the
340            composite load.  */
341         composite = NULL;
342       else
343         {
344           /* Now we have loaded all the new data.  Put it in place.  */
345           for (category = 0; category < LC_ALL; ++category)
346             {
347               setdata (category, newdata[category]);
348               setname (category, newnames[category]);
349             }
350           setname (LC_ALL, composite);
351         }
352
353       /* Critical section left.  */
354       __libc_lock_unlock (__libc_setlocale_lock);
355
356       return composite;
357     }
358   else
359     {
360       const struct locale_data *newdata = NULL;
361       char *newname = (char *) locale;
362
363       /* Protect global data.  */
364       __libc_lock_lock (__libc_setlocale_lock);
365
366       if (_nl_current[category] != NULL)
367         {
368           /* Only actually load the data if anything will use it.  */
369           newdata = _nl_find_locale (locale_path, locale_path_len, category,
370                                      (char **) &newname);
371           if (newdata == NULL)
372             goto abort_single;
373         }
374
375       /* Create new composite name.  */
376       composite = new_composite_name (category, &newname);
377       if (composite == NULL)
378         {
379           /* Say that we don't have any data loaded.  */
380         abort_single:
381           newname = NULL;
382         }
383       else
384         {
385           if (_nl_current[category] != NULL)
386             setdata (category, newdata);
387
388           setname (category, newname);
389           setname (LC_ALL, composite);
390         }
391
392       /* Critical section left.  */
393       __libc_lock_unlock (__libc_setlocale_lock);
394
395       return newname;
396     }
397 }