Add libteken (from FreeBSD) to third_party.

third_party/libteken/teken/teken.c

+/*-
+ * Copyright (c) 2008-2009 Ed Schouten <ed@FreeBSD.org>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * $FreeBSD: head/sys/teken/teken.c 261551 2014-02-06 13:28:06Z ray $
+ */
+
+#include <sys/cdefs.h>
+#if defined(__FreeBSD__) && defined(_KERNEL)
+#include <sys/param.h>
+#include <sys/lock.h>
+#include <sys/systm.h>
+#define teken_assert(x)         MPASS(x)
+#else /* !(__FreeBSD__ && _KERNEL) */
+#include <sys/types.h>
+#include <assert.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <string.h>
+#define teken_assert(x)         assert(x)
+#endif /* __FreeBSD__ && _KERNEL */
+
+/* debug messages */
+#define teken_printf(x,...)
+
+/* Private flags for t_stateflags. */
+#define TS_FIRSTDIGIT   0x0001  /* First numeric digit in escape sequence. */
+#define TS_INSERT       0x0002  /* Insert mode. */
+#define TS_AUTOWRAP     0x0004  /* Autowrap. */
+#define TS_ORIGIN       0x0008  /* Origin mode. */
+#define TS_WRAPPED      0x0010  /* Next character should be printed on col 0. */
+#define TS_8BIT         0x0020  /* UTF-8 disabled. */
+#define TS_CONS25       0x0040  /* cons25 emulation. */
+#define TS_INSTRING     0x0080  /* Inside string. */
+#define TS_CURSORKEYS   0x0100  /* Cursor keys mode. */
+
+/* Character that blanks a cell. */
+#define BLANK   ' '
+
+#include "teken.h"
+#include "teken_wcwidth.h"
+#include "teken_scs.h"
+
+static teken_state_t    teken_state_init;
+
+/*
+ * Wrappers for hooks.
+ */
+
+static inline void
+teken_funcs_bell(teken_t *t)
+{
+
+        t->t_funcs->tf_bell(t->t_softc);
+}
+
+static inline void
+teken_funcs_cursor(teken_t *t)
+{
+
+        teken_assert(t->t_cursor.tp_row < t->t_winsize.tp_row);
+        teken_assert(t->t_cursor.tp_col < t->t_winsize.tp_col);
+
+        t->t_funcs->tf_cursor(t->t_softc, &t->t_cursor);
+}
+
+static inline void
+teken_funcs_putchar(teken_t *t, const teken_pos_t *p, teken_char_t c,
+    const teken_attr_t *a)
+{
+
+        teken_assert(p->tp_row < t->t_winsize.tp_row);
+        teken_assert(p->tp_col < t->t_winsize.tp_col);
+
+        t->t_funcs->tf_putchar(t->t_softc, p, c, a);
+}
+
+static inline void
+teken_funcs_fill(teken_t *t, const teken_rect_t *r,
+    const teken_char_t c, const teken_attr_t *a)
+{
+
+        teken_assert(r->tr_end.tp_row > r->tr_begin.tp_row);
+        teken_assert(r->tr_end.tp_row <= t->t_winsize.tp_row);
+        teken_assert(r->tr_end.tp_col > r->tr_begin.tp_col);
+        teken_assert(r->tr_end.tp_col <= t->t_winsize.tp_col);
+
+        t->t_funcs->tf_fill(t->t_softc, r, c, a);
+}
+
+static inline void
+teken_funcs_copy(teken_t *t, const teken_rect_t *r, const teken_pos_t *p)
+{
+
+        teken_assert(r->tr_end.tp_row > r->tr_begin.tp_row);
+        teken_assert(r->tr_end.tp_row <= t->t_winsize.tp_row);
+        teken_assert(r->tr_end.tp_col > r->tr_begin.tp_col);
+        teken_assert(r->tr_end.tp_col <= t->t_winsize.tp_col);
+        teken_assert(p->tp_row + (r->tr_end.tp_row - r->tr_begin.tp_row) <= t->t_winsize.tp_row);
+        teken_assert(p->tp_col + (r->tr_end.tp_col - r->tr_begin.tp_col) <= t->t_winsize.tp_col);
+
+        t->t_funcs->tf_copy(t->t_softc, r, p);
+}
+
+static inline void
+teken_funcs_param(teken_t *t, int cmd, unsigned int value)
+{
+
+        t->t_funcs->tf_param(t->t_softc, cmd, value);
+}
+
+static inline void
+teken_funcs_respond(teken_t *t, const void *buf, size_t len)
+{
+
+        t->t_funcs->tf_respond(t->t_softc, buf, len);
+}
+
+#include "teken_subr.h"
+#include "teken_subr_compat.h"
+
+/*
+ * Programming interface.
+ */
+
+void
+teken_init(teken_t *t, const teken_funcs_t *tf, void *softc)
+{
+        teken_pos_t tp = { .tp_row = 24, .tp_col = 80 };
+
+        t->t_funcs = tf;
+        t->t_softc = softc;
+
+        t->t_nextstate = teken_state_init;
+        t->t_stateflags = 0;
+        t->t_utf8_left = 0;
+
+        t->t_defattr.ta_format = 0;
+        t->t_defattr.ta_fgcolor = TC_WHITE;
+        t->t_defattr.ta_bgcolor = TC_BLACK;
+        teken_subr_do_reset(t);
+
+        teken_set_winsize(t, &tp);
+}
+
+static void
+teken_input_char(teken_t *t, teken_char_t c)
+{
+
+        /*
+         * There is no support for DCS and OSC.  Just discard strings
+         * until we receive characters that may indicate string
+         * termination.
+         */
+        if (t->t_stateflags & TS_INSTRING) {
+                switch (c) {
+                case '\x1B':
+                        t->t_stateflags &= ~TS_INSTRING;
+                        break;
+                case '\a':
+                        t->t_stateflags &= ~TS_INSTRING;
+                        return;
+                default:
+                        return;
+                }
+        }
+
+        switch (c) {
+        case '\0':
+                break;
+        case '\a':
+                teken_subr_bell(t);
+                break;
+        case '\b':
+                teken_subr_backspace(t);
+                break;
+        case '\n':
+        case '\x0B':
+                teken_subr_newline(t);
+                break;
+        case '\x0C':
+                teken_subr_newpage(t);
+                break;
+        case '\x0E':
+                if (t->t_stateflags & TS_CONS25)
+                        t->t_nextstate(t, c);
+                else
+                        t->t_curscs = 1;
+                break;
+        case '\x0F':
+                if (t->t_stateflags & TS_CONS25)
+                        t->t_nextstate(t, c);
+                else
+                        t->t_curscs = 0;
+                break;
+        case '\r':
+                teken_subr_carriage_return(t);
+                break;
+        case '\t':
+                teken_subr_horizontal_tab(t);
+                break;
+        default:
+                t->t_nextstate(t, c);
+                break;
+        }
+
+        /* Post-processing assertions. */
+        teken_assert(t->t_cursor.tp_row >= t->t_originreg.ts_begin);
+        teken_assert(t->t_cursor.tp_row < t->t_originreg.ts_end);
+        teken_assert(t->t_cursor.tp_row < t->t_winsize.tp_row);
+        teken_assert(t->t_cursor.tp_col < t->t_winsize.tp_col);
+        teken_assert(t->t_saved_cursor.tp_row < t->t_winsize.tp_row);
+        teken_assert(t->t_saved_cursor.tp_col < t->t_winsize.tp_col);
+        teken_assert(t->t_scrollreg.ts_end <= t->t_winsize.tp_row);
+        teken_assert(t->t_scrollreg.ts_begin < t->t_scrollreg.ts_end);
+        /* Origin region has to be window size or the same as scrollreg. */
+        teken_assert((t->t_originreg.ts_begin == t->t_scrollreg.ts_begin &&
+            t->t_originreg.ts_end == t->t_scrollreg.ts_end) ||
+            (t->t_originreg.ts_begin == 0 &&
+            t->t_originreg.ts_end == t->t_winsize.tp_row));
+}
+
+static void
+teken_input_byte(teken_t *t, unsigned char c)
+{
+
+        /*
+         * UTF-8 handling.
+         */
+        if ((c & 0x80) == 0x00 || t->t_stateflags & TS_8BIT) {
+                /* One-byte sequence. */
+                t->t_utf8_left = 0;
+                teken_input_char(t, c);
+        } else if ((c & 0xe0) == 0xc0) {
+                /* Two-byte sequence. */
+                t->t_utf8_left = 1;
+                t->t_utf8_partial = c & 0x1f;
+        } else if ((c & 0xf0) == 0xe0) {
+                /* Three-byte sequence. */
+                t->t_utf8_left = 2;
+                t->t_utf8_partial = c & 0x0f;
+        } else if ((c & 0xf8) == 0xf0) {
+                /* Four-byte sequence. */
+                t->t_utf8_left = 3;
+                t->t_utf8_partial = c & 0x07;
+        } else if ((c & 0xc0) == 0x80) {
+                if (t->t_utf8_left == 0)
+                        return;
+                t->t_utf8_left--;
+                t->t_utf8_partial = (t->t_utf8_partial << 6) | (c & 0x3f);
+                if (t->t_utf8_left == 0) {
+                        teken_printf("Got UTF-8 char %x\n", t->t_utf8_partial);
+                        teken_input_char(t, t->t_utf8_partial);
+                }
+        }
+}
+
+void
+teken_input(teken_t *t, const void *buf, size_t len)
+{
+        const char *c = buf;
+
+        while (len-- > 0)
+                teken_input_byte(t, *c++);
+}
+
+const teken_pos_t *
+teken_get_cursor(teken_t *t)
+{
+
+        return (&t->t_cursor);
+}
+
+void
+teken_set_cursor(teken_t *t, const teken_pos_t *p)
+{
+
+        /* XXX: bounds checking with originreg! */
+        teken_assert(p->tp_row < t->t_winsize.tp_row);
+        teken_assert(p->tp_col < t->t_winsize.tp_col);
+
+        t->t_cursor = *p;
+}
+
+const teken_attr_t *
+teken_get_curattr(teken_t *t)
+{
+
+        return (&t->t_curattr);
+}
+
+void
+teken_set_curattr(teken_t *t, const teken_attr_t *a)
+{
+
+        t->t_curattr = *a;
+}
+
+const teken_attr_t *
+teken_get_defattr(teken_t *t)
+{
+
+        return (&t->t_defattr);
+}
+
+void
+teken_set_defattr(teken_t *t, const teken_attr_t *a)
+{
+
+        t->t_curattr = t->t_saved_curattr = t->t_defattr = *a;
+}
+
+const teken_pos_t *
+teken_get_winsize(teken_t *t)
+{
+
+        return (&t->t_winsize);
+}
+
+static void
+teken_trim_cursor_pos(teken_t *t, const teken_pos_t *new)
+{
+        const teken_pos_t *cur;
+
+        cur = &t->t_winsize;
+
+        if (cur->tp_row < new->tp_row || cur->tp_col < new->tp_col)
+                return;
+        if (t->t_cursor.tp_row >= new->tp_row)
+                t->t_cursor.tp_row = new->tp_row - 1;
+        if (t->t_cursor.tp_col >= new->tp_col)
+                t->t_cursor.tp_col = new->tp_col - 1;
+}
+
+void
+teken_set_winsize(teken_t *t, const teken_pos_t *p)
+{
+
+        teken_trim_cursor_pos(t, p);
+        t->t_winsize = *p;
+        teken_subr_do_reset(t);
+}
+
+void
+teken_set_winsize_noreset(teken_t *t, const teken_pos_t *p)
+{
+
+        teken_trim_cursor_pos(t, p);
+        t->t_winsize = *p;
+        teken_subr_do_resize(t);
+}
+
+void
+teken_set_8bit(teken_t *t)
+{
+
+        t->t_stateflags |= TS_8BIT;
+}
+
+void
+teken_set_cons25(teken_t *t)
+{
+
+        t->t_stateflags |= TS_CONS25;
+}
+
+/*
+ * State machine.
+ */
+
+static void
+teken_state_switch(teken_t *t, teken_state_t *s)
+{
+
+        t->t_nextstate = s;
+        t->t_curnum = 0;
+        t->t_stateflags |= TS_FIRSTDIGIT;
+}
+
+static int
+teken_state_numbers(teken_t *t, teken_char_t c)
+{
+
+        teken_assert(t->t_curnum < T_NUMSIZE);
+
+        if (c >= '0' && c <= '9') {
+                /*
+                 * Don't do math with the default value of 1 when a
+                 * custom number is inserted.
+                 */
+                if (t->t_stateflags & TS_FIRSTDIGIT) {
+                        t->t_stateflags &= ~TS_FIRSTDIGIT;
+                        t->t_nums[t->t_curnum] = 0;
+                } else {
+                        t->t_nums[t->t_curnum] *= 10;
+                }
+
+                t->t_nums[t->t_curnum] += c - '0';
+                return (1);
+        } else if (c == ';') {
+                if (t->t_stateflags & TS_FIRSTDIGIT)
+                        t->t_nums[t->t_curnum] = 0;
+
+                /* Only allow a limited set of arguments. */
+                if (++t->t_curnum == T_NUMSIZE) {
+                        teken_state_switch(t, teken_state_init);
+                        return (1);
+                }
+
+                t->t_stateflags |= TS_FIRSTDIGIT;
+                return (1);
+        } else {
+                if (t->t_stateflags & TS_FIRSTDIGIT && t->t_curnum > 0) {
+                        /* Finish off the last empty argument. */
+                        t->t_nums[t->t_curnum] = 0;
+                        t->t_curnum++;
+                } else if ((t->t_stateflags & TS_FIRSTDIGIT) == 0) {
+                        /* Also count the last argument. */
+                        t->t_curnum++;
+                }
+        }
+
+        return (0);
+}
+
+teken_color_t
+teken_256to8(teken_color_t c)
+{
+        unsigned int r, g, b;
+
+        if (c < 16) {
+                /* Traditional color indices. */
+                return (c % 8);
+        } else if (c >= 244) {
+                /* Upper grayscale colors. */
+                return (TC_WHITE);
+        } else if (c >= 232) {
+                /* Lower grayscale colors. */
+                return (TC_BLACK);
+        }
+
+        /* Convert to RGB. */
+        c -= 16;
+        b = c % 6;
+        g = (c / 6) % 6;
+        r = c / 36;
+
+        if (r < g) {
+                /* Possibly green. */
+                if (g < b)
+                        return (TC_BLUE);
+                else if (g > b)
+                        return (TC_GREEN);
+                else
+                        return (TC_CYAN);
+        } else if (r > g) {
+                /* Possibly red. */
+                if (r < b)
+                        return (TC_BLUE);
+                else if (r > b)
+                        return (TC_RED);
+                else
+                        return (TC_MAGENTA);
+        } else {
+                /* Possibly brown. */
+                if (g < b)
+                        return (TC_BLUE);
+                else if (g > b)
+                        return (TC_BROWN);
+                else if (r < 3)
+                        return (TC_BLACK);
+                else
+                        return (TC_WHITE);
+        }
+}
+
+static const char * const special_strings_cons25[] = {
+        [TKEY_UP] = "\x1B[A",           [TKEY_DOWN] = "\x1B[B",
+        [TKEY_LEFT] = "\x1B[D",         [TKEY_RIGHT] = "\x1B[C",
+
+        [TKEY_HOME] = "\x1B[H",         [TKEY_END] = "\x1B[F",
+        [TKEY_INSERT] = "\x1B[L",       [TKEY_DELETE] = "\x7F",
+        [TKEY_PAGE_UP] = "\x1B[I",      [TKEY_PAGE_DOWN] = "\x1B[G",
+
+        [TKEY_F1] = "\x1B[M",           [TKEY_F2] = "\x1B[N",
+        [TKEY_F3] = "\x1B[O",           [TKEY_F4] = "\x1B[P",
+        [TKEY_F5] = "\x1B[Q",           [TKEY_F6] = "\x1B[R",
+        [TKEY_F7] = "\x1B[S",           [TKEY_F8] = "\x1B[T",
+        [TKEY_F9] = "\x1B[U",           [TKEY_F10] = "\x1B[V",
+        [TKEY_F11] = "\x1B[W",          [TKEY_F12] = "\x1B[X",
+};
+
+static const char * const special_strings_ckeys[] = {
+        [TKEY_UP] = "\x1BOA",           [TKEY_DOWN] = "\x1BOB",
+        [TKEY_LEFT] = "\x1BOD",         [TKEY_RIGHT] = "\x1BOC",
+
+        [TKEY_HOME] = "\x1BOH",         [TKEY_END] = "\x1BOF",
+};
+
+static const char * const special_strings_normal[] = {
+        [TKEY_UP] = "\x1B[A",           [TKEY_DOWN] = "\x1B[B",
+        [TKEY_LEFT] = "\x1B[D",         [TKEY_RIGHT] = "\x1B[C",
+
+        [TKEY_HOME] = "\x1B[H",         [TKEY_END] = "\x1B[F",
+        [TKEY_INSERT] = "\x1B[2~",      [TKEY_DELETE] = "\x1B[3~",
+        [TKEY_PAGE_UP] = "\x1B[5~",     [TKEY_PAGE_DOWN] = "\x1B[6~",
+
+        [TKEY_F1] = "\x1BOP",           [TKEY_F2] = "\x1BOQ",
+        [TKEY_F3] = "\x1BOR",           [TKEY_F4] = "\x1BOS",
+        [TKEY_F5] = "\x1B[15~",         [TKEY_F6] = "\x1B[17~",
+        [TKEY_F7] = "\x1B[18~",         [TKEY_F8] = "\x1B[19~",
+        [TKEY_F9] = "\x1B[20~",         [TKEY_F10] = "\x1B[21~",
+        [TKEY_F11] = "\x1B[23~",        [TKEY_F12] = "\x1B[24~",
+};
+
+const char *
+teken_get_sequence(teken_t *t, unsigned int k)
+{
+
+        /* Cons25 mode. */
+        if (t->t_stateflags & TS_CONS25 &&
+            k < sizeof special_strings_cons25 / sizeof(char *))
+                return (special_strings_cons25[k]);
+
+        /* Cursor keys mode. */
+        if (t->t_stateflags & TS_CURSORKEYS &&
+            k < sizeof special_strings_ckeys / sizeof(char *))
+                return (special_strings_ckeys[k]);
+
+        /* Default xterm sequences. */
+        if (k < sizeof special_strings_normal / sizeof(char *))
+                return (special_strings_normal[k]);
+
+        return (NULL);
+}
+
+#include "teken_state.h"