From 1ef81bf07f9bfbfd57c8c4d408f037eb20ce7367 Mon Sep 17 00:00:00 2001
From: Jake McGinty <me@jake.su>
Date: Fri, 9 Feb 2018 14:21:10 +0000
Subject: router refactoring

---
 src/ip_packet.rs | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 src/ip_packet.rs

(limited to 'src/ip_packet.rs')

diff --git a/src/ip_packet.rs b/src/ip_packet.rs
new file mode 100644
index 0000000..4aa7ecb
--- /dev/null
+++ b/src/ip_packet.rs
@@ -0,0 +1,33 @@
+use failure::Error;
+use pnet::packet::ipv4::Ipv4Packet;
+use pnet::packet::ipv6::Ipv6Packet;
+use std::net::{Ipv4Addr, Ipv6Addr, IpAddr, SocketAddr};
+
+pub enum IpPacket<'a> {
+    V4(Ipv4Packet<'a>),
+    V6(Ipv6Packet<'a>),
+}
+
+impl<'a> IpPacket<'a> {
+    pub fn new(packet: &'a [u8]) -> Option<Self> {
+        match packet[0] >> 4 {
+            4 => Ipv4Packet::new(&packet).map(|packet| IpPacket::V4(packet)),
+            6 => Ipv6Packet::new(&packet).map(|packet| IpPacket::V6(packet)),
+            _ => None
+        }
+    }
+
+    pub fn get_source(&self) -> IpAddr {
+        match *self {
+            IpPacket::V4(ref packet) => packet.get_source().into(),
+            IpPacket::V6(ref packet) => packet.get_source().into(),
+        }
+    }
+
+    pub fn get_destination(&self) -> IpAddr {
+        match *self {
+            IpPacket::V4(ref packet) => packet.get_destination().into(),
+            IpPacket::V6(ref packet) => packet.get_destination().into(),
+        }
+    }
+}
-- 
cgit v1.2.3-59-g8ed1b