2021-12-03 09:00:31 -08:00
|
|
|
use {
|
|
|
|
bincode::Result,
|
|
|
|
serde::Serialize,
|
|
|
|
std::{
|
|
|
|
fmt, io,
|
2022-01-02 08:13:57 -08:00
|
|
|
net::{IpAddr, Ipv4Addr, SocketAddr},
|
2021-12-03 09:00:31 -08:00
|
|
|
},
|
2020-01-28 16:11:22 -08:00
|
|
|
};
|
2019-10-17 10:37:30 -07:00
|
|
|
|
2018-11-16 08:04:46 -08:00
|
|
|
/// Maximum over-the-wire size of a Transaction
|
2019-05-21 21:45:38 -07:00
|
|
|
/// 1280 is IPv6 minimum MTU
|
|
|
|
/// 40 bytes is the size of the IPv6 header
|
|
|
|
/// 8 bytes is the size of the fragment header
|
|
|
|
pub const PACKET_DATA_SIZE: usize = 1280 - 40 - 8;
|
2019-10-17 10:37:30 -07:00
|
|
|
|
2022-01-02 08:13:57 -08:00
|
|
|
#[derive(Clone, Debug, PartialEq)]
|
2019-10-17 10:37:30 -07:00
|
|
|
#[repr(C)]
|
|
|
|
pub struct Meta {
|
|
|
|
pub size: usize,
|
2021-12-29 10:34:31 -08:00
|
|
|
pub forwarded: bool,
|
2019-10-17 10:37:30 -07:00
|
|
|
pub repair: bool,
|
2019-10-17 16:26:29 -07:00
|
|
|
pub discard: bool,
|
2022-01-02 08:13:57 -08:00
|
|
|
pub addr: IpAddr,
|
2019-10-17 10:37:30 -07:00
|
|
|
pub port: u16,
|
2021-03-08 19:31:00 -08:00
|
|
|
pub is_tracer_tx: bool,
|
2021-09-20 21:00:33 -07:00
|
|
|
pub is_simple_vote_tx: bool,
|
2019-10-17 10:37:30 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
#[repr(C)]
|
|
|
|
pub struct Packet {
|
|
|
|
pub data: [u8; PACKET_DATA_SIZE],
|
|
|
|
pub meta: Meta,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Packet {
|
|
|
|
pub fn new(data: [u8; PACKET_DATA_SIZE], meta: Meta) -> Self {
|
|
|
|
Self { data, meta }
|
|
|
|
}
|
2019-11-14 10:24:53 -08:00
|
|
|
|
2021-04-12 23:28:08 -07:00
|
|
|
pub fn from_data<T: Serialize>(dest: Option<&SocketAddr>, data: T) -> Result<Self> {
|
2020-11-08 07:10:03 -08:00
|
|
|
let mut packet = Packet::default();
|
2021-04-12 23:28:08 -07:00
|
|
|
Self::populate_packet(&mut packet, dest, &data)?;
|
2020-11-08 07:10:03 -08:00
|
|
|
Ok(packet)
|
2019-11-14 10:24:53 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn populate_packet<T: Serialize>(
|
|
|
|
packet: &mut Packet,
|
|
|
|
dest: Option<&SocketAddr>,
|
|
|
|
data: &T,
|
|
|
|
) -> Result<()> {
|
|
|
|
let mut wr = io::Cursor::new(&mut packet.data[..]);
|
|
|
|
bincode::serialize_into(&mut wr, data)?;
|
|
|
|
let len = wr.position() as usize;
|
|
|
|
packet.meta.size = len;
|
|
|
|
if let Some(dest) = dest {
|
|
|
|
packet.meta.set_addr(dest);
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
2019-10-17 10:37:30 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Debug for Packet {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
|
|
write!(
|
|
|
|
f,
|
|
|
|
"Packet {{ size: {:?}, addr: {:?} }}",
|
|
|
|
self.meta.size,
|
|
|
|
self.meta.addr()
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-05-19 07:31:47 -07:00
|
|
|
#[allow(clippy::uninit_assumed_init)]
|
2019-10-17 10:37:30 -07:00
|
|
|
impl Default for Packet {
|
|
|
|
fn default() -> Packet {
|
|
|
|
Packet {
|
|
|
|
data: unsafe { std::mem::MaybeUninit::uninit().assume_init() },
|
|
|
|
meta: Meta::default(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl PartialEq for Packet {
|
|
|
|
fn eq(&self, other: &Packet) -> bool {
|
|
|
|
let self_data: &[u8] = self.data.as_ref();
|
|
|
|
let other_data: &[u8] = other.data.as_ref();
|
2020-12-13 17:26:34 -08:00
|
|
|
self.meta == other.meta && self_data[..self.meta.size] == other_data[..self.meta.size]
|
2019-10-17 10:37:30 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Meta {
|
|
|
|
pub fn addr(&self) -> SocketAddr {
|
2022-01-02 08:13:57 -08:00
|
|
|
SocketAddr::new(self.addr, self.port)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn set_addr(&mut self, socket_addr: &SocketAddr) {
|
|
|
|
self.addr = socket_addr.ip();
|
|
|
|
self.port = socket_addr.port();
|
2019-10-17 10:37:30 -07:00
|
|
|
}
|
2022-01-02 08:13:57 -08:00
|
|
|
}
|
2019-10-17 10:37:30 -07:00
|
|
|
|
2022-01-02 08:13:57 -08:00
|
|
|
impl Default for Meta {
|
|
|
|
fn default() -> Self {
|
|
|
|
Self {
|
|
|
|
size: 0,
|
|
|
|
forwarded: false,
|
|
|
|
repair: false,
|
|
|
|
discard: false,
|
|
|
|
addr: IpAddr::V4(Ipv4Addr::UNSPECIFIED),
|
|
|
|
port: 0,
|
|
|
|
is_tracer_tx: false,
|
|
|
|
is_simple_vote_tx: false,
|
2019-10-17 10:37:30 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|