Mirror/masscanned
1
0
Fork 0
mirror of https://github.com/ivre/masscanned.git synced 2025-10-02 06:38:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test to emphasis bug in SSH

Browse source
This commit is contained in:
_Frky 2022-04-06 15:34:16 +02:00
parent b465a7f78e
commit 43c5407622
2 changed files with 41 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

39
src/proto/mod.rs
View file

@ -177,6 +177,8 @@ mod tests {
use crate::logger::MetaLogger;
use crate::proto::ssh::SSH_SERVER_BANNER;
#[test]
fn test_proto_dispatch_stun() {
let mut client_info = ClientInfo::new();
@ -338,4 +340,41 @@ mod tests {
panic!("expected no answer, got one");
}
}
#[test]
fn test_proto_repl_ssh() {
/* ensure that SSH proto returns a banner after the client's banner
* but only once */
let mut client_info = ClientInfo::new();
let test_ip_addr = Ipv4Addr::new(3, 2, 1, 0);
client_info.ip.src = Some(IpAddr::V4(test_ip_addr));
client_info.port.src = Some(65000);
let masscanned_ip_addr = Ipv4Addr::new(0, 1, 2, 3);
let mut ips = HashSet::new();
ips.insert(IpAddr::V4(masscanned_ip_addr));
/* Construct masscanned context object */
let masscanned = Masscanned {
synack_key: [0, 0],
mac: MacAddr::from_str("00:11:22:33:44:55").expect("error parsing MAC address"),
iface: None,
ip_addresses: Some(&ips),
log: MetaLogger::new(),
};
/***** SEND BANNER A FIRST TIME *****/
let payload = b"SSH-2.0-PUTTY";
match repl(&payload.to_vec(), &masscanned, &mut client_info, None) {
None => { panic!("expected an answer, got nothing") }
Some(banner) => { if banner != SSH_SERVER_BANNER { panic!("unexpected banner: {:?}", banner); } }
}
/***** SEND ONE ADDITIONAL BYTE *****/
let payload = b"X";
if let Some(banner) = repl(&payload.to_vec(), &masscanned, &mut client_info, None) {
panic!("unexpected banner: {:?}", banner);
}
/***** SEND A SECOND BANNER *****/
let payload = b"SSH-2.0-PUTTY";
if let Some(banner) = repl(&payload.to_vec(), &masscanned, &mut client_info, None) {
panic!("unexpected banner: {:?}", banner);
}
}
}

3
src/proto/ssh.rs
View file

@ -23,6 +23,7 @@ use crate::proto::TCPControlBlock;
use crate::Masscanned;
pub const SSH_PATTERN_CLIENT_PROTOCOL: &[u8; 7] = b"SSH-2.0";
pub const SSH_SERVER_BANNER: &[u8; 11] = b"SSH-2.0-1\r\n";
pub fn repl<'a>(
data: &'a [u8],
@ -31,7 +32,7 @@ pub fn repl<'a>(
_tcb: Option<&mut TCPControlBlock>,
) -> Option<Vec<u8>> {
debug!("receiving SSH data");
let repl_data = b"SSH-2.0-1\r\n".to_vec();
let repl_data = SSH_SERVER_BANNER.to_vec();
debug!("sending SSH answer");
warn!(
"SSH server banner to {}",