lending: validate pyth price trading status (solana-labs#2618)

Nikolay-everstake · Dec 4, 2021 · 0b89615 · 0b89615
1 parent eaca634
commit 0b89615
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/token-lending/program/src/processor.rs b/token-lending/program/src/processor.rs
@@ -1775,6 +1775,11 @@ fn get_pyth_price(pyth_price_info: &AccountInfo, clock: &Clock) -> Result<Decima
         return Err(LendingError::InvalidOracleConfig.into());
     }
 
+    if pyth_price.agg.status != pyth::PriceStatus::Trading {
+        msg!("Oracle price status is invalid");
+        return Err(LendingError::InvalidOracleConfig.into());
+    }
+
     let slots_elapsed = clock
         .slot
         .checked_sub(pyth_price.valid_slot)

diff --git a/token-lending/program/src/pyth.rs b/token-lending/program/src/pyth.rs
@@ -20,7 +20,7 @@ pub struct AccKey {
     pub val: [u8; 32],
 }
 
-#[derive(Copy, Clone)]
+#[derive(PartialEq, Copy, Clone)]
 #[repr(C)]
 pub enum AccountType {
     Unknown,
@@ -29,7 +29,7 @@ pub enum AccountType {
     Price,
 }
 
-#[derive(Copy, Clone)]
+#[derive(PartialEq, Copy, Clone)]
 #[repr(C)]
 pub enum PriceStatus {
     Unknown,
@@ -38,7 +38,7 @@ pub enum PriceStatus {
     Auction,
 }
 
-#[derive(Copy, Clone)]
+#[derive(PartialEq, Copy, Clone)]
 #[repr(C)]
 pub enum CorpAction {
     NoCorpAct,