another invariant

drift-labs · Oct 29, 2024 · 08bc6ec · 08bc6ec
1 parent 8b1abfc
commit 08bc6ec
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/programs/drift_vaults/src/state/vault.rs b/programs/drift_vaults/src/state/vault.rs
@@ -311,6 +311,12 @@ impl Vault {
         }
 
         // this will underflow if there is an issue with protocol fee calc
+        validate!(
+            self.total_shares >= self.user_shares,
+            ErrorCode::InvalidVaultSharesDetected,
+            "total_shares must be >= user_shares"
+        )?;
+
         self.get_manager_shares(vault_protocol)?;
 
         Ok(VaultFee {