feat: added ip to sysinfo

This commit is contained in:
Andrew Rioux 2023-09-04 01:30:01 -04:00
parent 7f1f43a601
commit fb43a27ba3
Signed by: andrew.rioux
GPG Key ID: 9B8BAC47C17ABB94
3 changed files with 11 additions and 4 deletions

View File

@ -1,6 +1,8 @@
use std::net::IpAddr;
use sparse_05_common::messages::{Capabilities, OperatingSystem}; use sparse_05_common::messages::{Capabilities, OperatingSystem};
pub fn print_capabilities(capabilities: &Capabilities) { pub fn print_capabilities(capabilities: &Capabilities, ip: &IpAddr) {
use ansi_term::Colour as Color; use ansi_term::Colour as Color;
println!("Capabilities of remote host:"); println!("Capabilities of remote host:");
@ -65,7 +67,8 @@ pub fn print_capabilities(capabilities: &Capabilities) {
} }
println!("\tTransport type: \t{:?}", capabilities.transport); println!("\tTransport type: \t{:?}", capabilities.transport);
println!( println!(
"\tHost name: \t\t{}", "\tHost name (IP): \t{} ({})",
capabilities.hostname.as_deref().unwrap_or("<unknown>") capabilities.hostname.as_deref().unwrap_or("<unknown>"),
ip.ip()
); );
} }

View File

@ -88,7 +88,7 @@ pub(super) async fn shell(
cmd_matcher!( cmd_matcher!(
cmd[..amount], cmd[..amount],
(_sysinfo, b"#sysinfo") => { (_sysinfo, b"#sysinfo") => {
commands::sysinfo::print_capabilities(&capabilities) commands::sysinfo::print_capabilities(&capabilities, &connection.ip.ip())
}, },
(_help, b"#help") => {}, (_help, b"#help") => {},
(_exit, b"#exit") => { (_exit, b"#exit") => {

View File

@ -223,6 +223,10 @@ fn handle_full_connection<F>(
where where
F: Fn(), F: Fn(),
{ {
loop {
let msg = packet_handler.recv()?;
}
close(); close();
Ok(()) Ok(())