andrew.rioux/sparse-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: finished adding handler for new beacons

Browse Source
This commit is contained in:
Andrew Rioux 2025-02-22 21:19:51 -05:00
parent e103fa9f28
commit 9fee4009f2
Signed by: andrew.rioux
GPG Key ID: 9B8BAC47C17ABB94
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
sparse-server/src/webserver.rs
View File

@ -439,7 +439,7 @@ async fn handle_listener_events(
socket.send(ws::Message::Text(serde_json::to_string(&SidebarEvents::Checkin(bid.clone()))?)).await?; socket.send(ws::Message::Text(serde_json::to_string(&SidebarEvents::Checkin(bid.clone()))?)).await?;
} }
Ok(BeaconEvent::NewBeacon(bid)) => { Ok(BeaconEvent::NewBeacon(bid)) => {
let beacons = sqlx::query!( let beacon = sqlx::query!(
"SELECT template_id, peer_ip, nickname, cwd, operating_system, beacon_userent, hostname, config_id FROM beacon_instance "SELECT template_id, peer_ip, nickname, cwd, operating_system, beacon_userent, hostname, config_id FROM beacon_instance
WHERE beacon_id = ?", WHERE beacon_id = ?",
bid bid
@ -452,8 +452,24 @@ async fn handle_listener_events(
WHERE beacon_id = ?", WHERE beacon_id = ?",
bid bid
) )
.fetch_one(&state.db) .fetch_all(&state.db)
.await?; .await?;
let beacon = CurrentBeaconInstance {
beacon_id: bid,
template_id: beacon.template_id,
ip: beacon.peer_ip,
nickname: beacon.nickname,
cwd: beacon.cwd,
operating_system: beacon.operating_system,
userent: beacon.beacon_userent,
hostname: beacon.hostname,
config_id: beacon.config_id,
last_checkin: chrono::Utc::now(),
category_ids: category_ids.iter().map(|r| r.category_id).collect()
};
socket.send(ws::Message::Text(serde_json::to_string(&beacon)?)).await?;
} }
Err(e) => { Err(e) => {
tracing::warn!("Unable to handle general event: {e:?}"); tracing::warn!("Unable to handle general event: {e:?}");