adjusted program usage

Author: wfeldt

json.c

@@ -7,6 +7,7 @@
 #include <json-c/json.h>
 
 #include "json.h"
+#include "util.h"
 
 json_object *json_root;
 
@@ -20,27 +21,24 @@ void json_init()
 void json_done()
 {
   if(json_root) json_object_put(json_root);
+
   json_root = 0;
 }
 
 
-void json_print(char *file_name)
+void json_print()
 {
-  if(!file_name) return;
+  if(!opt.json) return;
 
   json_object *json_obj = json_root;
 
   if(json_object_array_length(json_obj) == 1) json_obj = json_object_array_get_idx(json_obj, 0);
 
   int flags = JSON_C_TO_STRING_PRETTY + JSON_C_TO_STRING_NOSLASHESCAPE + JSON_C_TO_STRING_SPACED;
 
-  if(strcmp(file_name, "-")) {
-    json_object_to_file_ext(file_name, json_obj, flags);
-  }
-  else {
-    json_object_to_fd(STDOUT_FILENO, json_obj, flags);
-    printf("\n");
-  }
+  json_object_to_fd(STDOUT_FILENO, json_obj, flags);
+
+  printf("\n");
 }
 
 

json.h

@@ -2,5 +2,5 @@
 
 void json_init();
 void json_done();
-void json_print(char *file_name);
+void json_print();
 json_object *json_object_new_format(const char *format, ...) __attribute__ ((format (printf, 1, 2)));

parti.c

@@ -27,7 +27,7 @@ struct option options[] = {
   { "version",     0, NULL, 1002 },
   { "export-disk", 1, NULL, 1003 },
   { "import-disk", 1, NULL, 1004 },
-  { "json",        1, NULL, 1005 },
+  { "json",        0, NULL, 1005 },
   { }
 };
 
@@ -66,7 +66,7 @@ int main(int argc, char **argv)
         break;
 
       case 1005:
-        opt.json_file = optarg;
+        opt.json = 1;
         break;
 
       default:
@@ -102,7 +102,7 @@ int main(int argc, char **argv)
     }
   }
 
-  json_print(opt.json_file);
+  json_print();
 
   json_done();
 
@@ -120,7 +120,7 @@ void help()
     "\n"
     "Options:\n"
     "\n"
-    "  --json FILE         Use JSON format for output.\n"
+    "  --json              Use JSON format for output.\n"
     "  --export-disk FILE  Export all relevant disk data to FILE. FILE can then be used\n"
     "                      with --import-disk to reproduce the results.\n"
     "  --import-disk FILE  Import relevant disk data from FILE.\n"

util.c

@@ -90,6 +90,8 @@ uint64_t read_qword_be(void *buf)
 
 void log_info(const char *format, ...)
 {
+  if(opt.json) return;
+
   va_list args;
   va_start(args, format);
   vfprintf(stdout, format, args);

util.h

@@ -18,7 +18,7 @@ typedef struct {
     unsigned raw:1;
   } show;
   char *export_file;
-  char *json_file;
+  unsigned json:1;
 } opt_t;
 
 extern opt_t opt;