1 /* 2 * Command line editing and history 3 * Copyright (c) 2010-2011, Jouni Malinen <j@w1.fi> 4 * 5 * This software may be distributed under the terms of the BSD license. 6 * See README for more details. 7 */ 8 9 #include "includes.h" 10 #include <termios.h> 11 12 #include "common.h" 13 #include "eloop.h" 14 #include "list.h" 15 #include "edit.h" 16 17 #define CMD_BUF_LEN 4096 18 static char cmdbuf[CMD_BUF_LEN]; 19 static int cmdbuf_pos = 0; 20 static int cmdbuf_len = 0; 21 static char currbuf[CMD_BUF_LEN]; 22 static int currbuf_valid = 0; 23 static const char *ps2 = NULL; 24 25 #define HISTORY_MAX 100 26 27 struct edit_history { 28 struct dl_list list; 29 char str[1]; 30 }; 31 32 static struct dl_list history_list; 33 static struct edit_history *history_curr; 34 35 static void *edit_cb_ctx; 36 static void (*edit_cmd_cb)(void *ctx, char *cmd); 37 static void (*edit_eof_cb)(void *ctx); 38 static char ** (*edit_completion_cb)(void *ctx, const char *cmd, int pos) = 39 NULL; 40 41 static struct termios prevt, newt; 42 43 44 #define CLEAR_END_LINE "\e[K" 45 46 edit_clear_line(void)47 void edit_clear_line(void) 48 { 49 int i; 50 putchar('\r'); 51 for (i = 0; i < cmdbuf_len + 2 + (ps2 ? (int) os_strlen(ps2) : 0); i++) 52 putchar(' '); 53 } 54 55 move_start(void)56 static void move_start(void) 57 { 58 cmdbuf_pos = 0; 59 edit_redraw(); 60 } 61 62 move_end(void)63 static void move_end(void) 64 { 65 cmdbuf_pos = cmdbuf_len; 66 edit_redraw(); 67 } 68 69 move_left(void)70 static void move_left(void) 71 { 72 if (cmdbuf_pos > 0) { 73 cmdbuf_pos--; 74 edit_redraw(); 75 } 76 } 77 78 move_right(void)79 static void move_right(void) 80 { 81 if (cmdbuf_pos < cmdbuf_len) { 82 cmdbuf_pos++; 83 edit_redraw(); 84 } 85 } 86 87 move_word_left(void)88 static void move_word_left(void) 89 { 90 while (cmdbuf_pos > 0 && cmdbuf[cmdbuf_pos - 1] == ' ') 91 cmdbuf_pos--; 92 while (cmdbuf_pos > 0 && cmdbuf[cmdbuf_pos - 1] != ' ') 93 cmdbuf_pos--; 94 edit_redraw(); 95 } 96 97 move_word_right(void)98 static void move_word_right(void) 99 { 100 while (cmdbuf_pos < cmdbuf_len && cmdbuf[cmdbuf_pos] == ' ') 101 cmdbuf_pos++; 102 while (cmdbuf_pos < cmdbuf_len && cmdbuf[cmdbuf_pos] != ' ') 103 cmdbuf_pos++; 104 edit_redraw(); 105 } 106 107 delete_left(void)108 static void delete_left(void) 109 { 110 if (cmdbuf_pos == 0) 111 return; 112 113 edit_clear_line(); 114 os_memmove(cmdbuf + cmdbuf_pos - 1, cmdbuf + cmdbuf_pos, 115 cmdbuf_len - cmdbuf_pos); 116 cmdbuf_pos--; 117 cmdbuf_len--; 118 edit_redraw(); 119 } 120 121 delete_current(void)122 static void delete_current(void) 123 { 124 if (cmdbuf_pos == cmdbuf_len) 125 return; 126 127 edit_clear_line(); 128 os_memmove(cmdbuf + cmdbuf_pos, cmdbuf + cmdbuf_pos + 1, 129 cmdbuf_len - cmdbuf_pos); 130 cmdbuf_len--; 131 edit_redraw(); 132 } 133 134 delete_word(void)135 static void delete_word(void) 136 { 137 int pos; 138 139 edit_clear_line(); 140 pos = cmdbuf_pos; 141 while (pos > 0 && cmdbuf[pos - 1] == ' ') 142 pos--; 143 while (pos > 0 && cmdbuf[pos - 1] != ' ') 144 pos--; 145 os_memmove(cmdbuf + pos, cmdbuf + cmdbuf_pos, cmdbuf_len - cmdbuf_pos); 146 cmdbuf_len -= cmdbuf_pos - pos; 147 cmdbuf_pos = pos; 148 edit_redraw(); 149 } 150 151 clear_left(void)152 static void clear_left(void) 153 { 154 if (cmdbuf_pos == 0) 155 return; 156 157 edit_clear_line(); 158 os_memmove(cmdbuf, cmdbuf + cmdbuf_pos, cmdbuf_len - cmdbuf_pos); 159 cmdbuf_len -= cmdbuf_pos; 160 cmdbuf_pos = 0; 161 edit_redraw(); 162 } 163 164 clear_right(void)165 static void clear_right(void) 166 { 167 if (cmdbuf_pos == cmdbuf_len) 168 return; 169 170 edit_clear_line(); 171 cmdbuf_len = cmdbuf_pos; 172 edit_redraw(); 173 } 174 175 history_add(const char * str)176 static void history_add(const char *str) 177 { 178 struct edit_history *h, *match = NULL, *last = NULL; 179 size_t len, count = 0; 180 181 if (str[0] == '\0') 182 return; 183 184 dl_list_for_each(h, &history_list, struct edit_history, list) { 185 if (os_strcmp(str, h->str) == 0) { 186 match = h; 187 break; 188 } 189 last = h; 190 count++; 191 } 192 193 if (match) { 194 dl_list_del(&h->list); 195 dl_list_add(&history_list, &h->list); 196 history_curr = h; 197 return; 198 } 199 200 if (count >= HISTORY_MAX && last) { 201 dl_list_del(&last->list); 202 os_free(last); 203 } 204 205 len = os_strlen(str); 206 h = os_zalloc(sizeof(*h) + len); 207 if (h == NULL) 208 return; 209 dl_list_add(&history_list, &h->list); 210 os_strlcpy(h->str, str, len + 1); 211 history_curr = h; 212 } 213 214 history_use(void)215 static void history_use(void) 216 { 217 edit_clear_line(); 218 cmdbuf_len = cmdbuf_pos = os_strlen(history_curr->str); 219 os_memcpy(cmdbuf, history_curr->str, cmdbuf_len); 220 edit_redraw(); 221 } 222 223 history_prev(void)224 static void history_prev(void) 225 { 226 if (history_curr == NULL) 227 return; 228 229 if (history_curr == 230 dl_list_first(&history_list, struct edit_history, list)) { 231 if (!currbuf_valid) { 232 cmdbuf[cmdbuf_len] = '\0'; 233 os_memcpy(currbuf, cmdbuf, cmdbuf_len + 1); 234 currbuf_valid = 1; 235 history_use(); 236 return; 237 } 238 } 239 240 if (history_curr == 241 dl_list_last(&history_list, struct edit_history, list)) 242 return; 243 244 history_curr = dl_list_entry(history_curr->list.next, 245 struct edit_history, list); 246 history_use(); 247 } 248 249 history_next(void)250 static void history_next(void) 251 { 252 if (history_curr == NULL || 253 history_curr == 254 dl_list_first(&history_list, struct edit_history, list)) { 255 if (currbuf_valid) { 256 currbuf_valid = 0; 257 edit_clear_line(); 258 cmdbuf_len = cmdbuf_pos = os_strlen(currbuf); 259 os_memcpy(cmdbuf, currbuf, cmdbuf_len); 260 edit_redraw(); 261 } 262 return; 263 } 264 265 history_curr = dl_list_entry(history_curr->list.prev, 266 struct edit_history, list); 267 history_use(); 268 } 269 270 history_read(const char * fname)271 static void history_read(const char *fname) 272 { 273 FILE *f; 274 char buf[CMD_BUF_LEN], *pos; 275 276 f = fopen(fname, "r"); 277 if (f == NULL) 278 return; 279 280 while (fgets(buf, CMD_BUF_LEN, f)) { 281 for (pos = buf; *pos; pos++) { 282 if (*pos == '\r' || *pos == '\n') { 283 *pos = '\0'; 284 break; 285 } 286 } 287 history_add(buf); 288 } 289 290 fclose(f); 291 } 292 293 history_write(const char * fname,int (* filter_cb)(void * ctx,const char * cmd))294 static void history_write(const char *fname, 295 int (*filter_cb)(void *ctx, const char *cmd)) 296 { 297 FILE *f; 298 struct edit_history *h; 299 300 f = fopen(fname, "w"); 301 if (f == NULL) 302 return; 303 304 dl_list_for_each_reverse(h, &history_list, struct edit_history, list) { 305 if (filter_cb && filter_cb(edit_cb_ctx, h->str)) 306 continue; 307 fprintf(f, "%s\n", h->str); 308 } 309 310 fclose(f); 311 } 312 313 history_debug_dump(void)314 static void history_debug_dump(void) 315 { 316 struct edit_history *h; 317 edit_clear_line(); 318 printf("\r"); 319 dl_list_for_each_reverse(h, &history_list, struct edit_history, list) 320 printf("%s%s\n", h == history_curr ? "[C]" : "", h->str); 321 if (currbuf_valid) 322 printf("{%s}\n", currbuf); 323 edit_redraw(); 324 } 325 326 insert_char(int c)327 static void insert_char(int c) 328 { 329 if (cmdbuf_len >= (int) sizeof(cmdbuf) - 1) 330 return; 331 if (cmdbuf_len == cmdbuf_pos) { 332 cmdbuf[cmdbuf_pos++] = c; 333 cmdbuf_len++; 334 putchar(c); 335 fflush(stdout); 336 } else { 337 os_memmove(cmdbuf + cmdbuf_pos + 1, cmdbuf + cmdbuf_pos, 338 cmdbuf_len - cmdbuf_pos); 339 cmdbuf[cmdbuf_pos++] = c; 340 cmdbuf_len++; 341 edit_redraw(); 342 } 343 } 344 345 process_cmd(void)346 static void process_cmd(void) 347 { 348 currbuf_valid = 0; 349 if (cmdbuf_len == 0) { 350 printf("\n%s> ", ps2 ? ps2 : ""); 351 fflush(stdout); 352 return; 353 } 354 printf("\n"); 355 cmdbuf[cmdbuf_len] = '\0'; 356 history_add(cmdbuf); 357 cmdbuf_pos = 0; 358 cmdbuf_len = 0; 359 edit_cmd_cb(edit_cb_ctx, cmdbuf); 360 printf("%s> ", ps2 ? ps2 : ""); 361 fflush(stdout); 362 } 363 364 free_completions(char ** c)365 static void free_completions(char **c) 366 { 367 int i; 368 if (c == NULL) 369 return; 370 for (i = 0; c[i]; i++) 371 os_free(c[i]); 372 os_free(c); 373 } 374 375 filter_strings(char ** c,char * str,size_t len)376 static int filter_strings(char **c, char *str, size_t len) 377 { 378 int i, j; 379 380 for (i = 0, j = 0; c[j]; j++) { 381 if (os_strncasecmp(c[j], str, len) == 0) { 382 if (i != j) { 383 c[i] = c[j]; 384 c[j] = NULL; 385 } 386 i++; 387 } else { 388 os_free(c[j]); 389 c[j] = NULL; 390 } 391 } 392 c[i] = NULL; 393 return i; 394 } 395 396 common_len(const char * a,const char * b)397 static int common_len(const char *a, const char *b) 398 { 399 int len = 0; 400 while (a[len] && a[len] == b[len]) 401 len++; 402 return len; 403 } 404 405 max_common_length(char ** c)406 static int max_common_length(char **c) 407 { 408 int len, i; 409 410 len = os_strlen(c[0]); 411 for (i = 1; c[i]; i++) { 412 int same = common_len(c[0], c[i]); 413 if (same < len) 414 len = same; 415 } 416 417 return len; 418 } 419 420 cmp_str(const void * a,const void * b)421 static int cmp_str(const void *a, const void *b) 422 { 423 return os_strcmp(* (const char **) a, * (const char **) b); 424 } 425 complete(int list)426 static void complete(int list) 427 { 428 char **c; 429 int i, len, count; 430 int start, end; 431 int room, plen, add_space; 432 433 if (edit_completion_cb == NULL) 434 return; 435 436 cmdbuf[cmdbuf_len] = '\0'; 437 c = edit_completion_cb(edit_cb_ctx, cmdbuf, cmdbuf_pos); 438 if (c == NULL) 439 return; 440 441 end = cmdbuf_pos; 442 start = end; 443 while (start > 0 && cmdbuf[start - 1] != ' ') 444 start--; 445 plen = end - start; 446 447 count = filter_strings(c, &cmdbuf[start], plen); 448 if (count == 0) { 449 free_completions(c); 450 return; 451 } 452 453 len = max_common_length(c); 454 if (len <= plen && count > 1) { 455 if (list) { 456 qsort(c, count, sizeof(char *), cmp_str); 457 edit_clear_line(); 458 printf("\r"); 459 for (i = 0; c[i]; i++) 460 printf("%s%s", i > 0 ? " " : "", c[i]); 461 printf("\n"); 462 edit_redraw(); 463 } 464 free_completions(c); 465 return; 466 } 467 len -= plen; 468 469 room = sizeof(cmdbuf) - 1 - cmdbuf_len; 470 if (room < len) 471 len = room; 472 add_space = count == 1 && len < room; 473 474 os_memmove(cmdbuf + cmdbuf_pos + len + add_space, cmdbuf + cmdbuf_pos, 475 cmdbuf_len - cmdbuf_pos); 476 os_memcpy(&cmdbuf[cmdbuf_pos - plen], c[0], plen + len); 477 if (add_space) 478 cmdbuf[cmdbuf_pos + len] = ' '; 479 480 cmdbuf_pos += len + add_space; 481 cmdbuf_len += len + add_space; 482 483 edit_redraw(); 484 485 free_completions(c); 486 } 487 488 489 enum edit_key_code { 490 EDIT_KEY_NONE = 256, 491 EDIT_KEY_TAB, 492 EDIT_KEY_UP, 493 EDIT_KEY_DOWN, 494 EDIT_KEY_RIGHT, 495 EDIT_KEY_LEFT, 496 EDIT_KEY_ENTER, 497 EDIT_KEY_BACKSPACE, 498 EDIT_KEY_INSERT, 499 EDIT_KEY_DELETE, 500 EDIT_KEY_HOME, 501 EDIT_KEY_END, 502 EDIT_KEY_PAGE_UP, 503 EDIT_KEY_PAGE_DOWN, 504 EDIT_KEY_F1, 505 EDIT_KEY_F2, 506 EDIT_KEY_F3, 507 EDIT_KEY_F4, 508 EDIT_KEY_F5, 509 EDIT_KEY_F6, 510 EDIT_KEY_F7, 511 EDIT_KEY_F8, 512 EDIT_KEY_F9, 513 EDIT_KEY_F10, 514 EDIT_KEY_F11, 515 EDIT_KEY_F12, 516 EDIT_KEY_CTRL_UP, 517 EDIT_KEY_CTRL_DOWN, 518 EDIT_KEY_CTRL_RIGHT, 519 EDIT_KEY_CTRL_LEFT, 520 EDIT_KEY_CTRL_A, 521 EDIT_KEY_CTRL_B, 522 EDIT_KEY_CTRL_D, 523 EDIT_KEY_CTRL_E, 524 EDIT_KEY_CTRL_F, 525 EDIT_KEY_CTRL_G, 526 EDIT_KEY_CTRL_H, 527 EDIT_KEY_CTRL_J, 528 EDIT_KEY_CTRL_K, 529 EDIT_KEY_CTRL_L, 530 EDIT_KEY_CTRL_N, 531 EDIT_KEY_CTRL_O, 532 EDIT_KEY_CTRL_P, 533 EDIT_KEY_CTRL_R, 534 EDIT_KEY_CTRL_T, 535 EDIT_KEY_CTRL_U, 536 EDIT_KEY_CTRL_V, 537 EDIT_KEY_CTRL_W, 538 EDIT_KEY_ALT_UP, 539 EDIT_KEY_ALT_DOWN, 540 EDIT_KEY_ALT_RIGHT, 541 EDIT_KEY_ALT_LEFT, 542 EDIT_KEY_SHIFT_UP, 543 EDIT_KEY_SHIFT_DOWN, 544 EDIT_KEY_SHIFT_RIGHT, 545 EDIT_KEY_SHIFT_LEFT, 546 EDIT_KEY_ALT_SHIFT_UP, 547 EDIT_KEY_ALT_SHIFT_DOWN, 548 EDIT_KEY_ALT_SHIFT_RIGHT, 549 EDIT_KEY_ALT_SHIFT_LEFT, 550 EDIT_KEY_EOF 551 }; 552 show_esc_buf(const char * esc_buf,char c,int i)553 static void show_esc_buf(const char *esc_buf, char c, int i) 554 { 555 edit_clear_line(); 556 printf("\rESC buffer '%s' c='%c' [%d]\n", esc_buf, c, i); 557 edit_redraw(); 558 } 559 560 esc_seq_to_key1_no(char last)561 static enum edit_key_code esc_seq_to_key1_no(char last) 562 { 563 switch (last) { 564 case 'A': 565 return EDIT_KEY_UP; 566 case 'B': 567 return EDIT_KEY_DOWN; 568 case 'C': 569 return EDIT_KEY_RIGHT; 570 case 'D': 571 return EDIT_KEY_LEFT; 572 default: 573 return EDIT_KEY_NONE; 574 } 575 } 576 577 esc_seq_to_key1_shift(char last)578 static enum edit_key_code esc_seq_to_key1_shift(char last) 579 { 580 switch (last) { 581 case 'A': 582 return EDIT_KEY_SHIFT_UP; 583 case 'B': 584 return EDIT_KEY_SHIFT_DOWN; 585 case 'C': 586 return EDIT_KEY_SHIFT_RIGHT; 587 case 'D': 588 return EDIT_KEY_SHIFT_LEFT; 589 default: 590 return EDIT_KEY_NONE; 591 } 592 } 593 594 esc_seq_to_key1_alt(char last)595 static enum edit_key_code esc_seq_to_key1_alt(char last) 596 { 597 switch (last) { 598 case 'A': 599 return EDIT_KEY_ALT_UP; 600 case 'B': 601 return EDIT_KEY_ALT_DOWN; 602 case 'C': 603 return EDIT_KEY_ALT_RIGHT; 604 case 'D': 605 return EDIT_KEY_ALT_LEFT; 606 default: 607 return EDIT_KEY_NONE; 608 } 609 } 610 611 esc_seq_to_key1_alt_shift(char last)612 static enum edit_key_code esc_seq_to_key1_alt_shift(char last) 613 { 614 switch (last) { 615 case 'A': 616 return EDIT_KEY_ALT_SHIFT_UP; 617 case 'B': 618 return EDIT_KEY_ALT_SHIFT_DOWN; 619 case 'C': 620 return EDIT_KEY_ALT_SHIFT_RIGHT; 621 case 'D': 622 return EDIT_KEY_ALT_SHIFT_LEFT; 623 default: 624 return EDIT_KEY_NONE; 625 } 626 } 627 628 esc_seq_to_key1_ctrl(char last)629 static enum edit_key_code esc_seq_to_key1_ctrl(char last) 630 { 631 switch (last) { 632 case 'A': 633 return EDIT_KEY_CTRL_UP; 634 case 'B': 635 return EDIT_KEY_CTRL_DOWN; 636 case 'C': 637 return EDIT_KEY_CTRL_RIGHT; 638 case 'D': 639 return EDIT_KEY_CTRL_LEFT; 640 default: 641 return EDIT_KEY_NONE; 642 } 643 } 644 645 esc_seq_to_key1(int param1,int param2,char last)646 static enum edit_key_code esc_seq_to_key1(int param1, int param2, char last) 647 { 648 /* ESC-[<param1>;<param2><last> */ 649 650 if (param1 < 0 && param2 < 0) 651 return esc_seq_to_key1_no(last); 652 653 if (param1 == 1 && param2 == 2) 654 return esc_seq_to_key1_shift(last); 655 656 if (param1 == 1 && param2 == 3) 657 return esc_seq_to_key1_alt(last); 658 659 if (param1 == 1 && param2 == 4) 660 return esc_seq_to_key1_alt_shift(last); 661 662 if (param1 == 1 && param2 == 5) 663 return esc_seq_to_key1_ctrl(last); 664 665 if (param2 < 0) { 666 if (last != '~') 667 return EDIT_KEY_NONE; 668 switch (param1) { 669 case 2: 670 return EDIT_KEY_INSERT; 671 case 3: 672 return EDIT_KEY_DELETE; 673 case 5: 674 return EDIT_KEY_PAGE_UP; 675 case 6: 676 return EDIT_KEY_PAGE_DOWN; 677 case 15: 678 return EDIT_KEY_F5; 679 case 17: 680 return EDIT_KEY_F6; 681 case 18: 682 return EDIT_KEY_F7; 683 case 19: 684 return EDIT_KEY_F8; 685 case 20: 686 return EDIT_KEY_F9; 687 case 21: 688 return EDIT_KEY_F10; 689 case 23: 690 return EDIT_KEY_F11; 691 case 24: 692 return EDIT_KEY_F12; 693 } 694 } 695 696 return EDIT_KEY_NONE; 697 } 698 699 esc_seq_to_key2(int param1,int param2,char last)700 static enum edit_key_code esc_seq_to_key2(int param1, int param2, char last) 701 { 702 /* ESC-O<param1>;<param2><last> */ 703 704 if (param1 >= 0 || param2 >= 0) 705 return EDIT_KEY_NONE; 706 707 switch (last) { 708 case 'F': 709 return EDIT_KEY_END; 710 case 'H': 711 return EDIT_KEY_HOME; 712 case 'P': 713 return EDIT_KEY_F1; 714 case 'Q': 715 return EDIT_KEY_F2; 716 case 'R': 717 return EDIT_KEY_F3; 718 case 'S': 719 return EDIT_KEY_F4; 720 default: 721 return EDIT_KEY_NONE; 722 } 723 } 724 725 esc_seq_to_key(char * seq)726 static enum edit_key_code esc_seq_to_key(char *seq) 727 { 728 char last, *pos; 729 int param1 = -1, param2 = -1; 730 enum edit_key_code ret = EDIT_KEY_NONE; 731 732 last = '\0'; 733 for (pos = seq; *pos; pos++) 734 last = *pos; 735 736 if (seq[1] >= '0' && seq[1] <= '9') { 737 param1 = atoi(&seq[1]); 738 pos = os_strchr(seq, ';'); 739 if (pos) 740 param2 = atoi(pos + 1); 741 } 742 743 if (seq[0] == '[') 744 ret = esc_seq_to_key1(param1, param2, last); 745 else if (seq[0] == 'O') 746 ret = esc_seq_to_key2(param1, param2, last); 747 748 if (ret != EDIT_KEY_NONE) 749 return ret; 750 751 edit_clear_line(); 752 printf("\rUnknown escape sequence '%s'\n", seq); 753 edit_redraw(); 754 return EDIT_KEY_NONE; 755 } 756 757 edit_read_key(int sock)758 static enum edit_key_code edit_read_key(int sock) 759 { 760 int c; 761 unsigned char buf[1]; 762 int res; 763 static int esc = -1; 764 static char esc_buf[7]; 765 766 res = read(sock, buf, 1); 767 if (res < 0) 768 perror("read"); 769 if (res <= 0) 770 return EDIT_KEY_EOF; 771 772 c = buf[0]; 773 774 if (esc >= 0) { 775 if (c == 27 /* ESC */) { 776 esc = 0; 777 return EDIT_KEY_NONE; 778 } 779 780 if (esc == 6) { 781 show_esc_buf(esc_buf, c, 0); 782 esc = -1; 783 } else { 784 esc_buf[esc++] = c; 785 esc_buf[esc] = '\0'; 786 } 787 } 788 789 if (esc == 1) { 790 if (esc_buf[0] != '[' && esc_buf[0] != 'O') { 791 show_esc_buf(esc_buf, c, 1); 792 esc = -1; 793 return EDIT_KEY_NONE; 794 } else 795 return EDIT_KEY_NONE; /* Escape sequence continues */ 796 } 797 798 if (esc > 1) { 799 if ((c >= '0' && c <= '9') || c == ';') 800 return EDIT_KEY_NONE; /* Escape sequence continues */ 801 802 if (c == '~' || (c >= 'A' && c <= 'Z')) { 803 esc = -1; 804 return esc_seq_to_key(esc_buf); 805 } 806 807 show_esc_buf(esc_buf, c, 2); 808 esc = -1; 809 return EDIT_KEY_NONE; 810 } 811 812 switch (c) { 813 case 1: 814 return EDIT_KEY_CTRL_A; 815 case 2: 816 return EDIT_KEY_CTRL_B; 817 case 4: 818 return EDIT_KEY_CTRL_D; 819 case 5: 820 return EDIT_KEY_CTRL_E; 821 case 6: 822 return EDIT_KEY_CTRL_F; 823 case 7: 824 return EDIT_KEY_CTRL_G; 825 case 8: 826 return EDIT_KEY_CTRL_H; 827 case 9: 828 return EDIT_KEY_TAB; 829 case 10: 830 return EDIT_KEY_CTRL_J; 831 case 13: /* CR */ 832 return EDIT_KEY_ENTER; 833 case 11: 834 return EDIT_KEY_CTRL_K; 835 case 12: 836 return EDIT_KEY_CTRL_L; 837 case 14: 838 return EDIT_KEY_CTRL_N; 839 case 15: 840 return EDIT_KEY_CTRL_O; 841 case 16: 842 return EDIT_KEY_CTRL_P; 843 case 18: 844 return EDIT_KEY_CTRL_R; 845 case 20: 846 return EDIT_KEY_CTRL_T; 847 case 21: 848 return EDIT_KEY_CTRL_U; 849 case 22: 850 return EDIT_KEY_CTRL_V; 851 case 23: 852 return EDIT_KEY_CTRL_W; 853 case 27: /* ESC */ 854 esc = 0; 855 return EDIT_KEY_NONE; 856 case 127: 857 return EDIT_KEY_BACKSPACE; 858 default: 859 return c; 860 } 861 } 862 863 864 static char search_buf[21]; 865 static int search_skip; 866 search_find(void)867 static char * search_find(void) 868 { 869 struct edit_history *h; 870 size_t len = os_strlen(search_buf); 871 int skip = search_skip; 872 873 if (len == 0) 874 return NULL; 875 876 dl_list_for_each(h, &history_list, struct edit_history, list) { 877 if (os_strstr(h->str, search_buf)) { 878 if (skip == 0) 879 return h->str; 880 skip--; 881 } 882 } 883 884 search_skip = 0; 885 return NULL; 886 } 887 888 search_redraw(void)889 static void search_redraw(void) 890 { 891 char *match = search_find(); 892 printf("\rsearch '%s': %s" CLEAR_END_LINE, 893 search_buf, match ? match : ""); 894 printf("\rsearch '%s", search_buf); 895 fflush(stdout); 896 } 897 898 search_start(void)899 static void search_start(void) 900 { 901 edit_clear_line(); 902 search_buf[0] = '\0'; 903 search_skip = 0; 904 search_redraw(); 905 } 906 907 search_clear(void)908 static void search_clear(void) 909 { 910 search_redraw(); 911 printf("\r" CLEAR_END_LINE); 912 } 913 914 search_stop(void)915 static void search_stop(void) 916 { 917 char *match = search_find(); 918 search_buf[0] = '\0'; 919 search_clear(); 920 if (match) { 921 os_strlcpy(cmdbuf, match, CMD_BUF_LEN); 922 cmdbuf_len = os_strlen(cmdbuf); 923 cmdbuf_pos = cmdbuf_len; 924 } 925 edit_redraw(); 926 } 927 928 search_cancel(void)929 static void search_cancel(void) 930 { 931 search_buf[0] = '\0'; 932 search_clear(); 933 edit_redraw(); 934 } 935 936 search_backspace(void)937 static void search_backspace(void) 938 { 939 size_t len; 940 len = os_strlen(search_buf); 941 if (len == 0) 942 return; 943 search_buf[len - 1] = '\0'; 944 search_skip = 0; 945 search_redraw(); 946 } 947 948 search_next(void)949 static void search_next(void) 950 { 951 search_skip++; 952 search_find(); 953 search_redraw(); 954 } 955 956 search_char(char c)957 static void search_char(char c) 958 { 959 size_t len; 960 len = os_strlen(search_buf); 961 if (len == sizeof(search_buf) - 1) 962 return; 963 search_buf[len] = c; 964 search_buf[len + 1] = '\0'; 965 search_skip = 0; 966 search_redraw(); 967 } 968 969 search_key(enum edit_key_code c)970 static enum edit_key_code search_key(enum edit_key_code c) 971 { 972 switch (c) { 973 case EDIT_KEY_ENTER: 974 case EDIT_KEY_CTRL_J: 975 case EDIT_KEY_LEFT: 976 case EDIT_KEY_RIGHT: 977 case EDIT_KEY_HOME: 978 case EDIT_KEY_END: 979 case EDIT_KEY_CTRL_A: 980 case EDIT_KEY_CTRL_E: 981 search_stop(); 982 return c; 983 case EDIT_KEY_DOWN: 984 case EDIT_KEY_UP: 985 search_cancel(); 986 return EDIT_KEY_EOF; 987 case EDIT_KEY_CTRL_H: 988 case EDIT_KEY_BACKSPACE: 989 search_backspace(); 990 break; 991 case EDIT_KEY_CTRL_R: 992 search_next(); 993 break; 994 default: 995 if (c >= 32 && c <= 255) 996 search_char(c); 997 break; 998 } 999 1000 return EDIT_KEY_NONE; 1001 } 1002 1003 edit_read_char(int sock,void * eloop_ctx,void * sock_ctx)1004 static void edit_read_char(int sock, void *eloop_ctx, void *sock_ctx) 1005 { 1006 static int last_tab = 0; 1007 static int search = 0; 1008 enum edit_key_code c; 1009 1010 c = edit_read_key(sock); 1011 1012 if (search) { 1013 c = search_key(c); 1014 if (c == EDIT_KEY_NONE) 1015 return; 1016 search = 0; 1017 if (c == EDIT_KEY_EOF) 1018 return; 1019 } 1020 1021 if (c != EDIT_KEY_TAB && c != EDIT_KEY_NONE) 1022 last_tab = 0; 1023 1024 switch (c) { 1025 case EDIT_KEY_NONE: 1026 break; 1027 case EDIT_KEY_EOF: 1028 edit_eof_cb(edit_cb_ctx); 1029 break; 1030 case EDIT_KEY_TAB: 1031 complete(last_tab); 1032 last_tab = 1; 1033 break; 1034 case EDIT_KEY_UP: 1035 case EDIT_KEY_CTRL_P: 1036 history_prev(); 1037 break; 1038 case EDIT_KEY_DOWN: 1039 case EDIT_KEY_CTRL_N: 1040 history_next(); 1041 break; 1042 case EDIT_KEY_RIGHT: 1043 case EDIT_KEY_CTRL_F: 1044 move_right(); 1045 break; 1046 case EDIT_KEY_LEFT: 1047 case EDIT_KEY_CTRL_B: 1048 move_left(); 1049 break; 1050 case EDIT_KEY_CTRL_RIGHT: 1051 move_word_right(); 1052 break; 1053 case EDIT_KEY_CTRL_LEFT: 1054 move_word_left(); 1055 break; 1056 case EDIT_KEY_DELETE: 1057 delete_current(); 1058 break; 1059 case EDIT_KEY_END: 1060 move_end(); 1061 break; 1062 case EDIT_KEY_HOME: 1063 case EDIT_KEY_CTRL_A: 1064 move_start(); 1065 break; 1066 case EDIT_KEY_F2: 1067 history_debug_dump(); 1068 break; 1069 case EDIT_KEY_CTRL_D: 1070 if (cmdbuf_len > 0) { 1071 delete_current(); 1072 return; 1073 } 1074 printf("\n"); 1075 edit_eof_cb(edit_cb_ctx); 1076 break; 1077 case EDIT_KEY_CTRL_E: 1078 move_end(); 1079 break; 1080 case EDIT_KEY_CTRL_H: 1081 case EDIT_KEY_BACKSPACE: 1082 delete_left(); 1083 break; 1084 case EDIT_KEY_ENTER: 1085 case EDIT_KEY_CTRL_J: 1086 process_cmd(); 1087 break; 1088 case EDIT_KEY_CTRL_K: 1089 clear_right(); 1090 break; 1091 case EDIT_KEY_CTRL_L: 1092 edit_clear_line(); 1093 edit_redraw(); 1094 break; 1095 case EDIT_KEY_CTRL_R: 1096 search = 1; 1097 search_start(); 1098 break; 1099 case EDIT_KEY_CTRL_U: 1100 clear_left(); 1101 break; 1102 case EDIT_KEY_CTRL_W: 1103 delete_word(); 1104 break; 1105 default: 1106 if (c >= 32 && c <= 255) 1107 insert_char(c); 1108 break; 1109 } 1110 } 1111 1112 edit_init(void (* cmd_cb)(void * ctx,char * cmd),void (* eof_cb)(void * ctx),char ** (* completion_cb)(void * ctx,const char * cmd,int pos),void * ctx,const char * history_file,const char * ps)1113 int edit_init(void (*cmd_cb)(void *ctx, char *cmd), 1114 void (*eof_cb)(void *ctx), 1115 char ** (*completion_cb)(void *ctx, const char *cmd, int pos), 1116 void *ctx, const char *history_file, const char *ps) 1117 { 1118 currbuf[0] = '\0'; 1119 dl_list_init(&history_list); 1120 history_curr = NULL; 1121 if (history_file) 1122 history_read(history_file); 1123 1124 edit_cb_ctx = ctx; 1125 edit_cmd_cb = cmd_cb; 1126 edit_eof_cb = eof_cb; 1127 edit_completion_cb = completion_cb; 1128 1129 tcgetattr(STDIN_FILENO, &prevt); 1130 newt = prevt; 1131 newt.c_lflag &= ~(ICANON | ECHO); 1132 tcsetattr(STDIN_FILENO, TCSANOW, &newt); 1133 1134 eloop_register_read_sock(STDIN_FILENO, edit_read_char, NULL, NULL); 1135 1136 ps2 = ps; 1137 printf("%s> ", ps2 ? ps2 : ""); 1138 fflush(stdout); 1139 1140 return 0; 1141 } 1142 1143 edit_deinit(const char * history_file,int (* filter_cb)(void * ctx,const char * cmd))1144 void edit_deinit(const char *history_file, 1145 int (*filter_cb)(void *ctx, const char *cmd)) 1146 { 1147 struct edit_history *h; 1148 if (history_file) 1149 history_write(history_file, filter_cb); 1150 while ((h = dl_list_first(&history_list, struct edit_history, list))) { 1151 dl_list_del(&h->list); 1152 os_free(h); 1153 } 1154 edit_clear_line(); 1155 putchar('\r'); 1156 fflush(stdout); 1157 eloop_unregister_read_sock(STDIN_FILENO); 1158 tcsetattr(STDIN_FILENO, TCSANOW, &prevt); 1159 } 1160 1161 edit_redraw(void)1162 void edit_redraw(void) 1163 { 1164 char tmp; 1165 cmdbuf[cmdbuf_len] = '\0'; 1166 printf("\r%s> %s", ps2 ? ps2 : "", cmdbuf); 1167 if (cmdbuf_pos != cmdbuf_len) { 1168 tmp = cmdbuf[cmdbuf_pos]; 1169 cmdbuf[cmdbuf_pos] = '\0'; 1170 printf("\r%s> %s", ps2 ? ps2 : "", cmdbuf); 1171 cmdbuf[cmdbuf_pos] = tmp; 1172 } 1173 fflush(stdout); 1174 } 1175