mirror of
https://gitlab.com/fabinfra/fabaccess/bffh.git
synced 2024-11-22 14:57:56 +01:00
Merge branch 'feature/new_initiator' into development
* feature/new_initiator: Process initiator working Reimplement the dummy initiator
This commit is contained in:
commit
728c33f444
23
Cargo.lock
generated
23
Cargo.lock
generated
@ -171,9 +171,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "async-io"
|
name = "async-io"
|
||||||
version = "1.6.0"
|
version = "1.7.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "a811e6a479f2439f0c04038796b5cfb3d2ad56c230e0f2d3f7b04d68cfee607b"
|
checksum = "e5e18f61464ae81cde0a23e713ae8fd299580c54d697a35820cfd0625b8b0e07"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"concurrent-queue",
|
"concurrent-queue",
|
||||||
"futures-lite",
|
"futures-lite",
|
||||||
@ -226,6 +226,23 @@ dependencies = [
|
|||||||
"futures-micro",
|
"futures-micro",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "async-process"
|
||||||
|
version = "1.4.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "cf2c06e30a24e8c78a3987d07f0930edf76ef35e027e7bdb063fccafdad1f60c"
|
||||||
|
dependencies = [
|
||||||
|
"async-io",
|
||||||
|
"blocking",
|
||||||
|
"cfg-if",
|
||||||
|
"event-listener",
|
||||||
|
"futures-lite",
|
||||||
|
"libc",
|
||||||
|
"once_cell",
|
||||||
|
"signal-hook",
|
||||||
|
"winapi",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "async-std"
|
name = "async-std"
|
||||||
version = "1.10.0"
|
version = "1.10.0"
|
||||||
@ -836,8 +853,10 @@ dependencies = [
|
|||||||
"api",
|
"api",
|
||||||
"async-channel",
|
"async-channel",
|
||||||
"async-compat",
|
"async-compat",
|
||||||
|
"async-io",
|
||||||
"async-net",
|
"async-net",
|
||||||
"async-oneshot",
|
"async-oneshot",
|
||||||
|
"async-process",
|
||||||
"async-trait",
|
"async-trait",
|
||||||
"backtrace",
|
"backtrace",
|
||||||
"capnp",
|
"capnp",
|
||||||
|
@ -37,6 +37,8 @@ pin-utils = "0.1.0"
|
|||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
futures-lite = "1.12.0"
|
futures-lite = "1.12.0"
|
||||||
async-net = "1.6.1"
|
async-net = "1.6.1"
|
||||||
|
async-io = "1.7.0"
|
||||||
|
async-process = "1.4.0"
|
||||||
backtrace = "0.3.65"
|
backtrace = "0.3.65"
|
||||||
miette = { version = "4.7.1", features = ["fancy"] }
|
miette = { version = "4.7.1", features = ["fancy"] }
|
||||||
thiserror = "1.0.31"
|
thiserror = "1.0.31"
|
||||||
|
118
bffhd/initiators/dummy.rs
Normal file
118
bffhd/initiators/dummy.rs
Normal file
@ -0,0 +1,118 @@
|
|||||||
|
use miette::{miette, Diagnostic};
|
||||||
|
use thiserror::Error;
|
||||||
|
|
||||||
|
use super::Initiator;
|
||||||
|
use crate::initiators::InitiatorCallbacks;
|
||||||
|
use crate::resources::modules::fabaccess::Status;
|
||||||
|
use crate::session::SessionHandle;
|
||||||
|
use crate::users::UserRef;
|
||||||
|
use async_io::Timer;
|
||||||
|
use futures_util::future::BoxFuture;
|
||||||
|
use futures_util::ready;
|
||||||
|
use lmdb::Stat;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::future::Future;
|
||||||
|
use std::mem;
|
||||||
|
use std::pin::Pin;
|
||||||
|
use std::task::{Context, Poll};
|
||||||
|
use std::time::{Duration, Instant};
|
||||||
|
|
||||||
|
pub struct Dummy {
|
||||||
|
callbacks: InitiatorCallbacks,
|
||||||
|
session: SessionHandle,
|
||||||
|
state: DummyState,
|
||||||
|
}
|
||||||
|
|
||||||
|
enum DummyState {
|
||||||
|
Empty,
|
||||||
|
Sleeping(Timer, Option<Status>),
|
||||||
|
Updating(BoxFuture<'static, Status>),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Dummy {
|
||||||
|
fn timer() -> Timer {
|
||||||
|
Timer::after(Duration::from_secs(2))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn flip(&self, status: Status) -> BoxFuture<'static, Status> {
|
||||||
|
let session = self.session.clone();
|
||||||
|
let mut callbacks = self.callbacks.clone();
|
||||||
|
Box::pin(async move {
|
||||||
|
let next = match &status {
|
||||||
|
Status::Free => Status::InUse(session.get_user_ref()),
|
||||||
|
Status::InUse(_) => Status::Free,
|
||||||
|
_ => Status::Free,
|
||||||
|
};
|
||||||
|
callbacks.try_update(session, status).await;
|
||||||
|
|
||||||
|
next
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Error, Diagnostic)]
|
||||||
|
pub enum DummyError {}
|
||||||
|
|
||||||
|
impl Future for Dummy {
|
||||||
|
type Output = ();
|
||||||
|
|
||||||
|
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
|
let span = tracing::debug_span!("Dummy initiator poll");
|
||||||
|
let _guard = span.enter();
|
||||||
|
tracing::trace!("polling Dummy initiator");
|
||||||
|
loop {
|
||||||
|
match &mut self.state {
|
||||||
|
DummyState::Empty => {
|
||||||
|
tracing::trace!("Dummy initiator is empty, initializing…");
|
||||||
|
mem::replace(
|
||||||
|
&mut self.state,
|
||||||
|
DummyState::Sleeping(Self::timer(), Some(Status::Free)),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
DummyState::Sleeping(timer, next) => {
|
||||||
|
tracing::trace!("Sleep timer exists, polling it.");
|
||||||
|
|
||||||
|
let _: Instant = ready!(Pin::new(timer).poll(cx));
|
||||||
|
|
||||||
|
tracing::trace!("Timer has fired, poking out an update!");
|
||||||
|
|
||||||
|
let status = next.take().unwrap();
|
||||||
|
let f = self.flip(status);
|
||||||
|
mem::replace(&mut self.state, DummyState::Updating(f));
|
||||||
|
}
|
||||||
|
DummyState::Updating(f) => {
|
||||||
|
tracing::trace!("Update future exists, polling it .");
|
||||||
|
|
||||||
|
let next = ready!(Pin::new(f).poll(cx));
|
||||||
|
|
||||||
|
tracing::trace!("Update future completed, sleeping!");
|
||||||
|
|
||||||
|
mem::replace(
|
||||||
|
&mut self.state,
|
||||||
|
DummyState::Sleeping(Self::timer(), Some(next)),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Initiator for Dummy {
|
||||||
|
fn new(params: &HashMap<String, String>, callbacks: InitiatorCallbacks) -> miette::Result<Self>
|
||||||
|
where
|
||||||
|
Self: Sized,
|
||||||
|
{
|
||||||
|
let uid = params
|
||||||
|
.get("uid")
|
||||||
|
.ok_or_else(|| miette!("Dummy initiator configured without an UID"))?;
|
||||||
|
let session = callbacks
|
||||||
|
.open_session(uid)
|
||||||
|
.ok_or_else(|| miette!("The configured user for the dummy initiator does not exist"))?;
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
callbacks,
|
||||||
|
session,
|
||||||
|
state: DummyState::Empty,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
@ -1,161 +1,157 @@
|
|||||||
|
use crate::initiators::dummy::Dummy;
|
||||||
|
use crate::initiators::process::Process;
|
||||||
|
use crate::resources::modules::fabaccess::Status;
|
||||||
|
use crate::session::SessionHandle;
|
||||||
|
use crate::{
|
||||||
|
AuthenticationHandle, Config, MachineState, Resource, ResourcesHandle, SessionManager,
|
||||||
|
};
|
||||||
|
use async_compat::CompatExt;
|
||||||
|
use executor::prelude::Executor;
|
||||||
|
use futures_util::ready;
|
||||||
|
use miette::IntoDiagnostic;
|
||||||
|
use rumqttc::ConnectReturnCode::Success;
|
||||||
|
use rumqttc::{AsyncClient, ConnectionError, Event, Incoming, MqttOptions};
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::fmt::Display;
|
||||||
use std::future::Future;
|
use std::future::Future;
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
use std::task::{Context, Poll};
|
use std::task::{Context, Poll};
|
||||||
use async_channel as channel;
|
use std::time::Duration;
|
||||||
use async_oneshot as oneshot;
|
use url::Url;
|
||||||
use futures_signals::signal::Signal;
|
|
||||||
use futures_util::future::BoxFuture;
|
|
||||||
use crate::resources::claim::{ResourceID, UserID};
|
|
||||||
use crate::resources::state::State;
|
|
||||||
|
|
||||||
pub enum UpdateError {
|
mod dummy;
|
||||||
/// We're not connected to anything anymore. You can't do anything about this error and the
|
mod process;
|
||||||
/// only reason why you even get it is because your future was called a last time before
|
|
||||||
/// being shelved so best way to handle this error is to just return from your loop entirely,
|
|
||||||
/// cleaning up any state that doesn't survive a freeze.
|
|
||||||
Closed,
|
|
||||||
|
|
||||||
Denied,
|
pub trait Initiator: Future<Output = ()> {
|
||||||
|
fn new(params: &HashMap<String, String>, callbacks: InitiatorCallbacks) -> miette::Result<Self>
|
||||||
Other(Box<dyn std::error::Error + Send>),
|
where
|
||||||
}
|
Self: Sized;
|
||||||
|
fn poll(self: Pin<&mut Self>, cx: &mut Context) -> Poll<()> {
|
||||||
pub trait InitiatorError: std::error::Error + Send {
|
<Self as Future>::poll(self, cx)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait Initiator {
|
|
||||||
fn start_for(&mut self, machine: ResourceID)
|
|
||||||
-> BoxFuture<'static, Result<(), Box<dyn InitiatorError>>>;
|
|
||||||
|
|
||||||
fn run(&mut self, request: &mut UpdateSink)
|
|
||||||
-> BoxFuture<'static, Result<(), Box<dyn InitiatorError>>>;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct UpdateSink {
|
pub struct InitiatorCallbacks {
|
||||||
tx: channel::Sender<(Option<UserID>, State)>,
|
resource: Resource,
|
||||||
rx: channel::Receiver<Result<(), Error>>,
|
sessions: SessionManager,
|
||||||
|
}
|
||||||
|
impl InitiatorCallbacks {
|
||||||
|
pub fn new(resource: Resource, sessions: SessionManager) -> Self {
|
||||||
|
Self { resource, sessions }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn try_update(&mut self, session: SessionHandle, status: Status) {
|
||||||
|
self.resource.try_update(session, status).await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn open_session(&self, uid: &str) -> Option<SessionHandle> {
|
||||||
|
self.sessions.open(uid)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl UpdateSink {
|
pub struct InitiatorDriver {
|
||||||
fn new(tx: channel::Sender<(Option<UserID>, State)>,
|
name: String,
|
||||||
rx: channel::Receiver<Result<(), Error>>)
|
initiator: Box<dyn Initiator + Unpin + Send>,
|
||||||
-> Self
|
}
|
||||||
|
|
||||||
|
impl InitiatorDriver {
|
||||||
|
pub fn new<I>(
|
||||||
|
name: String,
|
||||||
|
params: &HashMap<String, String>,
|
||||||
|
resource: Resource,
|
||||||
|
sessions: SessionManager,
|
||||||
|
) -> miette::Result<Self>
|
||||||
|
where
|
||||||
|
I: 'static + Initiator + Unpin + Send,
|
||||||
{
|
{
|
||||||
Self { tx, rx }
|
let callbacks = InitiatorCallbacks::new(resource, sessions);
|
||||||
}
|
let initiator = Box::new(I::new(params, callbacks)?);
|
||||||
|
Ok(Self { name, initiator })
|
||||||
async fn send(&mut self, userid: Option<UserID>, state: State)
|
|
||||||
-> Result<(), UpdateError>
|
|
||||||
{
|
|
||||||
if let Err(_e) = self.tx.send((userid, state)).await {
|
|
||||||
return Err(UpdateError::Closed);
|
|
||||||
}
|
|
||||||
|
|
||||||
match self.rx.recv().await {
|
|
||||||
Ok(Ok(())) => Ok(()),
|
|
||||||
Ok(Err(Error::Denied)) => Err(UpdateError::Denied),
|
|
||||||
Ok(Err(Error::Internal(e))) => Err(UpdateError::Other(e)),
|
|
||||||
// RecvError is send only when the channel is closed
|
|
||||||
Err(_) => Err(UpdateError::Closed),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
struct Resource;
|
impl Future for InitiatorDriver {
|
||||||
pub struct InitiatorDriver<S, I: Initiator> {
|
|
||||||
// TODO: make this a static reference to the resources because it's much easier and we don't
|
|
||||||
// need to replace resources at runtime at the moment.
|
|
||||||
resource_signal: S,
|
|
||||||
resource: Option<channel::Sender<Update>>,
|
|
||||||
|
|
||||||
// TODO: Initiators should instead
|
|
||||||
error_channel: Option<oneshot::Receiver<Error>>,
|
|
||||||
|
|
||||||
initiator: I,
|
|
||||||
initiator_future: Option<BoxFuture<'static, Result<(), Box<dyn InitiatorError>>>>,
|
|
||||||
update_sink: UpdateSink,
|
|
||||||
initiator_req_rx: channel::Receiver<(Option<UserID>, State)>,
|
|
||||||
initiator_reply_tx: channel::Sender<Result<(), Error>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct ResourceSink {
|
|
||||||
pub id: ResourceID,
|
|
||||||
pub state_sink: channel::Sender<Update>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<S: Signal<Item=ResourceSink>, I: Initiator> InitiatorDriver<S, I> {
|
|
||||||
pub fn new(resource_signal: S, initiator: I) -> Self {
|
|
||||||
let (initiator_reply_tx, initiator_reply_rx) = channel::bounded(1);
|
|
||||||
let (initiator_req_tx, initiator_req_rx) = async_channel::bounded(1);
|
|
||||||
let update_sink = UpdateSink::new(initiator_req_tx, initiator_reply_rx);
|
|
||||||
Self {
|
|
||||||
resource: None,
|
|
||||||
resource_signal,
|
|
||||||
error_channel: None,
|
|
||||||
|
|
||||||
initiator,
|
|
||||||
initiator_future: None,
|
|
||||||
update_sink,
|
|
||||||
initiator_req_rx,
|
|
||||||
initiator_reply_tx,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<S: Signal<Item=ResourceSink> + Unpin, I: Initiator + Unpin> Future for InitiatorDriver<S, I> {
|
|
||||||
type Output = ();
|
type Output = ();
|
||||||
|
|
||||||
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
match Pin::new(&mut self.resource_signal).poll_change(cx) {
|
let _guard = tracing::info_span!("initiator poll", initiator=%self.name);
|
||||||
Poll::Ready(Some(resource)) => {
|
tracing::trace!(initiator=%self.name, "polling initiator");
|
||||||
self.resource = Some(resource.state_sink);
|
|
||||||
self.error_channel = None;
|
|
||||||
let f = Box::pin(self.initiator.start_for(resource.id));
|
|
||||||
self.initiator_future.replace(f);
|
|
||||||
},
|
|
||||||
Poll::Ready(None) => self.resource = None,
|
|
||||||
Poll::Pending => {}
|
|
||||||
}
|
|
||||||
|
|
||||||
// do while there is work to do
|
ready!(Pin::new(&mut self.initiator).poll(cx));
|
||||||
while {
|
|
||||||
// First things first:
|
|
||||||
// If we've send an update to the resources in question we have error channel set, so
|
|
||||||
// we poll that first to determine if the resources has acted on it yet.
|
|
||||||
if let Some(ref mut errchan) = self.error_channel {
|
|
||||||
match Pin::new(errchan).poll(cx) {
|
|
||||||
// In case there's an ongoing
|
|
||||||
Poll::Pending => return Poll::Pending,
|
|
||||||
Poll::Ready(Ok(error)) => {
|
|
||||||
self.error_channel = None;
|
|
||||||
self.initiator_reply_tx.send(Err(error));
|
|
||||||
}
|
|
||||||
Poll::Ready(Err(_closed)) => {
|
|
||||||
// Error channel was dropped which means there was no error
|
|
||||||
self.error_channel = None;
|
|
||||||
self.initiator_reply_tx.send(Ok(()));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(ref mut init_fut) = self.initiator_future {
|
tracing::warn!(initiator=%self.name, "initiator module ran to completion!");
|
||||||
match init_fut.as_mut().poll(cx) {
|
|
||||||
Poll::Pending => return Poll::Pending,
|
|
||||||
Poll::Ready(Ok(())) => {},
|
|
||||||
Poll::Ready(Err(_e)) => {
|
|
||||||
// TODO: Log initiator error here
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if let Some(ref mut _resource) = self.resource {
|
|
||||||
let mut s = self.update_sink.clone();
|
|
||||||
let f = self.initiator.run(&mut s);
|
|
||||||
self.initiator_future.replace(f);
|
|
||||||
}
|
|
||||||
|
|
||||||
self.error_channel.is_some()
|
|
||||||
} {}
|
|
||||||
|
|
||||||
Poll::Ready(())
|
Poll::Ready(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn load(
|
||||||
|
executor: Executor,
|
||||||
|
config: &Config,
|
||||||
|
resources: ResourcesHandle,
|
||||||
|
sessions: SessionManager,
|
||||||
|
authentication: AuthenticationHandle,
|
||||||
|
) -> miette::Result<()> {
|
||||||
|
let span = tracing::info_span!("loading initiators");
|
||||||
|
let _guard = span.enter();
|
||||||
|
|
||||||
|
let mut initiator_map: HashMap<String, Resource> = config
|
||||||
|
.init_connections
|
||||||
|
.iter()
|
||||||
|
.filter_map(|(k, v)| {
|
||||||
|
if let Some(resource) = resources.get_by_id(v) {
|
||||||
|
Some((k.clone(), resource.clone()))
|
||||||
|
} else {
|
||||||
|
tracing::error!(initiator=%k, machine=%v,
|
||||||
|
"Machine configured for initiator not found!");
|
||||||
|
None
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
for (name, cfg) in config.initiators.iter() {
|
||||||
|
if let Some(resource) = initiator_map.remove(name) {
|
||||||
|
if let Some(driver) = load_single(name, &cfg.module, &cfg.params, resource, &sessions) {
|
||||||
|
tracing::debug!(module_name=%cfg.module, %name, "starting initiator task");
|
||||||
|
executor.spawn(driver);
|
||||||
|
} else {
|
||||||
|
tracing::error!(module_name=%cfg.module, %name, "Initiator module could not be configured");
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tracing::warn!(actor=%name, ?config, "Initiator has no machine configured. Skipping!");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn load_single(
|
||||||
|
name: &String,
|
||||||
|
module_name: &String,
|
||||||
|
params: &HashMap<String, String>,
|
||||||
|
resource: Resource,
|
||||||
|
sessions: &SessionManager,
|
||||||
|
) -> Option<InitiatorDriver> {
|
||||||
|
tracing::info!(%name, %module_name, ?params, "Loading initiator");
|
||||||
|
let o = match module_name.as_ref() {
|
||||||
|
"Dummy" => Some(InitiatorDriver::new::<Dummy>(
|
||||||
|
name.clone(),
|
||||||
|
params,
|
||||||
|
resource,
|
||||||
|
sessions.clone(),
|
||||||
|
)),
|
||||||
|
"Process" => Some(InitiatorDriver::new::<Process>(
|
||||||
|
name.clone(),
|
||||||
|
params,
|
||||||
|
resource,
|
||||||
|
sessions.clone(),
|
||||||
|
)),
|
||||||
|
_ => None,
|
||||||
|
};
|
||||||
|
|
||||||
|
o.transpose().unwrap_or_else(|error| {
|
||||||
|
tracing::error!(%error, "failed to configure initiator");
|
||||||
|
None
|
||||||
|
})
|
||||||
|
}
|
||||||
|
166
bffhd/initiators/process.rs
Normal file
166
bffhd/initiators/process.rs
Normal file
@ -0,0 +1,166 @@
|
|||||||
|
use super::Initiator;
|
||||||
|
use super::InitiatorCallbacks;
|
||||||
|
use crate::resources::state::State;
|
||||||
|
use crate::utils::linebuffer::LineBuffer;
|
||||||
|
use async_process::{Child, ChildStdout, Command, Stdio};
|
||||||
|
use futures_lite::{ready, AsyncRead};
|
||||||
|
use miette::{miette, IntoDiagnostic};
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::future::Future;
|
||||||
|
use std::io;
|
||||||
|
use std::pin::Pin;
|
||||||
|
use std::task::{Context, Poll};
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
|
pub enum InputMessage {
|
||||||
|
#[serde(rename = "state")]
|
||||||
|
SetState(State),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize)]
|
||||||
|
pub struct OutputLine {}
|
||||||
|
|
||||||
|
pub struct Process {
|
||||||
|
pub cmd: String,
|
||||||
|
pub args: Vec<String>,
|
||||||
|
state: Option<ProcessState>,
|
||||||
|
buffer: LineBuffer,
|
||||||
|
callbacks: InitiatorCallbacks,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Process {
|
||||||
|
fn spawn(&mut self) -> io::Result<()> {
|
||||||
|
let mut child = Command::new(&self.cmd)
|
||||||
|
.args(&self.args)
|
||||||
|
.stdin(Stdio::null())
|
||||||
|
.stdout(Stdio::piped())
|
||||||
|
.stderr(Stdio::piped())
|
||||||
|
.spawn()?;
|
||||||
|
self.state = Some(ProcessState::new(
|
||||||
|
child
|
||||||
|
.stdout
|
||||||
|
.take()
|
||||||
|
.expect("Child just spawned with piped stdout has no stdout"),
|
||||||
|
child,
|
||||||
|
));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct ProcessState {
|
||||||
|
pub child: Child,
|
||||||
|
pub stdout: ChildStdout,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ProcessState {
|
||||||
|
pub fn new(stdout: ChildStdout, child: Child) -> Self {
|
||||||
|
Self { stdout, child }
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_process(&mut self, buffer: &[u8], callbacks: &mut InitiatorCallbacks) -> usize {
|
||||||
|
tracing::trace!("trying to process current buffer");
|
||||||
|
|
||||||
|
let mut end = 0;
|
||||||
|
|
||||||
|
while let Some(idx) = buffer[end..].iter().position(|b| *b == b'\n') {
|
||||||
|
if idx == 0 {
|
||||||
|
end += 1;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
let line = &buffer[end..(end + idx)];
|
||||||
|
self.process_line(line, callbacks);
|
||||||
|
end = idx;
|
||||||
|
}
|
||||||
|
|
||||||
|
end
|
||||||
|
}
|
||||||
|
|
||||||
|
fn process_line(&mut self, line: &[u8], callbacks: &mut InitiatorCallbacks) {
|
||||||
|
match serde_json::from_slice::<InputMessage>(line) {
|
||||||
|
Ok(state) => {
|
||||||
|
tracing::trace!(?state, "got new state for process initiator");
|
||||||
|
}
|
||||||
|
Err(error) => tracing::warn!(%error, "process initiator did not send a valid line"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Future for Process {
|
||||||
|
type Output = ();
|
||||||
|
|
||||||
|
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||||
|
if let Process {
|
||||||
|
state: Some(state),
|
||||||
|
buffer,
|
||||||
|
callbacks,
|
||||||
|
..
|
||||||
|
} = self.get_mut()
|
||||||
|
{
|
||||||
|
match state.child.try_status() {
|
||||||
|
Err(error) => {
|
||||||
|
tracing::error!(%error, "checking child exit code returned an error");
|
||||||
|
return Poll::Ready(());
|
||||||
|
}
|
||||||
|
Ok(Some(exitcode)) => {
|
||||||
|
tracing::warn!(%exitcode, "child process exited");
|
||||||
|
return Poll::Ready(());
|
||||||
|
}
|
||||||
|
Ok(None) => {
|
||||||
|
tracing::trace!("process initiator checking on process");
|
||||||
|
|
||||||
|
let stdout = &mut state.stdout;
|
||||||
|
|
||||||
|
loop {
|
||||||
|
let buf = buffer.get_mut_write(512);
|
||||||
|
match AsyncRead::poll_read(Pin::new(stdout), cx, buf) {
|
||||||
|
Poll::Pending => break,
|
||||||
|
Poll::Ready(Ok(read)) => {
|
||||||
|
buffer.advance_valid(read);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
Poll::Ready(Err(error)) => {
|
||||||
|
tracing::warn!(%error, "reading from child stdout errored");
|
||||||
|
return Poll::Ready(());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let processed = state.try_process(buffer, callbacks);
|
||||||
|
buffer.consume(processed);
|
||||||
|
|
||||||
|
return Poll::Pending;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tracing::warn!("process initiator has no process attached!");
|
||||||
|
}
|
||||||
|
|
||||||
|
Poll::Ready(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Initiator for Process {
|
||||||
|
fn new(params: &HashMap<String, String>, callbacks: InitiatorCallbacks) -> miette::Result<Self>
|
||||||
|
where
|
||||||
|
Self: Sized,
|
||||||
|
{
|
||||||
|
let cmd = params
|
||||||
|
.get("cmd")
|
||||||
|
.ok_or(miette!("Process initiator requires a `cmd` parameter."))?
|
||||||
|
.clone();
|
||||||
|
let args = params
|
||||||
|
.get("args")
|
||||||
|
.map(|argv| argv.split_whitespace().map(|s| s.to_string()).collect())
|
||||||
|
.unwrap_or_else(Vec::new);
|
||||||
|
let mut this = Self {
|
||||||
|
cmd,
|
||||||
|
args,
|
||||||
|
state: None,
|
||||||
|
buffer: LineBuffer::new(),
|
||||||
|
callbacks,
|
||||||
|
};
|
||||||
|
this.spawn().into_diagnostic()?;
|
||||||
|
Ok(this)
|
||||||
|
}
|
||||||
|
}
|
14
bffhd/lib.rs
14
bffhd/lib.rs
@ -24,6 +24,7 @@ pub mod users;
|
|||||||
pub mod resources;
|
pub mod resources;
|
||||||
|
|
||||||
pub mod actors;
|
pub mod actors;
|
||||||
|
pub mod initiators;
|
||||||
|
|
||||||
pub mod sensors;
|
pub mod sensors;
|
||||||
|
|
||||||
@ -118,15 +119,22 @@ impl Diflouroborane {
|
|||||||
.into_diagnostic()
|
.into_diagnostic()
|
||||||
.wrap_err("Failed to construct signal handler")?;
|
.wrap_err("Failed to construct signal handler")?;
|
||||||
|
|
||||||
|
let sessionmanager = SessionManager::new(self.users.clone(), self.roles.clone());
|
||||||
|
let authentication = AuthenticationHandle::new(self.users.clone());
|
||||||
|
|
||||||
|
initiators::load(
|
||||||
|
self.executor.clone(),
|
||||||
|
&self.config,
|
||||||
|
self.resources.clone(),
|
||||||
|
sessionmanager.clone(),
|
||||||
|
authentication.clone(),
|
||||||
|
);
|
||||||
actors::load(self.executor.clone(), &self.config, self.resources.clone())?;
|
actors::load(self.executor.clone(), &self.config, self.resources.clone())?;
|
||||||
|
|
||||||
let tlsconfig = TlsConfig::new(self.config.tlskeylog.as_ref(), !self.config.is_quiet())
|
let tlsconfig = TlsConfig::new(self.config.tlskeylog.as_ref(), !self.config.is_quiet())
|
||||||
.into_diagnostic()?;
|
.into_diagnostic()?;
|
||||||
let acceptor = tlsconfig.make_tls_acceptor(&self.config.tlsconfig)?;
|
let acceptor = tlsconfig.make_tls_acceptor(&self.config.tlsconfig)?;
|
||||||
|
|
||||||
let sessionmanager = SessionManager::new(self.users.clone(), self.roles.clone());
|
|
||||||
let authentication = AuthenticationHandle::new(self.users.clone());
|
|
||||||
|
|
||||||
let apiserver = self.executor.run(APIServer::bind(
|
let apiserver = self.executor.run(APIServer::bind(
|
||||||
self.executor.clone(),
|
self.executor.clone(),
|
||||||
&self.config.listens,
|
&self.config.listens,
|
||||||
|
@ -25,6 +25,7 @@ pub mod modules;
|
|||||||
|
|
||||||
pub struct PermissionDenied;
|
pub struct PermissionDenied;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub(crate) struct Inner {
|
pub(crate) struct Inner {
|
||||||
id: String,
|
id: String,
|
||||||
db: StateDB,
|
db: StateDB,
|
||||||
@ -94,7 +95,7 @@ impl Inner {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct Resource {
|
pub struct Resource {
|
||||||
inner: Arc<Inner>,
|
inner: Arc<Inner>,
|
||||||
}
|
}
|
||||||
|
@ -1,9 +1,9 @@
|
|||||||
|
use crate::config::deser_option;
|
||||||
use crate::utils::oid::ObjectIdentifier;
|
use crate::utils::oid::ObjectIdentifier;
|
||||||
use once_cell::sync::Lazy;
|
use once_cell::sync::Lazy;
|
||||||
use rkyv::{Archive, Archived, Deserialize, Infallible};
|
use rkyv::{Archive, Archived, Deserialize, Infallible};
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
|
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
//use crate::oidvalue;
|
//use crate::oidvalue;
|
||||||
@ -54,6 +54,11 @@ pub enum Status {
|
|||||||
/// The status of the machine
|
/// The status of the machine
|
||||||
pub struct MachineState {
|
pub struct MachineState {
|
||||||
pub state: Status,
|
pub state: Status,
|
||||||
|
#[serde(
|
||||||
|
default,
|
||||||
|
skip_serializing_if = "Option::is_none",
|
||||||
|
deserialize_with = "deser_option"
|
||||||
|
)]
|
||||||
pub previous: Option<UserRef>,
|
pub previous: Option<UserRef>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
60
bffhd/utils/linebuffer.rs
Normal file
60
bffhd/utils/linebuffer.rs
Normal file
@ -0,0 +1,60 @@
|
|||||||
|
use std::ops::{Deref, DerefMut};
|
||||||
|
|
||||||
|
pub struct LineBuffer {
|
||||||
|
buffer: Vec<u8>,
|
||||||
|
valid: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl LineBuffer {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
buffer: Vec::new(),
|
||||||
|
valid: 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Resize the internal Vec so that buffer.len() == buffer.capacity()
|
||||||
|
fn resize(&mut self) {
|
||||||
|
// SAFETY: Whatever is in memory is always valid as u8.
|
||||||
|
unsafe { self.buffer.set_len(self.buffer.capacity()) }
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get an (initialized but empty) writeable buffer of at least `atleast` bytes
|
||||||
|
pub fn get_mut_write(&mut self, atleast: usize) -> &mut [u8] {
|
||||||
|
let avail = self.buffer.len() - self.valid;
|
||||||
|
if avail < atleast {
|
||||||
|
self.buffer.reserve(atleast - avail);
|
||||||
|
self.resize()
|
||||||
|
}
|
||||||
|
&mut self.buffer[self.valid..]
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn advance_valid(&mut self, amount: usize) {
|
||||||
|
self.valid += amount
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Mark `amount` bytes as 'consumed'
|
||||||
|
///
|
||||||
|
/// This will move any remaining data to the start of the buffer for future processing
|
||||||
|
pub fn consume(&mut self, amount: usize) {
|
||||||
|
assert!(amount <= self.valid);
|
||||||
|
|
||||||
|
if amount < self.valid {
|
||||||
|
self.buffer.copy_within(amount..self.valid, 0);
|
||||||
|
}
|
||||||
|
self.valid -= amount;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Deref for LineBuffer {
|
||||||
|
type Target = [u8];
|
||||||
|
|
||||||
|
fn deref(&self) -> &Self::Target {
|
||||||
|
&self.buffer[0..self.valid]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl DerefMut for LineBuffer {
|
||||||
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
||||||
|
&mut self.buffer[0..self.valid]
|
||||||
|
}
|
||||||
|
}
|
@ -8,3 +8,5 @@ pub mod varint;
|
|||||||
pub mod l10nstring;
|
pub mod l10nstring;
|
||||||
|
|
||||||
pub mod uuid;
|
pub mod uuid;
|
||||||
|
|
||||||
|
pub mod linebuffer;
|
||||||
|
13
examples/init.py
Executable file
13
examples/init.py
Executable file
@ -0,0 +1,13 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
|
||||||
|
while True:
|
||||||
|
print('{ "state": { "1.3.6.1.4.1.48398.612.2.4": { "state": "Free" } } }')
|
||||||
|
sys.stdout.flush()
|
||||||
|
time.sleep(2)
|
||||||
|
|
||||||
|
print('{ "state": { "1.3.6.1.4.1.48398.612.2.4": { "state": { "InUse": { "id": "Testuser" } } } } }')
|
||||||
|
sys.stdout.flush()
|
||||||
|
time.sleep(2)
|
Loading…
Reference in New Issue
Block a user