Don't add cylinders and weightsystems past the MAX

We actually should disable the 'Add' button, I guess.

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

1 files changed, 2 insertions, 0 deletions

diff --git a/qt-ui/models.cpp b/qt-ui/models.cpp
index 6756002e2..f1bb8d137 100644
--- a/qt-ui/models.cpp
+++ b/qt-ui/models.cpp
@@ -98,6 +98,7 @@ void CylindersModel::add(cylinder_t* cyl)
 {
 	if (usedRows[current_dive] >= MAX_CYLINDERS) {
 		free(cyl);
+		return;
 	}
 
 	int row = usedRows[current_dive];
@@ -203,6 +204,7 @@ void WeightModel::add(weightsystem_t* weight)
 {
 	if (usedRows[current_dive] >= MAX_WEIGHTSYSTEMS) {
 		free(weight);
+		return;
 	}
 
 	int row = usedRows[current_dive];