2018-05-14 16:36:19 -07:00
|
|
|
//! The `tpu` module implements the Transaction Processing Unit, a
|
2019-01-08 08:21:39 -08:00
|
|
|
//! multi-stage transaction processing pipeline in software.
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2021-12-03 09:00:31 -08:00
|
|
|
use {
|
|
|
|
crate::{
|
|
|
|
banking_stage::BankingStage,
|
|
|
|
broadcast_stage::{BroadcastStage, BroadcastStageType, RetransmitSlotsReceiver},
|
|
|
|
cluster_info_vote_listener::{
|
|
|
|
ClusterInfoVoteListener, GossipDuplicateConfirmedSlotsSender,
|
|
|
|
GossipVerifiedVoteHashSender, VerifiedVoteSender, VoteTracker,
|
|
|
|
},
|
|
|
|
fetch_stage::FetchStage,
|
2022-03-15 16:44:53 -07:00
|
|
|
find_packet_sender_stake_stage::FindPacketSenderStakeStage,
|
2021-12-03 09:00:31 -08:00
|
|
|
sigverify::TransactionSigVerifier,
|
|
|
|
sigverify_stage::SigVerifyStage,
|
2022-04-16 01:54:22 -07:00
|
|
|
staked_nodes_updater_service::StakedNodesUpdaterService,
|
2021-03-24 23:41:52 -07:00
|
|
|
},
|
2022-06-22 08:57:46 -07:00
|
|
|
crossbeam_channel::{unbounded, Receiver},
|
2022-06-08 04:57:12 -07:00
|
|
|
solana_client::connection_cache::ConnectionCache,
|
2021-12-03 09:00:31 -08:00
|
|
|
solana_gossip::cluster_info::ClusterInfo,
|
|
|
|
solana_ledger::{blockstore::Blockstore, blockstore_processor::TransactionStatusSender},
|
|
|
|
solana_poh::poh_recorder::{PohRecorder, WorkingBankEntry},
|
|
|
|
solana_rpc::{
|
|
|
|
optimistically_confirmed_bank_tracker::BankNotificationSender,
|
|
|
|
rpc_subscriptions::RpcSubscriptions,
|
|
|
|
},
|
|
|
|
solana_runtime::{
|
|
|
|
bank_forks::BankForks,
|
|
|
|
cost_model::CostModel,
|
|
|
|
vote_sender_types::{ReplayVoteReceiver, ReplayVoteSender},
|
|
|
|
},
|
2022-02-04 06:27:09 -08:00
|
|
|
solana_sdk::signature::Keypair,
|
2022-06-21 12:06:44 -07:00
|
|
|
solana_streamer::{
|
|
|
|
quic::{spawn_server, StreamStats, MAX_STAKED_CONNECTIONS, MAX_UNSTAKED_CONNECTIONS},
|
|
|
|
streamer::StakedNodes,
|
2022-06-02 11:14:58 -07:00
|
|
|
},
|
2021-12-03 09:00:31 -08:00
|
|
|
std::{
|
|
|
|
net::UdpSocket,
|
2022-07-05 07:29:44 -07:00
|
|
|
sync::{atomic::AtomicBool, Arc, RwLock},
|
2021-12-03 09:00:31 -08:00
|
|
|
thread,
|
2019-11-20 15:43:10 -08:00
|
|
|
},
|
|
|
|
};
|
2018-09-14 00:17:40 -07:00
|
|
|
|
2021-02-26 09:15:45 -08:00
|
|
|
pub const DEFAULT_TPU_COALESCE_MS: u64 = 5;
|
|
|
|
|
2022-03-09 01:52:31 -08:00
|
|
|
// allow multiple connections for NAT and any open/close overlap
|
2022-07-13 09:59:01 -07:00
|
|
|
pub const MAX_QUIC_CONNECTIONS_PER_PEER: usize = 8;
|
2022-03-09 01:52:31 -08:00
|
|
|
|
2022-01-10 09:29:48 -08:00
|
|
|
pub struct TpuSockets {
|
|
|
|
pub transactions: Vec<UdpSocket>,
|
|
|
|
pub transaction_forwards: Vec<UdpSocket>,
|
|
|
|
pub vote: Vec<UdpSocket>,
|
|
|
|
pub broadcast: Vec<UdpSocket>,
|
2022-02-04 06:27:09 -08:00
|
|
|
pub transactions_quic: UdpSocket,
|
2022-06-02 11:14:58 -07:00
|
|
|
pub transactions_forwards_quic: UdpSocket,
|
2022-01-10 09:29:48 -08:00
|
|
|
}
|
|
|
|
|
2019-03-04 20:08:21 -08:00
|
|
|
pub struct Tpu {
|
2019-02-16 18:03:55 -08:00
|
|
|
fetch_stage: FetchStage,
|
|
|
|
sigverify_stage: SigVerifyStage,
|
2021-10-07 02:38:23 -07:00
|
|
|
vote_sigverify_stage: SigVerifyStage,
|
2019-02-16 18:03:55 -08:00
|
|
|
banking_stage: BankingStage,
|
|
|
|
cluster_info_vote_listener: ClusterInfoVoteListener,
|
2019-03-01 19:43:30 -08:00
|
|
|
broadcast_stage: BroadcastStage,
|
2022-07-20 11:37:24 -07:00
|
|
|
tpu_quic_t: thread::JoinHandle<()>,
|
|
|
|
tpu_forwards_quic_t: thread::JoinHandle<()>,
|
2022-03-16 17:02:22 -07:00
|
|
|
find_packet_sender_stake_stage: FindPacketSenderStakeStage,
|
|
|
|
vote_find_packet_sender_stake_stage: FindPacketSenderStakeStage,
|
2022-04-16 01:54:22 -07:00
|
|
|
staked_nodes_updater_service: StakedNodesUpdaterService,
|
2019-01-26 00:28:08 -08:00
|
|
|
}
|
|
|
|
|
2018-05-14 16:36:19 -07:00
|
|
|
impl Tpu {
|
2019-03-09 02:47:41 -08:00
|
|
|
#[allow(clippy::too_many_arguments)]
|
2019-03-03 16:44:06 -08:00
|
|
|
pub fn new(
|
2020-04-21 12:54:45 -07:00
|
|
|
cluster_info: &Arc<ClusterInfo>,
|
2022-07-05 07:29:44 -07:00
|
|
|
poh_recorder: &Arc<RwLock<PohRecorder>>,
|
2019-09-18 12:16:22 -07:00
|
|
|
entry_receiver: Receiver<WorkingBankEntry>,
|
2020-03-19 23:35:01 -07:00
|
|
|
retransmit_slots_receiver: RetransmitSlotsReceiver,
|
2022-01-10 09:29:48 -08:00
|
|
|
sockets: TpuSockets,
|
2020-05-17 14:01:08 -07:00
|
|
|
subscriptions: &Arc<RpcSubscriptions>,
|
2019-11-20 15:43:10 -08:00
|
|
|
transaction_status_sender: Option<TransactionStatusSender>,
|
2020-01-13 13:13:52 -08:00
|
|
|
blockstore: &Arc<Blockstore>,
|
2019-06-19 00:13:19 -07:00
|
|
|
broadcast_type: &BroadcastStageType,
|
2019-03-04 16:33:14 -08:00
|
|
|
exit: &Arc<AtomicBool>,
|
2019-11-18 18:05:02 -08:00
|
|
|
shred_version: u16,
|
2020-03-09 22:03:09 -07:00
|
|
|
vote_tracker: Arc<VoteTracker>,
|
|
|
|
bank_forks: Arc<RwLock<BankForks>>,
|
2020-07-09 22:52:54 -07:00
|
|
|
verified_vote_sender: VerifiedVoteSender,
|
2021-04-10 17:34:45 -07:00
|
|
|
gossip_verified_vote_hash_sender: GossipVerifiedVoteHashSender,
|
2020-08-07 11:21:35 -07:00
|
|
|
replay_vote_receiver: ReplayVoteReceiver,
|
|
|
|
replay_vote_sender: ReplayVoteSender,
|
2020-09-28 19:43:05 -07:00
|
|
|
bank_notification_sender: Option<BankNotificationSender>,
|
2021-02-26 09:15:45 -08:00
|
|
|
tpu_coalesce_ms: u64,
|
2021-03-24 23:41:52 -07:00
|
|
|
cluster_confirmed_slot_sender: GossipDuplicateConfirmedSlotsSender,
|
2021-06-09 15:10:59 -07:00
|
|
|
cost_model: &Arc<RwLock<CostModel>>,
|
2022-06-08 04:57:12 -07:00
|
|
|
connection_cache: &Arc<ConnectionCache>,
|
2022-02-04 06:27:09 -08:00
|
|
|
keypair: &Keypair,
|
2022-07-11 08:53:18 -07:00
|
|
|
log_messages_bytes_limit: Option<usize>,
|
2019-03-03 16:44:06 -08:00
|
|
|
) -> Self {
|
2022-01-10 09:29:48 -08:00
|
|
|
let TpuSockets {
|
|
|
|
transactions: transactions_sockets,
|
|
|
|
transaction_forwards: tpu_forwards_sockets,
|
|
|
|
vote: tpu_vote_sockets,
|
|
|
|
broadcast: broadcast_sockets,
|
2022-02-04 06:27:09 -08:00
|
|
|
transactions_quic: transactions_quic_sockets,
|
2022-06-02 11:14:58 -07:00
|
|
|
transactions_forwards_quic: transactions_forwards_quic_sockets,
|
2022-01-10 09:29:48 -08:00
|
|
|
} = sockets;
|
|
|
|
|
2022-01-11 02:44:46 -08:00
|
|
|
let (packet_sender, packet_receiver) = unbounded();
|
|
|
|
let (vote_packet_sender, vote_packet_receiver) = unbounded();
|
2022-06-02 11:14:58 -07:00
|
|
|
let (forwarded_packet_sender, forwarded_packet_receiver) = unbounded();
|
2019-03-08 14:59:11 -08:00
|
|
|
let fetch_stage = FetchStage::new_with_sender(
|
|
|
|
transactions_sockets,
|
2019-07-30 14:50:02 -07:00
|
|
|
tpu_forwards_sockets,
|
2021-10-07 02:38:23 -07:00
|
|
|
tpu_vote_sockets,
|
2021-06-18 06:34:46 -07:00
|
|
|
exit,
|
2019-04-09 12:57:12 -07:00
|
|
|
&packet_sender,
|
2021-10-07 02:38:23 -07:00
|
|
|
&vote_packet_sender,
|
2022-06-02 11:14:58 -07:00
|
|
|
&forwarded_packet_sender,
|
|
|
|
forwarded_packet_receiver,
|
2021-06-18 06:34:46 -07:00
|
|
|
poh_recorder,
|
2021-02-26 09:15:45 -08:00
|
|
|
tpu_coalesce_ms,
|
2022-05-14 08:53:37 -07:00
|
|
|
Some(bank_forks.read().unwrap().get_vote_only_mode_signal()),
|
2019-03-08 14:59:11 -08:00
|
|
|
);
|
2021-12-16 06:47:55 -08:00
|
|
|
|
2022-06-21 12:06:44 -07:00
|
|
|
let staked_nodes = Arc::new(RwLock::new(StakedNodes::default()));
|
2022-05-20 12:51:07 -07:00
|
|
|
let staked_nodes_updater_service = StakedNodesUpdaterService::new(
|
|
|
|
exit.clone(),
|
|
|
|
cluster_info.clone(),
|
|
|
|
bank_forks.clone(),
|
|
|
|
staked_nodes.clone(),
|
|
|
|
);
|
|
|
|
|
2022-03-16 17:02:22 -07:00
|
|
|
let (find_packet_sender_stake_sender, find_packet_sender_stake_receiver) = unbounded();
|
2021-12-16 06:47:55 -08:00
|
|
|
|
2022-03-16 17:02:22 -07:00
|
|
|
let find_packet_sender_stake_stage = FindPacketSenderStakeStage::new(
|
2021-12-16 06:47:55 -08:00
|
|
|
packet_receiver,
|
2022-03-16 17:02:22 -07:00
|
|
|
find_packet_sender_stake_sender,
|
2022-05-20 12:51:07 -07:00
|
|
|
staked_nodes.clone(),
|
2022-05-06 12:16:13 -07:00
|
|
|
"tpu-find-packet-sender-stake",
|
2021-12-16 06:47:55 -08:00
|
|
|
);
|
|
|
|
|
2022-03-16 17:02:22 -07:00
|
|
|
let (vote_find_packet_sender_stake_sender, vote_find_packet_sender_stake_receiver) =
|
|
|
|
unbounded();
|
2021-12-16 06:47:55 -08:00
|
|
|
|
2022-03-16 17:02:22 -07:00
|
|
|
let vote_find_packet_sender_stake_stage = FindPacketSenderStakeStage::new(
|
2021-12-16 06:47:55 -08:00
|
|
|
vote_packet_receiver,
|
2022-03-16 17:02:22 -07:00
|
|
|
vote_find_packet_sender_stake_sender,
|
2022-05-20 12:51:07 -07:00
|
|
|
staked_nodes.clone(),
|
2022-05-06 12:16:13 -07:00
|
|
|
"tpu-vote-find-packet-sender-stake",
|
2021-12-16 06:47:55 -08:00
|
|
|
);
|
|
|
|
|
2019-06-26 18:42:27 -07:00
|
|
|
let (verified_sender, verified_receiver) = unbounded();
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2022-06-02 11:14:58 -07:00
|
|
|
let stats = Arc::new(StreamStats::default());
|
2022-07-20 11:37:24 -07:00
|
|
|
let tpu_quic_t = spawn_server(
|
|
|
|
transactions_quic_sockets,
|
|
|
|
keypair,
|
|
|
|
cluster_info.my_contact_info().tpu.ip(),
|
|
|
|
packet_sender,
|
|
|
|
exit.clone(),
|
|
|
|
MAX_QUIC_CONNECTIONS_PER_PEER,
|
|
|
|
staked_nodes.clone(),
|
|
|
|
MAX_STAKED_CONNECTIONS,
|
|
|
|
MAX_UNSTAKED_CONNECTIONS,
|
|
|
|
stats.clone(),
|
|
|
|
)
|
|
|
|
.unwrap();
|
2022-06-02 11:14:58 -07:00
|
|
|
|
2022-07-20 11:37:24 -07:00
|
|
|
let tpu_forwards_quic_t = spawn_server(
|
|
|
|
transactions_forwards_quic_sockets,
|
|
|
|
keypair,
|
|
|
|
cluster_info.my_contact_info().tpu_forwards.ip(),
|
|
|
|
forwarded_packet_sender,
|
|
|
|
exit.clone(),
|
|
|
|
MAX_QUIC_CONNECTIONS_PER_PEER,
|
|
|
|
staked_nodes,
|
|
|
|
MAX_STAKED_CONNECTIONS.saturating_add(MAX_UNSTAKED_CONNECTIONS),
|
|
|
|
0, // Prevent unstaked nodes from forwarding transactions
|
|
|
|
stats,
|
|
|
|
)
|
|
|
|
.unwrap();
|
2022-02-04 06:27:09 -08:00
|
|
|
|
2020-05-08 10:00:23 -07:00
|
|
|
let sigverify_stage = {
|
2022-05-24 14:01:41 -07:00
|
|
|
let verifier = TransactionSigVerifier::new(verified_sender);
|
|
|
|
SigVerifyStage::new(find_packet_sender_stake_receiver, verifier, "tpu-verifier")
|
2019-10-28 16:07:51 -07:00
|
|
|
};
|
2019-04-09 12:57:12 -07:00
|
|
|
|
2021-10-07 02:38:23 -07:00
|
|
|
let (verified_tpu_vote_packets_sender, verified_tpu_vote_packets_receiver) = unbounded();
|
|
|
|
|
|
|
|
let vote_sigverify_stage = {
|
2022-05-24 14:01:41 -07:00
|
|
|
let verifier =
|
|
|
|
TransactionSigVerifier::new_reject_non_vote(verified_tpu_vote_packets_sender);
|
2021-10-07 02:38:23 -07:00
|
|
|
SigVerifyStage::new(
|
2022-03-16 17:02:22 -07:00
|
|
|
vote_find_packet_sender_stake_receiver,
|
2021-10-07 02:38:23 -07:00
|
|
|
verifier,
|
2022-04-28 01:16:17 -07:00
|
|
|
"tpu-vote-verifier",
|
2021-10-07 02:38:23 -07:00
|
|
|
)
|
|
|
|
};
|
|
|
|
|
|
|
|
let (verified_gossip_vote_packets_sender, verified_gossip_vote_packets_receiver) =
|
|
|
|
unbounded();
|
2019-04-09 12:57:12 -07:00
|
|
|
let cluster_info_vote_listener = ClusterInfoVoteListener::new(
|
2021-12-30 07:03:14 -08:00
|
|
|
exit.clone(),
|
2019-04-09 12:57:12 -07:00
|
|
|
cluster_info.clone(),
|
2021-10-07 02:38:23 -07:00
|
|
|
verified_gossip_vote_packets_sender,
|
2021-12-30 07:03:14 -08:00
|
|
|
poh_recorder.clone(),
|
2020-03-09 22:03:09 -07:00
|
|
|
vote_tracker,
|
2021-07-07 08:21:12 -07:00
|
|
|
bank_forks.clone(),
|
2020-05-17 14:01:08 -07:00
|
|
|
subscriptions.clone(),
|
2020-07-09 22:52:54 -07:00
|
|
|
verified_vote_sender,
|
2021-04-10 17:34:45 -07:00
|
|
|
gossip_verified_vote_hash_sender,
|
2020-08-07 11:21:35 -07:00
|
|
|
replay_vote_receiver,
|
2020-07-28 02:33:27 -07:00
|
|
|
blockstore.clone(),
|
2020-09-28 19:43:05 -07:00
|
|
|
bank_notification_sender,
|
2021-03-24 23:41:52 -07:00
|
|
|
cluster_confirmed_slot_sender,
|
2019-04-09 12:57:12 -07:00
|
|
|
);
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2019-04-17 21:07:45 -07:00
|
|
|
let banking_stage = BankingStage::new(
|
2021-06-18 06:34:46 -07:00
|
|
|
cluster_info,
|
2019-04-17 21:07:45 -07:00
|
|
|
poh_recorder,
|
|
|
|
verified_receiver,
|
2021-10-07 02:38:23 -07:00
|
|
|
verified_tpu_vote_packets_receiver,
|
|
|
|
verified_gossip_vote_packets_receiver,
|
2019-11-20 15:43:10 -08:00
|
|
|
transaction_status_sender,
|
2020-08-07 11:21:35 -07:00
|
|
|
replay_vote_sender,
|
2021-10-19 12:37:33 -07:00
|
|
|
cost_model.clone(),
|
2022-07-11 08:53:18 -07:00
|
|
|
log_messages_bytes_limit,
|
2022-06-08 04:57:12 -07:00
|
|
|
connection_cache.clone(),
|
2022-07-05 21:24:58 -07:00
|
|
|
bank_forks.clone(),
|
2019-04-17 21:07:45 -07:00
|
|
|
);
|
2019-01-26 00:28:08 -08:00
|
|
|
|
2019-06-19 00:13:19 -07:00
|
|
|
let broadcast_stage = broadcast_type.new_broadcast_stage(
|
2019-12-16 17:11:18 -08:00
|
|
|
broadcast_sockets,
|
2019-03-03 16:44:06 -08:00
|
|
|
cluster_info.clone(),
|
2019-01-26 00:28:08 -08:00
|
|
|
entry_receiver,
|
2020-03-19 23:35:01 -07:00
|
|
|
retransmit_slots_receiver,
|
2022-06-15 06:44:24 -07:00
|
|
|
exit.clone(),
|
|
|
|
blockstore.clone(),
|
|
|
|
bank_forks,
|
2019-11-18 18:05:02 -08:00
|
|
|
shred_version,
|
2018-10-18 22:57:48 -07:00
|
|
|
);
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2019-03-04 20:08:21 -08:00
|
|
|
Self {
|
2018-07-03 21:14:08 -07:00
|
|
|
fetch_stage,
|
|
|
|
sigverify_stage,
|
2021-10-07 02:38:23 -07:00
|
|
|
vote_sigverify_stage,
|
2018-07-03 21:14:08 -07:00
|
|
|
banking_stage,
|
2019-01-31 15:51:29 -08:00
|
|
|
cluster_info_vote_listener,
|
2019-03-01 19:43:30 -08:00
|
|
|
broadcast_stage,
|
2022-02-04 06:27:09 -08:00
|
|
|
tpu_quic_t,
|
2022-06-02 11:14:58 -07:00
|
|
|
tpu_forwards_quic_t,
|
2022-03-16 17:02:22 -07:00
|
|
|
find_packet_sender_stake_stage,
|
|
|
|
vote_find_packet_sender_stake_stage,
|
2022-04-16 01:54:22 -07:00
|
|
|
staked_nodes_updater_service,
|
2019-03-03 16:44:06 -08:00
|
|
|
}
|
2019-01-26 00:28:08 -08:00
|
|
|
}
|
2018-07-03 21:14:08 -07:00
|
|
|
|
2019-11-13 10:12:09 -08:00
|
|
|
pub fn join(self) -> thread::Result<()> {
|
2021-01-23 11:55:15 -08:00
|
|
|
let results = vec![
|
|
|
|
self.fetch_stage.join(),
|
|
|
|
self.sigverify_stage.join(),
|
2021-10-07 02:38:23 -07:00
|
|
|
self.vote_sigverify_stage.join(),
|
2021-01-23 11:55:15 -08:00
|
|
|
self.cluster_info_vote_listener.join(),
|
|
|
|
self.banking_stage.join(),
|
2022-03-16 17:02:22 -07:00
|
|
|
self.find_packet_sender_stake_stage.join(),
|
|
|
|
self.vote_find_packet_sender_stake_stage.join(),
|
2022-04-16 01:54:22 -07:00
|
|
|
self.staked_nodes_updater_service.join(),
|
2022-07-20 11:37:24 -07:00
|
|
|
self.tpu_quic_t.join(),
|
|
|
|
self.tpu_forwards_quic_t.join(),
|
2021-01-23 11:55:15 -08:00
|
|
|
];
|
2019-03-04 20:08:21 -08:00
|
|
|
let broadcast_result = self.broadcast_stage.join();
|
|
|
|
for result in results {
|
|
|
|
result?;
|
|
|
|
}
|
|
|
|
let _ = broadcast_result?;
|
|
|
|
Ok(())
|
2018-05-14 16:36:19 -07:00
|
|
|
}
|
|
|
|
}
|