Dreaded_X/siranga
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improved send

Browse Source
This commit is contained in:
Dreaded_X 2025-04-05 03:42:42 +02:00
parent 4a1c7f0279
commit c2eabed3ea
Signed by: Dreaded_X
GPG Key ID: FA5F485356B0D2D4
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/ssh.rs
View File

@ -22,8 +22,12 @@ pub struct Handler {
} }
impl Handler { impl Handler {
fn send(&self, data: &str) { fn send(&self, data: impl AsRef<str>) {
let _ = self.tx.send(data.as_bytes().to_vec()); let _ = self.tx.send(data.as_ref().as_bytes().to_vec());
}
fn sendln(&self, data: impl AsRef<str>) {
self.send(format!("{}\n\r", data.as_ref()));
} }
} }
@ -104,13 +108,13 @@ impl russh::server::Handler for Handler {
let tunnel = Tunnel::new(session.handle(), address, *port); let tunnel = Tunnel::new(session.handle(), address, *port);
let Some(address) = self.all_tunnels.add_tunnel(address, tunnel).await else { let Some(address) = self.all_tunnels.add_tunnel(address, tunnel).await else {
self.send(&format!("FAILED: ({address} already in use)\r\n")); self.sendln(format!("FAILED: ({address} already in use)"));
return Ok(false); return Ok(false);
}; };
// NOTE: The port we receive might not be the port that is getting forwarded from the // NOTE: The port we receive might not be the port that is getting forwarded from the
// client, we could include it in the message we send // client, we could include it in the message we send
self.send(&format!("http://{address}\r\n")); self.sendln(format!("http://{address}"));
self.tunnels.insert(address); self.tunnels.insert(address);
Ok(true) Ok(true)