From d4f5c255832391ba6132959d1ded57ce9286e7d6 Mon Sep 17 00:00:00 2001

From: Dawid Zamirski <dzamirski@datto.com>

Date: Thu, 16 Feb 2017 18:17:25 -0500

Subject: [PATCH 11/16] hivexsh: add -u flag for HIVEX_OPEN_UNSAFE.

and pass it to hivex_open. Additionally make hivex_value_value failures

non-critical in this mode when iterating through node children/values.

(cherry picked from commit 62c53cda651f34f18a2b55e91012c0448acb70f1)

---

 sh/hivexsh.c | 26 ++++++++++++++++++++------

 1 file changed, 20 insertions(+), 6 deletions(-)

diff --git a/sh/hivexsh.c b/sh/hivexsh.c

index f578ccc..39ab9d0 100644

--- a/sh/hivexsh.c

+++ b/sh/hivexsh.c

@@ -67,6 +67,7 @@

 static int quit = 0;

 static int is_tty;

+static int unsafe = 0;

 static hive_h *h = NULL;

 static char *prompt_string = NULL; /* Normal prompt string. */

 static char *loaded = NULL;     /* Basename of loaded file, if any. */

@@ -97,7 +98,7 @@ static int cmd_setval (char *args);

 static void

 usage (void)

 {

-  fprintf (stderr, "hivexsh [-dfw] [hivefile]\n");

+  fprintf (stderr, "hivexsh [-dfwu] [hivefile]\n");

   exit (EXIT_FAILURE);

 }

@@ -115,7 +116,7 @@ main (int argc, char *argv[])

   set_prompt_string ();

-  while ((c = getopt (argc, argv, "df:w")) != EOF) {

+  while ((c = getopt (argc, argv, "df:wu")) != EOF) {

     switch (c) {

     case 'd':

       open_flags |= HIVEX_OPEN_DEBUG;

@@ -126,6 +127,10 @@ main (int argc, char *argv[])

     case 'w':

       open_flags |= HIVEX_OPEN_WRITE;

       break;

+    case 'u':

+      open_flags |= HIVEX_OPEN_UNSAFE;

+      unsafe = 1;

+      break;

     default:

       usage ();

     }

@@ -771,6 +776,7 @@ cmd_lsval (char *key)

       hive_type t;

       size_t len;

+

       if (hivex_value_type (h, values[i], &t, &len) == -1)

         goto error;

@@ -779,8 +785,12 @@ cmd_lsval (char *key)

       case hive_t_expand_string:

       case hive_t_link: {

         char *str = hivex_value_string (h, values[i]);

-        if (!str)

-          goto error;

+        if (!str) {

+          if (unsafe)

+            continue;

+          else

+            goto error;

+        }

         if (t != hive_t_string)

           printf ("str(%d):", t);

@@ -813,8 +823,12 @@ cmd_lsval (char *key)

       default: {

         unsigned char *data =

           (unsigned char *) hivex_value_value (h, values[i], &t, &len;;

-        if (!data)

-          goto error;

+        if (!data) {

+          if (unsafe)

+            continue;

+          else

+            goto error;

+        }

         printf ("hex(%d):", t);

         size_t j;

-- 

1.8.3.1