Skip to content
Snippets Groups Projects
Commit 5b219aa2 authored by Sebastien Luttringer's avatar Sebastien Luttringer
Browse files

select_payloads take a list of pattern

parent 8c4d69b9
No related branches found
No related tags found
No related merge requests found
......@@ -760,7 +760,7 @@ class RepositoryManager(object):
if len(s) > 0:
out(s)
def select_payloads(self, pattern):
def select_payloads(self, patterns):
'''
Return a list of available payloads
'''
......@@ -775,20 +775,19 @@ class RepositoryManager(object):
else:
paylist[md5]["images"].update(info["images"])
# check if pattern is md5 startpath
if pattern is not None:
ans = {}
for pattern in patterns:
for md5 in paylist.keys():
if not md5.startswith(pattern):
del paylist[md5]
return paylist
if md5.startswith(pattern):
ans[md5] = paylist[md5]
return ans
def show_payloads(self, patterns, o_images=False, o_json=False):
'''
Show payloads inside manager
'''
# get payload list
payloads = {}
for pattern in patterns:
payloads.update(self.select_payloads(pattern))
payloads = self.select_payloads(patterns)
# display result
if o_json:
s = json.dumps(payloads)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment