hivex: Add HIVEX_OPEN_WRITE flag to allow hive to be opened for writing.

[libguestfs.git] / hivex / hivexsh.c

diff --git a/hivex/hivexsh.c b/hivex/hivexsh.c
index 1cecaad..01a5ddc 100644 (file)
--- a/hivex/hivexsh.c
+++ b/hivex/hivexsh.c
@@ -57,8 +57,9 @@
 #include "hivex.h"
 
 static int quit = 0;
+static int is_tty;
 static hive_h *h = NULL;
-static char *prompt_string = NULL; /* Prompt string. */
+static char *prompt_string = NULL; /* Normal prompt string. */
 static char *loaded = NULL;     /* Basename of loaded file, if any. */
 static hive_node_h cwd;         /* Current node. */
 static int open_flags = 0;      /* Flags used when loading a hive file. */
@@ -69,7 +70,7 @@ static void set_prompt_string (void);
 static void initialize_readline (void);
 static void cleanup_readline (void);
 static void add_history_line (const char *);
-static char *rl_gets (int prompt);
+static char *rl_gets (const char *prompt_string);
 static void sort_strings (char **strings, int len);
 static int dispatch (char *cmd, char *args);
 static int cmd_cd (char *path);
@@ -128,10 +129,10 @@ main (int argc, char *argv[])
   }
 
   /* Main loop. */
+  is_tty = isatty (0);
   initialize_readline ();
-  int prompt = isatty (0);
 
-  if (prompt)
+  if (is_tty)
     printf (_(
 "\n"
 "Welcome to hivexsh, the hivex interactive shell for examining\n"
@@ -142,10 +143,11 @@ main (int argc, char *argv[])
 "\n"));
 
   while (!quit) {
-    char *buf = rl_gets (prompt);
+    char *buf = rl_gets (prompt_string);
     if (!buf) {
       quit = 1;
-      printf ("\n");
+      if (is_tty)
+        printf ("\n");
       break;
     }
 
@@ -192,7 +194,7 @@ main (int argc, char *argv[])
   got_command:
     /*printf ("command: '%s'  args: '%s'\n", cmd, args)*/;
     int r = dispatch (cmd, args);
-    if (!prompt && r == -1)
+    if (!is_tty && r == -1)
       exit (EXIT_FAILURE);
   }
 
@@ -218,7 +220,7 @@ set_prompt_string (void)
   fp = open_memstream (&ptr, &size);
   if (fp == NULL) {
     perror ("open_memstream");
-    exit (1);
+    exit (EXIT_FAILURE);
   }
 
   if (h) {
@@ -268,11 +270,11 @@ print_node_path (hive_node_h node, FILE *fp)
 static char *line_read = NULL;
 
 static char *
-rl_gets (int prompt)
+rl_gets (const char *prompt_string)
 {
 #ifdef HAVE_LIBREADLINE
 
-  if (prompt) {
+  if (is_tty) {
     if (line_read) {
       free (line_read);
       line_read = NULL;
@@ -291,7 +293,7 @@ rl_gets (int prompt)
   static char buf[8192];
   int len;
 
-  if (prompt)
+  if (is_tty)
     printf ("%s", prompt_string);
   line_read = fgets (buf, sizeof buf, stdin);
 
@@ -568,7 +570,7 @@ cmd_ls (char *args)
   char **names = calloc (len, sizeof (char *));
   if (names == NULL) {
     perror ("malloc");
-    exit (1);
+    exit (EXIT_FAILURE);
   }
 
   int ret = -1;