[shellcraft] Avoid recursive walk of all templates for command line

Fixes Gallopsled#1650
heapcrash · Jul 27, 2020 · 48f1952 · 48f1952
1 parent b6fc0f3
commit 48f1952
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pwnlib/commandline/shellcraft.py b/pwnlib/commandline/shellcraft.py
@@ -198,7 +198,9 @@ def main(args):
         common.parser.print_usage()
         exit()
 
-    if args.shellcode not in shellcraft.templates:
+    try:
+        template = get_template(args.shellcode)
+    except AttributeError:
         log.error("Unknown shellcraft template %r. Use --list to see available shellcodes." % args.shellcode)
 
     func = get_template(args.shellcode)