1 /*- 2 * Copyright (c) 1995 Alex Tatmanjants <alex@elvisti.kiev.ua> 3 * at Electronni Visti IA, Kiev, Ukraine. 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 * 27 * FreeBSD: src/lib/libc/string/strxfrm.c,v 1.15 2002/09/06 11:24:06 tjr Exp 28 * $FreeBSD: src/lib/libc/string/wcsxfrm.c,v 1.3 2004/04/07 09:47:56 tjr Exp $ 29 */ 30 31 #include <stdlib.h> 32 #include <string.h> 33 #include <wchar.h> 34 #include "collate.h" 35 36 static char *__mbsdup(const wchar_t *); 37 38 /* 39 * Placeholder wcsxfrm() implementation. See wcscoll.c for a description of 40 * the logic used. 41 */ 42 size_t 43 wcsxfrm(wchar_t * __restrict dest, const wchar_t * __restrict src, size_t len) 44 { 45 int prim, sec, l; 46 size_t slen; 47 char *mbsrc, *s, *ss; 48 49 if (*src == L'\0') { 50 if (len != 0) 51 *dest = L'\0'; 52 return (0); 53 } 54 55 if (__collate_load_error || MB_CUR_MAX > 1) { 56 slen = wcslen(src); 57 if (len > 0) { 58 if (slen < len) 59 wcscpy(dest, src); 60 else { 61 wcsncpy(dest, src, len - 1); 62 dest[len - 1] = L'\0'; 63 } 64 } 65 return (slen); 66 } 67 68 mbsrc = __mbsdup(src); 69 slen = 0; 70 prim = sec = 0; 71 ss = s = __collate_substitute(mbsrc); 72 while (*s != '\0') { 73 while (*s != '\0' && prim == 0) { 74 __collate_lookup(s, &l, &prim, &sec); 75 s += l; 76 } 77 if (prim != 0) { 78 if (len > 1) { 79 *dest++ = (wchar_t)prim; 80 len--; 81 } 82 slen++; 83 prim = 0; 84 } 85 } 86 free(ss); 87 free(mbsrc); 88 if (len != 0) 89 *dest = L'\0'; 90 91 return (slen); 92 } 93 94 static char * 95 __mbsdup(const wchar_t *ws) 96 { 97 static const mbstate_t initial; 98 mbstate_t st; 99 const wchar_t *wcp; 100 size_t len; 101 char *mbs; 102 103 wcp = ws; 104 st = initial; 105 if ((len = wcsrtombs(NULL, &wcp, 0, &st)) == (size_t)-1) 106 return (NULL); 107 if ((mbs = malloc(len + 1)) == NULL) 108 return (NULL); 109 st = initial; 110 wcsrtombs(mbs, &ws, len + 1, &st); 111 112 return (mbs); 113 } 114