krata 0.0.21

Client library and common services for the krata isolation engine
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

use std::{fmt::Display, str::FromStr};

use anyhow::anyhow;
use url::{Host, Url};

pub const KRATA_DEFAULT_TCP_PORT: u16 = 4350;
pub const KRATA_DEFAULT_TLS_PORT: u16 = 4353;

#[derive(Clone)]
pub enum ControlDialAddress {
    UnixSocket {
        path: String,
    },
    Tcp {
        host: String,
        port: u16,
    },
    Tls {
        host: String,
        port: u16,
        insecure: bool,
    },
}

impl FromStr for ControlDialAddress {
    type Err = anyhow::Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let url: Url = s.parse()?;

        let host = url.host().unwrap_or(Host::Domain("localhost")).to_string();

        match url.scheme() {
            "unix" => Ok(ControlDialAddress::UnixSocket {
                path: url.path().to_string(),
            }),

            "tcp" => {
                let port = url.port().unwrap_or(KRATA_DEFAULT_TCP_PORT);
                Ok(ControlDialAddress::Tcp { host, port })
            }

            "tls" | "tls-insecure" => {
                let insecure = url.scheme() == "tls-insecure";
                let port = url.port().unwrap_or(KRATA_DEFAULT_TLS_PORT);
                Ok(ControlDialAddress::Tls {
                    host,
                    port,
                    insecure,
                })
            }

            _ => Err(anyhow!("unknown control address scheme: {}", url.scheme())),
        }
    }
}

impl From<ControlDialAddress> for Url {
    fn from(val: ControlDialAddress) -> Self {
        match val {
            ControlDialAddress::UnixSocket { path } => {
                let mut url = Url::parse("unix:///").unwrap();
                url.set_path(&path);
                url
            }

            ControlDialAddress::Tcp { host, port } => {
                let mut url = Url::parse("tcp://").unwrap();
                url.set_host(Some(&host)).unwrap();
                if port != KRATA_DEFAULT_TCP_PORT {
                    url.set_port(Some(port)).unwrap();
                }
                url
            }

            ControlDialAddress::Tls {
                host,
                port,
                insecure,
            } => {
                let mut url = Url::parse("tls://").unwrap();
                if insecure {
                    url.set_scheme("tls-insecure").unwrap();
                }
                url.set_host(Some(&host)).unwrap();
                if port != KRATA_DEFAULT_TLS_PORT {
                    url.set_port(Some(port)).unwrap();
                }
                url
            }
        }
    }
}

impl Display for ControlDialAddress {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let url: Url = self.clone().into();
        write!(f, "{}", url)
    }
}