implement [v]swprintf
authorRich Felker <dalias@aerifal.cx>
Fri, 18 Mar 2011 13:19:09 +0000 (09:19 -0400)
committerRich Felker <dalias@aerifal.cx>
Fri, 18 Mar 2011 13:19:09 +0000 (09:19 -0400)
include/wchar.h
src/internal/stdio_impl.h
src/stdio/swprintf.c [new file with mode: 0644]
src/stdio/vswprintf.c [new file with mode: 0644]

index 53d619b..74f830d 100644 (file)
@@ -98,11 +98,11 @@ int fwide (FILE *, int);
 
 int wprintf (const wchar_t *, ...);
 int fwprintf (FILE *, const wchar_t *, ...);
-int swprintf (wchar_t *, const wchar_t *, ...);
+int swprintf (wchar_t *, size_t, const wchar_t *, ...);
 
 int vwprintf (const wchar_t *, va_list);
 int vfwprintf (FILE *, const wchar_t *, va_list);
-int vswprintf (wchar_t *, const wchar_t *, va_list);
+int vswprintf (wchar_t *, size_t, const wchar_t *, va_list);
 
 int wscanf (const wchar_t *, ...);
 int fwscanf (FILE *, const wchar_t *, ...);
index 13e5cfd..4e6d350 100644 (file)
@@ -50,7 +50,7 @@ struct __FILE_s {
        signed char lbf;
        int lock;
        int lockcount;
-       void *dummy5;
+       void *cookie;
        off_t off;
        int (*flush)(FILE *);
        void **wide_data; /* must be NULL */
diff --git a/src/stdio/swprintf.c b/src/stdio/swprintf.c
new file mode 100644 (file)
index 0000000..5ece97c
--- /dev/null
@@ -0,0 +1,14 @@
+#include <stdio.h>
+#include <stdarg.h>
+#include <wchar.h>
+
+int swprintf(wchar_t *s, size_t n, const wchar_t *fmt, ...)
+{
+       int ret;
+       va_list ap;
+       va_start(ap, fmt);
+       ret = vswprintf(s, n, fmt, ap);
+       va_end(ap);
+       return ret;
+}
+
diff --git a/src/stdio/vswprintf.c b/src/stdio/vswprintf.c
new file mode 100644 (file)
index 0000000..4ad581f
--- /dev/null
@@ -0,0 +1,46 @@
+#include "stdio_impl.h"
+
+struct cookie {
+       wchar_t *ws;
+       size_t l;
+};
+
+static size_t sw_write(FILE *f, const unsigned char *s, size_t l)
+{
+       size_t l0 = l;
+       int i = 0;
+       struct cookie *c = f->cookie;
+       while (c->l && l && (i=mbtowc(c->ws, s, l))>=0) {
+               s+=i;
+               l-=i;
+               c->l--;
+               c->ws++;
+       }
+       *c->ws = 0;
+       return i<0 ? i : l0;
+}
+
+int vswprintf(wchar_t *s, size_t n, const wchar_t *fmt, va_list ap)
+{
+       int r;
+       FILE f;
+       unsigned char buf[256];
+       struct cookie c = { s, n-1 };
+
+       memset(&f, 0, sizeof(FILE));
+       f.lbf = EOF;
+       f.write = sw_write;
+       f.buf_size = sizeof buf;
+       f.buf = buf;
+       f.owner = -1;
+       f.cookie = &c;
+       if (!n) {
+               return -1;
+       } else if (n > INT_MAX) {
+               errno = EOVERFLOW;
+               return -1;
+       }
+       r = vfwprintf(&f, fmt, ap);
+       __oflow(&f);
+       return r>=n ? -1 : r;
+}