viceroy-lib 0.17.0

Viceroy implementation details.
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143

use {
    crate::linking::ComponentCtx,
    wasmtime::component::{self, HasSelf},
};

pub(crate) mod bindings {
    wasmtime::component::bindgen!({
        path: "wasm_abi/wit",
        world: "fastly:adapter/adapter-service",
        imports: {
            default: tracing,

            "fastly:compute/backend.[constructor]dynamic-backend-options": tracing | trappable,
            "fastly:compute/shielding.[constructor]shield-backend-options": tracing | trappable,
            "fastly:compute/cache.[constructor]extra-lookup-options": tracing | trappable,
            "fastly:compute/cache.[constructor]extra-replace-options": tracing | trappable,
            "fastly:compute/cache.[constructor]extra-write-options": tracing | trappable,

            // The trap-test test depends on being able to induce an artificial
            // trap in `get-header-values`.
            "fastly:compute/http-resp.[method]response.get-header-values": tracing | trappable,

            "fastly:compute/http-body.append": async | tracing,
            "fastly:compute/kv-store.await-delete": async | tracing,
            "fastly:compute/cache.await-entry": async | tracing,
            "fastly:compute/kv-store.await-insert": async | tracing,
            "fastly:compute/kv-store.await-list": async | tracing,
            "fastly:compute/kv-store.await-lookup": async | tracing | trappable,
            "fastly:compute/http-downstream.await-request": async | tracing,
            "fastly:compute/http-req.await-response": async | tracing,
            "fastly:compute/cache.close-entry": async | tracing,
            "fastly:compute/cache.close-replace-entry": async | tracing,
            "fastly:compute/cache.insert": async | tracing,
            "fastly:compute/cache.[static]replace-entry.replace": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-age-ns": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-body": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-hits": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-length": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-max-age-ns": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-stale-while-revalidate-ns": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-state": async | tracing,
            "fastly:compute/cache.[method]replace-entry.get-user-metadata": async | tracing,
            "fastly:compute/cache.replace-insert": async | tracing,
            "fastly:compute/cache.[method]entry.get-age-ns": async | tracing,
            "fastly:compute/cache.[method]entry.get-body": async | tracing,
            "fastly:compute/cache.[method]entry.get-hits": async | tracing,
            "fastly:compute/cache.[method]entry.get-length": async | tracing,
            "fastly:compute/cache.[method]entry.get-max-age-ns": async | tracing,
            "fastly:compute/cache.[method]entry.get-stale-while-revalidate-ns": async | tracing,
            "fastly:compute/cache.[method]entry.get-state": async | tracing,
            "fastly:compute/cache.[method]entry.get-user-metadata": async | tracing,
            "fastly:compute/cache.[method]entry.transaction-cancel": async | tracing,
            "fastly:compute/cache.[method]entry.transaction-insert": async | tracing,
            "fastly:compute/cache.[method]entry.transaction-insert-and-stream-back": async | tracing,
            "fastly:compute/cache.[method]entry.transaction-update": async | tracing,
            "fastly:compute/kv-store.[method]store.delete": async | tracing,
            "fastly:compute/kv-store.[method]store.delete-async": async | tracing,
            "fastly:compute/kv-store.[method]store.insert": async | tracing,
            "fastly:compute/kv-store.[method]store.insert-async": async | tracing,
            "fastly:compute/kv-store.[method]store.list": async | tracing,
            "fastly:compute/kv-store.[method]store.list-async": async | tracing,
            "fastly:compute/kv-store.[method]store.lookup": async | tracing,
            "fastly:compute/kv-store.[method]store.lookup-async": async | tracing,
            "fastly:compute/http-downstream.next-request": async | tracing,
            "fastly:compute/http-body.read": async | tracing,
            "fastly:compute/backend.register-dynamic-backend": async | tracing,
            "fastly:compute/async-io.select": async | tracing | trappable,
            "fastly:compute/async-io.select-with-timeout": async | tracing,
            "fastly:compute/http-req.send": async | tracing,
            "fastly:compute/http-req.send-async": async | tracing,
            "fastly:compute/http-req.send-async-streaming": async | tracing,
            "fastly:compute/http-req.send-async-uncached": async | tracing,
            "fastly:compute/http-req.send-async-uncached-streaming": async | tracing,
            "fastly:compute/http-req.send-uncached": async | tracing,
            "fastly:compute/cache.[static]entry.lookup": async | tracing,
            "fastly:compute/cache.[static]entry.transaction-lookup": async | tracing,
            "fastly:compute/cache.[static]entry.transaction-lookup-async": async | tracing,
            "fastly:compute/http-body.write": async | tracing,
            "fastly:compute/http-body.write-front": async | tracing,

            // Match the `wasmtime-wasi` crate's bindings.
            "wasi:io/streams.[method]output-stream.write": tracing | trappable,
            "wasi:io/streams.[method]output-stream.blocking-write-and-flush": async | tracing | trappable,
            "wasi:io/streams.[method]output-stream.flush": tracing | trappable,
            "wasi:io/streams.[method]output-stream.blocking-flush": async | tracing | trappable,
            "wasi:io/streams.[method]output-stream.check-write": tracing | trappable,
            "wasi:io/streams.[method]output-stream.write-zeroes": tracing | trappable,
            "wasi:io/streams.[method]output-stream.blocking-write-zeroes-and-flush": async | tracing | trappable,
            "wasi:io/streams.[method]output-stream.splice": tracing | trappable,
            "wasi:io/streams.[method]output-stream.blocking-splice": async | tracing | trappable,
            "wasi:io/streams.[method]input-stream.read": tracing | trappable,
            "wasi:io/streams.[method]input-stream.blocking-read": async | tracing | trappable,
            "wasi:io/streams.[method]input-stream.skip": tracing | trappable,
            "wasi:io/streams.[method]input-stream.blocking-skip": async | tracing | trappable,
            "wasi:io/poll.poll": async | tracing,
            "wasi:io/poll.[method]pollable.block": async | tracing,
            "wasi:cli/exit.exit": tracing | trappable,
            "wasi:cli/exit.exit-with-code": tracing | trappable,
        },
        exports: {
            default: tracing,
            "fastly:compute/http-incoming.handle": async | tracing,
        },
        with: {
            // Match the `wasmtime-wasi` crate's bindings.
            "wasi:io/poll.pollable": wasmtime_wasi::p2::DynPollable,
            "wasi:io/streams.input-stream": wasmtime_wasi::p2::DynInputStream,
            "wasi:io/streams.output-stream": wasmtime_wasi::p2::DynOutputStream,
            "wasi:io/error.error": wasmtime_wasi_io::streams::Error,

            "fastly:compute/kv-store.entry": super::compute::kv_store::Entry,
            "fastly:compute/backend.dynamic-backend-options": super::compute::backend::BackendBuilder,
            "fastly:compute/backend.backend": String,
            "fastly:compute/erl.rate-counter": String,
            "fastly:compute/erl.penalty-box": String,
            "fastly:compute/http-downstream.extra-bot-category": super::compute::http_downstream::ExtraBotCategory,
        },

        trappable_error_type: {
            // Match the wasmtime-wasi crate's bindings.
            "wasi:io/streams.stream-error" => wasmtime_wasi::p2::StreamError,
        },
    });
}

pub fn link_host_functions(linker: &mut component::Linker<ComponentCtx>) -> anyhow::Result<()> {
    let options = bindings::LinkOptions::default();

    // Add the Viceroy host implementations.
    bindings::AdapterService::add_to_linker::<_, HasSelf<_>>(linker, &options, |x| x)?;

    Ok(())
}

pub mod adapter;
pub mod backend;
pub mod compute;
pub mod erl;
pub mod handles;
pub mod http_req;
pub mod image_optimizer;
pub mod shielding;
pub mod wasi;