wasm_bpf_rs/
handle.rs

1use std::sync::mpsc;
2
3use anyhow::{anyhow, bail, Context};
4use log::debug;
5use wasmtime::Engine;
6
7/// This is the signal that will be sended to the hanging epoch interruption callback function
8pub enum ProgramOperation {
9    /// Resume the program
10    Resume,
11    /// Terminate the program
12    Terminate,
13}
14
15/// This is a handle to the wasm program
16pub struct WasmProgramHandle {
17    operation_tx: mpsc::Sender<ProgramOperation>,
18    paused: bool,
19    engine: Engine,
20}
21
22impl WasmProgramHandle {
23    pub(crate) fn new(operation_tx: mpsc::Sender<ProgramOperation>, engine: Engine) -> Self {
24        Self {
25            operation_tx,
26            engine,
27            paused: false,
28        }
29    }
30    /// Pause the wasm program
31    /// Error will be returned when the program was already paused
32    pub fn pause(&mut self) -> anyhow::Result<()> {
33        if self.paused {
34            bail!("Already paused!");
35        }
36        self.engine.increment_epoch();
37        self.paused = true;
38        Ok(())
39    }
40    /// Resume the wasm program
41    /// Error will be returned when the program was already running, or when the program as terminated
42    pub fn resume(&mut self) -> anyhow::Result<()> {
43        if !self.paused {
44            bail!("Already running!");
45        }
46        self.operation_tx
47            .send(ProgramOperation::Resume)
48            .with_context(|| anyhow!("Failed to send resume operation"))?;
49        self.paused = false;
50        Ok(())
51    }
52    /// Terminate the wasm program
53    /// Error will be returned when the program was already terminated
54    pub fn terminate(self) -> anyhow::Result<()> {
55        debug!("Terminating wasm program");
56        self.engine.increment_epoch();
57        self.operation_tx
58            .send(ProgramOperation::Terminate)
59            .with_context(|| anyhow!("Failed to send terminate operation"))?;
60        Ok(())
61    }
62}
wasm_bpf_rs/handle.rs

wasm_bpf_rs/
handle.rs
