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