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: stable/9/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 */
56 typedef void *iconv_t;
57
58 static iconv_t (*my_iconv_open)(const char *, const char *);
59 static size_t(*my_iconv)(iconv_t, const char **, size_t *, char **, size_t *);
60 static int(*my_iconv_close)(iconv_t);
61
62 u_char nls_lower[256];
63 u_char nls_upper[256];
64
65 static iconv_t nls_toext, nls_toloc;
66 static int iconv_loaded;
67 static void *iconv_lib;
68
69 int
nls_setlocale(const char * name)70 nls_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
85 int
nls_setrecode(const char * local,const char * external)86 nls_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
127 char *
nls_str_toloc(char * dst,const char * src)128 nls_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
149 char *
nls_str_toext(char * dst,const char * src)150 nls_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
171 void *
nls_mem_toloc(void * dst,const void * src,int size)172 nls_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
196 void *
nls_mem_toext(void * dst,const void * src,int size)197 nls_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
219 char *
nls_str_upper(char * dst,const char * src)220 nls_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
230 char *
nls_str_lower(char * dst,const char * src)231 nls_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