DevComp/ssh-portfolio
1
1
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: top-level panic recovery mechanism

Browse source
This commit is contained in:
Erica Marigold 2025-08-27 13:37:09 +01:00
parent e7caac4dd7
commit 430ba54185
Signed by: DevComp
SSH key fingerprint: SHA256:jD3oMT4WL3WHPJQbrjC3l5feNCnkv7ndW8nYaHX5wFw
1 changed files with 21 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/main.rs
View file

@ -51,10 +51,27 @@ async fn main() -> Result<()> {
let ssh_socket_addr = SSH_SOCKET_ADDR.ok_or(eyre!("Invalid host IP provided"))?;
let web_server_addr = WEB_SERVER_ADDR.ok_or(eyre!("Invalid host IP provided"))?;
let ssh_config = tokio::task::block_in_place(ssh_config);
tokio::select! {
ssh_res = SshServer::start(ssh_socket_addr, ssh_config) => ssh_res,
web_res = WebLandingServer::start(web_server_addr) => web_res.map_err(|err| eyre!(err)),
let local_set = tokio::task::LocalSet::new();
loop {
let task = local_set.run_until(async {
tokio::task::spawn_local(async move {
let ssh_config = ssh_config();
tokio::select! {
ssh_res = SshServer::start(ssh_socket_addr, ssh_config) => ssh_res,
web_res = WebLandingServer::start(web_server_addr) => web_res.map_err(|err| eyre!(err))
}
}).await
});
match task.await {
Ok(res) => break res,
Err(err) => {
if err.is_panic() {
tracing::error!("Application panicked, recovering...");
std::panic::resume_unwind(err.into_panic());
}
}
}
}
}