test mbrtowc(3) in conjunction with mbsinit(3)

2 files changed, 93 insertions, 0 deletions

diff --git a/regress/lib/libc/locale/mbrtowc/Makefile b/regress/lib/libc/locale/mbrtowc/Makefile
new file mode 100644
index 00000000000..0aa1026e0a9
--- /dev/null
+++ b/regress/lib/libc/locale/mbrtowc/Makefile
@@ -0,0 +1,9 @@
+# $OpenBSD: Makefile,v 1.1 2016/09/03 18:14:18 schwarze Exp $
+
+PROG = test_mbrtowc
+NOMAN =
+
+regress: ${PROG}
+	./${PROG}
+
+.include <bsd.prog.mk>
diff --git a/regress/lib/libc/locale/mbrtowc/test_mbrtowc.c b/regress/lib/libc/locale/mbrtowc/test_mbrtowc.c
new file mode 100644
index 00000000000..5794791ffbe
--- /dev/null
+++ b/regress/lib/libc/locale/mbrtowc/test_mbrtowc.c
@@ -0,0 +1,84 @@
+/*
+ * Copyright (c) 2016 Ingo Schwarze <schwarze@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <sys/types.h>
+#include <err.h>
+#include <errno.h>
+#include <locale.h>
+#include <stdlib.h>
+#include <string.h>
+#include <wchar.h>
+
+static mbstate_t	 mbs;
+
+void
+onetest(const char *name, const char *in, size_t insz,
+    int outerr, size_t outsz, wint_t out)
+{
+	wchar_t		 wc;
+	size_t		 sz;
+
+	sz = mbrtowc(&wc, in, insz, &mbs);
+	if (errno != outerr)
+		err(1, "%d %s(%zd)", MB_CUR_MAX, name, insz);
+	if (sz != outsz || (out != WEOF && wc != out))
+		errx(1, "%d %s(%zd) = (%zd, %d) != (%zd, %d)",
+		    MB_CUR_MAX, name, insz, sz, wc, outsz, out);
+	if (mbsinit(&mbs) == (insz && outsz == (size_t)-2))
+		errx(1, "%d %s(%zd) mbsinit", MB_CUR_MAX, name, insz);
+	if (errno == 0 && outerr == 0)
+		return;
+	errno = 0;
+	memset(&mbs, 0, sizeof(mbs));
+}
+
+int
+main(void)
+{
+	onetest("NUL", "", 0, 0, -2, WEOF);
+	onetest("NUL", "", 2, 0, 0, L'\0');
+	onetest("BEL", "\a", 2, 0, 1, L'\a');
+	onetest("A", "A", 2, 0, 1, L'A');
+	onetest("DEL", "\177", 2, 0, 1, L'\177');
+	onetest("CSI", "\233", 2, 0, 1, L'\233');
+
+	if (setlocale(LC_CTYPE, "en_US.UTF-8") == NULL)
+		errx(1, "setlocale(UTF-8) failed"),
+
+	onetest("NUL", "", 0, 0, -2, WEOF);
+	onetest("NUL", "", 8, 0, 0, L'\0');
+	onetest("BEL", "\a", 8, 0, 1, L'\a');
+	onetest("A", "A", 8, 0, 1, L'A');
+	onetest("DEL", "\177", 8, 0, 1, L'\177');
+	onetest("0x80", "\200", 8, EILSEQ, -1, WEOF);
+	onetest("0xc3", "\303", 1, 0, -2, WEOF);
+	onetest("U+00E9", "\251", 8, 0, 1, 0xe9);
+	onetest("0xec", "\354", 1, 0, -2, WEOF);
+	onetest("0xecbf", "\277", 1, 0, -2, WEOF);
+	onetest("U+CFFF", "\277", 8, 0, 1, 0xcfff);
+
+	if (setlocale(LC_CTYPE, "POSIX") == NULL)
+		errx(1, "setlocale(POSIX) failed"),
+
+	onetest("0xff", "\277", 2, 0, 1, L'\277');
+
+	if (setlocale(LC_CTYPE, "en_US.UTF-8") == NULL)
+		errx(1, "second setlocale(UTF-8) failed"),
+
+	onetest("U+13000", "\360\223\200\200", 8, 0, 4, 0x13000);
+
+	return 0;
+}