Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Don't fail on invalid scriptlet name

This commit is contained in:
Oleksii Shevchuk 2017-04-25 15:54:28 +03:00
parent 0790f66665
commit 91bc695888
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pupy/pupygen.py
View File

@ -322,8 +322,11 @@ def parse_scriptlets(args_scriptlet, debug=False):
raise ValueError("usage: pupygen ... -s %s,arg1=value,arg2=value,..."%name)
if name not in scriptlets_dic:
print(colorize("[-] ","red")+"unknown scriptlet %s, valid choices are : %s"%(repr(name), [x for x in scriptlets_dic.iterkeys()]))
exit(1)
raise ValueError("unknown scriptlet %s, valid choices are : %s"%(
repr(name), [
x for x in scriptlets_dic.iterkeys()
]))
print colorize("[+] ","green")+"loading scriptlet %s with args %s"%(repr(name), sc_args)
try:
sp.add_scriptlet(scriptlets_dic[name](**sc_args))