diff options
| author |   Mathias Hall-Andersen <mathias@hall-andersen.dk> | 2019-10-28 14:48:24 +0100 |
|---|---|---|
| committer | Mathias Hall-Andersen <mathias@hall-andersen.dk> | 2019-10-28 14:48:24 +0100 |
| commit | 4ff328b7da876fb3305fefd83865553af9c8ab2c (patch) | |
| tree | bd1680d9f7316415e3044fd152ef503729d97239 /src/wireguard/router/device.rs | |
| parent | Fixed Ordering::Acquire -> Ordering::SeqCst typo (diff) | |
| download | wireguard-rs-4ff328b7da876fb3305fefd83865553af9c8ab2c.tar.xz wireguard-rs-4ff328b7da876fb3305fefd83865553af9c8ab2c.zip | |
First full test of pure WireGuard
Diffstat (limited to 'src/wireguard/router/device.rs')
| -rw-r--r-- | src/wireguard/router/device.rs | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/src/wireguard/router/device.rs b/src/wireguard/router/device.rs index b122bf4..254b3de 100644 --- a/src/wireguard/router/device.rs +++ b/src/wireguard/router/device.rs @@ -89,13 +89,7 @@ fn get_route<E: Endpoint, C: Callbacks, T: tun::Writer, B: bind::Writer<E>>( device: &Arc<DeviceInner<E, C, T, B>>, packet: &[u8], ) -> Option<Arc<PeerInner<E, C, T, B>>> { - // ensure version access within bounds - if packet.len() < 1 { - return None; - }; - - // cast to correct IP header - match packet[0] >> 4 { + match packet.get(0)? >> 4 { VERSION_IP4 => { // check length and cast to IPv4 header let (header, _): (LayoutVerified<&[u8], IPv4Header>, _) = @@ -176,7 +170,7 @@ impl<E: Endpoint, C: Callbacks, T: tun::Writer, B: bind::Writer<E>> Device<E, C, let packet = &msg[SIZE_MESSAGE_PREFIX..]; // lookup peer based on IP packet destination address - let peer = get_route(&self.state, packet).ok_or(RouterError::NoCryptKeyRoute)?; + let peer = get_route(&self.state, packet).ok_or(RouterError::NoCryptoKeyRoute)?; // schedule for encryption and transmission to peer if let Some(job) = peer.send_job(msg, true) { |