p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

PBA telem - Add fallback to ip & hostname collection

This commit is contained in:
itay 2019-07-07 12:07:04 +03:00
parent a9ec2a39bc
commit 6aca7d6f29
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
monkey/infection_monkey/telemetry/post_breach_telem.py
View File

@ -16,8 +16,7 @@ class PostBreachTelem(BaseTelem):
super(PostBreachTelem, self).__init__() super(PostBreachTelem, self).__init__()
self.pba = pba self.pba = pba
self.result = result self.result = result
self.hostname = socket.gethostname() self.hostname, self.ip = PostBreachTelem._get_hostname_and_ip()
self.ip = socket.gethostbyname(self.hostname)
telem_category = 'post_breach' telem_category = 'post_breach'
@ -29,3 +28,13 @@ class PostBreachTelem(BaseTelem):
'hostname': self.hostname, 'hostname': self.hostname,
'ip': self.ip 'ip': self.ip
} }
@staticmethod
def _get_hostname_and_ip():
try:
hostname = socket.gethostname()
ip = socket.gethostbyname(hostname)
except socket.error:
hostname = "Unknown"
ip = "Unknown"
return hostname, ip