mirror of
https://github.com/CyberSecurityUP/NeuroSploit.git
synced 2026-02-12 14:02:45 +00:00
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
"""
|
|
NeuroSploit v3 - Scan Schemas
|
|
"""
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class AuthConfig(BaseModel):
|
|
"""Authentication configuration for authenticated testing"""
|
|
auth_type: str = Field("none", description="Auth type: none, cookie, header, basic, bearer")
|
|
cookie: Optional[str] = Field(None, description="Session cookie value")
|
|
bearer_token: Optional[str] = Field(None, description="Bearer/JWT token")
|
|
username: Optional[str] = Field(None, description="Username for basic auth")
|
|
password: Optional[str] = Field(None, description="Password for basic auth")
|
|
header_name: Optional[str] = Field(None, description="Custom header name")
|
|
header_value: Optional[str] = Field(None, description="Custom header value")
|
|
|
|
|
|
class ScanCreate(BaseModel):
|
|
"""Schema for creating a new scan"""
|
|
name: Optional[str] = Field(None, max_length=255, description="Scan name")
|
|
targets: List[str] = Field(..., min_length=1, description="List of target URLs")
|
|
scan_type: str = Field("full", description="Scan type: quick, full, custom")
|
|
recon_enabled: bool = Field(True, description="Enable reconnaissance phase")
|
|
custom_prompt: Optional[str] = Field(None, max_length=32000, description="Custom prompt (up to 32k tokens)")
|
|
prompt_id: Optional[str] = Field(None, description="ID of preset prompt to use")
|
|
config: dict = Field(default_factory=dict, description="Additional configuration")
|
|
auth: Optional[AuthConfig] = Field(None, description="Authentication configuration")
|
|
custom_headers: Optional[dict] = Field(None, description="Custom HTTP headers to include")
|
|
|
|
|
|
class ScanUpdate(BaseModel):
|
|
"""Schema for updating a scan"""
|
|
name: Optional[str] = None
|
|
status: Optional[str] = None
|
|
progress: Optional[int] = None
|
|
current_phase: Optional[str] = None
|
|
error_message: Optional[str] = None
|
|
|
|
|
|
class ScanProgress(BaseModel):
|
|
"""Schema for scan progress updates"""
|
|
scan_id: str
|
|
status: str
|
|
progress: int
|
|
current_phase: Optional[str] = None
|
|
message: Optional[str] = None
|
|
total_endpoints: int = 0
|
|
total_vulnerabilities: int = 0
|
|
|
|
|
|
class ScanResponse(BaseModel):
|
|
"""Schema for scan response"""
|
|
id: str
|
|
name: Optional[str]
|
|
status: str
|
|
scan_type: str
|
|
recon_enabled: bool
|
|
progress: int
|
|
current_phase: Optional[str]
|
|
config: dict
|
|
custom_prompt: Optional[str]
|
|
prompt_id: Optional[str]
|
|
auth_type: Optional[str] = None
|
|
custom_headers: Optional[dict] = None
|
|
created_at: datetime
|
|
started_at: Optional[datetime]
|
|
completed_at: Optional[datetime]
|
|
error_message: Optional[str]
|
|
total_endpoints: int
|
|
total_vulnerabilities: int
|
|
critical_count: int
|
|
high_count: int
|
|
medium_count: int
|
|
low_count: int
|
|
info_count: int
|
|
targets: List[dict] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ScanListResponse(BaseModel):
|
|
"""Schema for list of scans"""
|
|
scans: List[ScanResponse]
|
|
total: int
|
|
page: int = 1
|
|
per_page: int = 10
|