1*f65a228fSRobert Mustacchi /*
2*f65a228fSRobert Mustacchi * Copyright (c) 2016 Ingo Schwarze <schwarze@openbsd.org>
3*f65a228fSRobert Mustacchi *
4*f65a228fSRobert Mustacchi * Permission to use, copy, modify, and distribute this software for any
5*f65a228fSRobert Mustacchi * purpose with or without fee is hereby granted, provided that the above
6*f65a228fSRobert Mustacchi * copyright notice and this permission notice appear in all copies.
7*f65a228fSRobert Mustacchi *
8*f65a228fSRobert Mustacchi * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9*f65a228fSRobert Mustacchi * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10*f65a228fSRobert Mustacchi * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11*f65a228fSRobert Mustacchi * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12*f65a228fSRobert Mustacchi * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13*f65a228fSRobert Mustacchi * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14*f65a228fSRobert Mustacchi * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15*f65a228fSRobert Mustacchi */
16*f65a228fSRobert Mustacchi
17*f65a228fSRobert Mustacchi #include <sys/types.h>
18*f65a228fSRobert Mustacchi #include <err.h>
19*f65a228fSRobert Mustacchi #include <errno.h>
20*f65a228fSRobert Mustacchi #include <locale.h>
21*f65a228fSRobert Mustacchi #include <stdlib.h>
22*f65a228fSRobert Mustacchi #include <string.h>
23*f65a228fSRobert Mustacchi #include <wchar.h>
24*f65a228fSRobert Mustacchi
25*f65a228fSRobert Mustacchi static mbstate_t mbs;
26*f65a228fSRobert Mustacchi
27*f65a228fSRobert Mustacchi void
onetest(const char * name,const char * in,size_t insz,int outerr,size_t outsz,wint_t out)28*f65a228fSRobert Mustacchi onetest(const char *name, const char *in, size_t insz,
29*f65a228fSRobert Mustacchi int outerr, size_t outsz, wint_t out)
30*f65a228fSRobert Mustacchi {
31*f65a228fSRobert Mustacchi wchar_t wc;
32*f65a228fSRobert Mustacchi size_t sz;
33*f65a228fSRobert Mustacchi
34*f65a228fSRobert Mustacchi sz = mbrtowc(&wc, in, insz, &mbs);
35*f65a228fSRobert Mustacchi if (errno != outerr)
36*f65a228fSRobert Mustacchi err(1, "%zu %s(%zd)", MB_CUR_MAX, name, insz);
37*f65a228fSRobert Mustacchi if (sz != outsz || (out != WEOF && wc != out))
38*f65a228fSRobert Mustacchi errx(1, "%zu %s(%zd) = (%zd, %d) != (%zd, %d)",
39*f65a228fSRobert Mustacchi MB_CUR_MAX, name, insz, sz, wc, outsz, out);
40*f65a228fSRobert Mustacchi if (mbsinit(&mbs) == (insz && outsz == (size_t)-2))
41*f65a228fSRobert Mustacchi errx(1, "%zu %s(%zd) mbsinit", MB_CUR_MAX, name, insz);
42*f65a228fSRobert Mustacchi if (errno == 0 && outerr == 0)
43*f65a228fSRobert Mustacchi return;
44*f65a228fSRobert Mustacchi errno = 0;
45*f65a228fSRobert Mustacchi memset(&mbs, 0, sizeof (mbs));
46*f65a228fSRobert Mustacchi }
47*f65a228fSRobert Mustacchi
48*f65a228fSRobert Mustacchi int
main(void)49*f65a228fSRobert Mustacchi main(void)
50*f65a228fSRobert Mustacchi {
51*f65a228fSRobert Mustacchi onetest("NUL", "", 0, 0, -2, WEOF);
52*f65a228fSRobert Mustacchi onetest("NUL", "", 2, 0, 0, L'\0');
53*f65a228fSRobert Mustacchi onetest("BEL", "\a", 2, 0, 1, L'\a');
54*f65a228fSRobert Mustacchi onetest("A", "A", 2, 0, 1, L'A');
55*f65a228fSRobert Mustacchi onetest("DEL", "\177", 2, 0, 1, L'\177');
56*f65a228fSRobert Mustacchi onetest("CSI", "\233", 2, 0, 1, L'\233');
57*f65a228fSRobert Mustacchi
58*f65a228fSRobert Mustacchi if (setlocale(LC_CTYPE, "en_US.UTF-8") == NULL)
59*f65a228fSRobert Mustacchi errx(1, "setlocale(UTF-8) failed");
60*f65a228fSRobert Mustacchi
61*f65a228fSRobert Mustacchi onetest("NUL", "", 0, 0, -2, WEOF);
62*f65a228fSRobert Mustacchi onetest("NUL", "", 8, 0, 0, L'\0');
63*f65a228fSRobert Mustacchi onetest("BEL", "\a", 8, 0, 1, L'\a');
64*f65a228fSRobert Mustacchi onetest("A", "A", 8, 0, 1, L'A');
65*f65a228fSRobert Mustacchi onetest("DEL", "\177", 8, 0, 1, L'\177');
66*f65a228fSRobert Mustacchi onetest("0x80", "\200", 8, EILSEQ, -1, WEOF);
67*f65a228fSRobert Mustacchi onetest("0xc3", "\303", 1, 0, -2, WEOF);
68*f65a228fSRobert Mustacchi onetest("U+00E9", "\251", 8, 0, 1, 0xe9);
69*f65a228fSRobert Mustacchi onetest("0xec", "\354", 1, 0, -2, WEOF);
70*f65a228fSRobert Mustacchi onetest("0xecbf", "\277", 1, 0, -2, WEOF);
71*f65a228fSRobert Mustacchi onetest("U+CFFF", "\277", 8, 0, 1, 0xcfff);
72*f65a228fSRobert Mustacchi
73*f65a228fSRobert Mustacchi if (setlocale(LC_CTYPE, "POSIX") == NULL)
74*f65a228fSRobert Mustacchi errx(1, "setlocale(POSIX) failed");
75*f65a228fSRobert Mustacchi
76*f65a228fSRobert Mustacchi onetest("0xff", "\277", 2, 0, 1, L'\277');
77*f65a228fSRobert Mustacchi
78*f65a228fSRobert Mustacchi if (setlocale(LC_CTYPE, "en_US.UTF-8") == NULL)
79*f65a228fSRobert Mustacchi errx(1, "second setlocale(UTF-8) failed");
80*f65a228fSRobert Mustacchi
81*f65a228fSRobert Mustacchi onetest("U+13000", "\360\223\200\200", 8, 0, 4, 0x13000);
82*f65a228fSRobert Mustacchi
83*f65a228fSRobert Mustacchi return (0);
84*f65a228fSRobert Mustacchi }
85