fix: renamed proto field to transport and setting protocol from banne…

…r grabber
evilsocket · Feb 15, 2024 · b007e43 · b007e43
1 parent f06e3f9
commit b007e43
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src/plugins/tcp_ports/grabbers/http.rs b/src/plugins/tcp_ports/grabbers/http.rs
@@ -46,6 +46,16 @@ pub(crate) async fn http_grabber(
     timeout: Duration,
 ) -> Banner {
     let mut banner = Banner::default();
+
+    banner.insert(
+        "protocol".to_owned(),
+        if ssl {
+            "https".to_owned()
+        } else {
+            "http".to_owned()
+        },
+    );
+
     let url = format!(
         "{}://{}:{}/",
         if ssl { "https" } else { "http" },

diff --git a/src/plugins/tcp_ports/mod.rs b/src/plugins/tcp_ports/mod.rs
@@ -77,7 +77,7 @@ impl Plugin for TcpPortScanner {
             crate::utils::net::async_tcp_stream(&address, timeout, false).await
         {
             let mut data = vec![
-                ("proto".to_owned(), "tcp".to_owned()),
+                ("transport".to_owned(), "tcp".to_owned()),
                 ("port".to_owned(), creds.username.to_owned()),
                 ("time".to_owned(), format!("{:?}", start.elapsed())),
             ];
@@ -93,7 +93,11 @@ impl Plugin for TcpPortScanner {
                 .await;
 
                 for (key, val) in banner {
-                    data.push((format!("banner.{}", key), val));
+                    if key == "proto" || key == "protocol" {
+                        data.push(("protocol".to_owned(), val));
+                    } else {
+                        data.push((format!("banner.{}", key), val));
+                    }
                 }
             }