use crate::PrimitiveValues;
use crate::ethernet::EtherType;
use alloc::vec::Vec;
use pnet_macros::packet;
use pnet_macros_support::types::*;
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct ClassOfService(pub u3);
impl ClassOfService {
pub fn new(value: u3) -> ClassOfService {
ClassOfService(value)
}
}
impl PrimitiveValues for ClassOfService {
type T = (u3,);
fn to_primitive_values(&self) -> (u3,) {
(self.0,)
}
}
#[allow(non_snake_case)]
#[allow(non_upper_case_globals)]
pub mod ClassesOfService {
use super::ClassOfService;
pub const BK: ClassOfService = ClassOfService(1);
pub const BE: ClassOfService = ClassOfService(0);
pub const EE: ClassOfService = ClassOfService(2);
pub const CA: ClassOfService = ClassOfService(3);
pub const VI: ClassOfService = ClassOfService(4);
pub const VO: ClassOfService = ClassOfService(5);
pub const IC: ClassOfService = ClassOfService(6);
pub const NC: ClassOfService = ClassOfService(7);
}
#[packet]
pub struct Vlan {
#[construct_with(u3)]
pub priority_code_point: ClassOfService,
pub drop_eligible_indicator: u1,
pub vlan_identifier: u12be,
#[construct_with(u16be)]
pub ethertype: EtherType,
#[payload]
pub payload: Vec<u8>,
}
#[cfg(test)]
mod tests {
use crate::ethernet::EtherTypes;
use super::*;
#[test]
fn vlan_packet_test() {
let mut packet = [0u8; 4];
{
let mut vlan_header = MutableVlanPacket::new(&mut packet[..]).unwrap();
vlan_header.set_priority_code_point(ClassesOfService::BE);
assert_eq!(vlan_header.get_priority_code_point(), ClassesOfService::BE);
vlan_header.set_drop_eligible_indicator(0);
assert_eq!(vlan_header.get_drop_eligible_indicator(), 0);
vlan_header.set_ethertype(EtherTypes::Ipv4);
assert_eq!(vlan_header.get_ethertype(), EtherTypes::Ipv4);
vlan_header.set_vlan_identifier(0x100);
assert_eq!(vlan_header.get_vlan_identifier(), 0x100);
}
let ref_packet = [0x01, 0x00, 0x08, 0x00]; assert_eq!(&ref_packet[..], &packet[..]);
}
}