add packet flag for staked node

jito-foundation · Nov 8, 2024 · 1e9bb98 · 1e9bb98
1 parent 5b10a31
commit 1e9bb98
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
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

diff --git a/jito-protos/protos b/jito-protos/protos
+18 −5		README.md
+1 −0		auth.proto
+11 −7		block_engine.proto
+95 −0		bundle.proto
+245 −0		json_rpc/http.md
+1 −0		packet.proto
+42 −26		searcher.proto
+25 −0		shredstream.proto
+15 −0		trace_shred.proto