Significant improvements for events - switch to event queue in orbital

This commit is contained in:
Jeremy Soller 2016-10-14 20:12:21 -06:00
parent 6d19f59c08
commit 77810d6713
9 changed files with 69 additions and 42 deletions

View file

@ -6,10 +6,10 @@ extern crate alloc;
extern crate orbclient; extern crate orbclient;
extern crate syscall; extern crate syscall;
use std::{env, thread}; use std::{env, mem, thread};
use std::fs::File; use std::fs::File;
use std::io::{Read, Write}; use std::io::{Read, Write};
use syscall::{physmap, physunmap, Packet, Scheme, MAP_WRITE, MAP_WRITE_COMBINE}; use syscall::{physmap, physunmap, Packet, Scheme, EVENT_READ, MAP_WRITE, MAP_WRITE_COMBINE};
use mode_info::VBEModeInfo; use mode_info::VBEModeInfo;
use primitive::fast_set64; use primitive::fast_set64;
@ -87,22 +87,22 @@ fn main() {
} }
} }
// If there are requested events, and data is available, send a notification for (screen_id, screen) in scheme.screens.borrow().iter() {
/* TODO if ! screen.will_block() {
if (! scheme.screen.borrow().input.is_empty() || scheme.screen.borrow().end_of_input) && scheme.screen.borrow().requested & EVENT_READ == EVENT_READ { let event_packet = Packet {
let event_packet = Packet { id: 0,
id: 0, pid: 0,
pid: 0, uid: 0,
uid: 0, gid: 0,
gid: 0, a: syscall::number::SYS_FEVENT,
a: syscall::number::SYS_FEVENT, b: *screen_id,
b: 0, c: EVENT_READ,
c: EVENT_READ, d: mem::size_of::<Packet>()
d: scheme.screen.borrow().input.len() };
};
socket.write(&event_packet).expect("vesad: failed to write display scheme"); socket.write(&event_packet).expect("vesad: failed to write display event");
}
} }
*/
} }
}); });
} }

View file

@ -14,7 +14,7 @@ pub struct DisplayScheme {
onscreen: usize, onscreen: usize,
active: Cell<usize>, active: Cell<usize>,
next_screen: Cell<usize>, next_screen: Cell<usize>,
screens: RefCell<BTreeMap<usize, Box<Screen>>> pub screens: RefCell<BTreeMap<usize, Box<Screen>>>
} }
impl DisplayScheme { impl DisplayScheme {
@ -77,7 +77,7 @@ impl Scheme for DisplayScheme {
fn fevent(&self, id: usize, flags: usize) -> Result<usize> { fn fevent(&self, id: usize, flags: usize) -> Result<usize> {
let mut screens = self.screens.borrow_mut(); let mut screens = self.screens.borrow_mut();
if let Some(mut screen) = screens.get_mut(&id) { if let Some(mut screen) = screens.get_mut(&id) {
screen.event(flags) screen.event(flags).and(Ok(id))
} else { } else {
Err(Error::new(EBADF)) Err(Error::new(EBADF))
} }

View file

@ -27,7 +27,7 @@ pub fn registry_mut() -> RwLockWriteGuard<'static, Registry> {
REGISTRY.call_once(init_registry).write() REGISTRY.call_once(init_registry).write()
} }
pub fn register(fd: usize, scheme_id: usize, id: usize) -> bool { pub fn register(fd: usize, scheme_id: usize, event_id: usize) -> bool {
let (context_id, events) = { let (context_id, events) = {
let contexts = context::contexts(); let contexts = context::contexts();
let context_lock = contexts.current().expect("event::register: No context"); let context_lock = contexts.current().expect("event::register: No context");
@ -36,7 +36,7 @@ pub fn register(fd: usize, scheme_id: usize, id: usize) -> bool {
}; };
let mut registry = registry_mut(); let mut registry = registry_mut();
let entry = registry.entry((scheme_id, id)).or_insert_with(|| { let entry = registry.entry((scheme_id, event_id)).or_insert_with(|| {
BTreeMap::new() BTreeMap::new()
}); });
if entry.contains_key(&(context_id, fd)) { if entry.contains_key(&(context_id, fd)) {
@ -47,11 +47,11 @@ pub fn register(fd: usize, scheme_id: usize, id: usize) -> bool {
} }
} }
pub fn unregister(fd: usize, scheme_id: usize, id: usize) { pub fn unregister(fd: usize, scheme_id: usize, event_id: usize) {
let mut registry = registry_mut(); let mut registry = registry_mut();
let mut remove = false; let mut remove = false;
if let Some(entry) = registry.get_mut(&(scheme_id, id)) { if let Some(entry) = registry.get_mut(&(scheme_id, event_id)) {
entry.remove(&(context::context_id(), fd)); entry.remove(&(context::context_id(), fd));
if entry.is_empty() { if entry.is_empty() {
@ -60,13 +60,13 @@ pub fn unregister(fd: usize, scheme_id: usize, id: usize) {
} }
if remove { if remove {
registry.remove(&(scheme_id, id)); registry.remove(&(scheme_id, event_id));
} }
} }
pub fn trigger(scheme_id: usize, id: usize, flags: usize, data: usize) { pub fn trigger(scheme_id: usize, event_id: usize, flags: usize, data: usize) {
let registry = registry(); let registry = registry();
if let Some(event_lists) = registry.get(&(scheme_id, id)) { if let Some(event_lists) = registry.get(&(scheme_id, event_id)) {
for entry in event_lists.iter() { for entry in event_lists.iter() {
if let Some(event_list) = entry.1.upgrade() { if let Some(event_list) = entry.1.upgrade() {
event_list.send(Event { event_list.send(Event {

View file

@ -8,4 +8,6 @@ pub struct File {
pub scheme: usize, pub scheme: usize,
/// The number the scheme uses to refer to this file /// The number the scheme uses to refer to this file
pub number: usize, pub number: usize,
/// If events are on, this is the event ID
pub event: Option<usize>,
} }

View file

@ -93,8 +93,8 @@ impl Scheme for IrqScheme {
} }
} }
fn fevent(&self, _file: usize, _flags: usize) -> Result<usize> { fn fevent(&self, file: usize, _flags: usize) -> Result<usize> {
Ok(0) Ok(file)
} }
fn fsync(&self, _file: usize) -> Result<usize> { fn fsync(&self, _file: usize) -> Result<usize> {

View file

@ -89,8 +89,8 @@ impl Scheme for RootScheme {
inner.write(buf) inner.write(buf)
} }
fn fevent(&self, _file: usize, _flags: usize) -> Result<usize> { fn fevent(&self, file: usize, _flags: usize) -> Result<usize> {
Ok(0) Ok(file)
} }
fn fsync(&self, _file: usize) -> Result<usize> { fn fsync(&self, _file: usize) -> Result<usize> {

View file

@ -98,7 +98,8 @@ pub fn open(path: &[u8], flags: usize) -> Result<usize> {
let context = context_lock.read(); let context = context_lock.read();
context.add_file(::context::file::File { context.add_file(::context::file::File {
scheme: scheme_id, scheme: scheme_id,
number: file_id number: file_id,
event: None,
}).ok_or(Error::new(EMFILE)) }).ok_or(Error::new(EMFILE))
} }
@ -113,12 +114,14 @@ pub fn pipe2(fds: &mut [usize], flags: usize) -> Result<usize> {
let read_fd = context.add_file(::context::file::File { let read_fd = context.add_file(::context::file::File {
scheme: scheme_id, scheme: scheme_id,
number: read_id number: read_id,
event: None,
}).ok_or(Error::new(EMFILE))?; }).ok_or(Error::new(EMFILE))?;
let write_fd = context.add_file(::context::file::File { let write_fd = context.add_file(::context::file::File {
scheme: scheme_id, scheme: scheme_id,
number: write_id number: write_id,
event: None,
}).ok_or(Error::new(EMFILE))?; }).ok_or(Error::new(EMFILE))?;
fds[0] = read_fd; fds[0] = read_fd;
@ -206,7 +209,9 @@ pub fn close(fd: usize) -> Result<usize> {
file file
}; };
context::event::unregister(fd, file.scheme, file.number); if let Some(event_id) = file.event {
context::event::unregister(fd, file.scheme, event_id);
}
let scheme = { let scheme = {
let schemes = scheme::schemes(); let schemes = scheme::schemes();
@ -240,7 +245,8 @@ pub fn dup(fd: usize) -> Result<usize> {
let context = context_lock.read(); let context = context_lock.read();
context.add_file(::context::file::File { context.add_file(::context::file::File {
scheme: file.scheme, scheme: file.scheme,
number: new_id number: new_id,
event: None,
}).ok_or(Error::new(EMFILE)) }).ok_or(Error::new(EMFILE))
} }
@ -250,8 +256,13 @@ pub fn fevent(fd: usize, flags: usize) -> Result<usize> {
let contexts = context::contexts(); let contexts = context::contexts();
let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; let context_lock = contexts.current().ok_or(Error::new(ESRCH))?;
let context = context_lock.read(); let context = context_lock.read();
let file = context.get_file(fd).ok_or(Error::new(EBADF))?; let mut files = context.files.lock();
file let mut file = files.get_mut(fd).ok_or(Error::new(EBADF))?.ok_or(Error::new(EBADF))?;
if let Some(event_id) = file.event.take() {
println!("{}: {}:{}: events already registered: {}", fd, file.scheme, file.number, event_id);
context::event::unregister(fd, file.scheme, event_id);
}
file.clone()
}; };
let scheme = { let scheme = {
@ -259,7 +270,15 @@ pub fn fevent(fd: usize, flags: usize) -> Result<usize> {
let scheme = schemes.get(file.scheme).ok_or(Error::new(EBADF))?; let scheme = schemes.get(file.scheme).ok_or(Error::new(EBADF))?;
scheme.clone() scheme.clone()
}; };
scheme.fevent(file.number, flags)?; let event_id = scheme.fevent(file.number, flags)?;
context::event::register(fd, file.scheme, file.number); {
let contexts = context::contexts();
let context_lock = contexts.current().ok_or(Error::new(ESRCH))?;
let context = context_lock.read();
let mut files = context.files.lock();
let mut file = files.get_mut(fd).ok_or(Error::new(EBADF))?.ok_or(Error::new(EBADF))?;
file.event = Some(event_id);
}
context::event::register(fd, file.scheme, event_id);
Ok(0) Ok(0)
} }

View file

@ -232,7 +232,11 @@ pub fn clone(flags: usize, stack_base: usize) -> Result<usize> {
}; };
match result { match result {
Ok(new_number) => { Ok(new_number) => {
Some(context::file::File { scheme: file.scheme, number: new_number }) Some(context::file::File {
scheme: file.scheme,
number: new_number,
event: None,
})
}, },
Err(err) => { Err(err) => {
println!("clone: failed to dup {}: {:?}", fd, err); println!("clone: failed to dup {}: {:?}", fd, err);
@ -616,7 +620,9 @@ pub fn exit(status: usize) -> ! {
/// Files must be closed while context is valid so that messages can be passed /// Files must be closed while context is valid so that messages can be passed
for (fd, file_option) in close_files.drain(..).enumerate() { for (fd, file_option) in close_files.drain(..).enumerate() {
if let Some(file) = file_option { if let Some(file) = file_option {
context::event::unregister(fd, file.scheme, file.number); if let Some(event_id) = file.event {
context::event::unregister(fd, file.scheme, event_id);
}
let scheme_option = { let scheme_option = {
let schemes = scheme::schemes(); let schemes = scheme::schemes();

@ -1 +1 @@
Subproject commit 3667b8960255928363b88ac9ac5902f228823512 Subproject commit c21bd7618907c5d146c947e9b38d64fdb7479528