csv: Support -o parameter per issue #28

Signed-off-by: Ikey Doherty <[email protected]>

Author: Ikey Doherty

src/plugins/output/csv/csv.c

@@ -12,6 +12,7 @@
 #define _GNU_SOURCE
 
 #include <stdio.h>
+#include <errno.h>
 
 #include "config.h"
 #include "util.h"
@@ -87,6 +88,19 @@ static bool csv_write_report(CveCheckTool *self)
         gchar *key = NULL;
         struct source_package_t *v = NULL;
 
+        FILE *fd = NULL;
+        bool ret = false;
+
+        if (self->output_file) {
+                fd = fopen(self->output_file, "w");
+                if (!fd) {
+                        fprintf(stderr, "Unable to open %s for writing: %s\n", self->output_file, strerror(errno));
+                        return false;
+                }
+        } else {
+                fd = stdout;
+        }
+
         /* package,version,unpatched CVE numbers space delimited,patched CVE numbers space delimited, open bug count */
         g_hash_table_iter_init(&iter, self->db);
         while (g_hash_table_iter_next(&iter, (void**)&key, (void**)&v)) {
@@ -117,9 +131,23 @@ static bool csv_write_report(CveCheckTool *self)
                 } else {
                         open_bug_count = g_strdup("0");
                 }
-                printf("%s,%s,%s,%s,%s\n", key, (char*)v->version, is, pa, open_bug_count);
+                if (fprintf(fd, "%s,%s,%s,%s,%s\n", key, (char*)v->version, is, pa, open_bug_count) < 0) {
+                        goto io_error;
+                }
         }
-        return true;
+
+        ret = true;
+        goto success;
+
+io_error:
+        fprintf(stderr, "Error writing to file: %s\n", strerror(errno));
+success:
+        ret = true;
+        if (fd != stdout && self->output_file) {
+                fclose(fd);
+        }
+
+        return ret;
 }
 
 _module_export_ bool cve_plugin_module_init(CvePlugin *self)