There is no need for us to support parsing values that are larger than the maximum terminal window size. In this case that would be the maximum of unsigned short. The problem with parsing larger values is that they can cause integer overflows when adjusting the cursor position, leading to all sorts of failing assertions. PR: 202326 Reported by: kcwu csie org MFC after: 1 month
564 lines
13 KiB
C
564 lines
13 KiB
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$
|
|
*/
|
|
|
|
#include <sys/cdefs.h>
|
|
#if defined(__FreeBSD__) && defined(_KERNEL)
|
|
#include <sys/param.h>
|
|
#include <sys/limits.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 <limits.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') {
|
|
if (t->t_stateflags & TS_FIRSTDIGIT) {
|
|
/* First digit. */
|
|
t->t_stateflags &= ~TS_FIRSTDIGIT;
|
|
t->t_nums[t->t_curnum] = c - '0';
|
|
} else if (t->t_nums[t->t_curnum] < USHRT_MAX) {
|
|
/*
|
|
* Screen positions are stored as unsigned
|
|
* shorts. There is no need to continue parsing
|
|
* input once the value exceeds USHRT_MAX. It
|
|
* would only allow for integer overflows when
|
|
* performing arithmetic on the cursor position.
|
|
*/
|
|
t->t_nums[t->t_curnum] =
|
|
t->t_nums[t->t_curnum] * 10 + 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"
|