Orbital (#16)
* Port previous ethernet scheme * Add ipd * Fix initfs rebuilds, use QEMU user networking addresses in ipd * Add tcp/udp, netutils, dns, and network config * Add fsync to network driver * Add dns, router, subnet by default * Fix e1000 driver. Make ethernet and IP non-blocking to avoid deadlocks * Add orbital server, WIP * Add futex * Add orbutils and orbital * Update libstd, orbutils, and orbital Move ANSI key encoding to vesad * Add orbital assets * Update orbital * Update to add login manager * Add blocking primitives, block for most things except waitpid, update orbital * Wait in waitpid and IRQ, improvements for other waits * Fevent in root scheme * WIP: Switch to using fevent * Reorganize * Event based e1000d driver * Superuser-only access to some network schemes, display, and disk * Superuser root and irq schemes * Fix orbital
This commit is contained in:
Jeremy Soller
GitHub
parent
372d44f88c
commit
224c43f761
1 changed files with 3418 additions and 476 deletions
155
schemes/ipd/src/scheme.rs
Normal file
155
schemes/ipd/src/scheme.rs
Normal file
|
|
@ -0,0 +1,155 @@
|
|||
use std::cell::RefCell;
|
||||
use std::rand;
|
||||
use std::{str, u16};
|
||||
|
||||
use resource_scheme::ResourceScheme;
|
||||
use syscall;
|
||||
use syscall::error::{Error, Result, EACCES, ENOENT, EINVAL};
|
||||
use syscall::flag::O_RDWR;
|
||||
|
||||
use common::{n16, MacAddr, Ipv4Addr, ArpHeader, Arp, Ipv4, MAC_ADDR, BROADCAST_MAC_ADDR, BROADCAST_IP_ADDR, IP_ADDR, IP_ROUTER_ADDR, IP_SUBNET};
|
||||
use resource::IpResource;
|
||||
|
||||
/// A ARP entry (MAC + IP)
|
||||
pub struct ArpEntry {
|
||||
ip: Ipv4Addr,
|
||||
mac: MacAddr,
|
||||
}
|
||||
|
||||
/// A IP scheme
|
||||
pub struct IpScheme {
|
||||
pub arp: RefCell<Vec<ArpEntry>>,
|
||||
}
|
||||
|
||||
impl IpScheme {
|
||||
pub fn new() -> IpScheme {
|
||||
IpScheme {
|
||||
arp: RefCell::new(Vec::new())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl ResourceScheme<IpResource> for IpScheme {
|
||||
fn open_resource(&self, url: &[u8], _flags: usize, uid: u32, _gid: u32) -> Result<Box<IpResource>> {
|
||||
if uid == 0 {
|
||||
let path = try!(str::from_utf8(url).or(Err(Error::new(EINVAL))));
|
||||
let mut parts = path.split('/');
|
||||
if let Some(host_string) = parts.next() {
|
||||
if let Some(proto_string) = parts.next() {
|
||||
let proto = u8::from_str_radix(proto_string, 16).unwrap_or(0);
|
||||
|
||||
if ! host_string.is_empty() {
|
||||
let peer_addr = Ipv4Addr::from_str(host_string);
|
||||
let mut route_mac = BROADCAST_MAC_ADDR;
|
||||
|
||||
if ! peer_addr.equals(BROADCAST_IP_ADDR) {
|
||||
let mut needs_routing = false;
|
||||
|
||||
for octet in 0..4 {
|
||||
let me = unsafe { IP_ADDR.bytes[octet] };
|
||||
let mask = unsafe { IP_SUBNET.bytes[octet] };
|
||||
let them = peer_addr.bytes[octet];
|
||||
if me & mask != them & mask {
|
||||
needs_routing = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
let route_addr = if needs_routing {
|
||||
unsafe { IP_ROUTER_ADDR }
|
||||
} else {
|
||||
peer_addr
|
||||
};
|
||||
|
||||
for entry in self.arp.borrow().iter() {
|
||||
if entry.ip.equals(route_addr) {
|
||||
route_mac = entry.mac;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if route_mac.equals(BROADCAST_MAC_ADDR) {
|
||||
if let Ok(link) = syscall::open(&format!("ethernet:{}/806", &route_mac.to_string()), O_RDWR) {
|
||||
let arp = Arp {
|
||||
header: ArpHeader {
|
||||
htype: n16::new(1),
|
||||
ptype: n16::new(0x800),
|
||||
hlen: 6,
|
||||
plen: 4,
|
||||
oper: n16::new(1),
|
||||
src_mac: unsafe { MAC_ADDR },
|
||||
src_ip: unsafe { IP_ADDR },
|
||||
dst_mac: route_mac,
|
||||
dst_ip: route_addr,
|
||||
},
|
||||
data: Vec::new(),
|
||||
};
|
||||
|
||||
match syscall::write(link, &arp.to_bytes()) {
|
||||
Ok(_) => loop {
|
||||
let mut bytes = [0; 65536];
|
||||
match syscall::read(link, &mut bytes) {
|
||||
Ok(count) => if let Some(packet) = Arp::from_bytes(&bytes[..count]) {
|
||||
if packet.header.oper.get() == 2 &&
|
||||
packet.header.src_ip.equals(route_addr) {
|
||||
route_mac = packet.header.src_mac;
|
||||
self.arp.borrow_mut().push(ArpEntry {
|
||||
ip: route_addr,
|
||||
mac: route_mac,
|
||||
});
|
||||
break;
|
||||
}
|
||||
},
|
||||
Err(_) => (),
|
||||
}
|
||||
},
|
||||
Err(err) => println!("IP: ARP Write Failed: {}", err),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if let Ok(link) = syscall::open(&format!("ethernet:{}/800", &route_mac.to_string()), O_RDWR) {
|
||||
return Ok(Box::new(IpResource {
|
||||
link: link,
|
||||
data: Vec::new(),
|
||||
peer_addr: peer_addr,
|
||||
proto: proto,
|
||||
id: (rand() % 65536) as u16,
|
||||
}));
|
||||
}
|
||||
} else {
|
||||
while let Ok(link) = syscall::open("ethernet:/800", O_RDWR) {
|
||||
let mut bytes = [0; 65536];
|
||||
match syscall::read(link, &mut bytes) {
|
||||
Ok(count) => {
|
||||
if let Some(packet) = Ipv4::from_bytes(&bytes[..count]) {
|
||||
if packet.header.proto == proto &&
|
||||
(packet.header.dst.equals(unsafe { IP_ADDR }) || packet.header.dst.equals(BROADCAST_IP_ADDR)) {
|
||||
return Ok(Box::new(IpResource {
|
||||
link: link,
|
||||
data: packet.data,
|
||||
peer_addr: packet.header.src,
|
||||
proto: proto,
|
||||
id: (rand() % 65536) as u16,
|
||||
}));
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(_) => break,
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
println!("IP: No protocol provided");
|
||||
}
|
||||
} else {
|
||||
println!("IP: No host provided");
|
||||
}
|
||||
|
||||
Err(Error::new(ENOENT))
|
||||
} else {
|
||||
Err(Error::new(EACCES))
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue