477 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			477 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			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 "readline.h"
 | |
| #include "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 */
 | |
| 	free(rs->history[0]);
 | |
| 	memcpy(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 = 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++] = qemu_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 = qemu_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);
 | |
|     qemu_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);
 | |
|     }
 | |
| }
 | |
| 
 | |
| /* 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 = qemu_mallocz(sizeof(*rs));
 | |
| 
 | |
|     rs->hist_entry = -1;
 | |
|     rs->mon = mon;
 | |
|     rs->completion_finder = completion_finder;
 | |
| 
 | |
|     return rs;
 | |
| }
 |