blob: 7cf1159f57db7b0307e2ed0602287574cf767ead (
plain)
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
|
use super::ClientEndpointId;
pub(crate) enum EndpointControlMessage {
HealthPong,
Snapshot(Box<crate::protocol::ClientShellSnapshot>),
Ignored,
}
pub(crate) fn decode_endpoint_control(
kind: &str,
data: &str,
) -> Result<EndpointControlMessage, String> {
if kind == crate::protocol::endpoint::HEALTH_PONG_KIND {
return Ok(EndpointControlMessage::HealthPong);
}
if kind == crate::protocol::endpoint::ENDPOINT_SNAPSHOT_KIND {
let snapshot = serde_json::from_str(data)
.map_err(|error| format!("invalid endpoint snapshot: {error}"))?;
return Ok(EndpointControlMessage::Snapshot(Box::new(snapshot)));
}
if kind.starts_with("shell.snapshot.") {
return Err(format!(
"unsupported mandatory endpoint snapshot codec {kind:?}"
));
}
Ok(EndpointControlMessage::Ignored)
}
pub(crate) fn protocol_failure_is_fatal(endpoint_id: &ClientEndpointId) -> bool {
endpoint_id.is_local()
}
#[cfg(test)]
mod tests {
use super::*;
use crate::client::endpoint::ProfileId;
#[test]
fn unknown_optional_controls_are_ignored() {
assert!(matches!(
decode_endpoint_control("future.optional", "not json").unwrap(),
EndpointControlMessage::Ignored
));
}
#[test]
fn unknown_snapshot_codecs_are_rejected() {
assert_eq!(
decode_endpoint_control("shell.snapshot.v2", "{}")
.err()
.as_deref(),
Some("unsupported mandatory endpoint snapshot codec \"shell.snapshot.v2\"")
);
}
#[test]
fn only_local_protocol_failures_end_the_client() {
let remote =
ClientEndpointId::Ssh(ProfileId::parse("0123456789abcdef0123456789abcdef").unwrap());
assert!(protocol_failure_is_fatal(&ClientEndpointId::Local));
assert!(!protocol_failure_is_fatal(&remote));
}
}
|
