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