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/types.h>
36 #include <sys/iconv.h>
37 #include <sys/sysctl.h>
38 #include <ctype.h>
39 #ifndef APPLE
40 #include <dlfcn.h>
41 #endif
42 #include <errno.h>
43 #include <stdio.h>
44 #include <string.h>
45 #include <strings.h>
46 #include <stdlib.h>
47 #include <locale.h>
48 #include <err.h>
49 #include <netsmb/smb_lib.h>
50
51 /*
52 * prototype iconv* functions
53 */
54 typedef void *iconv_t;
55
56 static iconv_t (*my_iconv_open)(const char *, const char *);
57 static size_t(*my_iconv)(iconv_t, char **, size_t *, char **, size_t *);
58 static int(*my_iconv_close)(iconv_t);
59
60 u_char nls_lower[256];
61 u_char nls_upper[256];
62
63 static iconv_t nls_toext, nls_toloc;
64 static int iconv_loaded;
65 static void *iconv_lib;
66
67 int
nls_setlocale(const char * name)68 nls_setlocale(const char *name)
69 {
70 int i;
71
72 if (setlocale(LC_CTYPE, name) == NULL) {
73 warnx("can't set locale '%s'\n", name);
74 return EINVAL;
75 }
76 for (i = 0; i < 256; i++) {
77 nls_lower[i] = tolower(i);
78 nls_upper[i] = toupper(i);
79 }
80 return 0;
81 }
82
83 int
nls_setrecode(const char * local,const char * external)84 nls_setrecode(const char *local, const char *external)
85 {
86 #ifdef APPLE
87 return ENOENT;
88 #else
89 iconv_t icd;
90
91 if (iconv_loaded == 2)
92 return ENOENT;
93 else if (iconv_loaded == 0) {
94 iconv_loaded++;
95 iconv_lib = dlopen("libiconv.so", RTLD_LAZY | RTLD_GLOBAL);
96 if (iconv_lib == NULL) {
97 warn("Unable to load iconv library: %s\n", dlerror());
98 iconv_loaded++;
99 return ENOENT;
100 }
101 my_iconv_open = dlsym(iconv_lib, "iconv_open");
102 my_iconv = dlsym(iconv_lib, "iconv");
103 my_iconv_close = dlsym(iconv_lib, "iconv_close");
104 }
105 if (nls_toext)
106 my_iconv_close(nls_toext);
107 if (nls_toloc)
108 my_iconv_close(nls_toloc);
109 nls_toext = nls_toloc = (iconv_t)0;
110 icd = my_iconv_open(external, local);
111 if (icd == (iconv_t)-1)
112 return errno;
113 nls_toext = icd;
114 icd = my_iconv_open(local, external);
115 if (icd == (iconv_t)-1) {
116 my_iconv_close(nls_toext);
117 nls_toext = (iconv_t)0;
118 return errno;
119 }
120 nls_toloc = icd;
121 return 0;
122 #endif
123 }
124
125 char *
nls_str_toloc(char * dst,char * src)126 nls_str_toloc(char *dst, char *src)
127 {
128 char *p = dst;
129 size_t inlen, outlen;
130
131 if (!iconv_loaded)
132 return strcpy(dst, src);
133
134 if (nls_toloc == (iconv_t)0)
135 return strcpy(dst, src);
136 inlen = outlen = strlen(src);
137 my_iconv(nls_toloc, NULL, NULL, &p, &outlen);
138 while (my_iconv(nls_toloc, &src, &inlen, &p, &outlen) == -1) {
139 *p++ = *src++;
140 inlen--;
141 outlen--;
142 }
143 *p = 0;
144 return dst;
145 }
146
147 char *
nls_str_toext(char * dst,char * src)148 nls_str_toext(char *dst, char *src)
149 {
150 char *p = dst;
151 size_t inlen, outlen;
152
153 if (!iconv_loaded)
154 return strcpy(dst, src);
155
156 if (nls_toext == (iconv_t)0)
157 return strcpy(dst, src);
158 inlen = outlen = strlen(src);
159 my_iconv(nls_toext, NULL, NULL, &p, &outlen);
160 while (my_iconv(nls_toext, &src, &inlen, &p, &outlen) == -1) {
161 *p++ = *src++;
162 inlen--;
163 outlen--;
164 }
165 *p = 0;
166 return dst;
167 }
168
169 void *
nls_mem_toloc(void * dst,void * src,int size)170 nls_mem_toloc(void *dst, void *src, int size)
171 {
172 char *p = dst;
173 char *s = src;
174 size_t inlen, outlen;
175
176 if (!iconv_loaded)
177 return memcpy(dst, src, size);
178
179 if (size == 0)
180 return NULL;
181
182 if (nls_toloc == (iconv_t)0)
183 return memcpy(dst, src, size);
184 inlen = outlen = size;
185 my_iconv(nls_toloc, NULL, NULL, &p, &outlen);
186 while (my_iconv(nls_toloc, &s, &inlen, &p, &outlen) == -1) {
187 *p++ = *s++;
188 inlen--;
189 outlen--;
190 }
191 return dst;
192 }
193
194 void *
nls_mem_toext(void * dst,void * src,int size)195 nls_mem_toext(void *dst, void *src, int size)
196 {
197 char *p = dst;
198 char *s = src;
199 size_t inlen, outlen;
200
201 if (size == 0)
202 return NULL;
203
204 if (!iconv_loaded || nls_toext == (iconv_t)0)
205 return memcpy(dst, src, size);
206
207 inlen = outlen = size;
208 my_iconv(nls_toext, NULL, NULL, &p, &outlen);
209 while (my_iconv(nls_toext, &s, &inlen, &p, &outlen) == -1) {
210 *p++ = *s++;
211 inlen--;
212 outlen--;
213 }
214 return dst;
215 }
216
217 char *
nls_str_upper(char * dst,const char * src)218 nls_str_upper(char *dst, const char *src)
219 {
220 char *p = dst;
221
222 while (*src)
223 *dst++ = toupper(*src++);
224 *dst = 0;
225 return p;
226 }
227
228 char *
nls_str_lower(char * dst,const char * src)229 nls_str_lower(char *dst, const char *src)
230 {
231 char *p = dst;
232
233 while (*src)
234 *dst++ = tolower(*src++);
235 *dst = 0;
236 return p;
237 }
238