mirror of
https://github.com/trustedsec/hate_crack.git
synced 2026-07-28 22:51:14 -07:00
Justin Bollinger
3bc85d9b8a
Extract _select_rules() helper from quick_crack/loopback_attack and wire it into omen_attack so OMEN can run with rule chains. Extend hcatOmen() to accept and apply an hcatChains argument including debug mode injection. Fix resolve_path() to honour HATE_CRACK_ORIG_CWD (set by the install shim) so relative hash/wordlist paths resolve against the caller's working directory instead of the repo root. Increase default omenMaxCandidates to 50M.
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
from typing import Optional
|
|
|
|
|
|
def resolve_path(value: Optional[str]) -> Optional[str]:
|
|
"""Expand user and return an absolute path, or None.
|
|
|
|
When invoked via the bash shim (``uv run --directory``), the working
|
|
directory is changed to the repo root before Python starts.
|
|
``HATE_CRACK_ORIG_CWD`` preserves the caller's real working directory
|
|
so that relative paths on the command line resolve correctly.
|
|
"""
|
|
if not value:
|
|
return None
|
|
value = os.path.expanduser(value)
|
|
if os.path.isabs(value):
|
|
return value
|
|
orig_cwd = os.environ.get("HATE_CRACK_ORIG_CWD")
|
|
if orig_cwd:
|
|
return os.path.normpath(os.path.join(orig_cwd, value))
|
|
return os.path.abspath(value)
|
|
|
|
|
|
def add_common_args(parser) -> None:
|
|
pass # All config items are now set via config file only
|
|
|
|
|
|
def setup_logging(logger: logging.Logger, hate_path: str, debug_mode: bool) -> None:
|
|
if not debug_mode:
|
|
return
|
|
logger.setLevel(logging.DEBUG)
|
|
# Debug logs go to stderr by default (no log file side effects).
|
|
has_stream = any(
|
|
isinstance(h, logging.StreamHandler) and not isinstance(h, logging.FileHandler)
|
|
for h in logger.handlers
|
|
)
|
|
if not has_stream:
|
|
stream_handler = logging.StreamHandler(sys.stderr)
|
|
stream_handler.setFormatter(
|
|
logging.Formatter("%(asctime)s %(levelname)s %(message)s")
|
|
)
|
|
logger.addHandler(stream_handler)
|
|
# Show HTTP requests made by the requests/urllib3 library.
|
|
debug_handler = logging.StreamHandler(sys.stderr)
|
|
debug_handler.setFormatter(logging.Formatter("%(asctime)s %(levelname)s %(name)s %(message)s"))
|
|
urllib3_logger = logging.getLogger("urllib3")
|
|
urllib3_logger.setLevel(logging.DEBUG)
|
|
urllib3_logger.addHandler(debug_handler)
|