From 343676fcf0870812b6eabe7a67b6897501e16f12 Mon Sep 17 00:00:00 2001
From: buffalu <85544055+buffalu@users.noreply.github.com>
Date: Wed, 7 Aug 2024 19:31:41 -0400
Subject: [PATCH] Add packet flag for from staked sender (v1.18) (#649)

Co-authored-by: Jed <4679729+jedleggett@users.noreply.github.com>
---
 .buildkite/scripts/build-stable.test.sh | 2 +-
 core/benches/proto_to_packet.rs         | 1 +
 core/src/lib.rs                         | 6 ++++++
 jito-protos/protos                      | 2 +-
 4 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/.buildkite/scripts/build-stable.test.sh b/.buildkite/scripts/build-stable.test.sh
index 0df2094586..5393bcc5c4 100755
--- a/.buildkite/scripts/build-stable.test.sh
+++ b/.buildkite/scripts/build-stable.test.sh
@@ -12,7 +12,7 @@ want=$(
     steps:
       - name: "partitions"
         command: ". ci/rust-version.sh; ci/docker-run.sh $$rust_stable_docker_image ci/stable/run-partition.sh"
-        timeout_in_minutes: 40
+        timeout_in_minutes: 60
         agents:
           queue: "solana"
         parallelism: 3
diff --git a/core/benches/proto_to_packet.rs b/core/benches/proto_to_packet.rs
index 87f85f9c7f..1727d8af9c 100644
--- 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
index b072b1e791..70c52d85ca 100644
--- a/core/src/lib.rs
+++ b/core/src/lib.rs
@@ -108,6 +108,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
index 05d210980f..b74a23ff23 160000
--- a/jito-protos/protos
+++ b/jito-protos/protos
@@ -1 +1 @@
-Subproject commit 05d210980f34a7c974d7ed1a4dbcb2ce1fca00b3
+Subproject commit b74a23ff236c0c223b1bc56daf7c5065bb585428