use core::mem::MaybeUninit; use smoltcp::{ iface::EthernetInterface, socket::{SocketSet, SocketHandle, TcpSocket, TcpSocketBuffer, SocketRef}, time::Instant, wire::{IpAddress, IpCidr, Ipv4Address, Ipv4Cidr}, }; use crate::command_parser::Ipv4Config; use crate::net::split_ipv4_config; pub struct SocketState { handle: SocketHandle, state: S, } /// Number of server sockets and therefore concurrent client /// sessions. Many data structures in `Server::run()` correspond to /// this const. const SOCKET_COUNT: usize = 4; const TCP_RX_BUFFER_SIZE: usize = 2048; const TCP_TX_BUFFER_SIZE: usize = 2048; /// Contains a number of server sockets that get all sent the same /// data (through `fmt::Write`). pub struct Server<'a, 'b, S> { net: EthernetInterface<'a, 'a, 'a, &'a mut stm32_eth::Eth<'static, 'static>>, sockets: SocketSet<'b, 'b, 'b>, states: [SocketState; SOCKET_COUNT], } impl<'a, 'b, S: Default> Server<'a, 'b, S> { /// Run a server with stack-allocated sockets pub fn run(net: EthernetInterface<'a, 'a, 'a, &'a mut stm32_eth::Eth<'static, 'static>>, f: F) where F: FnOnce(&mut Server<'a, '_, S>), { let mut sockets_storage: [_; SOCKET_COUNT] = Default::default(); let mut sockets = SocketSet::new(&mut sockets_storage[..]); let mut states: [SocketState; SOCKET_COUNT] = unsafe { MaybeUninit::uninit().assume_init() }; macro_rules! create_socket { ($set:ident, $rx_storage:ident, $tx_storage:ident, $target:expr) => { let mut $rx_storage = [0; TCP_RX_BUFFER_SIZE]; let mut $tx_storage = [0; TCP_TX_BUFFER_SIZE]; let tcp_rx_buffer = TcpSocketBuffer::new(&mut $rx_storage[..]); let tcp_tx_buffer = TcpSocketBuffer::new(&mut $tx_storage[..]); let tcp_socket = TcpSocket::new(tcp_rx_buffer, tcp_tx_buffer); $target = $set.add(tcp_socket); } } create_socket!(sockets, tcp_rx_storage0, tcp_tx_storage0, states[0].handle); create_socket!(sockets, tcp_rx_storage1, tcp_tx_storage1, states[1].handle); create_socket!(sockets, tcp_rx_storage2, tcp_tx_storage2, states[2].handle); create_socket!(sockets, tcp_rx_storage3, tcp_tx_storage3, states[3].handle); for state in &mut states { state.state = S::default(); } let mut server = Server { states, sockets, net, }; f(&mut server); } /// Poll the interface and the sockets pub fn poll(&mut self, now: Instant) -> Result<(), smoltcp::Error> { // Poll smoltcp EthernetInterface, // pass only unexpected smoltcp errors to the caller match self.net.poll(&mut self.sockets, now) { Ok(_) => Ok(()), Err(smoltcp::Error::Malformed) => Ok(()), Err(smoltcp::Error::Unrecognized) => Ok(()), Err(e) => Err(e), } } /// Iterate over all sockets managed by this server pub fn for_each, &mut S)>(&mut self, mut callback: F) { for state in &mut self.states { let socket = self.sockets.get::(state.handle); callback(socket, &mut state.state); } } fn set_ipv4_address(&mut self, ipv4_address: Ipv4Cidr) { self.net.update_ip_addrs(|addrs| { for addr in addrs.iter_mut() { match addr { IpCidr::Ipv4(_) => { *addr = IpCidr::Ipv4(ipv4_address); // done break } _ => { // skip } } } }); } fn set_gateway(&mut self, gateway: Option) { let routes = self.net.routes_mut(); match gateway { None => routes.update(|routes_storage| { routes_storage.remove(&IpCidr::new(IpAddress::v4(0, 0, 0, 0), 0)); }), Some(gateway) => { routes.add_default_ipv4_route(gateway).unwrap(); } } } pub fn set_ipv4_config(&mut self, config: Ipv4Config) { let (address, gateway) = split_ipv4_config(config); self.set_ipv4_address(address); self.set_gateway(gateway); } }