diff --git a/aircraft_cmd.c b/aircraft_cmd.c index 2478a8e108..f096b92d4b 100644 --- a/aircraft_cmd.c +++ b/aircraft_cmd.c @@ -121,29 +121,6 @@ void DrawAircraftEngine(int x, int y, EngineID engine, uint32 image_ormod) } } -/* Allocate many vehicles */ -static bool AllocateVehicles(Vehicle **vl, int num) -{ - int i; - Vehicle *v; - bool success = true; - - for(i=0; i!=num; i++) { - vl[i] = v = AllocateVehicle(); - if (v == NULL) { - success = false; - break; - } - v->type = 1; - } - - while (--i >= 0) { - vl[i]->type = 0; - } - - return success; -} - int32 EstimateAircraftCost(EngineID engine_type) { return AircraftVehInfo(engine_type)->base_cost * (_price.aircraft_base>>3)>>5; diff --git a/vehicle.c b/vehicle.c index b88811bdf7..f633689125 100644 --- a/vehicle.c +++ b/vehicle.c @@ -315,6 +315,34 @@ Vehicle *AllocateVehicle(void) return NULL; } +/** Allocates a lot of vehicles and frees them again +* @param vl pointer to an array of vehicles that can store all the vehicles in the test +* @param num number of vehicles to test for +* returns true if there is room to allocate all the vehicles +*/ +bool AllocateVehicles(Vehicle **vl, int num) +{ + int i; + Vehicle *v; + bool success = true; + + for(i = 0; i != num; i++) { + vl[i] = v = AllocateVehicle(); + if (v == NULL) { + success = false; + break; + } + v->type = 1; + } + + while (--i >= 0) { + vl[i]->type = 0; + } + + return success; +} + + void *VehicleFromPos(TileIndex tile, void *data, VehicleFromPosProc *proc) { int x,y,x2,y2; @@ -1482,7 +1510,6 @@ int32 CmdCloneVehicle(int x, int y, uint32 flags, uint32 p1, uint32 p2) Vehicle *v_front, *v; Vehicle *w_front, *w, *w_rear; int cost, total_cost = 0; -// VehicleID *new_id; if (!IsVehicleIndex(p1)) return CMD_ERROR; @@ -1506,6 +1533,25 @@ int32 CmdCloneVehicle(int x, int y, uint32 flags, uint32 p1, uint32 p2) if (v->type == VEH_Train && v->subtype != TS_Front_Engine) return CMD_ERROR; + // check that we can allocate enough vehicles + if (!(flags & DC_EXEC)) { + int veh_counter = 0; + do { + veh_counter++; + } while ((v = v->next) != NULL); + + { + Vehicle **vl = malloc(sizeof(Vehicle*) * veh_counter); // some platforms do not support Vehicle *vl[veh_counter] + bool can_allocate_vehicles = AllocateVehicles(vl, veh_counter); + free(vl); + if (!can_allocate_vehicles) { + return_cmd_error(STR_00E1_TOO_MANY_VEHICLES_IN_GAME); + } + } + } + + v = v_front; + do { cost = DoCommand(x, y, v->engine_type, 3, flags, CMD_BUILD_VEH(v->type)); diff --git a/vehicle.h b/vehicle.h index 0773d92313..be2b120d4e 100644 --- a/vehicle.h +++ b/vehicle.h @@ -274,6 +274,7 @@ typedef void *VehicleFromPosProc(Vehicle *v, void *data); void VehicleServiceInDepot(Vehicle *v); Vehicle *AllocateVehicle(void); +bool AllocateVehicles(Vehicle **vl, int num); Vehicle *ForceAllocateVehicle(void); Vehicle *ForceAllocateSpecialVehicle(void); void UpdateVehiclePosHash(Vehicle *v, int x, int y);