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): ...@@ -760,7 +760,7 @@ class RepositoryManager(object):
if len(s) > 0: if len(s) > 0:
out(s) out(s)
def select_payloads(self, pattern): def select_payloads(self, patterns):
''' '''
Return a list of available payloads Return a list of available payloads
''' '''
...@@ -775,20 +775,19 @@ class RepositoryManager(object): ...@@ -775,20 +775,19 @@ class RepositoryManager(object):
else: else:
paylist[md5]["images"].update(info["images"]) paylist[md5]["images"].update(info["images"])
# check if pattern is md5 startpath # check if pattern is md5 startpath
if pattern is not None: ans = {}
for pattern in patterns:
for md5 in paylist.keys(): for md5 in paylist.keys():
if not md5.startswith(pattern): if md5.startswith(pattern):
del paylist[md5] ans[md5] = paylist[md5]
return paylist return ans
def show_payloads(self, patterns, o_images=False, o_json=False): def show_payloads(self, patterns, o_images=False, o_json=False):
''' '''
Show payloads inside manager Show payloads inside manager
''' '''
# get payload list # get payload list
payloads = {} payloads = self.select_payloads(patterns)
for pattern in patterns:
payloads.update(self.select_payloads(pattern))
# display result # display result
if o_json: if o_json:
s = json.dumps(payloads) 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