From 0b8961597b3adf7355e48506d7e81b3925bbacd0 Mon Sep 17 00:00:00 2001
From: Jordan Sexton <jordan@jordansexton.com>
Date: Sat, 4 Dec 2021 17:59:20 -0600
Subject: [PATCH] lending: validate pyth price trading status (#2618)

---
 token-lending/program/src/processor.rs | 5 +++++
 token-lending/program/src/pyth.rs      | 6 +++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/token-lending/program/src/processor.rs b/token-lending/program/src/processor.rs
index ac29f6531f6..325fe6e38c5 100644
--- 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
index 0a44062f6c8..10ec0408a62 100644
--- 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,