xref: /freebsd/lib/libc/locale/setrunelocale.c (revision 39beb93c3f8bdbf72a61fda42300b5ebed7390c8)
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