packet_hasher.rs

// Get a unique hash value for a packet
// Used in retransmit and shred fetch to prevent dos with same packet data.

use {
    ahash::AHasher,
    rand::{thread_rng, Rng},
    solana_ledger::shred::Shred,
    solana_perf::packet::Packet,
    std::hash::Hasher,
};

#[derive(Clone)]
pub struct PacketHasher {
    seed1: u128,
    seed2: u128,
}

impl Default for PacketHasher {
    fn default() -> Self {
        Self {
            seed1: thread_rng().gen::<u128>(),
            seed2: thread_rng().gen::<u128>(),
        }
    }
}

impl PacketHasher {
    pub(crate) fn hash_packet(&self, packet: &Packet) -> u64 {
        let size = packet.data.len().min(packet.meta.size);
        self.hash_data(&packet.data[..size])
    }

    pub(crate) fn hash_shred(&self, shred: &Shred) -> u64 {
        self.hash_data(&shred.payload)
    }

    fn hash_data(&self, data: &[u8]) -> u64 {
        let mut hasher = AHasher::new_with_keys(self.seed1, self.seed2);
        hasher.write(data);
        hasher.finish()
    }

    pub fn reset(&mut self) {
        *self = Self::default();
    }
}