icmpv6.rs 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898
  1. use core::{cmp, fmt};
  2. use byteorder::{ByteOrder, NetworkEndian};
  3. use crate::{Error, Result};
  4. use crate::phy::ChecksumCapabilities;
  5. use crate::wire::ip::checksum;
  6. use crate::wire::{IpAddress, IpProtocol, Ipv6Packet, Ipv6Repr};
  7. use crate::wire::MldRepr;
  8. #[cfg(feature = "medium-ethernet")]
  9. use crate::wire::NdiscRepr;
  10. enum_with_unknown! {
  11. /// Internet protocol control message type.
  12. pub enum Message(u8) {
  13. /// Destination Unreachable.
  14. DstUnreachable = 0x01,
  15. /// Packet Too Big.
  16. PktTooBig = 0x02,
  17. /// Time Exceeded.
  18. TimeExceeded = 0x03,
  19. /// Parameter Problem.
  20. ParamProblem = 0x04,
  21. /// Echo Request
  22. EchoRequest = 0x80,
  23. /// Echo Reply
  24. EchoReply = 0x81,
  25. /// Multicast Listener Query
  26. MldQuery = 0x82,
  27. /// Router Solicitation
  28. RouterSolicit = 0x85,
  29. /// Router Advertisement
  30. RouterAdvert = 0x86,
  31. /// Neighbor Solicitation
  32. NeighborSolicit = 0x87,
  33. /// Neighbor Advertisement
  34. NeighborAdvert = 0x88,
  35. /// Redirect
  36. Redirect = 0x89,
  37. /// Multicast Listener Report
  38. MldReport = 0x8f
  39. }
  40. }
  41. impl Message {
  42. /// Per [RFC 4443 § 2.1] ICMPv6 message types with the highest order
  43. /// bit set are informational messages while message types without
  44. /// the highest order bit set are error messages.
  45. ///
  46. /// [RFC 4443 § 2.1]: https://tools.ietf.org/html/rfc4443#section-2.1
  47. pub fn is_error(&self) -> bool {
  48. (u8::from(*self) & 0x80) != 0x80
  49. }
  50. /// Return a boolean value indicating if the given message type
  51. /// is an [NDISC] message type.
  52. ///
  53. /// [NDISC]: https://tools.ietf.org/html/rfc4861
  54. pub fn is_ndisc(&self) -> bool {
  55. match *self {
  56. Message::RouterSolicit | Message::RouterAdvert | Message::NeighborSolicit |
  57. Message::NeighborAdvert | Message::Redirect => true,
  58. _ => false,
  59. }
  60. }
  61. /// Return a boolean value indicating if the given message type
  62. /// is an [MLD] message type.
  63. ///
  64. /// [MLD]: https://tools.ietf.org/html/rfc3810
  65. pub fn is_mld(&self) -> bool {
  66. match *self {
  67. Message::MldQuery | Message::MldReport => true,
  68. _ => false,
  69. }
  70. }
  71. }
  72. impl fmt::Display for Message {
  73. fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
  74. match *self {
  75. Message::DstUnreachable => write!(f, "destination unreachable"),
  76. Message::PktTooBig => write!(f, "packet too big"),
  77. Message::TimeExceeded => write!(f, "time exceeded"),
  78. Message::ParamProblem => write!(f, "parameter problem"),
  79. Message::EchoReply => write!(f, "echo reply"),
  80. Message::EchoRequest => write!(f, "echo request"),
  81. Message::RouterSolicit => write!(f, "router solicitation"),
  82. Message::RouterAdvert => write!(f, "router advertisement"),
  83. Message::NeighborSolicit => write!(f, "neighbor solicitation"),
  84. Message::NeighborAdvert => write!(f, "neighbor advert"),
  85. Message::Redirect => write!(f, "redirect"),
  86. Message::MldQuery => write!(f, "multicast listener query"),
  87. Message::MldReport => write!(f, "multicast listener report"),
  88. Message::Unknown(id) => write!(f, "{}", id)
  89. }
  90. }
  91. }
  92. enum_with_unknown! {
  93. /// Internet protocol control message subtype for type "Destination Unreachable".
  94. pub enum DstUnreachable(u8) {
  95. /// No Route to destination.
  96. NoRoute = 0,
  97. /// Communication with destination administratively prohibited.
  98. AdminProhibit = 1,
  99. /// Beyond scope of source address.
  100. BeyondScope = 2,
  101. /// Address unreachable.
  102. AddrUnreachable = 3,
  103. /// Port unreachable.
  104. PortUnreachable = 4,
  105. /// Source address failed ingress/egress policy.
  106. FailedPolicy = 5,
  107. /// Reject route to destination.
  108. RejectRoute = 6
  109. }
  110. }
  111. impl fmt::Display for DstUnreachable {
  112. fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
  113. match *self {
  114. DstUnreachable::NoRoute =>
  115. write!(f, "no route to destination"),
  116. DstUnreachable::AdminProhibit =>
  117. write!(f, "communication with destination administratively prohibited"),
  118. DstUnreachable::BeyondScope =>
  119. write!(f, "beyond scope of source address"),
  120. DstUnreachable::AddrUnreachable =>
  121. write!(f, "address unreachable"),
  122. DstUnreachable::PortUnreachable =>
  123. write!(f, "port unreachable"),
  124. DstUnreachable::FailedPolicy =>
  125. write!(f, "source address failed ingress/egress policy"),
  126. DstUnreachable::RejectRoute =>
  127. write!(f, "reject route to destination"),
  128. DstUnreachable::Unknown(id) =>
  129. write!(f, "{}", id)
  130. }
  131. }
  132. }
  133. enum_with_unknown! {
  134. /// Internet protocol control message subtype for the type "Parameter Problem".
  135. pub enum ParamProblem(u8) {
  136. /// Erroneous header field encountered.
  137. ErroneousHdrField = 0,
  138. /// Unrecognized Next Header type encountered.
  139. UnrecognizedNxtHdr = 1,
  140. /// Unrecognized IPv6 option encountered.
  141. UnrecognizedOption = 2
  142. }
  143. }
  144. impl fmt::Display for ParamProblem {
  145. fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
  146. match *self {
  147. ParamProblem::ErroneousHdrField =>
  148. write!(f, "erroneous header field."),
  149. ParamProblem::UnrecognizedNxtHdr =>
  150. write!(f, "unrecognized next header type."),
  151. ParamProblem::UnrecognizedOption =>
  152. write!(f, "unrecognized IPv6 option."),
  153. ParamProblem::Unknown(id) =>
  154. write!(f, "{}", id)
  155. }
  156. }
  157. }
  158. enum_with_unknown! {
  159. /// Internet protocol control message subtype for the type "Time Exceeded".
  160. pub enum TimeExceeded(u8) {
  161. /// Hop limit exceeded in transit.
  162. HopLimitExceeded = 0,
  163. /// Fragment reassembly time exceeded.
  164. FragReassemExceeded = 1
  165. }
  166. }
  167. impl fmt::Display for TimeExceeded {
  168. fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
  169. match *self {
  170. TimeExceeded::HopLimitExceeded =>
  171. write!(f, "hop limit exceeded in transit"),
  172. TimeExceeded::FragReassemExceeded =>
  173. write!(f, "fragment reassembly time exceeded"),
  174. TimeExceeded::Unknown(id) =>
  175. write!(f, "{}", id)
  176. }
  177. }
  178. }
  179. /// A read/write wrapper around an Internet Control Message Protocol version 6 packet buffer.
  180. #[derive(Debug, PartialEq, Clone)]
  181. #[cfg_attr(feature = "defmt", derive(defmt::Format))]
  182. pub struct Packet<T: AsRef<[u8]>> {
  183. pub(super) buffer: T
  184. }
  185. // Ranges and constants describing key boundaries in the ICMPv6 header.
  186. pub(super) mod field {
  187. use crate::wire::field::*;
  188. // ICMPv6: See https://tools.ietf.org/html/rfc4443
  189. pub const TYPE: usize = 0;
  190. pub const CODE: usize = 1;
  191. pub const CHECKSUM: Field = 2..4;
  192. pub const UNUSED: Field = 4..8;
  193. pub const MTU: Field = 4..8;
  194. pub const POINTER: Field = 4..8;
  195. pub const ECHO_IDENT: Field = 4..6;
  196. pub const ECHO_SEQNO: Field = 6..8;
  197. pub const HEADER_END: usize = 8;
  198. // NDISC: See https://tools.ietf.org/html/rfc4861
  199. // Router Advertisement message offsets
  200. pub const CUR_HOP_LIMIT: usize = 4;
  201. pub const ROUTER_FLAGS: usize = 5;
  202. pub const ROUTER_LT: Field = 6..8;
  203. pub const REACHABLE_TM: Field = 8..12;
  204. pub const RETRANS_TM: Field = 12..16;
  205. // Neighbor Solicitation message offsets
  206. pub const TARGET_ADDR: Field = 8..24;
  207. // Neighbor Advertisement message offsets
  208. pub const NEIGH_FLAGS: usize = 4;
  209. // Redirected Header message offsets
  210. pub const DEST_ADDR: Field = 24..40;
  211. // MLD:
  212. // - https://tools.ietf.org/html/rfc3810
  213. // - https://tools.ietf.org/html/rfc3810
  214. // Multicast Listener Query message
  215. pub const MAX_RESP_CODE: Field = 4..6;
  216. pub const QUERY_RESV: Field = 6..8;
  217. pub const QUERY_MCAST_ADDR: Field = 8..24;
  218. pub const SQRV: usize = 24;
  219. pub const QQIC: usize = 25;
  220. pub const QUERY_NUM_SRCS: Field = 26..28;
  221. // Multicast Listener Report Message
  222. pub const RECORD_RESV: Field = 4..6;
  223. pub const NR_MCAST_RCRDS: Field = 6..8;
  224. // Multicast Address Record Offsets
  225. pub const RECORD_TYPE: usize = 0;
  226. pub const AUX_DATA_LEN: usize = 1;
  227. pub const RECORD_NUM_SRCS: Field = 2..4;
  228. pub const RECORD_MCAST_ADDR: Field = 4..20;
  229. }
  230. impl<T: AsRef<[u8]>> Packet<T> {
  231. /// Imbue a raw octet buffer with ICMPv6 packet structure.
  232. pub fn new_unchecked(buffer: T) -> Packet<T> {
  233. Packet { buffer }
  234. }
  235. /// Shorthand for a combination of [new_unchecked] and [check_len].
  236. ///
  237. /// [new_unchecked]: #method.new_unchecked
  238. /// [check_len]: #method.check_len
  239. pub fn new_checked(buffer: T) -> Result<Packet<T>> {
  240. let packet = Self::new_unchecked(buffer);
  241. packet.check_len()?;
  242. Ok(packet)
  243. }
  244. /// Ensure that no accessor method will panic if called.
  245. /// Returns `Err(Error::Truncated)` if the buffer is too short.
  246. pub fn check_len(&self) -> Result<()> {
  247. let len = self.buffer.as_ref().len();
  248. if len < field::HEADER_END || len < self.header_len() {
  249. Err(Error::Truncated)
  250. } else {
  251. Ok(())
  252. }
  253. }
  254. /// Consume the packet, returning the underlying buffer.
  255. pub fn into_inner(self) -> T {
  256. self.buffer
  257. }
  258. /// Return the message type field.
  259. #[inline]
  260. pub fn msg_type(&self) -> Message {
  261. let data = self.buffer.as_ref();
  262. Message::from(data[field::TYPE])
  263. }
  264. /// Return the message code field.
  265. #[inline]
  266. pub fn msg_code(&self) -> u8 {
  267. let data = self.buffer.as_ref();
  268. data[field::CODE]
  269. }
  270. /// Return the checksum field.
  271. #[inline]
  272. pub fn checksum(&self) -> u16 {
  273. let data = self.buffer.as_ref();
  274. NetworkEndian::read_u16(&data[field::CHECKSUM])
  275. }
  276. /// Return the identifier field (for echo request and reply packets).
  277. #[inline]
  278. pub fn echo_ident(&self) -> u16 {
  279. let data = self.buffer.as_ref();
  280. NetworkEndian::read_u16(&data[field::ECHO_IDENT])
  281. }
  282. /// Return the sequence number field (for echo request and reply packets).
  283. #[inline]
  284. pub fn echo_seq_no(&self) -> u16 {
  285. let data = self.buffer.as_ref();
  286. NetworkEndian::read_u16(&data[field::ECHO_SEQNO])
  287. }
  288. /// Return the MTU field (for packet too big messages).
  289. #[inline]
  290. pub fn pkt_too_big_mtu(&self) -> u32 {
  291. let data = self.buffer.as_ref();
  292. NetworkEndian::read_u32(&data[field::MTU])
  293. }
  294. /// Return the pointer field (for parameter problem messages).
  295. #[inline]
  296. pub fn param_problem_ptr(&self) -> u32 {
  297. let data = self.buffer.as_ref();
  298. NetworkEndian::read_u32(&data[field::POINTER])
  299. }
  300. /// Return the header length. The result depends on the value of
  301. /// the message type field.
  302. pub fn header_len(&self) -> usize {
  303. match self.msg_type() {
  304. Message::DstUnreachable => field::UNUSED.end,
  305. Message::PktTooBig => field::MTU.end,
  306. Message::TimeExceeded => field::UNUSED.end,
  307. Message::ParamProblem => field::POINTER.end,
  308. Message::EchoRequest => field::ECHO_SEQNO.end,
  309. Message::EchoReply => field::ECHO_SEQNO.end,
  310. Message::RouterSolicit => field::UNUSED.end,
  311. Message::RouterAdvert => field::RETRANS_TM.end,
  312. Message::NeighborSolicit => field::TARGET_ADDR.end,
  313. Message::NeighborAdvert => field::TARGET_ADDR.end,
  314. Message::Redirect => field::DEST_ADDR.end,
  315. Message::MldQuery => field::QUERY_NUM_SRCS.end,
  316. Message::MldReport => field::NR_MCAST_RCRDS.end,
  317. // For packets that are not included in RFC 4443, do not
  318. // include the last 32 bits of the ICMPv6 header in
  319. // `header_bytes`. This must be done so that these bytes
  320. // can be accessed in the `payload`.
  321. _ => field::CHECKSUM.end
  322. }
  323. }
  324. /// Validate the header checksum.
  325. ///
  326. /// # Fuzzing
  327. /// This function always returns `true` when fuzzing.
  328. pub fn verify_checksum(&self, src_addr: &IpAddress, dst_addr: &IpAddress) -> bool {
  329. if cfg!(fuzzing) { return true }
  330. let data = self.buffer.as_ref();
  331. checksum::combine(&[
  332. checksum::pseudo_header(src_addr, dst_addr, IpProtocol::Icmpv6,
  333. data.len() as u32),
  334. checksum::data(data)
  335. ]) == !0
  336. }
  337. }
  338. impl<'a, T: AsRef<[u8]> + ?Sized> Packet<&'a T> {
  339. /// Return a pointer to the type-specific data.
  340. #[inline]
  341. pub fn payload(&self) -> &'a [u8] {
  342. let data = self.buffer.as_ref();
  343. &data[self.header_len()..]
  344. }
  345. }
  346. impl<T: AsRef<[u8]> + AsMut<[u8]>> Packet<T> {
  347. /// Set the message type field.
  348. #[inline]
  349. pub fn set_msg_type(&mut self, value: Message) {
  350. let data = self.buffer.as_mut();
  351. data[field::TYPE] = value.into()
  352. }
  353. /// Set the message code field.
  354. #[inline]
  355. pub fn set_msg_code(&mut self, value: u8) {
  356. let data = self.buffer.as_mut();
  357. data[field::CODE] = value
  358. }
  359. /// Clear any reserved fields in the message header.
  360. ///
  361. /// # Panics
  362. /// This function panics if the message type has not been set.
  363. /// See [set_msg_type].
  364. ///
  365. /// [set_msg_type]: #method.set_msg_type
  366. #[inline]
  367. pub fn clear_reserved(&mut self) {
  368. match self.msg_type() {
  369. Message::RouterSolicit | Message::NeighborSolicit |
  370. Message::NeighborAdvert | Message::Redirect => {
  371. let data = self.buffer.as_mut();
  372. NetworkEndian::write_u32(&mut data[field::UNUSED], 0);
  373. },
  374. Message::MldQuery => {
  375. let data = self.buffer.as_mut();
  376. NetworkEndian::write_u16(&mut data[field::QUERY_RESV], 0);
  377. data[field::SQRV] &= 0xf;
  378. },
  379. Message::MldReport => {
  380. let data = self.buffer.as_mut();
  381. NetworkEndian::write_u16(&mut data[field::RECORD_RESV], 0);
  382. }
  383. ty => panic!("Message type `{}` does not have any reserved fields.", ty),
  384. }
  385. }
  386. #[inline]
  387. pub fn set_checksum(&mut self, value: u16) {
  388. let data = self.buffer.as_mut();
  389. NetworkEndian::write_u16(&mut data[field::CHECKSUM], value)
  390. }
  391. /// Set the identifier field (for echo request and reply packets).
  392. ///
  393. /// # Panics
  394. /// This function may panic if this packet is not an echo request or reply packet.
  395. #[inline]
  396. pub fn set_echo_ident(&mut self, value: u16) {
  397. let data = self.buffer.as_mut();
  398. NetworkEndian::write_u16(&mut data[field::ECHO_IDENT], value)
  399. }
  400. /// Set the sequence number field (for echo request and reply packets).
  401. ///
  402. /// # Panics
  403. /// This function may panic if this packet is not an echo request or reply packet.
  404. #[inline]
  405. pub fn set_echo_seq_no(&mut self, value: u16) {
  406. let data = self.buffer.as_mut();
  407. NetworkEndian::write_u16(&mut data[field::ECHO_SEQNO], value)
  408. }
  409. /// Set the MTU field (for packet too big messages).
  410. ///
  411. /// # Panics
  412. /// This function may panic if this packet is not an packet too big packet.
  413. #[inline]
  414. pub fn set_pkt_too_big_mtu(&mut self, value: u32) {
  415. let data = self.buffer.as_mut();
  416. NetworkEndian::write_u32(&mut data[field::MTU], value)
  417. }
  418. /// Set the pointer field (for parameter problem messages).
  419. ///
  420. /// # Panics
  421. /// This function may panic if this packet is not a parameter problem message.
  422. #[inline]
  423. pub fn set_param_problem_ptr(&mut self, value: u32) {
  424. let data = self.buffer.as_mut();
  425. NetworkEndian::write_u32(&mut data[field::POINTER], value)
  426. }
  427. /// Compute and fill in the header checksum.
  428. pub fn fill_checksum(&mut self, src_addr: &IpAddress, dst_addr: &IpAddress) {
  429. self.set_checksum(0);
  430. let checksum = {
  431. let data = self.buffer.as_ref();
  432. !checksum::combine(&[
  433. checksum::pseudo_header(src_addr, dst_addr, IpProtocol::Icmpv6,
  434. data.len() as u32),
  435. checksum::data(data)
  436. ])
  437. };
  438. self.set_checksum(checksum)
  439. }
  440. /// Return a mutable pointer to the type-specific data.
  441. #[inline]
  442. pub fn payload_mut(&mut self) -> &mut [u8] {
  443. let range = self.header_len()..;
  444. let data = self.buffer.as_mut();
  445. &mut data[range]
  446. }
  447. }
  448. impl<T: AsRef<[u8]>> AsRef<[u8]> for Packet<T> {
  449. fn as_ref(&self) -> &[u8] {
  450. self.buffer.as_ref()
  451. }
  452. }
  453. /// A high-level representation of an Internet Control Message Protocol version 6 packet header.
  454. #[derive(Debug, PartialEq, Eq, Clone, Copy)]
  455. #[cfg_attr(feature = "defmt", derive(defmt::Format))]
  456. #[non_exhaustive]
  457. pub enum Repr<'a> {
  458. DstUnreachable {
  459. reason: DstUnreachable,
  460. header: Ipv6Repr,
  461. data: &'a [u8]
  462. },
  463. PktTooBig {
  464. mtu: u32,
  465. header: Ipv6Repr,
  466. data: &'a [u8]
  467. },
  468. TimeExceeded {
  469. reason: TimeExceeded,
  470. header: Ipv6Repr,
  471. data: &'a [u8]
  472. },
  473. ParamProblem {
  474. reason: ParamProblem,
  475. pointer: u32,
  476. header: Ipv6Repr,
  477. data: &'a [u8]
  478. },
  479. EchoRequest {
  480. ident: u16,
  481. seq_no: u16,
  482. data: &'a [u8]
  483. },
  484. EchoReply {
  485. ident: u16,
  486. seq_no: u16,
  487. data: &'a [u8]
  488. },
  489. #[cfg(feature = "medium-ethernet")]
  490. Ndisc(NdiscRepr<'a>),
  491. Mld(MldRepr<'a>),
  492. }
  493. impl<'a> Repr<'a> {
  494. /// Parse an Internet Control Message Protocol version 6 packet and return
  495. /// a high-level representation.
  496. pub fn parse<T>(src_addr: &IpAddress, dst_addr: &IpAddress,
  497. packet: &Packet<&'a T>, checksum_caps: &ChecksumCapabilities)
  498. -> Result<Repr<'a>>
  499. where T: AsRef<[u8]> + ?Sized {
  500. fn create_packet_from_payload<'a, T>(packet: &Packet<&'a T>)
  501. -> Result<(&'a [u8], Ipv6Repr)>
  502. where T: AsRef<[u8]> + ?Sized {
  503. let ip_packet = Ipv6Packet::new_checked(packet.payload())?;
  504. let payload = &packet.payload()[ip_packet.header_len() as usize..];
  505. if payload.len() < 8 { return Err(Error::Truncated) }
  506. let repr = Ipv6Repr {
  507. src_addr: ip_packet.src_addr(),
  508. dst_addr: ip_packet.dst_addr(),
  509. next_header: ip_packet.next_header(),
  510. payload_len: payload.len(),
  511. hop_limit: ip_packet.hop_limit()
  512. };
  513. Ok((payload, repr))
  514. }
  515. // Valid checksum is expected.
  516. if checksum_caps.icmpv6.rx() && !packet.verify_checksum(src_addr, dst_addr) {
  517. return Err(Error::Checksum)
  518. }
  519. match (packet.msg_type(), packet.msg_code()) {
  520. (Message::DstUnreachable, code) => {
  521. let (payload, repr) = create_packet_from_payload(packet)?;
  522. Ok(Repr::DstUnreachable {
  523. reason: DstUnreachable::from(code),
  524. header: repr,
  525. data: payload
  526. })
  527. },
  528. (Message::PktTooBig, 0) => {
  529. let (payload, repr) = create_packet_from_payload(packet)?;
  530. Ok(Repr::PktTooBig {
  531. mtu: packet.pkt_too_big_mtu(),
  532. header: repr,
  533. data: payload
  534. })
  535. },
  536. (Message::TimeExceeded, code) => {
  537. let (payload, repr) = create_packet_from_payload(packet)?;
  538. Ok(Repr::TimeExceeded {
  539. reason: TimeExceeded::from(code),
  540. header: repr,
  541. data: payload
  542. })
  543. },
  544. (Message::ParamProblem, code) => {
  545. let (payload, repr) = create_packet_from_payload(packet)?;
  546. Ok(Repr::ParamProblem {
  547. reason: ParamProblem::from(code),
  548. pointer: packet.param_problem_ptr(),
  549. header: repr,
  550. data: payload
  551. })
  552. },
  553. (Message::EchoRequest, 0) => {
  554. Ok(Repr::EchoRequest {
  555. ident: packet.echo_ident(),
  556. seq_no: packet.echo_seq_no(),
  557. data: packet.payload()
  558. })
  559. },
  560. (Message::EchoReply, 0) => {
  561. Ok(Repr::EchoReply {
  562. ident: packet.echo_ident(),
  563. seq_no: packet.echo_seq_no(),
  564. data: packet.payload()
  565. })
  566. },
  567. #[cfg(feature = "medium-ethernet")]
  568. (msg_type, 0) if msg_type.is_ndisc() => {
  569. NdiscRepr::parse(packet).map(Repr::Ndisc)
  570. },
  571. (msg_type, 0) if msg_type.is_mld() => {
  572. MldRepr::parse(packet).map(Repr::Mld)
  573. },
  574. _ => Err(Error::Unrecognized)
  575. }
  576. }
  577. /// Return the length of a packet that will be emitted from this high-level representation.
  578. pub fn buffer_len(&self) -> usize {
  579. match self {
  580. &Repr::DstUnreachable { header, data, .. } | &Repr::PktTooBig { header, data, .. } |
  581. &Repr::TimeExceeded { header, data, .. } | &Repr::ParamProblem { header, data, .. } => {
  582. field::UNUSED.end + header.buffer_len() + data.len()
  583. }
  584. &Repr::EchoRequest { data, .. } |
  585. &Repr::EchoReply { data, .. } => {
  586. field::ECHO_SEQNO.end + data.len()
  587. },
  588. #[cfg(feature = "medium-ethernet")]
  589. &Repr::Ndisc(ndisc) => {
  590. ndisc.buffer_len()
  591. },
  592. &Repr::Mld(mld) => {
  593. mld.buffer_len()
  594. },
  595. }
  596. }
  597. /// Emit a high-level representation into an Internet Control Message Protocol version 6
  598. /// packet.
  599. pub fn emit<T>(&self, src_addr: &IpAddress, dst_addr: &IpAddress,
  600. packet: &mut Packet<&mut T>, checksum_caps: &ChecksumCapabilities)
  601. where T: AsRef<[u8]> + AsMut<[u8]> + ?Sized {
  602. fn emit_contained_packet(buffer: &mut [u8], header: Ipv6Repr, data: &[u8]) {
  603. let mut ip_packet = Ipv6Packet::new_unchecked(buffer);
  604. header.emit(&mut ip_packet);
  605. let payload = &mut ip_packet.into_inner()[header.buffer_len()..];
  606. payload.copy_from_slice(&data[..]);
  607. }
  608. match *self {
  609. Repr::DstUnreachable { reason, header, data } => {
  610. packet.set_msg_type(Message::DstUnreachable);
  611. packet.set_msg_code(reason.into());
  612. emit_contained_packet(packet.payload_mut(), header, &data);
  613. },
  614. Repr::PktTooBig { mtu, header, data } => {
  615. packet.set_msg_type(Message::PktTooBig);
  616. packet.set_msg_code(0);
  617. packet.set_pkt_too_big_mtu(mtu);
  618. emit_contained_packet(packet.payload_mut(), header, &data);
  619. },
  620. Repr::TimeExceeded { reason, header, data } => {
  621. packet.set_msg_type(Message::TimeExceeded);
  622. packet.set_msg_code(reason.into());
  623. emit_contained_packet(packet.payload_mut(), header, &data);
  624. },
  625. Repr::ParamProblem { reason, pointer, header, data } => {
  626. packet.set_msg_type(Message::ParamProblem);
  627. packet.set_msg_code(reason.into());
  628. packet.set_param_problem_ptr(pointer);
  629. emit_contained_packet(packet.payload_mut(), header, &data);
  630. },
  631. Repr::EchoRequest { ident, seq_no, data } => {
  632. packet.set_msg_type(Message::EchoRequest);
  633. packet.set_msg_code(0);
  634. packet.set_echo_ident(ident);
  635. packet.set_echo_seq_no(seq_no);
  636. let data_len = cmp::min(packet.payload_mut().len(), data.len());
  637. packet.payload_mut()[..data_len].copy_from_slice(&data[..data_len])
  638. },
  639. Repr::EchoReply { ident, seq_no, data } => {
  640. packet.set_msg_type(Message::EchoReply);
  641. packet.set_msg_code(0);
  642. packet.set_echo_ident(ident);
  643. packet.set_echo_seq_no(seq_no);
  644. let data_len = cmp::min(packet.payload_mut().len(), data.len());
  645. packet.payload_mut()[..data_len].copy_from_slice(&data[..data_len])
  646. },
  647. #[cfg(feature = "medium-ethernet")]
  648. Repr::Ndisc(ndisc) => {
  649. ndisc.emit(packet)
  650. },
  651. Repr::Mld(mld) => {
  652. mld.emit(packet)
  653. },
  654. }
  655. if checksum_caps.icmpv6.tx() {
  656. packet.fill_checksum(src_addr, dst_addr);
  657. } else {
  658. // make sure we get a consistently zeroed checksum, since implementations might rely on it
  659. packet.set_checksum(0);
  660. }
  661. }
  662. }
  663. #[cfg(test)]
  664. mod test {
  665. use crate::wire::{Ipv6Address, Ipv6Repr, IpProtocol};
  666. use crate::wire::ip::test::{MOCK_IP_ADDR_1, MOCK_IP_ADDR_2};
  667. use super::*;
  668. static ECHO_PACKET_BYTES: [u8; 12] =
  669. [0x80, 0x00, 0x19, 0xb3,
  670. 0x12, 0x34, 0xab, 0xcd,
  671. 0xaa, 0x00, 0x00, 0xff];
  672. static ECHO_PACKET_PAYLOAD: [u8; 4] =
  673. [0xaa, 0x00, 0x00, 0xff];
  674. static PKT_TOO_BIG_BYTES: [u8; 60] =
  675. [0x02, 0x00, 0x0f, 0xc9,
  676. 0x00, 0x00, 0x05, 0xdc,
  677. 0x60, 0x00, 0x00, 0x00,
  678. 0x00, 0x0c, 0x11, 0x40,
  679. 0xfe, 0x80, 0x00, 0x00,
  680. 0x00, 0x00, 0x00, 0x00,
  681. 0x00, 0x00, 0x00, 0x00,
  682. 0x00, 0x00, 0x00, 0x01,
  683. 0xfe, 0x80, 0x00, 0x00,
  684. 0x00, 0x00, 0x00, 0x00,
  685. 0x00, 0x00, 0x00, 0x00,
  686. 0x00, 0x00, 0x00, 0x02,
  687. 0xbf, 0x00, 0x00, 0x35,
  688. 0x00, 0x0c, 0x12, 0x4d,
  689. 0xaa, 0x00, 0x00, 0xff];
  690. static PKT_TOO_BIG_IP_PAYLOAD: [u8; 52] =
  691. [0x60, 0x00, 0x00, 0x00,
  692. 0x00, 0x0c, 0x11, 0x40,
  693. 0xfe, 0x80, 0x00, 0x00,
  694. 0x00, 0x00, 0x00, 0x00,
  695. 0x00, 0x00, 0x00, 0x00,
  696. 0x00, 0x00, 0x00, 0x01,
  697. 0xfe, 0x80, 0x00, 0x00,
  698. 0x00, 0x00, 0x00, 0x00,
  699. 0x00, 0x00, 0x00, 0x00,
  700. 0x00, 0x00, 0x00, 0x02,
  701. 0xbf, 0x00, 0x00, 0x35,
  702. 0x00, 0x0c, 0x12, 0x4d,
  703. 0xaa, 0x00, 0x00, 0xff];
  704. static PKT_TOO_BIG_UDP_PAYLOAD: [u8; 12] =
  705. [0xbf, 0x00, 0x00, 0x35,
  706. 0x00, 0x0c, 0x12, 0x4d,
  707. 0xaa, 0x00, 0x00, 0xff];
  708. fn echo_packet_repr() -> Repr<'static> {
  709. Repr::EchoRequest {
  710. ident: 0x1234,
  711. seq_no: 0xabcd,
  712. data: &ECHO_PACKET_PAYLOAD
  713. }
  714. }
  715. fn too_big_packet_repr() -> Repr<'static> {
  716. Repr::PktTooBig {
  717. mtu: 1500,
  718. header: Ipv6Repr {
  719. src_addr: Ipv6Address([0xfe, 0x80, 0x00, 0x00,
  720. 0x00, 0x00, 0x00, 0x00,
  721. 0x00, 0x00, 0x00, 0x00,
  722. 0x00, 0x00, 0x00, 0x01]),
  723. dst_addr: Ipv6Address([0xfe, 0x80, 0x00, 0x00,
  724. 0x00, 0x00, 0x00, 0x00,
  725. 0x00, 0x00, 0x00, 0x00,
  726. 0x00, 0x00, 0x00, 0x02]),
  727. next_header: IpProtocol::Udp,
  728. payload_len: 12,
  729. hop_limit: 0x40
  730. },
  731. data: &PKT_TOO_BIG_UDP_PAYLOAD,
  732. }
  733. }
  734. #[test]
  735. fn test_echo_deconstruct() {
  736. let packet = Packet::new_unchecked(&ECHO_PACKET_BYTES[..]);
  737. assert_eq!(packet.msg_type(), Message::EchoRequest);
  738. assert_eq!(packet.msg_code(), 0);
  739. assert_eq!(packet.checksum(), 0x19b3);
  740. assert_eq!(packet.echo_ident(), 0x1234);
  741. assert_eq!(packet.echo_seq_no(), 0xabcd);
  742. assert_eq!(packet.payload(), &ECHO_PACKET_PAYLOAD[..]);
  743. assert_eq!(packet.verify_checksum(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2), true);
  744. assert!(!packet.msg_type().is_error());
  745. }
  746. #[test]
  747. fn test_echo_construct() {
  748. let mut bytes = vec![0xa5; 12];
  749. let mut packet = Packet::new_unchecked(&mut bytes);
  750. packet.set_msg_type(Message::EchoRequest);
  751. packet.set_msg_code(0);
  752. packet.set_echo_ident(0x1234);
  753. packet.set_echo_seq_no(0xabcd);
  754. packet.payload_mut().copy_from_slice(&ECHO_PACKET_PAYLOAD[..]);
  755. packet.fill_checksum(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2);
  756. assert_eq!(&packet.into_inner()[..], &ECHO_PACKET_BYTES[..]);
  757. }
  758. #[test]
  759. fn test_echo_repr_parse() {
  760. let packet = Packet::new_unchecked(&ECHO_PACKET_BYTES[..]);
  761. let repr = Repr::parse(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2,
  762. &packet, &ChecksumCapabilities::default()).unwrap();
  763. assert_eq!(repr, echo_packet_repr());
  764. }
  765. #[test]
  766. fn test_echo_emit() {
  767. let repr = echo_packet_repr();
  768. let mut bytes = vec![0xa5; repr.buffer_len()];
  769. let mut packet = Packet::new_unchecked(&mut bytes);
  770. repr.emit(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2,
  771. &mut packet, &ChecksumCapabilities::default());
  772. assert_eq!(&packet.into_inner()[..], &ECHO_PACKET_BYTES[..]);
  773. }
  774. #[test]
  775. fn test_too_big_deconstruct() {
  776. let packet = Packet::new_unchecked(&PKT_TOO_BIG_BYTES[..]);
  777. assert_eq!(packet.msg_type(), Message::PktTooBig);
  778. assert_eq!(packet.msg_code(), 0);
  779. assert_eq!(packet.checksum(), 0x0fc9);
  780. assert_eq!(packet.pkt_too_big_mtu(), 1500);
  781. assert_eq!(packet.payload(), &PKT_TOO_BIG_IP_PAYLOAD[..]);
  782. assert_eq!(packet.verify_checksum(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2), true);
  783. assert!(packet.msg_type().is_error());
  784. }
  785. #[test]
  786. fn test_too_big_construct() {
  787. let mut bytes = vec![0xa5; 60];
  788. let mut packet = Packet::new_unchecked(&mut bytes);
  789. packet.set_msg_type(Message::PktTooBig);
  790. packet.set_msg_code(0);
  791. packet.set_pkt_too_big_mtu(1500);
  792. packet.payload_mut().copy_from_slice(&PKT_TOO_BIG_IP_PAYLOAD[..]);
  793. packet.fill_checksum(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2);
  794. assert_eq!(&packet.into_inner()[..], &PKT_TOO_BIG_BYTES[..]);
  795. }
  796. #[test]
  797. fn test_too_big_repr_parse() {
  798. let packet = Packet::new_unchecked(&PKT_TOO_BIG_BYTES[..]);
  799. let repr = Repr::parse(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2,
  800. &packet, &ChecksumCapabilities::default()).unwrap();
  801. assert_eq!(repr, too_big_packet_repr());
  802. }
  803. #[test]
  804. fn test_too_big_emit() {
  805. let repr = too_big_packet_repr();
  806. let mut bytes = vec![0xa5; repr.buffer_len()];
  807. let mut packet = Packet::new_unchecked(&mut bytes);
  808. repr.emit(&MOCK_IP_ADDR_1, &MOCK_IP_ADDR_2,
  809. &mut packet, &ChecksumCapabilities::default());
  810. assert_eq!(&packet.into_inner()[..], &PKT_TOO_BIG_BYTES[..]);
  811. }
  812. }