Fix potential crash when recording end game unit stats

Warzone2100 · Feb 19, 2024 · ea751ab · ea751ab
1 parent cd03b07
commit ea751ab
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/droid.cpp b/src/droid.cpp
@@ -2279,12 +2279,20 @@ UDWORD	getNumDroidsForLevel(uint32_t player, UDWORD level)
 		}
 		for (const DROID* psDroid : *dList)
 		{
+			if (psDroid == nullptr || isDead(psDroid))
+			{
+				continue;
+			}
 			if (getDroidLevel(psDroid) == level)
 			{
 				++count;
 			}
 			if (psDroid->isTransporter())
 			{
+				if (psDroid->psGroup == nullptr)
+				{
+					continue;
+				}
 				for (const DROID *psCurr : psDroid->psGroup->psList)
 				{
 					if (psCurr != psDroid && getDroidLevel(psCurr) == level)

diff --git a/src/scores.cpp b/src/scores.cpp
@@ -310,8 +310,17 @@ END_GAME_STATS_DATA	collectEndGameStatsData()
 			}
 			for (const DROID* psDroid : *dList)
 			{
+				if (psDroid == nullptr || isDead(psDroid))
+				{
+					continue;
+				}
+				++fullStats.numUnits;
 				if (psDroid->isTransporter())
 				{
+					if (psDroid->psGroup == nullptr)
+					{
+						continue;
+					}
 					for (DROID *psCurr : psDroid->psGroup->psList)
 					{
 						if (psCurr != psDroid)
@@ -320,7 +329,6 @@ END_GAME_STATS_DATA	collectEndGameStatsData()
 						}
 					}
 				}
-				++fullStats.numUnits;
 			}
 		} while (++idx < 3);
 	}