2019-01-31 15:51:29 -08:00
|
|
|
use crate::cluster_info::{ClusterInfo, GOSSIP_SLEEP_MILLIS};
|
2019-04-09 22:06:32 -07:00
|
|
|
use crate::poh_recorder::PohRecorder;
|
2019-01-31 15:51:29 -08:00
|
|
|
use crate::result::Result;
|
|
|
|
use crate::service::Service;
|
2019-04-09 12:57:12 -07:00
|
|
|
use crate::sigverify_stage::VerifiedPackets;
|
|
|
|
use crate::{packet, sigverify};
|
2019-05-17 07:00:06 -07:00
|
|
|
use solana_metrics::inc_new_counter_debug;
|
2019-02-13 20:04:20 -08:00
|
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
2019-04-09 12:57:12 -07:00
|
|
|
use std::sync::mpsc::Sender;
|
2019-04-09 22:06:32 -07:00
|
|
|
use std::sync::{Arc, Mutex, RwLock};
|
2019-01-31 15:51:29 -08:00
|
|
|
use std::thread::{self, sleep, Builder, JoinHandle};
|
|
|
|
use std::time::Duration;
|
|
|
|
|
|
|
|
pub struct ClusterInfoVoteListener {
|
|
|
|
thread_hdls: Vec<JoinHandle<()>>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ClusterInfoVoteListener {
|
|
|
|
pub fn new(
|
2019-03-04 19:53:50 -08:00
|
|
|
exit: &Arc<AtomicBool>,
|
2019-01-31 15:51:29 -08:00
|
|
|
cluster_info: Arc<RwLock<ClusterInfo>>,
|
2019-04-09 12:57:12 -07:00
|
|
|
sigverify_disabled: bool,
|
|
|
|
sender: Sender<VerifiedPackets>,
|
2019-04-09 22:06:32 -07:00
|
|
|
poh_recorder: &Arc<Mutex<PohRecorder>>,
|
2019-01-31 15:51:29 -08:00
|
|
|
) -> Self {
|
2019-03-04 19:53:50 -08:00
|
|
|
let exit = exit.clone();
|
2019-04-09 22:06:32 -07:00
|
|
|
let poh_recorder = poh_recorder.clone();
|
2019-01-31 15:51:29 -08:00
|
|
|
let thread = Builder::new()
|
|
|
|
.name("solana-cluster_info_vote_listener".to_string())
|
|
|
|
.spawn(move || {
|
2019-04-09 22:06:32 -07:00
|
|
|
let _ = Self::recv_loop(
|
|
|
|
exit,
|
|
|
|
&cluster_info,
|
|
|
|
sigverify_disabled,
|
|
|
|
&sender,
|
|
|
|
poh_recorder,
|
|
|
|
);
|
2019-01-31 15:51:29 -08:00
|
|
|
})
|
|
|
|
.unwrap();
|
|
|
|
Self {
|
|
|
|
thread_hdls: vec![thread],
|
|
|
|
}
|
|
|
|
}
|
|
|
|
fn recv_loop(
|
2019-03-04 19:53:50 -08:00
|
|
|
exit: Arc<AtomicBool>,
|
2019-01-31 15:51:29 -08:00
|
|
|
cluster_info: &Arc<RwLock<ClusterInfo>>,
|
2019-04-09 12:57:12 -07:00
|
|
|
sigverify_disabled: bool,
|
|
|
|
sender: &Sender<VerifiedPackets>,
|
2019-04-09 22:06:32 -07:00
|
|
|
poh_recorder: Arc<Mutex<PohRecorder>>,
|
2019-01-31 15:51:29 -08:00
|
|
|
) -> Result<()> {
|
|
|
|
let mut last_ts = 0;
|
|
|
|
loop {
|
|
|
|
if exit.load(Ordering::Relaxed) {
|
|
|
|
return Ok(());
|
|
|
|
}
|
|
|
|
let (votes, new_ts) = cluster_info.read().unwrap().get_votes(last_ts);
|
2019-04-09 22:06:32 -07:00
|
|
|
if poh_recorder.lock().unwrap().bank().is_some() {
|
|
|
|
last_ts = new_ts;
|
2019-05-17 07:00:06 -07:00
|
|
|
inc_new_counter_debug!("cluster_info_vote_listener-recv_count", votes.len());
|
2019-04-09 22:06:32 -07:00
|
|
|
let msgs = packet::to_packets(&votes);
|
2019-04-26 17:27:31 -07:00
|
|
|
if !msgs.is_empty() {
|
|
|
|
let r = if sigverify_disabled {
|
|
|
|
sigverify::ed25519_verify_disabled(&msgs)
|
|
|
|
} else {
|
|
|
|
sigverify::ed25519_verify_cpu(&msgs)
|
|
|
|
};
|
|
|
|
sender.send(msgs.into_iter().zip(r).collect())?;
|
|
|
|
}
|
2019-04-09 22:06:32 -07:00
|
|
|
}
|
2019-01-31 15:51:29 -08:00
|
|
|
sleep(Duration::from_millis(GOSSIP_SLEEP_MILLIS));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Service for ClusterInfoVoteListener {
|
|
|
|
type JoinReturnType = ();
|
|
|
|
|
|
|
|
fn join(self) -> thread::Result<()> {
|
|
|
|
for thread_hdl in self.thread_hdls {
|
|
|
|
thread_hdl.join()?;
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|