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

use crate::component::bindings::fastly::compute::{shielding, types};
use crate::config::Backend;
use crate::error::Error;
use crate::session::Session;
use http::Uri;
use std::str::FromStr;
use wasmtime::component::{Resource, ResourceTable};

pub(crate) fn backend_for_shield(
    session: &mut Session,
    _table: &mut ResourceTable,
    name: &str,
    _options: Option<Resource<shielding::ShieldBackendOptions>>,
    max_len: u64,
) -> Result<String, types::Error> {
    let shield_uri = name;

    let Ok(uri) = Uri::from_str(shield_uri) else {
        return Err(Error::InvalidArgument.into());
    };

    let new_name = format!("******{uri}*****");
    let new_backend = Backend {
        uri,
        override_host: None,
        cert_host: None,
        use_sni: false,
        grpc: false,
        client_cert: None,
        ca_certs: Vec::new(),
        health: crate::config::BackendHealth::Unknown,
    };

    if !session.add_backend(&new_name, new_backend) {
        return Err(Error::BackendNameRegistryError(new_name).into());
    }

    let target_len = new_name.len() as u64;

    if target_len > max_len {
        return Err(Error::BufferLengthError {
            buf: "shielding_backend",
            len: "name.len()",
        }
        .into());
    }

    Ok(new_name)
}