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, const 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 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 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 * 126 nls_str_toloc(char *dst, const 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 * 148 nls_str_toext(char *dst, const 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 * 170 nls_mem_toloc(void *dst, const void *src, int size) 171 { 172 char *p = dst; 173 const 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 * 195 nls_mem_toext(void *dst, const void *src, int size) 196 { 197 char *p = dst; 198 const 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 * 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 * 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