/*
 * UTF8BIN.C
 *
 * cc utf8bin.c -o ~/bin/utf8bin
 * rehash
 * setenv LANG en_US.UTF-8
 * dd if=/dev/urandom bs=32k count=1024 | utf8bin
 *
 * Test round-trip UTF8-B binary escaping functions.
 */
#include <sys/types.h>
#include <sys/file.h>
#include <locale.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <wchar.h>

int
main(int ac, char **av)
{
	char ibuf[1024];
	char obuf[1024];
	wchar_t warray[1024];
	ssize_t r1;
	ssize_t r2;
	ssize_t x;
	ssize_t w1;
	ssize_t w2;
	ssize_t i;
	ssize_t o;
	int failed;
	int flags;

	/*
	 * NOTE: If we use WCSBIN_SURRO the round-trip will not be 8-bit
	 *	 clean.
	 *
	 *	 Typically use either 0 or WCSBIN_LONGCODE, both of which
	 *	 are 8-bit clean, will escape, and cannot error on input
	 *	 (or generate wchars that might error on output) given the
	 *	 same flags).
	 */
	flags = 0;
	flags = WCSBIN_LONGCODES;

	setlocale(LC_ALL, "");
	x = 0;
	while ((flags & WCSBIN_EOF) == 0 &&
	       (r1 = read(0, ibuf + x, sizeof(ibuf) - x)) >= 0) {
		/* allow final loop for loose ends */
		if (r1 == 0)
			flags |= WCSBIN_EOF;
		r1 += x;
		r2 = r1;
		w1 = mbintowcr(warray, ibuf, 1024, &r2, flags);

		/* round-trip, output buffer can be same size as input buffer */
		w2 = w1;
		o = wcrtombin(obuf, warray, sizeof(obuf), &w2, flags);
		fflush(stdout);

		printf("read %4d/%-4d wc=%4d/%-4d write=%-4d\t",
		       r2, r1, w2, w1, o);
		if (r2 == o) {
			if (bcmp(ibuf, obuf, o) == 0) {
				printf("ok%c",
					(flags & WCSBIN_EOF) ? '\n' : '\r');
				fflush(stdout);
			} else {
				printf("compare-fail\n");
			}
		} else if (r2 < o) {
			if (bcmp(ibuf, obuf, r2) == 0)
				printf("len-fail, rest-ok\n");
			else
				printf("len-fail, compare-fail\n");
		} else if (o < r2) {
			if (bcmp(ibuf, obuf, o) == 0)
				printf("len-fail, rest-ok\n");
			else
				printf("len-fail, compare-fail\n");
		}
		for (i = failed = 0; i < r2 && i < o; ++i) {
			if (ibuf[i] != obuf[i]) {
				printf("    @%04x %02x %02x\n",
				       i, (uint8_t)ibuf[i], (uint8_t)obuf[i]);
				failed = 16;
			} else if (failed) {
				--failed;
				printf("    @%04x %02x %02x\n",
				       i, (uint8_t)ibuf[i], (uint8_t)obuf[i]);
			}
		}

		x = r1 - r2;
		if (x)
			bcopy(ibuf + r2, ibuf, x);
	}
}