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

Raise exception instead of using boolean

This commit is contained in:
Shreya 2020-10-25 16:41:50 +05:30
parent fc9d21201f
commit b3e9695289
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
monkey/infection_monkey/network/windowsserver_fingerprint.py
View File

@ -39,7 +39,6 @@ class WindowsServerFinger(HostFinger):
"""
Checks if the Windows Server is vulnerable to Zerologon.
"""
unexpected_error_encountered = False
def try_zero_authenticate(DC_HANDLE, DC_IP, DC_NAME):
# Connect to the DC's Netlogon service.
@ -75,12 +74,10 @@ class WindowsServerFinger(HostFinger):
if ex.get_error_code() == 0xc0000022: # STATUS_ACCESS_DENIED error; if not this, probably some other issue.
pass
else:
LOG.error(f'Unexpected error code: {ex.get_error_code()}.')
unexpected_error_encountered = True
raise Exception(f'Unexpected error code: {ex.get_error_code()}.')
except BaseException as ex:
LOG.error(f'Unexpected error: {ex}.')
unexpected_error_encountered = True
raise Exception(f'Unexpected error: {ex}.')
DC_IP = host.ip_addr
DC_NAME = self.get_dc_name(DC_IP)
@ -93,8 +90,12 @@ class WindowsServerFinger(HostFinger):
LOG.info('Performing Zerologon authentication attempts...')
rpc_con = None
for _ in range(0, self.MAX_ATTEMPTS):
rpc_con = try_zero_authenticate(DC_HANDLE, DC_IP, DC_NAME)
if (rpc_con is not None) or (unexpected_error_encountered):
try:
rpc_con = try_zero_authenticate(DC_HANDLE, DC_IP, DC_NAME)
if rpc_con is not None:
break
except Exception as ex:
LOG.info(ex)
break
self.init_service(host.services, self._SCANNED_SERVICE, '')