use super::*;
use crate::managed::Slice;
use crate::nic::{external::External, Device};
use crate::layer::{arp, eth, ip};
use crate::wire::{ethernet, ip::v4, ip::v6};
use crate::wire::ip::{Address, Cidr, Protocol, Subnet};
use crate::wire::{Payload, PayloadMut};
static PAYLOAD_BYTES: [u8; 50] =
[0xaa, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0xff];
struct SimpleSend {
dst_addr: Address,
}
#[test]
fn simple_ipv4() {
const MAC_ADDR_SRC: ethernet::Address = ethernet::Address([0, 1, 2, 3, 4, 5]);
const IP_ADDR_SRC: v4::Address = v4::Address::new(10, 0, 0, 1);
const MAC_ADDR_DST: ethernet::Address = ethernet::Address([6, 5, 4, 3, 2, 1]);
const IP_ADDR_DST: v4::Address = v4::Address::new(10, 0, 0, 2);
let mut nic = External::new_send(Slice::One(vec![0; 1024]));
let mut eth = eth::Endpoint::new(MAC_ADDR_SRC);
let mut neighbors = [arp::Neighbor::default(); 1];
let neighbors = {
let mut eth_cache = arp::NeighborCache::new(&mut neighbors[..]);
eth_cache.fill(IP_ADDR_DST.into(), MAC_ADDR_DST, None).unwrap();
eth_cache
};
let mut ip = [ip::Route::unspecified(); 2];
let mut ip = ip::Endpoint::new(Cidr::new(IP_ADDR_SRC.into(), 24),
ip::Routes::new(&mut ip[..]),
neighbors);
let sent = nic.tx(1, eth.send(ip.send(SimpleSend {
dst_addr: IP_ADDR_DST.into(),
})));
assert_eq!(sent, Ok(1));
{
let buffer = nic.get_mut(0).unwrap();
let eth = ethernet::frame::new_unchecked_mut(buffer);
eth.set_dst_addr(MAC_ADDR_SRC);
eth.set_src_addr(MAC_ADDR_DST);
let ip = v4::packet::new_unchecked_mut(eth.payload_mut_slice());
ip.set_dst_addr(IP_ADDR_SRC);
ip.set_src_addr(IP_ADDR_DST);
ip.fill_checksum();
}
nic.receive_all();
let recv = nic.rx(1,
eth.recv(ip.recv_with(simple_recv)));
assert_eq!(recv, Ok(1));
}
#[test]
fn simple_ipv6() {
const MAC_ADDR_SRC: ethernet::Address = ethernet::Address([0, 1, 2, 3, 4, 5]);
const IP_ADDR_SRC: v6::Address = v6::Address::from_link_local_id(v6::InterfaceId::from_generated_ether(MAC_ADDR_SRC));
const MAC_ADDR_DST: ethernet::Address = ethernet::Address([6, 5, 4, 3, 2, 1]);
const IP_ADDR_DST: v6::Address = v6::Address::from_link_local_id(v6::InterfaceId::from_generated_ether(MAC_ADDR_DST));
let mut nic = External::new_send(Slice::One(vec![0; 1024]));
let mut eth = eth::Endpoint::new(MAC_ADDR_SRC);
let mut neighbors = [arp::Neighbor::default(); 1];
let neighbors = {
let mut eth_cache = arp::NeighborCache::new(&mut neighbors[..]);
eth_cache.fill(IP_ADDR_DST.into(), MAC_ADDR_DST, None).unwrap();
eth_cache
};
let mut ip = [ip::Route::unspecified(); 2];
let mut ip = ip::Endpoint::new(Cidr::new(IP_ADDR_SRC.into(), 24),
ip::Routes::new(&mut ip[..]),
neighbors);
let sent = nic.tx(1, eth.send(ip.send(SimpleSend {
dst_addr: IP_ADDR_DST.into(),
})));
assert_eq!(sent, Ok(1));
{
let buffer = nic.get_mut(0).unwrap();
let eth = ethernet::frame::new_unchecked_mut(buffer);
eth.set_dst_addr(MAC_ADDR_SRC);
eth.set_src_addr(MAC_ADDR_DST);
let ip = v6::packet::new_unchecked_mut(eth.payload_mut_slice());
ip.set_dst_addr(IP_ADDR_SRC);
ip.set_src_addr(IP_ADDR_DST);
}
nic.receive_all();
let recv = nic.rx(1,
eth.recv(ip.recv_with(simple_recv)));
assert_eq!(recv, Ok(1));
}
fn simple_recv<P: Payload>(frame: InPacket<P>) {
assert_eq!(frame.packet.payload().as_slice(), &PAYLOAD_BYTES[..]);
}
impl<P: PayloadMut> ip::Send<P> for SimpleSend {
fn send(&mut self, packet: RawPacket<P>) {
let init = ip::Init {
source: match self.dst_addr {
Address::Ipv4(_) => Subnet::from(v4::Subnet::ANY),
Address::Ipv6(_) => Subnet::from(v6::Subnet::ANY),
_ => unreachable!(),
}.into(),
dst_addr: self.dst_addr,
payload: PAYLOAD_BYTES.len(),
protocol: Protocol::Unknown(0xEF),
};
let mut prepared = packet.prepare(init)
.expect("Found no valid routes");
prepared
.payload_mut_slice()
.copy_from_slice(&PAYLOAD_BYTES[..]);
prepared.send()
.expect("Could actuall egress packet");
}
}