Compare commits

...

4 Commits

Author SHA1 Message Date
DuProcess fdbd58b628 Fetch remote server trace reports
ci / test (push) Has been cancelled
ci / fuzz-smoke (push) Has been cancelled
ci / windows-client (push) Has been cancelled
ci / package-release (linux-x86_64, ubuntu-latest) (push) Has been cancelled
ci / package-release (macos-aarch64, macos-14) (push) Has been cancelled
ci / package-release (macos-x86_64, macos-13) (push) Has been cancelled
ci / package-release (windows-x86_64, windows-latest) (push) Has been cancelled
ci / remote-bench (push) Has been cancelled
2026-07-12 11:55:23 -04:00
DuProcess c65aba9d7a Repaint terminal after local sleep
ci / test (push) Has been cancelled
ci / fuzz-smoke (push) Has been cancelled
ci / windows-client (push) Has been cancelled
ci / package-release (linux-x86_64, ubuntu-latest) (push) Has been cancelled
ci / package-release (macos-aarch64, macos-14) (push) Has been cancelled
ci / package-release (macos-x86_64, macos-13) (push) Has been cancelled
ci / package-release (windows-x86_64, windows-latest) (push) Has been cancelled
ci / remote-bench (push) Has been cancelled
2026-07-12 11:48:18 -04:00
DuProcess 818b481154 Cover terminal mouse input filtering
ci / test (push) Has been cancelled
ci / fuzz-smoke (push) Has been cancelled
ci / windows-client (push) Has been cancelled
ci / package-release (linux-x86_64, ubuntu-latest) (push) Has been cancelled
ci / package-release (macos-aarch64, macos-14) (push) Has been cancelled
ci / package-release (macos-x86_64, macos-13) (push) Has been cancelled
ci / package-release (windows-x86_64, windows-latest) (push) Has been cancelled
ci / remote-bench (push) Has been cancelled
2026-07-12 01:08:38 -04:00
DuProcess 70650e221b Flag trace terminal input anomalies
ci / test (push) Has been cancelled
ci / fuzz-smoke (push) Has been cancelled
ci / windows-client (push) Has been cancelled
ci / package-release (linux-x86_64, ubuntu-latest) (push) Has been cancelled
ci / package-release (macos-aarch64, macos-14) (push) Has been cancelled
ci / package-release (macos-x86_64, macos-13) (push) Has been cancelled
ci / package-release (windows-x86_64, windows-latest) (push) Has been cancelled
ci / remote-bench (push) Has been cancelled
2026-07-12 01:04:37 -04:00
4 changed files with 365 additions and 76 deletions
Generated
+1 -1
View File
@@ -436,7 +436,7 @@ dependencies = [
[[package]] [[package]]
name = "dosh" name = "dosh"
version = "1.0.0-rc31" version = "1.0.0-rc34"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"base64", "base64",
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "dosh" name = "dosh"
version = "1.0.0-rc31" version = "1.0.0-rc34"
edition = "2024" edition = "2024"
license = "MIT" license = "MIT"
+5 -4
View File
@@ -85,12 +85,13 @@ dosh trace --client-log /tmp/dosh-client.log HOST
Summarize collected traces with: Summarize collected traces with:
```sh ```sh
dosh trace report --client-log /tmp/dosh-client.log dosh trace report HOST --client-log /tmp/dosh-client.log
``` ```
Set `DOSH_TRACE=/tmp/dosh-server.log` on `dosh-server` for matching server When `HOST` is given, Dosh fetches `/tmp/dosh-server.log` over SSH before
events. Trace byte prefixes are enabled for `dosh trace`, so use it only for building the report. Set `DOSH_TRACE=/tmp/dosh-server.log` on `dosh-server` for
short reproductions. matching server events. Trace byte prefixes are enabled for `dosh trace`, so use
it only for short reproductions.
## VS Code ## VS Code
+358 -70
View File
@@ -74,6 +74,7 @@ const STALE_TERMINAL_INPUT_AFTER: Duration = Duration::from_secs(2);
const POST_RECONNECT_STALE_INPUT_GRACE: Duration = Duration::from_secs(2); const POST_RECONNECT_STALE_INPUT_GRACE: Duration = Duration::from_secs(2);
const FOCUS_REPAINT_COOLDOWN: Duration = Duration::from_secs(1); const FOCUS_REPAINT_COOLDOWN: Duration = Duration::from_secs(1);
const ALT_SCREEN_IDLE_REPAINT_AFTER: Duration = Duration::from_secs(15); const ALT_SCREEN_IDLE_REPAINT_AFTER: Duration = Duration::from_secs(15);
const LOCAL_SLEEP_REPAINT_AFTER: Duration = Duration::from_secs(5);
/// Sentinel `target_host` the server uses on a server-initiated `StreamOpen` that /// Sentinel `target_host` the server uses on a server-initiated `StreamOpen` that
/// represents an SSH-agent connection (rather than a TCP target). The client /// represents an SSH-agent connection (rather than a TCP target). The client
@@ -791,8 +792,9 @@ struct TraceOptions {
#[derive(Debug, Clone, PartialEq, Eq)] #[derive(Debug, Clone, PartialEq, Eq)]
struct TraceReportOptions { struct TraceReportOptions {
host: Option<String>,
client_log: Option<PathBuf>, client_log: Option<PathBuf>,
server_log: Option<PathBuf>, server_log: Option<String>,
tail: usize, tail: usize,
} }
@@ -809,6 +811,10 @@ struct TraceReport {
escape_events: usize, escape_events: usize,
hex_events: usize, hex_events: usize,
stripped_mouse_events: usize, stripped_mouse_events: usize,
client_mouseish_sent_events: usize,
server_mouseish_pty_write_events: usize,
server_rejected_input_events: usize,
replay_drop_events: usize,
queued_input_events: usize, queued_input_events: usize,
flushed_input_events: usize, flushed_input_events: usize,
sent_input_events: usize, sent_input_events: usize,
@@ -883,15 +889,16 @@ fn parse_trace_options(command: &[String]) -> Result<TraceOptions> {
} }
fn parse_trace_report_options(command: &[String]) -> Result<TraceReportOptions> { fn parse_trace_report_options(command: &[String]) -> Result<TraceReportOptions> {
let mut host: Option<String> = None;
let mut client_log: Option<PathBuf> = None; let mut client_log: Option<PathBuf> = None;
let mut server_log: Option<PathBuf> = Some(PathBuf::from("/tmp/dosh-server.log")); let mut server_log: Option<String> = Some("/tmp/dosh-server.log".to_string());
let mut tail = 20usize; let mut tail = 20usize;
let mut index = 0usize; let mut index = 0usize;
while index < command.len() { while index < command.len() {
match command[index].as_str() { match command[index].as_str() {
"--client-log" => { "--client-log" => {
let path = command.get(index + 1).ok_or_else(|| { let path = command.get(index + 1).ok_or_else(|| {
anyhow!("usage: dosh trace report [--client-log PATH] [--server-log PATH|none] [--tail N]") anyhow!("usage: dosh trace report [HOST] [--client-log PATH] [--server-log PATH|none] [--tail N]")
})?; })?;
client_log = Some(expand_tilde(path)); client_log = Some(expand_tilde(path));
index += 2; index += 2;
@@ -906,7 +913,7 @@ fn parse_trace_report_options(command: &[String]) -> Result<TraceReportOptions>
} }
"--server-log" => { "--server-log" => {
let path = command.get(index + 1).ok_or_else(|| { let path = command.get(index + 1).ok_or_else(|| {
anyhow!("usage: dosh trace report [--client-log PATH] [--server-log PATH|none] [--tail N]") anyhow!("usage: dosh trace report [HOST] [--client-log PATH] [--server-log PATH|none] [--tail N]")
})?; })?;
server_log = parse_optional_trace_path(path); server_log = parse_optional_trace_path(path);
index += 2; index += 2;
@@ -924,7 +931,7 @@ fn parse_trace_report_options(command: &[String]) -> Result<TraceReportOptions>
} }
"--tail" => { "--tail" => {
let value = command.get(index + 1).ok_or_else(|| { let value = command.get(index + 1).ok_or_else(|| {
anyhow!("usage: dosh trace report [--client-log PATH] [--server-log PATH|none] [--tail N]") anyhow!("usage: dosh trace report [HOST] [--client-log PATH] [--server-log PATH|none] [--tail N]")
})?; })?;
tail = parse_trace_tail(value)?; tail = parse_trace_tail(value)?;
index += 2; index += 2;
@@ -936,25 +943,35 @@ fn parse_trace_report_options(command: &[String]) -> Result<TraceReportOptions>
tail = parse_trace_tail(value)?; tail = parse_trace_tail(value)?;
index += 1; index += 1;
} }
value => { value if value.starts_with('-') => {
return Err(anyhow!( return Err(anyhow!(
"usage: dosh trace report [--client-log PATH] [--server-log PATH|none] [--tail N] (unknown option {value})" "usage: dosh trace report [HOST] [--client-log PATH] [--server-log PATH|none] [--tail N] (unknown option {value})"
)); ));
} }
value => {
if host.is_some() {
return Err(anyhow!(
"usage: dosh trace report [host] [--client-log PATH] [--server-log PATH|none] [--tail N]"
));
}
host = Some(value.to_string());
index += 1;
}
} }
} }
Ok(TraceReportOptions { Ok(TraceReportOptions {
host,
client_log, client_log,
server_log, server_log,
tail, tail,
}) })
} }
fn parse_optional_trace_path(value: &str) -> Option<PathBuf> { fn parse_optional_trace_path(value: &str) -> Option<String> {
if matches!(value, "" | "none" | "off" | "0") { if matches!(value, "" | "none" | "off" | "0") {
None None
} else { } else {
Some(expand_tilde(value)) Some(value.to_string())
} }
} }
@@ -1001,7 +1018,8 @@ fn sanitize_trace_name(value: &str) -> String {
fn run_trace_command(args: &Args) -> Result<()> { fn run_trace_command(args: &Args) -> Result<()> {
if args.command.first().is_some_and(|value| value == "report") { if args.command.first().is_some_and(|value| value == "report") {
return run_trace_report_command(&args.command[1..]); let config = load_client_config(None).unwrap_or_default();
return run_trace_report_command(&config, args, &args.command[1..]);
} }
let options = parse_trace_options(&args.command)?; let options = parse_trace_options(&args.command)?;
if let Some(parent) = options.client_log.parent() { if let Some(parent) = options.client_log.parent() {
@@ -1037,7 +1055,11 @@ fn run_trace_command(args: &Args) -> Result<()> {
Ok(()) Ok(())
} }
fn run_trace_report_command(command: &[String]) -> Result<()> { fn run_trace_report_command(
config: &dosh::config::ClientConfig,
args: &Args,
command: &[String],
) -> Result<()> {
let options = parse_trace_report_options(command)?; let options = parse_trace_report_options(command)?;
let client_log = match options.client_log { let client_log = match options.client_log {
Some(path) => Some(path), Some(path) => Some(path),
@@ -1056,6 +1078,20 @@ fn run_trace_report_command(command: &[String]) -> Result<()> {
); );
} }
if let Some(path) = options.server_log { if let Some(path) = options.server_log {
let server_log = if let Some(host) = options.host.as_deref() {
match fetch_trace_server_log(config, args, host, &path) {
Ok(path) => Some(path),
Err(err) => {
println!("[warn] server log {host}:{path}: {err:#}");
None
}
}
} else {
Some(expand_tilde(&path))
};
let Some(path) = server_log else {
return Ok(());
};
match summarize_trace_file(&path, options.tail) { match summarize_trace_file(&path, options.tail) {
Ok(report) => print_trace_report("server", &report), Ok(report) => print_trace_report("server", &report),
Err(err) Err(err)
@@ -1071,6 +1107,66 @@ fn run_trace_report_command(command: &[String]) -> Result<()> {
Ok(()) Ok(())
} }
fn fetch_trace_server_log(
config: &dosh::config::ClientConfig,
args: &Args,
requested: &str,
remote_log: &str,
) -> Result<PathBuf> {
let hosts = load_hosts_config(None).unwrap_or_default();
let host_config = hosts.hosts.get(requested).cloned().unwrap_or_default();
let server = status_ssh_target(requested, &host_config);
let ssh_port = args.ssh_port.or(host_config.ssh_port).or(config.ssh_port);
if is_local_status_target(&server) {
return Ok(expand_tilde(remote_log));
}
let local_path = fetched_trace_server_log_path(requested);
if let Some(parent) = local_path.parent() {
fs::create_dir_all(parent)
.with_context(|| format!("create trace log directory {}", parent.to_string_lossy()))?;
}
let mut command = Command::new("ssh");
if let Some(ssh_port) = ssh_port {
command.arg("-p").arg(ssh_port.to_string());
}
command.arg("-T");
if let Some(key) = args.ssh_key.as_deref() {
command.arg("-i").arg(key);
}
if let Some(known_hosts) = args.ssh_known_hosts.as_deref() {
command
.arg("-o")
.arg(format!("UserKnownHostsFile={}", known_hosts.display()));
}
let script = format!("test -r {0} && cat {0}", shell_word(remote_log));
let output = command
.arg(&server)
.arg(format!("sh -c {}", shell_word(&script)))
.output()
.with_context(|| format!("fetch server trace log from {server}"))?;
anyhow::ensure!(
output.status.success(),
"{}",
String::from_utf8_lossy(&output.stderr)
);
fs::write(&local_path, &output.stdout)
.with_context(|| format!("write fetched server trace {}", local_path.display()))?;
Ok(local_path)
}
fn fetched_trace_server_log_path(host: &str) -> PathBuf {
let now = SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|duration| duration.as_secs())
.unwrap_or(0);
trace_cache_dir().join(format!(
"server-{}-{}-{}.log",
sanitize_trace_name(host),
now,
std::process::id()
))
}
fn trace_cache_dir() -> PathBuf { fn trace_cache_dir() -> PathBuf {
dirs::cache_dir() dirs::cache_dir()
.unwrap_or_else(|| expand_tilde("~/.cache")) .unwrap_or_else(|| expand_tilde("~/.cache"))
@@ -1159,6 +1255,11 @@ fn summarize_trace_file(path: &Path, tail: usize) -> Result<TraceReport> {
if event.contains("mouse_stripped") || event == "client.stale_strip" { if event.contains("mouse_stripped") || event == "client.stale_strip" {
report.stripped_mouse_events += 1; report.stripped_mouse_events += 1;
} }
if event == "client.input_send"
&& summary.get("mouseish").is_some_and(|value| value == "true")
{
report.client_mouseish_sent_events += 1;
}
if event.starts_with("client.queue_") { if event.starts_with("client.queue_") {
report.queued_input_events += 1; report.queued_input_events += 1;
} }
@@ -1170,6 +1271,15 @@ fn summarize_trace_file(path: &Path, tail: usize) -> Result<TraceReport> {
} }
if event == "server.pty_write" { if event == "server.pty_write" {
report.pty_write_events += 1; report.pty_write_events += 1;
if summary.get("mouseish").is_some_and(|value| value == "true") {
report.server_mouseish_pty_write_events += 1;
}
}
if event == "server.input_rejected_mode" {
report.server_rejected_input_events += 1;
}
if event.contains("replay_drop") {
report.replay_drop_events += 1;
} }
if event.contains("reconnect") || event.contains("resume") { if event.contains("reconnect") || event.contains("resume") {
report.reconnect_events += 1; report.reconnect_events += 1;
@@ -1202,15 +1312,19 @@ fn print_trace_report(label: &str, report: &TraceReport) {
.unwrap_or_else(|| "unknown".to_string()) .unwrap_or_else(|| "unknown".to_string())
); );
println!( println!(
" input: sent={} queued={} flushed={} pty_writes={}", " input: sent={} queued={} flushed={} pty_writes={} rejected={} replay_drops={}",
report.sent_input_events, report.sent_input_events,
report.queued_input_events, report.queued_input_events,
report.flushed_input_events, report.flushed_input_events,
report.pty_write_events report.pty_write_events,
report.server_rejected_input_events,
report.replay_drop_events
); );
println!( println!(
" terminal: mouseish={} stripped={} focus={} esc={} hex={}", " terminal: mouseish={} client_mouse_sent={} server_mouse_pty={} stripped={} focus={} esc={} hex={}",
report.mouseish_events, report.mouseish_events,
report.client_mouseish_sent_events,
report.server_mouseish_pty_write_events,
report.stripped_mouse_events, report.stripped_mouse_events,
report.focus_events, report.focus_events,
report.escape_events, report.escape_events,
@@ -1220,6 +1334,9 @@ fn print_trace_report(label: &str, report: &TraceReport) {
" reconnect: events={} roam={}", " reconnect: events={} roam={}",
report.reconnect_events, report.roam_events report.reconnect_events, report.roam_events
); );
for warning in trace_report_warnings(report) {
println!(" alert: {warning}");
}
if !report.events.is_empty() { if !report.events.is_empty() {
println!(" top events:"); println!(" top events:");
for (event, count) in top_trace_events(&report.events, 8) { for (event, count) in top_trace_events(&report.events, 8) {
@@ -1234,6 +1351,38 @@ fn print_trace_report(label: &str, report: &TraceReport) {
} }
} }
fn trace_report_warnings(report: &TraceReport) -> Vec<String> {
let mut warnings = Vec::new();
if report.server_mouseish_pty_write_events > 0 {
warnings.push(format!(
"{} mouse-like input event(s) reached the server PTY",
report.server_mouseish_pty_write_events
));
}
if report.client_mouseish_sent_events > report.stripped_mouse_events
&& report.server_mouseish_pty_write_events == 0
{
warnings.push(format!(
"{} mouse-like input event(s) left the client; check the matching server log",
report.client_mouseish_sent_events
));
}
if report.replay_drop_events > 0 {
warnings.push(format!(
"{} replay/drop event(s) observed",
report.replay_drop_events
));
}
if report.reconnect_events > 0
&& report.flushed_input_events == 0
&& report.queued_input_events > 0
{
warnings
.push("queued input was observed around reconnect without a later flush".to_string());
}
warnings
}
fn top_trace_events(events: &BTreeMap<String, usize>, limit: usize) -> Vec<(String, usize)> { fn top_trace_events(events: &BTreeMap<String, usize>, limit: usize) -> Vec<(String, usize)> {
let mut entries: Vec<_> = events let mut entries: Vec<_> = events
.iter() .iter()
@@ -5449,6 +5598,7 @@ async fn run_terminal(
let mut last_terminal_frame_at = Instant::now(); let mut last_terminal_frame_at = Instant::now();
let mut last_focus_repaint_at = Instant::now() - FOCUS_REPAINT_COOLDOWN; let mut last_focus_repaint_at = Instant::now() - FOCUS_REPAINT_COOLDOWN;
let mut last_idle_repaint_attempt_at = Instant::now() - ALT_SCREEN_IDLE_REPAINT_AFTER; let mut last_idle_repaint_attempt_at = Instant::now() - ALT_SCREEN_IDLE_REPAINT_AFTER;
let mut last_status_tick_at = Instant::now();
if let Some(frame) = first_frame { if let Some(frame) = first_frame {
if !forward_only { if !forward_only {
render_frame(&frame)?; render_frame(&frame)?;
@@ -5577,22 +5727,22 @@ async fn run_terminal(
if bytes.is_empty() { if bytes.is_empty() {
continue; continue;
} }
if should_strip_unowned_terminal_reports( let before_mouse_strip = bytes.len();
let (stripped_bytes, stripped_unowned_mouse) = strip_unowned_terminal_reports(
bytes,
predictor.alternate_screen, predictor.alternate_screen,
predictor.mouse_tracking, predictor.mouse_tracking,
) { );
let before_mouse_strip = bytes.len(); bytes = stripped_bytes;
bytes = strip_stale_mouse_reports(&bytes); if stripped_unowned_mouse {
if before_mouse_strip != bytes.len() { dosh::trace::event(
dosh::trace::event( "client.unowned_mouse_stripped",
"client.unowned_mouse_stripped", &[
&[ ("before", before_mouse_strip.to_string()),
("before", before_mouse_strip.to_string()), ("after", bytes.len().to_string()),
("after", bytes.len().to_string()), ("summary", dosh::trace::bytes_summary(&bytes)),
("summary", dosh::trace::bytes_summary(&bytes)), ],
], );
);
}
if bytes.is_empty() { if bytes.is_empty() {
continue; continue;
} }
@@ -6431,6 +6581,10 @@ async fn run_terminal(
} }
} }
_ = status_tick.tick() => { _ = status_tick.tick() => {
let status_tick_at = Instant::now();
let status_tick_gap = status_tick_at.duration_since(last_status_tick_at);
last_status_tick_at = status_tick_at;
let mut repainted_this_tick = false;
let stale = last_packet_at.elapsed(); let stale = last_packet_at.elapsed();
if stale >= Duration::from_secs(reconnect_timeout_secs.max(1)) { if stale >= Duration::from_secs(reconnect_timeout_secs.max(1)) {
if let Some(frame) = reconnect( if let Some(frame) = reconnect(
@@ -6451,6 +6605,8 @@ async fn run_terminal(
render_frame(&frame)?; render_frame(&frame)?;
predictor.observe_output(&frame.bytes); predictor.observe_output(&frame.bytes);
last_terminal_frame_at = Instant::now(); last_terminal_frame_at = Instant::now();
last_idle_repaint_attempt_at = Instant::now();
repainted_this_tick = true;
} }
last_packet_at = Instant::now(); last_packet_at = Instant::now();
flush_pending_user_input( flush_pending_user_input(
@@ -6493,13 +6649,22 @@ async fn run_terminal(
) )
.await?; .await?;
let now = Instant::now(); let now = Instant::now();
if should_repaint_idle_alternate_screen( if !repainted_this_tick && should_repaint_idle_terminal(
predictor.alternate_screen, predictor.alternate_screen,
last_terminal_frame_at, last_terminal_frame_at,
last_idle_repaint_attempt_at, last_idle_repaint_attempt_at,
status_tick_gap,
now, now,
) { ) {
last_idle_repaint_attempt_at = now; last_idle_repaint_attempt_at = now;
dosh::trace::event(
"client.idle_repaint_start",
&[
("alt", predictor.alternate_screen.to_string()),
("tick_gap_ms", status_tick_gap.as_millis().to_string()),
("silent_ms", stale.as_millis().to_string()),
],
);
if let Some(frame) = reconnect( if let Some(frame) = reconnect(
&socket, &socket,
&mut cred, &mut cred,
@@ -6926,6 +7091,20 @@ fn should_strip_unowned_terminal_reports(alternate_screen: bool, mouse_tracking:
!alternate_screen && !mouse_tracking !alternate_screen && !mouse_tracking
} }
fn strip_unowned_terminal_reports(
bytes: Vec<u8>,
alternate_screen: bool,
mouse_tracking: bool,
) -> (Vec<u8>, bool) {
if !should_strip_unowned_terminal_reports(alternate_screen, mouse_tracking) {
return (bytes, false);
}
let before = bytes.len();
let stripped = strip_stale_mouse_reports(&bytes);
let changed = stripped.len() != before;
(stripped, changed)
}
fn input_contains_focus_in(bytes: &[u8]) -> bool { fn input_contains_focus_in(bytes: &[u8]) -> bool {
contains_bytes(bytes, b"\x1b[I") || contains_bytes(bytes, b"\x9bI") contains_bytes(bytes, b"\x1b[I") || contains_bytes(bytes, b"\x9bI")
} }
@@ -6950,14 +7129,16 @@ fn strip_terminal_focus_reports(bytes: &[u8]) -> Vec<u8> {
out out
} }
fn should_repaint_idle_alternate_screen( fn should_repaint_idle_terminal(
alternate_screen: bool, alternate_screen: bool,
last_terminal_frame_at: Instant, last_terminal_frame_at: Instant,
last_attempt_at: Instant, last_attempt_at: Instant,
status_tick_gap: Duration,
now: Instant, now: Instant,
) -> bool { ) -> bool {
alternate_screen (alternate_screen
&& now.duration_since(last_terminal_frame_at) >= ALT_SCREEN_IDLE_REPAINT_AFTER && now.duration_since(last_terminal_frame_at) >= ALT_SCREEN_IDLE_REPAINT_AFTER
|| status_tick_gap >= LOCAL_SLEEP_REPAINT_AFTER)
&& now.duration_since(last_attempt_at) >= ALT_SCREEN_IDLE_REPAINT_AFTER && now.duration_since(last_attempt_at) >= ALT_SCREEN_IDLE_REPAINT_AFTER
} }
@@ -8901,30 +9082,31 @@ const TERMINAL_CLEANUP: &[u8] = concat!(
mod tests { mod tests {
use super::{ use super::{
ALT_SCREEN_IDLE_REPAINT_AFTER, CachedCredential, DisconnectStatus, DynamicForward, ALT_SCREEN_IDLE_REPAINT_AFTER, CachedCredential, DisconnectStatus, DynamicForward,
FRAME_GAP_RESYNC_AFTER_MS, FrameBuffer, LocalForward, MAX_PENDING_USER_INPUT_BYTES, FRAME_GAP_RESYNC_AFTER_MS, FrameBuffer, LOCAL_SLEEP_REPAINT_AFTER, LocalForward,
NativeIdentityContext, POST_SUBMIT_ALL_INPUT_HOLD, PendingStreamOpen, PredictMode, MAX_PENDING_USER_INPUT_BYTES, NativeIdentityContext, POST_SUBMIT_ALL_INPUT_HOLD,
Predictor, RESTART_STATUS_SCRIPT, RemoteForward, STARTUP_INPUT_HOLD, SshConfig, PendingStreamOpen, PredictMode, Predictor, RESTART_STATUS_SCRIPT, RemoteForward,
SshPathTokenContext, StartupGateMode, StatusAction, UpdateOptions, UpdateRole, auth_allows, STARTUP_INPUT_HOLD, SshConfig, SshPathTokenContext, StartupGateMode, StatusAction,
cache_key, cache_server_prefix, cleanup_stream_state, clear_cached_credentials, UpdateOptions, UpdateRole, auth_allows, cache_key, cache_server_prefix,
ensure_tui_safe_status_overlay, expand_ssh_path_tokens, input_contains_focus_in, cleanup_stream_state, clear_cached_credentials, ensure_tui_safe_status_overlay,
input_matches_escape, is_local_status_target, is_resume_response_for_client, expand_ssh_path_tokens, input_contains_focus_in, input_matches_escape,
latest_release_download_url, load_first_native_identity_with_prompt, is_local_status_target, is_resume_response_for_client, latest_release_download_url,
native_proxy_udp_warning, parse_dynamic_forward, parse_escape_key, parse_local_forward, load_first_native_identity_with_prompt, native_proxy_udp_warning, parse_dynamic_forward,
parse_remote_forward, parse_ssh_config, parse_trace_line, parse_trace_options, parse_escape_key, parse_local_forward, parse_remote_forward, parse_ssh_config,
parse_trace_report_options, parse_trace_summary, parse_update_options, parse_trace_line, parse_trace_options, parse_trace_report_options, parse_trace_summary,
post_submit_hold_duration, queue_pending_user_input, raw_contains_host_table, parse_update_options, post_submit_hold_duration, queue_pending_user_input,
recv_response_until, refresh_live_addr, release_tag_download_url, queue_stale_pending_user_input, raw_contains_host_table, recv_response_until,
release_tag_from_effective_url, release_version_from_tag, render_status_clear, refresh_live_addr, release_tag_download_url, release_tag_from_effective_url,
render_status_overlay, requested_env, resolved_startup_command, retire_stream_state, release_version_from_tag, render_status_clear, render_status_overlay, requested_env,
retransmit_stream_opens, rewrite_forward_command, sanitize_trace_name, resolved_startup_command, retire_stream_state, retransmit_stream_opens,
selected_predict_mode, selected_udp_host, server_version_mismatch, rewrite_forward_command, sanitize_trace_name, selected_predict_mode, selected_udp_host,
should_flush_terminal_input_after_contact, should_hold_during_startup_gate, server_version_mismatch, should_flush_terminal_input_after_contact,
should_hold_post_submit_input, should_repaint_idle_alternate_screen, should_hold_during_startup_gate, should_hold_post_submit_input,
should_strip_unowned_terminal_reports, split_after_command_submit, split_trace_tokens, should_repaint_idle_terminal, should_strip_unowned_terminal_reports,
ssh_command_target, ssh_config_uses_proxy, ssh_destination_host, ssh_username, split_after_command_submit, split_trace_tokens, ssh_command_target, ssh_config_uses_proxy,
ssh_with_user, startup_command, status_ssh_target, strip_stale_mouse_reports, ssh_destination_host, ssh_username, ssh_with_user, startup_command, status_ssh_target,
strip_terminal_focus_reports, summarize_trace_file, terminal_private_mode_transition, strip_stale_mouse_reports, strip_terminal_focus_reports, strip_unowned_terminal_reports,
toml_bare_key_or_quoted, top_trace_events, unix_update_script, update_installer_url, summarize_trace_file, terminal_private_mode_transition, toml_bare_key_or_quoted,
top_trace_events, trace_report_warnings, unix_update_script, update_installer_url,
update_version_status, upsert_managed_block, valid_forward_host, vscode_safe_alias, update_version_status, upsert_managed_block, valid_forward_host, vscode_safe_alias,
windows_update_script, windows_update_script,
}; };
@@ -10233,27 +10415,30 @@ mod tests {
#[test] #[test]
fn trace_report_options_parse_defaults_and_overrides() { fn trace_report_options_parse_defaults_and_overrides() {
let parsed = parse_trace_report_options(&[]).unwrap(); let parsed = parse_trace_report_options(&[]).unwrap();
assert_eq!(parsed.host, None);
assert_eq!(parsed.client_log, None); assert_eq!(parsed.client_log, None);
assert_eq!( assert_eq!(parsed.server_log, Some("/tmp/dosh-server.log".to_string()));
parsed.server_log,
Some(std::path::PathBuf::from("/tmp/dosh-server.log"))
);
assert_eq!(parsed.tail, 20); assert_eq!(parsed.tail, 20);
let parsed = parse_trace_report_options(&[ let parsed = parse_trace_report_options(&[
"palav".to_string(),
"--client-log=/tmp/client.log".to_string(), "--client-log=/tmp/client.log".to_string(),
"--server-log".to_string(), "--server-log".to_string(),
"none".to_string(), "/tmp/server.log".to_string(),
"--tail".to_string(), "--tail".to_string(),
"3".to_string(), "3".to_string(),
]) ])
.unwrap(); .unwrap();
assert_eq!(parsed.host.as_deref(), Some("palav"));
assert_eq!( assert_eq!(
parsed.client_log, parsed.client_log,
Some(std::path::PathBuf::from("/tmp/client.log")) Some(std::path::PathBuf::from("/tmp/client.log"))
); );
assert_eq!(parsed.server_log, None); assert_eq!(parsed.server_log.as_deref(), Some("/tmp/server.log"));
assert_eq!(parsed.tail, 3); assert_eq!(parsed.tail, 3);
let parsed = parse_trace_report_options(&["--server-log=none".to_string()]).unwrap();
assert_eq!(parsed.server_log, None);
} }
#[test] #[test]
@@ -10318,11 +10503,50 @@ mod tests {
assert_eq!(report.queued_input_events, 1); assert_eq!(report.queued_input_events, 1);
assert_eq!(report.sent_input_events, 1); assert_eq!(report.sent_input_events, 1);
assert_eq!(report.reconnect_events, 1); assert_eq!(report.reconnect_events, 1);
assert_eq!(report.client_mouseish_sent_events, 0);
assert_eq!(report.server_mouseish_pty_write_events, 0);
assert_eq!(report.recent_events.len(), 2); assert_eq!(report.recent_events.len(), 2);
assert_eq!( assert_eq!(
top_trace_events(&report.events, 1), top_trace_events(&report.events, 1),
vec![("client.input_send".to_string(), 1)] vec![("client.input_send".to_string(), 1)]
); );
assert_eq!(
trace_report_warnings(&report),
vec!["queued input was observed around reconnect without a later flush".to_string()]
);
}
#[test]
fn trace_report_warns_when_mouse_input_reaches_server_pty() {
let dir = tempfile::tempdir().unwrap();
let path = dir.path().join("server.log");
fs::write(
&path,
concat!(
"ts_ms=30 pid=2 event=server.input_roam session=term from=1.1.1.1:1 to=2.2.2.2:2\n",
"ts_ms=31 pid=2 event=server.pty_write session=term seq=9 summary=len=12,esc=false,focus=false,mouseish=true,printable=12\n",
"ts_ms=32 pid=2 event=server.input_replay_drop session=term seq=8 summary=len=1,esc=false,focus=false,mouseish=false,printable=1\n",
),
)
.unwrap();
let report = summarize_trace_file(&path, 10).unwrap();
let warnings = trace_report_warnings(&report);
assert_eq!(report.pty_write_events, 1);
assert_eq!(report.server_mouseish_pty_write_events, 1);
assert_eq!(report.replay_drop_events, 1);
assert_eq!(report.roam_events, 1);
assert!(
warnings
.iter()
.any(|warning| { warning == "1 mouse-like input event(s) reached the server PTY" })
);
assert!(
warnings
.iter()
.any(|warning| { warning == "1 replay/drop event(s) observed" })
);
} }
#[test] #[test]
@@ -10721,6 +10945,47 @@ mod tests {
assert_eq!(pending_bytes, MAX_PENDING_USER_INPUT_BYTES); assert_eq!(pending_bytes, MAX_PENDING_USER_INPUT_BYTES);
} }
#[test]
fn stale_pending_user_input_is_marked_for_mouse_stripping() {
let mut pending = VecDeque::new();
let mut pending_bytes = 0usize;
queue_pending_user_input(&mut pending, &mut pending_bytes, b"\x1b[<35;10;1M".to_vec())
.unwrap();
queue_stale_pending_user_input(
&mut pending,
&mut pending_bytes,
b"\x1b[<35;11;1M".to_vec(),
)
.unwrap();
let normal = pending.pop_front().unwrap();
assert!(!normal.strip_mouse_reports);
assert_eq!(strip_stale_mouse_reports(&normal.bytes), b"");
let stale = pending.pop_front().unwrap();
assert!(stale.strip_mouse_reports);
assert_eq!(strip_stale_mouse_reports(&stale.bytes), b"");
}
#[test]
fn unowned_terminal_mouse_reports_strip_only_without_mouse_owner() {
let input = b"\x1b[<35;10;1Mcmd\r".to_vec();
let (stripped, changed) = strip_unowned_terminal_reports(input.clone(), false, false);
assert!(changed);
assert_eq!(stripped, b"cmd\r");
let (preserved, changed) = strip_unowned_terminal_reports(input.clone(), false, true);
assert!(!changed);
assert_eq!(preserved, input);
let (preserved, changed) =
strip_unowned_terminal_reports(b"\x1b[<35;10;1M".to_vec(), true, false);
assert!(!changed);
assert_eq!(preserved, b"\x1b[<35;10;1M");
}
#[test] #[test]
fn transient_udp_send_errors_are_not_terminal_fatal() { fn transient_udp_send_errors_are_not_terminal_fatal() {
#[cfg(unix)] #[cfg(unix)]
@@ -10828,21 +11093,44 @@ mod tests {
} }
#[test] #[test]
fn idle_repaint_only_runs_for_stale_alternate_screen() { fn idle_repaint_runs_for_stale_alternate_screen_or_sleep_gap() {
let now = Instant::now(); let now = Instant::now();
let stale = now - ALT_SCREEN_IDLE_REPAINT_AFTER - Duration::from_secs(1); let stale = now - ALT_SCREEN_IDLE_REPAINT_AFTER - Duration::from_secs(1);
let recent = now - Duration::from_secs(1); let recent = now - Duration::from_secs(1);
assert!(should_repaint_idle_alternate_screen( assert!(should_repaint_idle_terminal(
true, stale, stale, now true,
stale,
stale,
Duration::from_secs(1),
now
)); ));
assert!(!should_repaint_idle_alternate_screen( assert!(should_repaint_idle_terminal(
false, stale, stale, now false,
recent,
stale,
LOCAL_SLEEP_REPAINT_AFTER + Duration::from_secs(1),
now
)); ));
assert!(!should_repaint_idle_alternate_screen( assert!(!should_repaint_idle_terminal(
true, recent, stale, now false,
stale,
stale,
Duration::from_secs(1),
now
)); ));
assert!(!should_repaint_idle_alternate_screen( assert!(!should_repaint_idle_terminal(
true, stale, recent, now true,
recent,
stale,
Duration::from_secs(1),
now
));
assert!(!should_repaint_idle_terminal(
true,
stale,
recent,
LOCAL_SLEEP_REPAINT_AFTER + Duration::from_secs(1),
now
)); ));
} }