Fix stale client rejection and terminal cursor mode

src/bin/dosh-client.rs

@@ -9,8 +9,8 @@ use dosh::auth::{
 use dosh::config::{expand_tilde, load_client_config, load_server_config};
 use dosh::crypto;
 use dosh::protocol::{
-    self, AttachOk, BootstrapAttachRequest, CLIENT_TO_SERVER, Frame, Input, PacketKind, Resize,
-    ResumeRequest, SERVER_TO_CLIENT, TicketAttachBody, TicketAttachEnvelope,
+    self, AttachOk, AttachReject, BootstrapAttachRequest, CLIENT_TO_SERVER, Frame, Input,
+    PacketKind, Resize, ResumeRequest, SERVER_TO_CLIENT, TicketAttachBody, TicketAttachEnvelope,
     TicketAttachOkEnvelope,
 };
 use serde::{Deserialize, Serialize};
@@ -633,6 +633,10 @@ async fn run_terminal(
                         }
                     }
                     PacketKind::Pong => {}
+                    PacketKind::AttachReject => {
+                        let reject: AttachReject = protocol::from_body(&packet.body)?;
+                        return Err(anyhow!("server rejected session: {}", reject.reason));
+                    }
                     _ => {}
                 }
             }
@@ -773,7 +777,7 @@ impl RawMode {
     fn enter() -> Result<Self> {
         enable_raw_mode()?;
         let mut stdout = std::io::stdout();
-        stdout.write_all(TERMINAL_CLEANUP)?;
+        stdout.write_all(TERMINAL_OPEN)?;
         stdout.flush()?;
         Ok(Self)
     }
@@ -788,6 +792,23 @@ impl Drop for RawMode {
     }
 }
 
+const TERMINAL_OPEN: &[u8] = concat!(
+    "\x1b[?1l",
+    "\x1b[0m",
+    "\x1b[?25h",
+    "\x1b[?1003l",
+    "\x1b[?1002l",
+    "\x1b[?1001l",
+    "\x1b[?1000l",
+    "\x1b[?1015l",
+    "\x1b[?1006l",
+    "\x1b[?1005l",
+    "\x1b[?2004l",
+    "\x1b[?1049l",
+    "\x1b[?1h"
+)
+.as_bytes();
+
 const TERMINAL_CLEANUP: &[u8] = concat!(
     "\x1b[?1l",
     "\x1b[0m",

src/bin/dosh-server.rs

@@ -226,6 +226,11 @@ async fn handle_packet(
             handle_ticket_attach(state, socket, peer, packet.body).await
         }
         PacketKind::ResumeRequest => handle_resume(state, socket, peer, &packet).await,
+        PacketKind::Input | PacketKind::Resize | PacketKind::Ping | PacketKind::Ack
+            if find_client_key(state, &packet.header.conn_id).is_err() =>
+        {
+            send_reject_to_client(socket, peer, packet.header.conn_id, "unknown client").await
+        }
         PacketKind::Input => handle_input(state, peer, &packet).await,
         PacketKind::Resize => handle_resize(state, peer, &packet).await,
         PacketKind::Ping => handle_ping(state, socket, peer, &packet).await,
@@ -425,10 +430,19 @@ async fn handle_ticket_attach(
 }
 
 async fn send_reject(socket: &UdpSocket, peer: SocketAddr, reason: &str) -> Result<()> {
+    send_reject_to_client(socket, peer, [0u8; 16], reason).await
+}
+
+async fn send_reject_to_client(
+    socket: &UdpSocket,
+    peer: SocketAddr,
+    client_id: [u8; 16],
+    reason: &str,
+) -> Result<()> {
     let body = protocol::to_body(&AttachReject {
         reason: reason.to_string(),
     })?;
-    let out = protocol::encode_plain(PacketKind::AttachReject, [0u8; 16], 0, 0, &body)?;
+    let out = protocol::encode_plain(PacketKind::AttachReject, client_id, 0, 0, &body)?;
     socket.send_to(&out, peer).await?;
     Ok(())
 }