interface.rs 144 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880
  1. // Heads up! Before working on this file you should read the parts
  2. // of RFC 1122 that discuss Ethernet, ARP and IP for any IPv4 work
  3. // and RFCs 8200 and 4861 for any IPv6 and NDISC work.
  4. use core::cmp;
  5. use managed::{ManagedMap, ManagedSlice};
  6. #[allow(unused)]
  7. #[cfg(feature = "proto-sixlowpan")]
  8. use super::fragmentation::PacketAssemblerSet;
  9. use super::socket_set::SocketSet;
  10. use crate::iface::Routes;
  11. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  12. use crate::iface::{NeighborAnswer, NeighborCache};
  13. use crate::phy::{ChecksumCapabilities, Device, DeviceCapabilities, Medium, RxToken, TxToken};
  14. use crate::rand::Rand;
  15. #[cfg(feature = "socket-dhcpv4")]
  16. use crate::socket::dhcpv4;
  17. #[cfg(feature = "socket-dns")]
  18. use crate::socket::dns;
  19. use crate::socket::*;
  20. use crate::time::{Duration, Instant};
  21. use crate::wire::*;
  22. use crate::{Error, Result};
  23. macro_rules! check {
  24. ($e:expr) => {
  25. match $e {
  26. Ok(x) => x,
  27. Err(_) => {
  28. // concat!/stringify! doesn't work with defmt macros
  29. #[cfg(not(feature = "defmt"))]
  30. net_trace!(concat!("iface: malformed ", stringify!($e)));
  31. #[cfg(feature = "defmt")]
  32. net_trace!("iface: malformed");
  33. return Default::default();
  34. }
  35. }
  36. };
  37. }
  38. /// A network interface.
  39. ///
  40. /// The network interface logically owns a number of other data structures; to avoid
  41. /// a dependency on heap allocation, it instead owns a `BorrowMut<[T]>`, which can be
  42. /// a `&mut [T]`, or `Vec<T>` if a heap is available.
  43. pub struct Interface<'a, DeviceT: for<'d> Device<'d>> {
  44. device: DeviceT,
  45. inner: InterfaceInner<'a>,
  46. }
  47. /// The device independent part of an Ethernet network interface.
  48. ///
  49. /// Separating the device from the data required for processing and dispatching makes
  50. /// it possible to borrow them independently. For example, the tx and rx tokens borrow
  51. /// the `device` mutably until they're used, which makes it impossible to call other
  52. /// methods on the `Interface` in this time (since its `device` field is borrowed
  53. /// exclusively). However, it is still possible to call methods on its `inner` field.
  54. pub struct InterfaceInner<'a> {
  55. caps: DeviceCapabilities,
  56. now: Instant,
  57. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  58. neighbor_cache: Option<NeighborCache<'a>>,
  59. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  60. hardware_addr: Option<HardwareAddress>,
  61. #[cfg(feature = "medium-ieee802154")]
  62. sequence_no: u8,
  63. #[cfg(feature = "medium-ieee802154")]
  64. pan_id: Option<Ieee802154Pan>,
  65. ip_addrs: ManagedSlice<'a, IpCidr>,
  66. #[cfg(feature = "proto-ipv4")]
  67. any_ip: bool,
  68. routes: Routes<'a>,
  69. #[cfg(feature = "proto-igmp")]
  70. ipv4_multicast_groups: ManagedMap<'a, Ipv4Address, ()>,
  71. /// When to report for (all or) the next multicast group membership via IGMP
  72. #[cfg(feature = "proto-igmp")]
  73. igmp_report_state: IgmpReportState,
  74. rand: Rand,
  75. }
  76. /// A builder structure used for creating a network interface.
  77. pub struct InterfaceBuilder<'a, DeviceT: for<'d> Device<'d>> {
  78. device: DeviceT,
  79. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  80. hardware_addr: Option<HardwareAddress>,
  81. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  82. neighbor_cache: Option<NeighborCache<'a>>,
  83. #[cfg(feature = "medium-ieee802154")]
  84. pan_id: Option<Ieee802154Pan>,
  85. ip_addrs: ManagedSlice<'a, IpCidr>,
  86. #[cfg(feature = "proto-ipv4")]
  87. any_ip: bool,
  88. routes: Routes<'a>,
  89. /// Does not share storage with `ipv6_multicast_groups` to avoid IPv6 size overhead.
  90. #[cfg(feature = "proto-igmp")]
  91. ipv4_multicast_groups: ManagedMap<'a, Ipv4Address, ()>,
  92. random_seed: u64,
  93. }
  94. impl<'a, DeviceT> InterfaceBuilder<'a, DeviceT>
  95. where
  96. DeviceT: for<'d> Device<'d>,
  97. {
  98. /// Create a builder used for creating a network interface using the
  99. /// given device and address.
  100. #[cfg_attr(
  101. feature = "medium-ethernet",
  102. doc = r##"
  103. # Examples
  104. ```
  105. # use std::collections::BTreeMap;
  106. use smoltcp::iface::{InterfaceBuilder, NeighborCache};
  107. # use smoltcp::phy::{Loopback, Medium};
  108. use smoltcp::wire::{EthernetAddress, IpCidr, IpAddress};
  109. let device = // ...
  110. # Loopback::new(Medium::Ethernet);
  111. let hw_addr = // ...
  112. # EthernetAddress::default();
  113. let neighbor_cache = // ...
  114. # NeighborCache::new(BTreeMap::new());
  115. let ip_addrs = // ...
  116. # [];
  117. let iface = InterfaceBuilder::new(device)
  118. .hardware_addr(hw_addr.into())
  119. .neighbor_cache(neighbor_cache)
  120. .ip_addrs(ip_addrs)
  121. .finalize();
  122. ```
  123. "##
  124. )]
  125. pub fn new(device: DeviceT) -> Self {
  126. InterfaceBuilder {
  127. device: device,
  128. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  129. hardware_addr: None,
  130. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  131. neighbor_cache: None,
  132. #[cfg(feature = "medium-ieee802154")]
  133. pan_id: None,
  134. ip_addrs: ManagedSlice::Borrowed(&mut []),
  135. #[cfg(feature = "proto-ipv4")]
  136. any_ip: false,
  137. routes: Routes::new(ManagedMap::Borrowed(&mut [])),
  138. #[cfg(feature = "proto-igmp")]
  139. ipv4_multicast_groups: ManagedMap::Borrowed(&mut []),
  140. random_seed: 0,
  141. }
  142. }
  143. /// Set the random seed for this interface.
  144. ///
  145. /// It is strongly recommended that the random seed is different on each boot,
  146. /// to avoid problems with TCP port/sequence collisions.
  147. ///
  148. /// The seed doesn't have to be cryptographically secure.
  149. pub fn random_seed(mut self, random_seed: u64) -> Self {
  150. self.random_seed = random_seed;
  151. self
  152. }
  153. /// Set the Hardware address the interface will use. See also
  154. /// [hardware_addr].
  155. ///
  156. /// # Panics
  157. /// This function panics if the address is not unicast.
  158. ///
  159. /// [hardware_addr]: struct.Interface.html#method.hardware_addr
  160. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  161. pub fn hardware_addr(mut self, addr: HardwareAddress) -> Self {
  162. InterfaceInner::check_hardware_addr(&addr);
  163. self.hardware_addr = Some(addr);
  164. self
  165. }
  166. /// Set the IEEE802.15.4 PAN ID the interface will use.
  167. ///
  168. /// **NOTE**: we use the same PAN ID for destination and source.
  169. #[cfg(feature = "medium-ieee802154")]
  170. pub fn pan_id(mut self, pan_id: Ieee802154Pan) -> Self {
  171. self.pan_id = Some(pan_id);
  172. self
  173. }
  174. /// Set the IP addresses the interface will use. See also
  175. /// [ip_addrs].
  176. ///
  177. /// # Panics
  178. /// This function panics if any of the addresses are not unicast.
  179. ///
  180. /// [ip_addrs]: struct.Interface.html#method.ip_addrs
  181. pub fn ip_addrs<T>(mut self, ip_addrs: T) -> Self
  182. where
  183. T: Into<ManagedSlice<'a, IpCidr>>,
  184. {
  185. let ip_addrs = ip_addrs.into();
  186. InterfaceInner::check_ip_addrs(&ip_addrs);
  187. self.ip_addrs = ip_addrs;
  188. self
  189. }
  190. /// Enable or disable the AnyIP capability, allowing packets to be received
  191. /// locally on IPv4 addresses other than the interface's configured [ip_addrs].
  192. /// When AnyIP is enabled and a route prefix in [routes] specifies one of
  193. /// the interface's [ip_addrs] as its gateway, the interface will accept
  194. /// packets addressed to that prefix.
  195. ///
  196. /// # IPv6
  197. ///
  198. /// This option is not available or required for IPv6 as packets sent to
  199. /// the interface are not filtered by IPv6 address.
  200. ///
  201. /// [routes]: struct.Interface.html#method.routes
  202. /// [ip_addrs]: struct.Interface.html#method.ip_addrs
  203. #[cfg(feature = "proto-ipv4")]
  204. pub fn any_ip(mut self, enabled: bool) -> Self {
  205. self.any_ip = enabled;
  206. self
  207. }
  208. /// Set the IP routes the interface will use. See also
  209. /// [routes].
  210. ///
  211. /// [routes]: struct.Interface.html#method.routes
  212. pub fn routes<T>(mut self, routes: T) -> InterfaceBuilder<'a, DeviceT>
  213. where
  214. T: Into<Routes<'a>>,
  215. {
  216. self.routes = routes.into();
  217. self
  218. }
  219. /// Provide storage for multicast groups.
  220. ///
  221. /// Join multicast groups by calling [`join_multicast_group()`] on an `Interface`.
  222. /// Using [`join_multicast_group()`] will send initial membership reports.
  223. ///
  224. /// A previously destroyed interface can be recreated by reusing the multicast group
  225. /// storage, i.e. providing a non-empty storage to `ipv4_multicast_groups()`.
  226. /// Note that this way initial membership reports are **not** sent.
  227. ///
  228. /// [`join_multicast_group()`]: struct.Interface.html#method.join_multicast_group
  229. #[cfg(feature = "proto-igmp")]
  230. pub fn ipv4_multicast_groups<T>(mut self, ipv4_multicast_groups: T) -> Self
  231. where
  232. T: Into<ManagedMap<'a, Ipv4Address, ()>>,
  233. {
  234. self.ipv4_multicast_groups = ipv4_multicast_groups.into();
  235. self
  236. }
  237. /// Set the Neighbor Cache the interface will use.
  238. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  239. pub fn neighbor_cache(mut self, neighbor_cache: NeighborCache<'a>) -> Self {
  240. self.neighbor_cache = Some(neighbor_cache);
  241. self
  242. }
  243. /// Create a network interface using the previously provided configuration.
  244. ///
  245. /// # Panics
  246. /// If a required option is not provided, this function will panic. Required
  247. /// options are:
  248. ///
  249. /// - [ethernet_addr]
  250. /// - [neighbor_cache]
  251. ///
  252. /// [ethernet_addr]: #method.ethernet_addr
  253. /// [neighbor_cache]: #method.neighbor_cache
  254. pub fn finalize(self) -> Interface<'a, DeviceT> {
  255. let device_capabilities = self.device.capabilities();
  256. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  257. let (hardware_addr, neighbor_cache) = match device_capabilities.medium {
  258. #[cfg(feature = "medium-ethernet")]
  259. Medium::Ethernet => (
  260. Some(
  261. self.hardware_addr
  262. .expect("hardware_addr required option was not set"),
  263. ),
  264. Some(
  265. self.neighbor_cache
  266. .expect("neighbor_cache required option was not set"),
  267. ),
  268. ),
  269. #[cfg(feature = "medium-ip")]
  270. Medium::Ip => {
  271. assert!(
  272. self.hardware_addr.is_none(),
  273. "hardware_addr is set, but device medium is IP"
  274. );
  275. assert!(
  276. self.neighbor_cache.is_none(),
  277. "neighbor_cache is set, but device medium is IP"
  278. );
  279. (None, None)
  280. }
  281. #[cfg(feature = "medium-ieee802154")]
  282. Medium::Ieee802154 => (
  283. Some(
  284. self.hardware_addr
  285. .expect("hardware_addr required option was not set"),
  286. ),
  287. Some(
  288. self.neighbor_cache
  289. .expect("neighbor_cache required option was not set"),
  290. ),
  291. ),
  292. };
  293. let caps = self.device.capabilities();
  294. #[cfg(feature = "medium-ieee802154")]
  295. let mut rand = Rand::new(self.random_seed);
  296. #[cfg(not(feature = "medium-ieee802154"))]
  297. let rand = Rand::new(self.random_seed);
  298. #[cfg(feature = "medium-ieee802154")]
  299. let mut sequence_no;
  300. #[cfg(feature = "medium-ieee802154")]
  301. loop {
  302. sequence_no = (rand.rand_u32() & 0xff) as u8;
  303. if sequence_no != 0 {
  304. break;
  305. }
  306. }
  307. Interface {
  308. device: self.device,
  309. inner: InterfaceInner {
  310. now: Instant::from_secs(0),
  311. caps,
  312. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  313. hardware_addr,
  314. ip_addrs: self.ip_addrs,
  315. #[cfg(feature = "proto-ipv4")]
  316. any_ip: self.any_ip,
  317. routes: self.routes,
  318. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  319. neighbor_cache,
  320. #[cfg(feature = "proto-igmp")]
  321. ipv4_multicast_groups: self.ipv4_multicast_groups,
  322. #[cfg(feature = "proto-igmp")]
  323. igmp_report_state: IgmpReportState::Inactive,
  324. #[cfg(feature = "medium-ieee802154")]
  325. sequence_no,
  326. #[cfg(feature = "medium-ieee802154")]
  327. pan_id: self.pan_id,
  328. rand,
  329. },
  330. }
  331. }
  332. }
  333. #[derive(Debug, PartialEq)]
  334. #[cfg_attr(feature = "defmt", derive(defmt::Format))]
  335. #[cfg(feature = "medium-ethernet")]
  336. enum EthernetPacket<'a> {
  337. #[cfg(feature = "proto-ipv4")]
  338. Arp(ArpRepr),
  339. Ip(IpPacket<'a>),
  340. }
  341. #[derive(Debug, PartialEq)]
  342. #[cfg_attr(feature = "defmt", derive(defmt::Format))]
  343. pub(crate) enum IpPacket<'a> {
  344. #[cfg(feature = "proto-ipv4")]
  345. Icmpv4((Ipv4Repr, Icmpv4Repr<'a>)),
  346. #[cfg(feature = "proto-igmp")]
  347. Igmp((Ipv4Repr, IgmpRepr)),
  348. #[cfg(feature = "proto-ipv6")]
  349. Icmpv6((Ipv6Repr, Icmpv6Repr<'a>)),
  350. #[cfg(feature = "socket-raw")]
  351. Raw((IpRepr, &'a [u8])),
  352. #[cfg(any(feature = "socket-udp", feature = "socket-dns"))]
  353. Udp((IpRepr, UdpRepr, &'a [u8])),
  354. #[cfg(feature = "socket-tcp")]
  355. Tcp((IpRepr, TcpRepr<'a>)),
  356. #[cfg(feature = "socket-dhcpv4")]
  357. Dhcpv4((Ipv4Repr, UdpRepr, DhcpRepr<'a>)),
  358. }
  359. impl<'a> IpPacket<'a> {
  360. pub(crate) fn ip_repr(&self) -> IpRepr {
  361. match self {
  362. #[cfg(feature = "proto-ipv4")]
  363. IpPacket::Icmpv4((ipv4_repr, _)) => IpRepr::Ipv4(*ipv4_repr),
  364. #[cfg(feature = "proto-igmp")]
  365. IpPacket::Igmp((ipv4_repr, _)) => IpRepr::Ipv4(*ipv4_repr),
  366. #[cfg(feature = "proto-ipv6")]
  367. IpPacket::Icmpv6((ipv6_repr, _)) => IpRepr::Ipv6(*ipv6_repr),
  368. #[cfg(feature = "socket-raw")]
  369. IpPacket::Raw((ip_repr, _)) => ip_repr.clone(),
  370. #[cfg(any(feature = "socket-udp", feature = "socket-dns"))]
  371. IpPacket::Udp((ip_repr, _, _)) => ip_repr.clone(),
  372. #[cfg(feature = "socket-tcp")]
  373. IpPacket::Tcp((ip_repr, _)) => ip_repr.clone(),
  374. #[cfg(feature = "socket-dhcpv4")]
  375. IpPacket::Dhcpv4((ipv4_repr, _, _)) => IpRepr::Ipv4(*ipv4_repr),
  376. }
  377. }
  378. pub(crate) fn emit_payload(
  379. &self,
  380. _ip_repr: IpRepr,
  381. payload: &mut [u8],
  382. caps: &DeviceCapabilities,
  383. ) {
  384. match self {
  385. #[cfg(feature = "proto-ipv4")]
  386. IpPacket::Icmpv4((_, icmpv4_repr)) => {
  387. icmpv4_repr.emit(&mut Icmpv4Packet::new_unchecked(payload), &caps.checksum)
  388. }
  389. #[cfg(feature = "proto-igmp")]
  390. IpPacket::Igmp((_, igmp_repr)) => {
  391. igmp_repr.emit(&mut IgmpPacket::new_unchecked(payload))
  392. }
  393. #[cfg(feature = "proto-ipv6")]
  394. IpPacket::Icmpv6((_, icmpv6_repr)) => icmpv6_repr.emit(
  395. &_ip_repr.src_addr(),
  396. &_ip_repr.dst_addr(),
  397. &mut Icmpv6Packet::new_unchecked(payload),
  398. &caps.checksum,
  399. ),
  400. #[cfg(feature = "socket-raw")]
  401. IpPacket::Raw((_, raw_packet)) => payload.copy_from_slice(raw_packet),
  402. #[cfg(any(feature = "socket-udp", feature = "socket-dns"))]
  403. IpPacket::Udp((_, udp_repr, inner_payload)) => udp_repr.emit(
  404. &mut UdpPacket::new_unchecked(payload),
  405. &_ip_repr.src_addr(),
  406. &_ip_repr.dst_addr(),
  407. inner_payload.len(),
  408. |buf| buf.copy_from_slice(inner_payload),
  409. &caps.checksum,
  410. ),
  411. #[cfg(feature = "socket-tcp")]
  412. IpPacket::Tcp((_, mut tcp_repr)) => {
  413. // This is a terrible hack to make TCP performance more acceptable on systems
  414. // where the TCP buffers are significantly larger than network buffers,
  415. // e.g. a 64 kB TCP receive buffer (and so, when empty, a 64k window)
  416. // together with four 1500 B Ethernet receive buffers. If left untreated,
  417. // this would result in our peer pushing our window and sever packet loss.
  418. //
  419. // I'm really not happy about this "solution" but I don't know what else to do.
  420. if let Some(max_burst_size) = caps.max_burst_size {
  421. let mut max_segment_size = caps.max_transmission_unit;
  422. max_segment_size -= _ip_repr.buffer_len();
  423. max_segment_size -= tcp_repr.header_len();
  424. let max_window_size = max_burst_size * max_segment_size;
  425. if tcp_repr.window_len as usize > max_window_size {
  426. tcp_repr.window_len = max_window_size as u16;
  427. }
  428. }
  429. tcp_repr.emit(
  430. &mut TcpPacket::new_unchecked(payload),
  431. &_ip_repr.src_addr(),
  432. &_ip_repr.dst_addr(),
  433. &caps.checksum,
  434. );
  435. }
  436. #[cfg(feature = "socket-dhcpv4")]
  437. IpPacket::Dhcpv4((_, udp_repr, dhcp_repr)) => udp_repr.emit(
  438. &mut UdpPacket::new_unchecked(payload),
  439. &_ip_repr.src_addr(),
  440. &_ip_repr.dst_addr(),
  441. dhcp_repr.buffer_len(),
  442. |buf| dhcp_repr.emit(&mut DhcpPacket::new_unchecked(buf)).unwrap(),
  443. &caps.checksum,
  444. ),
  445. }
  446. }
  447. }
  448. #[cfg(any(feature = "proto-ipv4", feature = "proto-ipv6"))]
  449. fn icmp_reply_payload_len(len: usize, mtu: usize, header_len: usize) -> usize {
  450. // Send back as much of the original payload as will fit within
  451. // the minimum MTU required by IPv4. See RFC 1812 § 4.3.2.3 for
  452. // more details.
  453. //
  454. // Since the entire network layer packet must fit within the minimum
  455. // MTU supported, the payload must not exceed the following:
  456. //
  457. // <min mtu> - IP Header Size * 2 - ICMPv4 DstUnreachable hdr size
  458. cmp::min(len, mtu - header_len * 2 - 8)
  459. }
  460. #[cfg(feature = "proto-igmp")]
  461. enum IgmpReportState {
  462. Inactive,
  463. ToGeneralQuery {
  464. version: IgmpVersion,
  465. timeout: Instant,
  466. interval: Duration,
  467. next_index: usize,
  468. },
  469. ToSpecificQuery {
  470. version: IgmpVersion,
  471. timeout: Instant,
  472. group: Ipv4Address,
  473. },
  474. }
  475. impl<'a, DeviceT> Interface<'a, DeviceT>
  476. where
  477. DeviceT: for<'d> Device<'d>,
  478. {
  479. /// Get the socket context.
  480. ///
  481. /// The context is needed for some socket methods.
  482. pub fn context(&mut self) -> &mut InterfaceInner<'a> {
  483. &mut self.inner
  484. }
  485. /// Get the HardwareAddress address of the interface.
  486. ///
  487. /// # Panics
  488. /// This function panics if the medium is not Ethernet or Ieee802154.
  489. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  490. pub fn hardware_addr(&self) -> HardwareAddress {
  491. #[cfg(all(feature = "medium-ethernet", not(feature = "medium-ieee802154")))]
  492. assert!(self.device().capabilities().medium == Medium::Ethernet);
  493. #[cfg(all(feature = "medium-ieee802154", not(feature = "medium-ethernet")))]
  494. assert!(self.device().capabilities().medium == Medium::Ieee802154);
  495. #[cfg(all(feature = "medium-ieee802154", feature = "medium-ethernet"))]
  496. assert!(
  497. self.device().capabilities().medium == Medium::Ethernet
  498. || self.device().capabilities().medium == Medium::Ieee802154
  499. );
  500. self.inner.hardware_addr.unwrap()
  501. }
  502. /// Set the HardwareAddress address of the interface.
  503. ///
  504. /// # Panics
  505. /// This function panics if the address is not unicast, and if the medium is not Ethernet or
  506. /// Ieee802154.
  507. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  508. pub fn set_hardware_addr(&mut self, addr: HardwareAddress) {
  509. #[cfg(all(feature = "medium-ethernet", not(feature = "medium-ieee802154")))]
  510. assert!(self.device().capabilities().medium == Medium::Ethernet);
  511. #[cfg(all(feature = "medium-ieee802154", not(feature = "medium-ethernet")))]
  512. assert!(self.device().capabilities().medium == Medium::Ieee802154);
  513. #[cfg(all(feature = "medium-ieee802154", feature = "medium-ethernet"))]
  514. assert!(
  515. self.device().capabilities().medium == Medium::Ethernet
  516. || self.device().capabilities().medium == Medium::Ieee802154
  517. );
  518. InterfaceInner::check_hardware_addr(&addr);
  519. self.inner.hardware_addr = Some(addr);
  520. }
  521. /// Get a reference to the inner device.
  522. pub fn device(&self) -> &DeviceT {
  523. &self.device
  524. }
  525. /// Get a mutable reference to the inner device.
  526. ///
  527. /// There are no invariants imposed on the device by the interface itself. Furthermore the
  528. /// trait implementations, required for references of all lifetimes, guarantees that the
  529. /// mutable reference can not invalidate the device as such. For some devices, such access may
  530. /// still allow modifications with adverse effects on the usability as a `phy` device. You
  531. /// should not use them this way.
  532. pub fn device_mut(&mut self) -> &mut DeviceT {
  533. &mut self.device
  534. }
  535. /// Add an address to a list of subscribed multicast IP addresses.
  536. ///
  537. /// Returns `Ok(announce_sent)` if the address was added successfully, where `annouce_sent`
  538. /// indicates whether an initial immediate announcement has been sent.
  539. pub fn join_multicast_group<T: Into<IpAddress>>(
  540. &mut self,
  541. addr: T,
  542. timestamp: Instant,
  543. ) -> Result<bool> {
  544. self.inner.now = timestamp;
  545. match addr.into() {
  546. #[cfg(feature = "proto-igmp")]
  547. IpAddress::Ipv4(addr) => {
  548. let is_not_new = self
  549. .inner
  550. .ipv4_multicast_groups
  551. .insert(addr, ())
  552. .map_err(|_| Error::Exhausted)?
  553. .is_some();
  554. if is_not_new {
  555. Ok(false)
  556. } else if let Some(pkt) = self.inner.igmp_report_packet(IgmpVersion::Version2, addr)
  557. {
  558. // Send initial membership report
  559. let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
  560. self.inner.dispatch_ip(tx_token, pkt)?;
  561. Ok(true)
  562. } else {
  563. Ok(false)
  564. }
  565. }
  566. // Multicast is not yet implemented for other address families
  567. #[allow(unreachable_patterns)]
  568. _ => Err(Error::Unaddressable),
  569. }
  570. }
  571. /// Remove an address from the subscribed multicast IP addresses.
  572. ///
  573. /// Returns `Ok(leave_sent)` if the address was removed successfully, where `leave_sent`
  574. /// indicates whether an immediate leave packet has been sent.
  575. pub fn leave_multicast_group<T: Into<IpAddress>>(
  576. &mut self,
  577. addr: T,
  578. timestamp: Instant,
  579. ) -> Result<bool> {
  580. self.inner.now = timestamp;
  581. match addr.into() {
  582. #[cfg(feature = "proto-igmp")]
  583. IpAddress::Ipv4(addr) => {
  584. let was_not_present = self.inner.ipv4_multicast_groups.remove(&addr).is_none();
  585. if was_not_present {
  586. Ok(false)
  587. } else if let Some(pkt) = self.inner.igmp_leave_packet(addr) {
  588. // Send group leave packet
  589. let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
  590. self.inner.dispatch_ip(tx_token, pkt)?;
  591. Ok(true)
  592. } else {
  593. Ok(false)
  594. }
  595. }
  596. // Multicast is not yet implemented for other address families
  597. #[allow(unreachable_patterns)]
  598. _ => Err(Error::Unaddressable),
  599. }
  600. }
  601. /// Check whether the interface listens to given destination multicast IP address.
  602. pub fn has_multicast_group<T: Into<IpAddress>>(&self, addr: T) -> bool {
  603. self.inner.has_multicast_group(addr)
  604. }
  605. /// Get the IP addresses of the interface.
  606. pub fn ip_addrs(&self) -> &[IpCidr] {
  607. self.inner.ip_addrs.as_ref()
  608. }
  609. /// Get the first IPv4 address if present.
  610. #[cfg(feature = "proto-ipv4")]
  611. pub fn ipv4_addr(&self) -> Option<Ipv4Address> {
  612. self.ip_addrs()
  613. .iter()
  614. .filter_map(|cidr| match cidr.address() {
  615. IpAddress::Ipv4(addr) => Some(addr),
  616. #[allow(unreachable_patterns)]
  617. _ => None,
  618. })
  619. .next()
  620. }
  621. /// Update the IP addresses of the interface.
  622. ///
  623. /// # Panics
  624. /// This function panics if any of the addresses are not unicast.
  625. pub fn update_ip_addrs<F: FnOnce(&mut ManagedSlice<'a, IpCidr>)>(&mut self, f: F) {
  626. f(&mut self.inner.ip_addrs);
  627. InterfaceInner::flush_cache(&mut self.inner);
  628. InterfaceInner::check_ip_addrs(&self.inner.ip_addrs)
  629. }
  630. /// Check whether the interface has the given IP address assigned.
  631. pub fn has_ip_addr<T: Into<IpAddress>>(&self, addr: T) -> bool {
  632. self.inner.has_ip_addr(addr)
  633. }
  634. /// Get the first IPv4 address of the interface.
  635. #[cfg(feature = "proto-ipv4")]
  636. pub fn ipv4_address(&self) -> Option<Ipv4Address> {
  637. self.inner.ipv4_address()
  638. }
  639. pub fn routes(&self) -> &Routes<'a> {
  640. &self.inner.routes
  641. }
  642. pub fn routes_mut(&mut self) -> &mut Routes<'a> {
  643. &mut self.inner.routes
  644. }
  645. /// Transmit packets queued in the given sockets, and receive packets queued
  646. /// in the device.
  647. ///
  648. /// This function returns a boolean value indicating whether any packets were
  649. /// processed or emitted, and thus, whether the readiness of any socket might
  650. /// have changed.
  651. ///
  652. /// # Errors
  653. /// This method will routinely return errors in response to normal network
  654. /// activity as well as certain boundary conditions such as buffer exhaustion.
  655. /// These errors are provided as an aid for troubleshooting, and are meant
  656. /// to be logged and ignored.
  657. ///
  658. /// As a special case, `Err(Error::Unrecognized)` is returned in response to
  659. /// packets containing any unsupported protocol, option, or form, which is
  660. /// a very common occurrence and on a production system it should not even
  661. /// be logged.
  662. pub fn poll(&mut self, timestamp: Instant, sockets: &mut SocketSet<'_>) -> Result<bool> {
  663. self.inner.now = timestamp;
  664. let mut readiness_may_have_changed = false;
  665. loop {
  666. let processed_any = self.socket_ingress(sockets);
  667. let emitted_any = self.socket_egress(sockets);
  668. #[cfg(feature = "proto-igmp")]
  669. self.igmp_egress()?;
  670. if processed_any || emitted_any {
  671. readiness_may_have_changed = true;
  672. } else {
  673. break;
  674. }
  675. }
  676. Ok(readiness_may_have_changed)
  677. }
  678. /// Return a _soft deadline_ for calling [poll] the next time.
  679. /// The [Instant] returned is the time at which you should call [poll] next.
  680. /// It is harmless (but wastes energy) to call it before the [Instant], and
  681. /// potentially harmful (impacting quality of service) to call it after the
  682. /// [Instant]
  683. ///
  684. /// [poll]: #method.poll
  685. /// [Instant]: struct.Instant.html
  686. pub fn poll_at(&mut self, timestamp: Instant, sockets: &SocketSet<'_>) -> Option<Instant> {
  687. self.inner.now = timestamp;
  688. let inner = &mut self.inner;
  689. sockets
  690. .items()
  691. .filter_map(move |item| {
  692. let socket_poll_at = item.socket.poll_at(inner);
  693. match item
  694. .meta
  695. .poll_at(socket_poll_at, |ip_addr| inner.has_neighbor(&ip_addr))
  696. {
  697. PollAt::Ingress => None,
  698. PollAt::Time(instant) => Some(instant),
  699. PollAt::Now => Some(Instant::from_millis(0)),
  700. }
  701. })
  702. .min()
  703. }
  704. /// Return an _advisory wait time_ for calling [poll] the next time.
  705. /// The [Duration] returned is the time left to wait before calling [poll] next.
  706. /// It is harmless (but wastes energy) to call it before the [Duration] has passed,
  707. /// and potentially harmful (impacting quality of service) to call it after the
  708. /// [Duration] has passed.
  709. ///
  710. /// [poll]: #method.poll
  711. /// [Duration]: struct.Duration.html
  712. pub fn poll_delay(&mut self, timestamp: Instant, sockets: &SocketSet<'_>) -> Option<Duration> {
  713. match self.poll_at(timestamp, sockets) {
  714. Some(poll_at) if timestamp < poll_at => Some(poll_at - timestamp),
  715. Some(_) => Some(Duration::from_millis(0)),
  716. _ => None,
  717. }
  718. }
  719. fn socket_ingress(&mut self, sockets: &mut SocketSet<'_>) -> bool {
  720. let mut processed_any = false;
  721. let Self { device, inner, .. } = self;
  722. while let Some((rx_token, tx_token)) = device.receive() {
  723. let res = rx_token.consume(inner.now, |frame| {
  724. match inner.caps.medium {
  725. #[cfg(feature = "medium-ethernet")]
  726. Medium::Ethernet => {
  727. if let Some(packet) = inner.process_ethernet(sockets, &frame) {
  728. if let Err(err) = inner.dispatch(tx_token, packet) {
  729. net_debug!("Failed to send response: {}", err);
  730. }
  731. }
  732. }
  733. #[cfg(feature = "medium-ip")]
  734. Medium::Ip => {
  735. if let Some(packet) = inner.process_ip(sockets, &frame) {
  736. if let Err(err) = inner.dispatch_ip(tx_token, packet) {
  737. net_debug!("Failed to send response: {}", err);
  738. }
  739. }
  740. }
  741. #[cfg(feature = "medium-ieee802154")]
  742. Medium::Ieee802154 => {
  743. if let Some(packet) = inner.process_ieee802154(sockets, &frame) {
  744. if let Err(err) = inner.dispatch_ieee802154(tx_token, packet) {
  745. net_debug!("Failed to send response: {}", err);
  746. }
  747. }
  748. }
  749. }
  750. processed_any = true;
  751. Ok(())
  752. });
  753. if let Err(err) = res {
  754. net_debug!("Failed to consume RX token: {}", err);
  755. }
  756. }
  757. processed_any
  758. }
  759. fn socket_egress(&mut self, sockets: &mut SocketSet<'_>) -> bool {
  760. let Self { device, inner, .. } = self;
  761. let _caps = device.capabilities();
  762. let mut emitted_any = false;
  763. for item in sockets.items_mut() {
  764. if !item
  765. .meta
  766. .egress_permitted(inner.now, |ip_addr| inner.has_neighbor(&ip_addr))
  767. {
  768. continue;
  769. }
  770. let mut neighbor_addr = None;
  771. let mut respond = |inner: &mut InterfaceInner, response: IpPacket| {
  772. neighbor_addr = Some(response.ip_repr().dst_addr());
  773. let tx_token = device.transmit().ok_or(Error::Exhausted)?;
  774. inner.dispatch_ip(tx_token, response)?;
  775. emitted_any = true;
  776. Ok(())
  777. };
  778. let result = match &mut item.socket {
  779. #[cfg(feature = "socket-raw")]
  780. Socket::Raw(socket) => socket.dispatch(inner, |inner, response| {
  781. respond(inner, IpPacket::Raw(response))
  782. }),
  783. #[cfg(feature = "socket-icmp")]
  784. Socket::Icmp(socket) => socket.dispatch(inner, |inner, response| match response {
  785. #[cfg(feature = "proto-ipv4")]
  786. (IpRepr::Ipv4(ipv4_repr), IcmpRepr::Ipv4(icmpv4_repr)) => {
  787. respond(inner, IpPacket::Icmpv4((ipv4_repr, icmpv4_repr)))
  788. }
  789. #[cfg(feature = "proto-ipv6")]
  790. (IpRepr::Ipv6(ipv6_repr), IcmpRepr::Ipv6(icmpv6_repr)) => {
  791. respond(inner, IpPacket::Icmpv6((ipv6_repr, icmpv6_repr)))
  792. }
  793. #[allow(unreachable_patterns)]
  794. _ => unreachable!(),
  795. }),
  796. #[cfg(feature = "socket-udp")]
  797. Socket::Udp(socket) => socket.dispatch(inner, |inner, response| {
  798. respond(inner, IpPacket::Udp(response))
  799. }),
  800. #[cfg(feature = "socket-tcp")]
  801. Socket::Tcp(socket) => socket.dispatch(inner, |inner, response| {
  802. respond(inner, IpPacket::Tcp(response))
  803. }),
  804. #[cfg(feature = "socket-dhcpv4")]
  805. Socket::Dhcpv4(socket) => socket.dispatch(inner, |inner, response| {
  806. respond(inner, IpPacket::Dhcpv4(response))
  807. }),
  808. #[cfg(feature = "socket-dns")]
  809. Socket::Dns(ref mut socket) => socket.dispatch(inner, |inner, response| {
  810. respond(inner, IpPacket::Udp(response))
  811. }),
  812. };
  813. match result {
  814. Err(Error::Exhausted) => break, // Device buffer full.
  815. Err(Error::Unaddressable) => {
  816. // `NeighborCache` already takes care of rate limiting the neighbor discovery
  817. // requests from the socket. However, without an additional rate limiting
  818. // mechanism, we would spin on every socket that has yet to discover its
  819. // neighbor.
  820. item.meta.neighbor_missing(
  821. inner.now,
  822. neighbor_addr.expect("non-IP response packet"),
  823. );
  824. break;
  825. }
  826. Err(err) => {
  827. net_debug!(
  828. "{}: cannot dispatch egress packet: {}",
  829. item.meta.handle,
  830. err
  831. );
  832. }
  833. Ok(()) => {}
  834. }
  835. }
  836. emitted_any
  837. }
  838. /// Depending on `igmp_report_state` and the therein contained
  839. /// timeouts, send IGMP membership reports.
  840. #[cfg(feature = "proto-igmp")]
  841. fn igmp_egress(&mut self) -> Result<bool> {
  842. match self.inner.igmp_report_state {
  843. IgmpReportState::ToSpecificQuery {
  844. version,
  845. timeout,
  846. group,
  847. } if self.inner.now >= timeout => {
  848. if let Some(pkt) = self.inner.igmp_report_packet(version, group) {
  849. // Send initial membership report
  850. let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
  851. self.inner.dispatch_ip(tx_token, pkt)?;
  852. }
  853. self.inner.igmp_report_state = IgmpReportState::Inactive;
  854. Ok(true)
  855. }
  856. IgmpReportState::ToGeneralQuery {
  857. version,
  858. timeout,
  859. interval,
  860. next_index,
  861. } if self.inner.now >= timeout => {
  862. let addr = self
  863. .inner
  864. .ipv4_multicast_groups
  865. .iter()
  866. .nth(next_index)
  867. .map(|(addr, ())| *addr);
  868. match addr {
  869. Some(addr) => {
  870. if let Some(pkt) = self.inner.igmp_report_packet(version, addr) {
  871. // Send initial membership report
  872. let tx_token = self.device.transmit().ok_or(Error::Exhausted)?;
  873. self.inner.dispatch_ip(tx_token, pkt)?;
  874. }
  875. let next_timeout = (timeout + interval).max(self.inner.now);
  876. self.inner.igmp_report_state = IgmpReportState::ToGeneralQuery {
  877. version,
  878. timeout: next_timeout,
  879. interval,
  880. next_index: next_index + 1,
  881. };
  882. Ok(true)
  883. }
  884. None => {
  885. self.inner.igmp_report_state = IgmpReportState::Inactive;
  886. Ok(false)
  887. }
  888. }
  889. }
  890. _ => Ok(false),
  891. }
  892. }
  893. }
  894. impl<'a> InterfaceInner<'a> {
  895. #[allow(unused)] // unused depending on which sockets are enabled
  896. pub(crate) fn now(&self) -> Instant {
  897. self.now
  898. }
  899. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  900. #[allow(unused)] // unused depending on which sockets are enabled
  901. pub(crate) fn hardware_addr(&self) -> Option<HardwareAddress> {
  902. self.hardware_addr
  903. }
  904. #[allow(unused)] // unused depending on which sockets are enabled
  905. pub(crate) fn checksum_caps(&self) -> ChecksumCapabilities {
  906. self.caps.checksum.clone()
  907. }
  908. #[allow(unused)] // unused depending on which sockets are enabled
  909. pub(crate) fn ip_mtu(&self) -> usize {
  910. self.caps.ip_mtu()
  911. }
  912. #[allow(unused)] // unused depending on which sockets are enabled, and in tests
  913. pub(crate) fn rand(&mut self) -> &mut Rand {
  914. &mut self.rand
  915. }
  916. #[allow(unused)] // unused depending on which sockets are enabled
  917. pub(crate) fn get_source_address(&mut self, dst_addr: IpAddress) -> Option<IpAddress> {
  918. let v = dst_addr.version();
  919. for cidr in self.ip_addrs.iter() {
  920. let addr = cidr.address();
  921. if addr.version() == v {
  922. return Some(addr);
  923. }
  924. }
  925. None
  926. }
  927. #[cfg(feature = "proto-ipv4")]
  928. #[allow(unused)]
  929. pub(crate) fn get_source_address_ipv4(
  930. &mut self,
  931. _dst_addr: Ipv4Address,
  932. ) -> Option<Ipv4Address> {
  933. for cidr in self.ip_addrs.iter() {
  934. #[allow(irrefutable_let_patterns)] // if only ipv4 is enabled
  935. if let IpCidr::Ipv4(cidr) = cidr {
  936. return Some(cidr.address());
  937. }
  938. }
  939. None
  940. }
  941. #[cfg(feature = "proto-ipv6")]
  942. #[allow(unused)]
  943. pub(crate) fn get_source_address_ipv6(
  944. &mut self,
  945. _dst_addr: Ipv6Address,
  946. ) -> Option<Ipv6Address> {
  947. for cidr in self.ip_addrs.iter() {
  948. #[allow(irrefutable_let_patterns)] // if only ipv6 is enabled
  949. if let IpCidr::Ipv6(cidr) = cidr {
  950. return Some(cidr.address());
  951. }
  952. }
  953. None
  954. }
  955. #[cfg(test)]
  956. pub(crate) fn mock() -> Self {
  957. Self {
  958. caps: DeviceCapabilities {
  959. #[cfg(feature = "medium-ethernet")]
  960. medium: crate::phy::Medium::Ethernet,
  961. #[cfg(not(feature = "medium-ethernet"))]
  962. medium: crate::phy::Medium::Ip,
  963. checksum: crate::phy::ChecksumCapabilities {
  964. #[cfg(feature = "proto-ipv4")]
  965. icmpv4: crate::phy::Checksum::Both,
  966. #[cfg(feature = "proto-ipv6")]
  967. icmpv6: crate::phy::Checksum::Both,
  968. ipv4: crate::phy::Checksum::Both,
  969. tcp: crate::phy::Checksum::Both,
  970. udp: crate::phy::Checksum::Both,
  971. },
  972. max_burst_size: None,
  973. #[cfg(feature = "medium-ethernet")]
  974. max_transmission_unit: 1514,
  975. #[cfg(not(feature = "medium-ethernet"))]
  976. max_transmission_unit: 1500,
  977. },
  978. now: Instant::from_millis_const(0),
  979. ip_addrs: ManagedSlice::Owned(vec![
  980. #[cfg(feature = "proto-ipv4")]
  981. IpCidr::Ipv4(Ipv4Cidr::new(Ipv4Address::new(192, 168, 1, 1), 24)),
  982. #[cfg(feature = "proto-ipv6")]
  983. IpCidr::Ipv6(Ipv6Cidr::new(
  984. Ipv6Address([0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1]),
  985. 64,
  986. )),
  987. ]),
  988. rand: Rand::new(1234),
  989. routes: Routes::new(&mut [][..]),
  990. #[cfg(feature = "proto-ipv4")]
  991. any_ip: false,
  992. #[cfg(feature = "medium-ieee802154")]
  993. pan_id: Some(crate::wire::Ieee802154Pan(0xabcd)),
  994. #[cfg(feature = "medium-ieee802154")]
  995. sequence_no: 0,
  996. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  997. hardware_addr: Some(crate::wire::HardwareAddress::Ethernet(
  998. crate::wire::EthernetAddress([0x02, 0x02, 0x02, 0x02, 0x02, 0x02]),
  999. )),
  1000. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  1001. neighbor_cache: None,
  1002. #[cfg(feature = "proto-igmp")]
  1003. igmp_report_state: IgmpReportState::Inactive,
  1004. #[cfg(feature = "proto-igmp")]
  1005. ipv4_multicast_groups: ManagedMap::Borrowed(&mut []),
  1006. }
  1007. }
  1008. #[cfg(test)]
  1009. #[allow(unused)] // unused depending on which sockets are enabled
  1010. pub(crate) fn set_now(&mut self, now: Instant) {
  1011. self.now = now
  1012. }
  1013. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  1014. fn check_hardware_addr(addr: &HardwareAddress) {
  1015. if !addr.is_unicast() {
  1016. panic!("Ethernet address {} is not unicast", addr)
  1017. }
  1018. }
  1019. fn check_ip_addrs(addrs: &[IpCidr]) {
  1020. for cidr in addrs {
  1021. if !cidr.address().is_unicast() && !cidr.address().is_unspecified() {
  1022. panic!("IP address {} is not unicast", cidr.address())
  1023. }
  1024. }
  1025. }
  1026. #[cfg(feature = "medium-ieee802154")]
  1027. fn get_sequence_number(&mut self) -> u8 {
  1028. let no = self.sequence_no;
  1029. self.sequence_no = self.sequence_no.wrapping_add(1);
  1030. no
  1031. }
  1032. /// Determine if the given `Ipv6Address` is the solicited node
  1033. /// multicast address for a IPv6 addresses assigned to the interface.
  1034. /// See [RFC 4291 § 2.7.1] for more details.
  1035. ///
  1036. /// [RFC 4291 § 2.7.1]: https://tools.ietf.org/html/rfc4291#section-2.7.1
  1037. #[cfg(feature = "proto-ipv6")]
  1038. pub fn has_solicited_node(&self, addr: Ipv6Address) -> bool {
  1039. self.ip_addrs.iter().any(|cidr| {
  1040. match *cidr {
  1041. IpCidr::Ipv6(cidr) if cidr.address() != Ipv6Address::LOOPBACK => {
  1042. // Take the lower order 24 bits of the IPv6 address and
  1043. // append those bits to FF02:0:0:0:0:1:FF00::/104.
  1044. addr.as_bytes()[14..] == cidr.address().as_bytes()[14..]
  1045. }
  1046. _ => false,
  1047. }
  1048. })
  1049. }
  1050. /// Check whether the interface has the given IP address assigned.
  1051. fn has_ip_addr<T: Into<IpAddress>>(&self, addr: T) -> bool {
  1052. let addr = addr.into();
  1053. self.ip_addrs.iter().any(|probe| probe.address() == addr)
  1054. }
  1055. /// Get the first IPv4 address of the interface.
  1056. #[cfg(feature = "proto-ipv4")]
  1057. pub fn ipv4_address(&self) -> Option<Ipv4Address> {
  1058. self.ip_addrs
  1059. .iter()
  1060. .filter_map(|addr| match *addr {
  1061. IpCidr::Ipv4(cidr) => Some(cidr.address()),
  1062. #[cfg(feature = "proto-ipv6")]
  1063. IpCidr::Ipv6(_) => None,
  1064. })
  1065. .next()
  1066. }
  1067. /// Check whether the interface listens to given destination multicast IP address.
  1068. ///
  1069. /// If built without feature `proto-igmp` this function will
  1070. /// always return `false`.
  1071. pub fn has_multicast_group<T: Into<IpAddress>>(&self, addr: T) -> bool {
  1072. match addr.into() {
  1073. #[cfg(feature = "proto-igmp")]
  1074. IpAddress::Ipv4(key) => {
  1075. key == Ipv4Address::MULTICAST_ALL_SYSTEMS
  1076. || self.ipv4_multicast_groups.get(&key).is_some()
  1077. }
  1078. #[allow(unreachable_patterns)]
  1079. _ => false,
  1080. }
  1081. }
  1082. #[cfg(feature = "medium-ethernet")]
  1083. fn process_ethernet<'frame, T: AsRef<[u8]>>(
  1084. &mut self,
  1085. sockets: &mut SocketSet,
  1086. frame: &'frame T,
  1087. ) -> Option<EthernetPacket<'frame>> {
  1088. let eth_frame = check!(EthernetFrame::new_checked(frame));
  1089. // Ignore any packets not directed to our hardware address or any of the multicast groups.
  1090. if !eth_frame.dst_addr().is_broadcast()
  1091. && !eth_frame.dst_addr().is_multicast()
  1092. && HardwareAddress::Ethernet(eth_frame.dst_addr()) != self.hardware_addr.unwrap()
  1093. {
  1094. return None;
  1095. }
  1096. match eth_frame.ethertype() {
  1097. #[cfg(feature = "proto-ipv4")]
  1098. EthernetProtocol::Arp => self.process_arp(self.now, &eth_frame),
  1099. #[cfg(feature = "proto-ipv4")]
  1100. EthernetProtocol::Ipv4 => {
  1101. let ipv4_packet = check!(Ipv4Packet::new_checked(eth_frame.payload()));
  1102. self.process_ipv4(sockets, &ipv4_packet)
  1103. .map(EthernetPacket::Ip)
  1104. }
  1105. #[cfg(feature = "proto-ipv6")]
  1106. EthernetProtocol::Ipv6 => {
  1107. let ipv6_packet = check!(Ipv6Packet::new_checked(eth_frame.payload()));
  1108. self.process_ipv6(sockets, &ipv6_packet)
  1109. .map(EthernetPacket::Ip)
  1110. }
  1111. // Drop all other traffic.
  1112. _ => None,
  1113. }
  1114. }
  1115. #[cfg(feature = "medium-ip")]
  1116. fn process_ip<'frame, T: AsRef<[u8]>>(
  1117. &mut self,
  1118. sockets: &mut SocketSet,
  1119. ip_payload: &'frame T,
  1120. ) -> Option<IpPacket<'frame>> {
  1121. match IpVersion::of_packet(ip_payload.as_ref()) {
  1122. #[cfg(feature = "proto-ipv4")]
  1123. Ok(IpVersion::Ipv4) => {
  1124. let ipv4_packet = check!(Ipv4Packet::new_checked(ip_payload));
  1125. self.process_ipv4(sockets, &ipv4_packet)
  1126. }
  1127. #[cfg(feature = "proto-ipv6")]
  1128. Ok(IpVersion::Ipv6) => {
  1129. let ipv6_packet = check!(Ipv6Packet::new_checked(ip_payload));
  1130. self.process_ipv6(sockets, &ipv6_packet)
  1131. }
  1132. // Drop all other traffic.
  1133. _ => None,
  1134. }
  1135. }
  1136. #[cfg(feature = "medium-ieee802154")]
  1137. fn process_ieee802154<'frame, T: AsRef<[u8]> + ?Sized>(
  1138. &mut self,
  1139. sockets: &mut SocketSet,
  1140. sixlowpan_payload: &'frame T,
  1141. ) -> Option<IpPacket<'frame>> {
  1142. let ieee802154_frame = check!(Ieee802154Frame::new_checked(sixlowpan_payload));
  1143. let ieee802154_repr = check!(Ieee802154Repr::parse(&ieee802154_frame));
  1144. if ieee802154_repr.frame_type != Ieee802154FrameType::Data {
  1145. return None;
  1146. }
  1147. // Drop frames when the user has set a PAN id and the PAN id from frame is not equal to this
  1148. // When the user didn't set a PAN id (so it is None), then we accept all PAN id's.
  1149. // We always accept the broadcast PAN id.
  1150. if self.pan_id.is_some()
  1151. && ieee802154_repr.dst_pan_id != self.pan_id
  1152. && ieee802154_repr.dst_pan_id != Some(Ieee802154Pan::BROADCAST)
  1153. {
  1154. net_debug!(
  1155. "dropping {:?} because not our PAN id (or not broadcast)",
  1156. ieee802154_repr
  1157. );
  1158. return None;
  1159. }
  1160. match ieee802154_frame.payload() {
  1161. Some(payload) => self.process_sixlowpan(sockets, &ieee802154_repr, payload),
  1162. None => None,
  1163. }
  1164. }
  1165. #[cfg(feature = "proto-sixlowpan")]
  1166. fn process_sixlowpan<'frame, T: AsRef<[u8]> + ?Sized>(
  1167. &mut self,
  1168. sockets: &mut SocketSet,
  1169. ieee802154_repr: &Ieee802154Repr,
  1170. payload: &'frame T,
  1171. ) -> Option<IpPacket<'frame>> {
  1172. // The first header needs to be an IPHC header.
  1173. let iphc_packet = check!(SixlowpanIphcPacket::new_checked(payload));
  1174. let iphc_repr = check!(SixlowpanIphcRepr::parse(
  1175. &iphc_packet,
  1176. ieee802154_repr.src_addr,
  1177. ieee802154_repr.dst_addr,
  1178. ));
  1179. let payload = iphc_packet.payload();
  1180. let mut ipv6_repr = Ipv6Repr {
  1181. src_addr: iphc_repr.src_addr,
  1182. dst_addr: iphc_repr.dst_addr,
  1183. hop_limit: iphc_repr.hop_limit,
  1184. next_header: IpProtocol::Unknown(0),
  1185. payload_len: iphc_repr.buffer_len(),
  1186. };
  1187. // Currently we assume the next header is a UDP, so we ignore everything else.
  1188. match iphc_repr.next_header {
  1189. SixlowpanNextHeader::Compressed => {
  1190. match check!(SixlowpanNhcPacket::dispatch(payload)) {
  1191. SixlowpanNhcPacket::ExtensionHeader(_) => {
  1192. net_debug!("Extension headers are currently not supported for 6LoWPAN");
  1193. None
  1194. }
  1195. #[cfg(not(feature = "socket-udp"))]
  1196. SixlowpanNhcPacket::UdpHeader(_) => {
  1197. net_debug!("UDP support is disabled, enable cargo feature `socket-udp`.");
  1198. None
  1199. }
  1200. #[cfg(feature = "socket-udp")]
  1201. SixlowpanNhcPacket::UdpHeader(udp_packet) => {
  1202. ipv6_repr.next_header = IpProtocol::Udp;
  1203. // Handle the UDP
  1204. let udp_repr = check!(SixlowpanUdpRepr::parse(
  1205. &udp_packet,
  1206. &iphc_repr.src_addr,
  1207. &iphc_repr.dst_addr,
  1208. udp_packet.checksum(),
  1209. ));
  1210. // Look for UDP sockets that will accept the UDP packet.
  1211. // If it does not accept the packet, then send an ICMP message.
  1212. for udp_socket in sockets
  1213. .items_mut()
  1214. .filter_map(|i| udp::Socket::downcast(&mut i.socket))
  1215. {
  1216. if udp_socket.accepts(self, &IpRepr::Ipv6(ipv6_repr), &udp_repr) {
  1217. udp_socket.process(
  1218. self,
  1219. &IpRepr::Ipv6(ipv6_repr),
  1220. &udp_repr,
  1221. udp_packet.payload(),
  1222. );
  1223. return None;
  1224. }
  1225. }
  1226. let payload_len = icmp_reply_payload_len(
  1227. payload.len(),
  1228. IPV6_MIN_MTU,
  1229. ipv6_repr.buffer_len(),
  1230. );
  1231. let icmpv6_reply_repr = Icmpv6Repr::DstUnreachable {
  1232. reason: Icmpv6DstUnreachable::PortUnreachable,
  1233. header: ipv6_repr,
  1234. data: &payload[0..payload_len],
  1235. };
  1236. self.icmpv6_reply(ipv6_repr, icmpv6_reply_repr)
  1237. }
  1238. }
  1239. }
  1240. SixlowpanNextHeader::Uncompressed(nxt_hdr) => match nxt_hdr {
  1241. IpProtocol::Icmpv6 => {
  1242. ipv6_repr.next_header = IpProtocol::Icmpv6;
  1243. self.process_icmpv6(sockets, IpRepr::Ipv6(ipv6_repr), iphc_packet.payload())
  1244. }
  1245. _ => {
  1246. net_debug!("Headers other than ICMPv6 and compressed headers are currently not supported for 6LoWPAN");
  1247. None
  1248. }
  1249. },
  1250. }
  1251. }
  1252. #[cfg(all(feature = "medium-ethernet", feature = "proto-ipv4"))]
  1253. fn process_arp<'frame, T: AsRef<[u8]>>(
  1254. &mut self,
  1255. timestamp: Instant,
  1256. eth_frame: &EthernetFrame<&'frame T>,
  1257. ) -> Option<EthernetPacket<'frame>> {
  1258. let arp_packet = check!(ArpPacket::new_checked(eth_frame.payload()));
  1259. let arp_repr = check!(ArpRepr::parse(&arp_packet));
  1260. match arp_repr {
  1261. ArpRepr::EthernetIpv4 {
  1262. operation,
  1263. source_hardware_addr,
  1264. source_protocol_addr,
  1265. target_protocol_addr,
  1266. ..
  1267. } => {
  1268. // Only process ARP packets for us.
  1269. if !self.has_ip_addr(target_protocol_addr) {
  1270. return None;
  1271. }
  1272. // Only process REQUEST and RESPONSE.
  1273. if let ArpOperation::Unknown(_) = operation {
  1274. net_debug!("arp: unknown operation code");
  1275. return None;
  1276. }
  1277. // Discard packets with non-unicast source addresses.
  1278. if !source_protocol_addr.is_unicast() || !source_hardware_addr.is_unicast() {
  1279. net_debug!("arp: non-unicast source address");
  1280. return None;
  1281. }
  1282. if !self.in_same_network(&IpAddress::Ipv4(source_protocol_addr)) {
  1283. net_debug!("arp: source IP address not in same network as us");
  1284. return None;
  1285. }
  1286. // Fill the ARP cache from any ARP packet aimed at us (both request or response).
  1287. // We fill from requests too because if someone is requesting our address they
  1288. // are probably going to talk to us, so we avoid having to request their address
  1289. // when we later reply to them.
  1290. self.neighbor_cache.as_mut().unwrap().fill(
  1291. source_protocol_addr.into(),
  1292. source_hardware_addr.into(),
  1293. timestamp,
  1294. );
  1295. if operation == ArpOperation::Request {
  1296. let src_hardware_addr = match self.hardware_addr {
  1297. Some(HardwareAddress::Ethernet(addr)) => addr,
  1298. _ => unreachable!(),
  1299. };
  1300. Some(EthernetPacket::Arp(ArpRepr::EthernetIpv4 {
  1301. operation: ArpOperation::Reply,
  1302. source_hardware_addr: src_hardware_addr,
  1303. source_protocol_addr: target_protocol_addr,
  1304. target_hardware_addr: source_hardware_addr,
  1305. target_protocol_addr: source_protocol_addr,
  1306. }))
  1307. } else {
  1308. None
  1309. }
  1310. }
  1311. }
  1312. }
  1313. #[cfg(feature = "socket-raw")]
  1314. fn raw_socket_filter<'frame>(
  1315. &mut self,
  1316. sockets: &mut SocketSet,
  1317. ip_repr: &IpRepr,
  1318. ip_payload: &'frame [u8],
  1319. ) -> bool {
  1320. let mut handled_by_raw_socket = false;
  1321. // Pass every IP packet to all raw sockets we have registered.
  1322. for raw_socket in sockets
  1323. .items_mut()
  1324. .filter_map(|i| raw::Socket::downcast(&mut i.socket))
  1325. {
  1326. if raw_socket.accepts(ip_repr) {
  1327. raw_socket.process(self, ip_repr, ip_payload);
  1328. handled_by_raw_socket = true;
  1329. }
  1330. }
  1331. handled_by_raw_socket
  1332. }
  1333. #[cfg(feature = "proto-ipv6")]
  1334. fn process_ipv6<'frame, T: AsRef<[u8]> + ?Sized>(
  1335. &mut self,
  1336. sockets: &mut SocketSet,
  1337. ipv6_packet: &Ipv6Packet<&'frame T>,
  1338. ) -> Option<IpPacket<'frame>> {
  1339. let ipv6_repr = check!(Ipv6Repr::parse(ipv6_packet));
  1340. if !ipv6_repr.src_addr.is_unicast() {
  1341. // Discard packets with non-unicast source addresses.
  1342. net_debug!("non-unicast source address");
  1343. return None;
  1344. }
  1345. let ip_payload = ipv6_packet.payload();
  1346. #[cfg(feature = "socket-raw")]
  1347. let handled_by_raw_socket = self.raw_socket_filter(sockets, &ipv6_repr.into(), ip_payload);
  1348. #[cfg(not(feature = "socket-raw"))]
  1349. let handled_by_raw_socket = false;
  1350. self.process_nxt_hdr(
  1351. sockets,
  1352. ipv6_repr,
  1353. ipv6_repr.next_header,
  1354. handled_by_raw_socket,
  1355. ip_payload,
  1356. )
  1357. }
  1358. /// Given the next header value forward the payload onto the correct process
  1359. /// function.
  1360. #[cfg(feature = "proto-ipv6")]
  1361. fn process_nxt_hdr<'frame>(
  1362. &mut self,
  1363. sockets: &mut SocketSet,
  1364. ipv6_repr: Ipv6Repr,
  1365. nxt_hdr: IpProtocol,
  1366. handled_by_raw_socket: bool,
  1367. ip_payload: &'frame [u8],
  1368. ) -> Option<IpPacket<'frame>> {
  1369. match nxt_hdr {
  1370. IpProtocol::Icmpv6 => self.process_icmpv6(sockets, ipv6_repr.into(), ip_payload),
  1371. #[cfg(any(feature = "socket-udp", feature = "socket-dns"))]
  1372. IpProtocol::Udp => {
  1373. self.process_udp(sockets, ipv6_repr.into(), handled_by_raw_socket, ip_payload)
  1374. }
  1375. #[cfg(feature = "socket-tcp")]
  1376. IpProtocol::Tcp => self.process_tcp(sockets, ipv6_repr.into(), ip_payload),
  1377. IpProtocol::HopByHop => {
  1378. self.process_hopbyhop(sockets, ipv6_repr, handled_by_raw_socket, ip_payload)
  1379. }
  1380. #[cfg(feature = "socket-raw")]
  1381. _ if handled_by_raw_socket => None,
  1382. _ => {
  1383. // Send back as much of the original payload as we can.
  1384. let payload_len =
  1385. icmp_reply_payload_len(ip_payload.len(), IPV6_MIN_MTU, ipv6_repr.buffer_len());
  1386. let icmp_reply_repr = Icmpv6Repr::ParamProblem {
  1387. reason: Icmpv6ParamProblem::UnrecognizedNxtHdr,
  1388. // The offending packet is after the IPv6 header.
  1389. pointer: ipv6_repr.buffer_len() as u32,
  1390. header: ipv6_repr,
  1391. data: &ip_payload[0..payload_len],
  1392. };
  1393. self.icmpv6_reply(ipv6_repr, icmp_reply_repr)
  1394. }
  1395. }
  1396. }
  1397. #[cfg(feature = "proto-ipv4")]
  1398. fn process_ipv4<'frame, T: AsRef<[u8]> + ?Sized>(
  1399. &mut self,
  1400. sockets: &mut SocketSet,
  1401. ipv4_packet: &Ipv4Packet<&'frame T>,
  1402. ) -> Option<IpPacket<'frame>> {
  1403. let ipv4_repr = check!(Ipv4Repr::parse(ipv4_packet, &self.caps.checksum));
  1404. if !self.is_unicast_v4(ipv4_repr.src_addr) {
  1405. // Discard packets with non-unicast source addresses.
  1406. net_debug!("non-unicast source address");
  1407. return None;
  1408. }
  1409. let ip_repr = IpRepr::Ipv4(ipv4_repr);
  1410. let ip_payload = ipv4_packet.payload();
  1411. #[cfg(feature = "socket-raw")]
  1412. let handled_by_raw_socket = self.raw_socket_filter(sockets, &ip_repr, ip_payload);
  1413. #[cfg(not(feature = "socket-raw"))]
  1414. let handled_by_raw_socket = false;
  1415. #[cfg(feature = "socket-dhcpv4")]
  1416. {
  1417. if ipv4_repr.next_header == IpProtocol::Udp && self.hardware_addr.is_some() {
  1418. // First check for source and dest ports, then do `UdpRepr::parse` if they match.
  1419. // This way we avoid validating the UDP checksum twice for all non-DHCP UDP packets (one here, one in `process_udp`)
  1420. let udp_packet = check!(UdpPacket::new_checked(ip_payload));
  1421. if udp_packet.src_port() == DHCP_SERVER_PORT
  1422. && udp_packet.dst_port() == DHCP_CLIENT_PORT
  1423. {
  1424. if let Some(dhcp_socket) = sockets
  1425. .items_mut()
  1426. .filter_map(|i| dhcpv4::Socket::downcast(&mut i.socket))
  1427. .next()
  1428. {
  1429. let (src_addr, dst_addr) = (ip_repr.src_addr(), ip_repr.dst_addr());
  1430. let udp_repr = check!(UdpRepr::parse(
  1431. &udp_packet,
  1432. &src_addr,
  1433. &dst_addr,
  1434. &self.caps.checksum
  1435. ));
  1436. let udp_payload = udp_packet.payload();
  1437. dhcp_socket.process(self, &ipv4_repr, &udp_repr, udp_payload);
  1438. return None;
  1439. }
  1440. }
  1441. }
  1442. }
  1443. if !self.has_ip_addr(ipv4_repr.dst_addr)
  1444. && !self.has_multicast_group(ipv4_repr.dst_addr)
  1445. && !self.is_broadcast_v4(ipv4_repr.dst_addr)
  1446. {
  1447. // Ignore IP packets not directed at us, or broadcast, or any of the multicast groups.
  1448. // If AnyIP is enabled, also check if the packet is routed locally.
  1449. if !self.any_ip
  1450. || !ipv4_repr.dst_addr.is_unicast()
  1451. || self
  1452. .routes
  1453. .lookup(&IpAddress::Ipv4(ipv4_repr.dst_addr), self.now)
  1454. .map_or(true, |router_addr| !self.has_ip_addr(router_addr))
  1455. {
  1456. return None;
  1457. }
  1458. }
  1459. match ipv4_repr.next_header {
  1460. IpProtocol::Icmp => self.process_icmpv4(sockets, ip_repr, ip_payload),
  1461. #[cfg(feature = "proto-igmp")]
  1462. IpProtocol::Igmp => self.process_igmp(ipv4_repr, ip_payload),
  1463. #[cfg(any(feature = "socket-udp", feature = "socket-dns"))]
  1464. IpProtocol::Udp => {
  1465. self.process_udp(sockets, ip_repr, handled_by_raw_socket, ip_payload)
  1466. }
  1467. #[cfg(feature = "socket-tcp")]
  1468. IpProtocol::Tcp => self.process_tcp(sockets, ip_repr, ip_payload),
  1469. _ if handled_by_raw_socket => None,
  1470. _ => {
  1471. // Send back as much of the original payload as we can.
  1472. let payload_len =
  1473. icmp_reply_payload_len(ip_payload.len(), IPV4_MIN_MTU, ipv4_repr.buffer_len());
  1474. let icmp_reply_repr = Icmpv4Repr::DstUnreachable {
  1475. reason: Icmpv4DstUnreachable::ProtoUnreachable,
  1476. header: ipv4_repr,
  1477. data: &ip_payload[0..payload_len],
  1478. };
  1479. self.icmpv4_reply(ipv4_repr, icmp_reply_repr)
  1480. }
  1481. }
  1482. }
  1483. /// Checks if an incoming packet has a broadcast address for the interfaces
  1484. /// associated ipv4 addresses.
  1485. #[cfg(feature = "proto-ipv4")]
  1486. fn is_subnet_broadcast(&self, address: Ipv4Address) -> bool {
  1487. self.ip_addrs
  1488. .iter()
  1489. .filter_map(|own_cidr| match own_cidr {
  1490. IpCidr::Ipv4(own_ip) => Some(own_ip.broadcast()?),
  1491. #[cfg(feature = "proto-ipv6")]
  1492. IpCidr::Ipv6(_) => None,
  1493. })
  1494. .any(|broadcast_address| address == broadcast_address)
  1495. }
  1496. /// Checks if an ipv4 address is broadcast, taking into account subnet broadcast addresses
  1497. #[cfg(feature = "proto-ipv4")]
  1498. fn is_broadcast_v4(&self, address: Ipv4Address) -> bool {
  1499. address.is_broadcast() || self.is_subnet_broadcast(address)
  1500. }
  1501. /// Checks if an ipv4 address is unicast, taking into account subnet broadcast addresses
  1502. #[cfg(feature = "proto-ipv4")]
  1503. fn is_unicast_v4(&self, address: Ipv4Address) -> bool {
  1504. address.is_unicast() && !self.is_subnet_broadcast(address)
  1505. }
  1506. /// Host duties of the **IGMPv2** protocol.
  1507. ///
  1508. /// Sets up `igmp_report_state` for responding to IGMP general/specific membership queries.
  1509. /// Membership must not be reported immediately in order to avoid flooding the network
  1510. /// after a query is broadcasted by a router; this is not currently done.
  1511. #[cfg(feature = "proto-igmp")]
  1512. fn process_igmp<'frame>(
  1513. &mut self,
  1514. ipv4_repr: Ipv4Repr,
  1515. ip_payload: &'frame [u8],
  1516. ) -> Option<IpPacket<'frame>> {
  1517. let igmp_packet = check!(IgmpPacket::new_checked(ip_payload));
  1518. let igmp_repr = check!(IgmpRepr::parse(&igmp_packet));
  1519. // FIXME: report membership after a delay
  1520. match igmp_repr {
  1521. IgmpRepr::MembershipQuery {
  1522. group_addr,
  1523. version,
  1524. max_resp_time,
  1525. } => {
  1526. // General query
  1527. if group_addr.is_unspecified()
  1528. && ipv4_repr.dst_addr == Ipv4Address::MULTICAST_ALL_SYSTEMS
  1529. {
  1530. // Are we member in any groups?
  1531. if self.ipv4_multicast_groups.iter().next().is_some() {
  1532. let interval = match version {
  1533. IgmpVersion::Version1 => Duration::from_millis(100),
  1534. IgmpVersion::Version2 => {
  1535. // No dependence on a random generator
  1536. // (see [#24](https://github.com/m-labs/smoltcp/issues/24))
  1537. // but at least spread reports evenly across max_resp_time.
  1538. let intervals = self.ipv4_multicast_groups.len() as u32 + 1;
  1539. max_resp_time / intervals
  1540. }
  1541. };
  1542. self.igmp_report_state = IgmpReportState::ToGeneralQuery {
  1543. version,
  1544. timeout: self.now + interval,
  1545. interval,
  1546. next_index: 0,
  1547. };
  1548. }
  1549. } else {
  1550. // Group-specific query
  1551. if self.has_multicast_group(group_addr) && ipv4_repr.dst_addr == group_addr {
  1552. // Don't respond immediately
  1553. let timeout = max_resp_time / 4;
  1554. self.igmp_report_state = IgmpReportState::ToSpecificQuery {
  1555. version,
  1556. timeout: self.now + timeout,
  1557. group: group_addr,
  1558. };
  1559. }
  1560. }
  1561. }
  1562. // Ignore membership reports
  1563. IgmpRepr::MembershipReport { .. } => (),
  1564. // Ignore hosts leaving groups
  1565. IgmpRepr::LeaveGroup { .. } => (),
  1566. }
  1567. None
  1568. }
  1569. #[cfg(feature = "proto-ipv6")]
  1570. fn process_icmpv6<'frame>(
  1571. &mut self,
  1572. _sockets: &mut SocketSet,
  1573. ip_repr: IpRepr,
  1574. ip_payload: &'frame [u8],
  1575. ) -> Option<IpPacket<'frame>> {
  1576. let icmp_packet = check!(Icmpv6Packet::new_checked(ip_payload));
  1577. let icmp_repr = check!(Icmpv6Repr::parse(
  1578. &ip_repr.src_addr(),
  1579. &ip_repr.dst_addr(),
  1580. &icmp_packet,
  1581. &self.caps.checksum,
  1582. ));
  1583. #[cfg(feature = "socket-icmp")]
  1584. let mut handled_by_icmp_socket = false;
  1585. #[cfg(all(feature = "socket-icmp", feature = "proto-ipv6"))]
  1586. for icmp_socket in _sockets
  1587. .items_mut()
  1588. .filter_map(|i| icmp::Socket::downcast(&mut i.socket))
  1589. {
  1590. if icmp_socket.accepts(self, &ip_repr, &icmp_repr.into()) {
  1591. icmp_socket.process(self, &ip_repr, &icmp_repr.into());
  1592. handled_by_icmp_socket = true;
  1593. }
  1594. }
  1595. match icmp_repr {
  1596. // Respond to echo requests.
  1597. Icmpv6Repr::EchoRequest {
  1598. ident,
  1599. seq_no,
  1600. data,
  1601. } => match ip_repr {
  1602. IpRepr::Ipv6(ipv6_repr) => {
  1603. let icmp_reply_repr = Icmpv6Repr::EchoReply {
  1604. ident,
  1605. seq_no,
  1606. data,
  1607. };
  1608. self.icmpv6_reply(ipv6_repr, icmp_reply_repr)
  1609. }
  1610. #[allow(unreachable_patterns)]
  1611. _ => unreachable!(),
  1612. },
  1613. // Ignore any echo replies.
  1614. Icmpv6Repr::EchoReply { .. } => None,
  1615. // Forward any NDISC packets to the ndisc packet handler
  1616. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  1617. Icmpv6Repr::Ndisc(repr) if ip_repr.hop_limit() == 0xff => match ip_repr {
  1618. IpRepr::Ipv6(ipv6_repr) => self.process_ndisc(ipv6_repr, repr),
  1619. #[allow(unreachable_patterns)]
  1620. _ => unreachable!(),
  1621. },
  1622. // Don't report an error if a packet with unknown type
  1623. // has been handled by an ICMP socket
  1624. #[cfg(feature = "socket-icmp")]
  1625. _ if handled_by_icmp_socket => None,
  1626. // FIXME: do something correct here?
  1627. _ => None,
  1628. }
  1629. }
  1630. #[cfg(all(
  1631. any(feature = "medium-ethernet", feature = "medium-ieee802154"),
  1632. feature = "proto-ipv6"
  1633. ))]
  1634. fn process_ndisc<'frame>(
  1635. &mut self,
  1636. ip_repr: Ipv6Repr,
  1637. repr: NdiscRepr<'frame>,
  1638. ) -> Option<IpPacket<'frame>> {
  1639. match repr {
  1640. NdiscRepr::NeighborAdvert {
  1641. lladdr,
  1642. target_addr,
  1643. flags,
  1644. } => {
  1645. let ip_addr = ip_repr.src_addr.into();
  1646. if let Some(lladdr) = lladdr {
  1647. let lladdr = check!(lladdr.parse(self.caps.medium));
  1648. if !lladdr.is_unicast() || !target_addr.is_unicast() {
  1649. return None;
  1650. }
  1651. if flags.contains(NdiscNeighborFlags::OVERRIDE)
  1652. || !self
  1653. .neighbor_cache
  1654. .as_mut()
  1655. .unwrap()
  1656. .lookup(&ip_addr, self.now)
  1657. .found()
  1658. {
  1659. self.neighbor_cache
  1660. .as_mut()
  1661. .unwrap()
  1662. .fill(ip_addr, lladdr, self.now)
  1663. }
  1664. }
  1665. None
  1666. }
  1667. NdiscRepr::NeighborSolicit {
  1668. target_addr,
  1669. lladdr,
  1670. ..
  1671. } => {
  1672. if let Some(lladdr) = lladdr {
  1673. let lladdr = check!(lladdr.parse(self.caps.medium));
  1674. if !lladdr.is_unicast() || !target_addr.is_unicast() {
  1675. return None;
  1676. }
  1677. self.neighbor_cache.as_mut().unwrap().fill(
  1678. ip_repr.src_addr.into(),
  1679. lladdr,
  1680. self.now,
  1681. );
  1682. }
  1683. if self.has_solicited_node(ip_repr.dst_addr) && self.has_ip_addr(target_addr) {
  1684. let advert = Icmpv6Repr::Ndisc(NdiscRepr::NeighborAdvert {
  1685. flags: NdiscNeighborFlags::SOLICITED,
  1686. target_addr,
  1687. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  1688. lladdr: Some(self.hardware_addr.unwrap().into()),
  1689. });
  1690. let ip_repr = Ipv6Repr {
  1691. src_addr: target_addr,
  1692. dst_addr: ip_repr.src_addr,
  1693. next_header: IpProtocol::Icmpv6,
  1694. hop_limit: 0xff,
  1695. payload_len: advert.buffer_len(),
  1696. };
  1697. Some(IpPacket::Icmpv6((ip_repr, advert)))
  1698. } else {
  1699. None
  1700. }
  1701. }
  1702. _ => None,
  1703. }
  1704. }
  1705. #[cfg(feature = "proto-ipv6")]
  1706. fn process_hopbyhop<'frame>(
  1707. &mut self,
  1708. sockets: &mut SocketSet,
  1709. ipv6_repr: Ipv6Repr,
  1710. handled_by_raw_socket: bool,
  1711. ip_payload: &'frame [u8],
  1712. ) -> Option<IpPacket<'frame>> {
  1713. let hbh_pkt = check!(Ipv6HopByHopHeader::new_checked(ip_payload));
  1714. let hbh_repr = check!(Ipv6HopByHopRepr::parse(&hbh_pkt));
  1715. for opt_repr in hbh_repr.options() {
  1716. let opt_repr = check!(opt_repr);
  1717. match opt_repr {
  1718. Ipv6OptionRepr::Pad1 | Ipv6OptionRepr::PadN(_) => (),
  1719. Ipv6OptionRepr::Unknown { type_, .. } => {
  1720. match Ipv6OptionFailureType::from(type_) {
  1721. Ipv6OptionFailureType::Skip => (),
  1722. Ipv6OptionFailureType::Discard => {
  1723. return None;
  1724. }
  1725. _ => {
  1726. // FIXME(dlrobertson): Send an ICMPv6 parameter problem message
  1727. // here.
  1728. return None;
  1729. }
  1730. }
  1731. }
  1732. }
  1733. }
  1734. self.process_nxt_hdr(
  1735. sockets,
  1736. ipv6_repr,
  1737. hbh_repr.next_header,
  1738. handled_by_raw_socket,
  1739. &ip_payload[hbh_repr.buffer_len()..],
  1740. )
  1741. }
  1742. #[cfg(feature = "proto-ipv4")]
  1743. fn process_icmpv4<'frame>(
  1744. &mut self,
  1745. _sockets: &mut SocketSet,
  1746. ip_repr: IpRepr,
  1747. ip_payload: &'frame [u8],
  1748. ) -> Option<IpPacket<'frame>> {
  1749. let icmp_packet = check!(Icmpv4Packet::new_checked(ip_payload));
  1750. let icmp_repr = check!(Icmpv4Repr::parse(&icmp_packet, &self.caps.checksum));
  1751. #[cfg(feature = "socket-icmp")]
  1752. let mut handled_by_icmp_socket = false;
  1753. #[cfg(all(feature = "socket-icmp", feature = "proto-ipv4"))]
  1754. for icmp_socket in _sockets
  1755. .items_mut()
  1756. .filter_map(|i| icmp::Socket::downcast(&mut i.socket))
  1757. {
  1758. if icmp_socket.accepts(self, &ip_repr, &icmp_repr.into()) {
  1759. icmp_socket.process(self, &ip_repr, &icmp_repr.into());
  1760. handled_by_icmp_socket = true;
  1761. }
  1762. }
  1763. match icmp_repr {
  1764. // Respond to echo requests.
  1765. #[cfg(feature = "proto-ipv4")]
  1766. Icmpv4Repr::EchoRequest {
  1767. ident,
  1768. seq_no,
  1769. data,
  1770. } => {
  1771. let icmp_reply_repr = Icmpv4Repr::EchoReply {
  1772. ident,
  1773. seq_no,
  1774. data,
  1775. };
  1776. match ip_repr {
  1777. IpRepr::Ipv4(ipv4_repr) => self.icmpv4_reply(ipv4_repr, icmp_reply_repr),
  1778. #[allow(unreachable_patterns)]
  1779. _ => unreachable!(),
  1780. }
  1781. }
  1782. // Ignore any echo replies.
  1783. Icmpv4Repr::EchoReply { .. } => None,
  1784. // Don't report an error if a packet with unknown type
  1785. // has been handled by an ICMP socket
  1786. #[cfg(feature = "socket-icmp")]
  1787. _ if handled_by_icmp_socket => None,
  1788. // FIXME: do something correct here?
  1789. _ => None,
  1790. }
  1791. }
  1792. #[cfg(feature = "proto-ipv4")]
  1793. fn icmpv4_reply<'frame, 'icmp: 'frame>(
  1794. &self,
  1795. ipv4_repr: Ipv4Repr,
  1796. icmp_repr: Icmpv4Repr<'icmp>,
  1797. ) -> Option<IpPacket<'frame>> {
  1798. if !self.is_unicast_v4(ipv4_repr.src_addr) {
  1799. // Do not send ICMP replies to non-unicast sources
  1800. None
  1801. } else if self.is_unicast_v4(ipv4_repr.dst_addr) {
  1802. // Reply as normal when src_addr and dst_addr are both unicast
  1803. let ipv4_reply_repr = Ipv4Repr {
  1804. src_addr: ipv4_repr.dst_addr,
  1805. dst_addr: ipv4_repr.src_addr,
  1806. next_header: IpProtocol::Icmp,
  1807. payload_len: icmp_repr.buffer_len(),
  1808. hop_limit: 64,
  1809. };
  1810. Some(IpPacket::Icmpv4((ipv4_reply_repr, icmp_repr)))
  1811. } else if self.is_broadcast_v4(ipv4_repr.dst_addr) {
  1812. // Only reply to broadcasts for echo replies and not other ICMP messages
  1813. match icmp_repr {
  1814. Icmpv4Repr::EchoReply { .. } => match self.ipv4_address() {
  1815. Some(src_addr) => {
  1816. let ipv4_reply_repr = Ipv4Repr {
  1817. src_addr: src_addr,
  1818. dst_addr: ipv4_repr.src_addr,
  1819. next_header: IpProtocol::Icmp,
  1820. payload_len: icmp_repr.buffer_len(),
  1821. hop_limit: 64,
  1822. };
  1823. Some(IpPacket::Icmpv4((ipv4_reply_repr, icmp_repr)))
  1824. }
  1825. None => None,
  1826. },
  1827. _ => None,
  1828. }
  1829. } else {
  1830. None
  1831. }
  1832. }
  1833. #[cfg(feature = "proto-ipv6")]
  1834. fn icmpv6_reply<'frame, 'icmp: 'frame>(
  1835. &self,
  1836. ipv6_repr: Ipv6Repr,
  1837. icmp_repr: Icmpv6Repr<'icmp>,
  1838. ) -> Option<IpPacket<'frame>> {
  1839. if ipv6_repr.dst_addr.is_unicast() {
  1840. let ipv6_reply_repr = Ipv6Repr {
  1841. src_addr: ipv6_repr.dst_addr,
  1842. dst_addr: ipv6_repr.src_addr,
  1843. next_header: IpProtocol::Icmpv6,
  1844. payload_len: icmp_repr.buffer_len(),
  1845. hop_limit: 64,
  1846. };
  1847. Some(IpPacket::Icmpv6((ipv6_reply_repr, icmp_repr)))
  1848. } else {
  1849. // Do not send any ICMP replies to a broadcast destination address.
  1850. None
  1851. }
  1852. }
  1853. #[cfg(any(feature = "socket-udp", feature = "socket-dns"))]
  1854. fn process_udp<'frame>(
  1855. &mut self,
  1856. sockets: &mut SocketSet,
  1857. ip_repr: IpRepr,
  1858. handled_by_raw_socket: bool,
  1859. ip_payload: &'frame [u8],
  1860. ) -> Option<IpPacket<'frame>> {
  1861. let (src_addr, dst_addr) = (ip_repr.src_addr(), ip_repr.dst_addr());
  1862. let udp_packet = check!(UdpPacket::new_checked(ip_payload));
  1863. let udp_repr = check!(UdpRepr::parse(
  1864. &udp_packet,
  1865. &src_addr,
  1866. &dst_addr,
  1867. &self.caps.checksum
  1868. ));
  1869. let udp_payload = udp_packet.payload();
  1870. #[cfg(feature = "socket-udp")]
  1871. for udp_socket in sockets
  1872. .items_mut()
  1873. .filter_map(|i| udp::Socket::downcast(&mut i.socket))
  1874. {
  1875. if udp_socket.accepts(self, &ip_repr, &udp_repr) {
  1876. udp_socket.process(self, &ip_repr, &udp_repr, udp_payload);
  1877. return None;
  1878. }
  1879. }
  1880. #[cfg(feature = "socket-dns")]
  1881. for dns_socket in sockets
  1882. .items_mut()
  1883. .filter_map(|i| dns::Socket::downcast(&mut i.socket))
  1884. {
  1885. if dns_socket.accepts(&ip_repr, &udp_repr) {
  1886. dns_socket.process(self, &ip_repr, &udp_repr, udp_payload);
  1887. return None;
  1888. }
  1889. }
  1890. // The packet wasn't handled by a socket, send an ICMP port unreachable packet.
  1891. match ip_repr {
  1892. #[cfg(feature = "proto-ipv4")]
  1893. IpRepr::Ipv4(_) if handled_by_raw_socket => None,
  1894. #[cfg(feature = "proto-ipv6")]
  1895. IpRepr::Ipv6(_) if handled_by_raw_socket => None,
  1896. #[cfg(feature = "proto-ipv4")]
  1897. IpRepr::Ipv4(ipv4_repr) => {
  1898. let payload_len =
  1899. icmp_reply_payload_len(ip_payload.len(), IPV4_MIN_MTU, ipv4_repr.buffer_len());
  1900. let icmpv4_reply_repr = Icmpv4Repr::DstUnreachable {
  1901. reason: Icmpv4DstUnreachable::PortUnreachable,
  1902. header: ipv4_repr,
  1903. data: &ip_payload[0..payload_len],
  1904. };
  1905. self.icmpv4_reply(ipv4_repr, icmpv4_reply_repr)
  1906. }
  1907. #[cfg(feature = "proto-ipv6")]
  1908. IpRepr::Ipv6(ipv6_repr) => {
  1909. let payload_len =
  1910. icmp_reply_payload_len(ip_payload.len(), IPV6_MIN_MTU, ipv6_repr.buffer_len());
  1911. let icmpv6_reply_repr = Icmpv6Repr::DstUnreachable {
  1912. reason: Icmpv6DstUnreachable::PortUnreachable,
  1913. header: ipv6_repr,
  1914. data: &ip_payload[0..payload_len],
  1915. };
  1916. self.icmpv6_reply(ipv6_repr, icmpv6_reply_repr)
  1917. }
  1918. }
  1919. }
  1920. #[cfg(feature = "socket-tcp")]
  1921. fn process_tcp<'frame>(
  1922. &mut self,
  1923. sockets: &mut SocketSet,
  1924. ip_repr: IpRepr,
  1925. ip_payload: &'frame [u8],
  1926. ) -> Option<IpPacket<'frame>> {
  1927. let (src_addr, dst_addr) = (ip_repr.src_addr(), ip_repr.dst_addr());
  1928. let tcp_packet = check!(TcpPacket::new_checked(ip_payload));
  1929. let tcp_repr = check!(TcpRepr::parse(
  1930. &tcp_packet,
  1931. &src_addr,
  1932. &dst_addr,
  1933. &self.caps.checksum
  1934. ));
  1935. for tcp_socket in sockets
  1936. .items_mut()
  1937. .filter_map(|i| tcp::Socket::downcast(&mut i.socket))
  1938. {
  1939. if tcp_socket.accepts(self, &ip_repr, &tcp_repr) {
  1940. return tcp_socket
  1941. .process(self, &ip_repr, &tcp_repr)
  1942. .map(IpPacket::Tcp);
  1943. }
  1944. }
  1945. if tcp_repr.control == TcpControl::Rst {
  1946. // Never reply to a TCP RST packet with another TCP RST packet.
  1947. None
  1948. } else {
  1949. // The packet wasn't handled by a socket, send a TCP RST packet.
  1950. Some(IpPacket::Tcp(tcp::Socket::rst_reply(&ip_repr, &tcp_repr)))
  1951. }
  1952. }
  1953. #[cfg(feature = "medium-ethernet")]
  1954. fn dispatch<Tx>(&mut self, tx_token: Tx, packet: EthernetPacket) -> Result<()>
  1955. where
  1956. Tx: TxToken,
  1957. {
  1958. match packet {
  1959. #[cfg(feature = "proto-ipv4")]
  1960. EthernetPacket::Arp(arp_repr) => {
  1961. let dst_hardware_addr = match arp_repr {
  1962. ArpRepr::EthernetIpv4 {
  1963. target_hardware_addr,
  1964. ..
  1965. } => target_hardware_addr,
  1966. };
  1967. self.dispatch_ethernet(tx_token, arp_repr.buffer_len(), |mut frame| {
  1968. frame.set_dst_addr(dst_hardware_addr);
  1969. frame.set_ethertype(EthernetProtocol::Arp);
  1970. let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
  1971. arp_repr.emit(&mut packet);
  1972. })
  1973. }
  1974. EthernetPacket::Ip(packet) => self.dispatch_ip(tx_token, packet),
  1975. }
  1976. }
  1977. #[cfg(feature = "medium-ethernet")]
  1978. fn dispatch_ethernet<Tx, F>(&mut self, tx_token: Tx, buffer_len: usize, f: F) -> Result<()>
  1979. where
  1980. Tx: TxToken,
  1981. F: FnOnce(EthernetFrame<&mut [u8]>),
  1982. {
  1983. let tx_len = EthernetFrame::<&[u8]>::buffer_len(buffer_len);
  1984. tx_token.consume(self.now, tx_len, |tx_buffer| {
  1985. debug_assert!(tx_buffer.as_ref().len() == tx_len);
  1986. let mut frame = EthernetFrame::new_unchecked(tx_buffer);
  1987. let src_addr = if let Some(HardwareAddress::Ethernet(addr)) = self.hardware_addr {
  1988. addr
  1989. } else {
  1990. return Err(Error::Malformed);
  1991. };
  1992. frame.set_src_addr(src_addr);
  1993. f(frame);
  1994. Ok(())
  1995. })
  1996. }
  1997. fn in_same_network(&self, addr: &IpAddress) -> bool {
  1998. self.ip_addrs.iter().any(|cidr| cidr.contains_addr(addr))
  1999. }
  2000. fn route(&self, addr: &IpAddress, timestamp: Instant) -> Result<IpAddress> {
  2001. // Send directly.
  2002. if self.in_same_network(addr) || addr.is_broadcast() {
  2003. return Ok(*addr);
  2004. }
  2005. // Route via a router.
  2006. match self.routes.lookup(addr, timestamp) {
  2007. Some(router_addr) => Ok(router_addr),
  2008. None => Err(Error::Unaddressable),
  2009. }
  2010. }
  2011. fn has_neighbor(&self, addr: &IpAddress) -> bool {
  2012. match self.route(addr, self.now) {
  2013. Ok(_routed_addr) => match self.caps.medium {
  2014. #[cfg(feature = "medium-ethernet")]
  2015. Medium::Ethernet => self
  2016. .neighbor_cache
  2017. .as_ref()
  2018. .unwrap()
  2019. .lookup(&_routed_addr, self.now)
  2020. .found(),
  2021. #[cfg(feature = "medium-ieee802154")]
  2022. Medium::Ieee802154 => self
  2023. .neighbor_cache
  2024. .as_ref()
  2025. .unwrap()
  2026. .lookup(&_routed_addr, self.now)
  2027. .found(),
  2028. #[cfg(feature = "medium-ip")]
  2029. Medium::Ip => true,
  2030. },
  2031. Err(_) => false,
  2032. }
  2033. }
  2034. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  2035. fn lookup_hardware_addr<Tx>(
  2036. &mut self,
  2037. tx_token: Tx,
  2038. src_addr: &IpAddress,
  2039. dst_addr: &IpAddress,
  2040. ) -> Result<(HardwareAddress, Tx)>
  2041. where
  2042. Tx: TxToken,
  2043. {
  2044. if dst_addr.is_broadcast() {
  2045. let hardware_addr = match self.caps.medium {
  2046. #[cfg(feature = "medium-ethernet")]
  2047. Medium::Ethernet => HardwareAddress::Ethernet(EthernetAddress::BROADCAST),
  2048. #[cfg(feature = "medium-ieee802154")]
  2049. Medium::Ieee802154 => HardwareAddress::Ieee802154(Ieee802154Address::BROADCAST),
  2050. #[cfg(feature = "medium-ip")]
  2051. Medium::Ip => unreachable!(),
  2052. };
  2053. return Ok((hardware_addr, tx_token));
  2054. }
  2055. if dst_addr.is_multicast() {
  2056. let b = dst_addr.as_bytes();
  2057. let hardware_addr = match *dst_addr {
  2058. #[cfg(feature = "proto-ipv4")]
  2059. IpAddress::Ipv4(_addr) => {
  2060. HardwareAddress::Ethernet(EthernetAddress::from_bytes(&[
  2061. 0x01,
  2062. 0x00,
  2063. 0x5e,
  2064. b[1] & 0x7F,
  2065. b[2],
  2066. b[3],
  2067. ]))
  2068. }
  2069. #[cfg(feature = "proto-ipv6")]
  2070. IpAddress::Ipv6(_addr) => match self.caps.medium {
  2071. #[cfg(feature = "medium-ethernet")]
  2072. Medium::Ethernet => HardwareAddress::Ethernet(EthernetAddress::from_bytes(&[
  2073. 0x33, 0x33, b[12], b[13], b[14], b[15],
  2074. ])),
  2075. #[cfg(feature = "medium-ieee802154")]
  2076. Medium::Ieee802154 => {
  2077. // Not sure if this is correct
  2078. HardwareAddress::Ieee802154(Ieee802154Address::BROADCAST)
  2079. }
  2080. #[cfg(feature = "medium-ip")]
  2081. Medium::Ip => unreachable!(),
  2082. },
  2083. };
  2084. return Ok((hardware_addr, tx_token));
  2085. }
  2086. let dst_addr = self.route(dst_addr, self.now)?;
  2087. match self
  2088. .neighbor_cache
  2089. .as_mut()
  2090. .unwrap()
  2091. .lookup(&dst_addr, self.now)
  2092. {
  2093. NeighborAnswer::Found(hardware_addr) => return Ok((hardware_addr, tx_token)),
  2094. NeighborAnswer::RateLimited => return Err(Error::Unaddressable),
  2095. _ => (), // XXX
  2096. }
  2097. match (src_addr, dst_addr) {
  2098. #[cfg(feature = "proto-ipv4")]
  2099. (&IpAddress::Ipv4(src_addr), IpAddress::Ipv4(dst_addr)) => {
  2100. net_debug!(
  2101. "address {} not in neighbor cache, sending ARP request",
  2102. dst_addr
  2103. );
  2104. let src_hardware_addr =
  2105. if let Some(HardwareAddress::Ethernet(addr)) = self.hardware_addr {
  2106. addr
  2107. } else {
  2108. return Err(Error::Malformed);
  2109. };
  2110. let arp_repr = ArpRepr::EthernetIpv4 {
  2111. operation: ArpOperation::Request,
  2112. source_hardware_addr: src_hardware_addr,
  2113. source_protocol_addr: src_addr,
  2114. target_hardware_addr: EthernetAddress::BROADCAST,
  2115. target_protocol_addr: dst_addr,
  2116. };
  2117. self.dispatch_ethernet(tx_token, arp_repr.buffer_len(), |mut frame| {
  2118. frame.set_dst_addr(EthernetAddress::BROADCAST);
  2119. frame.set_ethertype(EthernetProtocol::Arp);
  2120. arp_repr.emit(&mut ArpPacket::new_unchecked(frame.payload_mut()))
  2121. })?;
  2122. }
  2123. #[cfg(feature = "proto-ipv6")]
  2124. (&IpAddress::Ipv6(src_addr), IpAddress::Ipv6(dst_addr)) => {
  2125. net_debug!(
  2126. "address {} not in neighbor cache, sending Neighbor Solicitation",
  2127. dst_addr
  2128. );
  2129. let solicit = Icmpv6Repr::Ndisc(NdiscRepr::NeighborSolicit {
  2130. target_addr: dst_addr,
  2131. lladdr: Some(self.hardware_addr.unwrap().into()),
  2132. });
  2133. let packet = IpPacket::Icmpv6((
  2134. Ipv6Repr {
  2135. src_addr,
  2136. dst_addr: dst_addr.solicited_node(),
  2137. next_header: IpProtocol::Icmpv6,
  2138. payload_len: solicit.buffer_len(),
  2139. hop_limit: 0xff,
  2140. },
  2141. solicit,
  2142. ));
  2143. self.dispatch_ip(tx_token, packet)?;
  2144. }
  2145. #[allow(unreachable_patterns)]
  2146. _ => (),
  2147. }
  2148. // The request got dispatched, limit the rate on the cache.
  2149. self.neighbor_cache.as_mut().unwrap().limit_rate(self.now);
  2150. Err(Error::Unaddressable)
  2151. }
  2152. fn flush_cache(&mut self) {
  2153. #[cfg(any(feature = "medium-ethernet", feature = "medium-ieee802154"))]
  2154. if let Some(cache) = self.neighbor_cache.as_mut() {
  2155. cache.flush()
  2156. }
  2157. }
  2158. fn dispatch_ip<Tx: TxToken>(&mut self, tx_token: Tx, packet: IpPacket) -> Result<()> {
  2159. let ip_repr = packet.ip_repr();
  2160. assert!(!ip_repr.dst_addr().is_unspecified());
  2161. match self.caps.medium {
  2162. #[cfg(feature = "medium-ethernet")]
  2163. Medium::Ethernet => {
  2164. let (dst_hardware_addr, tx_token) = match self.lookup_hardware_addr(
  2165. tx_token,
  2166. &ip_repr.src_addr(),
  2167. &ip_repr.dst_addr(),
  2168. )? {
  2169. (HardwareAddress::Ethernet(addr), tx_token) => (addr, tx_token),
  2170. #[cfg(feature = "medium-ieee802154")]
  2171. (HardwareAddress::Ieee802154(_), _) => unreachable!(),
  2172. };
  2173. let caps = self.caps.clone();
  2174. self.dispatch_ethernet(tx_token, ip_repr.total_len(), |mut frame| {
  2175. frame.set_dst_addr(dst_hardware_addr);
  2176. match ip_repr {
  2177. #[cfg(feature = "proto-ipv4")]
  2178. IpRepr::Ipv4(_) => frame.set_ethertype(EthernetProtocol::Ipv4),
  2179. #[cfg(feature = "proto-ipv6")]
  2180. IpRepr::Ipv6(_) => frame.set_ethertype(EthernetProtocol::Ipv6),
  2181. }
  2182. ip_repr.emit(frame.payload_mut(), &caps.checksum);
  2183. let payload = &mut frame.payload_mut()[ip_repr.buffer_len()..];
  2184. packet.emit_payload(ip_repr, payload, &caps);
  2185. })
  2186. }
  2187. #[cfg(feature = "medium-ip")]
  2188. Medium::Ip => {
  2189. let tx_len = ip_repr.total_len();
  2190. tx_token.consume(self.now, tx_len, |mut tx_buffer| {
  2191. debug_assert!(tx_buffer.as_ref().len() == tx_len);
  2192. ip_repr.emit(&mut tx_buffer, &self.caps.checksum);
  2193. let payload = &mut tx_buffer[ip_repr.buffer_len()..];
  2194. packet.emit_payload(ip_repr, payload, &self.caps);
  2195. Ok(())
  2196. })
  2197. }
  2198. #[cfg(feature = "medium-ieee802154")]
  2199. Medium::Ieee802154 => self.dispatch_ieee802154(tx_token, packet),
  2200. }
  2201. }
  2202. #[cfg(feature = "medium-ieee802154")]
  2203. fn dispatch_ieee802154<Tx: TxToken>(&mut self, tx_token: Tx, packet: IpPacket) -> Result<()> {
  2204. let ip_repr = packet.ip_repr();
  2205. assert!(!ip_repr.dst_addr().is_unspecified());
  2206. match self.caps.medium {
  2207. #[cfg(feature = "medium-ieee802154")]
  2208. Medium::Ieee802154 => {
  2209. let (dst_hardware_addr, tx_token) = match self.lookup_hardware_addr(
  2210. tx_token,
  2211. &ip_repr.src_addr(),
  2212. &ip_repr.dst_addr(),
  2213. )? {
  2214. (HardwareAddress::Ieee802154(addr), tx_token) => (addr, tx_token),
  2215. _ => unreachable!(),
  2216. };
  2217. let ack_request = dst_hardware_addr.is_unicast();
  2218. let ack_request = match packet {
  2219. IpPacket::Icmpv6(_) => false,
  2220. _ => ack_request,
  2221. };
  2222. let mut tx_len = 0;
  2223. let ll_src_addr =
  2224. if let Some(HardwareAddress::Ieee802154(addr)) = self.hardware_addr {
  2225. Some(addr)
  2226. } else {
  2227. return Err(Error::Malformed);
  2228. };
  2229. let ieee_repr = Ieee802154Repr {
  2230. frame_type: Ieee802154FrameType::Data,
  2231. security_enabled: false,
  2232. frame_pending: false,
  2233. ack_request,
  2234. sequence_number: Some(self.get_sequence_number()),
  2235. pan_id_compression: true,
  2236. frame_version: Ieee802154FrameVersion::Ieee802154_2003,
  2237. dst_pan_id: self.pan_id,
  2238. dst_addr: Some(dst_hardware_addr),
  2239. src_pan_id: self.pan_id,
  2240. src_addr: ll_src_addr,
  2241. };
  2242. let (src_addr, dst_addr) = match (ip_repr.src_addr(), ip_repr.dst_addr()) {
  2243. (IpAddress::Ipv6(src_addr), IpAddress::Ipv6(dst_addr)) => (src_addr, dst_addr),
  2244. #[allow(unreachable_patterns)]
  2245. _ => return Err(Error::Unaddressable),
  2246. };
  2247. #[allow(unreachable_patterns)]
  2248. let (next_header, hop_limit) = match &packet {
  2249. #[cfg(feature = "socket-udp")]
  2250. IpPacket::Udp(_) => (SixlowpanNextHeader::Compressed, 64),
  2251. IpPacket::Icmpv6((_, repr)) => (
  2252. SixlowpanNextHeader::Uncompressed(IpProtocol::Icmpv6),
  2253. match repr {
  2254. Icmpv6Repr::Ndisc(_) => 255,
  2255. _ => 64,
  2256. },
  2257. ),
  2258. _ => return Err(Error::Unrecognized),
  2259. };
  2260. let iphc_repr = SixlowpanIphcRepr {
  2261. src_addr,
  2262. ll_src_addr,
  2263. dst_addr,
  2264. ll_dst_addr: Some(dst_hardware_addr),
  2265. next_header,
  2266. hop_limit,
  2267. };
  2268. tx_len += ieee_repr.buffer_len();
  2269. tx_len += iphc_repr.buffer_len();
  2270. #[allow(unreachable_patterns)]
  2271. match &packet {
  2272. #[cfg(feature = "socket-udp")]
  2273. IpPacket::Udp((_, udp_repr, payload)) => {
  2274. let udp_repr = SixlowpanUdpRepr(*udp_repr);
  2275. tx_len += udp_repr.header_len() + payload.len();
  2276. }
  2277. IpPacket::Icmpv6((_, icmp)) => {
  2278. tx_len += icmp.buffer_len();
  2279. }
  2280. _ => return Err(Error::Unrecognized),
  2281. }
  2282. tx_token.consume(self.now, tx_len, |mut tx_buffer| {
  2283. // 1. Create the header of 802.15.4
  2284. let mut ieee_packet = Ieee802154Frame::new_unchecked(&mut tx_buffer);
  2285. ieee_repr.emit(&mut ieee_packet);
  2286. let mut start = ieee_repr.buffer_len();
  2287. // 2. Create the header for 6LoWPAN IPHC
  2288. let mut iphc_packet =
  2289. SixlowpanIphcPacket::new_unchecked(&mut tx_buffer[start..tx_len]);
  2290. iphc_repr.emit(&mut iphc_packet);
  2291. start += iphc_repr.buffer_len();
  2292. #[allow(unreachable_patterns)]
  2293. match packet {
  2294. #[cfg(feature = "socket-udp")]
  2295. IpPacket::Udp((_, udp_repr, payload)) => {
  2296. // 3. Create the header for 6LoWPAN UDP
  2297. let mut udp_packet =
  2298. SixlowpanUdpPacket::new_unchecked(&mut tx_buffer[start..tx_len]);
  2299. SixlowpanUdpRepr(udp_repr).emit(
  2300. &mut udp_packet,
  2301. &iphc_repr.src_addr,
  2302. &iphc_repr.dst_addr,
  2303. payload.len(),
  2304. |buf| buf.copy_from_slice(payload),
  2305. );
  2306. }
  2307. #[cfg(feature = "proto-ipv6")]
  2308. IpPacket::Icmpv6((_, icmp_repr)) => {
  2309. // 3. Create the header for ICMPv6
  2310. let mut icmp_packet =
  2311. Icmpv6Packet::new_unchecked(&mut tx_buffer[start..tx_len]);
  2312. icmp_repr.emit(
  2313. &iphc_repr.src_addr.into(),
  2314. &iphc_repr.dst_addr.into(),
  2315. &mut icmp_packet,
  2316. &self.caps.checksum,
  2317. );
  2318. }
  2319. _ => return Err(Error::Unrecognized),
  2320. }
  2321. Ok(())
  2322. })
  2323. }
  2324. #[allow(unreachable_patterns)]
  2325. _ => Err(Error::NotSupported),
  2326. }
  2327. }
  2328. #[cfg(feature = "proto-igmp")]
  2329. fn igmp_report_packet<'any>(
  2330. &self,
  2331. version: IgmpVersion,
  2332. group_addr: Ipv4Address,
  2333. ) -> Option<IpPacket<'any>> {
  2334. let iface_addr = self.ipv4_address()?;
  2335. let igmp_repr = IgmpRepr::MembershipReport {
  2336. group_addr,
  2337. version,
  2338. };
  2339. let pkt = IpPacket::Igmp((
  2340. Ipv4Repr {
  2341. src_addr: iface_addr,
  2342. // Send to the group being reported
  2343. dst_addr: group_addr,
  2344. next_header: IpProtocol::Igmp,
  2345. payload_len: igmp_repr.buffer_len(),
  2346. hop_limit: 1,
  2347. // TODO: add Router Alert IPv4 header option. See
  2348. // [#183](https://github.com/m-labs/smoltcp/issues/183).
  2349. },
  2350. igmp_repr,
  2351. ));
  2352. Some(pkt)
  2353. }
  2354. #[cfg(feature = "proto-igmp")]
  2355. fn igmp_leave_packet<'any>(&self, group_addr: Ipv4Address) -> Option<IpPacket<'any>> {
  2356. self.ipv4_address().map(|iface_addr| {
  2357. let igmp_repr = IgmpRepr::LeaveGroup { group_addr };
  2358. IpPacket::Igmp((
  2359. Ipv4Repr {
  2360. src_addr: iface_addr,
  2361. dst_addr: Ipv4Address::MULTICAST_ALL_ROUTERS,
  2362. next_header: IpProtocol::Igmp,
  2363. payload_len: igmp_repr.buffer_len(),
  2364. hop_limit: 1,
  2365. },
  2366. igmp_repr,
  2367. ))
  2368. })
  2369. }
  2370. }
  2371. #[cfg(test)]
  2372. mod test {
  2373. use std::collections::BTreeMap;
  2374. #[cfg(feature = "proto-igmp")]
  2375. use std::vec::Vec;
  2376. use super::*;
  2377. use crate::iface::Interface;
  2378. #[cfg(feature = "medium-ethernet")]
  2379. use crate::iface::NeighborCache;
  2380. use crate::phy::{ChecksumCapabilities, Loopback};
  2381. #[cfg(feature = "proto-igmp")]
  2382. use crate::time::Instant;
  2383. use crate::{Error, Result};
  2384. #[allow(unused)]
  2385. fn fill_slice(s: &mut [u8], val: u8) {
  2386. for x in s.iter_mut() {
  2387. *x = val
  2388. }
  2389. }
  2390. fn create_loopback<'a>() -> (Interface<'a, Loopback>, SocketSet<'a>) {
  2391. #[cfg(feature = "medium-ethernet")]
  2392. return create_loopback_ethernet();
  2393. #[cfg(not(feature = "medium-ethernet"))]
  2394. return create_loopback_ip();
  2395. }
  2396. #[cfg(all(feature = "medium-ip"))]
  2397. #[allow(unused)]
  2398. fn create_loopback_ip<'a>() -> (Interface<'a, Loopback>, SocketSet<'a>) {
  2399. // Create a basic device
  2400. let device = Loopback::new(Medium::Ip);
  2401. let ip_addrs = [
  2402. #[cfg(feature = "proto-ipv4")]
  2403. IpCidr::new(IpAddress::v4(127, 0, 0, 1), 8),
  2404. #[cfg(feature = "proto-ipv6")]
  2405. IpCidr::new(IpAddress::v6(0, 0, 0, 0, 0, 0, 0, 1), 128),
  2406. #[cfg(feature = "proto-ipv6")]
  2407. IpCidr::new(IpAddress::v6(0xfdbe, 0, 0, 0, 0, 0, 0, 1), 64),
  2408. ];
  2409. let iface_builder = InterfaceBuilder::new(device).ip_addrs(ip_addrs);
  2410. #[cfg(feature = "proto-igmp")]
  2411. let iface_builder = iface_builder.ipv4_multicast_groups(BTreeMap::new());
  2412. let iface = iface_builder.finalize();
  2413. (iface, SocketSet::new(vec![]))
  2414. }
  2415. #[cfg(all(feature = "medium-ethernet"))]
  2416. fn create_loopback_ethernet<'a>() -> (Interface<'a, Loopback>, SocketSet<'a>) {
  2417. // Create a basic device
  2418. let device = Loopback::new(Medium::Ethernet);
  2419. let ip_addrs = [
  2420. #[cfg(feature = "proto-ipv4")]
  2421. IpCidr::new(IpAddress::v4(127, 0, 0, 1), 8),
  2422. #[cfg(feature = "proto-ipv6")]
  2423. IpCidr::new(IpAddress::v6(0, 0, 0, 0, 0, 0, 0, 1), 128),
  2424. #[cfg(feature = "proto-ipv6")]
  2425. IpCidr::new(IpAddress::v6(0xfdbe, 0, 0, 0, 0, 0, 0, 1), 64),
  2426. ];
  2427. let iface_builder = InterfaceBuilder::new(device)
  2428. .hardware_addr(EthernetAddress::default().into())
  2429. .neighbor_cache(NeighborCache::new(BTreeMap::new()))
  2430. .ip_addrs(ip_addrs);
  2431. #[cfg(feature = "proto-igmp")]
  2432. let iface_builder = iface_builder.ipv4_multicast_groups(BTreeMap::new());
  2433. let iface = iface_builder.finalize();
  2434. (iface, SocketSet::new(vec![]))
  2435. }
  2436. #[cfg(feature = "proto-igmp")]
  2437. fn recv_all(iface: &mut Interface<'_, Loopback>, timestamp: Instant) -> Vec<Vec<u8>> {
  2438. let mut pkts = Vec::new();
  2439. while let Some((rx, _tx)) = iface.device.receive() {
  2440. rx.consume(timestamp, |pkt| {
  2441. pkts.push(pkt.to_vec());
  2442. Ok(())
  2443. })
  2444. .unwrap();
  2445. }
  2446. pkts
  2447. }
  2448. #[derive(Debug, PartialEq)]
  2449. #[cfg_attr(feature = "defmt", derive(defmt::Format))]
  2450. struct MockTxToken;
  2451. impl TxToken for MockTxToken {
  2452. fn consume<R, F>(self, _: Instant, _: usize, _: F) -> Result<R>
  2453. where
  2454. F: FnOnce(&mut [u8]) -> Result<R>,
  2455. {
  2456. Err(Error::Unaddressable)
  2457. }
  2458. }
  2459. #[test]
  2460. #[should_panic(expected = "hardware_addr required option was not set")]
  2461. #[cfg(all(feature = "medium-ethernet"))]
  2462. fn test_builder_initialization_panic() {
  2463. InterfaceBuilder::new(Loopback::new(Medium::Ethernet)).finalize();
  2464. }
  2465. #[test]
  2466. #[cfg(feature = "proto-ipv4")]
  2467. fn test_no_icmp_no_unicast_ipv4() {
  2468. let (mut iface, mut sockets) = create_loopback();
  2469. // Unknown Ipv4 Protocol
  2470. //
  2471. // Because the destination is the broadcast address
  2472. // this should not trigger and Destination Unreachable
  2473. // response. See RFC 1122 § 3.2.2.
  2474. let repr = IpRepr::Ipv4(Ipv4Repr {
  2475. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2476. dst_addr: Ipv4Address::BROADCAST,
  2477. next_header: IpProtocol::Unknown(0x0c),
  2478. payload_len: 0,
  2479. hop_limit: 0x40,
  2480. });
  2481. let mut bytes = vec![0u8; 54];
  2482. repr.emit(&mut bytes, &ChecksumCapabilities::default());
  2483. let frame = Ipv4Packet::new_unchecked(&bytes);
  2484. // Ensure that the unknown protocol frame does not trigger an
  2485. // ICMP error response when the destination address is a
  2486. // broadcast address
  2487. assert_eq!(iface.inner.process_ipv4(&mut sockets, &frame), None);
  2488. }
  2489. #[test]
  2490. #[cfg(feature = "proto-ipv6")]
  2491. fn test_no_icmp_no_unicast_ipv6() {
  2492. let (mut iface, mut sockets) = create_loopback();
  2493. // Unknown Ipv6 Protocol
  2494. //
  2495. // Because the destination is the broadcast address
  2496. // this should not trigger and Destination Unreachable
  2497. // response. See RFC 1122 § 3.2.2.
  2498. let repr = IpRepr::Ipv6(Ipv6Repr {
  2499. src_addr: Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1),
  2500. dst_addr: Ipv6Address::LINK_LOCAL_ALL_NODES,
  2501. next_header: IpProtocol::Unknown(0x0c),
  2502. payload_len: 0,
  2503. hop_limit: 0x40,
  2504. });
  2505. let mut bytes = vec![0u8; 54];
  2506. repr.emit(&mut bytes, &ChecksumCapabilities::default());
  2507. let frame = Ipv6Packet::new_unchecked(&bytes);
  2508. // Ensure that the unknown protocol frame does not trigger an
  2509. // ICMP error response when the destination address is a
  2510. // broadcast address
  2511. assert_eq!(iface.inner.process_ipv6(&mut sockets, &frame), None);
  2512. }
  2513. #[test]
  2514. #[cfg(feature = "proto-ipv4")]
  2515. fn test_icmp_error_no_payload() {
  2516. static NO_BYTES: [u8; 0] = [];
  2517. let (mut iface, mut sockets) = create_loopback();
  2518. // Unknown Ipv4 Protocol with no payload
  2519. let repr = IpRepr::Ipv4(Ipv4Repr {
  2520. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2521. dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2522. next_header: IpProtocol::Unknown(0x0c),
  2523. payload_len: 0,
  2524. hop_limit: 0x40,
  2525. });
  2526. let mut bytes = vec![0u8; 34];
  2527. repr.emit(&mut bytes, &ChecksumCapabilities::default());
  2528. let frame = Ipv4Packet::new_unchecked(&bytes);
  2529. // The expected Destination Unreachable response due to the
  2530. // unknown protocol
  2531. let icmp_repr = Icmpv4Repr::DstUnreachable {
  2532. reason: Icmpv4DstUnreachable::ProtoUnreachable,
  2533. header: Ipv4Repr {
  2534. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2535. dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2536. next_header: IpProtocol::Unknown(12),
  2537. payload_len: 0,
  2538. hop_limit: 64,
  2539. },
  2540. data: &NO_BYTES,
  2541. };
  2542. let expected_repr = IpPacket::Icmpv4((
  2543. Ipv4Repr {
  2544. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2545. dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2546. next_header: IpProtocol::Icmp,
  2547. payload_len: icmp_repr.buffer_len(),
  2548. hop_limit: 64,
  2549. },
  2550. icmp_repr,
  2551. ));
  2552. // Ensure that the unknown protocol triggers an error response.
  2553. // And we correctly handle no payload.
  2554. assert_eq!(
  2555. iface.inner.process_ipv4(&mut sockets, &frame),
  2556. Some(expected_repr)
  2557. );
  2558. }
  2559. #[test]
  2560. #[cfg(feature = "proto-ipv4")]
  2561. fn test_local_subnet_broadcasts() {
  2562. let (mut iface, _) = create_loopback();
  2563. iface.update_ip_addrs(|addrs| {
  2564. addrs.iter_mut().next().map(|addr| {
  2565. *addr = IpCidr::Ipv4(Ipv4Cidr::new(Ipv4Address([192, 168, 1, 23]), 24));
  2566. });
  2567. });
  2568. assert!(iface
  2569. .inner
  2570. .is_subnet_broadcast(Ipv4Address([192, 168, 1, 255])),);
  2571. assert!(!iface
  2572. .inner
  2573. .is_subnet_broadcast(Ipv4Address([192, 168, 1, 254])),);
  2574. iface.update_ip_addrs(|addrs| {
  2575. addrs.iter_mut().next().map(|addr| {
  2576. *addr = IpCidr::Ipv4(Ipv4Cidr::new(Ipv4Address([192, 168, 23, 24]), 16));
  2577. });
  2578. });
  2579. assert!(!iface
  2580. .inner
  2581. .is_subnet_broadcast(Ipv4Address([192, 168, 23, 255])),);
  2582. assert!(!iface
  2583. .inner
  2584. .is_subnet_broadcast(Ipv4Address([192, 168, 23, 254])),);
  2585. assert!(!iface
  2586. .inner
  2587. .is_subnet_broadcast(Ipv4Address([192, 168, 255, 254])),);
  2588. assert!(iface
  2589. .inner
  2590. .is_subnet_broadcast(Ipv4Address([192, 168, 255, 255])),);
  2591. iface.update_ip_addrs(|addrs| {
  2592. addrs.iter_mut().next().map(|addr| {
  2593. *addr = IpCidr::Ipv4(Ipv4Cidr::new(Ipv4Address([192, 168, 23, 24]), 8));
  2594. });
  2595. });
  2596. assert!(!iface
  2597. .inner
  2598. .is_subnet_broadcast(Ipv4Address([192, 23, 1, 255])),);
  2599. assert!(!iface
  2600. .inner
  2601. .is_subnet_broadcast(Ipv4Address([192, 23, 1, 254])),);
  2602. assert!(!iface
  2603. .inner
  2604. .is_subnet_broadcast(Ipv4Address([192, 255, 255, 254])),);
  2605. assert!(iface
  2606. .inner
  2607. .is_subnet_broadcast(Ipv4Address([192, 255, 255, 255])),);
  2608. }
  2609. #[test]
  2610. #[cfg(all(feature = "socket-udp", feature = "proto-ipv4"))]
  2611. fn test_icmp_error_port_unreachable() {
  2612. static UDP_PAYLOAD: [u8; 12] = [
  2613. 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2c, 0x20, 0x57, 0x6f, 0x6c, 0x64, 0x21,
  2614. ];
  2615. let (mut iface, mut sockets) = create_loopback();
  2616. let mut udp_bytes_unicast = vec![0u8; 20];
  2617. let mut udp_bytes_broadcast = vec![0u8; 20];
  2618. let mut packet_unicast = UdpPacket::new_unchecked(&mut udp_bytes_unicast);
  2619. let mut packet_broadcast = UdpPacket::new_unchecked(&mut udp_bytes_broadcast);
  2620. let udp_repr = UdpRepr {
  2621. src_port: 67,
  2622. dst_port: 68,
  2623. };
  2624. let ip_repr = IpRepr::Ipv4(Ipv4Repr {
  2625. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2626. dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2627. next_header: IpProtocol::Udp,
  2628. payload_len: udp_repr.header_len() + UDP_PAYLOAD.len(),
  2629. hop_limit: 64,
  2630. });
  2631. // Emit the representations to a packet
  2632. udp_repr.emit(
  2633. &mut packet_unicast,
  2634. &ip_repr.src_addr(),
  2635. &ip_repr.dst_addr(),
  2636. UDP_PAYLOAD.len(),
  2637. |buf| buf.copy_from_slice(&UDP_PAYLOAD),
  2638. &ChecksumCapabilities::default(),
  2639. );
  2640. let data = packet_unicast.into_inner();
  2641. // The expected Destination Unreachable ICMPv4 error response due
  2642. // to no sockets listening on the destination port.
  2643. let icmp_repr = Icmpv4Repr::DstUnreachable {
  2644. reason: Icmpv4DstUnreachable::PortUnreachable,
  2645. header: Ipv4Repr {
  2646. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2647. dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2648. next_header: IpProtocol::Udp,
  2649. payload_len: udp_repr.header_len() + UDP_PAYLOAD.len(),
  2650. hop_limit: 64,
  2651. },
  2652. data: data,
  2653. };
  2654. let expected_repr = IpPacket::Icmpv4((
  2655. Ipv4Repr {
  2656. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  2657. dst_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2658. next_header: IpProtocol::Icmp,
  2659. payload_len: icmp_repr.buffer_len(),
  2660. hop_limit: 64,
  2661. },
  2662. icmp_repr,
  2663. ));
  2664. // Ensure that the unknown protocol triggers an error response.
  2665. // And we correctly handle no payload.
  2666. assert_eq!(
  2667. iface.inner.process_udp(&mut sockets, ip_repr, false, data),
  2668. Some(expected_repr)
  2669. );
  2670. let ip_repr = IpRepr::Ipv4(Ipv4Repr {
  2671. src_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x02]),
  2672. dst_addr: Ipv4Address::BROADCAST,
  2673. next_header: IpProtocol::Udp,
  2674. payload_len: udp_repr.header_len() + UDP_PAYLOAD.len(),
  2675. hop_limit: 64,
  2676. });
  2677. // Emit the representations to a packet
  2678. udp_repr.emit(
  2679. &mut packet_broadcast,
  2680. &ip_repr.src_addr(),
  2681. &IpAddress::Ipv4(Ipv4Address::BROADCAST),
  2682. UDP_PAYLOAD.len(),
  2683. |buf| buf.copy_from_slice(&UDP_PAYLOAD),
  2684. &ChecksumCapabilities::default(),
  2685. );
  2686. // Ensure that the port unreachable error does not trigger an
  2687. // ICMP error response when the destination address is a
  2688. // broadcast address and no socket is bound to the port.
  2689. assert_eq!(
  2690. iface
  2691. .inner
  2692. .process_udp(&mut sockets, ip_repr, false, packet_broadcast.into_inner()),
  2693. None
  2694. );
  2695. }
  2696. #[test]
  2697. #[cfg(feature = "socket-udp")]
  2698. fn test_handle_udp_broadcast() {
  2699. use crate::wire::IpEndpoint;
  2700. static UDP_PAYLOAD: [u8; 5] = [0x48, 0x65, 0x6c, 0x6c, 0x6f];
  2701. let (mut iface, mut sockets) = create_loopback();
  2702. let rx_buffer = udp::PacketBuffer::new(vec![udp::PacketMetadata::EMPTY], vec![0; 15]);
  2703. let tx_buffer = udp::PacketBuffer::new(vec![udp::PacketMetadata::EMPTY], vec![0; 15]);
  2704. let udp_socket = udp::Socket::new(rx_buffer, tx_buffer);
  2705. let mut udp_bytes = vec![0u8; 13];
  2706. let mut packet = UdpPacket::new_unchecked(&mut udp_bytes);
  2707. let socket_handle = sockets.add(udp_socket);
  2708. #[cfg(feature = "proto-ipv6")]
  2709. let src_ip = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1);
  2710. #[cfg(all(not(feature = "proto-ipv6"), feature = "proto-ipv4"))]
  2711. let src_ip = Ipv4Address::new(0x7f, 0x00, 0x00, 0x02);
  2712. let udp_repr = UdpRepr {
  2713. src_port: 67,
  2714. dst_port: 68,
  2715. };
  2716. #[cfg(feature = "proto-ipv6")]
  2717. let ip_repr = IpRepr::Ipv6(Ipv6Repr {
  2718. src_addr: src_ip,
  2719. dst_addr: Ipv6Address::LINK_LOCAL_ALL_NODES,
  2720. next_header: IpProtocol::Udp,
  2721. payload_len: udp_repr.header_len() + UDP_PAYLOAD.len(),
  2722. hop_limit: 0x40,
  2723. });
  2724. #[cfg(all(not(feature = "proto-ipv6"), feature = "proto-ipv4"))]
  2725. let ip_repr = IpRepr::Ipv4(Ipv4Repr {
  2726. src_addr: src_ip,
  2727. dst_addr: Ipv4Address::BROADCAST,
  2728. next_header: IpProtocol::Udp,
  2729. payload_len: udp_repr.header_len() + UDP_PAYLOAD.len(),
  2730. hop_limit: 0x40,
  2731. });
  2732. // Bind the socket to port 68
  2733. let socket = sockets.get::<udp::Socket>(socket_handle);
  2734. assert_eq!(socket.bind(68), Ok(()));
  2735. assert!(!socket.can_recv());
  2736. assert!(socket.can_send());
  2737. udp_repr.emit(
  2738. &mut packet,
  2739. &ip_repr.src_addr(),
  2740. &ip_repr.dst_addr(),
  2741. UDP_PAYLOAD.len(),
  2742. |buf| buf.copy_from_slice(&UDP_PAYLOAD),
  2743. &ChecksumCapabilities::default(),
  2744. );
  2745. // Packet should be handled by bound UDP socket
  2746. assert_eq!(
  2747. iface
  2748. .inner
  2749. .process_udp(&mut sockets, ip_repr, false, packet.into_inner()),
  2750. None
  2751. );
  2752. // Make sure the payload to the UDP packet processed by process_udp is
  2753. // appended to the bound sockets rx_buffer
  2754. let socket = sockets.get::<udp::Socket>(socket_handle);
  2755. assert!(socket.can_recv());
  2756. assert_eq!(
  2757. socket.recv(),
  2758. Ok((&UDP_PAYLOAD[..], IpEndpoint::new(src_ip.into(), 67)))
  2759. );
  2760. }
  2761. #[test]
  2762. #[cfg(feature = "proto-ipv4")]
  2763. fn test_handle_ipv4_broadcast() {
  2764. use crate::wire::{Icmpv4Packet, Icmpv4Repr, Ipv4Packet};
  2765. let (mut iface, mut sockets) = create_loopback();
  2766. let our_ipv4_addr = iface.ipv4_address().unwrap();
  2767. let src_ipv4_addr = Ipv4Address([127, 0, 0, 2]);
  2768. // ICMPv4 echo request
  2769. let icmpv4_data: [u8; 4] = [0xaa, 0x00, 0x00, 0xff];
  2770. let icmpv4_repr = Icmpv4Repr::EchoRequest {
  2771. ident: 0x1234,
  2772. seq_no: 0xabcd,
  2773. data: &icmpv4_data,
  2774. };
  2775. // Send to IPv4 broadcast address
  2776. let ipv4_repr = Ipv4Repr {
  2777. src_addr: src_ipv4_addr,
  2778. dst_addr: Ipv4Address::BROADCAST,
  2779. next_header: IpProtocol::Icmp,
  2780. hop_limit: 64,
  2781. payload_len: icmpv4_repr.buffer_len(),
  2782. };
  2783. // Emit to ip frame
  2784. let mut bytes = vec![0u8; ipv4_repr.buffer_len() + icmpv4_repr.buffer_len()];
  2785. let frame = {
  2786. ipv4_repr.emit(
  2787. &mut Ipv4Packet::new_unchecked(&mut bytes),
  2788. &ChecksumCapabilities::default(),
  2789. );
  2790. icmpv4_repr.emit(
  2791. &mut Icmpv4Packet::new_unchecked(&mut bytes[ipv4_repr.buffer_len()..]),
  2792. &ChecksumCapabilities::default(),
  2793. );
  2794. Ipv4Packet::new_unchecked(&bytes)
  2795. };
  2796. // Expected ICMPv4 echo reply
  2797. let expected_icmpv4_repr = Icmpv4Repr::EchoReply {
  2798. ident: 0x1234,
  2799. seq_no: 0xabcd,
  2800. data: &icmpv4_data,
  2801. };
  2802. let expected_ipv4_repr = Ipv4Repr {
  2803. src_addr: our_ipv4_addr,
  2804. dst_addr: src_ipv4_addr,
  2805. next_header: IpProtocol::Icmp,
  2806. hop_limit: 64,
  2807. payload_len: expected_icmpv4_repr.buffer_len(),
  2808. };
  2809. let expected_packet = IpPacket::Icmpv4((expected_ipv4_repr, expected_icmpv4_repr));
  2810. assert_eq!(
  2811. iface.inner.process_ipv4(&mut sockets, &frame),
  2812. Some(expected_packet)
  2813. );
  2814. }
  2815. #[test]
  2816. #[cfg(feature = "socket-udp")]
  2817. fn test_icmp_reply_size() {
  2818. #[cfg(feature = "proto-ipv6")]
  2819. use crate::wire::Icmpv6DstUnreachable;
  2820. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2821. use crate::wire::IPV4_MIN_MTU as MIN_MTU;
  2822. #[cfg(feature = "proto-ipv6")]
  2823. use crate::wire::IPV6_MIN_MTU as MIN_MTU;
  2824. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2825. const MAX_PAYLOAD_LEN: usize = 528;
  2826. #[cfg(feature = "proto-ipv6")]
  2827. const MAX_PAYLOAD_LEN: usize = 1192;
  2828. let (mut iface, mut sockets) = create_loopback();
  2829. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2830. let src_addr = Ipv4Address([192, 168, 1, 1]);
  2831. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2832. let dst_addr = Ipv4Address([192, 168, 1, 2]);
  2833. #[cfg(feature = "proto-ipv6")]
  2834. let src_addr = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1);
  2835. #[cfg(feature = "proto-ipv6")]
  2836. let dst_addr = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 2);
  2837. // UDP packet that if not tructated will cause a icmp port unreachable reply
  2838. // to exeed the minimum mtu bytes in length.
  2839. let udp_repr = UdpRepr {
  2840. src_port: 67,
  2841. dst_port: 68,
  2842. };
  2843. let mut bytes = vec![0xff; udp_repr.header_len() + MAX_PAYLOAD_LEN];
  2844. let mut packet = UdpPacket::new_unchecked(&mut bytes[..]);
  2845. udp_repr.emit(
  2846. &mut packet,
  2847. &src_addr.into(),
  2848. &dst_addr.into(),
  2849. MAX_PAYLOAD_LEN,
  2850. |buf| fill_slice(buf, 0x2a),
  2851. &ChecksumCapabilities::default(),
  2852. );
  2853. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2854. let ip_repr = Ipv4Repr {
  2855. src_addr: src_addr,
  2856. dst_addr: dst_addr,
  2857. next_header: IpProtocol::Udp,
  2858. hop_limit: 64,
  2859. payload_len: udp_repr.header_len() + MAX_PAYLOAD_LEN,
  2860. };
  2861. #[cfg(feature = "proto-ipv6")]
  2862. let ip_repr = Ipv6Repr {
  2863. src_addr: src_addr,
  2864. dst_addr: dst_addr,
  2865. next_header: IpProtocol::Udp,
  2866. hop_limit: 64,
  2867. payload_len: udp_repr.header_len() + MAX_PAYLOAD_LEN,
  2868. };
  2869. let payload = packet.into_inner();
  2870. // Expected packets
  2871. #[cfg(feature = "proto-ipv6")]
  2872. let expected_icmp_repr = Icmpv6Repr::DstUnreachable {
  2873. reason: Icmpv6DstUnreachable::PortUnreachable,
  2874. header: ip_repr,
  2875. data: &payload[..MAX_PAYLOAD_LEN],
  2876. };
  2877. #[cfg(feature = "proto-ipv6")]
  2878. let expected_ip_repr = Ipv6Repr {
  2879. src_addr: dst_addr,
  2880. dst_addr: src_addr,
  2881. next_header: IpProtocol::Icmpv6,
  2882. hop_limit: 64,
  2883. payload_len: expected_icmp_repr.buffer_len(),
  2884. };
  2885. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2886. let expected_icmp_repr = Icmpv4Repr::DstUnreachable {
  2887. reason: Icmpv4DstUnreachable::PortUnreachable,
  2888. header: ip_repr,
  2889. data: &payload[..MAX_PAYLOAD_LEN],
  2890. };
  2891. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2892. let expected_ip_repr = Ipv4Repr {
  2893. src_addr: dst_addr,
  2894. dst_addr: src_addr,
  2895. next_header: IpProtocol::Icmp,
  2896. hop_limit: 64,
  2897. payload_len: expected_icmp_repr.buffer_len(),
  2898. };
  2899. // The expected packet does not exceed the IPV4_MIN_MTU
  2900. #[cfg(feature = "proto-ipv6")]
  2901. assert_eq!(
  2902. expected_ip_repr.buffer_len() + expected_icmp_repr.buffer_len(),
  2903. MIN_MTU
  2904. );
  2905. // The expected packet does not exceed the IPV4_MIN_MTU
  2906. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2907. assert_eq!(
  2908. expected_ip_repr.buffer_len() + expected_icmp_repr.buffer_len(),
  2909. MIN_MTU
  2910. );
  2911. // The expected packet and the generated packet are equal
  2912. #[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
  2913. assert_eq!(
  2914. iface
  2915. .inner
  2916. .process_udp(&mut sockets, ip_repr.into(), false, payload),
  2917. Some(IpPacket::Icmpv4((expected_ip_repr, expected_icmp_repr)))
  2918. );
  2919. #[cfg(feature = "proto-ipv6")]
  2920. assert_eq!(
  2921. iface
  2922. .inner
  2923. .process_udp(&mut sockets, ip_repr.into(), false, payload),
  2924. Some(IpPacket::Icmpv6((expected_ip_repr, expected_icmp_repr)))
  2925. );
  2926. }
  2927. #[test]
  2928. #[cfg(all(feature = "medium-ethernet", feature = "proto-ipv4"))]
  2929. fn test_handle_valid_arp_request() {
  2930. let (mut iface, mut sockets) = create_loopback_ethernet();
  2931. let mut eth_bytes = vec![0u8; 42];
  2932. let local_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x01]);
  2933. let remote_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x02]);
  2934. let local_hw_addr = EthernetAddress([0x00, 0x00, 0x00, 0x00, 0x00, 0x00]);
  2935. let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
  2936. let repr = ArpRepr::EthernetIpv4 {
  2937. operation: ArpOperation::Request,
  2938. source_hardware_addr: remote_hw_addr,
  2939. source_protocol_addr: remote_ip_addr,
  2940. target_hardware_addr: EthernetAddress::default(),
  2941. target_protocol_addr: local_ip_addr,
  2942. };
  2943. let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
  2944. frame.set_dst_addr(EthernetAddress::BROADCAST);
  2945. frame.set_src_addr(remote_hw_addr);
  2946. frame.set_ethertype(EthernetProtocol::Arp);
  2947. let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
  2948. repr.emit(&mut packet);
  2949. // Ensure an ARP Request for us triggers an ARP Reply
  2950. assert_eq!(
  2951. iface
  2952. .inner
  2953. .process_ethernet(&mut sockets, frame.into_inner()),
  2954. Some(EthernetPacket::Arp(ArpRepr::EthernetIpv4 {
  2955. operation: ArpOperation::Reply,
  2956. source_hardware_addr: local_hw_addr,
  2957. source_protocol_addr: local_ip_addr,
  2958. target_hardware_addr: remote_hw_addr,
  2959. target_protocol_addr: remote_ip_addr
  2960. }))
  2961. );
  2962. // Ensure the address of the requestor was entered in the cache
  2963. assert_eq!(
  2964. iface.inner.lookup_hardware_addr(
  2965. MockTxToken,
  2966. &IpAddress::Ipv4(local_ip_addr),
  2967. &IpAddress::Ipv4(remote_ip_addr)
  2968. ),
  2969. Ok((HardwareAddress::Ethernet(remote_hw_addr), MockTxToken))
  2970. );
  2971. }
  2972. #[test]
  2973. #[cfg(all(feature = "medium-ethernet", feature = "proto-ipv6"))]
  2974. fn test_handle_valid_ndisc_request() {
  2975. let (mut iface, mut sockets) = create_loopback_ethernet();
  2976. let mut eth_bytes = vec![0u8; 86];
  2977. let local_ip_addr = Ipv6Address::new(0xfdbe, 0, 0, 0, 0, 0, 0, 1);
  2978. let remote_ip_addr = Ipv6Address::new(0xfdbe, 0, 0, 0, 0, 0, 0, 2);
  2979. let local_hw_addr = EthernetAddress([0x00, 0x00, 0x00, 0x00, 0x00, 0x00]);
  2980. let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
  2981. let solicit = Icmpv6Repr::Ndisc(NdiscRepr::NeighborSolicit {
  2982. target_addr: local_ip_addr,
  2983. lladdr: Some(remote_hw_addr.into()),
  2984. });
  2985. let ip_repr = IpRepr::Ipv6(Ipv6Repr {
  2986. src_addr: remote_ip_addr,
  2987. dst_addr: local_ip_addr.solicited_node(),
  2988. next_header: IpProtocol::Icmpv6,
  2989. hop_limit: 0xff,
  2990. payload_len: solicit.buffer_len(),
  2991. });
  2992. let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
  2993. frame.set_dst_addr(EthernetAddress([0x33, 0x33, 0x00, 0x00, 0x00, 0x00]));
  2994. frame.set_src_addr(remote_hw_addr);
  2995. frame.set_ethertype(EthernetProtocol::Ipv6);
  2996. ip_repr.emit(frame.payload_mut(), &ChecksumCapabilities::default());
  2997. solicit.emit(
  2998. &remote_ip_addr.into(),
  2999. &local_ip_addr.solicited_node().into(),
  3000. &mut Icmpv6Packet::new_unchecked(&mut frame.payload_mut()[ip_repr.buffer_len()..]),
  3001. &ChecksumCapabilities::default(),
  3002. );
  3003. let icmpv6_expected = Icmpv6Repr::Ndisc(NdiscRepr::NeighborAdvert {
  3004. flags: NdiscNeighborFlags::SOLICITED,
  3005. target_addr: local_ip_addr,
  3006. lladdr: Some(local_hw_addr.into()),
  3007. });
  3008. let ipv6_expected = Ipv6Repr {
  3009. src_addr: local_ip_addr,
  3010. dst_addr: remote_ip_addr,
  3011. next_header: IpProtocol::Icmpv6,
  3012. hop_limit: 0xff,
  3013. payload_len: icmpv6_expected.buffer_len(),
  3014. };
  3015. // Ensure an Neighbor Solicitation triggers a Neighbor Advertisement
  3016. assert_eq!(
  3017. iface
  3018. .inner
  3019. .process_ethernet(&mut sockets, frame.into_inner()),
  3020. Some(EthernetPacket::Ip(IpPacket::Icmpv6((
  3021. ipv6_expected,
  3022. icmpv6_expected
  3023. ))))
  3024. );
  3025. // Ensure the address of the requestor was entered in the cache
  3026. assert_eq!(
  3027. iface.inner.lookup_hardware_addr(
  3028. MockTxToken,
  3029. &IpAddress::Ipv6(local_ip_addr),
  3030. &IpAddress::Ipv6(remote_ip_addr)
  3031. ),
  3032. Ok((HardwareAddress::Ethernet(remote_hw_addr), MockTxToken))
  3033. );
  3034. }
  3035. #[test]
  3036. #[cfg(all(feature = "medium-ethernet", feature = "proto-ipv4"))]
  3037. fn test_handle_other_arp_request() {
  3038. let (mut iface, mut sockets) = create_loopback_ethernet();
  3039. let mut eth_bytes = vec![0u8; 42];
  3040. let remote_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x02]);
  3041. let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
  3042. let repr = ArpRepr::EthernetIpv4 {
  3043. operation: ArpOperation::Request,
  3044. source_hardware_addr: remote_hw_addr,
  3045. source_protocol_addr: remote_ip_addr,
  3046. target_hardware_addr: EthernetAddress::default(),
  3047. target_protocol_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x03]),
  3048. };
  3049. let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
  3050. frame.set_dst_addr(EthernetAddress::BROADCAST);
  3051. frame.set_src_addr(remote_hw_addr);
  3052. frame.set_ethertype(EthernetProtocol::Arp);
  3053. let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
  3054. repr.emit(&mut packet);
  3055. // Ensure an ARP Request for someone else does not trigger an ARP Reply
  3056. assert_eq!(
  3057. iface
  3058. .inner
  3059. .process_ethernet(&mut sockets, frame.into_inner()),
  3060. None
  3061. );
  3062. // Ensure the address of the requestor was NOT entered in the cache
  3063. assert_eq!(
  3064. iface.inner.lookup_hardware_addr(
  3065. MockTxToken,
  3066. &IpAddress::Ipv4(Ipv4Address([0x7f, 0x00, 0x00, 0x01])),
  3067. &IpAddress::Ipv4(remote_ip_addr)
  3068. ),
  3069. Err(Error::Unaddressable)
  3070. );
  3071. }
  3072. #[test]
  3073. #[cfg(all(feature = "medium-ethernet", feature = "proto-ipv4"))]
  3074. fn test_arp_flush_after_update_ip() {
  3075. let (mut iface, mut sockets) = create_loopback_ethernet();
  3076. let mut eth_bytes = vec![0u8; 42];
  3077. let local_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x01]);
  3078. let remote_ip_addr = Ipv4Address([0x7f, 0x00, 0x00, 0x02]);
  3079. let local_hw_addr = EthernetAddress([0x00, 0x00, 0x00, 0x00, 0x00, 0x00]);
  3080. let remote_hw_addr = EthernetAddress([0x52, 0x54, 0x00, 0x00, 0x00, 0x00]);
  3081. let repr = ArpRepr::EthernetIpv4 {
  3082. operation: ArpOperation::Request,
  3083. source_hardware_addr: remote_hw_addr,
  3084. source_protocol_addr: remote_ip_addr,
  3085. target_hardware_addr: EthernetAddress::default(),
  3086. target_protocol_addr: Ipv4Address([0x7f, 0x00, 0x00, 0x01]),
  3087. };
  3088. let mut frame = EthernetFrame::new_unchecked(&mut eth_bytes);
  3089. frame.set_dst_addr(EthernetAddress::BROADCAST);
  3090. frame.set_src_addr(remote_hw_addr);
  3091. frame.set_ethertype(EthernetProtocol::Arp);
  3092. {
  3093. let mut packet = ArpPacket::new_unchecked(frame.payload_mut());
  3094. repr.emit(&mut packet);
  3095. }
  3096. // Ensure an ARP Request for us triggers an ARP Reply
  3097. assert_eq!(
  3098. iface
  3099. .inner
  3100. .process_ethernet(&mut sockets, frame.into_inner()),
  3101. Some(EthernetPacket::Arp(ArpRepr::EthernetIpv4 {
  3102. operation: ArpOperation::Reply,
  3103. source_hardware_addr: local_hw_addr,
  3104. source_protocol_addr: local_ip_addr,
  3105. target_hardware_addr: remote_hw_addr,
  3106. target_protocol_addr: remote_ip_addr
  3107. }))
  3108. );
  3109. // Ensure the address of the requestor was entered in the cache
  3110. assert_eq!(
  3111. iface.inner.lookup_hardware_addr(
  3112. MockTxToken,
  3113. &IpAddress::Ipv4(local_ip_addr),
  3114. &IpAddress::Ipv4(remote_ip_addr)
  3115. ),
  3116. Ok((HardwareAddress::Ethernet(remote_hw_addr), MockTxToken))
  3117. );
  3118. // Update IP addrs to trigger ARP cache flush
  3119. let local_ip_addr_new = Ipv4Address([0x7f, 0x00, 0x00, 0x01]);
  3120. iface.update_ip_addrs(|addrs| {
  3121. addrs.iter_mut().next().map(|addr| {
  3122. *addr = IpCidr::Ipv4(Ipv4Cidr::new(local_ip_addr_new, 24));
  3123. });
  3124. });
  3125. // ARP cache flush after address change
  3126. assert!(!iface.inner.has_neighbor(&IpAddress::Ipv4(remote_ip_addr)));
  3127. }
  3128. #[test]
  3129. #[cfg(all(feature = "socket-icmp", feature = "proto-ipv4"))]
  3130. fn test_icmpv4_socket() {
  3131. use crate::wire::Icmpv4Packet;
  3132. let (mut iface, mut sockets) = create_loopback();
  3133. let rx_buffer = icmp::PacketBuffer::new(vec![icmp::PacketMetadata::EMPTY], vec![0; 24]);
  3134. let tx_buffer = icmp::PacketBuffer::new(vec![icmp::PacketMetadata::EMPTY], vec![0; 24]);
  3135. let icmpv4_socket = icmp::Socket::new(rx_buffer, tx_buffer);
  3136. let socket_handle = sockets.add(icmpv4_socket);
  3137. let ident = 0x1234;
  3138. let seq_no = 0x5432;
  3139. let echo_data = &[0xff; 16];
  3140. let socket = sockets.get::<icmp::Socket>(socket_handle);
  3141. // Bind to the ID 0x1234
  3142. assert_eq!(socket.bind(icmp::Endpoint::Ident(ident)), Ok(()));
  3143. // Ensure the ident we bound to and the ident of the packet are the same.
  3144. let mut bytes = [0xff; 24];
  3145. let mut packet = Icmpv4Packet::new_unchecked(&mut bytes);
  3146. let echo_repr = Icmpv4Repr::EchoRequest {
  3147. ident,
  3148. seq_no,
  3149. data: echo_data,
  3150. };
  3151. echo_repr.emit(&mut packet, &ChecksumCapabilities::default());
  3152. let icmp_data = &packet.into_inner()[..];
  3153. let ipv4_repr = Ipv4Repr {
  3154. src_addr: Ipv4Address::new(0x7f, 0x00, 0x00, 0x02),
  3155. dst_addr: Ipv4Address::new(0x7f, 0x00, 0x00, 0x01),
  3156. next_header: IpProtocol::Icmp,
  3157. payload_len: 24,
  3158. hop_limit: 64,
  3159. };
  3160. let ip_repr = IpRepr::Ipv4(ipv4_repr);
  3161. // Open a socket and ensure the packet is handled due to the listening
  3162. // socket.
  3163. assert!(!sockets.get::<icmp::Socket>(socket_handle).can_recv());
  3164. // Confirm we still get EchoReply from `smoltcp` even with the ICMP socket listening
  3165. let echo_reply = Icmpv4Repr::EchoReply {
  3166. ident,
  3167. seq_no,
  3168. data: echo_data,
  3169. };
  3170. let ipv4_reply = Ipv4Repr {
  3171. src_addr: ipv4_repr.dst_addr,
  3172. dst_addr: ipv4_repr.src_addr,
  3173. ..ipv4_repr
  3174. };
  3175. assert_eq!(
  3176. iface.inner.process_icmpv4(&mut sockets, ip_repr, icmp_data),
  3177. Some(IpPacket::Icmpv4((ipv4_reply, echo_reply)))
  3178. );
  3179. let socket = sockets.get::<icmp::Socket>(socket_handle);
  3180. assert!(socket.can_recv());
  3181. assert_eq!(
  3182. socket.recv(),
  3183. Ok((
  3184. icmp_data,
  3185. IpAddress::Ipv4(Ipv4Address::new(0x7f, 0x00, 0x00, 0x02))
  3186. ))
  3187. );
  3188. }
  3189. #[test]
  3190. #[cfg(feature = "proto-ipv6")]
  3191. fn test_solicited_node_addrs() {
  3192. let (mut iface, _) = create_loopback();
  3193. let mut new_addrs = vec![
  3194. IpCidr::new(IpAddress::v6(0xfe80, 0, 0, 0, 1, 2, 0, 2), 64),
  3195. IpCidr::new(IpAddress::v6(0xfe80, 0, 0, 0, 3, 4, 0, 0xffff), 64),
  3196. ];
  3197. iface.update_ip_addrs(|addrs| {
  3198. new_addrs.extend(addrs.to_vec());
  3199. *addrs = From::from(new_addrs);
  3200. });
  3201. assert!(iface
  3202. .inner
  3203. .has_solicited_node(Ipv6Address::new(0xff02, 0, 0, 0, 0, 1, 0xff00, 0x0002)));
  3204. assert!(iface
  3205. .inner
  3206. .has_solicited_node(Ipv6Address::new(0xff02, 0, 0, 0, 0, 1, 0xff00, 0xffff)));
  3207. assert!(!iface
  3208. .inner
  3209. .has_solicited_node(Ipv6Address::new(0xff02, 0, 0, 0, 0, 1, 0xff00, 0x0003)));
  3210. }
  3211. #[test]
  3212. #[cfg(feature = "proto-ipv6")]
  3213. fn test_icmpv6_nxthdr_unknown() {
  3214. let (mut iface, mut sockets) = create_loopback();
  3215. let remote_ip_addr = Ipv6Address::new(0xfe80, 0, 0, 0, 0, 0, 0, 1);
  3216. let payload = [0x12, 0x34, 0x56, 0x78];
  3217. let ipv6_repr = Ipv6Repr {
  3218. src_addr: remote_ip_addr,
  3219. dst_addr: Ipv6Address::LOOPBACK,
  3220. next_header: IpProtocol::HopByHop,
  3221. payload_len: 12,
  3222. hop_limit: 0x40,
  3223. };
  3224. let mut bytes = vec![0; 52];
  3225. let frame = {
  3226. let ip_repr = IpRepr::Ipv6(ipv6_repr);
  3227. ip_repr.emit(&mut bytes, &ChecksumCapabilities::default());
  3228. let mut offset = ipv6_repr.buffer_len();
  3229. {
  3230. let mut hbh_pkt = Ipv6HopByHopHeader::new_unchecked(&mut bytes[offset..]);
  3231. hbh_pkt.set_next_header(IpProtocol::Unknown(0x0c));
  3232. hbh_pkt.set_header_len(0);
  3233. offset += 8;
  3234. {
  3235. let mut pad_pkt = Ipv6Option::new_unchecked(&mut *hbh_pkt.options_mut());
  3236. Ipv6OptionRepr::PadN(3).emit(&mut pad_pkt);
  3237. }
  3238. {
  3239. let mut pad_pkt = Ipv6Option::new_unchecked(&mut hbh_pkt.options_mut()[5..]);
  3240. Ipv6OptionRepr::Pad1.emit(&mut pad_pkt);
  3241. }
  3242. }
  3243. bytes[offset..].copy_from_slice(&payload);
  3244. Ipv6Packet::new_unchecked(&bytes)
  3245. };
  3246. let reply_icmp_repr = Icmpv6Repr::ParamProblem {
  3247. reason: Icmpv6ParamProblem::UnrecognizedNxtHdr,
  3248. pointer: 40,
  3249. header: ipv6_repr,
  3250. data: &payload[..],
  3251. };
  3252. let reply_ipv6_repr = Ipv6Repr {
  3253. src_addr: Ipv6Address::LOOPBACK,
  3254. dst_addr: remote_ip_addr,
  3255. next_header: IpProtocol::Icmpv6,
  3256. payload_len: reply_icmp_repr.buffer_len(),
  3257. hop_limit: 0x40,
  3258. };
  3259. // Ensure the unknown next header causes a ICMPv6 Parameter Problem
  3260. // error message to be sent to the sender.
  3261. assert_eq!(
  3262. iface.inner.process_ipv6(&mut sockets, &frame),
  3263. Some(IpPacket::Icmpv6((reply_ipv6_repr, reply_icmp_repr)))
  3264. );
  3265. }
  3266. #[test]
  3267. #[cfg(feature = "proto-igmp")]
  3268. fn test_handle_igmp() {
  3269. fn recv_igmp(
  3270. iface: &mut Interface<'_, Loopback>,
  3271. timestamp: Instant,
  3272. ) -> Vec<(Ipv4Repr, IgmpRepr)> {
  3273. let caps = iface.device.capabilities();
  3274. let checksum_caps = &caps.checksum;
  3275. recv_all(iface, timestamp)
  3276. .iter()
  3277. .filter_map(|frame| {
  3278. let ipv4_packet = match caps.medium {
  3279. #[cfg(feature = "medium-ethernet")]
  3280. Medium::Ethernet => {
  3281. let eth_frame = EthernetFrame::new_checked(frame).ok()?;
  3282. Ipv4Packet::new_checked(eth_frame.payload()).ok()?
  3283. }
  3284. #[cfg(feature = "medium-ip")]
  3285. Medium::Ip => Ipv4Packet::new_checked(&frame[..]).ok()?,
  3286. #[cfg(feature = "medium-ieee802154")]
  3287. Medium::Ieee802154 => todo!(),
  3288. };
  3289. let ipv4_repr = Ipv4Repr::parse(&ipv4_packet, checksum_caps).ok()?;
  3290. let ip_payload = ipv4_packet.payload();
  3291. let igmp_packet = IgmpPacket::new_checked(ip_payload).ok()?;
  3292. let igmp_repr = IgmpRepr::parse(&igmp_packet).ok()?;
  3293. Some((ipv4_repr, igmp_repr))
  3294. })
  3295. .collect::<Vec<_>>()
  3296. }
  3297. let groups = [
  3298. Ipv4Address::new(224, 0, 0, 22),
  3299. Ipv4Address::new(224, 0, 0, 56),
  3300. ];
  3301. let (mut iface, mut sockets) = create_loopback();
  3302. // Join multicast groups
  3303. let timestamp = Instant::now();
  3304. for group in &groups {
  3305. iface.join_multicast_group(*group, timestamp).unwrap();
  3306. }
  3307. let reports = recv_igmp(&mut iface, timestamp);
  3308. assert_eq!(reports.len(), 2);
  3309. for (i, group_addr) in groups.iter().enumerate() {
  3310. assert_eq!(reports[i].0.next_header, IpProtocol::Igmp);
  3311. assert_eq!(reports[i].0.dst_addr, *group_addr);
  3312. assert_eq!(
  3313. reports[i].1,
  3314. IgmpRepr::MembershipReport {
  3315. group_addr: *group_addr,
  3316. version: IgmpVersion::Version2,
  3317. }
  3318. );
  3319. }
  3320. // General query
  3321. let timestamp = Instant::now();
  3322. const GENERAL_QUERY_BYTES: &[u8] = &[
  3323. 0x46, 0xc0, 0x00, 0x24, 0xed, 0xb4, 0x00, 0x00, 0x01, 0x02, 0x47, 0x43, 0xac, 0x16,
  3324. 0x63, 0x04, 0xe0, 0x00, 0x00, 0x01, 0x94, 0x04, 0x00, 0x00, 0x11, 0x64, 0xec, 0x8f,
  3325. 0x00, 0x00, 0x00, 0x00, 0x02, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  3326. 0x00, 0x00, 0x00, 0x00,
  3327. ];
  3328. {
  3329. // Transmit GENERAL_QUERY_BYTES into loopback
  3330. let tx_token = iface.device.transmit().unwrap();
  3331. tx_token
  3332. .consume(timestamp, GENERAL_QUERY_BYTES.len(), |buffer| {
  3333. buffer.copy_from_slice(GENERAL_QUERY_BYTES);
  3334. Ok(())
  3335. })
  3336. .unwrap();
  3337. }
  3338. // Trigger processing until all packets received through the
  3339. // loopback have been processed, including responses to
  3340. // GENERAL_QUERY_BYTES. Therefore `recv_all()` would return 0
  3341. // pkts that could be checked.
  3342. iface.socket_ingress(&mut sockets);
  3343. // Leave multicast groups
  3344. let timestamp = Instant::now();
  3345. for group in &groups {
  3346. iface.leave_multicast_group(*group, timestamp).unwrap();
  3347. }
  3348. let leaves = recv_igmp(&mut iface, timestamp);
  3349. assert_eq!(leaves.len(), 2);
  3350. for (i, group_addr) in groups.iter().cloned().enumerate() {
  3351. assert_eq!(leaves[i].0.next_header, IpProtocol::Igmp);
  3352. assert_eq!(leaves[i].0.dst_addr, Ipv4Address::MULTICAST_ALL_ROUTERS);
  3353. assert_eq!(leaves[i].1, IgmpRepr::LeaveGroup { group_addr });
  3354. }
  3355. }
  3356. #[test]
  3357. #[cfg(all(feature = "proto-ipv4", feature = "socket-raw"))]
  3358. fn test_raw_socket_no_reply() {
  3359. use crate::wire::{IpVersion, Ipv4Packet, UdpPacket, UdpRepr};
  3360. let (mut iface, mut sockets) = create_loopback();
  3361. let packets = 1;
  3362. let rx_buffer =
  3363. raw::PacketBuffer::new(vec![raw::PacketMetadata::EMPTY; packets], vec![0; 48 * 1]);
  3364. let tx_buffer = raw::PacketBuffer::new(
  3365. vec![raw::PacketMetadata::EMPTY; packets],
  3366. vec![0; 48 * packets],
  3367. );
  3368. let raw_socket = raw::Socket::new(IpVersion::Ipv4, IpProtocol::Udp, rx_buffer, tx_buffer);
  3369. sockets.add(raw_socket);
  3370. let src_addr = Ipv4Address([127, 0, 0, 2]);
  3371. let dst_addr = Ipv4Address([127, 0, 0, 1]);
  3372. const PAYLOAD_LEN: usize = 10;
  3373. let udp_repr = UdpRepr {
  3374. src_port: 67,
  3375. dst_port: 68,
  3376. };
  3377. let mut bytes = vec![0xff; udp_repr.header_len() + PAYLOAD_LEN];
  3378. let mut packet = UdpPacket::new_unchecked(&mut bytes[..]);
  3379. udp_repr.emit(
  3380. &mut packet,
  3381. &src_addr.into(),
  3382. &dst_addr.into(),
  3383. PAYLOAD_LEN,
  3384. |buf| fill_slice(buf, 0x2a),
  3385. &ChecksumCapabilities::default(),
  3386. );
  3387. let ipv4_repr = Ipv4Repr {
  3388. src_addr: src_addr,
  3389. dst_addr: dst_addr,
  3390. next_header: IpProtocol::Udp,
  3391. hop_limit: 64,
  3392. payload_len: udp_repr.header_len() + PAYLOAD_LEN,
  3393. };
  3394. // Emit to frame
  3395. let mut bytes = vec![0u8; ipv4_repr.buffer_len() + udp_repr.header_len() + PAYLOAD_LEN];
  3396. let frame = {
  3397. ipv4_repr.emit(
  3398. &mut Ipv4Packet::new_unchecked(&mut bytes),
  3399. &ChecksumCapabilities::default(),
  3400. );
  3401. udp_repr.emit(
  3402. &mut UdpPacket::new_unchecked(&mut bytes[ipv4_repr.buffer_len()..]),
  3403. &src_addr.into(),
  3404. &dst_addr.into(),
  3405. PAYLOAD_LEN,
  3406. |buf| fill_slice(buf, 0x2a),
  3407. &ChecksumCapabilities::default(),
  3408. );
  3409. Ipv4Packet::new_unchecked(&bytes)
  3410. };
  3411. assert_eq!(iface.inner.process_ipv4(&mut sockets, &frame), None);
  3412. }
  3413. #[test]
  3414. #[cfg(all(feature = "proto-ipv4", feature = "socket-raw", feature = "socket-udp"))]
  3415. fn test_raw_socket_with_udp_socket() {
  3416. use crate::wire::{IpEndpoint, IpVersion, Ipv4Packet, UdpPacket, UdpRepr};
  3417. static UDP_PAYLOAD: [u8; 5] = [0x48, 0x65, 0x6c, 0x6c, 0x6f];
  3418. let (mut iface, mut sockets) = create_loopback();
  3419. let udp_rx_buffer = udp::PacketBuffer::new(vec![udp::PacketMetadata::EMPTY], vec![0; 15]);
  3420. let udp_tx_buffer = udp::PacketBuffer::new(vec![udp::PacketMetadata::EMPTY], vec![0; 15]);
  3421. let udp_socket = udp::Socket::new(udp_rx_buffer, udp_tx_buffer);
  3422. let udp_socket_handle = sockets.add(udp_socket);
  3423. // Bind the socket to port 68
  3424. let socket = sockets.get::<udp::Socket>(udp_socket_handle);
  3425. assert_eq!(socket.bind(68), Ok(()));
  3426. assert!(!socket.can_recv());
  3427. assert!(socket.can_send());
  3428. let packets = 1;
  3429. let raw_rx_buffer =
  3430. raw::PacketBuffer::new(vec![raw::PacketMetadata::EMPTY; packets], vec![0; 48 * 1]);
  3431. let raw_tx_buffer = raw::PacketBuffer::new(
  3432. vec![raw::PacketMetadata::EMPTY; packets],
  3433. vec![0; 48 * packets],
  3434. );
  3435. let raw_socket = raw::Socket::new(
  3436. IpVersion::Ipv4,
  3437. IpProtocol::Udp,
  3438. raw_rx_buffer,
  3439. raw_tx_buffer,
  3440. );
  3441. sockets.add(raw_socket);
  3442. let src_addr = Ipv4Address([127, 0, 0, 2]);
  3443. let dst_addr = Ipv4Address([127, 0, 0, 1]);
  3444. let udp_repr = UdpRepr {
  3445. src_port: 67,
  3446. dst_port: 68,
  3447. };
  3448. let mut bytes = vec![0xff; udp_repr.header_len() + UDP_PAYLOAD.len()];
  3449. let mut packet = UdpPacket::new_unchecked(&mut bytes[..]);
  3450. udp_repr.emit(
  3451. &mut packet,
  3452. &src_addr.into(),
  3453. &dst_addr.into(),
  3454. UDP_PAYLOAD.len(),
  3455. |buf| buf.copy_from_slice(&UDP_PAYLOAD),
  3456. &ChecksumCapabilities::default(),
  3457. );
  3458. let ipv4_repr = Ipv4Repr {
  3459. src_addr: src_addr,
  3460. dst_addr: dst_addr,
  3461. next_header: IpProtocol::Udp,
  3462. hop_limit: 64,
  3463. payload_len: udp_repr.header_len() + UDP_PAYLOAD.len(),
  3464. };
  3465. // Emit to frame
  3466. let mut bytes =
  3467. vec![0u8; ipv4_repr.buffer_len() + udp_repr.header_len() + UDP_PAYLOAD.len()];
  3468. let frame = {
  3469. ipv4_repr.emit(
  3470. &mut Ipv4Packet::new_unchecked(&mut bytes),
  3471. &ChecksumCapabilities::default(),
  3472. );
  3473. udp_repr.emit(
  3474. &mut UdpPacket::new_unchecked(&mut bytes[ipv4_repr.buffer_len()..]),
  3475. &src_addr.into(),
  3476. &dst_addr.into(),
  3477. UDP_PAYLOAD.len(),
  3478. |buf| buf.copy_from_slice(&UDP_PAYLOAD),
  3479. &ChecksumCapabilities::default(),
  3480. );
  3481. Ipv4Packet::new_unchecked(&bytes)
  3482. };
  3483. assert_eq!(iface.inner.process_ipv4(&mut sockets, &frame), None);
  3484. // Make sure the UDP socket can still receive in presence of a Raw socket that handles UDP
  3485. let socket = sockets.get::<udp::Socket>(udp_socket_handle);
  3486. assert!(socket.can_recv());
  3487. assert_eq!(
  3488. socket.recv(),
  3489. Ok((&UDP_PAYLOAD[..], IpEndpoint::new(src_addr.into(), 67)))
  3490. );
  3491. }
  3492. }