Add category filtering to module listing

Author: NeffIsBack

nxc/cli.py

@@ -76,7 +76,7 @@ def gen_cli_args():
     mgroup = module_parser.add_argument_group("Modules", "Options for nxc modules")
     mgroup.add_argument("-M", "--module", choices=get_module_names(), action="append", metavar="MODULE", help="module to use")
     mgroup.add_argument("-o", metavar="MODULE_OPTION", nargs="+", default=[], dest="module_options", help="module options")
-    mgroup.add_argument("-L", "--list-modules", action="store_true", help="list available modules")
+    mgroup.add_argument("-L", "--list-modules", nargs="?", type=str, const="", help="list available modules")
     mgroup.add_argument("--options", dest="show_module_options", action="store_true", help="display module options")
 
     subparsers = parser.add_subparsers(title="Available Protocols", dest="protocol")

nxc/netexec.py

@@ -159,17 +159,20 @@ def main():
     # with the new nxc/config.py this can be eventually removed, as it can be imported anywhere
     protocol_object.config = nxc_config
 
-    if args.module or args.list_modules:
+    if args.module or args.list_modules is not None:
         loader = ModuleLoader(args, db, nxc_logger)
         modules = loader.list_modules()
 
-    if args.list_modules:
+    if args.list_modules is not None:
         low_privilege_modules = {m: props for m, props in modules.items() if args.protocol in props["supported_protocols"] and not props["requires_admin"]}
         high_privilege_modules = {m: props for m, props in modules.items() if args.protocol in props["supported_protocols"] and props["requires_admin"]}
 
         # List low privilege modules
         nxc_logger.highlight("LOW PRIVILEGE MODULES")
         for category, color in {CATEGORY.ENUMERATION: "green", CATEGORY.CREDENTIAL_DUMPING: "cyan", CATEGORY.PRIVILEGE_ESCALATION: "magenta"}.items():
+            # Add category filter for module listing
+            if args.list_modules and args.list_modules.lower() != category.name.lower():
+                continue
             if len([module for module in low_privilege_modules.values() if module["category"] == category]) > 0:
                 nxc_logger.highlight(colored(f"{category.name}", color, attrs=["bold"]))
             for name, props in sorted(low_privilege_modules.items()):
@@ -179,6 +182,9 @@ def main():
         # List high privilege modules
         nxc_logger.highlight("\nHIGH PRIVILEGE MODULES (requires admin privs)")
         for category, color in {CATEGORY.ENUMERATION: "green", CATEGORY.CREDENTIAL_DUMPING: "cyan", CATEGORY.PRIVILEGE_ESCALATION: "magenta"}.items():
+            # Add category filter for module listing
+            if args.list_modules and args.list_modules.lower() != category.name.lower():
+                continue
             if len([module for module in high_privilege_modules.values() if module["category"] == category]) > 0:
                 nxc_logger.highlight(colored(f"{category.name}", color, attrs=["bold"]))
             for name, props in sorted(high_privilege_modules.items()):