stuff

ssh-transport/src/client.rs

@@ -19,6 +19,8 @@ pub struct ClientConnection {
     rng: Box<dyn SshRng + Send + Sync>,
 
     plaintext_packets: VecDeque<Packet>,
+
+    pub abort_for_dos: bool,
 }
 
 enum ClientState {
@@ -67,6 +69,7 @@ impl ClientConnection {
             rng: Box::new(rng),
 
             plaintext_packets: VecDeque::new(),
+            abort_for_dos: false,
         }
     }
 
@@ -215,6 +218,10 @@ impl ClientConnection {
                         ));
                     }
 
+                    if self.abort_for_dos {
+                        return Err(peer_error!("early abort"));
+                    }
+
                     let server_hostkey = dh.string()?;
                     let server_ephermal_key = dh.string()?;
                     let signature = dh.string()?;