Red Bear OS
67 lines
2.0 KiB
Rust
67 lines
2.0 KiB
Rust
pub mod ethernet;
|
|
pub mod loopback;
|
|
pub mod stp;
|
|
|
|
use std::rc::Rc;
|
|
|
|
use smoltcp::time::Instant;
|
|
use smoltcp::wire::{EthernetAddress, IpAddress, IpCidr};
|
|
|
|
/// Represent a link layer device (eth0, loopback...)
|
|
pub trait LinkDevice {
|
|
/// Send the given packet to the machine with the `next_hop` ip address
|
|
/// This method cannot fail so it's the implementor responsability
|
|
/// to buffer packets which can't be sent immediatly or decide to
|
|
/// drop them if necessary
|
|
fn send(&mut self, next_hop: IpAddress, packet: &[u8], now: Instant);
|
|
|
|
/// Returns None if nothing is received.
|
|
/// Returns an Ip packet otherwise
|
|
fn recv(&mut self, now: Instant) -> Option<&[u8]>;
|
|
|
|
/// Returns the LinkDevice display name used to refer to it and for lookups
|
|
fn name(&self) -> &Rc<str>;
|
|
|
|
/// Returns wether this device have packets pending
|
|
fn can_recv(&self) -> bool;
|
|
|
|
fn mac_address(&self) -> Option<EthernetAddress>;
|
|
fn set_mac_address(&mut self, addr: EthernetAddress);
|
|
|
|
fn ip_address(&self) -> Option<IpCidr>;
|
|
fn set_ip_address(&mut self, addr: IpCidr);
|
|
}
|
|
|
|
#[derive(Default)]
|
|
pub struct DeviceList {
|
|
inner: Vec<Box<dyn LinkDevice>>,
|
|
}
|
|
|
|
impl DeviceList {
|
|
pub fn push<T: LinkDevice + 'static>(&mut self, dev: T) {
|
|
self.inner.push(Box::new(dev))
|
|
}
|
|
|
|
pub fn get(&self, device_name: &str) -> Option<&dyn LinkDevice> {
|
|
self.inner
|
|
.iter()
|
|
.find(|dev| dev.name().as_ref() == device_name)
|
|
.map(|device| device.as_ref())
|
|
}
|
|
|
|
pub fn get_mut(&mut self, device_name: &str) -> Option<&mut (dyn LinkDevice + 'static)> {
|
|
self.inner
|
|
.iter_mut()
|
|
.find(|dev| dev.name().as_ref() == device_name)
|
|
.map(|device| device.as_mut())
|
|
}
|
|
|
|
pub fn iter(&self) -> impl Iterator<Item = &(dyn LinkDevice + 'static)> {
|
|
self.inner.iter().map(|b| b.as_ref())
|
|
}
|
|
|
|
pub fn iter_mut(&mut self) -> impl Iterator<Item = &mut (dyn LinkDevice + 'static)> {
|
|
self.inner.iter_mut().map(|b| b.as_mut())
|
|
}
|
|
}
|