xref: /dragonfly/lib/libc/locale/setrunelocale.c (revision 67640b13)
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  * Copyright (c) 2011 The FreeBSD Foundation
9  * All rights reserved.
10  * Portions of this software were developed by David Chisnall
11  * under sponsorship from the FreeBSD Foundation.
12  *
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  * 4. Neither the name of the University nor the names of its contributors
22  *    may be used to endorse or promote products derived from this software
23  *    without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35  * SUCH DAMAGE.
36  *
37  * $FreeBSD: head/lib/libc/locale/setrunelocale.c 264038 2014-04-02 11:10:46Z theraven $
38  */
39 
40 
41 #define __RUNETYPE_INTERNAL 1
42 
43 #include <runetype.h>
44 #include <errno.h>
45 #include <limits.h>
46 #include <string.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <unistd.h>
50 #include <wchar.h>
51 #include "ldpart.h"
52 #include "mblocal.h"
53 #include "setlocale.h"
54 
55 #undef _CurrentRuneLocale
56 extern _RuneLocale const *_CurrentRuneLocale;
57 #ifndef __NO_TLS
58 /*
59  * A cached version of the runes for this thread.  Used by ctype.h
60  */
61 __thread const _RuneLocale *_ThreadRuneLocale;
62 #endif
63 
64 extern int __mb_sb_limit;
65 
66 extern _RuneLocale	*_Read_RuneMagi(const char *);
67 
68 static int		__setrunelocale(struct xlocale_ctype *l, const char *);
69 
70 static void
71 destruct_ctype(void *v)
72 {
73 	struct xlocale_ctype *l = v;
74 
75 	if (&_DefaultRuneLocale != l->runes)
76 		free(l->runes);
77 	free(l);
78 }
79 
80 const _RuneLocale *
81 __getCurrentRuneLocale(void)
82 {
83 
84 	return XLOCALE_CTYPE(__get_locale())->runes;
85 }
86 
87 static void
88 free_runes(_RuneLocale *rl)
89 {
90 	if ((rl != &_DefaultRuneLocale) && (rl)) {
91 		free(rl);
92 	}
93 }
94 
95 static int
96 __setrunelocale(struct xlocale_ctype *l, const char *encoding)
97 {
98 	_RuneLocale *rl;
99 	int ret;
100 	char path[PATH_MAX];
101 	struct xlocale_ctype saved = *l;
102 
103 	/*
104 	 * The "C" and "POSIX" locale are always here.
105 	 */
106 	if (strcmp(encoding, "C") == 0 || strcmp(encoding, "POSIX") == 0) {
107 		free_runes(saved.runes);
108 		(void) _none_init(l, (_RuneLocale*)&_DefaultRuneLocale);
109 		return (0);
110 	}
111 
112 	/* Range checking not needed, encoding length already checked before */
113 	(void) snprintf(path, sizeof (path), "%s/%s/LC_CTYPE",
114 	    _PathLocale, encoding);
115 
116 	if ((rl = _Read_RuneMagi(path)) == NULL) {
117 		errno = EINVAL;
118 		return (errno);
119 	}
120 
121 	l->__mbrtowc = NULL;
122 	l->__mbsinit = NULL;
123 	l->__mbsnrtowcs = NULL;
124 	l->__wcrtomb = NULL;
125 	l->__wcsnrtombs = NULL;
126 
127 	rl->__sputrune = NULL;
128 	rl->__sgetrune = NULL;
129 	if (strcmp(rl->__encoding, "NONE:US-ASCII") == 0)
130 		ret = _ascii_init(l, rl);
131 	else if (strncmp(rl->__encoding, "NONE", 4) == 0)
132 		ret = _none_init(l, rl);
133 	else if (strcmp(rl->__encoding, "UTF-8") == 0)
134 		ret = _UTF8_init(l, rl);
135 	else if (strcmp(rl->__encoding, "EUC-CN") == 0)
136 		ret = _EUC_CN_init(l, rl);
137 	else if (strcmp(rl->__encoding, "EUC-JP") == 0)
138 		ret = _EUC_JP_init(l, rl);
139 	else if (strcmp(rl->__encoding, "EUC-KR") == 0)
140 		ret = _EUC_KR_init(l, rl);
141 	else if (strcmp(rl->__encoding, "EUC-TW") == 0)
142 		ret = _EUC_TW_init(l, rl);
143 	else if (strcmp(rl->__encoding, "GB18030") == 0)
144  		ret = _GB18030_init(l, rl);
145 	else if (strcmp(rl->__encoding, "GB2312") == 0)
146 		ret = _GB2312_init(l, rl);
147 	else if (strcmp(rl->__encoding, "GBK") == 0)
148 		ret = _GBK_init(l, rl);
149 	else if (strcmp(rl->__encoding, "BIG5") == 0)
150 		ret = _BIG5_init(l, rl);
151 	else if (strcmp(rl->__encoding, "MSKanji") == 0)
152 		ret = _MSKanji_init(l, rl);
153 	else
154 		ret = EFTYPE;
155 
156 	if (ret == 0) {
157 		/* Free the old runes if it exists. */
158 		free_runes(saved.runes);
159 	} else {
160 		/* Restore the saved version if this failed. */
161 		memcpy(l, &saved, sizeof(struct xlocale_ctype));
162 		free(rl);
163 	}
164 
165 	return (ret);
166 }
167 
168 int
169 __wrap_setrunelocale(const char *locale)
170 {
171 	int ret = __setrunelocale(&__xlocale_global_ctype, locale);
172 
173 	if (ret != 0) {
174 		errno = ret;
175 		return (_LDP_ERROR);
176 	}
177 	__mb_cur_max = __xlocale_global_ctype.__mb_cur_max;
178 	__mb_sb_limit = __xlocale_global_ctype.__mb_sb_limit;
179 	_CurrentRuneLocale = __xlocale_global_ctype.runes;
180 	return (_LDP_LOADED);
181 }
182 
183 #ifndef __NO_TLS
184 void
185 __set_thread_rune_locale(locale_t loc)
186 {
187 
188 	if (loc == NULL) {
189 		_ThreadRuneLocale = &_DefaultRuneLocale;
190 	} else if (loc == LC_GLOBAL_LOCALE) {
191 		_ThreadRuneLocale = 0;
192 	} else {
193 		_ThreadRuneLocale = XLOCALE_CTYPE(loc)->runes;
194 	}
195 }
196 #endif
197 
198 void *
199 __ctype_load(const char *locale, __unused locale_t unused)
200 {
201 	struct xlocale_ctype *l = calloc(sizeof(struct xlocale_ctype), 1);
202 
203 	l->header.header.destructor = destruct_ctype;
204 	if (__setrunelocale(l, locale))
205 	{
206 		free(l);
207 		return NULL;
208 	}
209 	return l;
210 }
211