1 /*
   2  * Copyright 2014 Garrett D'Amore <garrett@damore.org>
   3  * Copyright 2010 Nexenta Systems, Inc.  All rights reserved.
   4  * Copyright (c) 1996 - 2002 FreeBSD Project
   5  * Copyright (c) 1991, 1993
   6  *      The Regents of the University of California.  All rights reserved.
   7  *
   8  * This code is derived from software contributed to Berkeley by
   9  * Paul Borman at Krystal Technologies.
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions
  13  * are met:
  14  * 1. Redistributions of source code must retain the above copyright
  15  *    notice, this list of conditions and the following disclaimer.
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in the
  18  *    documentation and/or other materials provided with the distribution.
  19  * 4. Neither the name of the University nor the names of its contributors
  20  *    may be used to endorse or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35 
  36 #include "lint.h"
  37 #include <sys/types.h>
  38 #include <sys/stat.h>
  39 #include <errno.h>
  40 #include <limits.h>
  41 #include <locale.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45 #include <stdio.h>
  46 #include "mtlib.h"
  47 #include "collate.h"
  48 #include "lnumeric.h"   /* for struct lc_numeric */
  49 #include "lctype.h"     /* for struct lc_ctype */
  50 #include "setlocale.h"
  51 #include "../i18n/_loc_path.h"
  52 #include "localeimpl.h"
  53 #include "../i18n/_locale.h"
  54 
  55 /*
  56  * Path to locale storage directory.  See ../i18n/_loc_path.h
  57  */
  58 char    *_PathLocale = _DFLT_LOC_PATH;
  59 
  60 static char     *current_locale(locale_t, int);
  61 static void     install_legacy(locale_t, int);
  62 
  63 static mutex_t setlocale_lock = DEFAULTMUTEX;
  64 static locale_t setlocale_list = NULL;
  65 
  66 char *
  67 setlocale(int category, const char *locname)
  68 {
  69         locale_t loc;
  70         locale_t srch;
  71         int mask;
  72 
  73         if (category < 0 || category > LC_ALL) {
  74                 errno = EINVAL;
  75                 return (NULL);
  76         }
  77 
  78         if (locname == NULL)
  79                 return (current_locale(___global_locale, category));
  80 
  81         mask = (category == LC_ALL ? LC_ALL_MASK : (1 << category));
  82 
  83         loc = newlocale(mask, locname, NULL);
  84         if (loc == NULL) {
  85                 return (NULL);
  86         }
  87 
  88         /*
  89          * This next logic looks to see if we have ever used the same locale
  90          * settings before.  If so, we reuse it.  We avoid ever calling
  91          * freelocale() on a locale setting built up by setlocale, this
  92          * ensures that consumers (uselocale) will always be thread safe;
  93          * the actual locale data objects are never freed, and unique
  94          * locale objects are also never freed.  We reuse to avoid leaking
  95          * memory in applications that call setlocale repeatedly.
  96          */
  97         lmutex_lock(&setlocale_lock);
  98         for (srch = setlocale_list; srch != NULL; srch = srch->next) {
  99                 if (strcmp(srch->locname, loc->locname) == 0) {
 100                         break;
 101                 }
 102         }
 103 
 104         if (srch == NULL) {
 105                 /* this is a new locale, save it for reuse later */
 106                 loc->next = setlocale_list;
 107                 loc->on_list = 1;
 108                 setlocale_list = loc;
 109         } else {
 110                 /* we already had it, toss the new, and use what we found */
 111                 freelocale(loc);
 112                 loc = srch;
 113         }
 114         ___global_locale = loc;
 115 
 116         install_legacy(loc, mask);
 117         lmutex_unlock(&setlocale_lock);
 118 
 119         return (current_locale(loc, category));
 120 }
 121 
 122 static char *
 123 current_locale(locale_t loc, int cat)
 124 {
 125         switch (cat) {
 126         case LC_CTYPE:
 127         case LC_COLLATE:
 128         case LC_MESSAGES:
 129         case LC_MONETARY:
 130         case LC_NUMERIC:
 131         case LC_TIME:
 132                 return (loc->locdata[cat]->l_lname);
 133         case LC_ALL:
 134                 return (loc->locname);
 135         default:
 136                 return (NULL);
 137         }
 138 }
 139 
 140 static void
 141 install_legacy(locale_t loc, int mask)
 142 {
 143         /*
 144          * Update the legacy fixed variables that may be baked into
 145          * legacy programs.  This is really unfortunate, but we can't
 146          * solve for them otherwise.  Note that such legacy programs
 147          * are only going to see the global locale settings, and cannot
 148          * benefit from uselocale().
 149          */
 150         if (mask & LC_NUMERIC_MASK) {
 151                 struct lc_numeric *lnum;
 152                 lnum = loc->locdata[LC_NUMERIC]->l_data[0];
 153                 _numeric[0] = *lnum->decimal_point;
 154                 _numeric[1] = *lnum->thousands_sep;
 155         }
 156 
 157         if (mask & LC_CTYPE_MASK) {
 158                 struct lc_ctype *lct;
 159                 lct = loc->locdata[LC_CTYPE]->l_data[0];
 160                 for (int i = 0; i < _CACHED_RUNES; i++) {
 161                         /* ctype can only encode the lower 8 bits. */
 162                         __ctype[i+1] = lct->lc_ctype_mask[i] & 0xff;
 163                         __ctype_mask[i] = lct->lc_ctype_mask[i];
 164                 }
 165 
 166                 /* The bottom half is the toupper/lower array */
 167                 for (int i = 0; i < _CACHED_RUNES; i++) {
 168                         int u, l;
 169                         __ctype[258 + i] = i;
 170                         u = lct->lc_trans_upper[i];
 171                         l = lct->lc_trans_lower[i];
 172                         if (u && u != i)
 173                                 __ctype[258+i] = u;
 174                         if (l && l != i)
 175                                 __ctype[258+i] = l;
 176 
 177                         /* Don't forget these annoyances either! */
 178                         __trans_upper[i] = u;
 179                         __trans_lower[i] = l;
 180                 }
 181 
 182                 /* Maximum mblen, cswidth, weird legacy */
 183                 __ctype[520] = lct->lc_max_mblen;
 184         }
 185 }