use std::fmt;
use crate::error::*;
use crate::packet::{Packet as P, PacketMut as PM, AsPacket, AsPacketMut};
use crate::size;
use crate::ip;
use crate::icmp::Kind;
pub struct Packet<B> {
buffer: B,
}
sized!(Packet,
header {
min: 8,
max: 8,
size: 8,
}
payload {
min: <ip::v4::Packet<()> as size::header::Min>::min(),
max: <ip::v4::Packet<()> as size::header::Max>::max(),
size: p => {
if let Ok(ip) = p.packet() {
size::header::Size::size(&ip)
}
else {
p.buffer.as_ref().len() - 8
}
},
});
impl<B: AsRef<[u8]>> fmt::Debug for Packet<B> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("icmp::parameter_problem::Packet")
.field("pointer", &self.pointer())
.field("packet", &self.packet())
.finish()
}
}
impl<B: AsRef<[u8]>> Packet<B> {
pub fn unchecked(buffer: B) -> Packet<B> {
Packet { buffer }
}
pub fn new(buffer: B) -> Result<Packet<B>> {
use crate::size::header::Min;
let packet = Packet::unchecked(buffer);
if packet.buffer.as_ref().len() < Self::min() {
Err(Error::SmallBuffer)?
}
match Kind::from(packet.buffer.as_ref()[0]) {
Kind::ParameterProblem =>
(),
_ =>
Err(Error::InvalidPacket)?
}
Ok(packet)
}
}
impl<B: AsRef<[u8]>> Packet<B> {
pub fn to_owned(&self) -> Packet<Vec<u8>> {
Packet::unchecked(self.buffer.as_ref().to_vec())
}
}
impl<B: AsRef<[u8]>> AsRef<[u8]> for Packet<B> {
fn as_ref(&self) -> &[u8] {
use crate::size::Size;
&self.buffer.as_ref()[.. self.size()]
}
}
impl<B: AsRef<[u8]> + AsMut<[u8]>> AsMut<[u8]> for Packet<B> {
fn as_mut(&mut self) -> &mut [u8] {
use crate::size::Size;
let size = self.size();
&mut self.buffer.as_mut()[.. size]
}
}
impl<'a, B: AsRef<[u8]>> AsPacket<'a, Packet<&'a [u8]>> for B {
fn as_packet(&self) -> Result<Packet<&[u8]>> {
Packet::new(self.as_ref())
}
}
impl<'a, B: AsRef<[u8]> + AsMut<[u8]>> AsPacketMut<'a, Packet<&'a mut [u8]>> for B {
fn as_packet_mut(&mut self) -> Result<Packet<&mut [u8]>> {
Packet::new(self.as_mut())
}
}
impl<B: AsRef<[u8]>> P for Packet<B> {
fn split(&self) -> (&[u8], &[u8]) {
let (header, payload) = self.buffer.as_ref().split_at(8);
(&header[ .. 5], payload)
}
}
impl<B: AsRef<[u8]> + AsMut<[u8]>> PM for Packet<B> {
fn split_mut(&mut self) -> (&mut [u8], &mut [u8]) {
let (header, payload) = self.buffer.as_mut().split_at_mut(8);
(&mut header[.. 5], payload)
}
}
impl<B: AsRef<[u8]>> Packet<B> {
pub fn pointer(&self) -> u8 {
self.buffer.as_ref()[4]
}
pub fn packet(&self) -> Result<ip::v4::Packet<&[u8]>> {
ip::v4::Packet::new(&self.buffer.as_ref()[8 ..])
}
}