From 2023756023a589d5753cc1b27a3aa1a11d0ebccb Mon Sep 17 00:00:00 2001 From: Fiftytwo Date: Thu, 30 Jul 2020 11:11:51 +0200 Subject: [PATCH] Fix factory() being able to produce non-bot units --- src/level/robotmain.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/level/robotmain.cpp b/src/level/robotmain.cpp index 523f69c6..9c3323c5 100644 --- a/src/level/robotmain.cpp +++ b/src/level/robotmain.cpp @@ -5957,6 +5957,8 @@ Error CRobotMain::CanFactoryError(ObjectType type, int team) if (type == OBJECT_MOBILEst && !IsResearchDone(RESEARCH_SUBM, team)) return ERR_BUILD_DISABLED; if (type == OBJECT_MOBILEtg && !IsResearchDone(RESEARCH_TARGET, team)) return ERR_BUILD_RESEARCH; + if (tool == ToolType::Other && drive == DriveType::Other && type != OBJECT_MOBILEtg) return ERR_WRONG_OBJ; + return ERR_OK; }