Browse Source

wpa_cli: Use edit API as a wrapper for optional readline

Jouni Malinen 14 years ago
parent
commit
bdc45634f0
3 changed files with 53 additions and 45 deletions
  1. 3 0
      src/utils/edit.c
  2. 4 0
      src/utils/edit_simple.c
  3. 46 45
      wpa_supplicant/wpa_cli.c

+ 3 - 0
src/utils/edit.c

@@ -509,6 +509,9 @@ int edit_init(void (*cmd_cb)(void *ctx, char *cmd),
 
 	eloop_register_read_sock(STDIN_FILENO, edit_read_char, NULL, NULL);
 
+	printf("> ");
+	fflush(stdout);
+
 	return 0;
 }
 

+ 4 - 0
src/utils/edit_simple.c

@@ -68,6 +68,10 @@ int edit_init(void (*cmd_cb)(void *ctx, char *cmd),
 	edit_cmd_cb = cmd_cb;
 	edit_eof_cb = eof_cb;
 	eloop_register_read_sock(STDIN_FILENO, edit_read_char, NULL, NULL);
+
+	printf("> ");
+	fflush(stdout);
+
 	return 0;
 }
 

+ 46 - 45
wpa_supplicant/wpa_cli.c

@@ -124,17 +124,6 @@ static void usage(void)
 }
 
 
-static void readline_redraw()
-{
-#ifdef CONFIG_READLINE
-	rl_on_new_line();
-	rl_redisplay();
-#else /* CONFIG_READLINE */
-	edit_redraw();
-#endif /* CONFIG_READLINE */
-}
-
-
 static int str_starts(const char *src, const char *match)
 {
 	return os_strncmp(src, match, os_strlen(match)) == 0;
@@ -2636,11 +2625,9 @@ static void wpa_cli_recv_pending(struct wpa_ctrl *ctrl, int action_monitor)
 				wpa_cli_action_process(buf);
 			else {
 				if (wpa_cli_show_event(buf)) {
-#ifndef CONFIG_READLINE
 					edit_clear_line();
-#endif /* CONFIG_READLINE */
 					printf("\r%s\n", buf);
-					readline_redraw();
+					edit_redraw();
 				}
 			}
 		} else {
@@ -2715,6 +2702,11 @@ static void wpa_cli_mon_receive(int sock, void *eloop_ctx, void *sock_ctx)
 
 #ifdef CONFIG_READLINE
 
+static void *edit_cb_ctx;
+static void (*edit_cmd_cb)(void *ctx, char *cmd);
+static void (*edit_eof_cb)(void *ctx);
+
+
 static char * wpa_cli_cmd_gen(const char *text, int state)
 {
 	static int i, len;
@@ -2745,7 +2737,7 @@ static char * wpa_cli_dummy_gen(const char *text, int state)
 		if (os_strncasecmp(rl_line_buffer, cmd, len) == 0 &&
 		    rl_line_buffer[len] == ' ') {
 			printf("\n%s\n", wpa_cli_commands[i].usage);
-			readline_redraw();
+			edit_redraw();
 			break;
 		}
 	}
@@ -2814,9 +2806,6 @@ static void trunc_nl(char *str)
 
 static void readline_cmd_handler(char *cmd)
 {
-	int argc;
-	char *argv[max_args];
-
 	if (cmd && *cmd) {
 		HIST_ENTRY *h;
 		while (next_history())
@@ -2827,55 +2816,59 @@ static void readline_cmd_handler(char *cmd)
 		next_history();
 	}
 	if (cmd == NULL) {
-		eloop_terminate();
+		edit_eof_cb(edit_cb_ctx);
 		return;
 	}
 	trunc_nl(cmd);
-	argc = tokenize_cmd(cmd, argv);
-	if (argc)
-		wpa_request(ctrl_conn, argc, argv);
+	edit_cmd_cb(edit_cb_ctx, cmd);
 }
 
 
-static void wpa_cli_interactive(void)
+static char *readline_hfile = NULL;
+
+int edit_init(void (*cmd_cb)(void *ctx, char *cmd),
+	      void (*eof_cb)(void *ctx),
+	      void *ctx)
 {
-	char *home, *hfile = NULL;
+	char *home;
 
-	printf("\nInteractive mode\n\n");
+	edit_cb_ctx = ctx;
+	edit_cmd_cb = cmd_cb;
+	edit_eof_cb = eof_cb;
 
 	rl_attempted_completion_function = wpa_cli_completion;
 	home = getenv("HOME");
 	if (home) {
 		const char *fname = ".wpa_cli_history";
 		int hfile_len = os_strlen(home) + 1 + os_strlen(fname) + 1;
-		hfile = os_malloc(hfile_len);
-		if (hfile) {
+		readline_hfile = os_malloc(hfile_len);
+		if (readline_hfile) {
 			int res;
-			res = os_snprintf(hfile, hfile_len, "%s/%s", home,
-					  fname);
+			res = os_snprintf(readline_hfile, hfile_len, "%s/%s",
+					  home, fname);
 			if (res >= 0 && res < hfile_len) {
-				hfile[hfile_len - 1] = '\0';
-				read_history(hfile);
+				readline_hfile[hfile_len - 1] = '\0';
+				read_history(readline_hfile);
 				stifle_history(100);
 			}
 		}
 	}
 
-	eloop_register_signal_terminate(wpa_cli_eloop_terminate, NULL);
 	eloop_register_read_sock(STDIN_FILENO, wpa_cli_read_char, NULL, NULL);
-	eloop_register_timeout(ping_interval, 0, wpa_cli_ping, NULL, NULL);
 
 	rl_callback_handler_install("> ", readline_cmd_handler);
 
-	eloop_run();
+	return 0;
+}
+
 
+void edit_deinit(void)
+{
 	rl_callback_handler_remove();
 
 	eloop_unregister_read_sock(STDIN_FILENO);
-	eloop_cancel_timeout(wpa_cli_ping, NULL, NULL);
-	wpa_cli_close_connection();
 
-	if (hfile) {
+	if (readline_hfile) {
 		/* Save command history, excluding lines that may contain
 		 * passwords. */
 		HIST_ENTRY *h;
@@ -2895,12 +2888,25 @@ static void wpa_cli_interactive(void)
 			} else
 				next_history();
 		}
-		write_history(hfile);
-		os_free(hfile);
+		write_history(readline_hfile);
+		os_free(readline_hfile);
+		readline_hfile = NULL;
 	}
 }
 
-#else /* CONFIG_READLINE */
+
+void edit_clear_line(void)
+{
+}
+
+
+void edit_redraw(void)
+{
+	rl_on_new_line();
+	rl_redisplay();
+}
+
+#endif /* CONFIG_READLINE */
 
 
 static void wpa_cli_edit_cmd_cb(void *ctx, char *cmd)
@@ -2928,9 +2934,6 @@ static void wpa_cli_interactive(void)
 	edit_init(wpa_cli_edit_cmd_cb, wpa_cli_edit_eof_cb, NULL);
 	eloop_register_timeout(ping_interval, 0, wpa_cli_ping, NULL, NULL);
 
-	printf("> ");
-	fflush(stdout);
-
 	eloop_run();
 
 	edit_deinit();
@@ -2938,8 +2941,6 @@ static void wpa_cli_interactive(void)
 	wpa_cli_close_connection();
 }
 
-#endif /* CONFIG_READLINE */
-
 
 static void wpa_cli_action(struct wpa_ctrl *ctrl)
 {