mirror of
https://github.com/mblanke/GooseStrike.git
synced 2026-03-01 05:50:22 -05:00
Add roadmap API and mock dashboard
This commit is contained in:
40
hydra_runner.py
Normal file
40
hydra_runner.py
Normal file
@@ -0,0 +1,40 @@
|
||||
"""Wrapper for Hydra tasks."""
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any, Dict, List
|
||||
|
||||
from runner_utils import run_subprocess
|
||||
|
||||
|
||||
def build_command(task: Dict[str, Any]) -> List[str]:
|
||||
service = task.get("service")
|
||||
target = task.get("target")
|
||||
if not service or not target:
|
||||
raise ValueError("service and target are required")
|
||||
command = ["hydra", "-t", str(task.get("threads", 4))]
|
||||
if task.get("username"):
|
||||
command.extend(["-l", task["username"]])
|
||||
if task.get("password"):
|
||||
command.extend(["-p", task["password"]])
|
||||
if task.get("username_list"):
|
||||
command.extend(["-L", task["username_list"]])
|
||||
if task.get("password_list"):
|
||||
command.extend(["-P", task["password_list"]])
|
||||
if task.get("options"):
|
||||
for opt in task["options"]:
|
||||
command.append(opt)
|
||||
command.extend([f"{target}", service])
|
||||
return command
|
||||
|
||||
|
||||
def run_task(task: Dict[str, Any]) -> Dict[str, Any]:
|
||||
try:
|
||||
command = build_command(task)
|
||||
except ValueError as exc:
|
||||
return {"status": "error", "exit_code": None, "error": str(exc)}
|
||||
return run_subprocess(command, "hydra")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
example = {"service": "ssh", "target": "10.0.0.5", "username": "root", "password_list": "rockyou.txt"}
|
||||
print(run_task(example))
|
||||
Reference in New Issue
Block a user