#![allow(clippy::cast_lossless)]
use std::borrow::Cow;
use std::io::{Result as IoResult, Write};
use byteorder_slice::byteorder::WriteBytesExt;
use byteorder_slice::result::ReadSlice;
use byteorder_slice::ByteOrder;
use derive_into_owned::IntoOwned;
use super::block_common::{Block, PcapNgBlock};
use super::opt_common::{CustomBinaryOption, CustomUtf8Option, PcapNgOption, UnknownOption, WriteOptTo};
use crate::errors::PcapError;
use crate::DataLink;
#[derive(Clone, Debug, IntoOwned, Eq, PartialEq)]
pub struct InterfaceDescriptionBlock<'a> {
pub linktype: DataLink,
pub snaplen: u32,
pub options: Vec<InterfaceDescriptionOption<'a>>,
}
impl<'a> PcapNgBlock<'a> for InterfaceDescriptionBlock<'a> {
fn from_slice<B: ByteOrder>(mut slice: &'a [u8]) -> Result<(&'a [u8], Self), PcapError> {
if slice.len() < 8 {
return Err(PcapError::InvalidField("InterfaceDescriptionBlock: block length < 8"));
}
let linktype = (slice.read_u16::<B>().unwrap() as u32).into();
let reserved = slice.read_u16::<B>().unwrap();
if reserved != 0 {
return Err(PcapError::InvalidField("InterfaceDescriptionBlock: reserved != 0"));
}
let snaplen = slice.read_u32::<B>().unwrap();
let (slice, options) = InterfaceDescriptionOption::opts_from_slice::<B>(slice)?;
let block = InterfaceDescriptionBlock { linktype, snaplen, options };
Ok((slice, block))
}
fn write_to<B: ByteOrder, W: Write>(&self, writer: &mut W) -> IoResult<usize> {
writer.write_u16::<B>(u32::from(self.linktype) as u16)?;
writer.write_u16::<B>(0)?;
writer.write_u32::<B>(self.snaplen)?;
let opt_len = InterfaceDescriptionOption::write_opts_to::<B, W>(&self.options, writer)?;
Ok(8 + opt_len)
}
fn into_block(self) -> Block<'a> {
Block::InterfaceDescription(self)
}
}
impl InterfaceDescriptionBlock<'static> {
pub fn new(linktype: DataLink, snaplen: u32) -> Self {
Self { linktype, snaplen, options: vec![] }
}
}
#[derive(Clone, Debug, IntoOwned, Eq, PartialEq)]
pub enum InterfaceDescriptionOption<'a> {
Comment(Cow<'a, str>),
IfName(Cow<'a, str>),
IfDescription(Cow<'a, str>),
IfIpv4Addr(Cow<'a, [u8]>),
IfIpv6Addr(Cow<'a, [u8]>),
IfMacAddr(Cow<'a, [u8]>),
IfEuIAddr(u64),
IfSpeed(u64),
IfTsResol(u8),
IfTzone(u32),
IfFilter(Cow<'a, [u8]>),
IfOs(Cow<'a, str>),
IfFcsLen(u8),
IfTsOffset(u64),
IfHardware(Cow<'a, str>),
CustomBinary(CustomBinaryOption<'a>),
CustomUtf8(CustomUtf8Option<'a>),
Unknown(UnknownOption<'a>),
}
impl<'a> PcapNgOption<'a> for InterfaceDescriptionOption<'a> {
fn from_slice<B: ByteOrder>(code: u16, length: u16, mut slice: &'a [u8]) -> Result<Self, PcapError> {
let opt = match code {
1 => InterfaceDescriptionOption::Comment(Cow::Borrowed(std::str::from_utf8(slice)?)),
2 => InterfaceDescriptionOption::IfName(Cow::Borrowed(std::str::from_utf8(slice)?)),
3 => InterfaceDescriptionOption::IfDescription(Cow::Borrowed(std::str::from_utf8(slice)?)),
4 => {
if slice.len() != 8 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfIpv4Addr length != 8"));
}
InterfaceDescriptionOption::IfIpv4Addr(Cow::Borrowed(slice))
},
5 => {
if slice.len() != 17 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfIpv6Addr length != 17"));
}
InterfaceDescriptionOption::IfIpv6Addr(Cow::Borrowed(slice))
},
6 => {
if slice.len() != 6 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfMacAddr length != 6"));
}
InterfaceDescriptionOption::IfMacAddr(Cow::Borrowed(slice))
},
7 => {
if slice.len() != 8 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfEuIAddr length != 8"));
}
InterfaceDescriptionOption::IfEuIAddr(slice.read_u64::<B>().map_err(|_| PcapError::IncompleteBuffer)?)
},
8 => {
if slice.len() != 8 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfSpeed length != 8"));
}
InterfaceDescriptionOption::IfSpeed(slice.read_u64::<B>().map_err(|_| PcapError::IncompleteBuffer)?)
},
9 => {
if slice.len() != 1 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfTsResol length != 1"));
}
InterfaceDescriptionOption::IfTsResol(slice.read_u8().map_err(|_| PcapError::IncompleteBuffer)?)
},
10 => {
if slice.len() != 1 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfTzone length != 1"));
}
InterfaceDescriptionOption::IfTzone(slice.read_u32::<B>().map_err(|_| PcapError::IncompleteBuffer)?)
},
11 => {
if slice.is_empty() {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfFilter is empty"));
}
InterfaceDescriptionOption::IfFilter(Cow::Borrowed(slice))
},
12 => InterfaceDescriptionOption::IfOs(Cow::Borrowed(std::str::from_utf8(slice)?)),
13 => {
if slice.len() != 1 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfFcsLen length != 1"));
}
InterfaceDescriptionOption::IfFcsLen(slice.read_u8().map_err(|_| PcapError::IncompleteBuffer)?)
},
14 => {
if slice.len() != 8 {
return Err(PcapError::InvalidField("InterfaceDescriptionOption: IfTsOffset length != 8"));
}
InterfaceDescriptionOption::IfTsOffset(slice.read_u64::<B>().map_err(|_| PcapError::IncompleteBuffer)?)
},
15 => InterfaceDescriptionOption::IfHardware(Cow::Borrowed(std::str::from_utf8(slice)?)),
2988 | 19372 => InterfaceDescriptionOption::CustomUtf8(CustomUtf8Option::from_slice::<B>(code, slice)?),
2989 | 19373 => InterfaceDescriptionOption::CustomBinary(CustomBinaryOption::from_slice::<B>(code, slice)?),
_ => InterfaceDescriptionOption::Unknown(UnknownOption::new(code, length, slice)),
};
Ok(opt)
}
fn write_to<B: ByteOrder, W: Write>(&self, writer: &mut W) -> IoResult<usize> {
match self {
InterfaceDescriptionOption::Comment(a) => a.write_opt_to::<B, W>(1, writer),
InterfaceDescriptionOption::IfName(a) => a.write_opt_to::<B, W>(2, writer),
InterfaceDescriptionOption::IfDescription(a) => a.write_opt_to::<B, W>(3, writer),
InterfaceDescriptionOption::IfIpv4Addr(a) => a.write_opt_to::<B, W>(4, writer),
InterfaceDescriptionOption::IfIpv6Addr(a) => a.write_opt_to::<B, W>(5, writer),
InterfaceDescriptionOption::IfMacAddr(a) => a.write_opt_to::<B, W>(6, writer),
InterfaceDescriptionOption::IfEuIAddr(a) => a.write_opt_to::<B, W>(7, writer),
InterfaceDescriptionOption::IfSpeed(a) => a.write_opt_to::<B, W>(8, writer),
InterfaceDescriptionOption::IfTsResol(a) => a.write_opt_to::<B, W>(9, writer),
InterfaceDescriptionOption::IfTzone(a) => a.write_opt_to::<B, W>(10, writer),
InterfaceDescriptionOption::IfFilter(a) => a.write_opt_to::<B, W>(11, writer),
InterfaceDescriptionOption::IfOs(a) => a.write_opt_to::<B, W>(12, writer),
InterfaceDescriptionOption::IfFcsLen(a) => a.write_opt_to::<B, W>(13, writer),
InterfaceDescriptionOption::IfTsOffset(a) => a.write_opt_to::<B, W>(14, writer),
InterfaceDescriptionOption::IfHardware(a) => a.write_opt_to::<B, W>(15, writer),
InterfaceDescriptionOption::CustomBinary(a) => a.write_opt_to::<B, W>(a.code, writer),
InterfaceDescriptionOption::CustomUtf8(a) => a.write_opt_to::<B, W>(a.code, writer),
InterfaceDescriptionOption::Unknown(a) => a.write_opt_to::<B, W>(a.code, writer),
}
}
}