Add packet flag for from staked sender (#655)

Co-authored-by: Jed <4679729+jedleggett@users.noreply.github.com>
jito-foundation · Sep 6, 2024 · 9136e1b · 9136e1b
1 parent 80b5466
commit 9136e1b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/core/benches/proto_to_packet.rs b/core/benches/proto_to_packet.rs
@@ -25,6 +25,7 @@ fn get_proto_packet(i: u8) -> PbPacket {
                 repair: false,
                 simple_vote_tx: false,
                 tracer_packet: false,
+                from_staked_node: false,
             }),
             sender_stake: 0,
         }),

diff --git a/core/src/lib.rs b/core/src/lib.rs
@@ -106,6 +106,12 @@ pub fn proto_packet_to_packet(p: jito_protos::proto::packet::Packet) -> Packet {
             if flags.repair {
                 packet.meta_mut().flags.insert(PacketFlags::REPAIR);
             }
+            if flags.from_staked_node {
+                packet
+                    .meta_mut()
+                    .flags
+                    .insert(PacketFlags::FROM_STAKED_NODE)
+            }
         }
     }
     packet