usbvfiod 0.1.0

A vfio-user server for USB pass-through.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! Manage tokio runtime

use anyhow::{anyhow, Context, Result};
use std::sync::OnceLock;
use tokio::runtime::Handle;

static RUNTIME_HANDLE: OnceLock<Handle> = OnceLock::new();

pub fn runtime() -> &'static Handle {
    RUNTIME_HANDLE.get().expect("runtime not initialized")
}

pub fn init_runtime() -> Result<()> {
    let rt = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()
        .context("Failed to build tokio runtime")?;

    RUNTIME_HANDLE
        .set(rt.handle().clone())
        .map_err(|_| anyhow!("runtime already initialized"))?;

    // Keep runtime alive forever
    std::mem::forget(rt);

    Ok(())
}