Hotfixed telemetry feed for post breach actions

This commit is contained in:
VakarisZ 2019-05-30 21:04:30 +03:00
parent 9f0e3c8513
commit 053f9686c6
1 changed files with 5 additions and 1 deletions

View File

@ -1,5 +1,6 @@
import logging
import subprocess
import socket
from infection_monkey.control import ControlClient
from infection_monkey.utils import is_windows_os
from infection_monkey.config import WormConfiguration
@ -44,9 +45,12 @@ class PBA(object):
"""
exec_funct = self._execute_default
result = exec_funct()
hostname = socket.gethostname()
ControlClient.send_telemetry('post_breach', {'command': self.command,
'result': result,
'name': self.name})
'name': self.name,
'hostname': hostname,
'ip': socket.gethostbyname(hostname)})
def _execute_default(self):
"""