Loading...
string/FreeBSD/wcsxfrm.c Libc-1725.40.4 Libc-498
--- Libc/Libc-1725.40.4/string/FreeBSD/wcsxfrm.c
+++ Libc/Libc-498/string/FreeBSD/wcsxfrm.c
@@ -1,15 +1,7 @@
 /*-
- * SPDX-License-Identifier: BSD-2-Clause
- *
- * Copyright 2010 Nexenta Systems, Inc.  All rights reserved.
  * Copyright (c) 1995 Alex Tatmanjants <alex@elvisti.kiev.ua>
  *		at Electronni Visti IA, Kiev, Ukraine.
  *			All rights reserved.
- *
- * Copyright (c) 2011 The FreeBSD Foundation
- *
- * Portions of this software were developed by David Chisnall
- * under sponsorship from the FreeBSD Foundation.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
@@ -33,25 +25,29 @@
  * SUCH DAMAGE.
  */
 
-#include <errno.h>
+#include <sys/cdefs.h>
+#if 0
+__FBSDID("FreeBSD: src/lib/libc/string/strxfrm.c,v 1.15 2002/09/06 11:24:06 tjr Exp ");
+#endif
+__FBSDID("$FreeBSD: src/lib/libc/string/wcsxfrm.c,v 1.3 2004/04/07 09:47:56 tjr Exp $");
+
 #include <stdlib.h>
 #include <string.h>
 #include <wchar.h>
 #include "collate.h"
 
+static char *__mbsdup(const wchar_t *);
+
+/*
+ * Placeholder wcsxfrm() implementation. See wcscoll.c for a description of
+ * the logic used.
+ */
 size_t
-wcsxfrm_l(wchar_t * __restrict dest, const wchar_t * __restrict src, size_t len, locale_t locale)
+wcsxfrm(wchar_t * __restrict dest, const wchar_t * __restrict src, size_t len)
 {
+	int prim, sec, l;
 	size_t slen;
-#ifdef __APPLE__
-	int serrno;
-	NORMALIZE_LOCALE(locale);
-	struct xlocale_collate *table = XLOCALE_COLLATE(locale);
-#else
-	FIX_LOCALE(locale);
-	struct xlocale_collate *table =
-		(struct xlocale_collate*)locale->components[XLC_COLLATE];
-#endif
+	char *mbsrc, *s, *ss;
 
 	if (*src == L'\0') {
 		if (len != 0)
@@ -59,43 +55,62 @@
 		return (0);
 	}
 
-	if ((table->__collate_load_error) ||
-	    ((slen = _collate_wxfrm(table, src, dest, len)) == (size_t)-1)) {
-		goto error;
+	if (__collate_load_error || MB_CUR_MAX > 1) {
+		slen = wcslen(src);
+		if (len > 0) {
+			if (slen < len)
+				wcscpy(dest, src);
+			else {
+				wcsncpy(dest, src, len - 1);
+				dest[len - 1] = L'\0';
+			}
+		}
+		return (slen);
 	}
 
-	/* Add null termination at the correct location. */
-	if (len > slen) {
-		dest[slen] = 0;
-	} else if (len) {
-		dest[len-1] = 0;
+	mbsrc = __mbsdup(src);
+	slen = 0;
+	prim = sec = 0;
+	ss = s = __collate_substitute(mbsrc);
+	while (*s != '\0') {
+		while (*s != '\0' && prim == 0) {
+			__collate_lookup(s, &l, &prim, &sec);
+			s += l;
+		}
+		if (prim != 0) {
+			if (len > 1) {
+				*dest++ = (wchar_t)prim;
+				len--;
+			}
+			slen++;
+			prim = 0;
+		}
 	}
+	free(ss);
+	free(mbsrc);
+	if (len != 0)
+		*dest = L'\0';
 
-	return (slen);
-
-error:
-#ifdef __APPLE__
-	serrno = errno;
-#endif
-	slen = wcslen(src);
-	if (slen < len)
-		(void) wcscpy(dest, src);
-	else if (len > 0) {
-		(void) wcsncpy(dest, src, len - 1);
-		dest[len - 1] = L'\0';
-	}
-#ifdef __APPLE__
-	errno = serrno;
-#endif
 	return (slen);
 }
 
-size_t
-wcsxfrm(wchar_t * __restrict dest, const wchar_t * __restrict src, size_t len)
+static char *
+__mbsdup(const wchar_t *ws)
 {
-#ifdef __APPLE__
-	return wcsxfrm_l(dest, src, len, __current_locale());
-#else
-	return wcsxfrm_l(dest, src, len, __get_locale());
-#endif
+	static const mbstate_t initial;
+	mbstate_t st;
+	const wchar_t *wcp;
+	size_t len;
+	char *mbs;
+
+	wcp = ws;
+	st = initial;
+	if ((len = wcsrtombs(NULL, &wcp, 0, &st)) == (size_t)-1)
+		return (NULL);
+	if ((mbs = malloc(len + 1)) == NULL)
+		return (NULL);
+	st = initial;
+	wcsrtombs(mbs, &ws, len + 1, &st);
+
+	return (mbs);
 }