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

Merge pull request #251 from VakarisZ/wmi_handler_fix 

wmi info handler does not crash if no wmi info is collected
This commit is contained in:
Daniel Goldberg 2019-01-28 18:23:16 +02:00 committed by GitHub
parent 38276f4abb 072677ac10
commit bb30e37ef1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/monkey_island/cc/services/wmi_handler.py
View File

@ -13,6 +13,13 @@ class WMIHandler(object):
self.monkey_id = monkey_id self.monkey_id = monkey_id
self.info_for_mongo = {} self.info_for_mongo = {}
self.users_secrets = user_secrets self.users_secrets = user_secrets
if not wmi_info:
self.users_info = ""
self.groups_info = ""
self.groups_and_users = ""
self.services = ""
self.products = ""
else:
self.users_info = wmi_info['Win32_UserAccount'] self.users_info = wmi_info['Win32_UserAccount']
self.groups_info = wmi_info['Win32_Group'] self.groups_info = wmi_info['Win32_Group']
self.groups_and_users = wmi_info['Win32_GroupUser'] self.groups_and_users = wmi_info['Win32_GroupUser']
@ -25,6 +32,7 @@ class WMIHandler(object):
self.add_users_to_collection() self.add_users_to_collection()
self.create_group_user_connection() self.create_group_user_connection()
self.insert_info_to_mongo() self.insert_info_to_mongo()
if self.info_for_mongo:
self.add_admin(self.info_for_mongo[self.ADMINISTRATORS_GROUP_KNOWN_SID], self.monkey_id) self.add_admin(self.info_for_mongo[self.ADMINISTRATORS_GROUP_KNOWN_SID], self.monkey_id)
self.update_admins_retrospective() self.update_admins_retrospective()
self.update_critical_services() self.update_critical_services()