(
self,
msf_host: str = "127.0.0.1",
msf_port: int = 55553,
msf_password: str = "",
ollama_host: str = "http://localhost:11434",
ollama_model: str = "llama3.1:latest",
output_dir: str = "/tmp/secsuite-loop",
)
| 144 | |
| 145 | def __init__( |
| 146 | self, |
| 147 | msf_host: str = "127.0.0.1", |
| 148 | msf_port: int = 55553, |
| 149 | msf_password: str = "", |
| 150 | ollama_host: str = "http://localhost:11434", |
| 151 | ollama_model: str = "llama3.1:latest", |
| 152 | output_dir: str = "/tmp/secsuite-loop", |
| 153 | ) -> None: |
| 154 | self._exploit_runner = ExploitRunner( |
| 155 | msf_host=msf_host, msf_port=msf_port, msf_password=msf_password, |
| 156 | ) |
| 157 | self._remediation_ai = RemediationAI(model=ollama_model, ollama_host=ollama_host) |
| 158 | self._hardener = AutoHardener(snapshot_dir=f"{output_dir}/snapshots") |
| 159 | self._output_dir = output_dir |
| 160 | os.makedirs(output_dir, exist_ok=True) |
| 161 | |
| 162 | async def run( |
| 163 | self, |
| 164 | target: str, |
nothing calls this directly
no test coverage detected