rns-hooks 0.1.4

WASM hook system for the Reticulum Network Stack
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

use crate::runtime::StoreData;
use wasmtime::{Instance, Module, Store};

/// A compiled WASM hook program ready for execution.
pub struct LoadedProgram {
    pub name: String,
    pub module: Module,
    pub priority: i32,
    pub consecutive_traps: u32,
    pub enabled: bool,
    pub max_consecutive_traps: u32,
    pub export_name: String,
    /// Cached store and instance for cross-call state persistence.
    pub cached: Option<(Store<StoreData>, Instance)>,
}

impl LoadedProgram {
    pub fn new(name: String, module: Module, priority: i32) -> Self {
        LoadedProgram {
            name,
            module,
            priority,
            consecutive_traps: 0,
            enabled: true,
            max_consecutive_traps: 10,
            export_name: "on_hook".to_string(),
            cached: None,
        }
    }

    /// Reset the consecutive trap counter after a successful execution.
    pub fn record_success(&mut self) {
        self.consecutive_traps = 0;
    }

    /// Drop the cached store/instance (e.g. on reload).
    pub fn drop_cache(&mut self) {
        self.cached = None;
    }

    /// Increment the trap counter. Returns `true` if the program was auto-disabled.
    pub fn record_trap(&mut self) -> bool {
        self.consecutive_traps += 1;
        if self.consecutive_traps >= self.max_consecutive_traps {
            self.enabled = false;
            true
        } else {
            false
        }
    }
}