Add runtime tools and Red Bear service wiring
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
@@ -2,10 +2,13 @@ mod device_db;
|
||||
mod scheme;
|
||||
|
||||
use std::env;
|
||||
use std::process;
|
||||
use std::os::fd::{AsRawFd, FromRawFd, RawFd};
|
||||
|
||||
use log::{error, info, LevelFilter, Metadata, Record};
|
||||
use redox_scheme::{SignalBehavior, Socket};
|
||||
use redox_scheme::{
|
||||
scheme::{SchemeState, SchemeSync},
|
||||
SignalBehavior, Socket,
|
||||
};
|
||||
|
||||
use scheme::UdevScheme;
|
||||
|
||||
@@ -25,45 +28,35 @@ impl log::Log for StderrLogger {
|
||||
fn flush(&self) {}
|
||||
}
|
||||
|
||||
fn run() -> Result<(), String> {
|
||||
let mut scheme = UdevScheme::new();
|
||||
|
||||
match scheme.scan_pci_devices() {
|
||||
Ok(n) => info!("udev-shim: enumerated {} PCI device(s)", n),
|
||||
Err(e) => error!("udev-shim: PCI scan failed: {}", e),
|
||||
fn init_logging(level: LevelFilter) {
|
||||
if log::set_boxed_logger(Box::new(StderrLogger { level })).is_err() {
|
||||
return;
|
||||
}
|
||||
log::set_max_level(level);
|
||||
}
|
||||
|
||||
let socket =
|
||||
Socket::create("udev").map_err(|e| format!("failed to register udev scheme: {}", e))?;
|
||||
info!("udev-shim: registered scheme:udev");
|
||||
unsafe fn get_init_notify_fd() -> RawFd {
|
||||
let fd: RawFd = env::var("INIT_NOTIFY")
|
||||
.expect("udev-shim: INIT_NOTIFY not set")
|
||||
.parse()
|
||||
.expect("udev-shim: INIT_NOTIFY is not a valid fd");
|
||||
libc::fcntl(fd, libc::F_SETFD, libc::FD_CLOEXEC);
|
||||
fd
|
||||
}
|
||||
|
||||
loop {
|
||||
let request = match socket.next_request(SignalBehavior::Restart) {
|
||||
Ok(Some(r)) => r,
|
||||
Ok(None) => {
|
||||
info!("udev-shim: scheme unmounted, exiting");
|
||||
break;
|
||||
}
|
||||
Err(e) => {
|
||||
error!("udev-shim: failed to read scheme request: {}", e);
|
||||
continue;
|
||||
}
|
||||
};
|
||||
fn notify_scheme_ready(notify_fd: RawFd, socket: &Socket, scheme: &mut UdevScheme) {
|
||||
let cap_id = scheme.scheme_root().expect("udev-shim: scheme_root failed");
|
||||
let cap_fd = socket
|
||||
.create_this_scheme_fd(0, cap_id, 0, 0)
|
||||
.expect("udev-shim: create_this_scheme_fd failed");
|
||||
|
||||
let response = match request.handle_scheme_block_mut(&mut scheme) {
|
||||
Ok(r) => r,
|
||||
Err(_req) => {
|
||||
error!("udev-shim: failed to handle request");
|
||||
continue;
|
||||
}
|
||||
};
|
||||
|
||||
if let Err(e) = socket.write_response(response, SignalBehavior::Restart) {
|
||||
error!("udev-shim: failed to write response: {}", e);
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
syscall::call_wo(
|
||||
notify_fd as usize,
|
||||
&libredox::Fd::new(cap_fd).into_raw().to_ne_bytes(),
|
||||
syscall::CallFlags::FD,
|
||||
&[],
|
||||
)
|
||||
.expect("udev-shim: failed to notify init that scheme is ready");
|
||||
}
|
||||
|
||||
fn main() {
|
||||
@@ -72,11 +65,40 @@ fn main() {
|
||||
Ok("trace") => LevelFilter::Trace,
|
||||
_ => LevelFilter::Info,
|
||||
};
|
||||
let _ = log::set_boxed_logger(Box::new(StderrLogger { level: log_level }));
|
||||
log::set_max_level(log_level);
|
||||
|
||||
if let Err(e) = run() {
|
||||
error!("udev-shim: fatal error: {}", e);
|
||||
process::exit(1);
|
||||
init_logging(log_level);
|
||||
|
||||
let mut scheme = UdevScheme::new();
|
||||
|
||||
match scheme.scan_pci_devices() {
|
||||
Ok(n) => info!("udev-shim: enumerated {} PCI device(s)", n),
|
||||
Err(e) => error!("udev-shim: PCI scan failed: {}", e),
|
||||
}
|
||||
|
||||
let notify_fd = unsafe { get_init_notify_fd() };
|
||||
let socket = Socket::create().expect("udev-shim: failed to create udev scheme");
|
||||
let mut state = SchemeState::new();
|
||||
|
||||
notify_scheme_ready(notify_fd, &socket, &mut scheme);
|
||||
|
||||
libredox::call::setrens(0, 0).expect("udev-shim: failed to enter null namespace");
|
||||
|
||||
info!("udev-shim: registered scheme:udev");
|
||||
|
||||
while let Some(request) = socket
|
||||
.next_request(SignalBehavior::Restart)
|
||||
.expect("udev-shim: failed to read scheme request")
|
||||
{
|
||||
match request.kind() {
|
||||
redox_scheme::RequestKind::Call(request) => {
|
||||
let response = request.handle_sync(&mut scheme, &mut state);
|
||||
socket
|
||||
.write_response(response, SignalBehavior::Restart)
|
||||
.expect("udev-shim: failed to write response");
|
||||
}
|
||||
_ => (),
|
||||
}
|
||||
}
|
||||
|
||||
std::process::exit(0);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user