45bba35a71
This repository is no longer meant for just radarflow, thus it will be renamed. I have split the SDK from radarflow, allowing for simpler use with new projects. Other than that, radarflow is functionally the same. - Fixed bug in radarflow where the entity loop didn't include the last entity.
58 lines
1.4 KiB
Rust
58 lines
1.4 KiB
Rust
use std::sync::Arc;
|
|
|
|
use clap::Parser;
|
|
use cli::Cli;
|
|
use comms::RadarData;
|
|
use tokio::sync::RwLock;
|
|
|
|
mod comms;
|
|
mod cli;
|
|
|
|
mod dma;
|
|
mod websocket;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
let cli = Cli::parse();
|
|
|
|
simple_logger::SimpleLogger::new()
|
|
.with_level(cli.loglevel.into())
|
|
.init()
|
|
.expect("Initializing logger");
|
|
|
|
let rwlock = Arc::new(
|
|
RwLock::new(
|
|
RadarData::empty()
|
|
)
|
|
);
|
|
|
|
let rwlock_clone = rwlock.clone();
|
|
let dma_handle = tokio::spawn(async move {
|
|
if let Err(err) = dma::run(cli.connector, cli.pcileech_device, cli.poll_rate, rwlock_clone).await {
|
|
log::error!("Error in dma thread: [{}]", err.to_string());
|
|
}
|
|
});
|
|
|
|
tokio::spawn(async move {
|
|
let future = websocket::run(cli.web_path, cli.port, rwlock);
|
|
|
|
if let Ok(my_local_ip) = local_ip_address::local_ip() {
|
|
let address = format!("http://{}:{}", my_local_ip, cli.port);
|
|
println!("Launched webserver at {}", address);
|
|
} else {
|
|
let address = format!("http://0.0.0.0:{}", cli.port);
|
|
println!("launched webserver at! {}", address);
|
|
}
|
|
|
|
if let Err(err) = future.await {
|
|
log::error!("Error in websocket server: [{}]", err.to_string());
|
|
}
|
|
});
|
|
|
|
if let Err(err) = dma_handle.await {
|
|
log::error!("Error when waiting for dma thread: {}", err.to_string());
|
|
}
|
|
|
|
Ok(())
|
|
}
|