NaxdyOrg/embassy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Simplify UDP code

Browse source 
Drop unneeded APIs: remove impls of embedded_io error traits, remove
flush() and split() methods.
This commit is contained in:
Artur Kowalski 2022-08-11 08:23:18 +02:00
parent 865a91976c
commit b97983242d
1 changed files with 5 additions and 59 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

64
embassy-net/src/udp.rs
View file

@ -30,14 +30,6 @@ pub struct UdpSocket<'a> {
io: UdpIo<'a>,
}
pub struct UdpReader<'a> {
io: UdpIo<'a>,
}
pub struct UdpWriter<'a> {
io: UdpIo<'a>,
}
impl<'a> UdpSocket<'a> {
pub fn new<D: Device>(
stack: &'a Stack<D>,
@ -66,10 +58,6 @@ impl<'a> UdpSocket<'a> {
}
}
pub fn split(&mut self) -> (UdpReader<'_>, UdpWriter<'_>) {
(UdpReader { io: self.io }, UdpWriter { io: self.io })
}
pub fn bind<T>(&mut self, endpoint: T) -> Result<(), BindError>
where
T: Into<IpListenEndpoint>,
@ -90,21 +78,17 @@ impl<'a> UdpSocket<'a> {
}
}
pub async fn send_to<T>(&mut self, buf: &[u8], remote_endpoint: T) -> Result<(), Error>
pub async fn send_to<T>(&self, buf: &[u8], remote_endpoint: T) -> Result<(), Error>
where
T: Into<IpEndpoint>,
{
self.io.write(buf, remote_endpoint.into()).await
}
pub async fn recv_from(&mut self, buf: &mut [u8]) -> Result<(usize, IpEndpoint), Error> {
pub async fn recv_from(&self, buf: &mut [u8]) -> Result<(usize, IpEndpoint), Error> {
self.io.read(buf).await
}
pub async fn flush(&mut self) -> Result<(), Error> {
self.io.flush().await
}
pub fn endpoint(&self) -> IpListenEndpoint {
unsafe { self.io.with(|s, _| s.endpoint()) }
}
@ -149,7 +133,7 @@ impl UdpIo<'_> {
}
/// SAFETY: must not call reentrantly.
unsafe fn with_mut<R>(&mut self, f: impl FnOnce(&mut udp::Socket, &mut Interface) -> R) -> R {
unsafe fn with_mut<R>(&self, f: impl FnOnce(&mut udp::Socket, &mut Interface) -> R) -> R {
let s = &mut *self.stack.get();
let socket = s.sockets.get_mut::<udp::Socket>(self.handle);
let res = f(socket, &mut s.iface);
@ -157,7 +141,7 @@ impl UdpIo<'_> {
res
}
async fn read(&mut self, buf: &mut [u8]) -> Result<(usize, IpEndpoint), Error> {
async fn read(&self, buf: &mut [u8]) -> Result<(usize, IpEndpoint), Error> {
poll_fn(move |cx| unsafe {
self.with_mut(|s, _| match s.recv_slice(buf) {
Ok(x) => Poll::Ready(Ok(x)),
@ -172,7 +156,7 @@ impl UdpIo<'_> {
.await
}
async fn write(&mut self, buf: &[u8], ep: IpEndpoint) -> Result<(), Error> {
async fn write(&self, buf: &[u8], ep: IpEndpoint) -> Result<(), Error> {
poll_fn(move |cx| unsafe {
self.with_mut(|s, _| match s.send_slice(buf, ep) {
// Entire datagram has been sent
@ -186,42 +170,4 @@ impl UdpIo<'_> {
})
.await
}
async fn flush(&mut self) -> Result<(), Error> {
poll_fn(move |_| {
Poll::Ready(Ok(())) // TODO: Is there a better implementation for this?
})
.await
}
}
impl UdpReader<'_> {
pub async fn recv_from(&mut self, buf: &mut [u8]) -> Result<(usize, IpEndpoint), Error> {
self.io.read(buf).await
}
}
impl UdpWriter<'_> {
pub async fn send_to<T>(&mut self, buf: &[u8], remote_endpoint: T) -> Result<(), Error>
where
T: Into<IpEndpoint>,
{
self.io.write(buf, remote_endpoint.into()).await
}
pub async fn flush(&mut self) -> Result<(), Error> {
self.io.flush().await
}
}
impl embedded_io::Error for BindError {
fn kind(&self) -> embedded_io::ErrorKind {
embedded_io::ErrorKind::Other
}
}
impl embedded_io::Error for Error {
fn kind(&self) -> embedded_io::ErrorKind {
embedded_io::ErrorKind::Other
}
}