mirror of
https://github.com/trustedsec/hate_crack.git
synced 2026-04-28 12:03:11 -07:00
c51ae332c6
Add Wordlist Tools submenu (key 80) with 7 preprocessing utilities backed by hashcat-utils binaries: - Filter by length (len.bin) - #90 - Require/exclude character classes (req-include.bin, req-exclude.bin) - #90 - Extract substring (cutb.bin) - #90 - Remove matching lines (rli.bin, rli2.bin) - #91 - Split by length (splitlen.bin) - #92 - Shard wordlist (gate.bin) - #94 Three-layer pattern: - main.py: low-level wrappers (wordlist_filter_len, wordlist_filter_req_include, wordlist_filter_req_exclude, wordlist_cutb, wordlist_splitlen, wordlist_subtract, wordlist_subtract_single, wordlist_gate) return bool for testability - attacks.py: UI handlers with input validation and the wordlist_tools_submenu dispatcher - hate_crack.py + main.py: menu item "80" wired in both get_main_menu_items and get_main_menu_options Move interactive_menu import to attacks.py module level (was local import in combinator_submenu) to support patching in tests.
75 lines
3.0 KiB
Python
75 lines
3.0 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
CLI_SPEC = importlib.util.spec_from_file_location(
|
|
"hate_crack_cli", PROJECT_ROOT / "hate_crack.py"
|
|
)
|
|
CLI_MODULE = importlib.util.module_from_spec(CLI_SPEC)
|
|
CLI_SPEC.loader.exec_module(CLI_MODULE)
|
|
|
|
MENU_OPTION_TEST_CASES = [
|
|
("1", CLI_MODULE._attacks, "quick_crack", "quick-crack"),
|
|
("2", CLI_MODULE._attacks, "extensive_crack", "extensive-crack"),
|
|
("3", CLI_MODULE._attacks, "brute_force_crack", "brute-force"),
|
|
("4", CLI_MODULE._attacks, "top_mask_crack", "top-mask"),
|
|
("5", CLI_MODULE._attacks, "fingerprint_crack", "fingerprint"),
|
|
("6", CLI_MODULE._attacks, "combinator_submenu", "combinator"),
|
|
("7", CLI_MODULE._attacks, "hybrid_crack", "hybrid"),
|
|
("8", CLI_MODULE._attacks, "pathwell_crack", "pathwell"),
|
|
("9", CLI_MODULE._attacks, "prince_attack", "prince"),
|
|
("13", CLI_MODULE._attacks, "bandrel_method", "bandrel"),
|
|
("14", CLI_MODULE._attacks, "loopback_attack", "loopback"),
|
|
("15", CLI_MODULE._attacks, "ollama_attack", "ollama"),
|
|
("16", CLI_MODULE._attacks, "omen_attack", "omen"),
|
|
("17", CLI_MODULE._attacks, "adhoc_mask_crack", "adhoc-mask"),
|
|
("18", CLI_MODULE._attacks, "markov_brute_force", "markov-brute"),
|
|
("80", CLI_MODULE._attacks, "wordlist_tools_submenu", "wordlist-tools"),
|
|
("90", CLI_MODULE, "download_hashmob_rules", "hashmob-rules"),
|
|
("91", CLI_MODULE, "weakpass_wordlist_menu", "weakpass-menu"),
|
|
("92", CLI_MODULE, "download_hashmob_wordlists", "hashmob-wordlists"),
|
|
("93", CLI_MODULE, "weakpass_wordlist_menu", "weakpass-menu-secondary"),
|
|
("95", CLI_MODULE, "pipal", "pipal"),
|
|
("96", CLI_MODULE, "export_excel", "export-excel"),
|
|
("97", CLI_MODULE, "show_results", "show-results"),
|
|
("98", CLI_MODULE, "show_readme", "show-readme"),
|
|
("99", CLI_MODULE, "quit_hc", "quit"),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("option_key", "target_module", "target_attr", "expected_prefix"),
|
|
MENU_OPTION_TEST_CASES,
|
|
)
|
|
def test_main_menu_option_returns_expected(
|
|
monkeypatch, option_key, target_module, target_attr, expected_prefix
|
|
):
|
|
sentinel = f"{expected_prefix}-{option_key}"
|
|
monkeypatch.setattr(
|
|
target_module,
|
|
target_attr,
|
|
lambda *args, **kwargs: sentinel,
|
|
)
|
|
|
|
options = CLI_MODULE.get_main_menu_options()
|
|
assert option_key in options, f"Menu option {option_key} must exist"
|
|
handler = options[option_key]
|
|
assert handler() == sentinel
|
|
|
|
|
|
def test_main_menu_option_94_hashview_hidden_without_hashview_api_key(monkeypatch):
|
|
monkeypatch.setattr(CLI_MODULE, "hashview_api_key", "")
|
|
options = CLI_MODULE.get_main_menu_options()
|
|
assert "94" not in options
|
|
|
|
|
|
def test_main_menu_option_94_hashview_visible_with_hashview_api_key(monkeypatch):
|
|
monkeypatch.setattr(CLI_MODULE, "hashview_api_key", "test-key")
|
|
sentinel = "hashview-94"
|
|
monkeypatch.setattr(CLI_MODULE, "hashview_api", lambda *a, **k: sentinel)
|
|
options = CLI_MODULE.get_main_menu_options()
|
|
assert "94" in options
|
|
assert options["94"]() == sentinel
|