Significant improvements for events - switch to event queue in orbital
This commit is contained in:
parent
6d19f59c08
commit
77810d6713
|
@ -6,10 +6,10 @@ extern crate alloc;
|
|||
extern crate orbclient;
|
||||
extern crate syscall;
|
||||
|
||||
use std::{env, thread};
|
||||
use std::{env, mem, thread};
|
||||
use std::fs::File;
|
||||
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 primitive::fast_set64;
|
||||
|
@ -87,22 +87,22 @@ fn main() {
|
|||
}
|
||||
}
|
||||
|
||||
// If there are requested events, and data is available, send a notification
|
||||
/* TODO
|
||||
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 {
|
||||
id: 0,
|
||||
pid: 0,
|
||||
uid: 0,
|
||||
gid: 0,
|
||||
a: syscall::number::SYS_FEVENT,
|
||||
b: 0,
|
||||
c: EVENT_READ,
|
||||
d: scheme.screen.borrow().input.len()
|
||||
};
|
||||
socket.write(&event_packet).expect("vesad: failed to write display scheme");
|
||||
for (screen_id, screen) in scheme.screens.borrow().iter() {
|
||||
if ! screen.will_block() {
|
||||
let event_packet = Packet {
|
||||
id: 0,
|
||||
pid: 0,
|
||||
uid: 0,
|
||||
gid: 0,
|
||||
a: syscall::number::SYS_FEVENT,
|
||||
b: *screen_id,
|
||||
c: EVENT_READ,
|
||||
d: mem::size_of::<Packet>()
|
||||
};
|
||||
|
||||
socket.write(&event_packet).expect("vesad: failed to write display event");
|
||||
}
|
||||
}
|
||||
*/
|
||||
}
|
||||
});
|
||||
}
|
||||
|
|
|
@ -14,7 +14,7 @@ pub struct DisplayScheme {
|
|||
onscreen: usize,
|
||||
active: Cell<usize>,
|
||||
next_screen: Cell<usize>,
|
||||
screens: RefCell<BTreeMap<usize, Box<Screen>>>
|
||||
pub screens: RefCell<BTreeMap<usize, Box<Screen>>>
|
||||
}
|
||||
|
||||
impl DisplayScheme {
|
||||
|
@ -77,7 +77,7 @@ impl Scheme for DisplayScheme {
|
|||
fn fevent(&self, id: usize, flags: usize) -> Result<usize> {
|
||||
let mut screens = self.screens.borrow_mut();
|
||||
if let Some(mut screen) = screens.get_mut(&id) {
|
||||
screen.event(flags)
|
||||
screen.event(flags).and(Ok(id))
|
||||
} else {
|
||||
Err(Error::new(EBADF))
|
||||
}
|
||||
|
|
|
@ -27,7 +27,7 @@ pub fn registry_mut() -> RwLockWriteGuard<'static, Registry> {
|
|||
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 contexts = context::contexts();
|
||||
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 entry = registry.entry((scheme_id, id)).or_insert_with(|| {
|
||||
let entry = registry.entry((scheme_id, event_id)).or_insert_with(|| {
|
||||
BTreeMap::new()
|
||||
});
|
||||
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 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));
|
||||
|
||||
if entry.is_empty() {
|
||||
|
@ -60,13 +60,13 @@ pub fn unregister(fd: usize, scheme_id: usize, id: usize) {
|
|||
}
|
||||
|
||||
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();
|
||||
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() {
|
||||
if let Some(event_list) = entry.1.upgrade() {
|
||||
event_list.send(Event {
|
||||
|
|
|
@ -8,4 +8,6 @@ pub struct File {
|
|||
pub scheme: usize,
|
||||
/// The number the scheme uses to refer to this file
|
||||
pub number: usize,
|
||||
/// If events are on, this is the event ID
|
||||
pub event: Option<usize>,
|
||||
}
|
||||
|
|
|
@ -93,8 +93,8 @@ impl Scheme for IrqScheme {
|
|||
}
|
||||
}
|
||||
|
||||
fn fevent(&self, _file: usize, _flags: usize) -> Result<usize> {
|
||||
Ok(0)
|
||||
fn fevent(&self, file: usize, _flags: usize) -> Result<usize> {
|
||||
Ok(file)
|
||||
}
|
||||
|
||||
fn fsync(&self, _file: usize) -> Result<usize> {
|
||||
|
|
|
@ -89,8 +89,8 @@ impl Scheme for RootScheme {
|
|||
inner.write(buf)
|
||||
}
|
||||
|
||||
fn fevent(&self, _file: usize, _flags: usize) -> Result<usize> {
|
||||
Ok(0)
|
||||
fn fevent(&self, file: usize, _flags: usize) -> Result<usize> {
|
||||
Ok(file)
|
||||
}
|
||||
|
||||
fn fsync(&self, _file: usize) -> Result<usize> {
|
||||
|
|
|
@ -98,7 +98,8 @@ pub fn open(path: &[u8], flags: usize) -> Result<usize> {
|
|||
let context = context_lock.read();
|
||||
context.add_file(::context::file::File {
|
||||
scheme: scheme_id,
|
||||
number: file_id
|
||||
number: file_id,
|
||||
event: None,
|
||||
}).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 {
|
||||
scheme: scheme_id,
|
||||
number: read_id
|
||||
number: read_id,
|
||||
event: None,
|
||||
}).ok_or(Error::new(EMFILE))?;
|
||||
|
||||
let write_fd = context.add_file(::context::file::File {
|
||||
scheme: scheme_id,
|
||||
number: write_id
|
||||
number: write_id,
|
||||
event: None,
|
||||
}).ok_or(Error::new(EMFILE))?;
|
||||
|
||||
fds[0] = read_fd;
|
||||
|
@ -206,7 +209,9 @@ pub fn close(fd: usize) -> Result<usize> {
|
|||
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 schemes = scheme::schemes();
|
||||
|
@ -240,7 +245,8 @@ pub fn dup(fd: usize) -> Result<usize> {
|
|||
let context = context_lock.read();
|
||||
context.add_file(::context::file::File {
|
||||
scheme: file.scheme,
|
||||
number: new_id
|
||||
number: new_id,
|
||||
event: None,
|
||||
}).ok_or(Error::new(EMFILE))
|
||||
}
|
||||
|
||||
|
@ -250,8 +256,13 @@ pub fn fevent(fd: usize, flags: usize) -> Result<usize> {
|
|||
let contexts = context::contexts();
|
||||
let context_lock = contexts.current().ok_or(Error::new(ESRCH))?;
|
||||
let context = context_lock.read();
|
||||
let file = context.get_file(fd).ok_or(Error::new(EBADF))?;
|
||||
file
|
||||
let mut files = context.files.lock();
|
||||
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 = {
|
||||
|
@ -259,7 +270,15 @@ pub fn fevent(fd: usize, flags: usize) -> Result<usize> {
|
|||
let scheme = schemes.get(file.scheme).ok_or(Error::new(EBADF))?;
|
||||
scheme.clone()
|
||||
};
|
||||
scheme.fevent(file.number, flags)?;
|
||||
context::event::register(fd, file.scheme, file.number);
|
||||
let event_id = scheme.fevent(file.number, flags)?;
|
||||
{
|
||||
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)
|
||||
}
|
||||
|
|
|
@ -232,7 +232,11 @@ pub fn clone(flags: usize, stack_base: usize) -> Result<usize> {
|
|||
};
|
||||
match result {
|
||||
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) => {
|
||||
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
|
||||
for (fd, file_option) in close_files.drain(..).enumerate() {
|
||||
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 schemes = scheme::schemes();
|
||||
|
|
|
@ -1 +1 @@
|
|||
Subproject commit 3667b8960255928363b88ac9ac5902f228823512
|
||||
Subproject commit c21bd7618907c5d146c947e9b38d64fdb7479528
|
Loading…
Reference in a new issue