pupyps: add exe to unsafe fields

This commit is contained in:
Oleksii Shevchuk 2019-05-27 11:22:12 +03:00
parent c423b69905
commit 37a7145090
1 changed files with 15 additions and 9 deletions

View File

@ -60,16 +60,22 @@ def to_unicode(x):
def safe_as_dict(p, data): def safe_as_dict(p, data):
try: removed = set()
return p.as_dict(data)
except:
data = list(data)
if 'cmdline' in data:
data.remove('cmdline')
result = p.as_dict(data) for unsafe in (None, 'cmdline', 'exe'):
result['cmdline'] = None if unsafe is not None and unsafe in data:
return result data = list(data)
data.remove(unsafe)
removed.add(unsafe)
try:
result = p.as_dict(data)
for item in removed:
result[item] = None
return result
except WindowsError:
pass
def psinfo(pids): def psinfo(pids):