xref: /wasmtime-44.0.1/crates/test-programs/src/tls.rs (revision 90ac295e)

use crate::wasi::clocks::monotonic_clock;
use crate::wasi::io::error::Error as IoError;
use crate::wasi::io::streams::StreamError;
use crate::wasi::tls::types::{ClientConnection, ClientHandshake, InputStream, OutputStream};

const TIMEOUT_NS: u64 = 1_000_000_000;

impl ClientHandshake {
    pub fn blocking_finish(self) -> Result<(ClientConnection, InputStream, OutputStream), IoError> {
        let future = ClientHandshake::finish(self);
        let timeout = monotonic_clock::subscribe_duration(TIMEOUT_NS * 200);
        let pollable = future.subscribe();

        loop {
            match future.get() {
                None => pollable.block_until(&timeout).expect("timed out"),
                Some(Ok(r)) => return r,
                Some(Err(e)) => {
                    eprintln!("{e:?}");
                    unimplemented!()
                }
            }
        }
    }
}

impl ClientConnection {
    pub fn blocking_close_output(
        &self,
        output: &OutputStream,
    ) -> Result<(), crate::wasi::io::error::Error> {
        let timeout = monotonic_clock::subscribe_duration(TIMEOUT_NS);
        let pollable = output.subscribe();

        self.close_output();

        loop {
            match output.check_write() {
                Ok(0) => pollable.block_until(&timeout).expect("timed out"),
                Ok(_) => unreachable!(
                    "After calling close_output, the output stream should never accept new writes again."
                ),
                Err(StreamError::Closed) => return Ok(()),
                Err(StreamError::LastOperationFailed(e)) => return Err(e),
            }
        }
    }
}