deps: Replace pnet_packet with rips-packets

Closes #11

3 files changed, 13 insertions, 17 deletions

diff --git a/src/interface/mod.rs b/src/interface/mod.rs
index c45e70f..7036a09 100644
--- a/src/interface/mod.rs
+++ b/src/interface/mod.rs
@@ -14,7 +14,7 @@ use std::cell::RefCell;
 use std::collections::HashMap;
 use types::{InterfaceInfo};
 
-use pnet_packet::ipv4::Ipv4Packet;
+use rips_packets::ipv4::Ipv4Packet;
 
 use futures::{Future, Stream, Sink, unsync};
 use tokio_core::reactor::Core;
diff --git a/src/ip_packet.rs b/src/ip_packet.rs
index 3de6a60..9150f5b 100644
--- a/src/ip_packet.rs
+++ b/src/ip_packet.rs
@@ -1,5 +1,5 @@
-use pnet_packet::ipv4::Ipv4Packet;
-use pnet_packet::ipv6::Ipv6Packet;
+use rips_packets::ipv4::Ipv4Packet;
+use rips_packets::ipv6::Ipv6Packet;
 use std::net::IpAddr;
 
 pub enum IpPacket<'a> {
@@ -9,35 +9,31 @@ pub enum IpPacket<'a> {
 
 impl<'a> IpPacket<'a> {
     pub fn new(packet: &'a [u8]) -> Option<Self> {
-        if packet.is_empty() {
-            return None;
-        }
-
-        match packet[0] >> 4 {
-            4 => Ipv4Packet::new(packet).map(IpPacket::V4),
-            6 => Ipv6Packet::new(packet).map(IpPacket::V6),
+        match packet.get(0).map(|byte| *byte >> 4) {
+            Some(4) => Ipv4Packet::new(packet).map(IpPacket::V4),
+            Some(6) => Ipv6Packet::new(packet).map(IpPacket::V6),
             _ => None
         }
     }
 
     pub fn source(&self) -> IpAddr {
         match *self {
-            IpPacket::V4(ref packet) => packet.get_source().into(),
-            IpPacket::V6(ref packet) => packet.get_source().into(),
+            IpPacket::V4(ref packet) => packet.source().into(),
+            IpPacket::V6(ref packet) => packet.source().into(),
         }
     }
 
     pub fn destination(&self) -> IpAddr {
         match *self {
-            IpPacket::V4(ref packet) => packet.get_destination().into(),
-            IpPacket::V6(ref packet) => packet.get_destination().into(),
+            IpPacket::V4(ref packet) => packet.destination().into(),
+            IpPacket::V6(ref packet) => packet.destination().into(),
         }
     }
 
     pub fn length(&self) -> u16 {
         match *self {
-            IpPacket::V4(ref packet) => packet.get_total_length(),
-            IpPacket::V6(ref packet) => 40 + packet.get_payload_length(),
+            IpPacket::V4(ref packet) => packet.total_length(),
+            IpPacket::V6(ref packet) => 40 + packet.payload_length(),
         }
 
     }
diff --git a/src/lib.rs b/src/lib.rs
index 22b8fb3..5b41b8f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -26,8 +26,8 @@ extern crate libc;
 extern crate mio;
 extern crate nix;
 extern crate notify;
-extern crate pnet_packet;
 extern crate rand;
+extern crate rips_packets;
 extern crate snow;
 extern crate socket2;
 extern crate subtle;