[z180-stamp.git] / avr / cli_readline.c

/*
 * (C) Copyright 2014 Leo C. <erbl259-lmu@yahoo.de>
 *
 * (C) Copyright 2000
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * Add to readline cmdline-editing by
 * (C) Copyright 2005
 * JinHua Luo, GuangDong Linux Center, <luo.jinhua@gd-linux.com>
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include "common.h"
#include <string.h>
#include <stdio.h>
#include <ctype.h>

#include "config.h"
#include "con-utils.h"
#include "print-utils.h"
#include "command.h"
#include "cli_readline.h"


/************************************************************************************************/
/* TODO:
 *
 */

#define ESC						0x1b

#define KEY_TAB                 '\t'                                                // TAB key
#define KEY_CR                  '\r'                                                // RETURN key
#define KEY_BACKSPACE           '\b'                                                // Backspace key
#define KEY_ESCAPE              0x1B                                                // ESCAPE (pressed twice)

#define KEY_DOWN                0x80                                                // Down arrow key
#define KEY_UP                  0x81                                                // Up arrow key
#define KEY_LEFT                0x82                                                // Left arrow key
#define KEY_RIGHT               0x83                                                // Right arrow key
#define KEY_HOME                0x84                                                // Home key
#define KEY_DC                  0x85                                                // Delete character key
#define KEY_IC                  0x86                                                // Ins char/toggle ins mode key
#define KEY_NPAGE               0x87                                                // Next-page key
#define KEY_PPAGE               0x88                                                // Previous-page key
#define KEY_END                 0x89                                                // End key
#define KEY_BTAB                0x8A                                                // Back tab key
#define KEY_F1                  0x8B                                                // Function key F1
#define KEY_F(n)                (KEY_F1+(n)-1)                                      // Space for additional 12 function keys


struct fkey_tbl_s {
	const FLASH char *sequence;		/* ESC Sequence	*/
	int	code;						/* Keycode	*/
};

//typedef const FLASH struct fkey_tbl_s fkey_tbl_t;

#define FKEY_TBL_ITEM(_seq, _code)		{ FSTR(#_seq), _code }


static const FLASH struct fkey_tbl_s fkey_table[] = {

FKEY_TBL_ITEM(B,   KEY_DOWN),			// Down arrow key
FKEY_TBL_ITEM(A,   KEY_UP),			// Up arrow key
FKEY_TBL_ITEM(D,   KEY_LEFT),			// Left arrow key
FKEY_TBL_ITEM(C,   KEY_RIGHT),		// Right arrow key
FKEY_TBL_ITEM(1~,  KEY_HOME),			// Home key
FKEY_TBL_ITEM(3~,  KEY_DC),			// Delete character key
FKEY_TBL_ITEM(2~,  KEY_IC),			// Ins char/toggle ins mode key
FKEY_TBL_ITEM(6~,  KEY_NPAGE),		// Next-page key
FKEY_TBL_ITEM(5~,  KEY_PPAGE),		// Previous-page key
FKEY_TBL_ITEM(4~,  KEY_END),			// End key
FKEY_TBL_ITEM(Z,   KEY_BTAB),			// Back tab key
/* VT400: */
FKEY_TBL_ITEM(11~, KEY_F(1)),			// Function key F1
FKEY_TBL_ITEM(12~, KEY_F(2)),			// Function key F2
FKEY_TBL_ITEM(13~, KEY_F(3)),			// Function key F3
FKEY_TBL_ITEM(14~, KEY_F(4)),			// Function key F4
FKEY_TBL_ITEM(15~, KEY_F(5)),			// Function key F5
/* Linux consoe */
FKEY_TBL_ITEM([A,  KEY_F(1)),			// Function key F1
FKEY_TBL_ITEM([B,  KEY_F(2)),			// Function key F2
FKEY_TBL_ITEM([C,  KEY_F(3)),			// Function key F3
FKEY_TBL_ITEM([D,  KEY_F(4)),			// Function key F4
FKEY_TBL_ITEM([E,  KEY_F(5)),			// Function key F5

FKEY_TBL_ITEM(17~, KEY_F(6)),			// Function key F6
FKEY_TBL_ITEM(18~, KEY_F(7)),			// Function key F7
FKEY_TBL_ITEM(19~, KEY_F(8)),			// Function key F8
FKEY_TBL_ITEM(20~, KEY_F(9)),			// Function key F9
FKEY_TBL_ITEM(21~, KEY_F(10)),		// Function key F10
FKEY_TBL_ITEM(23~, KEY_F(11)),		// Function key F11
FKEY_TBL_ITEM(24~, KEY_F(12)),		// Function key F12
{ NULL }	/* Mark end of table */
};


typedef enum {
	STATE_GROUND,
	STATE_ESCAPE,
	STATE_CSI_ENTRY
} vtparse_state_t;

#define CHB_SIZE 	15

static
int vt_parse (void)
{
	static vtparse_state_t state = STATE_GROUND;
	char	buf[CHB_SIZE+1];
	uint8_t param[2];
	uint8_t i_buf;
	uint8_t i_param;
	int  ch;


	while (1) {
		ch = my_getchar(1);
//		debug_getch(state, ch);

		switch (state) {
		case STATE_GROUND:
			if (ch == ESC) {
				state = STATE_ESCAPE;
				continue;
			}
			if (ch == 0x7F)				// BACKSPACE on VT200 sends DEL char
				ch = KEY_BACKSPACE;		// map it to '\b'
			break;
		case STATE_ESCAPE:
			if (ch < 0)
				continue;

			if (ch == '[') {
				state = STATE_CSI_ENTRY;
				param[0] = param[1] = 0;
				i_buf = 0;
				i_param = 0;
				continue;
			}
			state = STATE_GROUND;
			break;
		case STATE_CSI_ENTRY:
			if (ch < 0)
				continue;

			buf[i_buf] = ch;
			if (i_buf < CHB_SIZE)
				i_buf++;
			if (ch == ';') {
				i_param++;
				continue;
			}
			if (isdigit(ch)) {
				if (i_param < 2)
					param[i_param] = param[i_param] * 10 + ch - '0';
				continue;
			}
			if (ch >= '@' && ch <= '~' && ch != '[') {
				buf[i_buf] = '\0';
				int_fast8_t i = 0;
				while (fkey_table[i].sequence) {
					if (! strcmp_P (buf, fkey_table[i].sequence)) {
						ch = fkey_table[i].code;
						break;
					}
					i++;
				}
				if (fkey_table[i].sequence == NULL) {
					ch = '$'; /* KEY_ESCAPE; */
				}
			}
			state = STATE_GROUND;
			break;
		}
		break; /* while */
	}

	return ch;
}

/************************************************************************************************/


char console_buffer[CONFIG_SYS_CBSIZE + 1];	/* console I/O buffer	*/

#ifndef CONFIG_CMDLINE_EDITING
static const FLASH char erase_seq[] = "\b \b";	  /* erase sequence */
static const FLASH char   tab_seq[] = "        "; /* used to expand TABs */

static char *delete_char (char *buffer, char *p, int *colp, int *np, int plen)
{
	char *s;

	if (*np == 0)
		return p;

	if (*(--p) == '\t') {		/* will retype the whole line */
		while (*colp > plen) {
			my_puts_P(erase_seq);
			(*colp)--;
		}
		for (s = buffer; s < p; ++s) {
			if (*s == '\t') {
				my_puts_P(tab_seq + ((*colp) & 07));
				*colp += 8 - ((*colp) & 07);
			} else {
				++(*colp);
				putchar(*s);
			}
		}
	} else {
		my_puts_P(erase_seq);
		(*colp)--;
	}
	(*np)--;

	return p;
}
#endif /* CONFIG_CMDLINE_EDITING */


#ifdef CONFIG_CMDLINE_EDITING

/*
 * cmdline-editing related codes from vivi.
 * Author: Janghoon Lyu <nandy@mizi.com>
 */

static void putnstr(char *str, int n)
{
	/* printf_P(PSTR("%.*s"), (int)n, str) */
	while (n-- && *str)
		putchar(*str++);
}


#define CTL_CH(c)			((c) - 'a' + 1)
#define CTL_BACKSPACE		('\b')
#define DEL					((char)255)
#define DEL7				((char)127)
#define CREAD_HIST_CHAR		('!')

#define getcmd_putch(ch)	putchar(ch)
#define getcmd_getch()		vt_parse()
#define getcmd_cbeep()		getcmd_putch('\a')

#define HIST_MAX			5
#define HIST_SIZE			CONFIG_SYS_CBSIZE

static int hist_max;
static int hist_add_idx;
static int hist_cur = -1;
static unsigned hist_num;

static char *hist_list[HIST_MAX];
static char hist_lines[HIST_MAX][HIST_SIZE + 1];	/* Save room for NULL */

#define add_idx_minus_one() ((hist_add_idx == 0) ? hist_max : hist_add_idx-1)

static void hist_init(void)
{
	int i;

	hist_max = 0;
	hist_add_idx = 0;
	hist_cur = -1;
	hist_num = 0;

	for (i = 0; i < HIST_MAX; i++) {
		hist_list[i] = hist_lines[i];
		hist_list[i][0] = '\0';
	}
}

static void cread_add_to_hist(char *line)
{
	strcpy(hist_list[hist_add_idx], line);

	if (++hist_add_idx >= HIST_MAX)
		hist_add_idx = 0;

	if (hist_add_idx > hist_max)
		hist_max = hist_add_idx;

	hist_num++;
}

static char *hist_prev(void)
{
	char *ret;
	int old_cur;

	if (hist_cur < 0)
		return NULL;

	old_cur = hist_cur;
	if (--hist_cur < 0)
		hist_cur = hist_max;

	if (hist_cur == hist_add_idx) {
		hist_cur = old_cur;
		ret = NULL;
	} else {
		ret = hist_list[hist_cur];
	}

	return ret;
}

static char *hist_next(void)
{
	char *ret;

	if (hist_cur < 0)
		return NULL;

	if (hist_cur == hist_add_idx)
		return NULL;

	if (++hist_cur > hist_max)
		hist_cur = 0;

	if (hist_cur == hist_add_idx)
		ret = "";
	else
		ret = hist_list[hist_cur];

	return ret;
}


#define BEGINNING_OF_LINE() {                                                  \
	while (num) {                                                              \
		getcmd_putch(CTL_BACKSPACE);                                           \
		num--;                                                                 \
	}                                                                          \
}

#define ERASE_TO_EOL() {                                                       \
	if (num < eol_num) {                                                       \
		/* printf_P(PSTR("%*S"), (int)(eol_num - num), PSTR("")); */           \
		print_blanks(eol_num - num);                                           \
		do {                                                                   \
			getcmd_putch(CTL_BACKSPACE);                                       \
		} while (--eol_num > num);                                             \
	}                                                                          \
}

#define REFRESH_TO_EOL() {                                                     \
	if (num < eol_num) {                                                       \
		wlen = eol_num - num;                                                  \
		putnstr(buf + num, wlen);                                              \
		num = eol_num;                                                         \
	}                                                                          \
}

static void cread_add_char(char ichar, int insert, unsigned int *num,
	       unsigned int *eol_num, char *buf, unsigned int len)
{
	unsigned int wlen;

	/* room ??? */
	if (insert || *num == *eol_num) {
		if (*eol_num > len - 1) {
			getcmd_cbeep();
			return;
		}
		(*eol_num)++;
	}

	if (insert) {
		wlen = *eol_num - *num;
		if (wlen > 1)
			memmove(&buf[*num+1], &buf[*num], wlen-1);

		buf[*num] = ichar;
		putnstr(buf + *num, wlen);
		(*num)++;
		while (--wlen)
			getcmd_putch(CTL_BACKSPACE);
	} else {
		/* echo the character */
		wlen = 1;
		buf[*num] = ichar;
		putnstr(buf + *num, wlen);
		(*num)++;
	}
}

static void cread_add_str(char *str, int strsize, int insert,
			  unsigned int *num, unsigned int *eol_num,
			  char *buf, unsigned int len)
{
	while (strsize--) {
		cread_add_char(*str, insert, num, eol_num, buf, len);
		str++;
	}
}

static int cread_line(const FLASH char *const prompt, char *buf, unsigned int *len)
{
	unsigned int num = 0;
	unsigned int eol_num = 0;
	unsigned int wlen;
	int ichar;
	int insert = 1;
	int init_len = strlen(buf);

	(void) prompt;

	if (init_len)
		cread_add_str(buf, init_len, 1, &num, &eol_num, buf, *len);

	while (1) {
		ichar = getcmd_getch();

		if ((ichar == '\n') || (ichar == '\r')) {
			putchar('\n');
			break;
		}


		switch (ichar) {

		case KEY_HOME:
		case CTL_CH('a'):
			BEGINNING_OF_LINE();
			break;
		case CTL_CH('c'):	/* ^C - break */
			*buf = '\0';	/* discard input */
			return -1;
		case KEY_RIGHT:
		case CTL_CH('f'):
			if (num < eol_num) {
				getcmd_putch(buf[num]);
				num++;
			}
			break;
		case KEY_LEFT:
		case CTL_CH('b'):
			if (num) {
				getcmd_putch(CTL_BACKSPACE);
				num--;
			}
			break;
		case KEY_DC:
		case CTL_CH('d'):
			if (num < eol_num) {
				wlen = eol_num - num - 1;
				if (wlen) {
					memmove(&buf[num], &buf[num+1], wlen);
					putnstr(buf + num, wlen);
				}

				getcmd_putch(' ');
				do {
					getcmd_putch(CTL_BACKSPACE);
				} while (wlen--);
				eol_num--;
			}
			break;
		case CTL_CH('k'):
			ERASE_TO_EOL();
			break;
		case CTL_CH('e'):
			REFRESH_TO_EOL();
			break;
		case KEY_IC:
		case CTL_CH('o'):
			insert = !insert;
			break;
		case CTL_CH('x'):
		case CTL_CH('u'):
			BEGINNING_OF_LINE();
			ERASE_TO_EOL();
			break;
		case DEL:
		case DEL7:
		case 8:
			if (num) {
				wlen = eol_num - num;
				num--;
				memmove(&buf[num], &buf[num+1], wlen);
				getcmd_putch(CTL_BACKSPACE);
				putnstr(buf + num, wlen);
				getcmd_putch(' ');
				do {
					getcmd_putch(CTL_BACKSPACE);
				} while (wlen--);
				eol_num--;
			}
			break;
		case KEY_UP:
		case CTL_CH('p'):
		case KEY_DOWN:
		case CTL_CH('n'):
		{
			char *hline;

			if (ichar == CTL_CH('p') || ichar == KEY_UP)
				hline = hist_prev();
			else
				hline = hist_next();

			if (!hline) {
				getcmd_cbeep();
				continue;
			}

			/* nuke the current line */
			/* first, go home */
			BEGINNING_OF_LINE();

			/* erase to end of line */
			ERASE_TO_EOL();

			/* copy new line into place and display */
			strcpy(buf, hline);
			eol_num = strlen(buf);
			REFRESH_TO_EOL();
			continue;
		}
#ifdef CONFIG_AUTO_COMPLETE
		case '\t': {
			int num2, col;

			/* do not autocomplete when in the middle */
			if (num < eol_num) {
				getcmd_cbeep();
				break;
			}

			buf[num] = '\0';
			col = strlen_P(prompt) + eol_num;
			num2 = num;
			if (cmd_auto_complete(prompt, buf, &num2, &col)) {
				col = num2 - num;
				num += col;
				eol_num += col;
			}
			break;
		}
#endif
		default:
			if (isprint(ichar))
				cread_add_char(ichar, insert, &num, &eol_num, buf,
								*len);
			break;
		}
	}
	*len = eol_num;
	buf[eol_num] = '\0';	/* lose the newline */

	if (buf[0] /* && buf[0] != CREAD_HIST_CHAR */)
		cread_add_to_hist(buf);
	hist_cur = hist_add_idx;

	return 0;
}

#endif /* CONFIG_CMDLINE_EDITING */

/****************************************************************************/

static int cli_readline_into_buffer(const FLASH char *const prompt, char *buffer)
{
	char *p = buffer;
#ifdef CONFIG_CMDLINE_EDITING
	unsigned int len = CONFIG_SYS_CBSIZE;
	int rc;
	static int initted;

	if (!initted) {
		hist_init();
		initted = 1;
	}

	if (prompt)
		my_puts_P(prompt);

	rc = cread_line(prompt, p, &len);
	return rc < 0 ? rc : (int) len;

#else	/* CONFIG_CMDLINE_EDITING */
	char *p_buf = p;
	int	n = 0;				/* buffer index		*/
	int	plen = 0;			/* prompt length	*/
	int	col;				/* output column cnt	*/
	char	c;

	/* print prompt */
	if (prompt) {
		plen = strlen_P(prompt);
		my_puts_P(prompt);
	}
	col = plen;

	for (;;) {

		c = my_getchar(1);

		/*
		 * Special character handling
		 */
		switch (c) {
		case '\r':			/* Enter		*/
		case '\n':
			*p = '\0';
			my_puts_P(PSTR("\r\n"));
			return p - p_buf;

		case '\0':			/* nul			*/
			continue;

		case 0x03:			/* ^C - break		*/
			p_buf[0] = '\0';	/* discard input */
			return -1;

		case 0x15:			/* ^U - erase line	*/
			while (col > plen) {
				my_puts_P(erase_seq);
				--col;
			}
			p = p_buf;
			n = 0;
			continue;

		case 0x17:			/* ^W - erase word	*/
			p = delete_char(p_buf, p, &col, &n, plen);
			while ((n > 0) && (*p != ' '))
				p = delete_char(p_buf, p, &col, &n, plen);
			continue;

		case 0x08:			/* ^H  - backspace	*/
		case 0x7F:			/* DEL - backspace	*/
			p = delete_char(p_buf, p, &col, &n, plen);
			continue;

		default:
			/*
			 * Must be a normal character then
			 */
			if (n < CONFIG_SYS_CBSIZE-2) {
				if (c == '\t') {	/* expand TABs */
#ifdef CONFIG_AUTO_COMPLETE
					/*
					 * if auto completion triggered just
					 * continue
					 */
					*p = '\0';
					if (cmd_auto_complete(prompt,
							      console_buffer,
							      &n, &col)) {
						p = p_buf + n;	/* reset */
						continue;
					}
#endif
					my_puts_P(tab_seq + (col & 07));
					col += 8 - (col & 07);
				} else {
					++col;
					putchar(c);
				}
				*p++ = c;
				++n;
			} else {			/* Buffer full */
				putchar('\a');
			}
		}
	}
#endif /* CONFIG_CMDLINE_EDITING */
}

int cli_readline(const FLASH char *const prompt)
{
	/*
	 * If console_buffer isn't 0-length the user will be prompted to modify
	 * it instead of entering it from scratch as desired.
	 */
	console_buffer[0] = '\0';

	return cli_readline_into_buffer(prompt, console_buffer);
}
Commit	Line	Data
d684c216	1	/*
35edb766 L	2	* (C) Copyright 2014 Leo C. <erbl259-lmu@yahoo.de>
35edb766 L	3	*
d684c216 L	4	* (C) Copyright 2000
	5	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	6	*
	7	* Add to readline cmdline-editing by
	8	* (C) Copyright 2005
	9	* JinHua Luo, GuangDong Linux Center, <luo.jinhua@gd-linux.com>
	10	*
	11	* SPDX-License-Identifier: GPL-2.0+
	12	*/
	13
	14	#include "common.h"
d684c216 L	15	#include <string.h>
d684c216 L	16	#include <stdio.h>
8506d791	17	#include <ctype.h>
d684c216 L	18
	19	#include "config.h"
	20	#include "con-utils.h"
8591c65b	21	#include "print-utils.h"
d684c216 L	22	#include "command.h"
	23	#include "cli_readline.h"
	24
8506d791 L	25
	26
	27	/************************************************************************************************/
	28	/* TODO:
	29	*
	30	*/
	31
	32	#define ESC 0x1b
	33
	34	#define KEY_TAB '\t' // TAB key
	35	#define KEY_CR '\r' // RETURN key
	36	#define KEY_BACKSPACE '\b' // Backspace key
	37	#define KEY_ESCAPE 0x1B // ESCAPE (pressed twice)
	38
	39	#define KEY_DOWN 0x80 // Down arrow key
	40	#define KEY_UP 0x81 // Up arrow key
	41	#define KEY_LEFT 0x82 // Left arrow key
	42	#define KEY_RIGHT 0x83 // Right arrow key
	43	#define KEY_HOME 0x84 // Home key
	44	#define KEY_DC 0x85 // Delete character key
	45	#define KEY_IC 0x86 // Ins char/toggle ins mode key
	46	#define KEY_NPAGE 0x87 // Next-page key
	47	#define KEY_PPAGE 0x88 // Previous-page key
	48	#define KEY_END 0x89 // End key
	49	#define KEY_BTAB 0x8A // Back tab key
	50	#define KEY_F1 0x8B // Function key F1
	51	#define KEY_F(n) (KEY_F1+(n)-1) // Space for additional 12 function keys
	52
	53
	54	struct fkey_tbl_s {
	55	const FLASH char sequence; / ESC Sequence */
	56	int code; /* Keycode */
	57	};
	58
	59	//typedef const FLASH struct fkey_tbl_s fkey_tbl_t;
	60
	61	#define FKEY_TBL_ITEM(_seq, _code) { FSTR(#_seq), _code }
	62
	63
	64
	65	static const FLASH struct fkey_tbl_s fkey_table[] = {
	66
	67	FKEY_TBL_ITEM(B, KEY_DOWN), // Down arrow key
	68	FKEY_TBL_ITEM(A, KEY_UP), // Up arrow key
	69	FKEY_TBL_ITEM(D, KEY_LEFT), // Left arrow key
	70	FKEY_TBL_ITEM(C, KEY_RIGHT), // Right arrow key
	71	FKEY_TBL_ITEM(1~, KEY_HOME), // Home key
	72	FKEY_TBL_ITEM(3~, KEY_DC), // Delete character key
	73	FKEY_TBL_ITEM(2~, KEY_IC), // Ins char/toggle ins mode key
	74	FKEY_TBL_ITEM(6~, KEY_NPAGE), // Next-page key
	75	FKEY_TBL_ITEM(5~, KEY_PPAGE), // Previous-page key
	76	FKEY_TBL_ITEM(4~, KEY_END), // End key
	77	FKEY_TBL_ITEM(Z, KEY_BTAB), // Back tab key
	78	/* VT400: */
	79	FKEY_TBL_ITEM(11~, KEY_F(1)), // Function key F1
	80	FKEY_TBL_ITEM(12~, KEY_F(2)), // Function key F2
	81	FKEY_TBL_ITEM(13~, KEY_F(3)), // Function key F3
	82	FKEY_TBL_ITEM(14~, KEY_F(4)), // Function key F4
	83	FKEY_TBL_ITEM(15~, KEY_F(5)), // Function key F5
	84	/* Linux consoe */
	85	FKEY_TBL_ITEM([A, KEY_F(1)), // Function key F1
	86	FKEY_TBL_ITEM([B, KEY_F(2)), // Function key F2
	87	FKEY_TBL_ITEM([C, KEY_F(3)), // Function key F3
	88	FKEY_TBL_ITEM([D, KEY_F(4)), // Function key F4
89	FKEY_TBL_ITEM([E, KEY_F(5)), // Function key F5
90
91	FKEY_TBL_ITEM(17~, KEY_F(6)), // Function key F6
92	FKEY_TBL_ITEM(18~, KEY_F(7)), // Function key F7
93	FKEY_TBL_ITEM(19~, KEY_F(8)), // Function key F8
94	FKEY_TBL_ITEM(20~, KEY_F(9)), // Function key F9
95	FKEY_TBL_ITEM(21~, KEY_F(10)), // Function key F10
96	FKEY_TBL_ITEM(23~, KEY_F(11)), // Function key F11
97	FKEY_TBL_ITEM(24~, KEY_F(12)), // Function key F12
98	{ NULL } /* Mark end of table */
99	};
100
101
102
103	typedef enum {
104	STATE_GROUND,
105	STATE_ESCAPE,
106	STATE_CSI_ENTRY
107	} vtparse_state_t;
108
109	#define CHB_SIZE 15
110
111	static
112	int vt_parse (void)
113	{
114	static vtparse_state_t state = STATE_GROUND;
115	char buf[CHB_SIZE+1];
116	uint8_t param[2];
117	uint8_t i_buf;
118	uint8_t i_param;
119	int ch;
120
121
122	while (1) {
123	ch = my_getchar(1);
124	// debug_getch(state, ch);
125
126	switch (state) {
127	case STATE_GROUND:
128	if (ch == ESC) {
129	state = STATE_ESCAPE;
130	continue;
131	}
132	if (ch == 0x7F) // BACKSPACE on VT200 sends DEL char
133	ch = KEY_BACKSPACE; // map it to '\b'
134	break;
135	case STATE_ESCAPE:
136	if (ch < 0)
137	continue;
138
139	if (ch == '[') {
140	state = STATE_CSI_ENTRY;
141	param[0] = param[1] = 0;
142	i_buf = 0;
143	i_param = 0;
144	continue;
145	}
146	state = STATE_GROUND;
147	break;
148	case STATE_CSI_ENTRY:
149	if (ch < 0)
150	continue;
151
152	buf[i_buf] = ch;
153	if (i_buf < CHB_SIZE)
154	i_buf++;
155	if (ch == ';') {
156	i_param++;
157	continue;
158	}
159	if (isdigit(ch)) {
160	if (i_param < 2)
161	param[i_param] = param[i_param] * 10 + ch - '0';
162	continue;
163	}
164	if (ch >= '@' && ch <= '~' && ch != '[') {
165	buf[i_buf] = '\0';
166	int_fast8_t i = 0;
167	while (fkey_table[i].sequence) {
168	if (! strcmp_P (buf, fkey_table[i].sequence)) {
169	ch = fkey_table[i].code;
170	break;
171	}
172	i++;
173	}
174	if (fkey_table[i].sequence == NULL) {
175	ch = '$'; /* KEY_ESCAPE; */
176	}
177	}
178	state = STATE_GROUND;
179	break;
180	}
181	break; /* while */
182	}
183
184	return ch;
185	}
186
187	/************************************************************************************************/
188
189
190
191
192
d684c216 L	193	char console_buffer[CONFIG_SYS_CBSIZE + 1]; /* console I/O buffer */
d684c216 L	194
8506d791	195	#ifndef CONFIG_CMDLINE_EDITING
d684c216 L	196	static const FLASH char erase_seq[] = "\b \b"; /* erase sequence */
	197	static const FLASH char tab_seq[] = " "; /* used to expand TABs */
	198
d684c216 L	199	static char delete_char (char buffer, char p, int colp, int *np, int plen)
	200	{
	201	char *s;
	202
	203	if (*np == 0)
	204	return p;
	205
	206	if ((--p) == '\t') { / will retype the whole line */
	207	while (*colp > plen) {
	208	my_puts_P(erase_seq);
	209	(*colp)--;
	210	}
	211	for (s = buffer; s < p; ++s) {
	212	if (*s == '\t') {
	213	my_puts_P(tab_seq + ((*colp) & 07));
	214	colp += 8 - ((colp) & 07);
	215	} else {
	216	++(*colp);
	217	putchar(*s);
	218	}
	219	}
	220	} else {
	221	my_puts_P(erase_seq);
	222	(*colp)--;
	223	}
	224	(*np)--;
	225
	226	return p;
	227	}
	228	#endif /* CONFIG_CMDLINE_EDITING */
	229
	230
	231	#ifdef CONFIG_CMDLINE_EDITING
	232
	233	/*
	234	* cmdline-editing related codes from vivi.
	235	* Author: Janghoon Lyu <nandy@mizi.com>
	236	*/
	237
e1a50c19 L	238	static void putnstr(char *str, int n)
	239	{
	240	/* printf_P(PSTR("%.s"), (int)n, str) /
8591c65b	241	while (n-- && *str)
e1a50c19 L	242	putchar(*str++);
e1a50c19 L	243	}
d684c216	244
e1a50c19 L	245
e1a50c19 L	246	#define CTL_CH(c) ((c) - 'a' + 1)
d684c216	247	#define CTL_BACKSPACE ('\b')
e1a50c19 L	248	#define DEL ((char)255)
e1a50c19 L	249	#define DEL7 ((char)127)
d684c216 L	250	#define CREAD_HIST_CHAR ('!')
	251
	252	#define getcmd_putch(ch) putchar(ch)
8506d791	253	#define getcmd_getch() vt_parse()
d684c216 L	254	#define getcmd_cbeep() getcmd_putch('\a')
d684c216 L	255
e1a50c19 L	256	#define HIST_MAX 5
e1a50c19 L	257	#define HIST_SIZE CONFIG_SYS_CBSIZE
d684c216 L	258
	259	static int hist_max;
	260	static int hist_add_idx;
	261	static int hist_cur = -1;
	262	static unsigned hist_num;
	263
	264	static char *hist_list[HIST_MAX];
	265	static char hist_lines[HIST_MAX][HIST_SIZE + 1]; /* Save room for NULL */
	266
	267	#define add_idx_minus_one() ((hist_add_idx == 0) ? hist_max : hist_add_idx-1)
	268
	269	static void hist_init(void)
	270	{
	271	int i;
	272
	273	hist_max = 0;
	274	hist_add_idx = 0;
	275	hist_cur = -1;
	276	hist_num = 0;
	277
	278	for (i = 0; i < HIST_MAX; i++) {
	279	hist_list[i] = hist_lines[i];
	280	hist_list[i][0] = '\0';
	281	}
	282	}
	283
	284	static void cread_add_to_hist(char *line)
	285	{
	286	strcpy(hist_list[hist_add_idx], line);
	287
	288	if (++hist_add_idx >= HIST_MAX)
	289	hist_add_idx = 0;
	290
	291	if (hist_add_idx > hist_max)
	292	hist_max = hist_add_idx;
	293
	294	hist_num++;
	295	}
	296
	297	static char *hist_prev(void)
	298	{
	299	char *ret;
	300	int old_cur;
	301
	302	if (hist_cur < 0)
	303	return NULL;
	304
	305	old_cur = hist_cur;
	306	if (--hist_cur < 0)
	307	hist_cur = hist_max;
	308
	309	if (hist_cur == hist_add_idx) {
	310	hist_cur = old_cur;
	311	ret = NULL;
	312	} else {
	313	ret = hist_list[hist_cur];
	314	}
	315
	316	return ret;
	317	}
	318
	319	static char *hist_next(void)
	320	{
	321	char *ret;
322
323	if (hist_cur < 0)
324	return NULL;
325
326	if (hist_cur == hist_add_idx)
327	return NULL;
328
329	if (++hist_cur > hist_max)
330	hist_cur = 0;
331
332	if (hist_cur == hist_add_idx)
333	ret = "";
334	else
335	ret = hist_list[hist_cur];
336
337	return ret;
338	}
339
d684c216	340
e1a50c19 L	341	#define BEGINNING_OF_LINE() { \
	342	while (num) { \
	343	getcmd_putch(CTL_BACKSPACE); \
	344	num--; \
	345	} \
d684c216 L	346	}
d684c216 L	347
e1a50c19 L	348	#define ERASE_TO_EOL() { \
	349	if (num < eol_num) { \
	350	/* printf_P(PSTR("%S"), (int)(eol_num - num), PSTR("")); / \
	351	print_blanks(eol_num - num); \
	352	do { \
	353	getcmd_putch(CTL_BACKSPACE); \
	354	} while (--eol_num > num); \
	355	} \
d684c216 L	356	}
d684c216 L	357
e1a50c19 L	358	#define REFRESH_TO_EOL() { \
	359	if (num < eol_num) { \
	360	wlen = eol_num - num; \
	361	putnstr(buf + num, wlen); \
	362	num = eol_num; \
	363	} \
d684c216 L	364	}
d684c216 L	365
8591c65b L	366	static void cread_add_char(char ichar, int insert, unsigned int *num,
8591c65b L	367	unsigned int eol_num, char buf, unsigned int len)
d684c216	368	{
8591c65b	369	unsigned int wlen;
d684c216 L	370
	371	/* room ??? */
	372	if (insert \|\| num == eol_num) {
	373	if (*eol_num > len - 1) {
	374	getcmd_cbeep();
	375	return;
	376	}
	377	(*eol_num)++;
	378	}
	379
	380	if (insert) {
	381	wlen = eol_num - num;
	382	if (wlen > 1)
	383	memmove(&buf[num+1], &buf[num], wlen-1);
	384
	385	buf[*num] = ichar;
	386	putnstr(buf + *num, wlen);
	387	(*num)++;
	388	while (--wlen)
	389	getcmd_putch(CTL_BACKSPACE);
	390	} else {
	391	/* echo the character */
	392	wlen = 1;
	393	buf[*num] = ichar;
	394	putnstr(buf + *num, wlen);
	395	(*num)++;
	396	}
	397	}
	398
	399	static void cread_add_str(char *str, int strsize, int insert,
8591c65b L	400	unsigned int num, unsigned int eol_num,
8591c65b L	401	char *buf, unsigned int len)
d684c216 L	402	{
	403	while (strsize--) {
	404	cread_add_char(*str, insert, num, eol_num, buf, len);
	405	str++;
	406	}
	407	}
	408
	409	static int cread_line(const FLASH char const prompt, char buf, unsigned int *len)
	410	{
8591c65b L	411	unsigned int num = 0;
	412	unsigned int eol_num = 0;
	413	unsigned int wlen;
8506d791	414	int ichar;
d684c216	415	int insert = 1;
d684c216 L	416	int init_len = strlen(buf);
	417
	418	(void) prompt;
	419
	420	if (init_len)
	421	cread_add_str(buf, init_len, 1, &num, &eol_num, buf, *len);
	422
	423	while (1) {
	424	ichar = getcmd_getch();
	425
	426	if ((ichar == '\n') \|\| (ichar == '\r')) {
	427	putchar('\n');
	428	break;
	429	}
	430
d684c216 L	431
d684c216 L	432	switch (ichar) {
d684c216	433
8506d791	434	case KEY_HOME:
d684c216 L	435	case CTL_CH('a'):
	436	BEGINNING_OF_LINE();
	437	break;
	438	case CTL_CH('c'): /* ^C - break */
	439	buf = '\0'; / discard input */
	440	return -1;
8506d791	441	case KEY_RIGHT:
d684c216 L	442	case CTL_CH('f'):
	443	if (num < eol_num) {
	444	getcmd_putch(buf[num]);
	445	num++;
	446	}
	447	break;
8506d791	448	case KEY_LEFT:
d684c216 L	449	case CTL_CH('b'):
	450	if (num) {
	451	getcmd_putch(CTL_BACKSPACE);
	452	num--;
	453	}
	454	break;
8506d791	455	case KEY_DC:
d684c216 L	456	case CTL_CH('d'):
	457	if (num < eol_num) {
	458	wlen = eol_num - num - 1;
	459	if (wlen) {
	460	memmove(&buf[num], &buf[num+1], wlen);
	461	putnstr(buf + num, wlen);
	462	}
	463
	464	getcmd_putch(' ');
	465	do {
	466	getcmd_putch(CTL_BACKSPACE);
	467	} while (wlen--);
	468	eol_num--;
	469	}
	470	break;
	471	case CTL_CH('k'):
	472	ERASE_TO_EOL();
	473	break;
	474	case CTL_CH('e'):
	475	REFRESH_TO_EOL();
	476	break;
8506d791	477	case KEY_IC:
d684c216 L	478	case CTL_CH('o'):
	479	insert = !insert;
	480	break;
	481	case CTL_CH('x'):
	482	case CTL_CH('u'):
	483	BEGINNING_OF_LINE();
	484	ERASE_TO_EOL();
	485	break;
	486	case DEL:
	487	case DEL7:
	488	case 8:
	489	if (num) {
	490	wlen = eol_num - num;
	491	num--;
	492	memmove(&buf[num], &buf[num+1], wlen);
	493	getcmd_putch(CTL_BACKSPACE);
	494	putnstr(buf + num, wlen);
	495	getcmd_putch(' ');
	496	do {
	497	getcmd_putch(CTL_BACKSPACE);
	498	} while (wlen--);
	499	eol_num--;
	500	}
	501	break;
8506d791	502	case KEY_UP:
d684c216	503	case CTL_CH('p'):
8506d791	504	case KEY_DOWN:
d684c216 L	505	case CTL_CH('n'):
	506	{
	507	char *hline;
	508
8506d791	509	if (ichar == CTL_CH('p') \|\| ichar == KEY_UP)
d684c216 L	510	hline = hist_prev();
	511	else
	512	hline = hist_next();
	513
	514	if (!hline) {
	515	getcmd_cbeep();
	516	continue;
	517	}
	518
	519	/* nuke the current line */
	520	/* first, go home */
	521	BEGINNING_OF_LINE();
	522
	523	/* erase to end of line */
	524	ERASE_TO_EOL();
	525
	526	/* copy new line into place and display */
	527	strcpy(buf, hline);
	528	eol_num = strlen(buf);
	529	REFRESH_TO_EOL();
	530	continue;
	531	}
	532	#ifdef CONFIG_AUTO_COMPLETE
	533	case '\t': {
	534	int num2, col;
	535
	536	/* do not autocomplete when in the middle */
	537	if (num < eol_num) {
	538	getcmd_cbeep();
	539	break;
	540	}
	541
	542	buf[num] = '\0';
	543	col = strlen_P(prompt) + eol_num;
	544	num2 = num;
	545	if (cmd_auto_complete(prompt, buf, &num2, &col)) {
	546	col = num2 - num;
	547	num += col;
	548	eol_num += col;
	549	}
	550	break;
	551	}
	552	#endif
	553	default:
8506d791 L	554	if (isprint(ichar))
	555	cread_add_char(ichar, insert, &num, &eol_num, buf,
	556	*len);
d684c216 L	557	break;
	558	}
	559	}
	560	*len = eol_num;
	561	buf[eol_num] = '\0'; /* lose the newline */
	562
8506d791	563	if (buf[0] /* && buf[0] != CREAD_HIST_CHAR */)
d684c216 L	564	cread_add_to_hist(buf);
	565	hist_cur = hist_add_idx;
	566
	567	return 0;
	568	}
	569
	570	#endif /* CONFIG_CMDLINE_EDITING */
	571
	572	/****************************************************************************/
	573
	574	static int cli_readline_into_buffer(const FLASH char const prompt, char buffer)
	575	{
	576	char *p = buffer;
	577	#ifdef CONFIG_CMDLINE_EDITING
	578	unsigned int len = CONFIG_SYS_CBSIZE;
	579	int rc;
	580	static int initted;
	581
	582	if (!initted) {
	583	hist_init();
	584	initted = 1;
	585	}
	586
	587	if (prompt)
	588	my_puts_P(prompt);
	589
	590	rc = cread_line(prompt, p, &len);
	591	return rc < 0 ? rc : (int) len;
	592
	593	#else /* CONFIG_CMDLINE_EDITING */
	594	char *p_buf = p;
	595	int n = 0; /* buffer index */
	596	int plen = 0; /* prompt length */
	597	int col; /* output column cnt */
	598	char c;
	599
	600	/* print prompt */
	601	if (prompt) {
	602	plen = strlen_P(prompt);
	603	my_puts_P(prompt);
	604	}
	605	col = plen;
	606
	607	for (;;) {
	608
424b184a	609	c = my_getchar(1);
d684c216 L	610
	611	/*
	612	* Special character handling
	613	*/
	614	switch (c) {
	615	case '\r': /* Enter */
	616	case '\n':
	617	*p = '\0';
	618	my_puts_P(PSTR("\r\n"));
	619	return p - p_buf;
	620
	621	case '\0': /* nul */
	622	continue;
	623
	624	case 0x03: /* ^C - break */
	625	p_buf[0] = '\0'; /* discard input */
	626	return -1;
	627
	628	case 0x15: /* ^U - erase line */
	629	while (col > plen) {
	630	my_puts_P(erase_seq);
	631	--col;
	632	}
	633	p = p_buf;
	634	n = 0;
	635	continue;
	636
	637	case 0x17: /* ^W - erase word */
	638	p = delete_char(p_buf, p, &col, &n, plen);
	639	while ((n > 0) && (*p != ' '))
	640	p = delete_char(p_buf, p, &col, &n, plen);
	641	continue;
	642
	643	case 0x08: /* ^H - backspace */
	644	case 0x7F: /* DEL - backspace */
	645	p = delete_char(p_buf, p, &col, &n, plen);
	646	continue;
	647
	648	default:
	649	/*
	650	* Must be a normal character then
	651	*/
	652	if (n < CONFIG_SYS_CBSIZE-2) {
	653	if (c == '\t') { /* expand TABs */
	654	#ifdef CONFIG_AUTO_COMPLETE
	655	/*
	656	* if auto completion triggered just
	657	* continue
	658	*/
	659	*p = '\0';
	660	if (cmd_auto_complete(prompt,
	661	console_buffer,
	662	&n, &col)) {
	663	p = p_buf + n; /* reset */
	664	continue;
	665	}
	666	#endif
	667	my_puts_P(tab_seq + (col & 07));
	668	col += 8 - (col & 07);
	669	} else {
d684c216	670	++col;
cb52e1e0	671	putchar(c);
d684c216 L	672	}
	673	*p++ = c;
	674	++n;
	675	} else { /* Buffer full */
	676	putchar('\a');
	677	}
	678	}
	679	}
	680	#endif /* CONFIG_CMDLINE_EDITING */
	681	}
	682
	683	int cli_readline(const FLASH char *const prompt)
	684	{
	685	/*
	686	* If console_buffer isn't 0-length the user will be prompted to modify
	687	* it instead of entering it from scratch as desired.
	688	*/
	689	console_buffer[0] = '\0';
	690
	691	return cli_readline_into_buffer(prompt, console_buffer);
	692	}