p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

add more wmi classeS

This commit is contained in:
Oran Nadler 2018-04-10 17:51:18 +03:00
parent 826df43708
commit 21cf786d51
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
infection_monkey/system_info/windows_info_collector.py
View File

@ -21,9 +21,10 @@ WMI_CLASSES = set(["Win32_OperatingSystem",
"Win32_UserProfile",
"Win32_Group",
"Win32_GroupUser",
#"Win32_Product",
"Win32_Product",
"Win32_Service",
"Win32_OptionalFeature",
#"Win32_Process",
#"Win32_Service"
])
def fix_obj_for_mongo(o):
@ -131,7 +132,7 @@ class WindowsInfoCollector(InfoCollector):
key = _winreg.ConnectRegistry(None, store)
subkey = _winreg.OpenKey(key, subkey_path)
d = dict([_winreg.EnumValue(subkey, i)[:2] for i in xrange(_winreg.QueryInfoKey(subkey)[1])])
d = dict([_winreg.EnumValue(subkey, i)[:2] for i in xrange(_winreg.QueryInfoKey(subkey)[0])])
d = fix_obj_for_mongo(d)
self.info[subkey_path] = d