1 /*- 2 * Copyright (c) 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * This code is derived from software contributed to Berkeley by 6 * Paul Borman at Krystal Technologies. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 4. Neither the name of the University nor the names of its contributors 17 * may be used to endorse or promote products derived from this software 18 * without specific prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 23 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 */ 32 33 #include <sys/cdefs.h> 34 __FBSDID("$FreeBSD$"); 35 36 #include <runetype.h> 37 #include <errno.h> 38 #include <limits.h> 39 #include <string.h> 40 #include <stdio.h> 41 #include <stdlib.h> 42 #include <unistd.h> 43 #include <wchar.h> 44 #include "ldpart.h" 45 #include "mblocal.h" 46 #include "setlocale.h" 47 48 extern int __mb_sb_limit; 49 50 extern _RuneLocale *_Read_RuneMagi(FILE *); 51 52 static int __setrunelocale(const char *); 53 54 static int 55 __setrunelocale(const char *encoding) 56 { 57 FILE *fp; 58 char name[PATH_MAX]; 59 _RuneLocale *rl; 60 int saverr, ret; 61 size_t (*old__mbrtowc)(wchar_t * __restrict, 62 const char * __restrict, size_t, mbstate_t * __restrict); 63 size_t (*old__wcrtomb)(char * __restrict, wchar_t, 64 mbstate_t * __restrict); 65 int (*old__mbsinit)(const mbstate_t *); 66 size_t (*old__mbsnrtowcs)(wchar_t * __restrict, 67 const char ** __restrict, size_t, size_t, mbstate_t * __restrict); 68 size_t (*old__wcsnrtombs)(char * __restrict, 69 const wchar_t ** __restrict, size_t, size_t, 70 mbstate_t * __restrict); 71 static char ctype_encoding[ENCODING_LEN + 1]; 72 static _RuneLocale *CachedRuneLocale; 73 static int Cached__mb_cur_max; 74 static int Cached__mb_sb_limit; 75 static size_t (*Cached__mbrtowc)(wchar_t * __restrict, 76 const char * __restrict, size_t, mbstate_t * __restrict); 77 static size_t (*Cached__wcrtomb)(char * __restrict, wchar_t, 78 mbstate_t * __restrict); 79 static int (*Cached__mbsinit)(const mbstate_t *); 80 static size_t (*Cached__mbsnrtowcs)(wchar_t * __restrict, 81 const char ** __restrict, size_t, size_t, mbstate_t * __restrict); 82 static size_t (*Cached__wcsnrtombs)(char * __restrict, 83 const wchar_t ** __restrict, size_t, size_t, 84 mbstate_t * __restrict); 85 86 /* 87 * The "C" and "POSIX" locale are always here. 88 */ 89 if (strcmp(encoding, "C") == 0 || strcmp(encoding, "POSIX") == 0) { 90 (void) _none_init(&_DefaultRuneLocale); 91 return (0); 92 } 93 94 /* 95 * If the locale name is the same as our cache, use the cache. 96 */ 97 if (CachedRuneLocale != NULL && 98 strcmp(encoding, ctype_encoding) == 0) { 99 _CurrentRuneLocale = CachedRuneLocale; 100 __mb_cur_max = Cached__mb_cur_max; 101 __mb_sb_limit = Cached__mb_sb_limit; 102 __mbrtowc = Cached__mbrtowc; 103 __mbsinit = Cached__mbsinit; 104 __mbsnrtowcs = Cached__mbsnrtowcs; 105 __wcrtomb = Cached__wcrtomb; 106 __wcsnrtombs = Cached__wcsnrtombs; 107 return (0); 108 } 109 110 /* 111 * Slurp the locale file into the cache. 112 */ 113 114 /* Range checking not needed, encoding length already checked before */ 115 (void) strcpy(name, _PathLocale); 116 (void) strcat(name, "/"); 117 (void) strcat(name, encoding); 118 (void) strcat(name, "/LC_CTYPE"); 119 120 if ((fp = fopen(name, "r")) == NULL) 121 return (errno == 0 ? ENOENT : errno); 122 123 if ((rl = _Read_RuneMagi(fp)) == NULL) { 124 saverr = (errno == 0 ? EFTYPE : errno); 125 (void)fclose(fp); 126 return (saverr); 127 } 128 (void)fclose(fp); 129 130 old__mbrtowc = __mbrtowc; 131 old__mbsinit = __mbsinit; 132 old__mbsnrtowcs = __mbsnrtowcs; 133 old__wcrtomb = __wcrtomb; 134 old__wcsnrtombs = __wcsnrtombs; 135 136 __mbrtowc = NULL; 137 __mbsinit = NULL; 138 __mbsnrtowcs = __mbsnrtowcs_std; 139 __wcrtomb = NULL; 140 __wcsnrtombs = __wcsnrtombs_std; 141 142 rl->__sputrune = NULL; 143 rl->__sgetrune = NULL; 144 if (strcmp(rl->__encoding, "NONE") == 0) 145 ret = _none_init(rl); 146 else if (strcmp(rl->__encoding, "ASCII") == 0) 147 ret = _ascii_init(rl); 148 else if (strcmp(rl->__encoding, "UTF-8") == 0) 149 ret = _UTF8_init(rl); 150 else if (strcmp(rl->__encoding, "EUC") == 0) 151 ret = _EUC_init(rl); 152 else if (strcmp(rl->__encoding, "GB18030") == 0) 153 ret = _GB18030_init(rl); 154 else if (strcmp(rl->__encoding, "GB2312") == 0) 155 ret = _GB2312_init(rl); 156 else if (strcmp(rl->__encoding, "GBK") == 0) 157 ret = _GBK_init(rl); 158 else if (strcmp(rl->__encoding, "BIG5") == 0) 159 ret = _BIG5_init(rl); 160 else if (strcmp(rl->__encoding, "MSKanji") == 0) 161 ret = _MSKanji_init(rl); 162 else 163 ret = EFTYPE; 164 165 if (ret == 0) { 166 if (CachedRuneLocale != NULL) { 167 /* See euc.c */ 168 if (strcmp(CachedRuneLocale->__encoding, "EUC") == 0) 169 free(CachedRuneLocale->__variable); 170 free(CachedRuneLocale); 171 } 172 CachedRuneLocale = _CurrentRuneLocale; 173 Cached__mb_cur_max = __mb_cur_max; 174 Cached__mb_sb_limit = __mb_sb_limit; 175 Cached__mbrtowc = __mbrtowc; 176 Cached__mbsinit = __mbsinit; 177 Cached__mbsnrtowcs = __mbsnrtowcs; 178 Cached__wcrtomb = __wcrtomb; 179 Cached__wcsnrtombs = __wcsnrtombs; 180 (void)strcpy(ctype_encoding, encoding); 181 } else { 182 __mbrtowc = old__mbrtowc; 183 __mbsinit = old__mbsinit; 184 __mbsnrtowcs = old__mbsnrtowcs; 185 __wcrtomb = old__wcrtomb; 186 __wcsnrtombs = old__wcsnrtombs; 187 free(rl); 188 } 189 190 return (ret); 191 } 192 193 int 194 __wrap_setrunelocale(const char *locale) 195 { 196 int ret = __setrunelocale(locale); 197 198 if (ret != 0) { 199 errno = ret; 200 return (_LDP_ERROR); 201 } 202 return (_LDP_LOADED); 203 } 204 205