ilia
12a6e2b6bc
FastAPI + Kali tools UI for authorized scans of levkin.ca / LAN targets. Gitignore local data/.env; add pytest + gitleaks CI.
85 lines
2.1 KiB
Python
85 lines
2.1 KiB
Python
from datetime import datetime
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ScanOptions(BaseModel):
|
|
port_range: str = "1-1000"
|
|
include_web: bool = True
|
|
include_brute: bool = False
|
|
include_ssl: bool = True
|
|
llm_model: str = "qwen2.5:14b"
|
|
|
|
|
|
class ScanCreate(BaseModel):
|
|
target: str
|
|
profile: Literal["passive", "standard", "redteam"]
|
|
options: ScanOptions = Field(default_factory=ScanOptions)
|
|
authorized: bool = False
|
|
|
|
|
|
class BulkScanCreate(BaseModel):
|
|
targets: list[str]
|
|
profile: Literal["passive", "standard", "redteam"]
|
|
options: ScanOptions = Field(default_factory=ScanOptions)
|
|
authorized: bool = False
|
|
|
|
|
|
class Scan(BaseModel):
|
|
id: str
|
|
target: str
|
|
profile: Literal["passive", "standard", "redteam"]
|
|
options: ScanOptions
|
|
status: Literal["queued", "running", "complete", "failed", "cancelled"]
|
|
created_at: datetime
|
|
completed_at: datetime | None = None
|
|
llm_unavailable: bool = False
|
|
bulk_id: str | None = None
|
|
|
|
|
|
class BulkRun(BaseModel):
|
|
id: str
|
|
profile: Literal["passive", "standard", "redteam"]
|
|
status: Literal["queued", "running", "complete", "cancelled"]
|
|
created_at: datetime
|
|
completed_at: datetime | None = None
|
|
total: int = 0
|
|
complete: int = 0
|
|
failed: int = 0
|
|
running: int = 0
|
|
queued: int = 0
|
|
|
|
|
|
class Phase(BaseModel):
|
|
id: str
|
|
scan_id: str
|
|
name: str
|
|
tool: str
|
|
command: str
|
|
status: Literal["pending", "running", "complete", "failed", "skipped", "blocked", "inconclusive"]
|
|
started_at: datetime | None = None
|
|
completed_at: datetime | None = None
|
|
raw_output: str = ""
|
|
exit_code: int | None = None
|
|
|
|
|
|
class Finding(BaseModel):
|
|
id: str
|
|
scan_id: str
|
|
phase_id: str
|
|
severity: Literal["critical", "high", "medium", "low", "info"]
|
|
title: str
|
|
description: str
|
|
evidence: str
|
|
recommendation: str
|
|
cve_refs: list[str] = Field(default_factory=list)
|
|
confidence: Literal["high", "medium", "low"]
|
|
|
|
|
|
class ToolResult(BaseModel):
|
|
stdout: str
|
|
stderr: str
|
|
exit_code: int
|
|
duration: float
|