mirror of
https://github.com/EFForg/rayhunter.git
synced 2026-05-30 07:29:27 -07:00
f8824ce7e7
Rayhunter uses a mixture of spawn and spawn_blocking, then also does some blocking operations inside of async code. Move everything to async. This allows us to use the single-threaded runtime. Now the binary is 100kB smaller, and the memory usage also improved by ~100kB on tplink.
52 lines
1.3 KiB
Rust
52 lines
1.3 KiB
Rust
use crate::config;
|
|
use crate::display::DisplayState;
|
|
use crate::display::generic_framebuffer::{self, Dimensions, GenericFramebuffer};
|
|
use async_trait::async_trait;
|
|
|
|
use tokio::sync::mpsc::Receiver;
|
|
use tokio::sync::oneshot;
|
|
use tokio_util::task::TaskTracker;
|
|
|
|
const FB_PATH: &str = "/dev/fb0";
|
|
|
|
#[derive(Copy, Clone, Default)]
|
|
struct Framebuffer;
|
|
|
|
#[async_trait]
|
|
impl GenericFramebuffer for Framebuffer {
|
|
fn dimensions(&self) -> Dimensions {
|
|
// TODO actually poll for this, maybe w/ fbset?
|
|
Dimensions {
|
|
height: 128,
|
|
width: 128,
|
|
}
|
|
}
|
|
|
|
async fn write_buffer(&mut self, buffer: Vec<(u8, u8, u8)>) {
|
|
let mut raw_buffer = Vec::new();
|
|
for (r, g, b) in buffer {
|
|
let mut rgb565: u16 = (r as u16 & 0b11111000) << 8;
|
|
rgb565 |= (g as u16 & 0b11111100) << 3;
|
|
rgb565 |= (b as u16) >> 3;
|
|
raw_buffer.extend(rgb565.to_le_bytes());
|
|
}
|
|
|
|
tokio::fs::write(FB_PATH, &raw_buffer).await.unwrap();
|
|
}
|
|
}
|
|
|
|
pub fn update_ui(
|
|
task_tracker: &TaskTracker,
|
|
config: &config::Config,
|
|
ui_shutdown_rx: oneshot::Receiver<()>,
|
|
ui_update_rx: Receiver<DisplayState>,
|
|
) {
|
|
generic_framebuffer::update_ui(
|
|
task_tracker,
|
|
config,
|
|
Framebuffer,
|
|
ui_shutdown_rx,
|
|
ui_update_rx,
|
|
)
|
|
}
|