updating portscan PR to capture all IPs

MythicAgents · Nov 13, 2024 · 740126f · 740126f
1 parent 35a2ffa
commit 740126f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/Payload_Type/poseidon/poseidon/agent_code/portscan/scanutils.go b/Payload_Type/poseidon/poseidon/agent_code/portscan/scanutils.go
@@ -43,6 +43,7 @@ func NewHost(hostName string) (*host, error) {
 	if net.ParseIP(hostName) != nil {
 		return &host{
 			IP:         hostName,
+			Hostname:   hostName,
 			PrettyName: hostName,
 			mutex:      mtx,
 			lock:       semaphore.NewWeighted(100), // yeah i hardcoded don't @me
@@ -53,8 +54,7 @@ func NewHost(hostName string) (*host, error) {
 		if err != nil {
 			return nil, err
 		}
-		hostStr := fmt.Sprintf("%s (%s)", ips[0], hostName)
-
+		hostStr := strings.Join(ips, "\n")
 		return &host{
 			IP:         ips[0],
 			Hostname:   hostName,

diff --git a/Payload_Type/poseidon/poseidon/browserscripts/portscan_new.js b/Payload_Type/poseidon/poseidon/browserscripts/portscan_new.js
@@ -1,9 +1,8 @@
 function(task, response){
 	let tables = [];
 	let headers = [
+			{"plaintext": "ips", "type": "button", "width": 80, "disableSort": true},
             {"plaintext": "hostname", "type": "string", "fillWidth": true, "disableSort": true},
-			{"plaintext": "ip", "type": "string", "fillWidth": true, "disableSort": true},
-            {"plaintext": "pretty name", "type": "string", "fillWidth": true, "disableSort": true},
 			{"plaintext": "open ports", "type": "string", "fillWidth": true,"disableSort": true}
 
         ];
@@ -16,9 +15,18 @@ function(task, response){
 			for(let j = 0; j < data.length; j++){
 				let rows = [];
 				for(let k = 0; k < data[j]["hosts"].length; k++){
+					if(data[j]["hosts"][k]["open_ports"] === null){continue}
 					rows.push({
 						"hostname": {"plaintext": data[j]["hosts"][k]['hostname']},
-						"ip": {"plaintext": data[j]["hosts"][k]["ip"]},
+						"ips": {
+							"button": {
+								"name": "View IPs",
+								"type": "string",
+								"value": data[j]["hosts"][k]["pretty_name"],
+								"hoverText": "View all IPs",
+								"title": "All IPs associated with this host",
+							}
+						},
 						"pretty name": {"plaintext":data[j]["hosts"][k]["pretty_name"]},
 						"open ports": {"plaintext": JSON.stringify(data[j]["hosts"][k]["open_ports"])}
 					});