1
2 /*
3 * UTF8BIN.C
4 *
5 * cc utf8bin.c -o ~/bin/utf8bin
6 * rehash
7 * setenv LANG en_US.UTF-8
8 * dd if=/dev/urandom bs=32k count=1024 | utf8bin
9 *
10 * Test round-trip UTF8-B binary escaping functions.
11 */
12 #include <sys/types.h>
13 #include <sys/file.h>
14 #include <locale.h>
15 #include <stdio.h>
16 #include <unistd.h>
17 #include <string.h>
18 #include <wchar.h>
19
20 int
main(int ac,char ** av)21 main(int ac, char **av)
22 {
23 char ibuf[1024];
24 char obuf[1024];
25 wchar_t warray[1024];
26 ssize_t r1;
27 ssize_t r2;
28 ssize_t x;
29 ssize_t w1;
30 ssize_t w2;
31 ssize_t i;
32 ssize_t o;
33 int failed;
34 int flags;
35
36 /*
37 * NOTE: If we use WCSBIN_SURRO the round-trip will not be 8-bit
38 * clean.
39 *
40 * Typically use either 0 or WCSBIN_LONGCODE, both of which
41 * are 8-bit clean, will escape, and cannot error on input
42 * (or generate wchars that might error on output) given the
43 * same flags).
44 */
45 flags = 0;
46 flags = WCSBIN_LONGCODES;
47
48 setlocale(LC_ALL, "");
49 x = 0;
50 while ((flags & WCSBIN_EOF) == 0 &&
51 (r1 = read(0, ibuf + x, sizeof(ibuf) - x)) >= 0) {
52 /* allow final loop for loose ends */
53 if (r1 == 0)
54 flags |= WCSBIN_EOF;
55 r1 += x;
56 r2 = r1;
57 w1 = mbintowcr(warray, ibuf, 1024, &r2, flags);
58
59 /* round-trip, output buffer can be same size as input buffer */
60 w2 = w1;
61 o = wcrtombin(obuf, warray, sizeof(obuf), &w2, flags);
62 fflush(stdout);
63
64 printf("read %4d/%-4d wc=%4d/%-4d write=%-4d\t",
65 r2, r1, w2, w1, o);
66 if (r2 == o) {
67 if (bcmp(ibuf, obuf, o) == 0) {
68 printf("ok%c",
69 (flags & WCSBIN_EOF) ? '\n' : '\r');
70 fflush(stdout);
71 } else {
72 printf("compare-fail\n");
73 }
74 } else if (r2 < o) {
75 if (bcmp(ibuf, obuf, r2) == 0)
76 printf("len-fail, rest-ok\n");
77 else
78 printf("len-fail, compare-fail\n");
79 } else if (o < r2) {
80 if (bcmp(ibuf, obuf, o) == 0)
81 printf("len-fail, rest-ok\n");
82 else
83 printf("len-fail, compare-fail\n");
84 }
85 for (i = failed = 0; i < r2 && i < o; ++i) {
86 if (ibuf[i] != obuf[i]) {
87 printf(" @%04x %02x %02x\n",
88 i, (uint8_t)ibuf[i], (uint8_t)obuf[i]);
89 failed = 16;
90 } else if (failed) {
91 --failed;
92 printf(" @%04x %02x %02x\n",
93 i, (uint8_t)ibuf[i], (uint8_t)obuf[i]);
94 }
95 }
96
97 x = r1 - r2;
98 if (x)
99 bcopy(ibuf + r2, ibuf, x);
100 }
101 }
102