147 lines
5.5 KiB
Python
Executable File
147 lines
5.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import nxc
|
|
import importlib
|
|
import traceback
|
|
import sys
|
|
|
|
from os import listdir
|
|
from os.path import dirname
|
|
from os.path import join as path_join
|
|
|
|
from nxc.context import Context
|
|
from nxc.logger import NXCAdapter
|
|
from nxc.paths import NXC_PATH
|
|
|
|
|
|
class ModuleLoader:
|
|
def __init__(self, args, db, logger):
|
|
self.args = args
|
|
self.db = db
|
|
self.logger = logger
|
|
|
|
def module_is_sane(self, module, module_path):
|
|
"""
|
|
Check if a module has the proper attributes
|
|
"""
|
|
module_error = False
|
|
if not hasattr(module, "name"):
|
|
self.logger.fail(f"{module_path} missing the name variable")
|
|
module_error = True
|
|
elif not hasattr(module, "description"):
|
|
self.logger.fail(f"{module_path} missing the description variable")
|
|
module_error = True
|
|
elif not hasattr(module, "supported_protocols"):
|
|
self.logger.fail(f"{module_path} missing the supported_protocols variable")
|
|
module_error = True
|
|
elif not hasattr(module, "opsec_safe"):
|
|
self.logger.fail(f"{module_path} missing the opsec_safe variable")
|
|
module_error = True
|
|
elif not hasattr(module, "multiple_hosts"):
|
|
self.logger.fail(f"{module_path} missing the multiple_hosts variable")
|
|
module_error = True
|
|
elif not hasattr(module, "options"):
|
|
self.logger.fail(f"{module_path} missing the options function")
|
|
module_error = True
|
|
elif not hasattr(module, "on_login") and not (module, "on_admin_login"):
|
|
self.logger.fail(f"{module_path} missing the on_login/on_admin_login function(s)")
|
|
module_error = True
|
|
# elif not hasattr(module, 'chain_support'):
|
|
# self.logger.fail('{} missing the chain_support variable'.format(module_path))
|
|
# module_error = True
|
|
|
|
if module_error:
|
|
return False
|
|
return True
|
|
|
|
def load_module(self, module_path):
|
|
"""
|
|
Load a module, initializing it and checking that it has the proper attributes
|
|
"""
|
|
try:
|
|
spec = importlib.util.spec_from_file_location("NXCModule", module_path)
|
|
module = spec.loader.load_module().NXCModule()
|
|
|
|
if self.module_is_sane(module, module_path):
|
|
return module
|
|
except Exception as e:
|
|
self.logger.fail(f"Failed loading module at {module_path}: {e}")
|
|
self.logger.debug(traceback.format_exc())
|
|
return None
|
|
|
|
def init_module(self, module_path):
|
|
"""
|
|
Initialize a module for execution
|
|
"""
|
|
module = None
|
|
module = self.load_module(module_path)
|
|
|
|
if module:
|
|
self.logger.debug(f"Supported protocols: {module.supported_protocols}")
|
|
self.logger.debug(f"Protocol: {self.args.protocol}")
|
|
if self.args.protocol in module.supported_protocols:
|
|
try:
|
|
module_logger = NXCAdapter(extra={"module_name": module.name.upper()})
|
|
except Exception as e:
|
|
self.logger.fail(f"Error loading NXCAdaptor for module {module.name.upper()}: {e}")
|
|
context = Context(self.db, module_logger, self.args)
|
|
module_options = {}
|
|
|
|
for option in self.args.module_options:
|
|
key, value = option.split("=", 1)
|
|
module_options[str(key).upper()] = value
|
|
|
|
module.options(context, module_options)
|
|
return module
|
|
else:
|
|
self.logger.fail(f"Module {module.name.upper()} is not supported for protocol {self.args.protocol}")
|
|
sys.exit(1)
|
|
|
|
def get_module_info(self, module_path):
|
|
"""
|
|
Get the path, description, and options from a module
|
|
"""
|
|
try:
|
|
spec = importlib.util.spec_from_file_location("NXCModule", module_path)
|
|
module_spec = spec.loader.load_module().NXCModule
|
|
|
|
module = {
|
|
f"{module_spec.name.lower()}": {
|
|
"path": module_path,
|
|
"description": module_spec.description,
|
|
"options": module_spec.options.__doc__,
|
|
"supported_protocols": module_spec.supported_protocols,
|
|
"opsec_safe": module_spec.opsec_safe,
|
|
"multiple_hosts": module_spec.multiple_hosts,
|
|
}
|
|
}
|
|
if self.module_is_sane(module_spec, module_path):
|
|
return module
|
|
except Exception as e:
|
|
self.logger.fail(f"Failed loading module at {module_path}: {e}")
|
|
self.logger.debug(traceback.format_exc())
|
|
return None
|
|
|
|
def list_modules(self):
|
|
"""
|
|
List modules without initializing them
|
|
"""
|
|
modules = {}
|
|
modules_paths = [
|
|
path_join(dirname(nxc.__file__), "modules"),
|
|
path_join(NXC_PATH, "modules"),
|
|
]
|
|
|
|
for path in modules_paths:
|
|
for module in listdir(path):
|
|
if module[-3:] == ".py" and module != "example_module.py":
|
|
try:
|
|
module_path = path_join(path, module)
|
|
module_data = self.get_module_info(module_path)
|
|
modules.update(module_data)
|
|
except Exception as e:
|
|
self.logger.debug(f"Error loading module {module}: {e}")
|
|
pass
|
|
return modules
|