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