1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596 |
- // Heads up! Before working on this file you should read the parts
- // of RFC 1122 that discuss Ethernet, ARP and IP for any IPv4 work
- // and RFCs 8200 and 4861 for any IPv6 and NDISC work.
- use core::cmp;
- use managed::{ManagedSlice, ManagedMap};
- #[cfg(not(feature = "proto-igmp"))]
- use core::marker::PhantomData;
- use {Error, Result};
- use phy::{Device, DeviceCapabilities, RxToken, TxToken};
- use time::{Duration, Instant};
- use wire::pretty_print::PrettyPrinter;
- use wire::{EthernetAddress, EthernetProtocol, EthernetFrame};
- use wire::{IpAddress, IpProtocol, IpRepr, IpCidr};
- #[cfg(feature = "proto-ipv6")]
- use wire::{Ipv6Address, Ipv6Packet, Ipv6Repr, IPV6_MIN_MTU};
- #[cfg(feature = "proto-ipv4")]
- use wire::{Ipv4Address, Ipv4Packet, Ipv4Repr, IPV4_MIN_MTU};
- #[cfg(feature = "proto-ipv4")]
- use wire::{ArpPacket, ArpRepr, ArpOperation};
- #[cfg(feature = "proto-ipv4")]
- use wire::{Icmpv4Packet, Icmpv4Repr, Icmpv4DstUnreachable};
- #[cfg(feature = "proto-igmp")]
- use wire::{IgmpPacket, IgmpRepr, IgmpVersion};
- #[cfg(feature = "proto-ipv6")]
- use wire::{Icmpv6Packet, Icmpv6Repr, Icmpv6ParamProblem};
- #[cfg(all(feature = "socket-icmp", any(feature = "proto-ipv4", feature = "proto-ipv6")))]
- use wire::IcmpRepr;
- #[cfg(feature = "proto-ipv6")]
- use wire::{Ipv6HopByHopHeader, Ipv6HopByHopRepr};
- #[cfg(feature = "proto-ipv6")]
- use wire::{Ipv6OptionRepr, Ipv6OptionFailureType};
- #[cfg(feature = "proto-ipv6")]
- use wire::{NdiscNeighborFlags, NdiscRepr};
- #[cfg(all(feature = "proto-ipv6", feature = "socket-udp"))]
- use wire::Icmpv6DstUnreachable;
- #[cfg(feature = "socket-udp")]
- use wire::{UdpPacket, UdpRepr};
- #[cfg(feature = "socket-tcp")]
- use wire::{TcpPacket, TcpRepr, TcpControl};
- use socket::{Socket, SocketSet, AnySocket, PollAt};
- #[cfg(feature = "socket-raw")]
- use socket::RawSocket;
- #[cfg(all(feature = "socket-icmp", any(feature = "proto-ipv4", feature = "proto-ipv6")))]
- use socket::IcmpSocket;
- #[cfg(feature = "socket-udp")]
- use socket::UdpSocket;
- #[cfg(feature = "socket-tcp")]
- use socket::TcpSocket;
- use super::{NeighborCache, NeighborAnswer};
- use super::Routes;
- /// An Ethernet network interface.
- ///
- /// The network interface logically owns a number of other data structures; to avoid
- /// a dependency on heap allocation, it instead owns a `BorrowMut<[T]>`, which can be
- /// a `&mut [T]`, or `Vec<T>` if a heap is available.
- pub struct Interface<'b, 'c, 'e, DeviceT: for<'d> Device<'d>> {
- device: DeviceT,
- inner: InterfaceInner<'b, 'c, 'e>,
- }
- /// The device independent part of an Ethernet network interface.
- ///
- /// Separating the device from the data required for prorcessing and dispatching makes
- /// it possible to borrow them independently. For example, the tx and rx tokens borrow
- /// the `device` mutably until they're used, which makes it impossible to call other
- /// methods on the `Interface` in this time (since its `device` field is borrowed
- /// exclusively). However, it is still possible to call methods on its `inner` field.
- struct InterfaceInner<'b, 'c, 'e> {
- neighbor_cache: NeighborCache<'b>,
- ethernet_addr: EthernetAddress,
- ip_addrs: ManagedSlice<'c, IpCidr>,
- #[cfg(feature = "proto-ipv4")]
- any_ip: bool,
- routes: Routes<'e>,
- #[cfg(feature = "proto-igmp")]
- ipv4_multicast_groups: ManagedMap<'e, Ipv4Address, ()>,
- #[cfg(not(feature = "proto-igmp"))]
- _ipv4_multicast_groups: PhantomData<&'e ()>,
- /// When to report for (all or) the next multicast group membership via IGMP
- #[cfg(feature = "proto-igmp")]
- igmp_report_state: IgmpReportState,
- device_capabilities: DeviceCapabilities,
- }
- /// A builder structure used for creating a Ethernet network
- /// interface.
- pub struct InterfaceBuilder <'b, 'c, 'e, DeviceT: for<'d> Device<'d>> {
- device: DeviceT,
- ethernet_addr: Option<EthernetAddress>,
- neighbor_cache: Option<NeighborCache<'b>>,
- ip_addrs: ManagedSlice<'c, IpCidr>,
- #[cfg(feature = "proto-ipv4")]
- any_ip: bool,
- routes: Routes<'e>,
- /// Does not share storage with `ipv6_multicast_groups` to avoid IPv6 size overhead.
- #[cfg(feature = "proto-igmp")]
- ipv4_multicast_groups: ManagedMap<'e, Ipv4Address, ()>,
- #[cfg(not(feature = "proto-igmp"))]
- _ipv4_multicast_groups: PhantomData<&'e ()>,
- }
- impl<'b, 'c, 'e, DeviceT> InterfaceBuilder<'b, 'c, 'e, DeviceT>
- where DeviceT: for<'d> Device<'d> {
- /// Create a builder used for creating a network interface using the
- /// given device and address.
- ///
- /// # Examples
- ///
- /// ```
- /// # use std::collections::BTreeMap;
- /// use smoltcp::iface::{EthernetInterfaceBuilder, NeighborCache};
- /// # use smoltcp::phy::Loopback;
- /// use smoltcp::wire::{EthernetAddress, IpCidr, IpAddress};
- ///
- /// let device = // ...
- /// # Loopback::new();
- /// let hw_addr = // ...
- /// # EthernetAddress::default();
- /// let neighbor_cache = // ...
- /// # NeighborCache::new(BTreeMap::new());
- /// let ip_addrs = // ...
- /// # [];
- /// let iface = EthernetInterfaceBuilder::new(device)
- /// .ethernet_addr(hw_addr)
- /// .neighbor_cache(neighbor_cache)
- /// .ip_addrs(ip_addrs)
- /// .finalize();
- /// ```
- pub fn new(device: DeviceT) -> Self {
- InterfaceBuilder {
- device: device,
- ethernet_addr: None,
- neighbor_cache: None,
- ip_addrs: ManagedSlice::Borrowed(&mut []),
- #[cfg(feature = "proto-ipv4")]
- any_ip: false,
- routes: Routes::new(ManagedMap::Borrowed(&mut [])),
- #[cfg(feature = "proto-igmp")]
- ipv4_multicast_groups: ManagedMap::Borrowed(&mut []),
- #[cfg(not(feature = "proto-igmp"))]
- _ipv4_multicast_groups: PhantomData,
- }
- }
- /// Set the Ethernet address the interface will use. See also
- /// [ethernet_addr].
- ///
- /// # Panics
- /// This function panics if the address is not unicast.
- ///
- /// [ethernet_addr]: struct.EthernetInterface.html#method.ethernet_addr
- pub fn ethernet_addr(mut self, addr: EthernetAddress) -> Self {
- InterfaceInner::check_ethernet_addr(&addr);
- self.ethernet_addr = Some(addr);
- self
- }
- /// Set the IP addresses the interface will use. See also
- /// [ip_addrs].
- ///
- /// # Panics
- /// This function panics if any of the addresses are not unicast.
- ///
- /// [ip_addrs]: struct.EthernetInterface.html#method.ip_addrs
- pub fn ip_addrs<T>(mut self, ip_addrs: T) -> Self
- where T: Into<ManagedSlice<'c, IpCidr>>
- {
- let ip_addrs = ip_addrs.into();
- InterfaceInner::check_ip_addrs(&ip_addrs);
- self.ip_addrs = ip_addrs;
- self
- }
- /// Enable or disable the AnyIP capability, allowing packets to be received
- /// locally on IPv4 addresses other than the interface's configured [ip_addrs].
- /// When AnyIP is enabled and a route prefix in [routes] specifies one of
- /// the interface's [ip_addrs] as its gateway, the interface will accept
- /// packets addressed to that prefix.
- ///
- /// # IPv6
- ///
- /// This option is not available or required for IPv6 as packets sent to
- /// the interface are not filtered by IPv6 address.
- ///
- /// [routes]: struct.EthernetInterface.html#method.routes
- /// [ip_addrs]: struct.EthernetInterface.html#method.ip_addrs
- #[cfg(feature = "proto-ipv4")]
- pub fn any_ip(mut self, enabled: bool) -> Self {
- self.any_ip = enabled;
- self
- }
- /// Set the IP routes the interface will use. See also
- /// [routes].
- ///
- /// [routes]: struct.EthernetInterface.html#method.routes
- pub fn routes<T>(mut self, routes: T) -> InterfaceBuilder<'b, 'c, 'e, DeviceT>
- where T: Into<Routes<'e>>
- {
- self.routes = routes.into();
- self
- }
- /// Provide storage for multicast groups.
- ///
- /// Join multicast groups by calling [`join_multicast_group()`] on an `Interface`.
- /// Using [`join_multicast_group()`] will send initial membership reports.
- ///
- /// A previously destroyed interface can be recreated by reusing the multicast group
- /// storage, i.e. providing a non-empty storage to `ipv4_multicast_groups()`.
- /// Note that this way initial membership reports are **not** sent.
- ///
- /// [`join_multicast_group()`]: struct.EthernetInterface.html#method.join_multicast_group
- #[cfg(feature = "proto-igmp")]
- pub fn ipv4_multicast_groups<T>(mut self, ipv4_multicast_groups: T) -> Self
- where T: Into<ManagedMap<'e, Ipv4Address, ()>>
- {
- self.ipv4_multicast_groups = ipv4_multicast_groups.into();
- self
- }
- /// Set the Neighbor Cache the interface will use.
- pub fn neighbor_cache(mut self, neighbor_cache: NeighborCache<'b>) -> Self {
- self.neighbor_cache = Some(neighbor_cache);
- self
- }
- /// Create a network interface using the previously provided configuration.
- ///
- /// # Panics
- /// If a required option is not provided, this function will panic. Required
- /// options are:
- ///
- /// - [ethernet_addr]
- /// - [neighbor_cache]
- ///
- /// [ethernet_addr]: #method.ethernet_addr
- /// [neighbor_cache]: #method.neighbor_cache
- pub fn finalize(self) -> Interface<'b, 'c, 'e, DeviceT> {
- match (self.ethernet_addr, self.neighbor_cache) {
- (Some(ethernet_addr), Some(neighbor_cache)) => {
- let device_capabilities = self.device.capabilities();
- Interface {
- device: self.device,
- inner: InterfaceInner {
- ethernet_addr, device_capabilities, neighbor_cache,
- ip_addrs: self.ip_addrs,
- #[cfg(feature = "proto-ipv4")]
- any_ip: self.any_ip,
- routes: self.routes,
- #[cfg(feature = "proto-igmp")]
- ipv4_multicast_groups: self.ipv4_multicast_groups,
- #[cfg(not(feature = "proto-igmp"))]
- _ipv4_multicast_groups: PhantomData,
- #[cfg(feature = "proto-igmp")]
- igmp_report_state: IgmpReportState::Inactive,
- }
- }
- },
- _ => panic!("a required option was not set"),
- }
- }
- }
- #[derive(Debug, PartialEq)]
- enum Packet<'a> {
- None,
- #[cfg(feature = "proto-ipv4")]
- Arp(ArpRepr),
- #[cfg(feature = "proto-ipv4")]
- Icmpv4((Ipv4Repr, Icmpv4Repr<'a>)),
- #[cfg(feature = "proto-igmp")]
- Igmp((Ipv4Repr, IgmpRepr)),
- #[cfg(feature = "proto-ipv6")]
- Icmpv6((Ipv6Repr, Icmpv6Repr<'a>)),
- #[cfg(feature = "socket-raw")]
- Raw((IpRepr, &'a [u8])),
- #[cfg(feature = "socket-udp")]
- Udp((IpRepr, UdpRepr<'a>)),
- #[cfg(feature = "socket-tcp")]
- Tcp((IpRepr, TcpRepr<'a>))
- }
- impl<'a> Packet<'a> {
- fn neighbor_addr(&self) -> Option<IpAddress> {
- match self {
- &Packet::None => None,
- #[cfg(feature = "proto-ipv4")]
- &Packet::Arp(_) => None,
- #[cfg(feature = "proto-ipv4")]
- &Packet::Icmpv4((ref ipv4_repr, _)) => Some(ipv4_repr.dst_addr.into()),
- #[cfg(feature = "proto-igmp")]
- &Packet::Igmp((ref ipv4_repr, _)) => Some(ipv4_repr.dst_addr.into()),
- #[cfg(feature = "proto-ipv6")]
- &Packet::Icmpv6((ref ipv6_repr, _)) => Some(ipv6_repr.dst_addr.into()),
- #[cfg(feature = "socket-raw")]
- &Packet::Raw((ref ip_repr, _)) => Some(ip_repr.dst_addr()),
- #[cfg(feature = "socket-udp")]
- &Packet::Udp((ref ip_repr, _)) => Some(ip_repr.dst_addr()),
- #[cfg(feature = "socket-tcp")]
- &Packet::Tcp((ref ip_repr, _)) => Some(ip_repr.dst_addr())
- }
- }
- }
- #[cfg(any(feature = "proto-ipv4", feature = "proto-ipv6"))]
- fn icmp_reply_payload_len(len: usize, mtu: usize, header_len: usize) -> usize {
- // Send back as much of the original payload as will fit within
- // the minimum MTU required by IPv4. See RFC 1812 § 4.3.2.3 for
- // more details.
- //
- // Since the entire network layer packet must fit within the minumum
- // MTU supported, the payload must not exceed the following:
- //
- // <min mtu> - IP Header Size * 2 - ICMPv4 DstUnreachable hdr size
- cmp::min(len, mtu - header_len * 2 - 8)
- }
- #[cfg(feature = "proto-igmp")]
- enum IgmpReportState {
- Inactive,
- ToGeneralQuery {
- version: IgmpVersion,
- timeout: Instant,
- interval: Duration,
- next_index: usize
- },
- ToSpecificQuery {
- version: IgmpVersion,
- timeout: Instant,
- group: Ipv4Address
- },
- }
- impl<'b, 'c, 'e, DeviceT> Interface<'b, 'c, 'e, DeviceT>
- where DeviceT: for<'d> Device<'d> {
- /// Get the Ethernet address of the interface.
- pub fn ethernet_addr(&self) -> EthernetAddress {
- self.inner.ethernet_addr
- }
- /// Set the Ethernet address of the interface.
- ///
- /// # Panics
- /// This function panics if the address is not unicast.
- pub fn set_ethernet_addr(&mut self, addr: EthernetAddress) {
- self.inner.ethernet_addr = addr;
- InterfaceInner::check_ethernet_addr(&self.inner.ethernet_addr);
- }
- /// Get a reference to the inner device.
- pub fn device(&self) -> &DeviceT {
- &self.device
- }
- /// Get a mutable reference to the inner device.
- ///
- /// There are no invariants imposed on the device by the interface itself. Furthermore the
- /// trait implementations, required for references of all lifetimes, guarantees that the
- /// mutable reference can not invalidate the device as such. For some devices, such access may
- /// still allow modifications with adverse effects on the usability as a `phy` device. You
- /// should not use them this way.
- pub fn device_mut(&mut self) -> &mut DeviceT {
- &mut self.device
- }
- /// Add an address to a list of subscribed multicast IP addresses.
- ///
- /// Returns `Ok(announce_sent)` if the address was added successfully, where `annouce_sent`
- /// indicates whether an initial immediate announcement has been sent.
- pub fn join_multicast_group<T: Into<IpAddress>>(&mut self, addr: T, _timestamp: Instant) -> Result<bool> {
- match addr.into() {
- #[cfg(feature = "proto-igmp")]
- IpAddress::Ipv4(addr) => {
- let is_not_new = self.inner.ipv4_multicast_groups.insert(addr, ())
- .map_err(|_| Error::Exhausted)?
- .is_some();
- if is_not_new {
- Ok(false)
- } else if let Some(pkt) =
- self.inner.igmp_report_packet(IgmpVersion::Version2, addr) {
- // Send initial membership report
- let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
- self.inner.dispatch(tx_token, _timestamp, pkt)?;
- Ok(true)
- } else {
- Ok(false)
- }
- }
- // Multicast is not yet implemented for other address families
- _ => Err(Error::Unaddressable)
- }
- }
- /// Remove an address from the subscribed multicast IP addresses.
- ///
- /// Returns `Ok(leave_sent)` if the address was removed successfully, where `leave_sent`
- /// indicates whether an immediate leave packet has been sent.
- pub fn leave_multicast_group<T: Into<IpAddress>>(&mut self, addr: T, _timestamp: Instant) -> Result<bool> {
- match addr.into() {
- #[cfg(feature = "proto-igmp")]
- IpAddress::Ipv4(addr) => {
- let was_not_present = self.inner.ipv4_multicast_groups.remove(&addr)
- .is_none();
- if was_not_present {
- Ok(false)
- } else if let Some(pkt) = self.inner.igmp_leave_packet(addr) {
- // Send group leave packet
- let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
- self.inner.dispatch(tx_token, _timestamp, pkt)?;
- Ok(true)
- } else {
- Ok(false)
- }
- }
- // Multicast is not yet implemented for other address families
- _ => Err(Error::Unaddressable)
- }
- }
- /// Check whether the interface listens to given destination multicast IP address.
- pub fn has_multicast_group<T: Into<IpAddress>>(&self, addr: T) -> bool {
- self.inner.has_multicast_group(addr)
- }
- /// Get the IP addresses of the interface.
- pub fn ip_addrs(&self) -> &[IpCidr] {
- self.inner.ip_addrs.as_ref()
- }
- /// Update the IP addresses of the interface.
- ///
- /// # Panics
- /// This function panics if any of the addresses are not unicast.
- pub fn update_ip_addrs<F: FnOnce(&mut ManagedSlice<'c, IpCidr>)>(&mut self, f: F) {
- f(&mut self.inner.ip_addrs);
- InterfaceInner::check_ip_addrs(&self.inner.ip_addrs)
- }
- /// Check whether the interface has the given IP address assigned.
- pub fn has_ip_addr<T: Into<IpAddress>>(&self, addr: T) -> bool {
- self.inner.has_ip_addr(addr)
- }
- /// Get the first IPv4 address of the interface.
- #[cfg(feature = "proto-ipv4")]
- pub fn ipv4_address(&self) -> Option<Ipv4Address> {
- self.inner.ipv4_address()
- }
- pub fn routes(&self) -> &Routes<'e> {
- &self.inner.routes
- }
- pub fn routes_mut(&mut self) -> &mut Routes<'e> {
- &mut self.inner.routes
- }
- /// Transmit packets queued in the given sockets, and receive packets queued
- /// in the device.
- ///
- /// This function returns a boolean value indicating whether any packets were
- /// processed or emitted, and thus, whether the readiness of any socket might
- /// have changed.
- ///
- /// # Errors
- /// This method will routinely return errors in response to normal network
- /// activity as well as certain boundary conditions such as buffer exhaustion.
- /// These errors are provided as an aid for troubleshooting, and are meant
- /// to be logged and ignored.
- ///
- /// As a special case, `Err(Error::Unrecognized)` is returned in response to
- /// packets containing any unsupported protocol, option, or form, which is
- /// a very common occurrence and on a production system it should not even
- /// be logged.
- pub fn poll(&mut self, sockets: &mut SocketSet, timestamp: Instant) -> Result<bool> {
- let mut readiness_may_have_changed = false;
- loop {
- let processed_any = self.socket_ingress(sockets, timestamp)?;
- let emitted_any = self.socket_egress(sockets, timestamp)?;
- #[cfg(feature = "proto-igmp")]
- self.igmp_egress(timestamp)?;
- if processed_any || emitted_any {
- readiness_may_have_changed = true;
- } else {
- break
- }
- }
- Ok(readiness_may_have_changed)
- }
- /// Return a _soft deadline_ for calling [poll] the next time.
- /// The [Instant] returned is the time at which you should call [poll] next.
- /// It is harmless (but wastes energy) to call it before the [Instant], and
- /// potentially harmful (impacting quality of service) to call it after the
- /// [Instant]
- ///
- /// [poll]: #method.poll
- /// [Instant]: struct.Instant.html
- pub fn poll_at(&self, sockets: &SocketSet, timestamp: Instant) -> Option<Instant> {
- sockets.iter().filter_map(|socket| {
- let socket_poll_at = socket.poll_at();
- match socket.meta().poll_at(socket_poll_at, |ip_addr|
- self.inner.has_neighbor(&ip_addr, timestamp)) {
- PollAt::Ingress => None,
- PollAt::Time(instant) => Some(instant),
- PollAt::Now => Some(Instant::from_millis(0)),
- }
- }).min()
- }
- /// Return an _advisory wait time_ for calling [poll] the next time.
- /// The [Duration] returned is the time left to wait before calling [poll] next.
- /// It is harmless (but wastes energy) to call it before the [Duration] has passed,
- /// and potentially harmful (impacting quality of service) to call it after the
- /// [Duration] has passed.
- ///
- /// [poll]: #method.poll
- /// [Duration]: struct.Duration.html
- pub fn poll_delay(&self, sockets: &SocketSet, timestamp: Instant) -> Option<Duration> {
- match self.poll_at(sockets, timestamp) {
- Some(poll_at) if timestamp < poll_at => {
- Some(poll_at - timestamp)
- }
- Some(_) => {
- Some(Duration::from_millis(0))
- }
- _ => None
- }
- }
- fn socket_ingress(&mut self, sockets: &mut SocketSet, timestamp: Instant) -> Result<bool> {
- let mut processed_any = false;
- loop {
- let &mut Self { ref mut device, ref mut inner } = self;
- let (rx_token, tx_token) = match device.receive() {
- None => break,
- Some(tokens) => tokens,
- };
- rx_token.consume(timestamp, |frame| {
- inner.process_ethernet(sockets, timestamp, &frame).map_err(|err| {
- net_debug!("cannot process ingress packet: {}", err);
- net_debug!("packet dump follows:\n{}",
- PrettyPrinter::<EthernetFrame<&[u8]>>::new("", &frame));
- err
- }).and_then(|response| {
- processed_any = true;
- inner.dispatch(tx_token, timestamp, response).map_err(|err| {
- net_debug!("cannot dispatch response packet: {}", err);
- err
- })
- })
- })?;
- }
- Ok(processed_any)
- }
- fn socket_egress(&mut self, sockets: &mut SocketSet, timestamp: Instant) -> Result<bool> {
- let mut caps = self.device.capabilities();
- caps.max_transmission_unit -= EthernetFrame::<&[u8]>::header_len();
- let mut emitted_any = false;
- for mut socket in sockets.iter_mut() {
- if !socket.meta_mut().egress_permitted(|ip_addr|
- self.inner.has_neighbor(&ip_addr, timestamp)) {
- continue
- }
- let mut neighbor_addr = None;
- let mut device_result = Ok(());
- let &mut Self { ref mut device, ref mut inner } = self;
- macro_rules! respond {
- ($response:expr) => ({
- let response = $response;
- neighbor_addr = response.neighbor_addr();
- let tx_token = device.transmit().ok_or(Error::Exhausted)?;
- device_result = inner.dispatch(tx_token, timestamp, response);
- device_result
- })
- }
- let socket_result =
- match *socket {
- #[cfg(feature = "socket-raw")]
- Socket::Raw(ref mut socket) =>
- socket.dispatch(&caps.checksum, |response|
- respond!(Packet::Raw(response))),
- #[cfg(all(feature = "socket-icmp", any(feature = "proto-ipv4", feature = "proto-ipv6")))]
- Socket::Icmp(ref mut socket) =>
- socket.dispatch(&caps, |response| {
- match response {
- #[cfg(feature = "proto-ipv4")]
- (IpRepr::Ipv4(ipv4_repr), IcmpRepr::Ipv4(icmpv4_repr)) =>
- respond!(Packet::Icmpv4((ipv4_repr, icmpv4_repr))),
- #[cfg(feature = "proto-ipv6")]
- (IpRepr::Ipv6(ipv6_repr), IcmpRepr::Ipv6(icmpv6_repr)) =>
- respond!(Packet::Icmpv6((ipv6_repr, icmpv6_repr))),
- _ => Err(Error::Unaddressable)
- }
- }),
- #[cfg(feature = "socket-udp")]
- Socket::Udp(ref mut socket) =>
- socket.dispatch(|response|
- respond!(Packet::Udp(response))),
- #[cfg(feature = "socket-tcp")]
- Socket::Tcp(ref mut socket) =>
- socket.dispatch(timestamp, &caps, |response|
- respond!(Packet::Tcp(response))),
- Socket::__Nonexhaustive(_) => unreachable!()
- };
- match (device_result, socket_result) {
- (Err(Error::Exhausted), _) => break, // nowhere to transmit
- (Ok(()), Err(Error::Exhausted)) => (), // nothing to transmit
- (Err(Error::Unaddressable), _) => {
- // `NeighborCache` already takes care of rate limiting the neighbor discovery
- // requests from the socket. However, without an additional rate limiting
- // mechanism, we would spin on every socket that has yet to discover its
- // neighboor.
- socket.meta_mut().neighbor_missing(timestamp,
- neighbor_addr.expect("non-IP response packet"));
- break
- }
- (Err(err), _) | (_, Err(err)) => {
- net_debug!("{}: cannot dispatch egress packet: {}",
- socket.meta().handle, err);
- return Err(err)
- }
- (Ok(()), Ok(())) => emitted_any = true
- }
- }
- Ok(emitted_any)
- }
- /// Depending on `igmp_report_state` and the therein contained
- /// timeouts, send IGMP membership reports.
- #[cfg(feature = "proto-igmp")]
- fn igmp_egress(&mut self, timestamp: Instant) -> Result<bool> {
- match self.inner.igmp_report_state {
- IgmpReportState::ToSpecificQuery { version, timeout, group }
- if timestamp >= timeout => {
- if let Some(pkt) = self.inner.igmp_report_packet(version, group) {
- // Send initial membership report
- let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
- self.inner.dispatch(tx_token, timestamp, pkt)?;
- }
- self.inner.igmp_report_state = IgmpReportState::Inactive;
- Ok(true)
- }
- IgmpReportState::ToGeneralQuery { version, timeout, interval, next_index }
- if timestamp >= timeout => {
- let addr = self.inner.ipv4_multicast_groups
- .iter()
- .nth(next_index)
- .map(|(addr, ())| *addr);
- match addr {
- Some(addr) => {
- if let Some(pkt) = self.inner.igmp_report_packet(version, addr) {
- // Send initial membership report
- let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
- self.inner.dispatch(tx_token, timestamp, pkt)?;
- }
- let next_timeout = (timeout + interval).max(timestamp);
- self.inner.igmp_report_state = IgmpReportState::ToGeneralQuery {
- version, timeout: next_timeout, interval, next_index: next_index + 1
- };
- Ok(true)
- }
- None => {
- self.inner.igmp_report_state = IgmpReportState::Inactive;
- Ok(false)
- }
- }
- }
- _ => Ok(false)
- }
- }
- }
- impl<'b, 'c, 'e> InterfaceInner<'b, 'c, 'e> {
- fn check_ethernet_addr(addr: &EthernetAddress) {
- if addr.is_multicast() {
- panic!("Ethernet address {} is not unicast", addr)
- }
- }
- fn check_ip_addrs(addrs: &[IpCidr]) {
- for cidr in addrs {
- if !cidr.address().is_unicast() {
- panic!("IP address {} is not unicast", cidr.address())
- }
- }
- }
- /// Determine if the given `Ipv6Address` is the solicited node
- /// multicast address for a IPv6 addresses assigned to the interface.
- /// See [RFC 4291 § 2.7.1] for more details.
- ///
- /// [RFC 4291 § 2.7.1]: https://tools.ietf.org/html/rfc4291#section-2.7.1
- #[cfg(feature = "proto-ipv6")]
- pub fn has_solicited_node(&self, addr: Ipv6Address) -> bool {
- self.ip_addrs.iter().find(|cidr| {
- match *cidr {
- &IpCidr::Ipv6(cidr) if cidr.address() != Ipv6Address::LOOPBACK=> {
- // Take the lower order 24 bits of the IPv6 address and
- // append those bits to FF02:0:0:0:0:1:FF00::/104.
- addr.as_bytes()[14..] == cidr.address().as_bytes()[14..]
- }
- _ => false,
- }
- }).is_some()
- }
- /// Check whether the interface has the given IP address assigned.
- fn has_ip_addr<T: Into<IpAddress>>(&self, addr: T) -> bool {
- let addr = addr.into();
- self.ip_addrs.iter().any(|probe| probe.address() == addr)
- }
- /// Get the first IPv4 address of the interface.
- #[cfg(feature = "proto-ipv4")]
- pub fn ipv4_address(&self) -> Option<Ipv4Address> {
- self.ip_addrs.iter()
- .filter_map(
- |addr| match addr {
- &IpCidr::Ipv4(cidr) => Some(cidr.address()),
- _ => None,
- })
- .next()
- }
- /// Check whether the interface listens to given destination multicast IP address.
- ///
- /// If built without feature `proto-igmp` this function will
- /// always return `false`.
- pub fn has_multicast_group<T: Into<IpAddress>>(&self, addr: T) -> bool {
- match addr.into() {
- #[cfg(feature = "proto-igmp")]
- IpAddress::Ipv4(key) =>
- key == Ipv4Address::MULTICAST_ALL_SYSTEMS ||
- self.ipv4_multicast_groups.get(&key).is_some(),
- _ =>
- false,
- }
- }
- fn process_ethernet<'frame, T: AsRef<[u8]>>
- (&mut self, sockets: &mut SocketSet, timestamp: Instant, frame: &'frame T) ->
- Result<Packet<'frame>>
- {
- let eth_frame = EthernetFrame::new_checked(frame)?;
- // Ignore any packets not directed to our hardware address or any of the multicast groups.
- if !eth_frame.dst_addr().is_broadcast() &&
- !eth_frame.dst_addr().is_multicast() &&
- eth_frame.dst_addr() != self.ethernet_addr
- {
- return Ok(Packet::None)
- }
- match eth_frame.ethertype() {
- #[cfg(feature = "proto-ipv4")]
- EthernetProtocol::Arp =>
- self.process_arp(timestamp, ð_frame),
- #[cfg(feature = "proto-ipv4")]
- EthernetProtocol::Ipv4 =>
- self.process_ipv4(sockets, timestamp, ð_frame),
- #[cfg(feature = "proto-ipv6")]
- EthernetProtocol::Ipv6 =>
- self.process_ipv6(sockets, timestamp, ð_frame),
- // Drop all other traffic.
- _ => Err(Error::Unrecognized),
- }
- }
- #[cfg(feature = "proto-ipv4")]
- fn process_arp<'frame, T: AsRef<[u8]>>
- (&mut self, timestamp: Instant, eth_frame: &EthernetFrame<&'frame T>) ->
- Result<Packet<'frame>>
- {
- let arp_packet = ArpPacket::new_checked(eth_frame.payload())?;
- let arp_repr = ArpRepr::parse(&arp_packet)?;
- match arp_repr {
- // Respond to ARP requests aimed at us, and fill the ARP cache from all ARP
- // requests and replies, to minimize the chance that we have to perform
- // an explicit ARP request.
- ArpRepr::EthernetIpv4 {
- operation, source_hardware_addr, source_protocol_addr, target_protocol_addr, ..
- } => {
- if source_protocol_addr.is_unicast() && source_hardware_addr.is_unicast() {
- self.neighbor_cache.fill(source_protocol_addr.into(),
- source_hardware_addr,
- timestamp);
- } else {
- // Discard packets with non-unicast source addresses.
- net_debug!("non-unicast source address");
- return Err(Error::Malformed)
- }
- if operation == ArpOperation::Request && self.has_ip_addr(target_protocol_addr) {
- Ok(Packet::Arp(ArpRepr::EthernetIpv4 {
- operation: ArpOperation::Reply,
- source_hardware_addr: self.ethernet_addr,
- source_protocol_addr: target_protocol_addr,
- target_hardware_addr: source_hardware_addr,
- target_protocol_addr: source_protocol_addr
- }))
- } else {
- Ok(Packet::None)
- }
- }
- _ => Err(Error::Unrecognized)
- }
- }
- #[cfg(all(any(feature = "proto-ipv4", feature = "proto-ipv6"), feature = "socket-raw"))]
- fn raw_socket_filter<'frame>(&mut self, sockets: &mut SocketSet, ip_repr: &IpRepr,
- ip_payload: &'frame [u8]) -> bool {
- let checksum_caps = self.device_capabilities.checksum.clone();
- let mut handled_by_raw_socket = false;
- // Pass every IP packet to all raw sockets we have registered.
- for mut raw_socket in sockets.iter_mut().filter_map(RawSocket::downcast) {
- if !raw_socket.accepts(&ip_repr) { continue }
- match raw_socket.process(&ip_repr, ip_payload, &checksum_caps) {
- // The packet is valid and handled by socket.
- Ok(()) => handled_by_raw_socket = true,
- // The socket buffer is full.
- Err(Error::Exhausted) => (),
- // Raw sockets don't validate the packets in any way.
- Err(_) => unreachable!(),
- }
- }
- handled_by_raw_socket
- }
- #[cfg(feature = "proto-ipv6")]
- fn process_ipv6<'frame, T: AsRef<[u8]>>
- (&mut self, sockets: &mut SocketSet, timestamp: Instant,
- eth_frame: &EthernetFrame<&'frame T>) ->
- Result<Packet<'frame>>
- {
- let ipv6_packet = Ipv6Packet::new_checked(eth_frame.payload())?;
- let ipv6_repr = Ipv6Repr::parse(&ipv6_packet)?;
- if !ipv6_repr.src_addr.is_unicast() {
- // Discard packets with non-unicast source addresses.
- net_debug!("non-unicast source address");
- return Err(Error::Malformed)
- }
- if eth_frame.src_addr().is_unicast() {
- // Fill the neighbor cache from IP header of unicast frames.
- let ip_addr = IpAddress::Ipv6(ipv6_repr.src_addr);
- if self.in_same_network(&ip_addr) &&
- self.neighbor_cache.lookup_pure(&ip_addr, timestamp).is_none() {
- self.neighbor_cache.fill(ip_addr, eth_frame.src_addr(), timestamp);
- }
- }
- let ip_payload = ipv6_packet.payload();
- #[cfg(feature = "socket-raw")]
- let handled_by_raw_socket = self.raw_socket_filter(sockets, &ipv6_repr.into(), ip_payload);
- #[cfg(not(feature = "socket-raw"))]
- let handled_by_raw_socket = false;
- self.process_nxt_hdr(sockets, timestamp, ipv6_repr, ipv6_repr.next_header,
- handled_by_raw_socket, ip_payload)
- }
- /// Given the next header value forward the payload onto the correct process
- /// function.
- #[cfg(feature = "proto-ipv6")]
- fn process_nxt_hdr<'frame>
- (&mut self, sockets: &mut SocketSet, timestamp: Instant, ipv6_repr: Ipv6Repr,
- nxt_hdr: IpProtocol, handled_by_raw_socket: bool, ip_payload: &'frame [u8])
- -> Result<Packet<'frame>>
- {
- match nxt_hdr {
- IpProtocol::Icmpv6 =>
- self.process_icmpv6(sockets, timestamp, ipv6_repr.into(), ip_payload),
- #[cfg(feature = "socket-udp")]
- IpProtocol::Udp =>
- self.process_udp(sockets, ipv6_repr.into(), ip_payload),
- #[cfg(feature = "socket-tcp")]
- IpProtocol::Tcp =>
- self.process_tcp(sockets, timestamp, ipv6_repr.into(), ip_payload),
- IpProtocol::HopByHop =>
- self.process_hopbyhop(sockets, timestamp, ipv6_repr, handled_by_raw_socket, ip_payload),
- #[cfg(feature = "socket-raw")]
- _ if handled_by_raw_socket =>
- Ok(Packet::None),
- _ => {
- // Send back as much of the original payload as we can.
- let payload_len = icmp_reply_payload_len(ip_payload.len(), IPV6_MIN_MTU,
- ipv6_repr.buffer_len());
- let icmp_reply_repr = Icmpv6Repr::ParamProblem {
- reason: Icmpv6ParamProblem::UnrecognizedNxtHdr,
- // The offending packet is after the IPv6 header.
- pointer: ipv6_repr.buffer_len() as u32,
- header: ipv6_repr,
- data: &ip_payload[0..payload_len]
- };
- Ok(self.icmpv6_reply(ipv6_repr, icmp_reply_repr))
- },
- }
- }
- #[cfg(feature = "proto-ipv4")]
- fn process_ipv4<'frame, T: AsRef<[u8]>>
- (&mut self, sockets: &mut SocketSet, timestamp: Instant,
- eth_frame: &EthernetFrame<&'frame T>) ->
- Result<Packet<'frame>>
- {
- let ipv4_packet = Ipv4Packet::new_checked(eth_frame.payload())?;
- let checksum_caps = self.device_capabilities.checksum.clone();
- let ipv4_repr = Ipv4Repr::parse(&ipv4_packet, &checksum_caps)?;
- if !ipv4_repr.src_addr.is_unicast() {
- // Discard packets with non-unicast source addresses.
- net_debug!("non-unicast source address");
- return Err(Error::Malformed)
- }
- if eth_frame.src_addr().is_unicast() {
- // Fill the neighbor cache from IP header of unicast frames.
- let ip_addr = IpAddress::Ipv4(ipv4_repr.src_addr);
- if self.in_same_network(&ip_addr) {
- self.neighbor_cache.fill(ip_addr, eth_frame.src_addr(), timestamp);
- }
- }
- let ip_repr = IpRepr::Ipv4(ipv4_repr);
- let ip_payload = ipv4_packet.payload();
- #[cfg(feature = "socket-raw")]
- let handled_by_raw_socket = self.raw_socket_filter(sockets, &ip_repr, ip_payload);
- if !self.has_ip_addr(ipv4_repr.dst_addr) &&
- !ipv4_repr.dst_addr.is_broadcast() &&
- !self.has_multicast_group(ipv4_repr.dst_addr) {
- // Ignore IP packets not directed at us, or broadcast, or any of the multicast groups.
- // If AnyIP is enabled, also check if the packet is routed locally.
- if !self.any_ip {
- return Ok(Packet::None);
- } else if match self.routes.lookup(&IpAddress::Ipv4(ipv4_repr.dst_addr), timestamp) {
- Some(router_addr) => !self.has_ip_addr(router_addr),
- None => true,
- } {
- return Ok(Packet::None);
- }
- }
- match ipv4_repr.protocol {
- IpProtocol::Icmp =>
- self.process_icmpv4(sockets, ip_repr, ip_payload),
- #[cfg(feature = "proto-igmp")]
- IpProtocol::Igmp =>
- self.process_igmp(timestamp, ipv4_repr, ip_payload),
- #[cfg(feature = "socket-udp")]
- IpProtocol::Udp =>
- self.process_udp(sockets, ip_repr, ip_payload),
- #[cfg(feature = "socket-tcp")]
- IpProtocol::Tcp =>
- self.process_tcp(sockets, timestamp, ip_repr, ip_payload),
- #[cfg(feature = "socket-raw")]
- _ if handled_by_raw_socket =>
- Ok(Packet::None),
- _ => {
- // Send back as much of the original payload as we can.
- let payload_len = icmp_reply_payload_len(ip_payload.len(), IPV4_MIN_MTU,
- ipv4_repr.buffer_len());
- let icmp_reply_repr = Icmpv4Repr::DstUnreachable {
- reason: Icmpv4DstUnreachable::ProtoUnreachable,
- header: ipv4_repr,
- data: &ip_payload[0..payload_len]
- };
- Ok(self.icmpv4_reply(ipv4_repr, icmp_reply_repr))
- }
- }
- }
- /// Host duties of the **IGMPv2** protocol.
- ///
- /// Sets up `igmp_report_state` for responding to IGMP general/specific membership queries.
- /// Membership must not be reported immediately in order to avoid flooding the network
- /// after a query is broadcasted by a router; this is not currently done.
- #[cfg(feature = "proto-igmp")]
- fn process_igmp<'frame>(&mut self, timestamp: Instant, ipv4_repr: Ipv4Repr,
- ip_payload: &'frame [u8]) -> Result<Packet<'frame>> {
- let igmp_packet = IgmpPacket::new_checked(ip_payload)?;
- let igmp_repr = IgmpRepr::parse(&igmp_packet)?;
- // FIXME: report membership after a delay
- match igmp_repr {
- IgmpRepr::MembershipQuery { group_addr, version, max_resp_time } => {
- // General query
- if group_addr.is_unspecified() &&
- ipv4_repr.dst_addr == Ipv4Address::MULTICAST_ALL_SYSTEMS {
- // Are we member in any groups?
- if self.ipv4_multicast_groups.iter().next().is_some() {
- let interval = match version {
- IgmpVersion::Version1 =>
- Duration::from_millis(100),
- IgmpVersion::Version2 => {
- // No dependence on a random generator
- // (see [#24](https://github.com/m-labs/smoltcp/issues/24))
- // but at least spread reports evenly across max_resp_time.
- let intervals = self.ipv4_multicast_groups.len() as u32 + 1;
- max_resp_time / intervals
- }
- };
- self.igmp_report_state = IgmpReportState::ToGeneralQuery {
- version, timeout: timestamp + interval, interval, next_index: 0
- };
- }
- } else {
- // Group-specific query
- if self.has_multicast_group(group_addr) && ipv4_repr.dst_addr == group_addr {
- // Don't respond immediately
- let timeout = max_resp_time / 4;
- self.igmp_report_state = IgmpReportState::ToSpecificQuery {
- version, timeout: timestamp + timeout, group: group_addr
- };
- }
- }
- },
- // Ignore membership reports
- IgmpRepr::MembershipReport { .. } => (),
- // Ignore hosts leaving groups
- IgmpRepr::LeaveGroup{ .. } => (),
- }
- Ok(Packet::None)
- }
- #[cfg(feature = "proto-ipv6")]
- fn process_icmpv6<'frame>(&mut self, _sockets: &mut SocketSet, timestamp: Instant,
- ip_repr: IpRepr, ip_payload: &'frame [u8]) -> Result<Packet<'frame>>
- {
- let icmp_packet = Icmpv6Packet::new_checked(ip_payload)?;
- let checksum_caps = self.device_capabilities.checksum.clone();
- let icmp_repr = Icmpv6Repr::parse(&ip_repr.src_addr(), &ip_repr.dst_addr(),
- &icmp_packet, &checksum_caps)?;
- #[cfg(feature = "socket-icmp")]
- let mut handled_by_icmp_socket = false;
- #[cfg(all(feature = "socket-icmp", feature = "proto-ipv6"))]
- for mut icmp_socket in _sockets.iter_mut().filter_map(IcmpSocket::downcast) {
- if !icmp_socket.accepts(&ip_repr, &icmp_repr.into(), &checksum_caps) { continue }
- match icmp_socket.process(&ip_repr, &icmp_repr.into(), &checksum_caps) {
- // The packet is valid and handled by socket.
- Ok(()) => handled_by_icmp_socket = true,
- // The socket buffer is full.
- Err(Error::Exhausted) => (),
- // ICMP sockets don't validate the packets in any way.
- Err(_) => unreachable!(),
- }
- }
- match icmp_repr {
- // Respond to echo requests.
- Icmpv6Repr::EchoRequest { ident, seq_no, data } => {
- match ip_repr {
- IpRepr::Ipv6(ipv6_repr) => {
- let icmp_reply_repr = Icmpv6Repr::EchoReply {
- ident: ident,
- seq_no: seq_no,
- data: data
- };
- Ok(self.icmpv6_reply(ipv6_repr, icmp_reply_repr))
- },
- _ => Err(Error::Unrecognized),
- }
- }
- // Ignore any echo replies.
- Icmpv6Repr::EchoReply { .. } => Ok(Packet::None),
- // Forward any NDISC packets to the ndisc packet handler
- Icmpv6Repr::Ndisc(repr) if ip_repr.hop_limit() == 0xff => match ip_repr {
- IpRepr::Ipv6(ipv6_repr) => self.process_ndisc(timestamp, ipv6_repr, repr),
- _ => Ok(Packet::None)
- },
- // Don't report an error if a packet with unknown type
- // has been handled by an ICMP socket
- #[cfg(feature = "socket-icmp")]
- _ if handled_by_icmp_socket => Ok(Packet::None),
- // FIXME: do something correct here?
- _ => Err(Error::Unrecognized),
- }
- }
- #[cfg(feature = "proto-ipv6")]
- fn process_ndisc<'frame>(&mut self, timestamp: Instant, ip_repr: Ipv6Repr,
- repr: NdiscRepr<'frame>) -> Result<Packet<'frame>> {
- let packet = match repr {
- NdiscRepr::NeighborAdvert { lladdr, target_addr, flags } => {
- let ip_addr = ip_repr.src_addr.into();
- match lladdr {
- Some(lladdr) if lladdr.is_unicast() && target_addr.is_unicast() => {
- if flags.contains(NdiscNeighborFlags::OVERRIDE) {
- self.neighbor_cache.fill(ip_addr, lladdr, timestamp)
- } else {
- if self.neighbor_cache.lookup_pure(&ip_addr, timestamp).is_none() {
- self.neighbor_cache.fill(ip_addr, lladdr, timestamp)
- }
- }
- },
- _ => (),
- }
- Ok(Packet::None)
- }
- NdiscRepr::NeighborSolicit { target_addr, lladdr, .. } => {
- match lladdr {
- Some(lladdr) if lladdr.is_unicast() && target_addr.is_unicast() => {
- self.neighbor_cache.fill(ip_repr.src_addr.into(), lladdr, timestamp)
- },
- _ => (),
- }
- if self.has_solicited_node(ip_repr.dst_addr) && self.has_ip_addr(target_addr) {
- let advert = Icmpv6Repr::Ndisc(NdiscRepr::NeighborAdvert {
- flags: NdiscNeighborFlags::SOLICITED,
- target_addr: target_addr,
- lladdr: Some(self.ethernet_addr)
- });
- let ip_repr = Ipv6Repr {
- src_addr: target_addr,
- dst_addr: ip_repr.src_addr,
- next_header: IpProtocol::Icmpv6,
- hop_limit: 0xff,
- payload_len: advert.buffer_len()
- };
- Ok(Packet::Icmpv6((ip_repr, advert)))
- } else {
- Ok(Packet::None)
- }
- }
- _ => Ok(Packet::None)
- };
- packet
- }
- #[cfg(feature = "proto-ipv6")]
- fn process_hopbyhop<'frame>(&mut self, sockets: &mut SocketSet, timestamp: Instant,
- ipv6_repr: Ipv6Repr, handled_by_raw_socket: bool,
- ip_payload: &'frame [u8]) -> Result<Packet<'frame>>
- {
- let hbh_pkt = Ipv6HopByHopHeader::new_checked(ip_payload)?;
- let hbh_repr = Ipv6HopByHopRepr::parse(&hbh_pkt)?;
- for result in hbh_repr.options() {
- let opt_repr = result?;
- match opt_repr {
- Ipv6OptionRepr::Pad1 | Ipv6OptionRepr::PadN(_) => (),
- Ipv6OptionRepr::Unknown { type_, .. } => {
- match Ipv6OptionFailureType::from(type_) {
- Ipv6OptionFailureType::Skip => (),
- Ipv6OptionFailureType::Discard => {
- return Ok(Packet::None);
- },
- _ => {
- // FIXME(dlrobertson): Send an ICMPv6 parameter problem message
- // here.
- return Err(Error::Unrecognized);
- }
- }
- }
- _ => return Err(Error::Unrecognized),
- }
- }
- self.process_nxt_hdr(sockets, timestamp, ipv6_repr, hbh_repr.next_header,
- handled_by_raw_socket, &ip_payload[hbh_repr.buffer_len()..])
- }
- #[cfg(feature = "proto-ipv4")]
- fn process_icmpv4<'frame>(&self, _sockets: &mut SocketSet, ip_repr: IpRepr,
- ip_payload: &'frame [u8]) -> Result<Packet<'frame>>
- {
- let icmp_packet = Icmpv4Packet::new_checked(ip_payload)?;
- let checksum_caps = self.device_capabilities.checksum.clone();
- let icmp_repr = Icmpv4Repr::parse(&icmp_packet, &checksum_caps)?;
- #[cfg(feature = "socket-icmp")]
- let mut handled_by_icmp_socket = false;
- #[cfg(all(feature = "socket-icmp", feature = "proto-ipv4"))]
- for mut icmp_socket in _sockets.iter_mut().filter_map(IcmpSocket::downcast) {
- if !icmp_socket.accepts(&ip_repr, &icmp_repr.into(), &checksum_caps) { continue }
- match icmp_socket.process(&ip_repr, &icmp_repr.into(), &checksum_caps) {
- // The packet is valid and handled by socket.
- Ok(()) => handled_by_icmp_socket = true,
- // The socket buffer is full.
- Err(Error::Exhausted) => (),
- // ICMP sockets don't validate the packets in any way.
- Err(_) => unreachable!(),
- }
- }
- match icmp_repr {
- // Respond to echo requests.
- #[cfg(feature = "proto-ipv4")]
- Icmpv4Repr::EchoRequest { ident, seq_no, data } => {
- let icmp_reply_repr = Icmpv4Repr::EchoReply {
- ident: ident,
- seq_no: seq_no,
- data: data
- };
- match ip_repr {
- IpRepr::Ipv4(ipv4_repr) => Ok(self.icmpv4_reply(ipv4_repr, icmp_reply_repr)),
- _ => Err(Error::Unrecognized),
- }
- },
- // Ignore any echo replies.
- Icmpv4Repr::EchoReply { .. } => Ok(Packet::None),
- // Don't report an error if a packet with unknown type
- // has been handled by an ICMP socket
- #[cfg(feature = "socket-icmp")]
- _ if handled_by_icmp_socket => Ok(Packet::None),
- // FIXME: do something correct here?
- _ => Err(Error::Unrecognized),
- }
- }
- #[cfg(feature = "proto-ipv4")]
- fn icmpv4_reply<'frame, 'icmp: 'frame>
- (&self, ipv4_repr: Ipv4Repr, icmp_repr: Icmpv4Repr<'icmp>) ->
- Packet<'frame>
- {
- if !ipv4_repr.src_addr.is_unicast() {
- // Do not send ICMP replies to non-unicast sources
- Packet::None
- } else if ipv4_repr.dst_addr.is_unicast() {
- // Reply as normal when src_addr and dst_addr are both unicast
- let ipv4_reply_repr = Ipv4Repr {
- src_addr: ipv4_repr.dst_addr,
- dst_addr: ipv4_repr.src_addr,
- protocol: IpProtocol::Icmp,
- payload_len: icmp_repr.buffer_len(),
- hop_limit: 64
- };
- Packet::Icmpv4((ipv4_reply_repr, icmp_repr))
- } else if ipv4_repr.dst_addr.is_broadcast() {
- // Only reply to broadcasts for echo replies and not other ICMP messages
- match icmp_repr {
- Icmpv4Repr::EchoReply {..} => match self.ipv4_address() {
- Some(src_addr) => {
- let ipv4_reply_repr = Ipv4Repr {
- src_addr: src_addr,
- dst_addr: ipv4_repr.src_addr,
- protocol: IpProtocol::Icmp,
- payload_len: icmp_repr.buffer_len(),
- hop_limit: 64
- };
- Packet::Icmpv4((ipv4_reply_repr, icmp_repr))
- },
- None => Packet::None,
- },
- _ => Packet::None,
- }
- } else {
- Packet::None
- }
- }
- #[cfg(feature = "proto-ipv6")]
- fn icmpv6_reply<'frame, 'icmp: 'frame>
- (&self, ipv6_repr: Ipv6Repr, icmp_repr: Icmpv6Repr<'icmp>) ->
- Packet<'frame>
- {
- if ipv6_repr.dst_addr.is_unicast() {
- let ipv6_reply_repr = Ipv6Repr {
- src_addr: ipv6_repr.dst_addr,
- dst_addr: ipv6_repr.src_addr,
- next_header: IpProtocol::Icmpv6,
- payload_len: icmp_repr.buffer_len(),
- hop_limit: 64
- };
- Packet::Icmpv6((ipv6_reply_repr, icmp_repr))
- } else {
- // Do not send any ICMP replies to a broadcast destination address.
- Packet::None
- }
- }
- #[cfg(feature = "socket-udp")]
- fn process_udp<'frame>(&self, sockets: &mut SocketSet,
- ip_repr: IpRepr, ip_payload: &'frame [u8]) ->
- Result<Packet<'frame>>
- {
- let (src_addr, dst_addr) = (ip_repr.src_addr(), ip_repr.dst_addr());
- let udp_packet = UdpPacket::new_checked(ip_payload)?;
- let checksum_caps = self.device_capabilities.checksum.clone();
- let udp_repr = UdpRepr::parse(&udp_packet, &src_addr, &dst_addr, &checksum_caps)?;
- for mut udp_socket in sockets.iter_mut().filter_map(UdpSocket::downcast) {
- if !udp_socket.accepts(&ip_repr, &udp_repr) { continue }
- match udp_socket.process(&ip_repr, &udp_repr) {
- // The packet is valid and handled by socket.
- Ok(()) => return Ok(Packet::None),
- // The packet is malformed, or the socket buffer is full.
- Err(e) => return Err(e)
- }
- }
- // The packet wasn't handled by a socket, send an ICMP port unreachable packet.
- match ip_repr {
- #[cfg(feature = "proto-ipv4")]
- IpRepr::Ipv4(ipv4_repr) => {
- let payload_len = icmp_reply_payload_len(ip_payload.len(), IPV4_MIN_MTU,
- ipv4_repr.buffer_len());
- let icmpv4_reply_repr = Icmpv4Repr::DstUnreachable {
- reason: Icmpv4DstUnreachable::PortUnreachable,
- header: ipv4_repr,
- data: &ip_payload[0..payload_len]
- };
- Ok(self.icmpv4_reply(ipv4_repr, icmpv4_reply_repr))
- },
- #[cfg(feature = "proto-ipv6")]
- IpRepr::Ipv6(ipv6_repr) => {
- let payload_len = icmp_reply_payload_len(ip_payload.len(), IPV6_MIN_MTU,
- ipv6_repr.buffer_len());
- let icmpv6_reply_repr = Icmpv6Repr::DstUnreachable {
- reason: Icmpv6DstUnreachable::PortUnreachable,
- header: ipv6_repr,
- data: &ip_payload[0..payload_len]
- };
- Ok(self.icmpv6_reply(ipv6_repr, icmpv6_reply_repr))
- },
- IpRepr::Unspecified { .. } |
- IpRepr::__Nonexhaustive => Err(Error::Unaddressable),
- }
- }
- #[cfg(feature = "socket-tcp")]
- fn process_tcp<'frame>(&self, sockets: &mut SocketSet, timestamp: Instant,
- ip_repr: IpRepr, ip_payload: &'frame [u8]) ->
- Result<Packet<'frame>>
- {
- let (src_addr, dst_addr) = (ip_repr.src_addr(), ip_repr.dst_addr());
- let tcp_packet = TcpPacket::new_checked(ip_payload)?;
- let checksum_caps = self.device_capabilities.checksum.clone();
- let tcp_repr = TcpRepr::parse(&tcp_packet, &src_addr, &dst_addr, &checksum_caps)?;
- for mut tcp_socket in sockets.iter_mut().filter_map(TcpSocket::downcast) {
- if !tcp_socket.accepts(&ip_repr, &tcp_repr) { continue }
- match tcp_socket.process(timestamp, &ip_repr, &tcp_repr) {
- // The packet is valid and handled by socket.
- Ok(reply) => return Ok(reply.map_or(Packet::None, Packet::Tcp)),
- // The packet is malformed, or doesn't match the socket state,
- // or the socket buffer is full.
- Err(e) => return Err(e)
- }
- }
- if tcp_repr.control == TcpControl::Rst {
- // Never reply to a TCP RST packet with another TCP RST packet.
- Ok(Packet::None)
- } else {
- // The packet wasn't handled by a socket, send a TCP RST packet.
- Ok(Packet::Tcp(TcpSocket::rst_reply(&ip_repr, &tcp_repr)))
- }
- }
- fn dispatch<Tx>(&mut self, tx_token: Tx, timestamp: Instant,
- packet: Packet) -> Result<()>
- where Tx: TxToken
- {
- let checksum_caps = self.device_capabilities.checksum.clone();
- match packet {
- #[cfg(feature = "proto-ipv4")]
- Packet::Arp(arp_repr) => {
- let dst_hardware_addr =
- match arp_repr {
- ArpRepr::EthernetIpv4 { target_hardware_addr, .. } => target_hardware_addr,
- _ => unreachable!()
- };
- self.dispatch_ethernet(tx_token, timestamp, arp_repr.buffer_len(), |mut frame| {
- frame.set_dst_addr(dst_hardware_addr);
- frame.set_ethertype(EthernetProtocol::Arp);
- let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
- arp_repr.emit(&mut packet);
- })
- },
- #[cfg(feature = "proto-ipv4")]
- Packet::Icmpv4((ipv4_repr, icmpv4_repr)) => {
- self.dispatch_ip(tx_token, timestamp, IpRepr::Ipv4(ipv4_repr),
- |_ip_repr, payload| {
- icmpv4_repr.emit(&mut Icmpv4Packet::new_unchecked(payload), &checksum_caps);
- })
- }
- #[cfg(feature = "proto-igmp")]
- Packet::Igmp((ipv4_repr, igmp_repr)) => {
- self.dispatch_ip(tx_token, timestamp, IpRepr::Ipv4(ipv4_repr), |_ip_repr, payload| {
- igmp_repr.emit(&mut IgmpPacket::new_unchecked(payload));
- })
- }
- #[cfg(feature = "proto-ipv6")]
- Packet::Icmpv6((ipv6_repr, icmpv6_repr)) => {
- self.dispatch_ip(tx_token, timestamp, IpRepr::Ipv6(ipv6_repr),
- |ip_repr, payload| {
- icmpv6_repr.emit(&ip_repr.src_addr(), &ip_repr.dst_addr(),
- &mut Icmpv6Packet::new_unchecked(payload), &checksum_caps);
- })
- }
- #[cfg(feature = "socket-raw")]
- Packet::Raw((ip_repr, raw_packet)) => {
- self.dispatch_ip(tx_token, timestamp, ip_repr, |_ip_repr, payload| {
- payload.copy_from_slice(raw_packet);
- })
- }
- #[cfg(feature = "socket-udp")]
- Packet::Udp((ip_repr, udp_repr)) => {
- self.dispatch_ip(tx_token, timestamp, ip_repr, |ip_repr, payload| {
- udp_repr.emit(&mut UdpPacket::new_unchecked(payload),
- &ip_repr.src_addr(), &ip_repr.dst_addr(),
- &checksum_caps);
- })
- }
- #[cfg(feature = "socket-tcp")]
- Packet::Tcp((ip_repr, mut tcp_repr)) => {
- let caps = self.device_capabilities.clone();
- self.dispatch_ip(tx_token, timestamp, ip_repr, |ip_repr, payload| {
- // This is a terrible hack to make TCP performance more acceptable on systems
- // where the TCP buffers are significantly larger than network buffers,
- // e.g. a 64 kB TCP receive buffer (and so, when empty, a 64k window)
- // together with four 1500 B Ethernet receive buffers. If left untreated,
- // this would result in our peer pushing our window and sever packet loss.
- //
- // I'm really not happy about this "solution" but I don't know what else to do.
- if let Some(max_burst_size) = caps.max_burst_size {
- let mut max_segment_size = caps.max_transmission_unit;
- max_segment_size -= EthernetFrame::<&[u8]>::header_len();
- max_segment_size -= ip_repr.buffer_len();
- max_segment_size -= tcp_repr.header_len();
- let max_window_size = max_burst_size * max_segment_size;
- if tcp_repr.window_len as usize > max_window_size {
- tcp_repr.window_len = max_window_size as u16;
- }
- }
- tcp_repr.emit(&mut TcpPacket::new_unchecked(payload),
- &ip_repr.src_addr(), &ip_repr.dst_addr(),
- &checksum_caps);
- })
- }
- Packet::None => Ok(())
- }
- }
- fn dispatch_ethernet<Tx, F>(&mut self, tx_token: Tx, timestamp: Instant,
- buffer_len: usize, f: F) -> Result<()>
- where Tx: TxToken, F: FnOnce(EthernetFrame<&mut [u8]>)
- {
- let tx_len = EthernetFrame::<&[u8]>::buffer_len(buffer_len);
- tx_token.consume(timestamp, tx_len, |tx_buffer| {
- debug_assert!(tx_buffer.as_ref().len() == tx_len);
- let mut frame = EthernetFrame::new_unchecked(tx_buffer.as_mut());
- frame.set_src_addr(self.ethernet_addr);
- f(frame);
- Ok(())
- })
- }
- fn in_same_network(&self, addr: &IpAddress) -> bool {
- self.ip_addrs
- .iter()
- .find(|cidr| cidr.contains_addr(addr))
- .is_some()
- }
- fn route(&self, addr: &IpAddress, timestamp: Instant) -> Result<IpAddress> {
- // Send directly.
- if self.in_same_network(addr) || addr.is_broadcast() {
- return Ok(*addr)
- }
- // Route via a router.
- match self.routes.lookup(addr, timestamp) {
- Some(router_addr) => Ok(router_addr),
- None => Err(Error::Unaddressable),
- }
- }
- fn has_neighbor<'a>(&self, addr: &'a IpAddress, timestamp: Instant) -> bool {
- match self.route(addr, timestamp) {
- Ok(routed_addr) => {
- self.neighbor_cache
- .lookup_pure(&routed_addr, timestamp)
- .is_some()
- }
- Err(_) => false
- }
- }
- fn lookup_hardware_addr<Tx>(&mut self, tx_token: Tx, timestamp: Instant,
- src_addr: &IpAddress, dst_addr: &IpAddress) ->
- Result<(EthernetAddress, Tx)>
- where Tx: TxToken
- {
- if dst_addr.is_multicast() {
- let b = dst_addr.as_bytes();
- let hardware_addr =
- match dst_addr {
- &IpAddress::Unspecified =>
- None,
- #[cfg(feature = "proto-ipv4")]
- &IpAddress::Ipv4(_addr) =>
- Some(EthernetAddress::from_bytes(&[
- 0x01, 0x00,
- 0x5e, b[1] & 0x7F,
- b[2], b[3],
- ])),
- #[cfg(feature = "proto-ipv6")]
- &IpAddress::Ipv6(_addr) =>
- Some(EthernetAddress::from_bytes(&[
- 0x33, 0x33,
- b[12], b[13],
- b[14], b[15],
- ])),
- &IpAddress::__Nonexhaustive =>
- unreachable!()
- };
- match hardware_addr {
- Some(hardware_addr) =>
- // Destination is multicast
- return Ok((hardware_addr, tx_token)),
- None =>
- // Continue
- (),
- }
- }
- let dst_addr = self.route(dst_addr, timestamp)?;
- match self.neighbor_cache.lookup(&dst_addr, timestamp) {
- NeighborAnswer::Found(hardware_addr) =>
- return Ok((hardware_addr, tx_token)),
- NeighborAnswer::RateLimited =>
- return Err(Error::Unaddressable),
- NeighborAnswer::NotFound => (),
- }
- match (src_addr, dst_addr) {
- #[cfg(feature = "proto-ipv4")]
- (&IpAddress::Ipv4(src_addr), IpAddress::Ipv4(dst_addr)) => {
- net_debug!("address {} not in neighbor cache, sending ARP request",
- dst_addr);
- let arp_repr = ArpRepr::EthernetIpv4 {
- operation: ArpOperation::Request,
- source_hardware_addr: self.ethernet_addr,
- source_protocol_addr: src_addr,
- target_hardware_addr: EthernetAddress::BROADCAST,
- target_protocol_addr: dst_addr,
- };
- self.dispatch_ethernet(tx_token, timestamp, arp_repr.buffer_len(), |mut frame| {
- frame.set_dst_addr(EthernetAddress::BROADCAST);
- frame.set_ethertype(EthernetProtocol::Arp);
- arp_repr.emit(&mut ArpPacket::new_unchecked(frame.payload_mut()))
- })?;
- Err(Error::Unaddressable)
- }
- #[cfg(feature = "proto-ipv6")]
- (&IpAddress::Ipv6(src_addr), IpAddress::Ipv6(dst_addr)) => {
- net_debug!("address {} not in neighbor cache, sending Neighbor Solicitation",
- dst_addr);
- let checksum_caps = self.device_capabilities.checksum.clone();
- let solicit = Icmpv6Repr::Ndisc(NdiscRepr::NeighborSolicit {
- target_addr: src_addr,
- lladdr: Some(self.ethernet_addr),
- });
- let ip_repr = IpRepr::Ipv6(Ipv6Repr {
- src_addr: src_addr,
- dst_addr: dst_addr.solicited_node(),
- next_header: IpProtocol::Icmpv6,
- payload_len: solicit.buffer_len(),
- hop_limit: 0xff
- });
- self.dispatch_ip(tx_token, timestamp, ip_repr, |ip_repr, payload| {
- solicit.emit(&ip_repr.src_addr(), &ip_repr.dst_addr(),
- &mut Icmpv6Packet::new_unchecked(payload), &checksum_caps);
- })?;
- Err(Error::Unaddressable)
- }
- _ => Err(Error::Unaddressable)
- }
- }
- fn dispatch_ip<Tx, F>(&mut self, tx_token: Tx, timestamp: Instant,
- ip_repr: IpRepr, f: F) -> Result<()>
- where Tx: TxToken, F: FnOnce(IpRepr, &mut [u8])
- {
- let ip_repr = ip_repr.lower(&self.ip_addrs)?;
- let checksum_caps = self.device_capabilities.checksum.clone();
- let (dst_hardware_addr, tx_token) =
- self.lookup_hardware_addr(tx_token, timestamp,
- &ip_repr.src_addr(), &ip_repr.dst_addr())?;
- self.dispatch_ethernet(tx_token, timestamp, ip_repr.total_len(), |mut frame| {
- frame.set_dst_addr(dst_hardware_addr);
- match ip_repr {
- #[cfg(feature = "proto-ipv4")]
- IpRepr::Ipv4(_) => frame.set_ethertype(EthernetProtocol::Ipv4),
- #[cfg(feature = "proto-ipv6")]
- IpRepr::Ipv6(_) => frame.set_ethertype(EthernetProtocol::Ipv6),
- _ => return
- }
- ip_repr.emit(frame.payload_mut(), &checksum_caps);
- let payload = &mut frame.payload_mut()[ip_repr.buffer_len()..];
- f(ip_repr, payload)
- })
- }
- #[cfg(feature = "proto-igmp")]
- fn igmp_report_packet<'any>(&self, version: IgmpVersion, group_addr: Ipv4Address) -> Option<Packet<'any>> {
- let iface_addr = self.ipv4_address()?;
- let igmp_repr = IgmpRepr::MembershipReport {
- group_addr,
- version,
- };
- let pkt = Packet::Igmp((Ipv4Repr {
- src_addr: iface_addr,
- // Send to the group being reported
- dst_addr: group_addr,
- protocol: IpProtocol::Igmp,
- payload_len: igmp_repr.buffer_len(),
- hop_limit: 1,
- // TODO: add Router Alert IPv4 header option. See
- // [#183](https://github.com/m-labs/smoltcp/issues/183).
- }, igmp_repr));
- Some(pkt)
- }
- #[cfg(feature = "proto-igmp")]
- fn igmp_leave_packet<'any>(&self, group_addr: Ipv4Address) -> Option<Packet<'any>> {
- self.ipv4_address().map(|iface_addr| {
- let igmp_repr = IgmpRepr::LeaveGroup { group_addr };
- let pkt = Packet::Igmp((Ipv4Repr {
- src_addr: iface_addr,
- dst_addr: Ipv4Address::MULTICAST_ALL_ROUTERS,
- protocol: IpProtocol::Igmp,
- payload_len: igmp_repr.buffer_len(),
- hop_limit: 1,
- }, igmp_repr));
- pkt
- })
- }
- }
- #[cfg(test)]
- mod test {
- #[cfg(feature = "proto-igmp")]
- use std::vec::Vec;
- use std::collections::BTreeMap;
- use {Result, Error};
- use super::InterfaceBuilder;
- use iface::{NeighborCache, EthernetInterface};
- use phy::{self, Loopback, ChecksumCapabilities};
- #[cfg(feature = "proto-igmp")]
- use phy::{Device, RxToken, TxToken};
- use time::Instant;
- use socket::SocketSet;
- #[cfg(feature = "proto-ipv4")]
- use wire::{ArpOperation, ArpPacket, ArpRepr};
- use wire::{EthernetAddress, EthernetFrame, EthernetProtocol};
- use wire::{IpAddress, IpCidr, IpProtocol, IpRepr};
- #[cfg(feature = "proto-ipv4")]
- use wire::{Ipv4Address, Ipv4Repr};
- #[cfg(feature = "proto-igmp")]
- use wire::Ipv4Packet;
- #[cfg(feature = "proto-ipv4")]
- use wire::{Icmpv4Repr, Icmpv4DstUnreachable};
- #[cfg(feature = "proto-igmp")]
- use wire::{IgmpPacket, IgmpRepr, IgmpVersion};
- #[cfg(all(feature = "socket-udp", any(feature = "proto-ipv4", feature = "proto-ipv6")))]
- use wire::{UdpPacket, UdpRepr};
- #[cfg(feature = "proto-ipv6")]
- use wire::{Ipv6Address, Ipv6Repr};
- #[cfg(feature = "proto-ipv6")]
- use wire::{Icmpv6Packet, Icmpv6Repr, Icmpv6ParamProblem};
- #[cfg(feature = "proto-ipv6")]
- use wire::{NdiscNeighborFlags, NdiscRepr};
- #[cfg(feature = "proto-ipv6")]
- use wire::{Ipv6HopByHopHeader, Ipv6Option, Ipv6OptionRepr};
- use super::Packet;
- fn create_loopback<'a, 'b, 'c>() -> (EthernetInterface<'static, 'b, 'c, Loopback>,
- SocketSet<'static, 'a, 'b>) {
- // Create a basic device
- let device = Loopback::new();
- let ip_addrs = [
- #[cfg(feature = "proto-ipv4")]
- IpCidr::new(IpAddress::v4(127, 0, 0, 1), 8),
- #[cfg(feature = "proto-ipv6")]
- IpCidr::new(IpAddress::v6(0, 0, 0, 0, 0, 0, 0, 1), 128),
- #[cfg(feature = "proto-ipv6")]
- IpCidr::new(IpAddress::v6(0xfdbe, 0, 0, 0, 0, 0, 0, 1), 64),
- ];
- let iface_builder = InterfaceBuilder::new(device)
- .ethernet_addr(EthernetAddress::default())
- .neighbor_cache(NeighborCache::new(BTreeMap::new()))
- .ip_addrs(ip_addrs);
- #[cfg(feature = "proto-igmp")]
- let iface_builder = iface_builder
- .ipv4_multicast_groups(BTreeMap::new());
- let iface = iface_builder
- .finalize();
- (iface, SocketSet::new(vec![]))
- }
- #[cfg(feature = "proto-igmp")]
- fn recv_all<'b>(iface: &mut EthernetInterface<'static, 'b, 'static, Loopback>, timestamp: Instant) -> Vec<Vec<u8>> {
- let mut pkts = Vec::new();
- while let Some((rx, _tx)) = iface.device.receive() {
- rx.consume(timestamp, |pkt| {
- pkts.push(pkt.iter().cloned().collect());
- Ok(())
- }).unwrap();
- }
- pkts
- }
- #[derive(Debug, PartialEq)]
- struct MockTxToken;
- impl phy::TxToken for MockTxToken {
- fn consume<R, F>(self, _: Instant, _: usize, _: F) -> Result<R>
- where F: FnOnce(&mut [u8]) -> Result<R> {
- Err(Error::__Nonexhaustive)
- }
- }
- #[test]
- #[should_panic(expected = "a required option was not set")]
- fn test_builder_initialization_panic() {
- InterfaceBuilder::new(Loopback::new()).finalize();
- }
- #[test]
- fn test_no_icmp_no_unicast() {
- let (mut iface, mut socket_set) = create_loopback();
- let mut eth_bytes = vec![0u8; 54];
- // Unknown Ipv4 Protocol
- //
- // Because the destination is the broadcast address
- // this should not trigger and Destination Unreachable
- // response. See RFC 1122 § 3.2.2.
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- let repr = IpRepr::Ipv4(Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- dst_addr: Ipv4Address::BROADCAST,
- protocol: IpProtocol::Unknown(0x0c),
- payload_len: 0,
- hop_limit: 0x40
- });
- #[cfg(feature = "proto-ipv6")]
- let repr = IpRepr::Ipv6(Ipv6Repr {
- src_addr: Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1),
- dst_addr: Ipv6Address::LINK_LOCAL_ALL_NODES,
- next_header: IpProtocol::Unknown(0x0c),
- payload_len: 0,
- hop_limit: 0x40
- });
- let frame = {
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- frame.set_dst_addr(EthernetAddress::BROADCAST);
- frame.set_src_addr(EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]));
- frame.set_ethertype(EthernetProtocol::Ipv4);
- repr.emit(frame.payload_mut(), &ChecksumCapabilities::default());
- EthernetFrame::new_unchecked(&*frame.into_inner())
- };
- // Ensure that the unknown protocol frame does not trigger an
- // ICMP error response when the destination address is a
- // broadcast address
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- assert_eq!(iface.inner.process_ipv4(&mut socket_set, Instant::from_millis(0), &frame),
- Ok(Packet::None));
- #[cfg(feature = "proto-ipv6")]
- assert_eq!(iface.inner.process_ipv6(&mut socket_set, Instant::from_millis(0), &frame),
- Ok(Packet::None));
- }
- #[test]
- #[cfg(feature = "proto-ipv4")]
- fn test_icmp_error_no_payload() {
- static NO_BYTES: [u8; 0] = [];
- let (mut iface, mut socket_set) = create_loopback();
- let mut eth_bytes = vec![0u8; 34];
- // Unknown Ipv4 Protocol with no payload
- let repr = IpRepr::Ipv4(Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- protocol: IpProtocol::Unknown(0x0c),
- payload_len: 0,
- hop_limit: 0x40
- });
- // emit the above repr to a frame
- let frame = {
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- frame.set_dst_addr(EthernetAddress([0x00, 0x00, 0x00, 0x00, 0x00, 0x00]));
- frame.set_src_addr(EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]));
- frame.set_ethertype(EthernetProtocol::Ipv4);
- repr.emit(frame.payload_mut(), &ChecksumCapabilities::default());
- EthernetFrame::new_unchecked(&*frame.into_inner())
- };
- // The expected Destination Unreachable response due to the
- // unknown protocol
- let icmp_repr = Icmpv4Repr::DstUnreachable {
- reason: Icmpv4DstUnreachable::ProtoUnreachable,
- header: Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- protocol: IpProtocol::Unknown(12),
- payload_len: 0,
- hop_limit: 64
- },
- data: &NO_BYTES
- };
- let expected_repr = Packet::Icmpv4((
- Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- protocol: IpProtocol::Icmp,
- payload_len: icmp_repr.buffer_len(),
- hop_limit: 64
- },
- icmp_repr
- ));
- // Ensure that the unknown protocol triggers an error response.
- // And we correctly handle no payload.
- assert_eq!(iface.inner.process_ipv4(&mut socket_set, Instant::from_millis(0), &frame),
- Ok(expected_repr));
- }
- #[test]
- #[cfg(all(feature = "socket-udp", feature = "proto-ipv4"))]
- fn test_icmp_error_port_unreachable() {
- static UDP_PAYLOAD: [u8; 12] = [
- 0x48, 0x65, 0x6c, 0x6c,
- 0x6f, 0x2c, 0x20, 0x57,
- 0x6f, 0x6c, 0x64, 0x21
- ];
- let (iface, mut socket_set) = create_loopback();
- let mut udp_bytes_unicast = vec![0u8; 20];
- let mut udp_bytes_broadcast = vec![0u8; 20];
- let mut packet_unicast = UdpPacket::new_unchecked(&mut udp_bytes_unicast);
- let mut packet_broadcast = UdpPacket::new_unchecked(&mut udp_bytes_broadcast);
- let udp_repr = UdpRepr {
- src_port: 67,
- dst_port: 68,
- payload: &UDP_PAYLOAD
- };
- let ip_repr = IpRepr::Ipv4(Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- protocol: IpProtocol::Udp,
- payload_len: udp_repr.buffer_len(),
- hop_limit: 64
- });
- // Emit the representations to a packet
- udp_repr.emit(&mut packet_unicast, &ip_repr.src_addr(),
- &ip_repr.dst_addr(), &ChecksumCapabilities::default());
- let data = packet_unicast.into_inner();
- // The expected Destination Unreachable ICMPv4 error response due
- // to no sockets listening on the destination port.
- let icmp_repr = Icmpv4Repr::DstUnreachable {
- reason: Icmpv4DstUnreachable::PortUnreachable,
- header: Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- protocol: IpProtocol::Udp,
- payload_len: udp_repr.buffer_len(),
- hop_limit: 64
- },
- data: &data
- };
- let expected_repr = Packet::Icmpv4((
- Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
- dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- protocol: IpProtocol::Icmp,
- payload_len: icmp_repr.buffer_len(),
- hop_limit: 64
- },
- icmp_repr
- ));
- // Ensure that the unknown protocol triggers an error response.
- // And we correctly handle no payload.
- assert_eq!(iface.inner.process_udp(&mut socket_set, ip_repr, data),
- Ok(expected_repr));
- let ip_repr = IpRepr::Ipv4(Ipv4Repr {
- src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
- dst_addr: Ipv4Address::BROADCAST,
- protocol: IpProtocol::Udp,
- payload_len: udp_repr.buffer_len(),
- hop_limit: 64
- });
- // Emit the representations to a packet
- udp_repr.emit(&mut packet_broadcast, &ip_repr.src_addr(),
- &IpAddress::Ipv4(Ipv4Address::BROADCAST),
- &ChecksumCapabilities::default());
- // Ensure that the port unreachable error does not trigger an
- // ICMP error response when the destination address is a
- // broadcast address and no socket is bound to the port.
- assert_eq!(iface.inner.process_udp(&mut socket_set, ip_repr,
- packet_broadcast.into_inner()), Ok(Packet::None));
- }
- #[test]
- #[cfg(feature = "socket-udp")]
- fn test_handle_udp_broadcast() {
- use socket::{UdpSocket, UdpSocketBuffer, UdpPacketMetadata};
- use wire::IpEndpoint;
- static UDP_PAYLOAD: [u8; 5] = [0x48, 0x65, 0x6c, 0x6c, 0x6f];
- let (iface, mut socket_set) = create_loopback();
- let rx_buffer = UdpSocketBuffer::new(vec![UdpPacketMetadata::EMPTY], vec![0; 15]);
- let tx_buffer = UdpSocketBuffer::new(vec![UdpPacketMetadata::EMPTY], vec![0; 15]);
- let udp_socket = UdpSocket::new(rx_buffer, tx_buffer);
- let mut udp_bytes = vec![0u8; 13];
- let mut packet = UdpPacket::new_unchecked(&mut udp_bytes);
- let socket_handle = socket_set.add(udp_socket);
- #[cfg(feature = "proto-ipv6")]
- let src_ip = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1);
- #[cfg(all(not(feature = "proto-ipv6"), feature = "proto-ipv4"))]
- let src_ip = Ipv4Address::new(0x7f, 0x00, 0x00, 0x02);
- let udp_repr = UdpRepr {
- src_port: 67,
- dst_port: 68,
- payload: &UDP_PAYLOAD
- };
- #[cfg(feature = "proto-ipv6")]
- let ip_repr = IpRepr::Ipv6(Ipv6Repr {
- src_addr: src_ip,
- dst_addr: Ipv6Address::LINK_LOCAL_ALL_NODES,
- next_header: IpProtocol::Udp,
- payload_len: udp_repr.buffer_len(),
- hop_limit: 0x40
- });
- #[cfg(all(not(feature = "proto-ipv6"), feature = "proto-ipv4"))]
- let ip_repr = IpRepr::Ipv4(Ipv4Repr {
- src_addr: src_ip,
- dst_addr: Ipv4Address::BROADCAST,
- protocol: IpProtocol::Udp,
- payload_len: udp_repr.buffer_len(),
- hop_limit: 0x40
- });
- {
- // Bind the socket to port 68
- let mut socket = socket_set.get::<UdpSocket>(socket_handle);
- assert_eq!(socket.bind(68), Ok(()));
- assert!(!socket.can_recv());
- assert!(socket.can_send());
- }
- udp_repr.emit(&mut packet, &ip_repr.src_addr(), &ip_repr.dst_addr(),
- &ChecksumCapabilities::default());
- // Packet should be handled by bound UDP socket
- assert_eq!(iface.inner.process_udp(&mut socket_set, ip_repr, packet.into_inner()),
- Ok(Packet::None));
- {
- // Make sure the payload to the UDP packet processed by process_udp is
- // appended to the bound sockets rx_buffer
- let mut socket = socket_set.get::<UdpSocket>(socket_handle);
- assert!(socket.can_recv());
- assert_eq!(socket.recv(), Ok((&UDP_PAYLOAD[..], IpEndpoint::new(src_ip.into(), 67))));
- }
- }
- #[test]
- #[cfg(feature = "proto-ipv4")]
- fn test_handle_ipv4_broadcast() {
- use wire::{Ipv4Packet, Icmpv4Repr, Icmpv4Packet};
- let (mut iface, mut socket_set) = create_loopback();
- let our_ipv4_addr = iface.ipv4_address().unwrap();
- let src_ipv4_addr = Ipv4Address([127, 0, 0, 2]);
- // ICMPv4 echo request
- let icmpv4_data: [u8; 4] = [0xaa, 0x00, 0x00, 0xff];
- let icmpv4_repr = Icmpv4Repr::EchoRequest {
- ident: 0x1234, seq_no: 0xabcd, data: &icmpv4_data
- };
- // Send to IPv4 broadcast address
- let ipv4_repr = Ipv4Repr {
- src_addr: src_ipv4_addr,
- dst_addr: Ipv4Address::BROADCAST,
- protocol: IpProtocol::Icmp,
- hop_limit: 64,
- payload_len: icmpv4_repr.buffer_len(),
- };
- // Emit to ethernet frame
- let mut eth_bytes = vec![0u8;
- EthernetFrame::<&[u8]>::header_len() +
- ipv4_repr.buffer_len() + icmpv4_repr.buffer_len()
- ];
- let frame = {
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- ipv4_repr.emit(
- &mut Ipv4Packet::new_unchecked(frame.payload_mut()),
- &ChecksumCapabilities::default());
- icmpv4_repr.emit(
- &mut Icmpv4Packet::new_unchecked(
- &mut frame.payload_mut()[ipv4_repr.buffer_len()..]),
- &ChecksumCapabilities::default());
- EthernetFrame::new_unchecked(&*frame.into_inner())
- };
- // Expected ICMPv4 echo reply
- let expected_icmpv4_repr = Icmpv4Repr::EchoReply {
- ident: 0x1234, seq_no: 0xabcd, data: &icmpv4_data };
- let expected_ipv4_repr = Ipv4Repr {
- src_addr: our_ipv4_addr,
- dst_addr: src_ipv4_addr,
- protocol: IpProtocol::Icmp,
- hop_limit: 64,
- payload_len: expected_icmpv4_repr.buffer_len(),
- };
- let expected_packet = Packet::Icmpv4((expected_ipv4_repr, expected_icmpv4_repr));
- assert_eq!(iface.inner.process_ipv4(&mut socket_set, Instant::from_millis(0), &frame),
- Ok(expected_packet));
- }
- #[test]
- #[cfg(feature = "socket-udp")]
- fn test_icmp_reply_size() {
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- use wire::IPV4_MIN_MTU as MIN_MTU;
- #[cfg(feature = "proto-ipv6")]
- use wire::Icmpv6DstUnreachable;
- #[cfg(feature = "proto-ipv6")]
- use wire::IPV6_MIN_MTU as MIN_MTU;
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- const MAX_PAYLOAD_LEN: usize = 528;
- #[cfg(feature = "proto-ipv6")]
- const MAX_PAYLOAD_LEN: usize = 1192;
- let (iface, mut socket_set) = create_loopback();
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- let src_addr = Ipv4Address([192, 168, 1, 1]);
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- let dst_addr = Ipv4Address([192, 168, 1, 2]);
- #[cfg(feature = "proto-ipv6")]
- let src_addr = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1);
- #[cfg(feature = "proto-ipv6")]
- let dst_addr = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 2);
- // UDP packet that if not tructated will cause a icmp port unreachable reply
- // to exeed the minimum mtu bytes in length.
- let udp_repr = UdpRepr {
- src_port: 67,
- dst_port: 68,
- payload: &[0x2a; MAX_PAYLOAD_LEN]
- };
- let mut bytes = vec![0xff; udp_repr.buffer_len()];
- let mut packet = UdpPacket::new_unchecked(&mut bytes[..]);
- udp_repr.emit(&mut packet, &src_addr.into(), &dst_addr.into(), &ChecksumCapabilities::default());
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- let ip_repr = Ipv4Repr {
- src_addr: src_addr,
- dst_addr: dst_addr,
- protocol: IpProtocol::Udp,
- hop_limit: 64,
- payload_len: udp_repr.buffer_len()
- };
- #[cfg(feature = "proto-ipv6")]
- let ip_repr = Ipv6Repr {
- src_addr: src_addr,
- dst_addr: dst_addr,
- next_header: IpProtocol::Udp,
- hop_limit: 64,
- payload_len: udp_repr.buffer_len()
- };
- let payload = packet.into_inner();
- // Expected packets
- #[cfg(feature = "proto-ipv6")]
- let expected_icmp_repr = Icmpv6Repr::DstUnreachable {
- reason: Icmpv6DstUnreachable::PortUnreachable,
- header: ip_repr,
- data: &payload[..MAX_PAYLOAD_LEN]
- };
- #[cfg(feature = "proto-ipv6")]
- let expected_ip_repr = Ipv6Repr {
- src_addr: dst_addr,
- dst_addr: src_addr,
- next_header: IpProtocol::Icmpv6,
- hop_limit: 64,
- payload_len: expected_icmp_repr.buffer_len()
- };
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- let expected_icmp_repr = Icmpv4Repr::DstUnreachable {
- reason: Icmpv4DstUnreachable::PortUnreachable,
- header: ip_repr,
- data: &payload[..MAX_PAYLOAD_LEN]
- };
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- let expected_ip_repr = Ipv4Repr {
- src_addr: dst_addr,
- dst_addr: src_addr,
- protocol: IpProtocol::Icmp,
- hop_limit: 64,
- payload_len: expected_icmp_repr.buffer_len()
- };
- // The expected packet does not exceed the IPV4_MIN_MTU
- assert_eq!(expected_ip_repr.buffer_len() + expected_icmp_repr.buffer_len(), MIN_MTU);
- // The expected packet and the generated packet are equal
- #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
- assert_eq!(iface.inner.process_udp(&mut socket_set, ip_repr.into(), payload),
- Ok(Packet::Icmpv4((expected_ip_repr, expected_icmp_repr))));
- #[cfg(feature = "proto-ipv6")]
- assert_eq!(iface.inner.process_udp(&mut socket_set, ip_repr.into(), payload),
- Ok(Packet::Icmpv6((expected_ip_repr, expected_icmp_repr))));
- }
- #[test]
- #[cfg(feature = "proto-ipv4")]
- fn test_handle_valid_arp_request() {
- let (mut iface, mut socket_set) = create_loopback();
- let mut eth_bytes = vec![0u8; 42];
- let local_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x01]);
- let remote_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x02]);
- let local_hw_addr = EthernetAddress([0x00, 0x00, 0x00, 0x00, 0x00, 0x00]);
- let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
- let repr = ArpRepr::EthernetIpv4 {
- operation: ArpOperation::Request,
- source_hardware_addr: remote_hw_addr,
- source_protocol_addr: remote_ip_addr,
- target_hardware_addr: EthernetAddress::default(),
- target_protocol_addr: local_ip_addr,
- };
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- frame.set_dst_addr(EthernetAddress::BROADCAST);
- frame.set_src_addr(remote_hw_addr);
- frame.set_ethertype(EthernetProtocol::Arp);
- {
- let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
- repr.emit(&mut packet);
- }
- // Ensure an ARP Request for us triggers an ARP Reply
- assert_eq!(iface.inner.process_ethernet(&mut socket_set, Instant::from_millis(0), frame.into_inner()),
- Ok(Packet::Arp(ArpRepr::EthernetIpv4 {
- operation: ArpOperation::Reply,
- source_hardware_addr: local_hw_addr,
- source_protocol_addr: local_ip_addr,
- target_hardware_addr: remote_hw_addr,
- target_protocol_addr: remote_ip_addr
- })));
- // Ensure the address of the requestor was entered in the cache
- assert_eq!(iface.inner.lookup_hardware_addr(MockTxToken, Instant::from_secs(0),
- &IpAddress::Ipv4(local_ip_addr), &IpAddress::Ipv4(remote_ip_addr)),
- Ok((remote_hw_addr, MockTxToken)));
- }
- #[test]
- #[cfg(feature = "proto-ipv6")]
- fn test_handle_valid_ndisc_request() {
- let (mut iface, mut socket_set) = create_loopback();
- let mut eth_bytes = vec![0u8; 86];
- let local_ip_addr = Ipv6Address::new(0xfdbe, 0, 0, 0, 0, 0, 0, 1);
- let remote_ip_addr = Ipv6Address::new(0xfdbe, 0, 0, 0, 0, 0, 0, 2);
- let local_hw_addr = EthernetAddress([0x00, 0x00, 0x00, 0x00, 0x00, 0x00]);
- let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
- let solicit = Icmpv6Repr::Ndisc(NdiscRepr::NeighborSolicit {
- target_addr: local_ip_addr,
- lladdr: Some(remote_hw_addr),
- });
- let ip_repr = IpRepr::Ipv6(Ipv6Repr {
- src_addr: remote_ip_addr,
- dst_addr: local_ip_addr.solicited_node(),
- next_header: IpProtocol::Icmpv6,
- hop_limit: 0xff,
- payload_len: solicit.buffer_len()
- });
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- frame.set_dst_addr(EthernetAddress([0x33, 0x33, 0x00, 0x00, 0x00, 0x00]));
- frame.set_src_addr(remote_hw_addr);
- frame.set_ethertype(EthernetProtocol::Ipv6);
- {
- ip_repr.emit(frame.payload_mut(), &ChecksumCapabilities::default());
- solicit.emit(&remote_ip_addr.into(), &local_ip_addr.solicited_node().into(),
- &mut Icmpv6Packet::new_unchecked(
- &mut frame.payload_mut()[ip_repr.buffer_len()..]),
- &ChecksumCapabilities::default());
- }
- let icmpv6_expected = Icmpv6Repr::Ndisc(NdiscRepr::NeighborAdvert {
- flags: NdiscNeighborFlags::SOLICITED,
- target_addr: local_ip_addr,
- lladdr: Some(local_hw_addr)
- });
- let ipv6_expected = Ipv6Repr {
- src_addr: local_ip_addr,
- dst_addr: remote_ip_addr,
- next_header: IpProtocol::Icmpv6,
- hop_limit: 0xff,
- payload_len: icmpv6_expected.buffer_len()
- };
- // Ensure an Neighbor Solicitation triggers a Neighbor Advertisement
- assert_eq!(iface.inner.process_ethernet(&mut socket_set, Instant::from_millis(0), frame.into_inner()),
- Ok(Packet::Icmpv6((ipv6_expected, icmpv6_expected))));
- // Ensure the address of the requestor was entered in the cache
- assert_eq!(iface.inner.lookup_hardware_addr(MockTxToken, Instant::from_secs(0),
- &IpAddress::Ipv6(local_ip_addr), &IpAddress::Ipv6(remote_ip_addr)),
- Ok((remote_hw_addr, MockTxToken)));
- }
- #[test]
- #[cfg(feature = "proto-ipv4")]
- fn test_handle_other_arp_request() {
- let (mut iface, mut socket_set) = create_loopback();
- let mut eth_bytes = vec![0u8; 42];
- let remote_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x02]);
- let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
- let repr = ArpRepr::EthernetIpv4 {
- operation: ArpOperation::Request,
- source_hardware_addr: remote_hw_addr,
- source_protocol_addr: remote_ip_addr,
- target_hardware_addr: EthernetAddress::default(),
- target_protocol_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x03]),
- };
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- frame.set_dst_addr(EthernetAddress::BROADCAST);
- frame.set_src_addr(remote_hw_addr);
- frame.set_ethertype(EthernetProtocol::Arp);
- {
- let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
- repr.emit(&mut packet);
- }
- // Ensure an ARP Request for someone else does not trigger an ARP Reply
- assert_eq!(iface.inner.process_ethernet(&mut socket_set, Instant::from_millis(0), frame.into_inner()),
- Ok(Packet::None));
- // Ensure the address of the requestor was entered in the cache
- assert_eq!(iface.inner.lookup_hardware_addr(MockTxToken, Instant::from_secs(0),
- &IpAddress::Ipv4(Ipv4Address([0x7f, 0x00, 0x00, 0x01])),
- &IpAddress::Ipv4(remote_ip_addr)),
- Ok((remote_hw_addr, MockTxToken)));
- }
- #[test]
- #[cfg(all(feature = "socket-icmp", feature = "proto-ipv4"))]
- fn test_icmpv4_socket() {
- use socket::{IcmpSocket, IcmpEndpoint, IcmpSocketBuffer, IcmpPacketMetadata};
- use wire::Icmpv4Packet;
- let (iface, mut socket_set) = create_loopback();
- let rx_buffer = IcmpSocketBuffer::new(vec![IcmpPacketMetadata::EMPTY], vec![0; 24]);
- let tx_buffer = IcmpSocketBuffer::new(vec![IcmpPacketMetadata::EMPTY], vec![0; 24]);
- let icmpv4_socket = IcmpSocket::new(rx_buffer, tx_buffer);
- let socket_handle = socket_set.add(icmpv4_socket);
- let ident = 0x1234;
- let seq_no = 0x5432;
- let echo_data = &[0xff; 16];
- {
- let mut socket = socket_set.get::<IcmpSocket>(socket_handle);
- // Bind to the ID 0x1234
- assert_eq!(socket.bind(IcmpEndpoint::Ident(ident)), Ok(()));
- }
- // Ensure the ident we bound to and the ident of the packet are the same.
- let mut bytes = [0xff; 24];
- let mut packet = Icmpv4Packet::new_unchecked(&mut bytes);
- let echo_repr = Icmpv4Repr::EchoRequest{ ident, seq_no, data: echo_data };
- echo_repr.emit(&mut packet, &ChecksumCapabilities::default());
- let icmp_data = &packet.into_inner()[..];
- let ipv4_repr = Ipv4Repr {
- src_addr: Ipv4Address::new(0x7f, 0x00, 0x00, 0x02),
- dst_addr: Ipv4Address::new(0x7f, 0x00, 0x00, 0x01),
- protocol: IpProtocol::Icmp,
- payload_len: 24,
- hop_limit: 64
- };
- let ip_repr = IpRepr::Ipv4(ipv4_repr);
- // Open a socket and ensure the packet is handled due to the listening
- // socket.
- {
- assert!(!socket_set.get::<IcmpSocket>(socket_handle).can_recv());
- }
- // Confirm we still get EchoReply from `smoltcp` even with the ICMP socket listening
- let echo_reply = Icmpv4Repr::EchoReply{ ident, seq_no, data: echo_data };
- let ipv4_reply = Ipv4Repr {
- src_addr: ipv4_repr.dst_addr,
- dst_addr: ipv4_repr.src_addr,
- ..ipv4_repr
- };
- assert_eq!(iface.inner.process_icmpv4(&mut socket_set, ip_repr, icmp_data),
- Ok(Packet::Icmpv4((ipv4_reply, echo_reply))));
- {
- let mut socket = socket_set.get::<IcmpSocket>(socket_handle);
- assert!(socket.can_recv());
- assert_eq!(socket.recv(),
- Ok((&icmp_data[..],
- IpAddress::Ipv4(Ipv4Address::new(0x7f, 0x00, 0x00, 0x02)))));
- }
- }
- #[test]
- #[cfg(feature = "proto-ipv6")]
- fn test_solicited_node_addrs() {
- let (mut iface, _) = create_loopback();
- let mut new_addrs = vec![IpCidr::new(IpAddress::v6(0xfe80, 0, 0, 0, 1, 2, 0, 2), 64),
- IpCidr::new(IpAddress::v6(0xfe80, 0, 0, 0, 3, 4, 0, 0xffff), 64)];
- iface.update_ip_addrs(|addrs| {
- new_addrs.extend(addrs.to_vec());
- *addrs = From::from(new_addrs);
- });
- assert!(iface.inner.has_solicited_node(Ipv6Address::new(0xff02, 0, 0, 0, 0, 1, 0xff00, 0x0002)));
- assert!(iface.inner.has_solicited_node(Ipv6Address::new(0xff02, 0, 0, 0, 0, 1, 0xff00, 0xffff)));
- assert!(!iface.inner.has_solicited_node(Ipv6Address::new(0xff02, 0, 0, 0, 0, 1, 0xff00, 0x0003)));
- }
- #[test]
- #[cfg(feature = "proto-ipv6")]
- fn test_icmpv6_nxthdr_unknown() {
- let (mut iface, mut socket_set) = create_loopback();
- let remote_ip_addr = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1);
- let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x01]);
- let mut eth_bytes = vec![0; 66];
- let payload = [0x12, 0x34, 0x56, 0x78];
- let ipv6_repr = Ipv6Repr {
- src_addr: remote_ip_addr,
- dst_addr: Ipv6Address::LOOPBACK,
- next_header: IpProtocol::HopByHop,
- payload_len: 12,
- hop_limit: 0x40,
- };
- let frame = {
- let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
- let ip_repr = IpRepr::Ipv6(ipv6_repr);
- frame.set_dst_addr(EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]));
- frame.set_src_addr(remote_hw_addr);
- frame.set_ethertype(EthernetProtocol::Ipv6);
- ip_repr.emit(frame.payload_mut(), &ChecksumCapabilities::default());
- let mut offset = ipv6_repr.buffer_len();
- {
- let mut hbh_pkt =
- Ipv6HopByHopHeader::new_unchecked(&mut frame.payload_mut()[offset..]);
- hbh_pkt.set_next_header(IpProtocol::Unknown(0x0c));
- hbh_pkt.set_header_len(0);
- offset += 8;
- {
- let mut pad_pkt = Ipv6Option::new_unchecked(&mut hbh_pkt.options_mut()[..]);
- Ipv6OptionRepr::PadN(3).emit(&mut pad_pkt);
- }
- {
- let mut pad_pkt = Ipv6Option::new_unchecked(&mut hbh_pkt.options_mut()[5..]);
- Ipv6OptionRepr::Pad1.emit(&mut pad_pkt);
- }
- }
- frame.payload_mut()[offset..].copy_from_slice(&payload);
- EthernetFrame::new_unchecked(&*frame.into_inner())
- };
- let reply_icmp_repr = Icmpv6Repr::ParamProblem {
- reason: Icmpv6ParamProblem::UnrecognizedNxtHdr,
- pointer: 40,
- header: ipv6_repr,
- data: &payload[..]
- };
- let reply_ipv6_repr = Ipv6Repr {
- src_addr: Ipv6Address::LOOPBACK,
- dst_addr: remote_ip_addr,
- next_header: IpProtocol::Icmpv6,
- payload_len: reply_icmp_repr.buffer_len(),
- hop_limit: 0x40,
- };
- // Ensure the unknown next header causes a ICMPv6 Parameter Problem
- // error message to be sent to the sender.
- assert_eq!(iface.inner.process_ipv6(&mut socket_set, Instant::from_millis(0), &frame),
- Ok(Packet::Icmpv6((reply_ipv6_repr, reply_icmp_repr))));
- // Ensure the address of the requestor was entered in the cache
- assert_eq!(iface.inner.lookup_hardware_addr(MockTxToken, Instant::from_secs(0),
- &IpAddress::Ipv6(Ipv6Address::LOOPBACK),
- &IpAddress::Ipv6(remote_ip_addr)),
- Ok((remote_hw_addr, MockTxToken)));
- }
- #[test]
- #[cfg(feature = "proto-igmp")]
- fn test_handle_igmp() {
- fn recv_igmp<'b>(mut iface: &mut EthernetInterface<'static, 'b, 'static, Loopback>, timestamp: Instant) -> Vec<(Ipv4Repr, IgmpRepr)> {
- let checksum_caps = &iface.device.capabilities().checksum;
- recv_all(&mut iface, timestamp)
- .iter()
- .filter_map(|frame| {
- let eth_frame = EthernetFrame::new_checked(frame).ok()?;
- let ipv4_packet = Ipv4Packet::new_checked(eth_frame.payload()).ok()?;
- let ipv4_repr = Ipv4Repr::parse(&ipv4_packet, &checksum_caps).ok()?;
- let ip_payload = ipv4_packet.payload();
- let igmp_packet = IgmpPacket::new_checked(ip_payload).ok()?;
- let igmp_repr = IgmpRepr::parse(&igmp_packet).ok()?;
- Some((ipv4_repr, igmp_repr))
- })
- .collect::<Vec<_>>()
- }
- let groups = [
- Ipv4Address::new(224, 0, 0, 22),
- Ipv4Address::new(224, 0, 0, 56),
- ];
- let (mut iface, mut socket_set) = create_loopback();
- // Join multicast groups
- let timestamp = Instant::now();
- for group in &groups {
- iface.join_multicast_group(*group, timestamp)
- .unwrap();
- }
- let reports = recv_igmp(&mut iface, timestamp);
- assert_eq!(reports.len(), 2);
- for (i, group_addr) in groups.iter().enumerate() {
- assert_eq!(reports[i].0.protocol, IpProtocol::Igmp);
- assert_eq!(reports[i].0.dst_addr, *group_addr);
- assert_eq!(reports[i].1, IgmpRepr::MembershipReport {
- group_addr: *group_addr,
- version: IgmpVersion::Version2,
- });
- }
- // General query
- let timestamp = Instant::now();
- const GENERAL_QUERY_BYTES: &[u8] = &[
- 0x01, 0x00, 0x5e, 0x00, 0x00, 0x01, 0x0a, 0x14,
- 0x48, 0x01, 0x21, 0x01, 0x08, 0x00, 0x46, 0xc0,
- 0x00, 0x24, 0xed, 0xb4, 0x00, 0x00, 0x01, 0x02,
- 0x47, 0x43, 0xac, 0x16, 0x63, 0x04, 0xe0, 0x00,
- 0x00, 0x01, 0x94, 0x04, 0x00, 0x00, 0x11, 0x64,
- 0xec, 0x8f, 0x00, 0x00, 0x00, 0x00, 0x02, 0x0c,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00
- ];
- {
- // Transmit GENERAL_QUERY_BYTES into loopback
- let tx_token = iface.device.transmit().unwrap();
- tx_token.consume(
- timestamp, GENERAL_QUERY_BYTES.len(),
- |buffer| {
- buffer.copy_from_slice(GENERAL_QUERY_BYTES);
- Ok(())
- }).unwrap();
- }
- // Trigger processing until all packets received through the
- // loopback have been processed, including responses to
- // GENERAL_QUERY_BYTES. Therefore `recv_all()` would return 0
- // pkts that could be checked.
- iface.socket_ingress(&mut socket_set, timestamp).unwrap();
- // Leave multicast groups
- let timestamp = Instant::now();
- for group in &groups {
- iface.leave_multicast_group(group.clone(), timestamp)
- .unwrap();
- }
- let leaves = recv_igmp(&mut iface, timestamp);
- assert_eq!(leaves.len(), 2);
- for (i, group_addr) in groups.iter().cloned().enumerate() {
- assert_eq!(leaves[i].0.protocol, IpProtocol::Igmp);
- assert_eq!(leaves[i].0.dst_addr, Ipv4Address::MULTICAST_ALL_ROUTERS);
- assert_eq!(leaves[i].1, IgmpRepr::LeaveGroup { group_addr });
- }
- }
- }
|