137 lines
5 KiB
Rust
137 lines
5 KiB
Rust
#![feature(asm)]
|
|
|
|
extern crate dma;
|
|
extern crate event;
|
|
extern crate netutils;
|
|
extern crate syscall;
|
|
|
|
use std::cell::RefCell;
|
|
use std::env;
|
|
use std::fs::File;
|
|
use std::io::{Read, Write, Result};
|
|
use std::os::unix::io::{AsRawFd, FromRawFd};
|
|
use std::sync::Arc;
|
|
|
|
use event::EventQueue;
|
|
use syscall::{Packet, Scheme, MAP_WRITE};
|
|
use syscall::error::EWOULDBLOCK;
|
|
|
|
pub mod device;
|
|
|
|
fn main() {
|
|
let mut args = env::args().skip(1);
|
|
|
|
let mut name = args.next().expect("e1000d: no name provided");
|
|
name.push_str("_e1000");
|
|
|
|
let bar_str = args.next().expect("e1000d: no address provided");
|
|
let bar = usize::from_str_radix(&bar_str, 16).expect("e1000d: failed to parse address");
|
|
|
|
let irq_str = args.next().expect("e1000d: no irq provided");
|
|
let irq = irq_str.parse::<u8>().expect("e1000d: failed to parse irq");
|
|
|
|
print!("{}", format!(" + E1000 {} on: {:X}, IRQ: {}\n", name, bar, irq));
|
|
|
|
// Daemonize
|
|
if unsafe { syscall::clone(0).unwrap() } == 0 {
|
|
let socket_fd = syscall::open(":network", syscall::O_RDWR | syscall::O_CREAT | syscall::O_NONBLOCK).expect("e1000d: failed to create network scheme");
|
|
let socket = Arc::new(RefCell::new(unsafe { File::from_raw_fd(socket_fd) }));
|
|
|
|
let address = unsafe { syscall::physmap(bar, 128*1024, MAP_WRITE).expect("e1000d: failed to map address") };
|
|
{
|
|
let device = Arc::new(unsafe { device::Intel8254x::new(address).expect("e1000d: failed to allocate device") });
|
|
|
|
let mut event_queue = EventQueue::<usize>::new().expect("e1000d: failed to create event queue");
|
|
|
|
let todo = Arc::new(RefCell::new(Vec::<Packet>::new()));
|
|
|
|
let device_irq = device.clone();
|
|
let socket_irq = socket.clone();
|
|
let todo_irq = todo.clone();
|
|
let mut irq_file = File::open(format!("irq:{}", irq)).expect("e1000d: failed to open IRQ file");
|
|
event_queue.add(irq_file.as_raw_fd(), move |_count: usize| -> Result<Option<usize>> {
|
|
let mut irq = [0; 8];
|
|
irq_file.read(&mut irq)?;
|
|
if unsafe { device_irq.irq() } {
|
|
irq_file.write(&mut irq)?;
|
|
|
|
let mut todo = todo_irq.borrow_mut();
|
|
let mut i = 0;
|
|
while i < todo.len() {
|
|
let a = todo[i].a;
|
|
device_irq.handle(&mut todo[i]);
|
|
if todo[i].a == (-EWOULDBLOCK) as usize {
|
|
todo[i].a = a;
|
|
i += 1;
|
|
} else {
|
|
socket_irq.borrow_mut().write(&mut todo[i])?;
|
|
todo.remove(i);
|
|
}
|
|
}
|
|
|
|
let next_read = device_irq.next_read();
|
|
if next_read > 0 {
|
|
return Ok(Some(next_read));
|
|
}
|
|
}
|
|
Ok(None)
|
|
}).expect("e1000d: failed to catch events on IRQ file");
|
|
|
|
let socket_packet = socket.clone();
|
|
event_queue.add(socket_fd, move |_count: usize| -> Result<Option<usize>> {
|
|
loop {
|
|
let mut packet = Packet::default();
|
|
if socket_packet.borrow_mut().read(&mut packet)? == 0 {
|
|
break;
|
|
}
|
|
|
|
let a = packet.a;
|
|
device.handle(&mut packet);
|
|
if packet.a == (-EWOULDBLOCK) as usize {
|
|
packet.a = a;
|
|
todo.borrow_mut().push(packet);
|
|
} else {
|
|
socket_packet.borrow_mut().write(&mut packet)?;
|
|
}
|
|
}
|
|
|
|
let next_read = device.next_read();
|
|
if next_read > 0 {
|
|
return Ok(Some(next_read));
|
|
}
|
|
|
|
Ok(None)
|
|
}).expect("e1000d: failed to catch events on IRQ file");
|
|
|
|
for event_count in event_queue.trigger_all(0).expect("e1000d: failed to trigger events") {
|
|
socket.borrow_mut().write(&Packet {
|
|
id: 0,
|
|
pid: 0,
|
|
uid: 0,
|
|
gid: 0,
|
|
a: syscall::number::SYS_FEVENT,
|
|
b: 0,
|
|
c: syscall::flag::EVENT_READ,
|
|
d: event_count
|
|
}).expect("e1000d: failed to write event");
|
|
}
|
|
|
|
loop {
|
|
let event_count = event_queue.run().expect("e1000d: failed to handle events");
|
|
|
|
socket.borrow_mut().write(&Packet {
|
|
id: 0,
|
|
pid: 0,
|
|
uid: 0,
|
|
gid: 0,
|
|
a: syscall::number::SYS_FEVENT,
|
|
b: 0,
|
|
c: syscall::flag::EVENT_READ,
|
|
d: event_count
|
|
}).expect("e1000d: failed to write event");
|
|
}
|
|
}
|
|
unsafe { let _ = syscall::physunmap(address); }
|
|
}
|
|
}
|