dig-capsule 0.5.0

The DIG Network .dig capsule data plane — one crate over the DIGS format, capsule read-crypto, compiler, staging, and the guest/host serve triad.
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

use crate::imp::core::abi::ErrorCode;
use crate::imp::core::config::HostImportsConfig;
use crate::imp::host::{ExecutionLimits, FixedClock, HostError, HostRuntime};

use super::common::test_deps;

fn cfg() -> HostImportsConfig {
    HostImportsConfig {
        return_buffer_capacity: 64 * 1024,
        max_return_buffer_size: 16 * 1024 * 1024,
        max_random_bytes: 1024,
        host_version: "dig-host-test/0.1".to_string(),
    }
}

fn echo_rt() -> HostRuntime {
    let module_bytes = wat::parse_str(include_str!(concat!(
        env!("CARGO_MANIFEST_DIR"),
        "/tests/fixtures/wat/serve_echo.wat"
    )))
    .unwrap();
    HostRuntime::new(
        &module_bytes,
        cfg(),
        ExecutionLimits::default(),
        test_deps(FixedClock::new(100)),
    )
    .unwrap()
}

#[test]
fn serve_content_round_trips_request_bytes() {
    let mut rt = echo_rt();
    let req = b"retrieval-key-and-root-and-range-bytes".to_vec();
    let out = rt.serve_content(&req).unwrap();
    assert_eq!(out, req);
}

#[test]
fn serve_proof_round_trips_request_bytes() {
    let mut rt = echo_rt();
    let req = vec![0xCDu8; 1024];
    let out = rt.serve_proof(&req).unwrap();
    assert_eq!(out, req);
}

#[test]
fn serve_content_empty_request_is_ok() {
    let mut rt = echo_rt();
    let out = rt.serve_content(&[]).unwrap();
    assert!(out.is_empty());
}

#[test]
fn serve_content_propagates_guest_error_sentinel() {
    let module_bytes = wat::parse_str(include_str!(concat!(
        env!("CARGO_MANIFEST_DIR"),
        "/tests/fixtures/wat/serve_err.wat"
    )))
    .unwrap();
    let mut rt = HostRuntime::new(
        &module_bytes,
        cfg(),
        ExecutionLimits::default(),
        test_deps(FixedClock::new(100)),
    )
    .unwrap();
    let err = rt.serve_content(b"anything").unwrap_err();
    assert!(
        matches!(err, HostError::GuestError(ErrorCode::NotFound)),
        "expected GuestError(NotFound), got {err:?}"
    );
}

#[test]
fn serve_proof_propagates_guest_error_sentinel() {
    let module_bytes = wat::parse_str(include_str!(concat!(
        env!("CARGO_MANIFEST_DIR"),
        "/tests/fixtures/wat/serve_err.wat"
    )))
    .unwrap();
    let mut rt = HostRuntime::new(
        &module_bytes,
        cfg(),
        ExecutionLimits::default(),
        test_deps(FixedClock::new(100)),
    )
    .unwrap();
    let err = rt.serve_proof(b"anything").unwrap_err();
    assert!(
        matches!(err, HostError::GuestError(ErrorCode::NotFound)),
        "expected GuestError(NotFound), got {err:?}"
    );
}