feat: add publisher caps to the validator

program/rust/Cargo.toml

@@ -17,7 +17,7 @@ num-traits = "0.2"
 byteorder = "1.4.3"
 serde = { version = "1.0", features = ["derive"], optional = true }
 strum = { version = "0.24.1", features = ["derive"], optional = true }
-pythnet-sdk = { git = "https://github.com/pyth-network/pyth-crosschain", rev="60144002053a93f424be70decd8a8ccb8d618d81"}
+pythnet-sdk = "2.2.0"
 solana-sdk = { version = "=1.14.17", optional = true }
 bitflags = { version = "2.6.0", features = ["bytemuck"] }
 
@@ -31,7 +31,7 @@ rand = "0.8.5"
 quickcheck_macros = "1"
 bincode = "1.3.3"
 serde = { version = "1.0", features = ["derive"] }
-pythnet-sdk = { git = "https://github.com/pyth-network/pyth-crosschain", rev="60144002053a93f424be70decd8a8ccb8d618d81", features = ["quickcheck"]}
+pythnet-sdk = { version = "2.2.0" , features = ["quickcheck"]}
 serde_json = "1.0"
 test-generator = "0.3.1"
 csv = "1.1"

program/rust/src/accounts/price.rs

@@ -13,10 +13,14 @@ use {
     },
     pythnet_sdk::messages::{
         PriceFeedMessage,
+        PublisherStakeCapsMessage,
         TwapMessage,
     },
     solana_program::pubkey::Pubkey,
-    std::mem::size_of,
+    std::{
+        mem::size_of,
+        u16,
+    },
 };
 
 /// Pythnet-specific PriceAccount implementation
@@ -111,7 +115,7 @@ mod price_pythnet {
             };
 
             PriceFeedMessage {
-                id: key.to_bytes(),
+                feed_id: key.to_bytes(),
                 price,
                 conf,
                 exponent: self.exponent,
@@ -141,7 +145,7 @@ mod price_pythnet {
             };
 
             TwapMessage {
-                id: key.to_bytes(),
+                feed_id: key.to_bytes(),
                 cumulative_price: self.price_cumulative.price,
                 cumulative_conf: self.price_cumulative.conf,
                 num_down_slots: self.price_cumulative.num_down_slots,
@@ -270,7 +274,7 @@ impl PythOracleSerialize for PriceFeedMessage {
         bytes[i..i + 1].clone_from_slice(&[DISCRIMINATOR]);
         i += 1;
 
-        bytes[i..i + 32].clone_from_slice(&self.id[..]);
+        bytes[i..i + 32].clone_from_slice(&self.feed_id[..]);
         i += 32;
 
         bytes[i..i + 8].clone_from_slice(&self.price.to_be_bytes());
@@ -313,7 +317,7 @@ impl PythOracleSerialize for TwapMessage {
         bytes[i..i + 1].clone_from_slice(&[DISCRIMINATOR]);
         i += 1;
 
-        bytes[i..i + 32].clone_from_slice(&self.id[..]);
+        bytes[i..i + 32].clone_from_slice(&self.feed_id[..]);
         i += 32;
 
         bytes[i..i + 16].clone_from_slice(&self.cumulative_price.to_be_bytes());
@@ -340,3 +344,23 @@ impl PythOracleSerialize for TwapMessage {
         bytes.to_vec()
     }
 }
+
+impl PythOracleSerialize for PublisherStakeCapsMessage {
+    fn to_bytes(self) -> Vec<u8> {
+        const DISCRIMINATOR: u8 = 2;
+        let mut result = vec![DISCRIMINATOR];
+        result.extend_from_slice(&self.publish_time.to_be_bytes());
+        result.extend_from_slice(
+            &u16::try_from(self.caps.as_ref().len())
+                .unwrap_or(u16::MAX)
+                .to_be_bytes(),
+        );
+
+        for cap in self.caps {
+            result.extend_from_slice(&cap.publisher);
+            result.extend_from_slice(&cap.cap.to_be_bytes());
+        }
+
+        result
+    }
+}

program/rust/src/tests/test_message.rs

@@ -5,10 +5,15 @@ use {
         messages::{
             Message,
             PriceFeedMessage,
+            PublisherStakeCapsMessage,
             TwapMessage,
         },
         wire::from_slice,
     },
+    quickcheck::{
+        Gen,
+        QuickCheck,
+    },
     quickcheck_macros::quickcheck,
 };
 
@@ -38,3 +43,25 @@ fn test_twap_message_roundtrip(input: TwapMessage) -> bool {
         _ => false,
     }
 }
+
+
+fn prop_publisher_caps_message_roundtrip(input: PublisherStakeCapsMessage) -> bool {
+    let reconstructed = from_slice::<BigEndian, Message>(&input.clone().to_bytes()).unwrap();
+
+    println!("Failed test case:");
+    println!("{:?}", input);
+    println!("{:?}", reconstructed);
+
+    match reconstructed {
+        Message::PublisherStakeCapsMessage(reconstructed) => reconstructed == input,
+        _ => false,
+    }
+}
+
+#[test]
+fn test_publisher_caps_message_roundtrip() {
+    // Configure the size parameter for the generator
+    QuickCheck::new()
+        .gen(Gen::new(1024))
+        .quickcheck(prop_publisher_caps_message_roundtrip as fn(PublisherStakeCapsMessage) -> bool);
+}

program/rust/src/tests/test_upd_price_with_validator.rs

@@ -130,7 +130,7 @@ fn test_upd_price_with_validator() {
             .unwrap();
     let expected_messages1 = [
         PriceFeedMessage {
-            id:                price_account.key.to_bytes(),
+            feed_id:           price_account.key.to_bytes(),
             price:             42,
             conf:              2,
             exponent:          0,
@@ -141,7 +141,7 @@ fn test_upd_price_with_validator() {
         }
         .to_bytes(),
         TwapMessage {
-            id:                price_account.key.to_bytes(),
+            feed_id:           price_account.key.to_bytes(),
             cumulative_price:  42,
             cumulative_conf:   2,
             num_down_slots:    0,
@@ -161,7 +161,7 @@ fn test_upd_price_with_validator() {
 
     let expected_messages2 = [
         PriceFeedMessage {
-            id:                price_account.key.to_bytes(),
+            feed_id:           price_account.key.to_bytes(),
             price:             42,
             conf:              2,
             exponent:          0,
@@ -172,7 +172,7 @@ fn test_upd_price_with_validator() {
         }
         .to_bytes(),
         TwapMessage {
-            id:                price_account.key.to_bytes(),
+            feed_id:           price_account.key.to_bytes(),
             cumulative_price:  84,
             cumulative_conf:   4,
             num_down_slots:    0,