use crate::{ metrics::NetworkMetrics, types::{Cid, Frame, Mid, Pid, Sid}, }; use async_std::{ net::{TcpStream, UdpSocket}, prelude::*, sync::RwLock, }; use futures::{ channel::{mpsc, oneshot}, future::FutureExt, select, sink::SinkExt, stream::StreamExt, }; use std::{net::SocketAddr, sync::Arc}; use tracing::*; // Reserving bytes 0, 10, 13 as i have enough space and want to make it easy to // detect a invalid client, e.g. sending an empty line would make 10 first char // const FRAME_RESERVED_1: u8 = 0; const FRAME_HANDSHAKE: u8 = 1; const FRAME_PARTICIPANT_ID: u8 = 2; const FRAME_SHUTDOWN: u8 = 3; const FRAME_OPEN_STREAM: u8 = 4; const FRAME_CLOSE_STREAM: u8 = 5; const FRAME_DATA_HEADER: u8 = 6; const FRAME_DATA: u8 = 7; const FRAME_RAW: u8 = 8; //const FRAME_RESERVED_2: u8 = 10; //const FRAME_RESERVED_3: u8 = 13; #[derive(Debug)] pub(crate) enum Protocols { Tcp(TcpProtocol), Udp(UdpProtocol), //Mpsc(MpscChannel), } #[derive(Debug)] pub(crate) struct TcpProtocol { stream: TcpStream, metrics: Arc, } #[derive(Debug)] pub(crate) struct UdpProtocol { socket: Arc, remote_addr: SocketAddr, metrics: Arc, data_in: RwLock>>, } impl TcpProtocol { pub(crate) fn new(stream: TcpStream, metrics: Arc) -> Self { Self { stream, metrics } } pub async fn read( &self, cid: Cid, mut from_wire_sender: mpsc::UnboundedSender<(Cid, Frame)>, end_receiver: oneshot::Receiver<()>, ) { trace!("starting up tcp write()"); let mut stream = self.stream.clone(); let mut end_receiver = end_receiver.fuse(); loop { let mut bytes = [0u8; 1]; let r = select! { r = stream.read_exact(&mut bytes).fuse() => r, _ = end_receiver => break, }; if r.is_err() { info!("tcp stream closed, shutting down read"); break; } let frame_no = bytes[0]; let frame = match frame_no { FRAME_HANDSHAKE => { let mut bytes = [0u8; 19]; stream.read_exact(&mut bytes).await.unwrap(); let magic_number = [ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], ]; Frame::Handshake { magic_number, version: [ u32::from_le_bytes([bytes[7], bytes[8], bytes[9], bytes[10]]), u32::from_le_bytes([bytes[11], bytes[12], bytes[13], bytes[14]]), u32::from_le_bytes([bytes[15], bytes[16], bytes[17], bytes[18]]), ], } }, FRAME_PARTICIPANT_ID => { let mut bytes = [0u8; 16]; stream.read_exact(&mut bytes).await.unwrap(); let pid = Pid::from_le_bytes(bytes); Frame::ParticipantId { pid } }, FRAME_SHUTDOWN => Frame::Shutdown, FRAME_OPEN_STREAM => { let mut bytes = [0u8; 10]; stream.read_exact(&mut bytes).await.unwrap(); let sid = Sid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); let prio = bytes[8]; let promises = bytes[9]; Frame::OpenStream { sid, prio, promises, } }, FRAME_CLOSE_STREAM => { let mut bytes = [0u8; 8]; stream.read_exact(&mut bytes).await.unwrap(); let sid = Sid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); Frame::CloseStream { sid } }, FRAME_DATA_HEADER => { let mut bytes = [0u8; 24]; stream.read_exact(&mut bytes).await.unwrap(); let mid = Mid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); let sid = Sid::from_le_bytes([ bytes[8], bytes[9], bytes[10], bytes[11], bytes[12], bytes[13], bytes[14], bytes[15], ]); let length = u64::from_le_bytes([ bytes[16], bytes[17], bytes[18], bytes[19], bytes[20], bytes[21], bytes[22], bytes[23], ]); Frame::DataHeader { mid, sid, length } }, FRAME_DATA => { let mut bytes = [0u8; 18]; stream.read_exact(&mut bytes).await.unwrap(); let mid = Mid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); let start = u64::from_le_bytes([ bytes[8], bytes[9], bytes[10], bytes[11], bytes[12], bytes[13], bytes[14], bytes[15], ]); let length = u16::from_le_bytes([bytes[16], bytes[17]]); let mut data = vec![0; length as usize]; stream.read_exact(&mut data).await.unwrap(); Frame::Data { mid, start, data } }, FRAME_RAW => { let mut bytes = [0u8; 2]; stream.read_exact(&mut bytes).await.unwrap(); let length = u16::from_le_bytes([bytes[0], bytes[1]]); let mut data = vec![0; length as usize]; stream.read_exact(&mut data).await.unwrap(); Frame::Raw(data) }, _ => { // report a RAW frame, but cannot rely on the next 2 bytes to be a size. // guessing 256 bytes, which might help to sort down issues let mut data = vec![0; 256]; stream.read(&mut data).await.unwrap(); Frame::Raw(data) }, }; self.metrics .frames_wire_in_total .with_label_values(&[&cid.to_string(), frame.get_string()]) .inc(); from_wire_sender.send((cid, frame)).await.unwrap(); } trace!("shutting down tcp read()"); } //dezerialize here as this is executed in a seperate thread PER channel. // Limites Throughput per single Receiver but stays in same thread (maybe as its // in a threadpool) for TCP, UDP and MPSC pub async fn write(&self, cid: Cid, mut to_wire_receiver: mpsc::UnboundedReceiver) { trace!("starting up tcp write()"); let mut stream = self.stream.clone(); let cid_string = cid.to_string(); while let Some(frame) = to_wire_receiver.next().await { self.metrics .frames_wire_out_total .with_label_values(&[&cid_string, frame.get_string()]) .inc(); match frame { Frame::Handshake { magic_number, version, } => { stream .write_all(&FRAME_HANDSHAKE.to_be_bytes()) .await .unwrap(); stream.write_all(&magic_number).await.unwrap(); stream.write_all(&version[0].to_le_bytes()).await.unwrap(); stream.write_all(&version[1].to_le_bytes()).await.unwrap(); stream.write_all(&version[2].to_le_bytes()).await.unwrap(); }, Frame::ParticipantId { pid } => { stream .write_all(&FRAME_PARTICIPANT_ID.to_be_bytes()) .await .unwrap(); stream.write_all(&pid.to_le_bytes()).await.unwrap(); }, Frame::Shutdown => { stream .write_all(&FRAME_SHUTDOWN.to_be_bytes()) .await .unwrap(); }, Frame::OpenStream { sid, prio, promises, } => { stream .write_all(&FRAME_OPEN_STREAM.to_be_bytes()) .await .unwrap(); stream.write_all(&sid.to_le_bytes()).await.unwrap(); stream.write_all(&prio.to_le_bytes()).await.unwrap(); stream.write_all(&promises.to_le_bytes()).await.unwrap(); }, Frame::CloseStream { sid } => { stream .write_all(&FRAME_CLOSE_STREAM.to_be_bytes()) .await .unwrap(); stream.write_all(&sid.to_le_bytes()).await.unwrap(); }, Frame::DataHeader { mid, sid, length } => { stream .write_all(&FRAME_DATA_HEADER.to_be_bytes()) .await .unwrap(); stream.write_all(&mid.to_le_bytes()).await.unwrap(); stream.write_all(&sid.to_le_bytes()).await.unwrap(); stream.write_all(&length.to_le_bytes()).await.unwrap(); }, Frame::Data { mid, start, data } => { stream.write_all(&FRAME_DATA.to_be_bytes()).await.unwrap(); stream.write_all(&mid.to_le_bytes()).await.unwrap(); stream.write_all(&start.to_le_bytes()).await.unwrap(); stream .write_all(&(data.len() as u16).to_le_bytes()) .await .unwrap(); stream.write_all(&data).await.unwrap(); }, Frame::Raw(data) => { stream.write_all(&FRAME_RAW.to_be_bytes()).await.unwrap(); stream .write_all(&(data.len() as u16).to_le_bytes()) .await .unwrap(); stream.write_all(&data).await.unwrap(); }, } } trace!("shutting down tcp write()"); } } impl UdpProtocol { pub(crate) fn new( socket: Arc, remote_addr: SocketAddr, metrics: Arc, data_in: mpsc::UnboundedReceiver>, ) -> Self { Self { socket, remote_addr, metrics, data_in: RwLock::new(data_in), } } pub async fn read( &self, cid: Cid, mut from_wire_sender: mpsc::UnboundedSender<(Cid, Frame)>, end_receiver: oneshot::Receiver<()>, ) { trace!("starting up udp read()"); let mut data_in = self.data_in.write().await; let mut end_receiver = end_receiver.fuse(); while let Some(bytes) = select! { r = data_in.next().fuse() => r, _ = end_receiver => None, } { trace!("got raw UDP message with len: {}", bytes.len()); let frame_no = bytes[0]; let frame = match frame_no { FRAME_HANDSHAKE => { let bytes = &bytes[1..20]; let magic_number = [ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], ]; Frame::Handshake { magic_number, version: [ u32::from_le_bytes([bytes[7], bytes[8], bytes[9], bytes[10]]), u32::from_le_bytes([bytes[11], bytes[12], bytes[13], bytes[14]]), u32::from_le_bytes([bytes[15], bytes[16], bytes[17], bytes[18]]), ], } }, FRAME_PARTICIPANT_ID => { let pid = Pid::from_le_bytes([ bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], bytes[8], bytes[9], bytes[10], bytes[11], bytes[12], bytes[13], bytes[14], bytes[15], bytes[16], ]); Frame::ParticipantId { pid } }, FRAME_SHUTDOWN => Frame::Shutdown, FRAME_OPEN_STREAM => { let bytes = &bytes[1..11]; let sid = Sid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); let prio = bytes[8]; let promises = bytes[9]; Frame::OpenStream { sid, prio, promises, } }, FRAME_CLOSE_STREAM => { let bytes = &bytes[1..9]; let sid = Sid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); Frame::CloseStream { sid } }, FRAME_DATA_HEADER => { let bytes = &bytes[1..25]; let mid = Mid::from_le_bytes([ bytes[0], bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], ]); let sid = Sid::from_le_bytes([ bytes[8], bytes[9], bytes[10], bytes[11], bytes[12], bytes[13], bytes[14], bytes[15], ]); let length = u64::from_le_bytes([ bytes[16], bytes[17], bytes[18], bytes[19], bytes[20], bytes[21], bytes[22], bytes[23], ]); Frame::DataHeader { mid, sid, length } }, FRAME_DATA => { let mid = Mid::from_le_bytes([ bytes[1], bytes[2], bytes[3], bytes[4], bytes[5], bytes[6], bytes[7], bytes[8], ]); let start = u64::from_le_bytes([ bytes[9], bytes[10], bytes[11], bytes[12], bytes[13], bytes[14], bytes[15], bytes[16], ]); let length = u16::from_le_bytes([bytes[17], bytes[18]]); let mut data = vec![0; length as usize]; data.copy_from_slice(&bytes[19..]); Frame::Data { mid, start, data } }, FRAME_RAW => { let length = u16::from_le_bytes([bytes[1], bytes[2]]); let mut data = vec![0; length as usize]; data.copy_from_slice(&bytes[3..]); Frame::Raw(data) }, _ => Frame::Raw(bytes), }; self.metrics .frames_wire_in_total .with_label_values(&[&cid.to_string(), frame.get_string()]) .inc(); from_wire_sender.send((cid, frame)).await.unwrap(); } trace!("shutting down udp read()"); } pub async fn write(&self, cid: Cid, mut to_wire_receiver: mpsc::UnboundedReceiver) { trace!("starting up udp write()"); let mut buffer = [0u8; 2000]; let cid_string = cid.to_string(); while let Some(frame) = to_wire_receiver.next().await { self.metrics .frames_wire_out_total .with_label_values(&[&cid_string, frame.get_string()]) .inc(); let len = match frame { Frame::Handshake { magic_number, version, } => { let x = FRAME_HANDSHAKE.to_be_bytes(); buffer[0] = x[0]; buffer[1] = magic_number[0]; buffer[2] = magic_number[1]; buffer[3] = magic_number[2]; buffer[4] = magic_number[3]; buffer[5] = magic_number[4]; buffer[6] = magic_number[5]; buffer[7] = magic_number[6]; let x = version[0].to_le_bytes(); buffer[8] = x[0]; buffer[9] = x[1]; buffer[10] = x[2]; buffer[11] = x[3]; let x = version[1].to_le_bytes(); buffer[12] = x[0]; buffer[13] = x[1]; buffer[14] = x[2]; buffer[15] = x[3]; let x = version[2].to_le_bytes(); buffer[16] = x[0]; buffer[17] = x[1]; buffer[18] = x[2]; buffer[19] = x[3]; 20 }, Frame::ParticipantId { pid } => { let x = FRAME_PARTICIPANT_ID.to_be_bytes(); buffer[0] = x[0]; let x = pid.to_le_bytes(); buffer[1] = x[0]; buffer[2] = x[1]; buffer[3] = x[2]; buffer[4] = x[3]; buffer[5] = x[4]; buffer[6] = x[5]; buffer[7] = x[6]; buffer[8] = x[7]; buffer[9] = x[8]; buffer[10] = x[9]; buffer[11] = x[10]; buffer[12] = x[11]; buffer[13] = x[12]; buffer[14] = x[13]; buffer[15] = x[14]; buffer[16] = x[15]; 17 }, Frame::Shutdown => { let x = FRAME_SHUTDOWN.to_be_bytes(); buffer[0] = x[0]; 1 }, Frame::OpenStream { sid, prio, promises, } => { let x = FRAME_OPEN_STREAM.to_be_bytes(); buffer[0] = x[0]; let x = sid.to_le_bytes(); buffer[1] = x[0]; buffer[2] = x[1]; buffer[3] = x[2]; buffer[4] = x[3]; buffer[5] = x[4]; buffer[6] = x[5]; buffer[7] = x[6]; buffer[8] = x[7]; let x = prio.to_le_bytes(); buffer[9] = x[0]; let x = promises.to_le_bytes(); buffer[10] = x[0]; 11 }, Frame::CloseStream { sid } => { let x = FRAME_CLOSE_STREAM.to_be_bytes(); buffer[0] = x[0]; let x = sid.to_le_bytes(); buffer[1] = x[0]; buffer[2] = x[1]; buffer[3] = x[2]; buffer[4] = x[3]; buffer[5] = x[4]; buffer[6] = x[5]; buffer[7] = x[6]; buffer[8] = x[7]; 9 }, Frame::DataHeader { mid, sid, length } => { let x = FRAME_DATA_HEADER.to_be_bytes(); buffer[0] = x[0]; let x = mid.to_le_bytes(); buffer[1] = x[0]; buffer[2] = x[1]; buffer[3] = x[2]; buffer[4] = x[3]; buffer[5] = x[4]; buffer[6] = x[5]; buffer[7] = x[6]; buffer[8] = x[7]; let x = sid.to_le_bytes(); buffer[9] = x[0]; buffer[10] = x[1]; buffer[11] = x[2]; buffer[12] = x[3]; buffer[13] = x[4]; buffer[14] = x[5]; buffer[15] = x[6]; buffer[16] = x[7]; let x = length.to_le_bytes(); buffer[17] = x[0]; buffer[18] = x[1]; buffer[19] = x[2]; buffer[20] = x[3]; buffer[21] = x[4]; buffer[22] = x[5]; buffer[23] = x[6]; buffer[24] = x[7]; 25 }, Frame::Data { mid, start, data } => { let x = FRAME_DATA.to_be_bytes(); buffer[0] = x[0]; let x = mid.to_le_bytes(); buffer[1] = x[0]; buffer[2] = x[1]; buffer[3] = x[2]; buffer[4] = x[3]; buffer[5] = x[4]; buffer[6] = x[5]; buffer[7] = x[6]; buffer[8] = x[7]; let x = start.to_le_bytes(); buffer[9] = x[0]; buffer[10] = x[1]; buffer[11] = x[2]; buffer[12] = x[3]; buffer[13] = x[4]; buffer[14] = x[5]; buffer[15] = x[6]; buffer[16] = x[7]; let x = (data.len() as u16).to_le_bytes(); buffer[17] = x[0]; buffer[18] = x[1]; for i in 0..data.len() { buffer[19 + i] = data[i]; } 19 + data.len() }, Frame::Raw(data) => { let x = FRAME_RAW.to_be_bytes(); buffer[0] = x[0]; let x = (data.len() as u16).to_le_bytes(); buffer[1] = x[0]; buffer[2] = x[1]; for i in 0..data.len() { buffer[3 + i] = data[i]; } 3 + data.len() }, }; let mut start = 0; while start < len { trace!(?start, ?len, "splitting up udp frame in multiple packages"); match self .socket .send_to(&buffer[start..len], self.remote_addr) .await { Ok(n) => { start += n; if n != len { error!( "THIS DOESNT WORK, as RECEIVER CURRENLTY ONLY HANDLES 1 FRAME per \ UDP message. splitting up will fail!" ); } }, Err(e) => error!(?e, "need to handle that error!"), } } } trace!("shutting down udp write()"); } }