setrunelocale.c revision 175584 
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: head/lib/libc/locale/setrunelocale.c 175584 2008-01-23 01:57:26Z ache $");
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
48extern int __mb_sb_limit;
49
50extern _RuneLocale	*_Read_RuneMagi(FILE *);
51
52static int		__setrunelocale(const char *);
53
54static int
55__setrunelocale(const char *encoding)
56{
57	FILE *fp;
58	char name[PATH_MAX];
59	_RuneLocale *rl;
60	int saverr, ret;
61	static char ctype_encoding[ENCODING_LEN + 1];
62	static _RuneLocale *CachedRuneLocale;
63	static int Cached__mb_cur_max;
64	static int Cached__mb_sb_limit;
65	static size_t (*Cached__mbrtowc)(wchar_t * __restrict,
66	    const char * __restrict, size_t, mbstate_t * __restrict);
67	static size_t (*Cached__wcrtomb)(char * __restrict, wchar_t,
68	    mbstate_t * __restrict);
69	static int (*Cached__mbsinit)(const mbstate_t *);
70	static size_t (*Cached__mbsnrtowcs)(wchar_t * __restrict,
71	    const char ** __restrict, size_t, size_t, mbstate_t * __restrict);
72	static size_t (*Cached__wcsnrtombs)(char * __restrict,
73	    const wchar_t ** __restrict, size_t, size_t,
74	    mbstate_t * __restrict);
75
76	/*
77	 * The "C" and "POSIX" locale are always here.
78	 */
79	if (strcmp(encoding, "C") == 0 || strcmp(encoding, "POSIX") == 0) {
80		(void) _none_init(&_DefaultRuneLocale);
81		return (0);
82	}
83
84	/*
85	 * If the locale name is the same as our cache, use the cache.
86	 */
87	if (CachedRuneLocale != NULL &&
88	    strcmp(encoding, ctype_encoding) == 0) {
89		_CurrentRuneLocale = CachedRuneLocale;
90		__mb_cur_max = Cached__mb_cur_max;
91		__mb_sb_limit = Cached__mb_sb_limit;
92		__mbrtowc = Cached__mbrtowc;
93		__mbsinit = Cached__mbsinit;
94		__mbsnrtowcs = Cached__mbsnrtowcs;
95		__wcrtomb = Cached__wcrtomb;
96		__wcsnrtombs = Cached__wcsnrtombs;
97		return (0);
98	}
99
100	/*
101	 * Slurp the locale file into the cache.
102	 */
103
104	/* Range checking not needed, encoding length already checked before */
105	(void) strcpy(name, _PathLocale);
106	(void) strcat(name, "/");
107	(void) strcat(name, encoding);
108	(void) strcat(name, "/LC_CTYPE");
109
110	if ((fp = fopen(name, "r")) == NULL)
111		return (errno == 0 ? ENOENT : errno);
112
113	if ((rl = _Read_RuneMagi(fp)) == NULL) {
114		saverr = (errno == 0 ? EFTYPE : errno);
115		(void)fclose(fp);
116		return (saverr);
117	}
118	(void)fclose(fp);
119
120	__mbrtowc = NULL;
121	__mbsinit = NULL;
122	__mbsnrtowcs = __mbsnrtowcs_std;
123	__wcrtomb = NULL;
124	__wcsnrtombs = __wcsnrtombs_std;
125	rl->__sputrune = NULL;
126	rl->__sgetrune = NULL;
127	if (strcmp(rl->__encoding, "NONE") == 0)
128		ret = _none_init(rl);
129	else if (strcmp(rl->__encoding, "ASCII") == 0)
130		ret = _ascii_init(rl);
131	else if (strcmp(rl->__encoding, "UTF-8") == 0)
132		ret = _UTF8_init(rl);
133	else if (strcmp(rl->__encoding, "EUC") == 0)
134		ret = _EUC_init(rl);
135 	else if (strcmp(rl->__encoding, "GB18030") == 0)
136 		ret = _GB18030_init(rl);
137	else if (strcmp(rl->__encoding, "GB2312") == 0)
138		ret = _GB2312_init(rl);
139	else if (strcmp(rl->__encoding, "GBK") == 0)
140		ret = _GBK_init(rl);
141	else if (strcmp(rl->__encoding, "BIG5") == 0)
142		ret = _BIG5_init(rl);
143	else if (strcmp(rl->__encoding, "MSKanji") == 0)
144		ret = _MSKanji_init(rl);
145	else {
146		(void) _none_init(&_DefaultRuneLocale);
147		ret = EFTYPE;
148	}
149	if (ret == 0) {
150		if (CachedRuneLocale != NULL) {
151			/* See euc.c */
152			if (strcmp(CachedRuneLocale->__encoding, "EUC") == 0)
153				free(CachedRuneLocale->__variable);
154			free(CachedRuneLocale);
155		}
156		CachedRuneLocale = _CurrentRuneLocale;
157		Cached__mb_cur_max = __mb_cur_max;
158		Cached__mb_sb_limit = __mb_sb_limit;
159		Cached__mbrtowc = __mbrtowc;
160		Cached__mbsinit = __mbsinit;
161		Cached__mbsnrtowcs = __mbsnrtowcs;
162		Cached__wcrtomb = __wcrtomb;
163		Cached__wcsnrtombs = __wcsnrtombs;
164		(void)strcpy(ctype_encoding, encoding);
165	} else
166		free(rl);
167
168	return (ret);
169}
170
171int
172__wrap_setrunelocale(const char *locale)
173{
174	int ret = __setrunelocale(locale);
175
176	if (ret != 0) {
177		errno = ret;
178		return (_LDP_ERROR);
179	}
180	return (_LDP_LOADED);
181}
182
183