From 5cfc0295236de7f1d32ca23519f35a3ee883f220 Mon Sep 17 00:00:00 2001
From: alberth <alberth@openttd.org>
Date: Fri, 6 Aug 2010 20:10:53 +0000
Subject: [PATCH] (svn r20391) -Fix [FS#3993]: Prevent buying more vehicles
 than allowed.

---
 src/vehicle.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/vehicle.cpp b/src/vehicle.cpp
index 8673baef13..0fde741282 100644
--- a/src/vehicle.cpp
+++ b/src/vehicle.cpp
@@ -1321,8 +1321,28 @@ UnitID FreeUnitIDGenerator::NextID()
 	return this->curid;
 }
 
+/**
+ * Get an unused unit number for a vehicle (if allowed).
+ * @param type Type of vehicle
+ * @return A unused unit number for the given type of vehicle if it is allowed to build one, else \c UINT16_MAX.
+ */
 UnitID GetFreeUnitNumber(VehicleType type)
 {
+	/* Check whether it is allowed to build another vehicle. */
+	uint max_veh;
+	switch (type) {
+		case VEH_TRAIN:    max_veh = _settings_game.vehicle.max_trains;   break;
+		case VEH_ROAD:     max_veh = _settings_game.vehicle.max_roadveh;  break;
+		case VEH_SHIP:     max_veh = _settings_game.vehicle.max_ships;    break;
+		case VEH_AIRCRAFT: max_veh = _settings_game.vehicle.max_aircraft; break;
+		default: NOT_REACHED();
+	}
+
+	uint amounts[4];
+	CountCompanyVehicles(_current_company, amounts);
+	assert((uint)type < lengthof(amounts));
+	if (amounts[type] >= max_veh) return UINT16_MAX; // Currently already at the limit, no room to make a new one.
+
 	FreeUnitIDGenerator gen(type, _current_company);
 
 	return gen.NextID();