mirror of
https://github.com/FuzzingLabs/fuzzforge_ai.git
synced 2026-07-14 01:07:20 +02:00
AFredefon
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
"""Models for the harness-validator module."""
|
|
|
|
from enum import Enum
|
|
|
|
from pydantic import BaseModel, Field
|
|
from fuzzforge_modules_sdk.api.models import FuzzForgeModuleInputBase, FuzzForgeModuleOutputBase
|
|
|
|
from module.settings import Settings
|
|
|
|
|
|
class HarnessStatus(str, Enum):
|
|
"""Status of harness validation."""
|
|
|
|
VALID = "valid"
|
|
COMPILE_ERROR = "compile_error"
|
|
NOT_FOUND = "not_found"
|
|
TIMEOUT = "timeout"
|
|
ERROR = "error"
|
|
|
|
|
|
class ValidationResult(BaseModel):
|
|
"""Result of validating a single harness."""
|
|
|
|
#: Name of the harness target
|
|
target: str
|
|
|
|
#: Path to the harness file
|
|
file_path: str
|
|
|
|
#: Validation status
|
|
status: HarnessStatus
|
|
|
|
#: Compilation errors (if any)
|
|
errors: list[str] = Field(default_factory=list)
|
|
|
|
#: Compilation warnings (if any)
|
|
warnings: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class Input(FuzzForgeModuleInputBase[Settings]):
|
|
"""Input for the harness-validator module.
|
|
|
|
Expects a fuzz project directory with:
|
|
- Cargo.toml
|
|
- fuzz_targets/ directory with .rs harness files
|
|
"""
|
|
|
|
|
|
class Output(FuzzForgeModuleOutputBase):
|
|
"""Output from the harness-validator module."""
|
|
|
|
#: Path to the fuzz project
|
|
fuzz_project: str = ""
|
|
|
|
#: Total number of harness targets
|
|
total_targets: int = 0
|
|
|
|
#: Number of valid (compilable) harnesses
|
|
valid_count: int = 0
|
|
|
|
#: Number of invalid harnesses
|
|
invalid_count: int = 0
|
|
|
|
#: List of valid target names (ready for fuzzing)
|
|
valid_targets: list[str] = Field(default_factory=list)
|
|
|
|
#: List of invalid target names (need fixes)
|
|
invalid_targets: list[str] = Field(default_factory=list)
|
|
|
|
#: Detailed validation results per target
|
|
results: list[ValidationResult] = Field(default_factory=list)
|