added MAX_TICKETS_ALLOWANCE check

multiversx · Nov 22, 2024 · d587001 · d587001
1 parent 1225e73
commit d587001
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/launchpad-guaranteed-tickets-v2/src/guaranteed_tickets_init.rs b/launchpad-guaranteed-tickets-v2/src/guaranteed_tickets_init.rs
@@ -1,6 +1,7 @@
 multiversx_sc::imports!();
 multiversx_sc::derive_imports!();
 
+pub const MAX_TICKETS_ALLOWANCE: usize = 128;
 pub const MAX_GUARANTEED_TICKETS_ENTRIES: usize = 10;
 
 #[derive(TopEncode, TopDecode, NestedEncode, NestedDecode, TypeAbi, ManagedVecItem)]
@@ -55,6 +56,10 @@ pub trait GuaranteedTicketsInitModule:
 
         for multi_arg in address_number_pairs.into_iter() {
             let (buyer, total_tickets_allowance, guaranteed_ticket_raw) = multi_arg.into_tuple();
+            require!(
+                total_tickets_allowance <= MAX_TICKETS_ALLOWANCE,
+                "Total number of tickets exceeds maximum allowed"
+            );
             require!(
                 guaranteed_ticket_raw.len() <= MAX_GUARANTEED_TICKETS_ENTRIES,
                 "Number of guaranteed tickets entries exceeds maximum allowed"