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