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/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 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 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 * 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 * 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 * 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 * 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 * 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 * 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