nls.c revision 136699 
1/*
2 * Copyright (c) 2000-2001, Boris Popov
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *    This product includes software developed by Boris Popov.
16 * 4. Neither the name of the author nor the names of any co-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 AUTHOR 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 AUTHOR 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 * $Id: nls.c,v 1.10 2002/07/22 08:33:59 bp Exp $
33 */
34
35#include <sys/cdefs.h>
36__FBSDID("$FreeBSD: head/contrib/smbfs/lib/smb/nls.c 136699 2004-10-19 17:39:15Z obrien $");
37
38#include <sys/types.h>
39#include <sys/iconv.h>
40#include <sys/sysctl.h>
41#include <ctype.h>
42#ifndef APPLE
43#include <dlfcn.h>
44#endif
45#include <errno.h>
46#include <stdio.h>
47#include <string.h>
48#include <stdlib.h>
49#include <locale.h>
50#include <err.h>
51#include <netsmb/smb_lib.h>
52
53/*
54 * prototype iconv* functions
55 */
56typedef void *iconv_t;
57
58static iconv_t (*my_iconv_open)(const char *, const char *);
59static size_t(*my_iconv)(iconv_t, const char **, size_t *, char **, size_t *);
60static int(*my_iconv_close)(iconv_t);
61
62u_char nls_lower[256];
63u_char nls_upper[256];
64
65static iconv_t nls_toext, nls_toloc;
66static int iconv_loaded;
67static void *iconv_lib;
68
69int
70nls_setlocale(const char *name)
71{
72	int i;
73
74	if (setlocale(LC_CTYPE, name) == NULL) {
75		warnx("can't set locale '%s'\n", name);
76		return EINVAL;
77	}
78	for (i = 0; i < 256; i++) {
79		nls_lower[i] = tolower(i);
80		nls_upper[i] = toupper(i);
81	}
82	return 0;
83}
84
85int
86nls_setrecode(const char *local, const char *external)
87{
88#ifdef APPLE
89	return ENOENT;
90#else
91	iconv_t icd;
92
93	if (iconv_loaded == 2)
94		return ENOENT;
95	else if (iconv_loaded == 0) {
96		iconv_loaded++;
97		iconv_lib = dlopen("libiconv.so", RTLD_LAZY | RTLD_GLOBAL);
98		if (iconv_lib == NULL) {
99			warn("Unable to load iconv library: %s\n", dlerror());
100			iconv_loaded++;
101			return ENOENT;
102		}
103		my_iconv_open = dlsym(iconv_lib, "iconv_open");
104		my_iconv = dlsym(iconv_lib, "iconv");
105		my_iconv_close = dlsym(iconv_lib, "iconv_close");
106	}
107	if (nls_toext)
108		my_iconv_close(nls_toext);
109	if (nls_toloc)
110		my_iconv_close(nls_toloc);
111	nls_toext = nls_toloc = (iconv_t)0;
112	icd = my_iconv_open(external, local);
113	if (icd == (iconv_t)-1)
114		return errno;
115	nls_toext = icd;
116	icd = my_iconv_open(local, external);
117	if (icd == (iconv_t)-1) {
118		my_iconv_close(nls_toext);
119		nls_toext = (iconv_t)0;
120		return errno;
121	}
122	nls_toloc = icd;
123	return 0;
124#endif
125}
126
127char *
128nls_str_toloc(char *dst, const char *src)
129{
130	char *p = dst;
131	size_t inlen, outlen;
132
133	if (!iconv_loaded)
134		return strcpy(dst, src);
135
136	if (nls_toloc == (iconv_t)0)
137		return strcpy(dst, src);
138	inlen = outlen = strlen(src);
139	my_iconv(nls_toloc, NULL, NULL, &p, &outlen);
140	while (my_iconv(nls_toloc, &src, &inlen, &p, &outlen) == -1) {
141		*p++ = *src++;
142		inlen--;
143		outlen--;
144	}
145	*p = 0;
146	return dst;
147}
148
149char *
150nls_str_toext(char *dst, const char *src)
151{
152	char *p = dst;
153	size_t inlen, outlen;
154
155	if (!iconv_loaded)
156		return strcpy(dst, src);
157
158	if (nls_toext == (iconv_t)0)
159		return strcpy(dst, src);
160	inlen = outlen = strlen(src);
161	my_iconv(nls_toext, NULL, NULL, &p, &outlen);
162	while (my_iconv(nls_toext, &src, &inlen, &p, &outlen) == -1) {
163		*p++ = *src++;
164		inlen--;
165		outlen--;
166	}
167	*p = 0;
168	return dst;
169}
170
171void *
172nls_mem_toloc(void *dst, const void *src, int size)
173{
174	char *p = dst;
175	const char *s = src;
176	size_t inlen, outlen;
177
178	if (!iconv_loaded)
179		return memcpy(dst, src, size);
180
181	if (size == 0)
182		return NULL;
183
184	if (nls_toloc == (iconv_t)0)
185		return memcpy(dst, src, size);
186	inlen = outlen = size;
187	my_iconv(nls_toloc, NULL, NULL, &p, &outlen);
188	while (my_iconv(nls_toloc, &s, &inlen, &p, &outlen) == -1) {
189		*p++ = *s++;
190		inlen--;
191		outlen--;
192	}
193	return dst;
194}
195
196void *
197nls_mem_toext(void *dst, const void *src, int size)
198{
199	char *p = dst;
200	const char *s = src;
201	size_t inlen, outlen;
202
203	if (size == 0)
204		return NULL;
205
206	if (!iconv_loaded || nls_toext == (iconv_t)0)
207		return memcpy(dst, src, size);
208
209	inlen = outlen = size;
210	my_iconv(nls_toext, NULL, NULL, &p, &outlen);
211	while (my_iconv(nls_toext, &s, &inlen, &p, &outlen) == -1) {
212		*p++ = *s++;
213		inlen--;
214		outlen--;
215	}
216	return dst;
217}
218
219char *
220nls_str_upper(char *dst, const char *src)
221{
222	char *p = dst;
223
224	while (*src)
225		*dst++ = toupper(*src++);
226	*dst = 0;
227	return p;
228}
229
230char *
231nls_str_lower(char *dst, const char *src)
232{
233	char *p = dst;
234
235	while (*src)
236		*dst++ = tolower(*src++);
237	*dst = 0;
238	return p;
239}
240