utf8.c - tmux - Rivoreo Source Code Repositories

 /* $OpenBSD$ */

 /*
  * Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
  *
  * 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 MIND, 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 <ctype.h>
 #include <errno.h>
 #include <stdlib.h>
 #include <string.h>
 #include <vis.h>
 #include <wchar.h>

 #include "tmux.h"

 struct utf8_item {
 	u_int			offset;
 	RB_ENTRY(utf8_item)	entry;

 	char			data[UTF8_SIZE];
 	u_char			size;
 };
 RB_HEAD(utf8_tree, utf8_item);

 static int
 utf8_cmp(struct utf8_item *ui1, struct utf8_item *ui2)
 {
 	if (ui1->size < ui2->size)
 		return (-1);
 	if (ui1->size > ui2->size)
 		return (1);
 	return (memcmp(ui1->data, ui2->data, ui1->size));
 }
 RB_GENERATE_STATIC(utf8_tree, utf8_item, entry, utf8_cmp);
 static struct utf8_tree utf8_tree = RB_INITIALIZER(utf8_tree);

 static struct utf8_item *utf8_list;
 static u_int		 utf8_list_size;
 static u_int		 utf8_list_used;

 union utf8_map {
 	utf8_char	uc;
 	struct {
 		u_char	flags;
 		u_char	data[3];
 	};
 } __packed;

 #define UTF8_GET_SIZE(flags) ((flags) & 0x1f)
 #define UTF8_GET_WIDTH(flags) (((flags) >> 5) - 1)

 #define UTF8_SET_SIZE(size) (size)
 #define UTF8_SET_WIDTH(width) ((width + 1) << 5)

 static const union utf8_map utf8_space0 = {
 	.flags = UTF8_SET_WIDTH(0)|UTF8_SET_SIZE(0),
 	.data = ""
 };
 static const union utf8_map utf8_space1 = {
 	.flags = UTF8_SET_WIDTH(1)|UTF8_SET_SIZE(1),
 	.data = " "
 };
 static const union utf8_map utf8_space2 = {
 	.flags = UTF8_SET_WIDTH(2)|UTF8_SET_SIZE(2),
 	.data = "  "
 };

 /* Get a UTF-8 item by offset. */
 static struct utf8_item *
 utf8_get_item(const char *data, size_t size)
 {
 	struct utf8_item	ui;

 	memcpy(ui.data, data, size);
 	ui.size = size;

 	return (RB_FIND(utf8_tree, &utf8_tree, &ui));
 }

 /* Expand UTF-8 list. */
 static int
 utf8_expand_list(void)
 {
 	if (utf8_list_size == 0xffffff)
 		return (-1);
 	if (utf8_list_size == 0)
 		utf8_list_size = 256;
 	else if (utf8_list_size > 0x7fffff)
 		utf8_list_size = 0xffffff;
 	else
 		utf8_list_size *= 2;
 	utf8_list = xreallocarray(utf8_list, utf8_list_size, sizeof *utf8_list);
 	return (0);
 }

 /* Add a UTF-8 item. */
 static int
 utf8_put_item(const char *data, size_t size, u_int *offset)
 {
 	struct utf8_item	*ui;

 	ui = utf8_get_item(data, size);
 	if (ui != NULL) {
 		*offset = ui->offset;
 		log_debug("%s: have %.*s at %u", __func__, (int)size, data,
 		    *offset);
 		return (0);
 	}

 	if (utf8_list_used == utf8_list_size && utf8_expand_list() != 0)
 		return (-1);
 	*offset = utf8_list_used++;

 	ui = &utf8_list[*offset];
 	ui->offset = *offset;
 	memcpy(ui->data, data, size);
 	ui->size = size;
 	RB_INSERT(utf8_tree, &utf8_tree, ui);

 	log_debug("%s: added %.*s at %u", __func__, (int)size, data, *offset);
 	return (0);
 }

 /* Get UTF-8 character from data. */
 enum utf8_state
 utf8_from_data(const struct utf8_data *ud, utf8_char *uc)
 {
 	union utf8_map	 m = { .uc = 0 };
 	u_int		 offset;

 	if (ud->width > 2)
 		fatalx("invalid UTF-8 width");

 	if (ud->size > UTF8_SIZE)
 		goto fail;
 	m.flags = UTF8_SET_SIZE(ud->size)|UTF8_SET_WIDTH(ud->width);
 	if (ud->size <= 3)
 		memcpy(m.data, ud->data, ud->size);
 	else {
 		if (utf8_put_item(ud->data, ud->size, &offset) != 0)
 			goto fail;
 		m.data[0] = (offset & 0xff);
 		m.data[1] = (offset >> 8) & 0xff;
 		m.data[2] = (offset >> 16);
 	}
 	*uc = htonl(m.uc);
 	return (UTF8_DONE);

 fail:
 	if (ud->width == 0)
 		*uc = htonl(utf8_space0.uc);
 	else if (ud->width == 1)
 		*uc = htonl(utf8_space1.uc);
 	else
 		*uc = htonl(utf8_space2.uc);
 	return (UTF8_ERROR);
 }

 /* Get UTF-8 data from character. */
 void
 utf8_to_data(utf8_char uc, struct utf8_data *ud)
 {
 	union utf8_map		 m = { .uc = ntohl(uc) };
 	struct utf8_item	*ui;
 	u_int			 offset;

 	memset(ud, 0, sizeof *ud);
 	ud->size = ud->have = UTF8_GET_SIZE(m.flags);
 	ud->width = UTF8_GET_WIDTH(m.flags);

 	if (ud->size <= 3) {
 		memcpy(ud->data, m.data, ud->size);
 		return;
 	}

 	offset = ((u_int)m.data[2] << 16)|((u_int)m.data[1] << 8)|m.data[0];
 	if (offset >= utf8_list_used)
 		memset(ud->data, ' ', ud->size);
 	else {
 		ui = &utf8_list[offset];
 		memcpy(ud->data, ui->data, ud->size);
 	}
 }

 /* Get UTF-8 character from a single ASCII character. */
 u_int
 utf8_build_one(u_char ch)
 {
 	union utf8_map	m;

 	m.flags = UTF8_SET_SIZE(1)|UTF8_SET_WIDTH(1);
 	m.data[0] = ch;
 	return (htonl(m.uc));
 }

 /* Set a single character. */
 void
 utf8_set(struct utf8_data *ud, u_char ch)
 {
 	static const struct utf8_data empty = { { 0 }, 1, 1, 1 };

 	memcpy(ud, &empty, sizeof *ud);
 	*ud->data = ch;
 }

 /* Copy UTF-8 character. */
 void
 utf8_copy(struct utf8_data *to, const struct utf8_data *from)
 {
 	u_int	i;

 	memcpy(to, from, sizeof *to);

 	for (i = to->size; i < sizeof to->data; i++)
 		to->data[i] = '\0';
 }

 /* Get width of Unicode character. */
 static enum utf8_state
 utf8_width(struct utf8_data *ud, int *width)
 {
 	wchar_t	wc;

 	switch (mbtowc(&wc, ud->data, ud->size)) {
 	case -1:
 		log_debug("UTF-8 %.*s, mbtowc() %d", (int)ud->size, ud->data,
 		    errno);
 		mbtowc(NULL, NULL, MB_CUR_MAX);
 		return (UTF8_ERROR);
 	case 0:
 		return (UTF8_ERROR);
 	}
 	*width = wcwidth(wc);
 	if (*width < 0 || *width > 0xff) {
 		log_debug("UTF-8 %.*s, wcwidth() %d", (int)ud->size, ud->data,
 		    *width);
 		return (UTF8_ERROR);
 	}
 	return (UTF8_DONE);
 }

 /*
  * Open UTF-8 sequence.
  *
  * 11000010-11011111 C2-DF start of 2-byte sequence
  * 11100000-11101111 E0-EF start of 3-byte sequence
  * 11110000-11110100 F0-F4 start of 4-byte sequence
  */
 enum utf8_state
 utf8_open(struct utf8_data *ud, u_char ch)
 {
 	memset(ud, 0, sizeof *ud);
 	if (ch >= 0xc2 && ch <= 0xdf)
 		ud->size = 2;
 	else if (ch >= 0xe0 && ch <= 0xef)
 		ud->size = 3;
 	else if (ch >= 0xf0 && ch <= 0xf4)
 		ud->size = 4;
 	else
 		return (UTF8_ERROR);
 	utf8_append(ud, ch);
 	return (UTF8_MORE);
 }

 /* Append character to UTF-8, closing if finished. */
 enum utf8_state
 utf8_append(struct utf8_data *ud, u_char ch)
 {
 	int	width;

 	if (ud->have >= ud->size)
 		fatalx("UTF-8 character overflow");
 	if (ud->size > sizeof ud->data)
 		fatalx("UTF-8 character size too large");

 	if (ud->have != 0 && (ch & 0xc0) != 0x80)
 		ud->width = 0xff;

 	ud->data[ud->have++] = ch;
 	if (ud->have != ud->size)
 		return (UTF8_MORE);

 	if (ud->width == 0xff)
 		return (UTF8_ERROR);
 	if (utf8_width(ud, &width) != UTF8_DONE)
 		return (UTF8_ERROR);
 	ud->width = width;

 	return (UTF8_DONE);
 }

 /*
  * Encode len characters from src into dst, which is guaranteed to have four
  * bytes available for each character from src (for \abc or UTF-8) plus space
  * for \0.
  */
 int
 utf8_strvis(char *dst, const char *src, size_t len, int flag)
 {
 	struct utf8_data	 ud;
 	const char		*start = dst, *end = src + len;
 	enum utf8_state		 more;
 	size_t			 i;

 	while (src < end) {
 		if ((more = utf8_open(&ud, *src)) == UTF8_MORE) {
 			while (++src < end && more == UTF8_MORE)
 				more = utf8_append(&ud, *src);
 			if (more == UTF8_DONE) {
 				/* UTF-8 character finished. */
 				for (i = 0; i < ud.size; i++)
 					*dst++ = ud.data[i];
 				continue;
 			}
 			/* Not a complete, valid UTF-8 character. */
 			src -= ud.have;
 		}
 		if (src[0] == '$' && src < end - 1) {
 			if (isalpha((u_char)src[1]) ||
 			    src[1] == '_' ||
 			    src[1] == '{')
 				*dst++ = '\\';
 			*dst++ = '$';
 		} else if (src < end - 1)
 			dst = vis(dst, src[0], flag, src[1]);
 		else if (src < end)
 			dst = vis(dst, src[0], flag, '\0');
 		src++;
 	}
 	*dst = '\0';
 	return (dst - start);
 }

 /* Same as utf8_strvis but allocate the buffer. */
 int
 utf8_stravis(char **dst, const char *src, int flag)
 {
 	char	*buf;
 	int	 len;

 	buf = xreallocarray(NULL, 4, strlen(src) + 1);
 	len = utf8_strvis(buf, src, strlen(src), flag);

 	*dst = xrealloc(buf, len + 1);
 	return (len);
 }

 /* Does this string contain anything that isn't valid UTF-8? */
 int
 utf8_isvalid(const char *s)
 {
 	struct utf8_data ud;
 	const char	*end;
 	enum utf8_state	 more;

 	end = s + strlen(s);
 	while (s < end) {
 		if ((more = utf8_open(&ud, *s)) == UTF8_MORE) {
 			while (++s < end && more == UTF8_MORE)
 				more = utf8_append(&ud, *s);
 			if (more == UTF8_DONE)
 				continue;
 			return (0);
 		}
 		if (*s < 0x20 || *s > 0x7e)
 			return (0);
 		s++;
 	}
 	return (1);
 }

 /*
  * Sanitize a string, changing any UTF-8 characters to '_'. Caller should free
  * the returned string. Anything not valid printable ASCII or UTF-8 is
  * stripped.
  */
 char *
 utf8_sanitize(const char *src)
 {
 	char		*dst = NULL;
 	size_t		 n = 0;
 	enum utf8_state	 more;
 	struct utf8_data ud;
 	u_int		 i;

 	while (*src != '\0') {
 		dst = xreallocarray(dst, n + 1, sizeof *dst);
 		if ((more = utf8_open(&ud, *src)) == UTF8_MORE) {
 			while (*++src != '\0' && more == UTF8_MORE)
 				more = utf8_append(&ud, *src);
 			if (more == UTF8_DONE) {
 				dst = xreallocarray(dst, n + ud.width,
 				    sizeof *dst);
 				for (i = 0; i < ud.width; i++)
 					dst[n++] = '_';
 				continue;
 			}
 			src -= ud.have;
 		}
 		if (*src > 0x1f && *src < 0x7f)
 			dst[n++] = *src;
 		else
 			dst[n++] = '_';
 		src++;
 	}
 	dst = xreallocarray(dst, n + 1, sizeof *dst);
 	dst[n] = '\0';
 	return (dst);
 }

 /* Get UTF-8 buffer length. */
 size_t
 utf8_strlen(const struct utf8_data *s)
 {
 	size_t	i;

 	for (i = 0; s[i].size != 0; i++)
 		/* nothing */;
 	return (i);
 }

 /* Get UTF-8 string width. */
 u_int
 utf8_strwidth(const struct utf8_data *s, ssize_t n)
 {
 	ssize_t	i;
 	u_int	width = 0;

 	for (i = 0; s[i].size != 0; i++) {
 		if (n != -1 && n == i)
 			break;
 		width += s[i].width;
 	}
 	return (width);
 }

 /*
  * Convert a string into a buffer of UTF-8 characters. Terminated by size == 0.
  * Caller frees.
  */
 struct utf8_data *
 utf8_fromcstr(const char *src)
 {
 	struct utf8_data	*dst = NULL;
 	size_t			 n = 0;
 	enum utf8_state		 more;

 	while (*src != '\0') {
 		dst = xreallocarray(dst, n + 1, sizeof *dst);
 		if ((more = utf8_open(&dst[n], *src)) == UTF8_MORE) {
 			while (*++src != '\0' && more == UTF8_MORE)
 				more = utf8_append(&dst[n], *src);
 			if (more == UTF8_DONE) {
 				n++;
 				continue;
 			}
 			src -= dst[n].have;
 		}
 		utf8_set(&dst[n], *src);
 		n++;
 		src++;
 	}
 	dst = xreallocarray(dst, n + 1, sizeof *dst);
 	dst[n].size = 0;
 	return (dst);
 }

 /* Convert from a buffer of UTF-8 characters into a string. Caller frees. */
 char *
 utf8_tocstr(struct utf8_data *src)
 {
 	char	*dst = NULL;
 	size_t	 n = 0;

 	for(; src->size != 0; src++) {
 		dst = xreallocarray(dst, n + src->size, 1);
 		memcpy(dst + n, src->data, src->size);
 		n += src->size;
 	}
 	dst = xreallocarray(dst, n + 1, 1);
 	dst[n] = '\0';
 	return (dst);
 }

 /* Get width of UTF-8 string. */
 u_int
 utf8_cstrwidth(const char *s)
 {
 	struct utf8_data	tmp;
 	u_int			width;
 	enum utf8_state		more;

 	width = 0;
 	while (*s != '\0') {
 		if ((more = utf8_open(&tmp, *s)) == UTF8_MORE) {
 			while (*++s != '\0' && more == UTF8_MORE)
 				more = utf8_append(&tmp, *s);
 			if (more == UTF8_DONE) {
 				width += tmp.width;
 				continue;
 			}
 			s -= tmp.have;
 		}
 		if (*s > 0x1f && *s != 0x7f)
 			width++;
 		s++;
 	}
 	return (width);
 }

 /* Pad UTF-8 string to width on the left. Caller frees. */
 char *
 utf8_padcstr(const char *s, u_int width)
 {
 	size_t	 slen;
 	char	*out;
 	u_int	 n, i;

 	n = utf8_cstrwidth(s);
 	if (n >= width)
 		return (xstrdup(s));

 	slen = strlen(s);
 	out = xmalloc(slen + 1 + (width - n));
 	memcpy(out, s, slen);
 	for (i = n; i < width; i++)
 		out[slen++] = ' ';
 	out[slen] = '\0';
 	return (out);
 }

 /* Pad UTF-8 string to width on the right. Caller frees. */
 char *
 utf8_rpadcstr(const char *s, u_int width)
 {
 	size_t	 slen;
 	char	*out;
 	u_int	 n, i;

 	n = utf8_cstrwidth(s);
 	if (n >= width)
 		return (xstrdup(s));

 	slen = strlen(s);
 	out = xmalloc(slen + 1 + (width - n));
 	for (i = 0; i < width - n; i++)
 		out[i] = ' ';
 	memcpy(out + i, s, slen);
 	out[i + slen] = '\0';
 	return (out);
 }

 int
 utf8_cstrhas(const char *s, const struct utf8_data *ud)
 {
 	struct utf8_data	*copy, *loop;
 	int			 found = 0;

 	copy = utf8_fromcstr(s);
 	for (loop = copy; loop->size != 0; loop++) {
 		if (loop->size != ud->size)
 			continue;
 		if (memcmp(loop->data, ud->data, loop->size) == 0) {
 			found = 1;
 			break;
 		}
 	}
 	free(copy);

 	return (found);
 }
	/* $OpenBSD$ */

	/*
	* Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
	*
	* 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 MIND, 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 <ctype.h>
	#include <errno.h>
	#include <stdlib.h>
	#include <string.h>
	#include <vis.h>
	#include <wchar.h>

	#include "tmux.h"

	struct utf8_item {
	u_int offset;
	RB_ENTRY(utf8_item) entry;

	char data[UTF8_SIZE];
	u_char size;
	};
	RB_HEAD(utf8_tree, utf8_item);

	static int
	utf8_cmp(struct utf8_item ui1, struct utf8_item ui2)
	{
	if (ui1->size < ui2->size)
	return (-1);
	if (ui1->size > ui2->size)
	return (1);
	return (memcmp(ui1->data, ui2->data, ui1->size));
	}
	RB_GENERATE_STATIC(utf8_tree, utf8_item, entry, utf8_cmp);
	static struct utf8_tree utf8_tree = RB_INITIALIZER(utf8_tree);

	static struct utf8_item *utf8_list;
	static u_int utf8_list_size;
	static u_int utf8_list_used;

	union utf8_map {
	utf8_char uc;
	struct {
	u_char flags;
	u_char data[3];
	};
	} __packed;

	#define UTF8_GET_SIZE(flags) ((flags) & 0x1f)
	#define UTF8_GET_WIDTH(flags) (((flags) >> 5) - 1)

	#define UTF8_SET_SIZE(size) (size)
	#define UTF8_SET_WIDTH(width) ((width + 1) << 5)

	static const union utf8_map utf8_space0 = {
	.flags = UTF8_SET_WIDTH(0)\|UTF8_SET_SIZE(0),
	.data = ""
	};
	static const union utf8_map utf8_space1 = {
	.flags = UTF8_SET_WIDTH(1)\|UTF8_SET_SIZE(1),
	.data = " "
	};
	static const union utf8_map utf8_space2 = {
	.flags = UTF8_SET_WIDTH(2)\|UTF8_SET_SIZE(2),
	.data = " "
	};

	/* Get a UTF-8 item by offset. */
	static struct utf8_item *
	utf8_get_item(const char *data, size_t size)
	{
	struct utf8_item ui;

	memcpy(ui.data, data, size);
	ui.size = size;

	return (RB_FIND(utf8_tree, &utf8_tree, &ui));
	}

	/* Expand UTF-8 list. */
	static int
	utf8_expand_list(void)
	{
	if (utf8_list_size == 0xffffff)
	return (-1);
	if (utf8_list_size == 0)
	utf8_list_size = 256;
	else if (utf8_list_size > 0x7fffff)
	utf8_list_size = 0xffffff;
	else
	utf8_list_size *= 2;
	utf8_list = xreallocarray(utf8_list, utf8_list_size, sizeof *utf8_list);
	return (0);
	}

	/* Add a UTF-8 item. */
	static int
	utf8_put_item(const char data, size_t size, u_int offset)
	{
	struct utf8_item *ui;

	ui = utf8_get_item(data, size);
	if (ui != NULL) {
	*offset = ui->offset;
	log_debug("%s: have %.*s at %u", __func__, (int)size, data,
	*offset);
	return (0);
	}

	if (utf8_list_used == utf8_list_size && utf8_expand_list() != 0)
	return (-1);
	*offset = utf8_list_used++;

	ui = &utf8_list[*offset];
	ui->offset = *offset;
	memcpy(ui->data, data, size);
	ui->size = size;
	RB_INSERT(utf8_tree, &utf8_tree, ui);

	log_debug("%s: added %.s at %u", __func__, (int)size, data, offset);
	return (0);
	}

	/* Get UTF-8 character from data. */
	enum utf8_state
	utf8_from_data(const struct utf8_data ud, utf8_char uc)
	{
	union utf8_map m = { .uc = 0 };
	u_int offset;

	if (ud->width > 2)
	fatalx("invalid UTF-8 width");

	if (ud->size > UTF8_SIZE)
	goto fail;
	m.flags = UTF8_SET_SIZE(ud->size)\|UTF8_SET_WIDTH(ud->width);
	if (ud->size <= 3)
	memcpy(m.data, ud->data, ud->size);
	else {
	if (utf8_put_item(ud->data, ud->size, &offset) != 0)
	goto fail;
	m.data[0] = (offset & 0xff);
	m.data[1] = (offset >> 8) & 0xff;
	m.data[2] = (offset >> 16);
	}
	*uc = htonl(m.uc);
	return (UTF8_DONE);

	fail:
	if (ud->width == 0)
	*uc = htonl(utf8_space0.uc);
	else if (ud->width == 1)
	*uc = htonl(utf8_space1.uc);
	else
	*uc = htonl(utf8_space2.uc);
	return (UTF8_ERROR);
	}

	/* Get UTF-8 data from character. */
	void
	utf8_to_data(utf8_char uc, struct utf8_data *ud)
	{
	union utf8_map m = { .uc = ntohl(uc) };
	struct utf8_item *ui;
	u_int offset;

	memset(ud, 0, sizeof *ud);
	ud->size = ud->have = UTF8_GET_SIZE(m.flags);
	ud->width = UTF8_GET_WIDTH(m.flags);

	if (ud->size <= 3) {
	memcpy(ud->data, m.data, ud->size);
	return;
	}

	offset = ((u_int)m.data[2] << 16)\|((u_int)m.data[1] << 8)\|m.data[0];
	if (offset >= utf8_list_used)
	memset(ud->data, ' ', ud->size);
	else {
	ui = &utf8_list[offset];
	memcpy(ud->data, ui->data, ud->size);
	}
	}

	/* Get UTF-8 character from a single ASCII character. */
	u_int
	utf8_build_one(u_char ch)
	{
	union utf8_map m;

	m.flags = UTF8_SET_SIZE(1)\|UTF8_SET_WIDTH(1);
	m.data[0] = ch;
	return (htonl(m.uc));
	}

	/* Set a single character. */
	void
	utf8_set(struct utf8_data *ud, u_char ch)
	{
	static const struct utf8_data empty = { { 0 }, 1, 1, 1 };

	memcpy(ud, &empty, sizeof *ud);
	*ud->data = ch;
	}

	/* Copy UTF-8 character. */
	void
	utf8_copy(struct utf8_data to, const struct utf8_data from)
	{
	u_int i;

	memcpy(to, from, sizeof *to);

	for (i = to->size; i < sizeof to->data; i++)
	to->data[i] = '\0';
	}

	/* Get width of Unicode character. */
	static enum utf8_state
	utf8_width(struct utf8_data ud, int width)
	{
	wchar_t wc;

	switch (mbtowc(&wc, ud->data, ud->size)) {
	case -1:
	log_debug("UTF-8 %.*s, mbtowc() %d", (int)ud->size, ud->data,
	errno);
	mbtowc(NULL, NULL, MB_CUR_MAX);
	return (UTF8_ERROR);
	case 0:
	return (UTF8_ERROR);
	}
	*width = wcwidth(wc);
	if (width < 0 \|\| width > 0xff) {
	log_debug("UTF-8 %.*s, wcwidth() %d", (int)ud->size, ud->data,
	*width);
	return (UTF8_ERROR);
	}
	return (UTF8_DONE);
	}

	/*
	* Open UTF-8 sequence.
	*
	* 11000010-11011111 C2-DF start of 2-byte sequence
	* 11100000-11101111 E0-EF start of 3-byte sequence
	* 11110000-11110100 F0-F4 start of 4-byte sequence
	*/
	enum utf8_state
	utf8_open(struct utf8_data *ud, u_char ch)
	{
	memset(ud, 0, sizeof *ud);
	if (ch >= 0xc2 && ch <= 0xdf)
	ud->size = 2;
	else if (ch >= 0xe0 && ch <= 0xef)
	ud->size = 3;
	else if (ch >= 0xf0 && ch <= 0xf4)
	ud->size = 4;
	else
	return (UTF8_ERROR);
	utf8_append(ud, ch);
	return (UTF8_MORE);
	}

	/* Append character to UTF-8, closing if finished. */
	enum utf8_state
	utf8_append(struct utf8_data *ud, u_char ch)
	{
	int width;

	if (ud->have >= ud->size)
	fatalx("UTF-8 character overflow");
	if (ud->size > sizeof ud->data)
	fatalx("UTF-8 character size too large");

	if (ud->have != 0 && (ch & 0xc0) != 0x80)
	ud->width = 0xff;

	ud->data[ud->have++] = ch;
	if (ud->have != ud->size)
	return (UTF8_MORE);

	if (ud->width == 0xff)
	return (UTF8_ERROR);
	if (utf8_width(ud, &width) != UTF8_DONE)
	return (UTF8_ERROR);
	ud->width = width;

	return (UTF8_DONE);
	}

	/*
	* Encode len characters from src into dst, which is guaranteed to have four
	* bytes available for each character from src (for \abc or UTF-8) plus space
	* for \0.
	*/
	int
	utf8_strvis(char dst, const char src, size_t len, int flag)
	{
	struct utf8_data ud;
	const char start = dst, end = src + len;
	enum utf8_state more;
	size_t i;

	while (src < end) {
	if ((more = utf8_open(&ud, *src)) == UTF8_MORE) {
	while (++src < end && more == UTF8_MORE)
	more = utf8_append(&ud, *src);
	if (more == UTF8_DONE) {
	/* UTF-8 character finished. */
	for (i = 0; i < ud.size; i++)
	*dst++ = ud.data[i];
	continue;
	}
	/* Not a complete, valid UTF-8 character. */
	src -= ud.have;
	}
	if (src[0] == '$' && src < end - 1) {
	if (isalpha((u_char)src[1]) \|\|
	src[1] == '_' \|\|
	src[1] == '{')
	*dst++ = '\\';
	*dst++ = '$';
	} else if (src < end - 1)
	dst = vis(dst, src[0], flag, src[1]);
	else if (src < end)
	dst = vis(dst, src[0], flag, '\0');
	src++;
	}
	*dst = '\0';
	return (dst - start);
	}

	/* Same as utf8_strvis but allocate the buffer. */
	int
	utf8_stravis(char *dst, const char src, int flag)
	{
	char *buf;
	int len;

	buf = xreallocarray(NULL, 4, strlen(src) + 1);
	len = utf8_strvis(buf, src, strlen(src), flag);

	*dst = xrealloc(buf, len + 1);
	return (len);
	}

	/* Does this string contain anything that isn't valid UTF-8? */
	int
	utf8_isvalid(const char *s)
	{
	struct utf8_data ud;
	const char *end;
	enum utf8_state more;

	end = s + strlen(s);
	while (s < end) {
	if ((more = utf8_open(&ud, *s)) == UTF8_MORE) {
	while (++s < end && more == UTF8_MORE)
	more = utf8_append(&ud, *s);
	if (more == UTF8_DONE)
	continue;
	return (0);
	}
	if (s < 0x20 \|\| s > 0x7e)
	return (0);
	s++;
	}
	return (1);
	}

	/*
	* Sanitize a string, changing any UTF-8 characters to '_'. Caller should free
	* the returned string. Anything not valid printable ASCII or UTF-8 is
	* stripped.
	*/
	char *
	utf8_sanitize(const char *src)
	{
	char *dst = NULL;
	size_t n = 0;
	enum utf8_state more;
	struct utf8_data ud;
	u_int i;

	while (*src != '\0') {
	dst = xreallocarray(dst, n + 1, sizeof *dst);
	if ((more = utf8_open(&ud, *src)) == UTF8_MORE) {
	while (*++src != '\0' && more == UTF8_MORE)
	more = utf8_append(&ud, *src);
	if (more == UTF8_DONE) {
	dst = xreallocarray(dst, n + ud.width,
	sizeof *dst);
	for (i = 0; i < ud.width; i++)
	dst[n++] = '_';
	continue;
	}
	src -= ud.have;
	}
	if (src > 0x1f && src < 0x7f)
	dst[n++] = *src;
	else
	dst[n++] = '_';
	src++;
	}
	dst = xreallocarray(dst, n + 1, sizeof *dst);
	dst[n] = '\0';
	return (dst);
	}

	/* Get UTF-8 buffer length. */
	size_t
	utf8_strlen(const struct utf8_data *s)
	{
	size_t i;

	for (i = 0; s[i].size != 0; i++)
	/* nothing */;
	return (i);
	}

	/* Get UTF-8 string width. */
	u_int
	utf8_strwidth(const struct utf8_data *s, ssize_t n)
	{
	ssize_t i;
	u_int width = 0;

	for (i = 0; s[i].size != 0; i++) {
	if (n != -1 && n == i)
	break;
	width += s[i].width;
	}
	return (width);
	}

	/*
	* Convert a string into a buffer of UTF-8 characters. Terminated by size == 0.
	* Caller frees.
	*/
	struct utf8_data *
	utf8_fromcstr(const char *src)
	{
	struct utf8_data *dst = NULL;
	size_t n = 0;
	enum utf8_state more;

	while (*src != '\0') {
	dst = xreallocarray(dst, n + 1, sizeof *dst);
	if ((more = utf8_open(&dst[n], *src)) == UTF8_MORE) {
	while (*++src != '\0' && more == UTF8_MORE)
	more = utf8_append(&dst[n], *src);
	if (more == UTF8_DONE) {
	n++;
	continue;
	}
	src -= dst[n].have;
	}
	utf8_set(&dst[n], *src);
	n++;
	src++;
	}
	dst = xreallocarray(dst, n + 1, sizeof *dst);
	dst[n].size = 0;
	return (dst);
	}

	/* Convert from a buffer of UTF-8 characters into a string. Caller frees. */
	char *
	utf8_tocstr(struct utf8_data *src)
	{
	char *dst = NULL;
	size_t n = 0;

	for(; src->size != 0; src++) {
	dst = xreallocarray(dst, n + src->size, 1);
	memcpy(dst + n, src->data, src->size);
	n += src->size;
	}
	dst = xreallocarray(dst, n + 1, 1);
	dst[n] = '\0';
	return (dst);
	}

	/* Get width of UTF-8 string. */
	u_int
	utf8_cstrwidth(const char *s)
	{
	struct utf8_data tmp;
	u_int width;
	enum utf8_state more;

	width = 0;
	while (*s != '\0') {
	if ((more = utf8_open(&tmp, *s)) == UTF8_MORE) {
	while (*++s != '\0' && more == UTF8_MORE)
	more = utf8_append(&tmp, *s);
	if (more == UTF8_DONE) {
	width += tmp.width;
	continue;
	}
	s -= tmp.have;
	}
	if (s > 0x1f && s != 0x7f)
	width++;
	s++;
	}
	return (width);
	}

	/* Pad UTF-8 string to width on the left. Caller frees. */
	char *
	utf8_padcstr(const char *s, u_int width)
	{
	size_t slen;
	char *out;
	u_int n, i;

	n = utf8_cstrwidth(s);
	if (n >= width)
	return (xstrdup(s));

	slen = strlen(s);
	out = xmalloc(slen + 1 + (width - n));
	memcpy(out, s, slen);
	for (i = n; i < width; i++)
	out[slen++] = ' ';
	out[slen] = '\0';
	return (out);
	}

	/* Pad UTF-8 string to width on the right. Caller frees. */
	char *
	utf8_rpadcstr(const char *s, u_int width)
	{
	size_t slen;
	char *out;
	u_int n, i;

	n = utf8_cstrwidth(s);
	if (n >= width)
	return (xstrdup(s));

	slen = strlen(s);
	out = xmalloc(slen + 1 + (width - n));
	for (i = 0; i < width - n; i++)
	out[i] = ' ';
	memcpy(out + i, s, slen);
	out[i + slen] = '\0';
	return (out);
	}

	int
	utf8_cstrhas(const char s, const struct utf8_data ud)
	{
	struct utf8_data copy, loop;
	int found = 0;

	copy = utf8_fromcstr(s);
	for (loop = copy; loop->size != 0; loop++) {
	if (loop->size != ud->size)
	continue;
	if (memcmp(loop->data, ud->data, loop->size) == 0) {
	found = 1;
	break;
	}
	}
	free(copy);

	return (found);
	}