From bf8d99c206a825232a17eb4f64afeaf0e05889a4 Mon Sep 17 00:00:00 2001
From: Greyscale <andrewborba10@hotmail.com>
Date: Tue, 29 Jun 2010 12:14:51 -0700
Subject: [PATCH] Fixed an error that happened when buying ammo and not having
 a weapon in prim/sec slot.

---
 src/zr/weapons/zmarket.inc | 29 ++++++++++++++++++++++++-----
 1 file changed, 24 insertions(+), 5 deletions(-)

diff --git a/src/zr/weapons/zmarket.inc b/src/zr/weapons/zmarket.inc
index ffa0f17..ba9a906 100644
--- a/src/zr/weapons/zmarket.inc
+++ b/src/zr/weapons/zmarket.inc
@@ -913,12 +913,31 @@ stock bool:ZMarketEquip(client, const String:weapon[], bool:rebuy = false)
             return false;
         }
         
-        // Give ammo and preserve client's clip ammo value on both primary/secondary weapons.
-        new prim_clip = WeaponAmmoGetAmmo(weapons[Slot_Primary], true);
-        new sec_clip = WeaponAmmoGetAmmo(weapons[Slot_Secondary], true);
+        // Prevent exploit by refilling ammo by re-buying weapon.
+        // Current clip amounts are stored, ammo is refilled, and then clips are reset.
+        
+        new prim_clip;
+        new sec_clip;
+        
+        if (weapons[Slot_Primary] > 0)
+        {
+            prim_clip = WeaponAmmoGetAmmo(weapons[Slot_Primary], true);
+        }
+        if (weapons[Slot_Secondary] > 0)
+        {
+            sec_clip = WeaponAmmoGetAmmo(weapons[Slot_Secondary], true);
+        }
+        
         GivePlayerItem(client, weaponammo);
-        WeaponAmmoSetAmmo(weapons[Slot_Primary], true, prim_clip);
-        WeaponAmmoSetAmmo(weapons[Slot_Secondary], true, sec_clip);
+        
+        if (weapons[Slot_Primary] > 0)
+        {
+            WeaponAmmoSetAmmo(weapons[Slot_Primary], true, prim_clip);
+        }
+        if (weapons[Slot_Secondary] > 0)
+        {
+            WeaponAmmoSetAmmo(weapons[Slot_Secondary], true, sec_clip);
+        }
     }
     
     return true;