9ae3a8
From 828d09abacc6ae7f0643957301cc3ddee269d7a0 Mon Sep 17 00:00:00 2001
9ae3a8
From: John Snow <jsnow@redhat.com>
9ae3a8
Date: Mon, 23 Nov 2015 17:38:31 +0100
9ae3a8
Subject: [PATCH 12/27] qemu-io: Move command_loop() and friends
9ae3a8
9ae3a8
RH-Author: John Snow <jsnow@redhat.com>
9ae3a8
Message-id: <1448300320-7772-13-git-send-email-jsnow@redhat.com>
9ae3a8
Patchwork-id: 68441
9ae3a8
O-Subject: [RHEL-7.3 qemu-kvm PATCH v2 12/21] qemu-io: Move command_loop() and friends
9ae3a8
Bugzilla: 1272523
9ae3a8
RH-Acked-by: Thomas Huth <thuth@redhat.com>
9ae3a8
RH-Acked-by: Laszlo Ersek <lersek@redhat.com>
9ae3a8
RH-Acked-by: Max Reitz <mreitz@redhat.com>
9ae3a8
9ae3a8
From: Kevin Wolf <kwolf@redhat.com>
9ae3a8
9ae3a8
Signed-off-by: Kevin Wolf <kwolf@redhat.com>
9ae3a8
Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com>
9ae3a8
(cherry picked from commit d1174f13e78e2f43f7ae33d59b62b0b94468c8db)
9ae3a8
Signed-off-by: John Snow <jsnow@redhat.com>
9ae3a8
Signed-off-by: Miroslav Rezanina <mrezanin@redhat.com>
9ae3a8
---
9ae3a8
 cmd.c     | 139 --------------------------------------------------------------
9ae3a8
 cmd.h     |   9 ----
9ae3a8
 qemu-io.c | 139 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
9ae3a8
 3 files changed, 139 insertions(+), 148 deletions(-)
9ae3a8
9ae3a8
diff --git a/cmd.c b/cmd.c
9ae3a8
index 6616d61..26d38a8 100644
9ae3a8
--- a/cmd.c
9ae3a8
+++ b/cmd.c
9ae3a8
@@ -31,145 +31,6 @@
9ae3a8
 
9ae3a8
 /* from libxcmd/command.c */
9ae3a8
 
9ae3a8
-static int		ncmdline;
9ae3a8
-static char		**cmdline;
9ae3a8
-
9ae3a8
-
9ae3a8
-void add_user_command(char *optarg)
9ae3a8
-{
9ae3a8
-    cmdline = g_realloc(cmdline, ++ncmdline * sizeof(char *));
9ae3a8
-    cmdline[ncmdline-1] = optarg;
9ae3a8
-}
9ae3a8
-
9ae3a8
-static void prep_fetchline(void *opaque)
9ae3a8
-{
9ae3a8
-    int *fetchable = opaque;
9ae3a8
-
9ae3a8
-    qemu_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL);
9ae3a8
-    *fetchable= 1;
9ae3a8
-}
9ae3a8
-
9ae3a8
-static char *get_prompt(void);
9ae3a8
-
9ae3a8
-void command_loop(void)
9ae3a8
-{
9ae3a8
-    int i, done = 0, fetchable = 0, prompted = 0;
9ae3a8
-    char *input;
9ae3a8
-
9ae3a8
-    for (i = 0; !done && i < ncmdline; i++) {
9ae3a8
-        done = qemuio_command(cmdline[i]);
9ae3a8
-    }
9ae3a8
-    if (cmdline) {
9ae3a8
-        g_free(cmdline);
9ae3a8
-        return;
9ae3a8
-    }
9ae3a8
-
9ae3a8
-    while (!done) {
9ae3a8
-        if (!prompted) {
9ae3a8
-            printf("%s", get_prompt());
9ae3a8
-            fflush(stdout);
9ae3a8
-            qemu_set_fd_handler(STDIN_FILENO, prep_fetchline, NULL, &fetchable);
9ae3a8
-            prompted = 1;
9ae3a8
-        }
9ae3a8
-
9ae3a8
-        main_loop_wait(false);
9ae3a8
-
9ae3a8
-        if (!fetchable) {
9ae3a8
-            continue;
9ae3a8
-        }
9ae3a8
-
9ae3a8
-        input = fetchline();
9ae3a8
-        if (input == NULL) {
9ae3a8
-            break;
9ae3a8
-        }
9ae3a8
-        done = qemuio_command(input);
9ae3a8
-        free(input);
9ae3a8
-
9ae3a8
-        prompted = 0;
9ae3a8
-        fetchable = 0;
9ae3a8
-    }
9ae3a8
-    qemu_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL);
9ae3a8
-}
9ae3a8
-
9ae3a8
-/* from libxcmd/input.c */
9ae3a8
-
9ae3a8
-#if defined(ENABLE_READLINE)
9ae3a8
-# include <readline history.h="">
9ae3a8
-# include <readline readline.h="">
9ae3a8
-#elif defined(ENABLE_EDITLINE)
9ae3a8
-# include <histedit.h>
9ae3a8
-#endif
9ae3a8
-
9ae3a8
-static char *
9ae3a8
-get_prompt(void)
9ae3a8
-{
9ae3a8
-	static char	prompt[FILENAME_MAX + 2 /*"> "*/ + 1 /*"\0"*/ ];
9ae3a8
-
9ae3a8
-	if (!prompt[0])
9ae3a8
-		snprintf(prompt, sizeof(prompt), "%s> ", progname);
9ae3a8
-	return prompt;
9ae3a8
-}
9ae3a8
-
9ae3a8
-#if defined(ENABLE_READLINE)
9ae3a8
-char *
9ae3a8
-fetchline(void)
9ae3a8
-{
9ae3a8
-	char	*line;
9ae3a8
-
9ae3a8
-	line = readline(get_prompt());
9ae3a8
-	if (line && *line)
9ae3a8
-		add_history(line);
9ae3a8
-	return line;
9ae3a8
-}
9ae3a8
-#elif defined(ENABLE_EDITLINE)
9ae3a8
-static char *el_get_prompt(EditLine *e) { return get_prompt(); }
9ae3a8
-char *
9ae3a8
-fetchline(void)
9ae3a8
-{
9ae3a8
-	static EditLine	*el;
9ae3a8
-	static History	*hist;
9ae3a8
-	HistEvent	hevent;
9ae3a8
-	char		*line;
9ae3a8
-	int		count;
9ae3a8
-
9ae3a8
-	if (!el) {
9ae3a8
-		hist = history_init();
9ae3a8
-		history(hist, &hevent, H_SETSIZE, 100);
9ae3a8
-		el = el_init(progname, stdin, stdout, stderr);
9ae3a8
-		el_source(el, NULL);
9ae3a8
-		el_set(el, EL_SIGNAL, 1);
9ae3a8
-		el_set(el, EL_PROMPT, el_get_prompt);
9ae3a8
-		el_set(el, EL_HIST, history, (const char *)hist);
9ae3a8
-	}
9ae3a8
-	line = strdup(el_gets(el, &count));
9ae3a8
-	if (line) {
9ae3a8
-		if (count > 0)
9ae3a8
-			line[count-1] = '\0';
9ae3a8
-		if (*line)
9ae3a8
-			history(hist, &hevent, H_ENTER, line);
9ae3a8
-	}
9ae3a8
-	return line;
9ae3a8
-}
9ae3a8
-#else
9ae3a8
-# define MAXREADLINESZ	1024
9ae3a8
-char *
9ae3a8
-fetchline(void)
9ae3a8
-{
9ae3a8
-	char	*p, *line = malloc(MAXREADLINESZ);
9ae3a8
-
9ae3a8
-	if (!line)
9ae3a8
-		return NULL;
9ae3a8
-	if (!fgets(line, MAXREADLINESZ, stdin)) {
9ae3a8
-		free(line);
9ae3a8
-		return NULL;
9ae3a8
-	}
9ae3a8
-	p = line + strlen(line);
9ae3a8
-	if (p != line && p[-1] == '\n')
9ae3a8
-		p[-1] = '\0';
9ae3a8
-	return line;
9ae3a8
-}
9ae3a8
-#endif
9ae3a8
-
9ae3a8
 #define EXABYTES(x)	((long long)(x) << 60)
9ae3a8
 #define PETABYTES(x)	((long long)(x) << 50)
9ae3a8
 #define TERABYTES(x)	((long long)(x) << 40)
9ae3a8
diff --git a/cmd.h b/cmd.h
9ae3a8
index 0d01a33..da0c7cf 100644
9ae3a8
--- a/cmd.h
9ae3a8
+++ b/cmd.h
9ae3a8
@@ -39,18 +39,11 @@ typedef struct cmdinfo {
9ae3a8
 	helpfunc_t      help;
9ae3a8
 } cmdinfo_t;
9ae3a8
 
9ae3a8
-typedef int (*checkfunc_t)(BlockDriverState *bs, const cmdinfo_t *ci);
9ae3a8
-
9ae3a8
 void qemuio_add_command(const cmdinfo_t *ci);
9ae3a8
-void add_user_command(char *optarg);
9ae3a8
-void add_check_command(checkfunc_t cf);
9ae3a8
 
9ae3a8
-void command_loop(void);
9ae3a8
 int qemuio_command_usage(const cmdinfo_t *ci);
9ae3a8
 
9ae3a8
 /* from input.h */
9ae3a8
-char *fetchline(void);
9ae3a8
-
9ae3a8
 void cvtstr(double value, char *str, size_t sz);
9ae3a8
 
9ae3a8
 struct timeval tsub(struct timeval t1, struct timeval t2);
9ae3a8
@@ -64,8 +57,6 @@ enum {
9ae3a8
 
9ae3a8
 void timestr(struct timeval *tv, char *str, size_t sz, int flags);
9ae3a8
 
9ae3a8
-extern char *progname;
9ae3a8
-
9ae3a8
 bool qemuio_command(const char *cmd);
9ae3a8
 
9ae3a8
 #endif	/* __COMMAND_H__ */
9ae3a8
diff --git a/qemu-io.c b/qemu-io.c
9ae3a8
index 97af39e..da9944e 100644
9ae3a8
--- a/qemu-io.c
9ae3a8
+++ b/qemu-io.c
9ae3a8
@@ -34,6 +34,10 @@ BlockDriverState *qemuio_bs;
9ae3a8
 
9ae3a8
 extern int qemuio_misalign;
9ae3a8
 
9ae3a8
+/* qemu-io commands passed using -c */
9ae3a8
+static int ncmdline;
9ae3a8
+static char **cmdline;
9ae3a8
+
9ae3a8
 static int close_f(BlockDriverState *bs, int argc, char **argv)
9ae3a8
 {
9ae3a8
     bdrv_unref(bs);
9ae3a8
@@ -207,6 +211,141 @@ static void usage(const char *name)
9ae3a8
 }
9ae3a8
 
9ae3a8
 
9ae3a8
+#if defined(ENABLE_READLINE)
9ae3a8
+# include <readline history.h="">
9ae3a8
+# include <readline readline.h="">
9ae3a8
+#elif defined(ENABLE_EDITLINE)
9ae3a8
+# include <histedit.h>
9ae3a8
+#endif
9ae3a8
+
9ae3a8
+static char *get_prompt(void)
9ae3a8
+{
9ae3a8
+    static char prompt[FILENAME_MAX + 2 /*"> "*/ + 1 /*"\0"*/ ];
9ae3a8
+
9ae3a8
+    if (!prompt[0]) {
9ae3a8
+        snprintf(prompt, sizeof(prompt), "%s> ", progname);
9ae3a8
+    }
9ae3a8
+
9ae3a8
+    return prompt;
9ae3a8
+}
9ae3a8
+
9ae3a8
+#if defined(ENABLE_READLINE)
9ae3a8
+static char *fetchline(void)
9ae3a8
+{
9ae3a8
+    char *line = readline(get_prompt());
9ae3a8
+    if (line && *line) {
9ae3a8
+        add_history(line);
9ae3a8
+    }
9ae3a8
+    return line;
9ae3a8
+}
9ae3a8
+#elif defined(ENABLE_EDITLINE)
9ae3a8
+static char *el_get_prompt(EditLine *e)
9ae3a8
+{
9ae3a8
+    return get_prompt();
9ae3a8
+}
9ae3a8
+
9ae3a8
+static char *fetchline(void)
9ae3a8
+{
9ae3a8
+    static EditLine *el;
9ae3a8
+    static History *hist;
9ae3a8
+    HistEvent hevent;
9ae3a8
+    char *line;
9ae3a8
+    int count;
9ae3a8
+
9ae3a8
+    if (!el) {
9ae3a8
+        hist = history_init();
9ae3a8
+        history(hist, &hevent, H_SETSIZE, 100);
9ae3a8
+        el = el_init(progname, stdin, stdout, stderr);
9ae3a8
+        el_source(el, NULL);
9ae3a8
+        el_set(el, EL_SIGNAL, 1);
9ae3a8
+        el_set(el, EL_PROMPT, el_get_prompt);
9ae3a8
+        el_set(el, EL_HIST, history, (const char *)hist);
9ae3a8
+    }
9ae3a8
+    line = strdup(el_gets(el, &count));
9ae3a8
+    if (line) {
9ae3a8
+        if (count > 0) {
9ae3a8
+            line[count-1] = '\0';
9ae3a8
+        }
9ae3a8
+        if (*line) {
9ae3a8
+            history(hist, &hevent, H_ENTER, line);
9ae3a8
+        }
9ae3a8
+    }
9ae3a8
+    return line;
9ae3a8
+}
9ae3a8
+#else
9ae3a8
+# define MAXREADLINESZ 1024
9ae3a8
+static char *fetchline(void)
9ae3a8
+{
9ae3a8
+    char *p, *line = g_malloc(MAXREADLINESZ);
9ae3a8
+
9ae3a8
+    if (!fgets(line, MAXREADLINESZ, stdin)) {
9ae3a8
+        g_free(line);
9ae3a8
+        return NULL;
9ae3a8
+    }
9ae3a8
+
9ae3a8
+    p = line + strlen(line);
9ae3a8
+    if (p != line && p[-1] == '\n') {
9ae3a8
+        p[-1] = '\0';
9ae3a8
+    }
9ae3a8
+
9ae3a8
+    return line;
9ae3a8
+}
9ae3a8
+#endif
9ae3a8
+
9ae3a8
+static void prep_fetchline(void *opaque)
9ae3a8
+{
9ae3a8
+    int *fetchable = opaque;
9ae3a8
+
9ae3a8
+    qemu_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL);
9ae3a8
+    *fetchable= 1;
9ae3a8
+}
9ae3a8
+
9ae3a8
+static void command_loop(void)
9ae3a8
+{
9ae3a8
+    int i, done = 0, fetchable = 0, prompted = 0;
9ae3a8
+    char *input;
9ae3a8
+
9ae3a8
+    for (i = 0; !done && i < ncmdline; i++) {
9ae3a8
+        done = qemuio_command(cmdline[i]);
9ae3a8
+    }
9ae3a8
+    if (cmdline) {
9ae3a8
+        g_free(cmdline);
9ae3a8
+        return;
9ae3a8
+    }
9ae3a8
+
9ae3a8
+    while (!done) {
9ae3a8
+        if (!prompted) {
9ae3a8
+            printf("%s", get_prompt());
9ae3a8
+            fflush(stdout);
9ae3a8
+            qemu_set_fd_handler(STDIN_FILENO, prep_fetchline, NULL, &fetchable);
9ae3a8
+            prompted = 1;
9ae3a8
+        }
9ae3a8
+
9ae3a8
+        main_loop_wait(false);
9ae3a8
+
9ae3a8
+        if (!fetchable) {
9ae3a8
+            continue;
9ae3a8
+        }
9ae3a8
+
9ae3a8
+        input = fetchline();
9ae3a8
+        if (input == NULL) {
9ae3a8
+            break;
9ae3a8
+        }
9ae3a8
+        done = qemuio_command(input);
9ae3a8
+        g_free(input);
9ae3a8
+
9ae3a8
+        prompted = 0;
9ae3a8
+        fetchable = 0;
9ae3a8
+    }
9ae3a8
+    qemu_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL);
9ae3a8
+}
9ae3a8
+
9ae3a8
+static void add_user_command(char *optarg)
9ae3a8
+{
9ae3a8
+    cmdline = g_realloc(cmdline, ++ncmdline * sizeof(char *));
9ae3a8
+    cmdline[ncmdline-1] = optarg;
9ae3a8
+}
9ae3a8
+
9ae3a8
 int main(int argc, char **argv)
9ae3a8
 {
9ae3a8
     int readonly = 0;
9ae3a8
-- 
9ae3a8
1.8.3.1
9ae3a8