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, 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, 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, void *src, int size) 152 { 153 #ifdef HAVE_ICONV 154 char *p = dst; 155 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, void *src, int size) 178 { 179 #ifdef HAVE_ICONV 180 char *p = dst; 181 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 } 224