1 /* $NetBSD: rune.c,v 1.43 2012/01/20 16:31:30 joerg Exp $ */ 2 3 /*- 4 * Copyright (c)2010 Citrus Project, 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 * SUCH DAMAGE. 27 */ 28 29 #include <sys/endian.h> 30 #include <sys/mman.h> 31 #include <sys/stat.h> 32 #include <assert.h> 33 #include <errno.h> 34 #include <fcntl.h> 35 #define __SETLOCALE_SOURCE__ 36 #include <locale.h> 37 #include <stddef.h> 38 #include <stdio.h> 39 #include <stdlib.h> 40 #include <string.h> 41 #include <unistd.h> 42 #include <wchar.h> 43 44 #include "setlocale_local.h" 45 46 #include "citrus_module.h" 47 #include "citrus_ctype.h" 48 49 #include "runetype_local.h" 50 51 #include "multibyte.h" 52 53 #include "_wctype_local.h" 54 #include "_wctrans_local.h" 55 56 typedef struct { 57 _RuneLocale rl; 58 unsigned char rlp_ctype_tab [_CTYPE_NUM_CHARS + 1]; 59 short rlp_tolower_tab[_CTYPE_NUM_CHARS + 1]; 60 short rlp_toupper_tab[_CTYPE_NUM_CHARS + 1]; 61 char rlp_codeset[33]; /* XXX */ 62 } _RuneLocalePriv; 63 64 static __inline void 65 _rune_wctype_init(_RuneLocale *rl) 66 { 67 memcpy(&rl->rl_wctype, &_DefaultRuneLocale.rl_wctype, 68 sizeof(rl->rl_wctype)); 69 } 70 71 static __inline void 72 _rune_wctrans_init(_RuneLocale *rl) 73 { 74 rl->rl_wctrans[_WCTRANS_INDEX_LOWER].te_name = "tolower"; 75 rl->rl_wctrans[_WCTRANS_INDEX_LOWER].te_cached = &rl->rl_maplower[0]; 76 rl->rl_wctrans[_WCTRANS_INDEX_LOWER].te_extmap = &rl->rl_maplower_ext; 77 rl->rl_wctrans[_WCTRANS_INDEX_UPPER].te_name = "toupper"; 78 rl->rl_wctrans[_WCTRANS_INDEX_UPPER].te_cached = &rl->rl_mapupper[0]; 79 rl->rl_wctrans[_WCTRANS_INDEX_UPPER].te_extmap = &rl->rl_mapupper_ext; 80 } 81 82 static __inline void 83 _rune_init_priv(_RuneLocalePriv *rlp) 84 { 85 #if _CTYPE_CACHE_SIZE != _CTYPE_NUM_CHARS 86 int i; 87 88 for (i = _CTYPE_CACHE_SIZE; i < _CTYPE_NUM_CHARS; ++i) { 89 rlp->rlp_ctype_tab [i + 1] = 0; 90 rlp->rlp_tolower_tab[i + 1] = i; 91 rlp->rlp_toupper_tab[i + 1] = i; 92 } 93 #endif 94 rlp->rlp_ctype_tab [0] = 0; 95 rlp->rlp_tolower_tab[0] = EOF; 96 rlp->rlp_toupper_tab[0] = EOF; 97 98 rlp->rl.rl_ctype_tab = (const unsigned char *)&rlp->rlp_ctype_tab[0]; 99 rlp->rl.rl_tolower_tab = (const short *)&rlp->rlp_tolower_tab[0]; 100 rlp->rl.rl_toupper_tab = (const short *)&rlp->rlp_toupper_tab[0]; 101 rlp->rl.rl_codeset = (const char *)&rlp->rlp_codeset[0]; 102 103 _rune_wctype_init(&rlp->rl); 104 _rune_wctrans_init(&rlp->rl); 105 } 106 107 static __inline void 108 _rune_find_codeset(char *s, size_t n, 109 char *var, size_t *plenvar) 110 { 111 size_t lenvar; 112 const char *endvar; 113 114 #define _RUNE_CODESET_LEN (sizeof(_RUNE_CODESET)-1) 115 116 lenvar = *plenvar; 117 for (/**/; lenvar > _RUNE_CODESET_LEN; ++var, --lenvar) { 118 if (!memcmp(var, _RUNE_CODESET, _RUNE_CODESET_LEN)) { 119 *var = '\0'; 120 *plenvar -= lenvar; 121 endvar = &var[_RUNE_CODESET_LEN]; 122 while (n-- > 1 && lenvar-- > _RUNE_CODESET_LEN) { 123 if (*endvar == ' ' || *endvar == '\t') 124 break; 125 *s++ = *endvar++; 126 } 127 break; 128 } 129 } 130 *s = '\0'; 131 } 132 133 static __inline int 134 _rune_read_file(const char * __restrict var, size_t lenvar, 135 _RuneLocale ** __restrict prl) 136 { 137 int ret, i; 138 const _FileRuneLocale *frl; 139 const _FileRuneEntry *fre; 140 const uint32_t *frune; 141 _RuneLocalePriv *rlp; 142 _RuneLocale *rl; 143 _RuneEntry *re; 144 uint32_t *rune; 145 uint32_t runetype_len, maplower_len, mapupper_len, variable_len; 146 size_t len, n; 147 148 if (lenvar < sizeof(*frl)) 149 return EFTYPE; 150 lenvar -= sizeof(*frl); 151 frl = (const _FileRuneLocale *)(const void *)var; 152 if (memcmp(_RUNECT10_MAGIC, &frl->frl_magic[0], sizeof(frl->frl_magic))) 153 return EFTYPE; 154 155 runetype_len = be32toh(frl->frl_runetype_ext.frr_nranges); 156 maplower_len = be32toh(frl->frl_maplower_ext.frr_nranges); 157 mapupper_len = be32toh(frl->frl_mapupper_ext.frr_nranges); 158 len = runetype_len + maplower_len + mapupper_len; 159 160 fre = (const _FileRuneEntry *)(const void *)(frl + 1); 161 frune = (const uint32_t *)(const void *)(fre + len); 162 163 variable_len = be32toh((uint32_t)frl->frl_variable_len); 164 165 n = (len * sizeof(*fre)) + variable_len; 166 if (lenvar < n) 167 return EFTYPE; 168 lenvar -= n; 169 170 n = sizeof(*rlp) + (len * sizeof(*re)) + lenvar; 171 rlp = (_RuneLocalePriv *)malloc(n); 172 if (rlp == NULL) 173 return ENOMEM; 174 _rune_init_priv(rlp); 175 176 rl = &rlp->rl; 177 re = (_RuneEntry *)(void *)(rlp + 1); 178 rune = (uint32_t *)(void *)(re + len); 179 180 for (i = 0; i < _CTYPE_CACHE_SIZE; ++i) { 181 rl->rl_runetype[i] = be32toh(frl->frl_runetype[i]); 182 rl->rl_maplower[i] = be32toh((uint32_t)frl->frl_maplower[i]); 183 rl->rl_mapupper[i] = be32toh((uint32_t)frl->frl_mapupper[i]); 184 } 185 186 #define READ_RANGE(name) \ 187 do { \ 188 const _FileRuneEntry *end_fre; \ 189 const uint32_t *end_frune; \ 190 \ 191 rl->rl_##name##_ext.rr_nranges = name##_len; \ 192 rl->rl_##name##_ext.rr_rune_ranges = re; \ 193 \ 194 end_fre = fre + name##_len; \ 195 while (fre < end_fre) { \ 196 re->re_min = be32toh((uint32_t)fre->fre_min); \ 197 re->re_max = be32toh((uint32_t)fre->fre_max); \ 198 re->re_map = be32toh((uint32_t)fre->fre_map); \ 199 if (re->re_map != 0) { \ 200 re->re_rune_types = NULL; \ 201 } else { \ 202 re->re_rune_types = rune; \ 203 len = re->re_max - re->re_min + 1; \ 204 n = len * sizeof(*frune); \ 205 if (lenvar < n) { \ 206 ret = EFTYPE; \ 207 goto err; \ 208 } \ 209 lenvar -= n; \ 210 end_frune = frune + len; \ 211 while (frune < end_frune) \ 212 *rune++ = be32toh(*frune++); \ 213 } \ 214 ++fre, ++re; \ 215 } \ 216 } while (/*CONSTCOND*/0) 217 218 READ_RANGE(runetype); 219 READ_RANGE(maplower); 220 READ_RANGE(mapupper); 221 222 memcpy((void *)rune, (void const *)frune, variable_len); 223 rl->rl_variable_len = variable_len; 224 rl->rl_variable = (void *)rune; 225 226 if (lenvar > 0) { 227 ret = EFTYPE; 228 goto err; 229 } 230 231 _rune_find_codeset(rlp->rlp_codeset, sizeof(rlp->rlp_codeset), 232 (char *)rl->rl_variable, &rl->rl_variable_len); 233 234 ret = _citrus_ctype_open(&rl->rl_citrus_ctype, frl->frl_encoding, 235 rl->rl_variable, rl->rl_variable_len, _PRIVSIZE); 236 if (ret) 237 goto err; 238 if (__mb_len_max_runtime < 239 _citrus_ctype_get_mb_cur_max(rl->rl_citrus_ctype)) { 240 ret = EINVAL; 241 goto err; 242 } 243 244 for (i = 0; i < _CTYPE_CACHE_SIZE; ++i) { 245 wint_t wc; 246 247 ret = _citrus_ctype_btowc(rl->rl_citrus_ctype, i, &wc); 248 if (ret) 249 goto err; 250 if (wc == WEOF) { 251 rlp->rlp_ctype_tab[i + 1] = 0; 252 rlp->rlp_tolower_tab[i + 1] = i; 253 rlp->rlp_toupper_tab[i + 1] = i; 254 } else { 255 rlp->rlp_ctype_tab[i + 1] = (unsigned char) 256 _runetype_to_ctype(_runetype_priv(rl, wc)); 257 258 #define CONVERT_MAP(name) \ 259 do { \ 260 wint_t map; \ 261 int c; \ 262 \ 263 map = _towctrans_priv(wc, _wctrans_##name(rl)); \ 264 if (map == wc || (_citrus_ctype_wctob(rl->rl_citrus_ctype, \ 265 map, &c) || c == EOF)) \ 266 c = i; \ 267 rlp->rlp_to##name##_tab[i + 1] = (short)c; \ 268 } while (/*CONSTCOND*/0) 269 270 CONVERT_MAP(lower); 271 CONVERT_MAP(upper); 272 } 273 } 274 *prl = rl; 275 return 0; 276 277 err: 278 free(rlp); 279 return ret; 280 } 281 282 int 283 _rune_load(const char * __restrict var, size_t lenvar, 284 _RuneLocale ** __restrict prl) 285 { 286 int ret; 287 288 _DIAGASSERT(var != NULL || lenvar < 1); 289 _DIAGASSERT(prl != NULL); 290 291 if (lenvar < 1) 292 return EFTYPE; 293 switch (*var) { 294 case 'R': 295 ret = _rune_read_file(var, lenvar, prl); 296 break; 297 default: 298 ret = EFTYPE; 299 } 300 return ret; 301 } 302