Переглянути джерело

examples/lockup: Disallow vesting accounts with invalid periods

Armani Ferrante 4 роки тому
батько
коміт
a078b683b8
1 змінених файлів з 3 додано та 0 видалено
  1. 3 0
      examples/lockup/programs/lockup/src/lib.rs

+ 3 - 0
examples/lockup/programs/lockup/src/lib.rs

@@ -78,6 +78,9 @@ pub mod lockup {
         if end_ts <= ctx.accounts.clock.unix_timestamp {
             return Err(ErrorCode::InvalidTimestamp.into());
         }
+        if period_count > (end_ts - ctx.accounts.clock.unix_timestamp) as u64 {
+            return Err(ErrorCode::InvalidPeriod.into());
+        }
         if period_count == 0 {
             return Err(ErrorCode::InvalidPeriod.into());
         }