89 lines
1.6 KiB
Rust
89 lines
1.6 KiB
Rust
|
use as_slice::{AsMutSlice, AsSlice};
|
||
|
|
use core::ops::{Deref, DerefMut, Range};
|
||
|
|
|
||
|
|
use super::pool::{BitPool, Box, StaticPool};
|
||
|
|
|
||
|
|
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])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Box<PacketPool> {
|
||
|
|
pub 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()]
|
||
|
|
}
|
||
|
|
}
|