1 files changed, 92 insertions, 0 deletions

diff --git a/embassy-net/src/packet_pool.rs b/embassy-net/src/packet_pool.rs
new file mode 100644
index 000000000..2c27d4013
--- /dev/null
+++ b/embassy-net/src/packet_pool.rs
@@ -0,0 +1,92 @@
+use as_slice::{AsMutSlice, AsSlice};
+use core::ops::{Deref, DerefMut, Range};
+
+use atomic_pool::{pool, Box};
+
+pub const MTU: usize = 1514;
+pub const PACKET_POOL_SIZE: usize = 4;
+
+pool!(pub PacketPool: [Packet; PACKET_POOL_SIZE]);
+pub type PacketBox = Box<PacketPool>;
+
+pub struct Packet(pub [u8; MTU]);
+
+impl Packet {
+    pub const fn new() -> Self {
+        Self([0; MTU])
+    }
+}
+
+pub trait PacketBoxExt {
+    fn slice(self, range: Range<usize>) -> PacketBuf;
+}
+
+impl PacketBoxExt for PacketBox {
+    fn slice(self, range: Range<usize>) -> PacketBuf {
+        PacketBuf {
+            packet: self,
+            range,
+        }
+    }
+}
+
+impl AsSlice for Packet {
+    type Element = u8;
+
+    fn as_slice(&self) -> &[Self::Element] {
+        &self.deref()[..]
+    }
+}
+
+impl AsMutSlice for Packet {
+    fn as_mut_slice(&mut self) -> &mut [Self::Element] {
+        &mut self.deref_mut()[..]
+    }
+}
+
+impl Deref for Packet {
+    type Target = [u8; MTU];
+
+    fn deref(&self) -> &[u8; MTU] {
+        &self.0
+    }
+}
+
+impl DerefMut for Packet {
+    fn deref_mut(&mut self) -> &mut [u8; MTU] {
+        &mut self.0
+    }
+}
+
+pub struct PacketBuf {
+    packet: PacketBox,
+    range: Range<usize>,
+}
+
+impl AsSlice for PacketBuf {
+    type Element = u8;
+
+    fn as_slice(&self) -> &[Self::Element] {
+        &self.packet[self.range.clone()]
+    }
+}
+
+impl AsMutSlice for PacketBuf {
+    fn as_mut_slice(&mut self) -> &mut [Self::Element] {
+        &mut self.packet[self.range.clone()]
+    }
+}
+
+impl Deref for PacketBuf {
+    type Target = [u8];
+
+    fn deref(&self) -> &[u8] {
+        &self.packet[self.range.clone()]
+    }
+}
+
+impl DerefMut for PacketBuf {
+    fn deref_mut(&mut self) -> &mut [u8] {
+        &mut self.packet[self.range.clone()]
+    }
+}