浏览代码

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 {
         if end_ts <= ctx.accounts.clock.unix_timestamp {
             return Err(ErrorCode::InvalidTimestamp.into());
             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 {
         if period_count == 0 {
             return Err(ErrorCode::InvalidPeriod.into());
             return Err(ErrorCode::InvalidPeriod.into());
         }
         }