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