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

wmi info handler does not crash if no wmi info is collected

This commit is contained in:
VakarisZ 2019-01-28 18:21:04 +02:00
parent ef0ccc9cc9
commit 072677ac10
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -13,11 +13,18 @@ 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
self.users_info = wmi_info['Win32_UserAccount'] if not wmi_info:
self.groups_info = wmi_info['Win32_Group'] self.users_info = ""
self.groups_and_users = wmi_info['Win32_GroupUser'] self.groups_info = ""
self.services = wmi_info['Win32_Service'] self.groups_and_users = ""
self.products = wmi_info['Win32_Product'] self.services = ""
self.products = ""
else:
self.users_info = wmi_info['Win32_UserAccount']
self.groups_info = wmi_info['Win32_Group']
self.groups_and_users = wmi_info['Win32_GroupUser']
self.services = wmi_info['Win32_Service']
self.products = wmi_info['Win32_Product']
def process_and_handle_wmi_info(self): def process_and_handle_wmi_info(self):
@ -25,7 +32,8 @@ 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()
self.add_admin(self.info_for_mongo[self.ADMINISTRATORS_GROUP_KNOWN_SID], self.monkey_id) if self.info_for_mongo:
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()