cc-lb-runtime-wasmtime 0.1.4

Wasmtime-based plugin runtime for cc-lb. Host-side wasm plugin admission + dispatch.
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
92
93
94
95
96
97
98

use std::sync::Arc;

use rkyv::util::AlignedVec;

use super::{HookFn, call_hook_scoped};
use crate::cell::PluginCell;
use crate::error::WasmtimeRuntimeError;

fn copy_to_aligned(slice: &[u8]) -> AlignedVec<16> {
    let mut aligned = AlignedVec::with_capacity(slice.len());
    aligned.extend_from_slice(slice);
    aligned
}

/// Synchronous filter call. Builds a fresh `Store`, runs the hook,
/// drops the `Store`.
pub fn call_filter_hook(
    cell: &Arc<PluginCell>,
    input: &[u8],
) -> Result<Vec<u8>, WasmtimeRuntimeError> {
    call_filter_hook_scoped(cell, input, <[u8]>::to_vec)
}

pub(crate) fn call_filter_hook_scoped<R, F>(
    cell: &Arc<PluginCell>,
    input: &[u8],
    with_output: F,
) -> Result<R, WasmtimeRuntimeError>
where
    F: for<'a> FnOnce(&'a [u8]) -> R,
{
    call_hook_scoped(cell, input, HookFn::Filter, with_output)
}

/// Synchronous shape call. Input is rkyv-encoded `ShapeRequest`,
/// output rkyv-encoded `ShapeResponse`.
pub fn call_shape_hook(
    cell: &Arc<PluginCell>,
    input: &[u8],
) -> Result<Vec<u8>, WasmtimeRuntimeError> {
    call_shape_hook_scoped(cell, input, <[u8]>::to_vec)
}

pub(crate) fn call_shape_hook_scoped<R, F>(
    cell: &Arc<PluginCell>,
    input: &[u8],
    with_output: F,
) -> Result<R, WasmtimeRuntimeError>
where
    F: for<'a> FnOnce(&'a [u8]) -> R,
{
    call_hook_scoped(cell, input, HookFn::Shape, with_output)
}

/// Synchronous observe call. Guest returns `(0, 0)`; the returned
/// `Vec<u8>` is always empty.
pub fn call_observe_hook(
    cell: &Arc<PluginCell>,
    input: &[u8],
) -> Result<Vec<u8>, WasmtimeRuntimeError> {
    call_hook_scoped(cell, input, HookFn::Observe, <[u8]>::to_vec)
}

pub fn call_transform_response_hook(
    cell: &Arc<PluginCell>,
    input: &[u8],
) -> Result<AlignedVec<16>, WasmtimeRuntimeError> {
    call_transform_response_hook_scoped(cell, input, copy_to_aligned)
}

pub(crate) fn call_transform_response_hook_scoped<R, F>(
    cell: &Arc<PluginCell>,
    input: &[u8],
    with_output: F,
) -> Result<R, WasmtimeRuntimeError>
where
    F: for<'a> FnOnce(&'a [u8]) -> R,
{
    call_hook_scoped(cell, input, HookFn::TransformResponse, with_output)
}

pub fn call_transform_sse_event_hook(
    cell: &Arc<PluginCell>,
    input: &[u8],
) -> Result<AlignedVec<16>, WasmtimeRuntimeError> {
    call_transform_sse_event_hook_scoped(cell, input, copy_to_aligned)
}

pub(crate) fn call_transform_sse_event_hook_scoped<R, F>(
    cell: &Arc<PluginCell>,
    input: &[u8],
    with_output: F,
) -> Result<R, WasmtimeRuntimeError>
where
    F: for<'a> FnOnce(&'a [u8]) -> R,
{
    call_hook_scoped(cell, input, HookFn::TransformSseEvent, with_output)
}