use jsont to produce JSON output; write json into given file

Author: Daniel Kroening

src/goto-analyzer/goto_analyzer_parse_options.cpp

@@ -487,17 +487,17 @@ int goto_analyzer_parse_optionst::doit()
   {
     const namespacet ns(symbol_table);
     std::string taint_file=cmdline.get_value("taint");
-    bool json=cmdline.isset("json");
 
     if(cmdline.isset("show-taint"))
     {
-      taint_analysis(goto_functions, ns, taint_file, get_message_handler(), true, json);
+      taint_analysis(goto_functions, ns, taint_file, get_message_handler(), true, "");
       return 0;
     }
     else
     {
+      std::string json_file=cmdline.get_value("json");
       bool result=
-        taint_analysis(goto_functions, ns, taint_file, get_message_handler(), false, json);
+        taint_analysis(goto_functions, ns, taint_file, get_message_handler(), false, json_file);
       return result?10:0;
     }
   }

src/goto-analyzer/goto_analyzer_parse_options.h

@@ -30,7 +30,7 @@ class optionst;
   "(verbosity):(version)" \
   "(gcc)(arch):" \
   "(taint):(show-taint)" \
-  "(json)"
+  "(json):"
 
 class goto_analyzer_parse_optionst:
   public parse_options_baset,

src/goto-analyzer/taint_analysis.cpp

@@ -7,9 +7,11 @@ Author: Daniel Kroening, [email protected]
 \*******************************************************************/
 
 #include <iostream>
+#include <fstream>
 
 #include <util/prefix.h>
 #include <util/simplify_expr.h>
+#include <util/json.h>
 
 #include <ansi-c/string_constant.h>
 
@@ -39,7 +41,7 @@ class taint_analysist:public messaget
     const std::string &taint_file_name,
     goto_functionst &goto_functions,
     bool show_full,
-    bool json);
+    const std::string &json_file_name);
 
 protected:
   const namespacet &ns;
@@ -218,10 +220,13 @@ bool taint_analysist::operator()(
   const std::string &taint_file_name,
   goto_functionst &goto_functions,
   bool show_full,
-  bool json)
+  const std::string &json_file_name)
 {
   try
   {
+    jsont json_result=jsont::json_array();
+    bool use_json=!json_file_name.empty();
+  
     status() << "Reading taint file `" << taint_file_name
              << "'" << eom;
 
@@ -323,18 +328,18 @@ bool taint_analysist::operator()(
         if(first)
         {
           first=false;
-          if(!json)
+          if(!use_json)
             std::cout << "\n"
                          "******** Function " << symbol.display_name() << '\n';
         }
 
-        if(json)
+        if(use_json)
         {
-          std::cout << "{\n";
-          std::cout << "  \"bug_class\": \"" << i_it->source_location.get_property_class() << "\",\n";
-          std::cout << "  \"file\": \"" << i_it->source_location.get_file() << "\",\n";
-          std::cout << "  \"line\": " << i_it->source_location.get_line() << "\n";
-          std::cout << "}\n";
+          jsont json=jsont::json_object();
+          json["bug_class"]=jsont::json_string(id2string(i_it->source_location.get_property_class()));
+          json["file"]=jsont::json_string(id2string(i_it->source_location.get_file()));
+          json["line"]=jsont::json_number(id2string(i_it->source_location.get_line()));
+          json_result.array.push_back(json);
         }
         else
         {
@@ -346,7 +351,24 @@ bool taint_analysist::operator()(
         }
       }
     }
+    
+    if(use_json)
+    {
+      std::ofstream json_out(json_file_name);
 
+      if(!json_out)
+      {
+        error() << "Failed to open json output `"
+                << json_file_name << "'" << eom;
+        return true;
+      }
+      
+      status() << "Analysis result is written to `"
+               << json_file_name << "'" << eom;
+      
+      json_out << json_result << '\n';
+    }
+  
     return false;
   }
   catch(const char *error_msg)
@@ -383,10 +405,11 @@ bool taint_analysis(
   const std::string &taint_file_name,
   message_handlert &message_handler,
   bool show_full,
-  bool json)
+  const std::string &json_file_name)
 {
   taint_analysist taint_analysis(ns);
   taint_analysis.set_message_handler(message_handler);
-  return taint_analysis(taint_file_name, goto_functions, show_full, json);
+  return taint_analysis(
+    taint_file_name, goto_functions, show_full, json_file_name);
 }
 

src/goto-analyzer/taint_analysis.h

@@ -20,6 +20,6 @@ bool taint_analysis(
   const std::string &taint_file_name,
   message_handlert &,
   bool show_full,
-  bool json);
+  const std::string &json_file_name);
 
 #endif