Function run

crates/openshell-sandbox/src/metadata_server.rs:48–87

Bind a TCP listener inside the sandbox network namespace. Run the metadata server accept loop. Signals `ready_tx` with the bound address before entering the loop. Returns when the listener encounters a fatal error or the runtime shuts down.

(
    listener: TcpListener,
    handler: H,
    ready_tx: oneshot::Sender<SocketAddr>,
)

/// Signals `ready_tx` with the bound address before entering the loop.
/// Returns when the listener encounters a fatal error or the runtime shuts down.
pub async fn run<H: MetadataHandler>(
    listener: TcpListener,
    handler: H,
    ready_tx: oneshot::Sender<SocketAddr>,
) {
    let local_addr = match listener.local_addr() {
        Ok(addr) => addr,
        Err(e) => {
            warn!("metadata server failed to get local address: {e}");
            return;
        }
    };

    let _ = ready_tx.send(local_addr);

    let handler = Arc::new(handler);
    let semaphore = Arc::new(Semaphore::new(MAX_CONCURRENT_CONNECTIONS));

    loop {
        let Ok(permit) = semaphore.clone().acquire_owned().await else {
            break;
        };

        match listener.accept().await {
            Ok((stream, _addr)) => {
                let handler = handler.clone();
                tokio::spawn(async move {
                    if let Err(e) = handle_connection(handler.as_ref(), stream).await {
                        debug!("metadata server connection error: {e}");
                    }
                    drop(permit);
                });
            }
            Err(e) => {
                warn!("metadata server accept error: {e}");
                tokio::time::sleep(std::time::Duration::from_millis(50)).await;
            }
        }
    }
}

const READ_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(5);