2021-05-05 01:52:41 +08:00
|
|
|
use minimq::embedded_nal;
|
2021-05-05 21:39:33 +08:00
|
|
|
use shared_bus::{AtomicCheckMutex, BusMutex};
|
2021-05-05 01:52:41 +08:00
|
|
|
|
|
|
|
use crate::hardware::NetworkStack;
|
|
|
|
|
|
|
|
pub struct NetworkStackProxy<'a, S> {
|
2021-05-05 21:39:33 +08:00
|
|
|
mutex: &'a AtomicCheckMutex<S>,
|
2021-05-05 01:52:41 +08:00
|
|
|
}
|
|
|
|
|
2021-05-05 21:39:33 +08:00
|
|
|
impl<'a, S> NetworkStackProxy<'a, S> {
|
|
|
|
pub fn lock<R, F: FnOnce(&mut S) -> R>(&mut self, f: F) -> R {
|
|
|
|
self.mutex.lock(|stack| f(stack))
|
2021-05-05 01:52:41 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
macro_rules! forward {
|
|
|
|
($func:ident($($v:ident: $IT:ty),*) -> $T:ty) => {
|
|
|
|
fn $func(&self, $($v: $IT),*) -> $T {
|
|
|
|
self.mutex.lock(|stack| stack.$func($($v),*))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a, S> embedded_nal::TcpStack for NetworkStackProxy<'a, S>
|
|
|
|
where
|
2021-05-05 21:39:33 +08:00
|
|
|
S: embedded_nal::TcpStack,
|
2021-05-05 01:52:41 +08:00
|
|
|
{
|
|
|
|
type TcpSocket = S::TcpSocket;
|
|
|
|
type Error = S::Error;
|
|
|
|
|
|
|
|
forward! {open(mode: embedded_nal::Mode) -> Result<S::TcpSocket, S::Error>}
|
|
|
|
forward! {connect(socket: S::TcpSocket, remote: embedded_nal::SocketAddr) -> Result<S::TcpSocket, S::Error>}
|
|
|
|
forward! {is_connected(socket: &S::TcpSocket) -> Result<bool, S::Error>}
|
|
|
|
forward! {write(socket: &mut S::TcpSocket, buffer: &[u8]) -> embedded_nal::nb::Result<usize, S::Error>}
|
|
|
|
forward! {read(socket: &mut S::TcpSocket, buffer: &mut [u8]) -> embedded_nal::nb::Result<usize, S::Error>}
|
|
|
|
forward! {close(socket: S::TcpSocket) -> Result<(), S::Error>}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub struct NetworkManager {
|
2021-05-05 21:39:33 +08:00
|
|
|
mutex: AtomicCheckMutex<NetworkStack>,
|
2021-05-05 01:52:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
impl NetworkManager {
|
|
|
|
pub fn new(stack: NetworkStack) -> Self {
|
2021-05-05 21:39:33 +08:00
|
|
|
Self {
|
|
|
|
mutex: AtomicCheckMutex::create(stack),
|
|
|
|
}
|
2021-05-05 01:52:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn acquire_stack<'a>(&'a self) -> NetworkStackProxy<'a, NetworkStack> {
|
2021-05-05 21:39:33 +08:00
|
|
|
NetworkStackProxy { mutex: &self.mutex }
|
2021-05-05 01:52:41 +08:00
|
|
|
}
|
|
|
|
}
|