Add remote_pubkey to Packet Meta

packet/Cargo.toml

@@ -35,8 +35,10 @@ serde_derive = { workspace = true, optional = true }
 serde_with = { workspace = true, optional = true, features = ["macros"] }
 solana-frozen-abi = { workspace = true, optional = true, features = ["frozen-abi"] }
 solana-frozen-abi-macro = { workspace = true, optional = true, features = ["frozen-abi"] }
+solana-pubkey = { workspace = true, features = ["serde"] }
 
 [dev-dependencies]
+solana-address = { workspace = true, features = ["atomic"] }
 solana-packet = { path = ".", features = ["dev-context-only-utils"] }
 static_assertions = { workspace = true }
 

packet/src/lib.rs

@@ -11,6 +11,7 @@ use {
 };
 use {
     bitflags::bitflags,
+    solana_pubkey::Pubkey,
     std::{
         fmt,
         net::{IpAddr, Ipv4Addr, SocketAddr},
@@ -72,6 +73,7 @@ pub struct Meta {
     pub addr: IpAddr,
     pub port: u16,
     pub flags: PacketFlags,
+    remote_pubkey: Pubkey,
 }
 
 #[cfg(feature = "frozen-abi")]
@@ -300,6 +302,20 @@ impl Meta {
     pub fn is_from_staked_node(&self) -> bool {
         self.flags.contains(PacketFlags::FROM_STAKED_NODE)
     }
+
+    #[inline]
+    pub fn remote_pubkey(&self) -> Option<Pubkey> {
+        if self.remote_pubkey == Pubkey::default() {
+            None
+        } else {
+            Some(self.remote_pubkey)
+        }
+    }
+
+    #[inline]
+    pub fn set_remote_pubkey(&mut self, pubkey: Pubkey) {
+        self.remote_pubkey = pubkey;
+    }
 }
 
 impl Default for Meta {
@@ -309,6 +325,7 @@ impl Default for Meta {
             addr: IpAddr::V4(Ipv4Addr::UNSPECIFIED),
             port: 0,
             flags: PacketFlags::empty(),
+            remote_pubkey: Pubkey::default(),
         }
     }
 }
@@ -355,4 +372,16 @@ mod tests {
             Err("the size limit has been reached".to_string()),
         );
     }
+
+    #[test]
+    fn test_remote_pubkey() {
+        let mut meta = Meta::default();
+        assert!(meta.remote_pubkey().is_none());
+        let pubkey = Pubkey::new_unique();
+        meta.set_remote_pubkey(pubkey);
+        assert_eq!(meta.remote_pubkey(), Some(pubkey));
+        let pubkey = Pubkey::default();
+        meta.set_remote_pubkey(pubkey);
+        assert!(meta.remote_pubkey().is_none());
+    }
 }