Don't allow a player to acquire more construction vehicles than the m…

…ax amount allowed. Fixes ticket:2175 git-svn-id: https://warzone2100.svn.sourceforge.net/svnroot/warzone2100/branches/2.3@11674 4a71c877-e1ca-e34f-864e-861f7616d084 (cherry picked from commit e6cd099)
Warzone2100 · Oct 19, 2010 · d6b47d1 · d6b47d1
1 parent 58c0cee
commit d6b47d1
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/droid.c b/src/droid.c
@@ -4298,6 +4298,14 @@ DROID * giftSingleDroid(DROID *psD, UDWORD to)
 
 		if (droidRemove(psD, apsDroidLists)) 		// remove droid from one list
 		{
+			if (psD->droidType == DROID_CYBORG_CONSTRUCT || psD->droidType == DROID_CONSTRUCT)
+			{
+				if (getNumConstructorDroids(selectedPlayer) > MAX_CONSTRUCTOR_DROIDS)
+				{
+					CONPRINTF(ConsoleString, (ConsoleString, _("%s wanted to give you a %s but you have too many!"), getPlayerName(psD->player), psD->aName));
+					return NULL;
+				}
+			}
 			if (!isHumanPlayer(psD->player))
 			{
 				droidSetName(psD, "Enemy Unit");
@@ -4332,6 +4340,11 @@ DROID * giftSingleDroid(DROID *psD, UDWORD to)
 				}
 			}
 		}
+		else
+		{
+			// if we couldn't remove it, then get rid of it.
+			return NULL;
+		}
 		// add back into cluster system
 		clustNewDroid(psD);