save reporters' results to abrt_vNN_reportresult table too

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>

2 files changed, 40 insertions, 5 deletions

diff --git a/lib/Plugins/SQLite3.cpp b/lib/Plugins/SQLite3.cpp
index b2be21a9..5819b98c 100644
--- a/lib/Plugins/SQLite3.cpp
+++ b/lib/Plugins/SQLite3.cpp
@@ -147,7 +147,7 @@ static void get_table(vector_database_rows_t& pTable,
     sqlite3_free_table(table);
 }
 
-static void execute_sql(sqlite3 *db, const char *fmt, ...)
+static int execute_sql(sqlite3 *db, const char *fmt, ...)
 {
     va_list p;
     va_start(p, fmt);
@@ -163,8 +163,11 @@ static void execute_sql(sqlite3 *db, const char *fmt, ...)
         sqlite3_free(err);
         throw CABRTException(EXCEP_PLUGIN, errstr.c_str());
     }
-    VERB2 log("%d rows affected by SQL:%s", sqlite3_changes(db), sql);
+    int affected = sqlite3_changes(db);
+    VERB2 log("%d rows affected by SQL:%s", affected, sql);
     free(sql);
+
+    return affected;
 }
 
 static bool exists_uuid_uid(sqlite3 *db, const char *pUUID, const char *pUID)
@@ -481,9 +484,6 @@ void CSQLite3::DeleteRows_by_dir(const char *dump_dir)
 
 void CSQLite3::SetReported(const char *pUUID, const char *pUID, const char *pMessage)
 {
-//    string escaped_msg = sql_escape(pMessage);
-//#define pMessage pMessage_must_not_be_used_below
-
     if (!is_string_safe(pUUID)
      || !is_string_safe(pUID)
      || !is_string_safe(pMessage)
@@ -528,6 +528,37 @@ void CSQLite3::SetReported(const char *pUUID, const char *pUID, const char *pMes
     }
 }
 
+void CSQLite3::SetReportedPerReporter(const char *pUUID,
+                                 const char *pUID,
+                                 const char *reporter,
+                                 const char *pMessage)
+{
+    if (!is_string_safe(pUUID)
+     || !is_string_safe(pUID)
+     || !is_string_safe(reporter)
+     || !is_string_safe(pMessage)
+    ) {
+        return;
+    }
+
+    int affected_rows = execute_sql(m_pDB,
+                "UPDATE "ABRT_REPRESULT_TABLE
+                " SET "COL_MESSAGE"='%s'"
+                " WHERE "COL_UUID"='%s' AND "COL_UID"='%s' AND "COL_REPORTER"='%s'",
+                pMessage,
+                pUUID, pUID, reporter
+    );
+    if (!affected_rows)
+    {
+        execute_sql(m_pDB,
+                "INSERT INTO "ABRT_REPRESULT_TABLE
+                " ("COL_UUID","COL_UID","COL_REPORTER","COL_MESSAGE")"
+                " VALUES ('%s','%s','%s','%s');",
+                pUUID, pUID, reporter, pMessage
+	);
+    }
+}
+
 vector_database_rows_t CSQLite3::GetUIDData(const char *pUID)
 {
     vector_database_rows_t table;
diff --git a/lib/Plugins/SQLite3.h b/lib/Plugins/SQLite3.h
index d4b321a2..94cb0422 100644
--- a/lib/Plugins/SQLite3.h
+++ b/lib/Plugins/SQLite3.h
@@ -45,6 +45,10 @@ class CSQLite3 : public CDatabase
         virtual void DeleteRow(const char *pUUID, const char *pUID);
         virtual void DeleteRows_by_dir(const char *dump_dir);
         virtual void SetReported(const char *pUUID, const char *pUID, const char *pMessage);
+        virtual void SetReportedPerReporter(const char *pUUID,
+                                 const char *pUID,
+                                 const char *reporter,
+                                 const char *pMessage);
         virtual vector_database_rows_t GetUIDData(const char *pUID);
         virtual database_row_t GetRow(const char *pUUID, const char *pUID);