Prevent a crash if droid in a group can't be found

Co-authored-by: past-due <past-due@users.noreply.github.com>
Warzone2100 · May 3, 2024 · 981df67 · 981df67
1 parent d8eebbd
commit 981df67
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/group.cpp b/src/group.cpp
@@ -177,8 +177,14 @@ void DROID_GROUP::remove(DROID *psDroid)
 		if (psDroid->droidType != DROID_COMMAND || type != GT_COMMAND)
 		{
 			auto it = std::find(psList.begin(), psList.end(), psDroid);
-			ASSERT(it != psList.end(), "grpLeave: droid not found");
-			psList.erase(it);
+			if (it != psList.end())
+			{
+				psList.erase(it);
+			}
+			else
+			{
+				ASSERT(false, "grpLeave: droid not found");
+			}
 		}
 
 		psDroid->psGroup = nullptr;