Fix potential crash in getColor functions

If the index is out of range, just return black.

Signed-off-by: Dirk Hohndel <dirk@hohndel.org>

2 files changed, 6 insertions, 2 deletions

diff --git a/qt-ui/diveplanner.cpp b/qt-ui/diveplanner.cpp
index 7fa1a8783..4ff6894e6 100644
--- a/qt-ui/diveplanner.cpp
+++ b/qt-ui/diveplanner.cpp
@@ -45,7 +45,9 @@ QString dpGasToStr(const divedatapoint& p)
 
 QColor getColor(const color_indice_t i)
 {
-	return profile_color[i].at(0);
+	if ( profile_color.count() > i && i >= 0)
+		return profile_color[i].at(0);
+	return QColor(Qt::black);
 }
 
 static DivePlannerPointsModel *plannerModel = DivePlannerPointsModel::instance();
diff --git a/qt-ui/graphicsview-common.cpp b/qt-ui/graphicsview-common.cpp
index b4b3b6bce..5415cfe1f 100644
--- a/qt-ui/graphicsview-common.cpp
+++ b/qt-ui/graphicsview-common.cpp
@@ -58,5 +58,7 @@ void fill_profile_color()
 
 QColor getColor(const color_indice_t i, bool isGrayscale = false)
 {
-	return profile_color[i].at((isGrayscale) ? 1 : 0);
+	if (profile_color.count() > i && i >= 0)
+		return profile_color[i].at((isGrayscale) ? 1 : 0);
+	return QColor(Qt::black);
 }