1 /*- 2 * Copyright (c) 1995 Alex Tatmanjants <alex@elvisti.kiev.ua> 3 * at Electronni Visti IA, Kiev, Ukraine. 4 * All rights reserved. 5 * 6 * Copyright (c) 2011 The FreeBSD Foundation 7 * All rights reserved. 8 * Portions of this software were developed by David Chisnall 9 * under sponsorship from the FreeBSD Foundation. 10 * 11 * Redistribution and use in source and binary forms, with or without 12 * modification, are permitted provided that the following conditions 13 * are met: 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 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 * $FreeBSD: head/lib/libc/string/wcsxfrm.c 227753 2011-11-20 14:45:42Z theraven $ 33 */ 34 35 #include <stdlib.h> 36 #include <string.h> 37 #include <wchar.h> 38 #include "collate.h" 39 40 static char *__mbsdup(const wchar_t *); 41 42 /* 43 * Placeholder wcsxfrm() implementation. See wcscoll.c for a description of 44 * the logic used. 45 */ 46 size_t 47 wcsxfrm_l(wchar_t * __restrict dest, const wchar_t * __restrict src, size_t len, locale_t locale) 48 { 49 int prim, sec, l; 50 size_t slen; 51 char *mbsrc, *s, *ss; 52 FIX_LOCALE(locale); 53 struct xlocale_collate *table = 54 (struct xlocale_collate*)locale->components[XLC_COLLATE]; 55 56 if (*src == L'\0') { 57 if (len != 0) 58 *dest = L'\0'; 59 return (0); 60 } 61 62 if (table->__collate_load_error || MB_CUR_MAX > 1) { 63 slen = wcslen(src); 64 if (len > 0) { 65 if (slen < len) 66 wcscpy(dest, src); 67 else { 68 wcsncpy(dest, src, len - 1); 69 dest[len - 1] = L'\0'; 70 } 71 } 72 return (slen); 73 } 74 75 mbsrc = __mbsdup(src); 76 slen = 0; 77 prim = sec = 0; 78 ss = s = __collate_substitute(table, mbsrc); 79 while (*s != '\0') { 80 while (*s != '\0' && prim == 0) { 81 __collate_lookup(table, s, &l, &prim, &sec); 82 s += l; 83 } 84 if (prim != 0) { 85 if (len > 1) { 86 *dest++ = (wchar_t)prim; 87 len--; 88 } 89 slen++; 90 prim = 0; 91 } 92 } 93 free(ss); 94 free(mbsrc); 95 if (len != 0) 96 *dest = L'\0'; 97 98 return (slen); 99 } 100 size_t 101 wcsxfrm(wchar_t * __restrict dest, const wchar_t * __restrict src, size_t len) 102 { 103 return wcsxfrm_l(dest, src, len, __get_locale()); 104 } 105 106 static char * 107 __mbsdup(const wchar_t *ws) 108 { 109 static const mbstate_t initial; 110 mbstate_t st; 111 const wchar_t *wcp; 112 size_t len; 113 char *mbs; 114 115 wcp = ws; 116 st = initial; 117 if ((len = wcsrtombs(NULL, &wcp, 0, &st)) == (size_t)-1) 118 return (NULL); 119 if ((mbs = malloc(len + 1)) == NULL) 120 return (NULL); 121 st = initial; 122 wcsrtombs(mbs, &ws, len + 1, &st); 123 124 return (mbs); 125 } 126