TBTR: Use virtual train head in CmdReplaceTemplateVehicle

Check that train is virtual
4 years ago · 34a6445e77
parent 08eba56a28
commit 34a6445e77
1 changed files with 5 additions and 1 deletions
--- a/src/vehicle_cmd.cpp
+++ b/src/vehicle_cmd.cpp
@ -1150,8 +1150,12 @@ CommandCost CmdReplaceTemplateVehicle(TileIndex tile, DoCommandFlag flags, uint3
 	CommandCost ret = CheckOwnership(vehicle->owner);
 	if (ret.Failed()) return ret;

-	Train* train = Train::From(vehicle);
+	vehicle = vehicle->First();

+	Train* train = Train::From(vehicle);
+	if (!train->IsVirtual()) {
+		return CMD_ERROR;
+	}
 	if (!TemplateVehicle::CanAllocateItem(CountVehiclesInChain(train))) {
 		return CMD_ERROR;
 	}