Loading...
stdio/FreeBSD/fputws.c Libc-320 Libc-1506.40.4
--- Libc/Libc-320/stdio/FreeBSD/fputws.c
+++ Libc/Libc-1506.40.4/stdio/FreeBSD/fputws.c
@@ -1,5 +1,5 @@
 /*-
- * Copyright (c) 2002 Tim J. Robbins.
+ * Copyright (c) 2002-2004 Tim J. Robbins.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -25,29 +25,60 @@
  */
 
 #include <sys/cdefs.h>
-__FBSDID("$FreeBSD: src/lib/libc/stdio/fputws.c,v 1.4 2002/09/20 13:25:40 tjr Exp $");
+__FBSDID("$FreeBSD: src/lib/libc/stdio/fputws.c,v 1.8 2009/01/15 18:53:52 rdivacky Exp $");
+
+#include "xlocale_private.h"
 
 #include "namespace.h"
 #include <errno.h>
+#include <limits.h>
 #include <stdio.h>
 #include <wchar.h>
 #include "un-namespace.h"
+#include "fvwrite.h"
 #include "libc_private.h"
 #include "local.h"
+#include "mblocal.h"
+
+int
+fputws_l(const wchar_t * __restrict ws, FILE * __restrict fp, locale_t loc)
+{
+	size_t nbytes;
+	char buf[BUFSIZ];
+	struct __suio uio;
+	struct __siov iov;
+	const wchar_t *wsp = ws;
+	size_t (*__wcsnrtombs)(char * __restrict, const wchar_t ** __restrict,
+	    size_t, size_t, mbstate_t * __restrict, locale_t);
+
+	NORMALIZE_LOCALE(loc);
+	__wcsnrtombs = loc->__lc_ctype->__wcsnrtombs;
+	FLOCKFILE(fp);
+	ORIENT(fp, 1);
+	if (prepwrite(fp) != 0)
+		goto error;
+	uio.uio_iov = &iov;
+	uio.uio_iovcnt = 1;
+	iov.iov_base = buf;
+	do {
+		nbytes = __wcsnrtombs(buf, &wsp, SIZE_T_MAX, sizeof(buf),
+		    &fp->_mbstate, loc);
+		if (nbytes == (size_t)-1)
+			goto error;
+		iov.iov_len = uio.uio_resid = nbytes;
+		if (__sfvwrite(fp, &uio) != 0)
+			goto error;
+	} while (wsp != NULL);
+	FUNLOCKFILE(fp);
+	return (0);
+
+error:
+	FUNLOCKFILE(fp);
+	return (-1);
+}
 
 int
 fputws(const wchar_t * __restrict ws, FILE * __restrict fp)
 {
-
-	FLOCKFILE(fp);
-	ORIENT(fp, 1);
-	/* XXX Inefficient */
-	while (*ws != '\0')
-		if (__fputwc(*ws++, fp) == WEOF) {
-			FUNLOCKFILE(fp);
-			return (-1);
-		}
-	FUNLOCKFILE(fp);
-
-	return (0);
+	return fputws_l(ws, fp, __current_locale());
 }