Gh0st
/
Nettacker
mirror of https://github.com/OWASP/Nettacker.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

maybe patching find_events will help

This commit is contained in:
pUrGe12 2025-09-02 18:24:53 +05:30
parent 62a9a3bdc0
commit 64dd5550d0
2 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
nettacker/config.py
View File

@ -7,7 +7,6 @@ from nettacker.core.utils.common import now, generate_random_token
CWD = Path.cwd() CWD = Path.cwd()
PACKAGE_PATH = Path(__file__).parent PACKAGE_PATH = Path(__file__).parent
ROOT_PATH = Path(__file__).parent.parent
@lru_cache(maxsize=128) @lru_cache(maxsize=128)
@ -115,19 +114,19 @@ class PathConfig:
a JSON contain the working, tmp and results path a JSON contain the working, tmp and results path
""" """
data_dir = ROOT_PATH / ".nettacker/data" data_dir = CWD / ".nettacker/data"
new_database_file = ROOT_PATH / ".nettacker/data/nettacker.db" new_database_file = CWD / ".nettacker/data/nettacker.db"
old_database_file = ROOT_PATH / ".data/nettacker.db" old_database_file = CWD / ".data/nettacker.db"
graph_dir = PACKAGE_PATH / "lib/graph" graph_dir = PACKAGE_PATH / "lib/graph"
home_dir = ROOT_PATH home_dir = CWD
locale_dir = PACKAGE_PATH / "locale" locale_dir = PACKAGE_PATH / "locale"
logo_file = PACKAGE_PATH / "logo.txt" logo_file = PACKAGE_PATH / "logo.txt"
module_protocols_dir = PACKAGE_PATH / "core/lib" module_protocols_dir = PACKAGE_PATH / "core/lib"
modules_dir = PACKAGE_PATH / "modules" modules_dir = PACKAGE_PATH / "modules"
payloads_dir = PACKAGE_PATH / "lib/payloads" payloads_dir = PACKAGE_PATH / "lib/payloads"
release_name_file = PACKAGE_PATH / "release_name.txt" release_name_file = PACKAGE_PATH / "release_name.txt"
results_dir = ROOT_PATH / ".nettacker/data/results" results_dir = CWD / ".nettacker/data/results"
tmp_dir = ROOT_PATH / ".nettacker/data/tmp" tmp_dir = CWD / ".nettacker/data/tmp"
web_static_dir = PACKAGE_PATH / "web/static" web_static_dir = PACKAGE_PATH / "web/static"
user_agents_file = PACKAGE_PATH / "lib/payloads/User-Agents/web_browsers_user_agents.txt" user_agents_file = PACKAGE_PATH / "lib/payloads/User-Agents/web_browsers_user_agents.txt"

21
tests/core/test_module.py
View File

@ -234,6 +234,7 @@ def loader_side_effect(name, inputs):
return mock_inst return mock_inst
@patch("nettacker.core.module.find_events", return_value=[])
@patch("nettacker.core.module.Thread") @patch("nettacker.core.module.Thread")
@patch("nettacker.core.module.importlib.import_module") @patch("nettacker.core.module.importlib.import_module")
@patch("nettacker.core.module.time.sleep", return_value=None) @patch("nettacker.core.module.time.sleep", return_value=None)
@ -243,15 +244,19 @@ def test_start_creates_threads_minimal(
mock_sleep, mock_sleep,
mock_import_module, mock_import_module,
mock_thread_cls, mock_thread_cls,
mock_find_events,
options, options,
module_args, module_args,
): ):
# Mock HttpEngine from the imported module
fake_engine = MagicMock() fake_engine = MagicMock()
mock_import_module.return_value = MagicMock(HttpEngine=MagicMock(return_value=fake_engine)) mock_import_module.return_value = MagicMock(HttpEngine=MagicMock(return_value=fake_engine))
# Mock thread instances
mock_thread_instance = MagicMock() mock_thread_instance = MagicMock()
mock_thread_cls.return_value = mock_thread_instance mock_thread_cls.return_value = mock_thread_instance
# Create module with mocked attributes
module = Module("test_module", options, **module_args) module = Module("test_module", options, **module_args)
module.libraries = ["http"] module.libraries = ["http"]
module.discovered_services = {"http": [80]} module.discovered_services = {"http": [80]}
@ -261,19 +266,23 @@ def test_start_creates_threads_minimal(
"payloads": [ "payloads": [
{ {
"library": "http", "library": "http",
"steps": [[{"response": {}, "id": 1}], [{"response": {}, "id": 2}]], "steps": [
[{"response": {}, "id": 1}],
[{"response": {}, "id": 2}],
],
} }
] ]
} }
# Run
module.start() module.start()
# Assert threads created twice
assert mock_thread_cls.call_count == 2 assert mock_thread_cls.call_count == 2
# Collect actual IDs passed to Thread
expected_ids = {1, 2} expected_ids = {1, 2}
actual_ids = set() actual_ids = {kwargs["args"][0]["id"] for _, kwargs in mock_thread_cls.call_args_list}
for _, kwargs in mock_thread_cls.call_args_list:
sub_step = kwargs["args"][0]
actual_ids.add(sub_step["id"])
assert actual_ids == expected_ids assert actual_ids == expected_ids
assert mock_thread_instance.start.call_count == 2 assert mock_thread_instance.start.call_count == 2