1*f05cddf9SRui Paulo /* 2*f05cddf9SRui Paulo * Minimal command line editing 3*f05cddf9SRui Paulo * Copyright (c) 2010, Jouni Malinen <j@w1.fi> 4*f05cddf9SRui Paulo * 5*f05cddf9SRui Paulo * This software may be distributed under the terms of the BSD license. 6*f05cddf9SRui Paulo * See README for more details. 7*f05cddf9SRui Paulo */ 8*f05cddf9SRui Paulo 9*f05cddf9SRui Paulo #include "includes.h" 10*f05cddf9SRui Paulo 11*f05cddf9SRui Paulo #include "common.h" 12*f05cddf9SRui Paulo #include "eloop.h" 13*f05cddf9SRui Paulo #include "edit.h" 14*f05cddf9SRui Paulo 15*f05cddf9SRui Paulo 16*f05cddf9SRui Paulo #define CMD_BUF_LEN 256 17*f05cddf9SRui Paulo static char cmdbuf[CMD_BUF_LEN]; 18*f05cddf9SRui Paulo static int cmdbuf_pos = 0; 19*f05cddf9SRui Paulo static const char *ps2 = NULL; 20*f05cddf9SRui Paulo 21*f05cddf9SRui Paulo static void *edit_cb_ctx; 22*f05cddf9SRui Paulo static void (*edit_cmd_cb)(void *ctx, char *cmd); 23*f05cddf9SRui Paulo static void (*edit_eof_cb)(void *ctx); 24*f05cddf9SRui Paulo 25*f05cddf9SRui Paulo 26*f05cddf9SRui Paulo static void edit_read_char(int sock, void *eloop_ctx, void *sock_ctx) 27*f05cddf9SRui Paulo { 28*f05cddf9SRui Paulo int c; 29*f05cddf9SRui Paulo unsigned char buf[1]; 30*f05cddf9SRui Paulo int res; 31*f05cddf9SRui Paulo 32*f05cddf9SRui Paulo res = read(sock, buf, 1); 33*f05cddf9SRui Paulo if (res < 0) 34*f05cddf9SRui Paulo perror("read"); 35*f05cddf9SRui Paulo if (res <= 0) { 36*f05cddf9SRui Paulo edit_eof_cb(edit_cb_ctx); 37*f05cddf9SRui Paulo return; 38*f05cddf9SRui Paulo } 39*f05cddf9SRui Paulo c = buf[0]; 40*f05cddf9SRui Paulo 41*f05cddf9SRui Paulo if (c == '\r' || c == '\n') { 42*f05cddf9SRui Paulo cmdbuf[cmdbuf_pos] = '\0'; 43*f05cddf9SRui Paulo cmdbuf_pos = 0; 44*f05cddf9SRui Paulo edit_cmd_cb(edit_cb_ctx, cmdbuf); 45*f05cddf9SRui Paulo printf("%s> ", ps2 ? ps2 : ""); 46*f05cddf9SRui Paulo fflush(stdout); 47*f05cddf9SRui Paulo return; 48*f05cddf9SRui Paulo } 49*f05cddf9SRui Paulo 50*f05cddf9SRui Paulo if (c >= 32 && c <= 255) { 51*f05cddf9SRui Paulo if (cmdbuf_pos < (int) sizeof(cmdbuf) - 1) { 52*f05cddf9SRui Paulo cmdbuf[cmdbuf_pos++] = c; 53*f05cddf9SRui Paulo } 54*f05cddf9SRui Paulo } 55*f05cddf9SRui Paulo } 56*f05cddf9SRui Paulo 57*f05cddf9SRui Paulo 58*f05cddf9SRui Paulo int edit_init(void (*cmd_cb)(void *ctx, char *cmd), 59*f05cddf9SRui Paulo void (*eof_cb)(void *ctx), 60*f05cddf9SRui Paulo char ** (*completion_cb)(void *ctx, const char *cmd, int pos), 61*f05cddf9SRui Paulo void *ctx, const char *history_file, const char *ps) 62*f05cddf9SRui Paulo { 63*f05cddf9SRui Paulo edit_cb_ctx = ctx; 64*f05cddf9SRui Paulo edit_cmd_cb = cmd_cb; 65*f05cddf9SRui Paulo edit_eof_cb = eof_cb; 66*f05cddf9SRui Paulo eloop_register_read_sock(STDIN_FILENO, edit_read_char, NULL, NULL); 67*f05cddf9SRui Paulo ps2 = ps; 68*f05cddf9SRui Paulo 69*f05cddf9SRui Paulo printf("%s> ", ps2 ? ps2 : ""); 70*f05cddf9SRui Paulo fflush(stdout); 71*f05cddf9SRui Paulo 72*f05cddf9SRui Paulo return 0; 73*f05cddf9SRui Paulo } 74*f05cddf9SRui Paulo 75*f05cddf9SRui Paulo 76*f05cddf9SRui Paulo void edit_deinit(const char *history_file, 77*f05cddf9SRui Paulo int (*filter_cb)(void *ctx, const char *cmd)) 78*f05cddf9SRui Paulo { 79*f05cddf9SRui Paulo eloop_unregister_read_sock(STDIN_FILENO); 80*f05cddf9SRui Paulo } 81*f05cddf9SRui Paulo 82*f05cddf9SRui Paulo 83*f05cddf9SRui Paulo void edit_clear_line(void) 84*f05cddf9SRui Paulo { 85*f05cddf9SRui Paulo } 86*f05cddf9SRui Paulo 87*f05cddf9SRui Paulo 88*f05cddf9SRui Paulo void edit_redraw(void) 89*f05cddf9SRui Paulo { 90*f05cddf9SRui Paulo cmdbuf[cmdbuf_pos] = '\0'; 91*f05cddf9SRui Paulo printf("\r> %s", cmdbuf); 92*f05cddf9SRui Paulo } 93