fcnet 0.12.0

An in-process Firecracker microVM networking backend based on fcnet-types.
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

#[cfg(feature = "smol-backend")]
use async_executor::{Executor, LocalExecutor};
use netlink_proto::Connection;
use rtnetlink::packet_route::RouteNetlinkMessage;
use std::future::Future;
#[cfg(feature = "smol-backend")]
use std::sync::{Arc, OnceLock};

/// The [Backend] trait encapsulates the async-runtime-dependent functionality that is needed for fcnet
/// to function.
pub trait Backend: Send + Sync + 'static {
    /// The [rtnetlink::sys::AsyncSocket] (async fd implementation) used by this backend.
    type NetlinkSocket: rtnetlink::sys::AsyncSocket + Send;
    /// The [nftables_async] driver used by this backend.
    type NftablesDriver: nftables_async::driver::Driver;

    /// Spawn a netlink [Connection] onto this async runtime, detaching the spawned task to have it run
    /// in the background.
    fn spawn_connection(connection: Connection<RouteNetlinkMessage, Self::NetlinkSocket>);

    /// Create a thread-local, !Send async executor from this runtime and block it on the given future.
    /// This will be called in a separate OS thread spawned by fcnet for the purposes of calling setns
    /// within it to operate within the context of another network namespace.
    fn block_on_current_thread<O, F: Future<Output = O>>(future: F) -> O;
}

/// A [Backend] implementation that uses the tokio crate for async I/O and its current-thread executor.
#[cfg(feature = "tokio-backend")]
#[cfg_attr(docsrs, doc(cfg(feature = "tokio-backend")))]
pub struct TokioBackend;

#[cfg(feature = "tokio-backend")]
#[cfg_attr(docsrs, doc(cfg(feature = "tokio-backend")))]
impl Backend for TokioBackend {
    type NetlinkSocket = netlink_proto::sys::TokioSocket;
    type NftablesDriver = nftables_async::driver::TokioDriver;

    fn spawn_connection(connection: Connection<RouteNetlinkMessage, Self::NetlinkSocket>) {
        tokio::task::spawn(connection);
    }

    fn block_on_current_thread<O, F: Future<Output = O>>(future: F) -> O {
        tokio::runtime::Builder::new_current_thread()
            .enable_all()
            .build()
            .expect("Could not build current-thread Tokio runtime. This panic should be isolated to another thread")
            .block_on(future)
    }
}

#[cfg(feature = "smol-backend")]
static SMOL_EXECUTOR: OnceLock<Arc<Executor>> = OnceLock::new();

/// A [Backend] implementation that uses the async-process and async-executor crates from the Smol stack.
/// [SmolBackend] must be initialized by passing it an [Arc] of an [Executor<'static>], otherwise functions that
/// spawn tasks on an executor will panic.
#[cfg(feature = "smol-backend")]
#[cfg_attr(docsrs, doc(cfg(feature = "smol-backend")))]
pub struct SmolBackend;

#[cfg(feature = "smol-backend")]
#[cfg_attr(docsrs, doc(cfg(feature = "smol-backend")))]
impl SmolBackend {
    /// Initializes the backend with the given [Arc] of an [Executor<'static>] from the async-executor crate.
    /// This can only be done once per application, as it uses [OnceLock] internally, and should only be performed
    /// within binary code.
    pub fn initialize(executor: impl Into<Arc<Executor<'static>>>) {
        SMOL_EXECUTOR
            .set(executor.into())
            .expect("Smol executor was already initialized");
    }
}

#[cfg(feature = "smol-backend")]
#[cfg_attr(docsrs, doc(cfg(feature = "smol-backend")))]
impl Backend for SmolBackend {
    type NetlinkSocket = netlink_proto::sys::SmolSocket;
    type NftablesDriver = nftables_async::driver::AsyncProcessDriver;

    fn spawn_connection(connection: Connection<RouteNetlinkMessage, Self::NetlinkSocket>) {
        SMOL_EXECUTOR
            .get()
            .expect("Smol executor wasn't initialized")
            .spawn(connection)
            .detach();
    }

    fn block_on_current_thread<O, F: Future<Output = O>>(future: F) -> O {
        async_io::block_on(LocalExecutor::new().run(future))
    }
}