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