vpncloud/src/ethernet.rs

138 lines
3.6 KiB
Rust
Raw Normal View History

2015-11-22 19:02:02 +00:00
use std::{mem, fmt};
2015-11-21 15:50:50 +00:00
use std::net::SocketAddr;
use std::collections::HashMap;
2015-11-22 19:02:02 +00:00
use std::io::Write;
2015-11-19 15:34:20 +00:00
2015-11-22 21:00:34 +00:00
use super::cloud::{Error, Table, Protocol, Address};
2015-11-22 18:00:56 +00:00
use super::util::as_obj;
2015-11-19 15:34:20 +00:00
2015-11-21 15:50:50 +00:00
use time::{Duration, SteadyTime};
2015-11-22 16:28:04 +00:00
2015-11-21 17:09:13 +00:00
#[derive(Clone, Copy, Hash, PartialEq, Eq, PartialOrd, Ord)]
pub struct Mac(pub [u8; 6]);
impl fmt::Debug for Mac {
fn fmt(&self, formatter: &mut fmt::Formatter) -> Result<(), fmt::Error> {
write!(formatter, "{:x}:{:x}:{:x}:{:x}:{:x}:{:x}",
self.0[0], self.0[1], self.0[2], self.0[3], self.0[4], self.0[5])
}
}
2015-11-20 17:09:51 +00:00
pub type VlanId = u16;
2015-11-21 17:09:13 +00:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct EthAddr {
pub mac: Mac,
2015-11-22 15:50:57 +00:00
pub vlan: Option<VlanId>
2015-11-21 17:09:13 +00:00
}
2015-11-22 21:00:34 +00:00
impl Address for EthAddr {
fn from_bytes(_bytes: &[u8]) -> Result<Self, Error> {
unimplemented!()
}
fn to_bytes(&self) -> Vec<u8> {
unimplemented!()
}
}
2015-11-22 16:28:04 +00:00
2015-11-19 15:34:20 +00:00
#[derive(PartialEq)]
2015-11-22 17:05:15 +00:00
pub struct Frame;
2015-11-21 17:09:13 +00:00
2015-11-22 17:05:15 +00:00
impl Protocol for Frame {
2015-11-21 17:09:13 +00:00
type Address = EthAddr;
2015-11-22 17:05:15 +00:00
fn parse(data: &[u8]) -> Result<(EthAddr, EthAddr), Error> {
2015-11-22 15:48:01 +00:00
if data.len() < 14 {
return Err(Error::ParseError("Frame is too short"));
}
let mut pos = 0;
let dst = *unsafe { as_obj::<Mac>(&data[pos..]) };
pos += mem::size_of::<Mac>();
let src = *unsafe { as_obj::<Mac>(&data[pos..]) };
pos += mem::size_of::<Mac>();
2015-11-22 15:50:57 +00:00
let mut vlan = None;
2015-11-22 15:48:01 +00:00
if data[pos] == 0x81 && data[pos+1] == 0x00 {
pos += 2;
if data.len() < pos + 2 {
return Err(Error::ParseError("Vlan frame is too short"));
}
2015-11-22 15:50:57 +00:00
vlan = Some(u16::from_be(* unsafe { as_obj::<u16>(&data[pos..]) }));
2015-11-22 15:48:01 +00:00
}
2015-11-22 17:05:15 +00:00
Ok((EthAddr{mac: src, vlan: vlan}, EthAddr{mac: dst, vlan: vlan}))
2015-11-22 15:48:01 +00:00
}
}
2015-11-22 16:28:04 +00:00
2015-11-21 15:50:50 +00:00
struct MacTableValue {
address: SocketAddr,
timeout: SteadyTime
}
2015-11-22 16:28:04 +00:00
2015-11-21 15:50:50 +00:00
pub struct MacTable {
2015-11-21 17:09:13 +00:00
table: HashMap<EthAddr, MacTableValue>,
2015-11-21 15:50:50 +00:00
timeout: Duration
}
impl MacTable {
pub fn new(timeout: Duration) -> MacTable {
MacTable{table: HashMap::new(), timeout: timeout}
}
2015-11-21 17:09:13 +00:00
}
impl Table for MacTable {
type Address = EthAddr;
2015-11-21 15:50:50 +00:00
2015-11-21 17:09:13 +00:00
fn housekeep(&mut self) {
2015-11-21 15:50:50 +00:00
let now = SteadyTime::now();
2015-11-21 17:09:13 +00:00
let mut del: Vec<Self::Address> = Vec::new();
2015-11-21 15:50:50 +00:00
for (&key, val) in &self.table {
if val.timeout < now {
del.push(key);
}
}
for key in del {
2015-11-22 15:50:57 +00:00
info!("Forgot mac: {:?} (vlan {:?})", key.mac, key.vlan);
2015-11-21 15:50:50 +00:00
self.table.remove(&key);
}
}
2015-11-21 17:09:13 +00:00
fn learn(&mut self, key: Self::Address, addr: SocketAddr) {
let value = MacTableValue{address: addr, timeout: SteadyTime::now()+self.timeout};
2015-11-21 15:50:50 +00:00
if self.table.insert(key, value).is_none() {
2015-11-22 15:50:57 +00:00
info!("Learned mac: {:?} (vlan {:?}) => {}", key.mac, key.vlan, addr);
2015-11-21 15:50:50 +00:00
}
}
2015-11-22 17:05:15 +00:00
fn lookup(&self, key: &Self::Address) -> Option<SocketAddr> {
match self.table.get(key) {
2015-11-21 15:50:50 +00:00
Some(value) => Some(value.address),
None => None
}
}
2015-11-22 21:00:34 +00:00
fn remove_all(&mut self, _addr: SocketAddr) {
unimplemented!()
}
2015-11-21 15:50:50 +00:00
}
2015-11-19 15:34:20 +00:00
#[test]
fn without_vlan() {
2015-11-22 17:05:15 +00:00
let data = [6,5,4,3,2,1,1,2,3,4,5,6,1,2,3,4,5,6,7,8];
let (src, dst) = Frame::parse(&data).unwrap();
assert_eq!(src, EthAddr{mac: Mac([1,2,3,4,5,6]), vlan: None});
assert_eq!(dst, EthAddr{mac: Mac([6,5,4,3,2,1]), vlan: None});
2015-11-19 15:34:20 +00:00
}
#[test]
fn with_vlan() {
2015-11-22 17:05:15 +00:00
let data = [6,5,4,3,2,1,1,2,3,4,5,6,0x81,0,4,210,1,2,3,4,5,6,7,8];
let (src, dst) = Frame::parse(&data).unwrap();
assert_eq!(src, EthAddr{mac: Mac([1,2,3,4,5,6]), vlan: Some(1234)});
assert_eq!(dst, EthAddr{mac: Mac([6,5,4,3,2,1]), vlan: Some(1234)});
2015-11-19 15:34:20 +00:00
}