[qemu.git] / readline.c

/*
 * QEMU readline utility
 *
 * Copyright (c) 2003-2004 Fabrice Bellard
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
#include "monitor/readline.h"
#include "monitor/monitor.h"

#define IS_NORM 0
#define IS_ESC  1
#define IS_CSI  2

#undef printf
#define printf do_not_use_printf

void readline_show_prompt(ReadLineState *rs)
{
    monitor_printf(rs->mon, "%s", rs->prompt);
    monitor_flush(rs->mon);
    rs->last_cmd_buf_index = 0;
    rs->last_cmd_buf_size = 0;
    rs->esc_state = IS_NORM;
}

/* update the displayed command line */
static void readline_update(ReadLineState *rs)
{
    int i, delta, len;

    if (rs->cmd_buf_size != rs->last_cmd_buf_size ||
        memcmp(rs->cmd_buf, rs->last_cmd_buf, rs->cmd_buf_size) != 0) {
        for(i = 0; i < rs->last_cmd_buf_index; i++) {
            monitor_printf(rs->mon, "\033[D");
        }
        rs->cmd_buf[rs->cmd_buf_size] = '\0';
        if (rs->read_password) {
            len = strlen(rs->cmd_buf);
            for(i = 0; i < len; i++)
                monitor_printf(rs->mon, "*");
        } else {
            monitor_printf(rs->mon, "%s", rs->cmd_buf);
        }
        monitor_printf(rs->mon, "\033[K");
        memcpy(rs->last_cmd_buf, rs->cmd_buf, rs->cmd_buf_size);
        rs->last_cmd_buf_size = rs->cmd_buf_size;
        rs->last_cmd_buf_index = rs->cmd_buf_size;
    }
    if (rs->cmd_buf_index != rs->last_cmd_buf_index) {
        delta = rs->cmd_buf_index - rs->last_cmd_buf_index;
        if (delta > 0) {
            for(i = 0;i < delta; i++) {
                monitor_printf(rs->mon, "\033[C");
            }
        } else {
            delta = -delta;
            for(i = 0;i < delta; i++) {
                monitor_printf(rs->mon, "\033[D");
            }
        }
        rs->last_cmd_buf_index = rs->cmd_buf_index;
    }
    monitor_flush(rs->mon);
}

static void readline_insert_char(ReadLineState *rs, int ch)
{
    if (rs->cmd_buf_index < READLINE_CMD_BUF_SIZE) {
        memmove(rs->cmd_buf + rs->cmd_buf_index + 1,
                rs->cmd_buf + rs->cmd_buf_index,
                rs->cmd_buf_size - rs->cmd_buf_index);
        rs->cmd_buf[rs->cmd_buf_index] = ch;
        rs->cmd_buf_size++;
        rs->cmd_buf_index++;
    }
}

static void readline_backward_char(ReadLineState *rs)
{
    if (rs->cmd_buf_index > 0) {
        rs->cmd_buf_index--;
    }
}

static void readline_forward_char(ReadLineState *rs)
{
    if (rs->cmd_buf_index < rs->cmd_buf_size) {
        rs->cmd_buf_index++;
    }
}

static void readline_delete_char(ReadLineState *rs)
{
    if (rs->cmd_buf_index < rs->cmd_buf_size) {
        memmove(rs->cmd_buf + rs->cmd_buf_index,
                rs->cmd_buf + rs->cmd_buf_index + 1,
                rs->cmd_buf_size - rs->cmd_buf_index - 1);
        rs->cmd_buf_size--;
    }
}

static void readline_backspace(ReadLineState *rs)
{
    if (rs->cmd_buf_index > 0) {
        readline_backward_char(rs);
        readline_delete_char(rs);
    }
}

static void readline_backword(ReadLineState *rs)
{
    int start;

    if (rs->cmd_buf_index == 0 || rs->cmd_buf_index > rs->cmd_buf_size) {
        return;
    }

    start = rs->cmd_buf_index - 1;

    /* find first word (backwards) */
    while (start > 0) {
        if (!qemu_isspace(rs->cmd_buf[start])) {
            break;
        }

        --start;
    }

    /* find first space (backwards) */
    while (start > 0) {
        if (qemu_isspace(rs->cmd_buf[start])) {
            ++start;
            break;
        }

        --start;
    }

    /* remove word */
    if (start < rs->cmd_buf_index) {
        memmove(rs->cmd_buf + start,
                rs->cmd_buf + rs->cmd_buf_index,
                rs->cmd_buf_size - rs->cmd_buf_index);
        rs->cmd_buf_size -= rs->cmd_buf_index - start;
        rs->cmd_buf_index = start;
    }
}

static void readline_bol(ReadLineState *rs)
{
    rs->cmd_buf_index = 0;
}

static void readline_eol(ReadLineState *rs)
{
    rs->cmd_buf_index = rs->cmd_buf_size;
}

static void readline_up_char(ReadLineState *rs)
{
    int idx;

    if (rs->hist_entry == 0)
	return;
    if (rs->hist_entry == -1) {
	/* Find latest entry */
	for (idx = 0; idx < READLINE_MAX_CMDS; idx++) {
	    if (rs->history[idx] == NULL)
		break;
	}
	rs->hist_entry = idx;
    }
    rs->hist_entry--;
    if (rs->hist_entry >= 0) {
	pstrcpy(rs->cmd_buf, sizeof(rs->cmd_buf),
                rs->history[rs->hist_entry]);
	rs->cmd_buf_index = rs->cmd_buf_size = strlen(rs->cmd_buf);
    }
}

static void readline_down_char(ReadLineState *rs)
{
    if (rs->hist_entry == -1)
        return;
    if (rs->hist_entry < READLINE_MAX_CMDS - 1 &&
        rs->history[++rs->hist_entry] != NULL) {
	pstrcpy(rs->cmd_buf, sizeof(rs->cmd_buf),
                rs->history[rs->hist_entry]);
    } else {
        rs->cmd_buf[0] = 0;
	rs->hist_entry = -1;
    }
    rs->cmd_buf_index = rs->cmd_buf_size = strlen(rs->cmd_buf);
}

static void readline_hist_add(ReadLineState *rs, const char *cmdline)
{
    char *hist_entry, *new_entry;
    int idx;

    if (cmdline[0] == '\0')
	return;
    new_entry = NULL;
    if (rs->hist_entry != -1) {
	/* We were editing an existing history entry: replace it */
	hist_entry = rs->history[rs->hist_entry];
	idx = rs->hist_entry;
	if (strcmp(hist_entry, cmdline) == 0) {
	    goto same_entry;
	}
    }
    /* Search cmdline in history buffers */
    for (idx = 0; idx < READLINE_MAX_CMDS; idx++) {
	hist_entry = rs->history[idx];
	if (hist_entry == NULL)
	    break;
	if (strcmp(hist_entry, cmdline) == 0) {
	same_entry:
	    new_entry = hist_entry;
	    /* Put this entry at the end of history */
	    memmove(&rs->history[idx], &rs->history[idx + 1],
		    (READLINE_MAX_CMDS - (idx + 1)) * sizeof(char *));
	    rs->history[READLINE_MAX_CMDS - 1] = NULL;
	    for (; idx < READLINE_MAX_CMDS; idx++) {
		if (rs->history[idx] == NULL)
		    break;
	    }
	    break;
	}
    }
    if (idx == READLINE_MAX_CMDS) {
	/* Need to get one free slot */
        g_free(rs->history[0]);
	memmove(rs->history, &rs->history[1],
	        (READLINE_MAX_CMDS - 1) * sizeof(char *));
	rs->history[READLINE_MAX_CMDS - 1] = NULL;
	idx = READLINE_MAX_CMDS - 1;
    }
    if (new_entry == NULL)
        new_entry = g_strdup(cmdline);
    rs->history[idx] = new_entry;
    rs->hist_entry = -1;
}

/* completion support */

void readline_add_completion(ReadLineState *rs, const char *str)
{
    if (rs->nb_completions < READLINE_MAX_COMPLETIONS) {
        rs->completions[rs->nb_completions++] = g_strdup(str);
    }
}

void readline_set_completion_index(ReadLineState *rs, int index)
{
    rs->completion_index = index;
}

static void readline_completion(ReadLineState *rs)
{
    Monitor *mon = cur_mon;
    int len, i, j, max_width, nb_cols, max_prefix;
    char *cmdline;

    rs->nb_completions = 0;

    cmdline = g_malloc(rs->cmd_buf_index + 1);
    memcpy(cmdline, rs->cmd_buf, rs->cmd_buf_index);
    cmdline[rs->cmd_buf_index] = '\0';
    rs->completion_finder(cmdline);
    g_free(cmdline);

    /* no completion found */
    if (rs->nb_completions <= 0)
        return;
    if (rs->nb_completions == 1) {
        len = strlen(rs->completions[0]);
        for(i = rs->completion_index; i < len; i++) {
            readline_insert_char(rs, rs->completions[0][i]);
        }
        /* extra space for next argument. XXX: make it more generic */
        if (len > 0 && rs->completions[0][len - 1] != '/')
            readline_insert_char(rs, ' ');
    } else {
        monitor_printf(mon, "\n");
        max_width = 0;
        max_prefix = 0;	
        for(i = 0; i < rs->nb_completions; i++) {
            len = strlen(rs->completions[i]);
            if (i==0) {
                max_prefix = len;
            } else {
                if (len < max_prefix)
                    max_prefix = len;
                for(j=0; j<max_prefix; j++) {
                    if (rs->completions[i][j] != rs->completions[0][j])
                        max_prefix = j;
                }
            }
            if (len > max_width)
                max_width = len;
        }
        if (max_prefix > 0) 
            for(i = rs->completion_index; i < max_prefix; i++) {
                readline_insert_char(rs, rs->completions[0][i]);
            }
        max_width += 2;
        if (max_width < 10)
            max_width = 10;
        else if (max_width > 80)
            max_width = 80;
        nb_cols = 80 / max_width;
        j = 0;
        for(i = 0; i < rs->nb_completions; i++) {
            monitor_printf(rs->mon, "%-*s", max_width, rs->completions[i]);
            if (++j == nb_cols || i == (rs->nb_completions - 1)) {
                monitor_printf(rs->mon, "\n");
                j = 0;
            }
        }
        readline_show_prompt(rs);
    }
    for (i = 0; i < rs->nb_completions; i++) {
        g_free(rs->completions[i]);
    }
}

/* return true if command handled */
void readline_handle_byte(ReadLineState *rs, int ch)
{
    switch(rs->esc_state) {
    case IS_NORM:
        switch(ch) {
        case 1:
            readline_bol(rs);
            break;
        case 4:
            readline_delete_char(rs);
            break;
        case 5:
            readline_eol(rs);
            break;
        case 9:
            readline_completion(rs);
            break;
        case 10:
        case 13:
            rs->cmd_buf[rs->cmd_buf_size] = '\0';
            if (!rs->read_password)
                readline_hist_add(rs, rs->cmd_buf);
            monitor_printf(rs->mon, "\n");
            rs->cmd_buf_index = 0;
            rs->cmd_buf_size = 0;
            rs->last_cmd_buf_index = 0;
            rs->last_cmd_buf_size = 0;
            rs->readline_func(rs->mon, rs->cmd_buf, rs->readline_opaque);
            break;
        case 23:
            /* ^W */
            readline_backword(rs);
            break;
        case 27:
            rs->esc_state = IS_ESC;
            break;
        case 127:
        case 8:
            readline_backspace(rs);
            break;
	case 155:
            rs->esc_state = IS_CSI;
	    break;
        default:
            if (ch >= 32) {
                readline_insert_char(rs, ch);
            }
            break;
        }
        break;
    case IS_ESC:
        if (ch == '[') {
            rs->esc_state = IS_CSI;
            rs->esc_param = 0;
        } else {
            rs->esc_state = IS_NORM;
        }
        break;
    case IS_CSI:
        switch(ch) {
	case 'A':
	case 'F':
	    readline_up_char(rs);
	    break;
	case 'B':
	case 'E':
	    readline_down_char(rs);
	    break;
        case 'D':
            readline_backward_char(rs);
            break;
        case 'C':
            readline_forward_char(rs);
            break;
        case '0' ... '9':
            rs->esc_param = rs->esc_param * 10 + (ch - '0');
            goto the_end;
        case '~':
            switch(rs->esc_param) {
            case 1:
                readline_bol(rs);
                break;
            case 3:
                readline_delete_char(rs);
                break;
            case 4:
                readline_eol(rs);
                break;
            }
            break;
        default:
            break;
        }
        rs->esc_state = IS_NORM;
    the_end:
        break;
    }
    readline_update(rs);
}

void readline_start(ReadLineState *rs, const char *prompt, int read_password,
                    ReadLineFunc *readline_func, void *opaque)
{
    pstrcpy(rs->prompt, sizeof(rs->prompt), prompt);
    rs->readline_func = readline_func;
    rs->readline_opaque = opaque;
    rs->read_password = read_password;
    readline_restart(rs);
}

void readline_restart(ReadLineState *rs)
{
    rs->cmd_buf_index = 0;
    rs->cmd_buf_size = 0;
}

const char *readline_get_history(ReadLineState *rs, unsigned int index)
{
    if (index >= READLINE_MAX_CMDS)
        return NULL;
    return rs->history[index];
}

ReadLineState *readline_init(Monitor *mon,
                             ReadLineCompletionFunc *completion_finder)
{
    ReadLineState *rs = g_malloc0(sizeof(*rs));

    rs->hist_entry = -1;
    rs->mon = mon;
    rs->completion_finder = completion_finder;

    return rs;
}
Commit	Line	Data
7e2515e8 FB	1	/*
7e2515e8 FB	2	* QEMU readline utility
5fafdf24	3	*
7e2515e8	4	* Copyright (c) 2003-2004 Fabrice Bellard
5fafdf24	5	*
7e2515e8 FB	6	* Permission is hereby granted, free of charge, to any person obtaining a copy
	7	* of this software and associated documentation files (the "Software"), to deal
	8	* in the Software without restriction, including without limitation the rights
	9	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	10	* copies of the Software, and to permit persons to whom the Software is
	11	* furnished to do so, subject to the following conditions:
	12	*
	13	* The above copyright notice and this permission notice shall be included in
	14	* all copies or substantial portions of the Software.
	15	*
	16	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	17	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	18	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
	19	* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	20	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	21	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
	22	* THE SOFTWARE.
	23	*/
83c9089e PB	24	#include "monitor/readline.h"
83c9089e PB	25	#include "monitor/monitor.h"
7e2515e8	26
7e2515e8 FB	27	#define IS_NORM 0
	28	#define IS_ESC 1
	29	#define IS_CSI 2
	30
30bf5f46	31	#undef printf
7e2515e8 FB	32	#define printf do_not_use_printf
7e2515e8 FB	33
4c36ba32	34	void readline_show_prompt(ReadLineState *rs)
7e2515e8	35	{
4c36ba32 AL	36	monitor_printf(rs->mon, "%s", rs->prompt);
	37	monitor_flush(rs->mon);
	38	rs->last_cmd_buf_index = 0;
	39	rs->last_cmd_buf_size = 0;
	40	rs->esc_state = IS_NORM;
7e2515e8 FB	41	}
7e2515e8 FB	42
7e2515e8	43	/* update the displayed command line */
4c36ba32	44	static void readline_update(ReadLineState *rs)
7e2515e8 FB	45	{
	46	int i, delta, len;
	47
4c36ba32 AL	48	if (rs->cmd_buf_size != rs->last_cmd_buf_size \|\|
	49	memcmp(rs->cmd_buf, rs->last_cmd_buf, rs->cmd_buf_size) != 0) {
	50	for(i = 0; i < rs->last_cmd_buf_index; i++) {
	51	monitor_printf(rs->mon, "\033[D");
7e2515e8	52	}
4c36ba32 AL	53	rs->cmd_buf[rs->cmd_buf_size] = '\0';
	54	if (rs->read_password) {
	55	len = strlen(rs->cmd_buf);
7e2515e8	56	for(i = 0; i < len; i++)
4c36ba32	57	monitor_printf(rs->mon, "*");
7e2515e8	58	} else {
4c36ba32	59	monitor_printf(rs->mon, "%s", rs->cmd_buf);
7e2515e8	60	}
4c36ba32 AL	61	monitor_printf(rs->mon, "\033[K");
	62	memcpy(rs->last_cmd_buf, rs->cmd_buf, rs->cmd_buf_size);
	63	rs->last_cmd_buf_size = rs->cmd_buf_size;
	64	rs->last_cmd_buf_index = rs->cmd_buf_size;
7e2515e8	65	}
4c36ba32 AL	66	if (rs->cmd_buf_index != rs->last_cmd_buf_index) {
4c36ba32 AL	67	delta = rs->cmd_buf_index - rs->last_cmd_buf_index;
7e2515e8 FB	68	if (delta > 0) {
7e2515e8 FB	69	for(i = 0;i < delta; i++) {
4c36ba32	70	monitor_printf(rs->mon, "\033[C");
7e2515e8 FB	71	}
	72	} else {
	73	delta = -delta;
	74	for(i = 0;i < delta; i++) {
4c36ba32	75	monitor_printf(rs->mon, "\033[D");
7e2515e8 FB	76	}
7e2515e8 FB	77	}
4c36ba32	78	rs->last_cmd_buf_index = rs->cmd_buf_index;
7e2515e8	79	}
4c36ba32	80	monitor_flush(rs->mon);
7e2515e8 FB	81	}
7e2515e8 FB	82
4c36ba32	83	static void readline_insert_char(ReadLineState *rs, int ch)
7e2515e8	84	{
4c36ba32 AL	85	if (rs->cmd_buf_index < READLINE_CMD_BUF_SIZE) {
	86	memmove(rs->cmd_buf + rs->cmd_buf_index + 1,
	87	rs->cmd_buf + rs->cmd_buf_index,
	88	rs->cmd_buf_size - rs->cmd_buf_index);
	89	rs->cmd_buf[rs->cmd_buf_index] = ch;
	90	rs->cmd_buf_size++;
	91	rs->cmd_buf_index++;
7e2515e8 FB	92	}
	93	}
	94
4c36ba32	95	static void readline_backward_char(ReadLineState *rs)
7e2515e8	96	{
4c36ba32 AL	97	if (rs->cmd_buf_index > 0) {
4c36ba32 AL	98	rs->cmd_buf_index--;
7e2515e8 FB	99	}
	100	}
	101
4c36ba32	102	static void readline_forward_char(ReadLineState *rs)
7e2515e8	103	{
4c36ba32 AL	104	if (rs->cmd_buf_index < rs->cmd_buf_size) {
4c36ba32 AL	105	rs->cmd_buf_index++;
7e2515e8 FB	106	}
	107	}
	108
4c36ba32	109	static void readline_delete_char(ReadLineState *rs)
7e2515e8	110	{
4c36ba32 AL	111	if (rs->cmd_buf_index < rs->cmd_buf_size) {
	112	memmove(rs->cmd_buf + rs->cmd_buf_index,
	113	rs->cmd_buf + rs->cmd_buf_index + 1,
	114	rs->cmd_buf_size - rs->cmd_buf_index - 1);
	115	rs->cmd_buf_size--;
7e2515e8 FB	116	}
	117	}
	118
4c36ba32	119	static void readline_backspace(ReadLineState *rs)
7e2515e8	120	{
4c36ba32 AL	121	if (rs->cmd_buf_index > 0) {
	122	readline_backward_char(rs);
	123	readline_delete_char(rs);
7e2515e8 FB	124	}
	125	}
	126
4c36ba32	127	static void readline_backword(ReadLineState *rs)
33fa11d4 TS	128	{
	129	int start;
	130
4c36ba32	131	if (rs->cmd_buf_index == 0 \|\| rs->cmd_buf_index > rs->cmd_buf_size) {
33fa11d4 TS	132	return;
	133	}
	134
4c36ba32	135	start = rs->cmd_buf_index - 1;
33fa11d4 TS	136
	137	/* find first word (backwards) */
	138	while (start > 0) {
4c36ba32	139	if (!qemu_isspace(rs->cmd_buf[start])) {
33fa11d4 TS	140	break;
	141	}
	142
	143	--start;
	144	}
	145
	146	/* find first space (backwards) */
	147	while (start > 0) {
4c36ba32	148	if (qemu_isspace(rs->cmd_buf[start])) {
33fa11d4 TS	149	++start;
	150	break;
	151	}
	152
	153	--start;
	154	}
	155
	156	/* remove word */
4c36ba32 AL	157	if (start < rs->cmd_buf_index) {
	158	memmove(rs->cmd_buf + start,
	159	rs->cmd_buf + rs->cmd_buf_index,
	160	rs->cmd_buf_size - rs->cmd_buf_index);
	161	rs->cmd_buf_size -= rs->cmd_buf_index - start;
	162	rs->cmd_buf_index = start;
33fa11d4 TS	163	}
	164	}
	165
4c36ba32	166	static void readline_bol(ReadLineState *rs)
7e2515e8	167	{
4c36ba32	168	rs->cmd_buf_index = 0;
7e2515e8 FB	169	}
7e2515e8 FB	170
4c36ba32	171	static void readline_eol(ReadLineState *rs)
7e2515e8	172	{
4c36ba32	173	rs->cmd_buf_index = rs->cmd_buf_size;
7e2515e8 FB	174	}
7e2515e8 FB	175
4c36ba32	176	static void readline_up_char(ReadLineState *rs)
7e2515e8 FB	177	{
	178	int idx;
	179
4c36ba32	180	if (rs->hist_entry == 0)
7e2515e8	181	return;
4c36ba32	182	if (rs->hist_entry == -1) {
7e2515e8	183	/* Find latest entry */
4c36ba32 AL	184	for (idx = 0; idx < READLINE_MAX_CMDS; idx++) {
4c36ba32 AL	185	if (rs->history[idx] == NULL)
7e2515e8 FB	186	break;
7e2515e8 FB	187	}
4c36ba32	188	rs->hist_entry = idx;
7e2515e8	189	}
4c36ba32 AL	190	rs->hist_entry--;
	191	if (rs->hist_entry >= 0) {
	192	pstrcpy(rs->cmd_buf, sizeof(rs->cmd_buf),
	193	rs->history[rs->hist_entry]);
	194	rs->cmd_buf_index = rs->cmd_buf_size = strlen(rs->cmd_buf);
7e2515e8 FB	195	}
	196	}
	197
4c36ba32	198	static void readline_down_char(ReadLineState *rs)
7e2515e8	199	{
5b0d2727 AL	200	if (rs->hist_entry == -1)
	201	return;
	202	if (rs->hist_entry < READLINE_MAX_CMDS - 1 &&
	203	rs->history[++rs->hist_entry] != NULL) {
4c36ba32 AL	204	pstrcpy(rs->cmd_buf, sizeof(rs->cmd_buf),
4c36ba32 AL	205	rs->history[rs->hist_entry]);
7e2515e8	206	} else {
5b0d2727	207	rs->cmd_buf[0] = 0;
4c36ba32	208	rs->hist_entry = -1;
7e2515e8	209	}
4c36ba32	210	rs->cmd_buf_index = rs->cmd_buf_size = strlen(rs->cmd_buf);
7e2515e8 FB	211	}
7e2515e8 FB	212
4c36ba32	213	static void readline_hist_add(ReadLineState rs, const char cmdline)
7e2515e8 FB	214	{
	215	char hist_entry, new_entry;
	216	int idx;
	217
	218	if (cmdline[0] == '\0')
	219	return;
	220	new_entry = NULL;
4c36ba32	221	if (rs->hist_entry != -1) {
7e2515e8	222	/* We were editing an existing history entry: replace it */
4c36ba32 AL	223	hist_entry = rs->history[rs->hist_entry];
4c36ba32 AL	224	idx = rs->hist_entry;
7e2515e8 FB	225	if (strcmp(hist_entry, cmdline) == 0) {
	226	goto same_entry;
	227	}
	228	}
	229	/* Search cmdline in history buffers */
4c36ba32 AL	230	for (idx = 0; idx < READLINE_MAX_CMDS; idx++) {
4c36ba32 AL	231	hist_entry = rs->history[idx];
7e2515e8 FB	232	if (hist_entry == NULL)
	233	break;
	234	if (strcmp(hist_entry, cmdline) == 0) {
	235	same_entry:
	236	new_entry = hist_entry;
	237	/* Put this entry at the end of history */
4c36ba32	238	memmove(&rs->history[idx], &rs->history[idx + 1],
8af42882	239	(READLINE_MAX_CMDS - (idx + 1)) * sizeof(char *));
4c36ba32 AL	240	rs->history[READLINE_MAX_CMDS - 1] = NULL;
	241	for (; idx < READLINE_MAX_CMDS; idx++) {
	242	if (rs->history[idx] == NULL)
7e2515e8 FB	243	break;
	244	}
	245	break;
	246	}
	247	}
4c36ba32	248	if (idx == READLINE_MAX_CMDS) {
7e2515e8	249	/* Need to get one free slot */
c3baa5f9	250	g_free(rs->history[0]);
7682e858 NZ	251	memmove(rs->history, &rs->history[1],
7682e858 NZ	252	(READLINE_MAX_CMDS - 1) * sizeof(char *));
4c36ba32 AL	253	rs->history[READLINE_MAX_CMDS - 1] = NULL;
4c36ba32 AL	254	idx = READLINE_MAX_CMDS - 1;
7e2515e8 FB	255	}
7e2515e8 FB	256	if (new_entry == NULL)
c3baa5f9	257	new_entry = g_strdup(cmdline);
4c36ba32 AL	258	rs->history[idx] = new_entry;
4c36ba32 AL	259	rs->hist_entry = -1;
7e2515e8 FB	260	}
	261
	262	/* completion support */
	263
4c36ba32	264	void readline_add_completion(ReadLineState rs, const char str)
7e2515e8	265	{
4c36ba32	266	if (rs->nb_completions < READLINE_MAX_COMPLETIONS) {
7267c094	267	rs->completions[rs->nb_completions++] = g_strdup(str);
7e2515e8 FB	268	}
	269	}
	270
4c36ba32	271	void readline_set_completion_index(ReadLineState *rs, int index)
376253ec	272	{
4c36ba32	273	rs->completion_index = index;
376253ec AL	274	}
376253ec AL	275
4c36ba32	276	static void readline_completion(ReadLineState *rs)
7e2515e8	277	{
376253ec	278	Monitor *mon = cur_mon;
b427c726	279	int len, i, j, max_width, nb_cols, max_prefix;
7e2515e8 FB	280	char *cmdline;
7e2515e8 FB	281
4c36ba32	282	rs->nb_completions = 0;
3b46e624	283
7267c094	284	cmdline = g_malloc(rs->cmd_buf_index + 1);
4c36ba32 AL	285	memcpy(cmdline, rs->cmd_buf, rs->cmd_buf_index);
	286	cmdline[rs->cmd_buf_index] = '\0';
	287	rs->completion_finder(cmdline);
7267c094	288	g_free(cmdline);
7e2515e8 FB	289
7e2515e8 FB	290	/* no completion found */
4c36ba32	291	if (rs->nb_completions <= 0)
7e2515e8	292	return;
4c36ba32 AL	293	if (rs->nb_completions == 1) {
	294	len = strlen(rs->completions[0]);
	295	for(i = rs->completion_index; i < len; i++) {
	296	readline_insert_char(rs, rs->completions[0][i]);
7e2515e8 FB	297	}
7e2515e8 FB	298	/* extra space for next argument. XXX: make it more generic */
4c36ba32 AL	299	if (len > 0 && rs->completions[0][len - 1] != '/')
4c36ba32 AL	300	readline_insert_char(rs, ' ');
7e2515e8	301	} else {
376253ec	302	monitor_printf(mon, "\n");
7e2515e8	303	max_width = 0;
b427c726	304	max_prefix = 0;
4c36ba32 AL	305	for(i = 0; i < rs->nb_completions; i++) {
4c36ba32 AL	306	len = strlen(rs->completions[i]);
b427c726 TS	307	if (i==0) {
	308	max_prefix = len;
	309	} else {
	310	if (len < max_prefix)
	311	max_prefix = len;
	312	for(j=0; j<max_prefix; j++) {
4c36ba32	313	if (rs->completions[i][j] != rs->completions[0][j])
b427c726 TS	314	max_prefix = j;
	315	}
	316	}
7e2515e8 FB	317	if (len > max_width)
	318	max_width = len;
	319	}
b427c726	320	if (max_prefix > 0)
4c36ba32 AL	321	for(i = rs->completion_index; i < max_prefix; i++) {
4c36ba32 AL	322	readline_insert_char(rs, rs->completions[0][i]);
b427c726	323	}
7e2515e8 FB	324	max_width += 2;
	325	if (max_width < 10)
	326	max_width = 10;
	327	else if (max_width > 80)
	328	max_width = 80;
	329	nb_cols = 80 / max_width;
	330	j = 0;
4c36ba32 AL	331	for(i = 0; i < rs->nb_completions; i++) {
	332	monitor_printf(rs->mon, "%-*s", max_width, rs->completions[i]);
	333	if (++j == nb_cols \|\| i == (rs->nb_completions - 1)) {
	334	monitor_printf(rs->mon, "\n");
7e2515e8 FB	335	j = 0;
	336	}
	337	}
4c36ba32	338	readline_show_prompt(rs);
7e2515e8	339	}
fc9fa4bd SW	340	for (i = 0; i < rs->nb_completions; i++) {
	341	g_free(rs->completions[i]);
	342	}
7e2515e8 FB	343	}
	344
	345	/* return true if command handled */
4c36ba32	346	void readline_handle_byte(ReadLineState *rs, int ch)
7e2515e8	347	{
4c36ba32	348	switch(rs->esc_state) {
7e2515e8 FB	349	case IS_NORM:
	350	switch(ch) {
	351	case 1:
4c36ba32	352	readline_bol(rs);
7e2515e8 FB	353	break;
7e2515e8 FB	354	case 4:
4c36ba32	355	readline_delete_char(rs);
7e2515e8 FB	356	break;
7e2515e8 FB	357	case 5:
4c36ba32	358	readline_eol(rs);
7e2515e8 FB	359	break;
7e2515e8 FB	360	case 9:
4c36ba32	361	readline_completion(rs);
7e2515e8 FB	362	break;
	363	case 10:
	364	case 13:
4c36ba32 AL	365	rs->cmd_buf[rs->cmd_buf_size] = '\0';
	366	if (!rs->read_password)
	367	readline_hist_add(rs, rs->cmd_buf);
	368	monitor_printf(rs->mon, "\n");
	369	rs->cmd_buf_index = 0;
	370	rs->cmd_buf_size = 0;
	371	rs->last_cmd_buf_index = 0;
	372	rs->last_cmd_buf_size = 0;
	373	rs->readline_func(rs->mon, rs->cmd_buf, rs->readline_opaque);
7e2515e8	374	break;
33fa11d4 TS	375	case 23:
33fa11d4 TS	376	/* ^W */
4c36ba32	377	readline_backword(rs);
33fa11d4	378	break;
7e2515e8	379	case 27:
4c36ba32	380	rs->esc_state = IS_ESC;
7e2515e8 FB	381	break;
	382	case 127:
	383	case 8:
4c36ba32	384	readline_backspace(rs);
7e2515e8 FB	385	break;
7e2515e8 FB	386	case 155:
4c36ba32	387	rs->esc_state = IS_CSI;
7e2515e8 FB	388	break;
	389	default:
	390	if (ch >= 32) {
4c36ba32	391	readline_insert_char(rs, ch);
7e2515e8 FB	392	}
	393	break;
	394	}
	395	break;
	396	case IS_ESC:
	397	if (ch == '[') {
4c36ba32 AL	398	rs->esc_state = IS_CSI;
4c36ba32 AL	399	rs->esc_param = 0;
7e2515e8	400	} else {
4c36ba32	401	rs->esc_state = IS_NORM;
7e2515e8 FB	402	}
	403	break;
	404	case IS_CSI:
	405	switch(ch) {
	406	case 'A':
	407	case 'F':
4c36ba32	408	readline_up_char(rs);
7e2515e8 FB	409	break;
	410	case 'B':
	411	case 'E':
4c36ba32	412	readline_down_char(rs);
7e2515e8 FB	413	break;
7e2515e8 FB	414	case 'D':
4c36ba32	415	readline_backward_char(rs);
7e2515e8 FB	416	break;
7e2515e8 FB	417	case 'C':
4c36ba32	418	readline_forward_char(rs);
7e2515e8 FB	419	break;
7e2515e8 FB	420	case '0' ... '9':
4c36ba32	421	rs->esc_param = rs->esc_param * 10 + (ch - '0');
7e2515e8 FB	422	goto the_end;
7e2515e8 FB	423	case '~':
4c36ba32	424	switch(rs->esc_param) {
7e2515e8	425	case 1:
4c36ba32	426	readline_bol(rs);
7e2515e8 FB	427	break;
7e2515e8 FB	428	case 3:
4c36ba32	429	readline_delete_char(rs);
7e2515e8 FB	430	break;
7e2515e8 FB	431	case 4:
4c36ba32	432	readline_eol(rs);
7e2515e8 FB	433	break;
	434	}
	435	break;
	436	default:
	437	break;
	438	}
4c36ba32	439	rs->esc_state = IS_NORM;
7e2515e8 FB	440	the_end:
	441	break;
	442	}
4c36ba32	443	readline_update(rs);
7e2515e8 FB	444	}
7e2515e8 FB	445
4c36ba32	446	void readline_start(ReadLineState rs, const char prompt, int read_password,
7e2515e8 FB	447	ReadLineFunc readline_func, void opaque)
7e2515e8 FB	448	{
4c36ba32 AL	449	pstrcpy(rs->prompt, sizeof(rs->prompt), prompt);
	450	rs->readline_func = readline_func;
	451	rs->readline_opaque = opaque;
	452	rs->read_password = read_password;
2724b180 AL	453	readline_restart(rs);
	454	}
	455
	456	void readline_restart(ReadLineState *rs)
	457	{
4c36ba32 AL	458	rs->cmd_buf_index = 0;
4c36ba32 AL	459	rs->cmd_buf_size = 0;
7e2515e8 FB	460	}
7e2515e8 FB	461
4c36ba32	462	const char readline_get_history(ReadLineState rs, unsigned int index)
7e2515e8	463	{
4c36ba32	464	if (index >= READLINE_MAX_CMDS)
7e2515e8	465	return NULL;
4c36ba32 AL	466	return rs->history[index];
	467	}
	468
	469	ReadLineState readline_init(Monitor mon,
	470	ReadLineCompletionFunc *completion_finder)
	471	{
7267c094	472	ReadLineState rs = g_malloc0(sizeof(rs));
4c36ba32	473
4c36ba32 AL	474	rs->hist_entry = -1;
	475	rs->mon = mon;
	476	rs->completion_finder = completion_finder;
	477
	478	return rs;
7e2515e8	479	}