1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
use std::convert::{TryFrom, TryInto}; use anyhow::{anyhow, Error}; pub const HEADER_SIZE: usize = 8; pub const UNREACHABLE: u8 = 1; pub const TIME_EXCEEDED: u8 = 3; pub const ECHO_REQUEST: u8 = 128; pub const ECHO_REPLY: u8 = 129; #[derive(Debug)] pub enum IcmpV6Packet<'a> { Unreachable(Unreachable<'a>), EchoRequest(Echo<'a>), EchoReply(Echo<'a>), HopLimitExceeded(&'a [u8]), ReassemblyTimeExceeded(&'a [u8]), Other(u8, u8, &'a [u8]), } #[derive(Debug)] pub struct Echo<'a> { pub id: u16, pub seq: u16, pub data: &'a [u8] } #[derive(Debug)] pub enum Unreachable<'a> { Address(&'a [u8]), Port(&'a [u8]), Other(u8, &'a [u8]), } impl<'a> TryFrom<&'a [u8]> for IcmpV6Packet<'a> { type Error = Error; fn try_from(slice: &'a [u8]) -> Result<Self, Self::Error> { if slice.len() < HEADER_SIZE { return Err(anyhow!("invalid slice")); } let kind = slice[0]; let code = slice[1]; let rest = &slice[4..]; Ok(match (kind, code) { (UNREACHABLE, _) => IcmpV6Packet::Unreachable((code, rest).try_into()?), (TIME_EXCEEDED, 0) => IcmpV6Packet::HopLimitExceeded(&rest[4..]), (TIME_EXCEEDED, 1) => IcmpV6Packet::ReassemblyTimeExceeded(&rest[4..]), (ECHO_REQUEST, 0) => IcmpV6Packet::EchoRequest(rest.try_into()?), (ECHO_REPLY, 0) => IcmpV6Packet::EchoReply(rest.try_into()?), _ => IcmpV6Packet::Other(kind, code, rest), }) } } impl<'a> TryFrom<&'a [u8]> for Echo<'a> { type Error = Error; fn try_from(slice: &'a [u8]) -> Result<Self, Self::Error> { Ok(Self { id: u16::from_be_bytes(slice[0..2].try_into()?), seq: u16::from_be_bytes(slice[2..4].try_into()?), data: &slice[4..] }) } } impl<'a> TryFrom<(u8, &'a [u8])> for Unreachable<'a> { type Error = Error; fn try_from((code, slice): (u8, &'a [u8])) -> Result<Self, Self::Error> { let data = &slice[4..]; Ok(match code { 3 => Unreachable::Address(data), 4 => Unreachable::Port(data), c => Unreachable::Other(c, data), }) } }