contrib/smbfs/lib/smb/nls.c

   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 }