renet/src/phy/raw_socket.rs

79 lines
1.9 KiB
Rust
Raw Normal View History

2016-12-12 15:19:53 +08:00
use std::cell::RefCell;
use std::vec::Vec;
2016-12-12 20:30:35 +08:00
use std::rc::Rc;
2016-12-12 15:19:53 +08:00
use std::io;
2016-12-12 20:30:35 +08:00
use Error;
use super::{sys, DeviceLimits, Device};
2016-12-11 07:15:26 +08:00
/// A socket that captures or transmits the complete frame.
#[derive(Debug)]
pub struct RawSocket {
2016-12-12 20:30:35 +08:00
lower: Rc<RefCell<sys::RawSocketDesc>>,
mtu: usize
}
impl RawSocket {
2016-12-11 07:15:26 +08:00
/// Creates a raw socket, bound to the interface called `name`.
///
/// This requires superuser privileges or a corresponding capability bit
/// set on the executable.
pub fn new(name: &str) -> io::Result<RawSocket> {
2016-12-11 07:15:26 +08:00
let mut lower = try!(sys::RawSocketDesc::new(name));
try!(lower.bind_interface());
2016-12-12 20:30:35 +08:00
let mtu = try!(lower.interface_mtu());
2016-12-11 07:15:26 +08:00
Ok(RawSocket {
2016-12-12 20:30:35 +08:00
lower: Rc::new(RefCell::new(lower)),
mtu: mtu
2016-12-11 07:15:26 +08:00
})
}
}
2016-12-11 07:15:26 +08:00
impl Device for RawSocket {
2016-12-12 20:30:35 +08:00
type RxBuffer = Vec<u8>;
type TxBuffer = TxBuffer;
fn limits(&self) -> DeviceLimits {
DeviceLimits {
max_transmission_unit: self.mtu,
..DeviceLimits::default()
}
}
2016-12-12 20:30:35 +08:00
fn receive(&mut self) -> Result<Self::RxBuffer, Error> {
2016-12-12 15:19:53 +08:00
let mut lower = self.lower.borrow_mut();
2016-12-12 20:30:35 +08:00
let mut buffer = vec![0; self.mtu];
2016-12-20 07:50:04 +08:00
let size = lower.recv(&mut buffer[..]).unwrap();
buffer.resize(size, 0);
2016-12-12 20:30:35 +08:00
Ok(buffer)
2016-12-12 15:19:53 +08:00
}
fn transmit(&mut self, length: usize) -> Result<Self::TxBuffer, Error> {
2016-12-12 20:30:35 +08:00
Ok(TxBuffer {
lower: self.lower.clone(),
buffer: vec![0; length]
2016-12-12 20:30:35 +08:00
})
}
2016-12-12 20:30:35 +08:00
}
2016-12-12 20:30:35 +08:00
#[doc(hidden)]
pub struct TxBuffer {
lower: Rc<RefCell<sys::RawSocketDesc>>,
buffer: Vec<u8>
}
impl AsRef<[u8]> for TxBuffer {
fn as_ref(&self) -> &[u8] { self.buffer.as_ref() }
}
impl AsMut<[u8]> for TxBuffer {
fn as_mut(&mut self) -> &mut [u8] { self.buffer.as_mut() }
}
impl Drop for TxBuffer {
fn drop(&mut self) {
let mut lower = self.lower.borrow_mut();
lower.send(&mut self.buffer[..]).unwrap();
}
}