Skip to content
This repository was archived by the owner on Jan 12, 2026. It is now read-only.

Fix tdx-quote-provider shutdown signal handling on non-unix #42

Open
kinderjoypresents wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix tdx-quote-provider shutdown signal handling on non-unix #42

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 23 additions & 12 deletions crates/tdx-quote-provider/src/server.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -163,18 +163,29 @@ async fn attest_handler(
}

async fn shutdown_signal() {
let mut sigterm = signal::unix::signal(signal::unix::SignalKind::terminate()).unwrap();
let mut sigint = signal::unix::signal(signal::unix::SignalKind::interrupt()).unwrap();

tokio::select! {
_ = signal::ctrl_c() => {
info!("Received Ctrl+C, shutting down gracefully");
}
_ = sigterm.recv() => {
info!("Received SIGTERM, shutting down gracefully");
}
_ = sigint.recv() => {
info!("Received SIGINT, shutting down gracefully");
#[cfg(unix)]
{
let mut sigterm = signal::unix::signal(signal::unix::SignalKind::terminate())
.expect("failed to install SIGTERM handler");
let mut sigint = signal::unix::signal(signal::unix::SignalKind::interrupt())
.expect("failed to install SIGINT handler");

tokio::select! {
_ = signal::ctrl_c() => {
info!("Received Ctrl+C, shutting down gracefully");
}
_ = sigterm.recv() => {
info!("Received SIGTERM, shutting down gracefully");
}
_ = sigint.recv() => {
info!("Received SIGINT, shutting down gracefully");
}
}
}

#[cfg(not(unix))]
{
let _ = signal::ctrl_c().await;
info!("Received Ctrl+C, shutting down gracefully");
}
}