From c148b70b55a798044839342fe5bb24cd8c19e7b8 Mon Sep 17 00:00:00 2001
From: Horst Schirmeier <horst.schirmeier@tu-dortmund.de>
Date: Mon, 23 Nov 2020 16:22:14 +0100
Subject: [PATCH] DatabaseCampaign: bail out on query failures

Change-Id: If76d4a2d391879e5e12b65ef593b35436cb0e34d
---
 src/core/cpn/DatabaseCampaign.cc | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/core/cpn/DatabaseCampaign.cc b/src/core/cpn/DatabaseCampaign.cc
index 72d6ab43..b028bd31 100644
--- a/src/core/cpn/DatabaseCampaign.cc
+++ b/src/core/cpn/DatabaseCampaign.cc
@@ -213,11 +213,17 @@ bool DatabaseCampaign::run_variant(Database::Variant variant) {
 
 	/* Get the number of unfinished experiments */
 	MYSQL_RES *count = db->query(("SELECT COUNT(*) " + sql_body).c_str(), true);
+	if (!count) {
+		exit(1);
+	}
 	MYSQL_ROW row = mysql_fetch_row(count);
 	experiment_count = strtoul(row[0], NULL, 10);
 
 
 	MYSQL_RES *pilots = db->query_stream ((sql_select + sql_body).c_str());
+	if (!pilots) {
+		exit(1);
+	}
 
 	log_send << "Found " << experiment_count << " jobs in database. ("
 			 << variant.variant << "/" << variant.benchmark << ")" << std::endl;
@@ -312,6 +318,9 @@ void DatabaseCampaign::load_completed_pilots(std::vector<Database::Variant> &var
 	    << "   AND fspmethod_id IN (SELECT id FROM fspmethod WHERE method LIKE '" << m_fspmethod << "')"
 	    << " GROUP BY pilot_id ";
 	MYSQL_RES *ids = db->query_stream(sql.str().c_str());
+	if (!ids) {
+		exit(1);
+	}
 	MYSQL_ROW row;
 	unsigned rowcount = 0;
 	while ((row = mysql_fetch_row(ids)) != 0) {