wip

solana-labs · May 30, 2023 · 9afe8eb · 9afe8eb
1 parent 6c52039
commit 9afe8eb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sdk/program/src/stake/state.rs b/sdk/program/src/stake/state.rs
@@ -564,9 +564,13 @@ impl Stake {
         Ok(new)
     }
 
-    pub fn deactivate(&mut self, epoch: Epoch) -> Result<(), StakeError> {
+    pub fn deactivate(&mut self, epoch: Epoch, flag: MustFullyActivateBeforeDeactivationIsPermittedFlag) -> Result<(), StakeError> {
         if self.delegation.deactivation_epoch != std::u64::MAX {
             Err(StakeError::AlreadyDeactivated)
+
+        else if flag == 1 && epoch < self.delegation.activation_epoch {
+            Err(StakeError::MustFullyActivateBeforeDeactivationIsPermitted)
+        }
         } else {
             self.delegation.deactivation_epoch = epoch;
             Ok(())