SaiTLS/src/set.rs

use smoltcp as net;

use managed::ManagedSlice;
use crate::tls::TlsSocket;
use net::socket::SocketSet;
use net::phy::Device;
use net::iface::EthernetInterface;
use net::time::Instant;

pub struct TlsSocketSet<'a, 'b, 'c> {
    tls_sockets: ManagedSlice<'a, Option<TlsSocket<'a, 'b, 'c>>>
}

#[derive(Clone, Copy, Debug)]
pub struct TlsSocketHandle(usize);

impl<'a, 'b, 'c> TlsSocketSet<'a, 'b, 'c> {
    pub fn new<T>(tls_sockets: T) -> Self
    where
        T: Into<ManagedSlice<'a, Option<TlsSocket<'a, 'b, 'c>>>>
    {
        Self {
            tls_sockets: tls_sockets.into()
        }
    }

    pub fn add(&mut self, socket: TlsSocket<'a, 'b, 'c>) -> TlsSocketHandle
    {
        for (index, slot) in self.tls_sockets.iter_mut().enumerate() {
            if slot.is_none() {
                *slot = Some(socket);
                return TlsSocketHandle(index);
            }
        }

        match self.tls_sockets {
            ManagedSlice::Borrowed(_) => {
                panic!("adding a socket to a full array")
            }

            ManagedSlice::Owned(ref mut sockets) => {
                sockets.push(Some(socket));
                let index = sockets.len() - 1;
                return TlsSocketHandle(index);
            }
        }
    }

    pub fn get(&mut self, handle: TlsSocketHandle) -> &mut TlsSocket<'a, 'b, 'c> {
        self.tls_sockets[handle.0].as_mut().unwrap()
    }

    pub(crate) fn polled_by<DeviceT>(
        &mut self,
        sockets: &mut SocketSet,
        iface: &mut EthernetInterface<DeviceT>,
        now: Instant
    ) -> smoltcp::Result<bool>
    where
        DeviceT: for<'d> Device<'d>
    {
        for socket in self.tls_sockets.iter_mut() {
            if socket.is_some() {
                socket.as_mut()
                    .unwrap()
                    .update_handshake(iface, now)?;
            }
        }

        Ok(true)
    }
    
}