Merge pull request #335 from VakarisZ/post_breach_hotfix

Hotfixed telemetry feed for post breach actions
This commit is contained in:
VakarisZ 2019-05-30 21:08:03 +03:00 committed by GitHub
commit 016b73c982
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions

View File

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