malmert
25bb7be29d
All checks were successful
Build and Deploy MITM Webserver / build (push) Successful in 1m40s
64 lines
2.7 KiB
Python
64 lines
2.7 KiB
Python
"""Pydantic model for packet rows returned by the backend."""
|
|
|
|
from datetime import datetime
|
|
from typing import Optional, Union
|
|
|
|
from pydantic import BaseModel, Field, IPvAnyAddress
|
|
|
|
|
|
class PacketDBModel(BaseModel):
|
|
"""Normalized packet representation used across DB and API layers."""
|
|
|
|
id: Union[int, str]
|
|
timestamp: datetime = Field(..., description="Packet timestamp in ISO format.")
|
|
iface: str
|
|
src_mac: Optional[str] = None
|
|
dst_mac: Optional[str] = None
|
|
eth_type: Optional[Union[int, str]] = None
|
|
ip_proto: Optional[Union[int, str]] = None
|
|
src_ip: Optional[IPvAnyAddress] = None
|
|
dst_ip: Optional[IPvAnyAddress] = None
|
|
src_port: Optional[int] = None
|
|
dst_port: Optional[int] = None
|
|
vlan_id: Optional[int] = None
|
|
length: Optional[int] = None
|
|
raw_b64: Optional[str] = Field(None, description="Base64-encoded packet bytes.")
|
|
app_protocol: Optional[str] = Field(None, description="Detected application protocol.")
|
|
app_master_protocol: Optional[str] = Field(None, description="Detected application master protocol.")
|
|
app_category: Optional[str] = Field(None, description="nDPI category, if available.")
|
|
app_confidence: Optional[str] = Field(None, description="nDPI confidence level, if available.")
|
|
app_hostname: Optional[str] = Field(None, description="Detected hostname/SNI, if available.")
|
|
app_is_encrypted: Optional[bool] = Field(None, description="Whether detected protocol appears encrypted.")
|
|
app_risk_score: Optional[int] = Field(None, description="Count/score of detected nDPI risks.")
|
|
dpi_metadata: Optional[dict] = Field(None, description="Raw DPI metadata from nDPI.")
|
|
direction: Optional[str] = None
|
|
packets: Optional[int] = None
|
|
|
|
class Config:
|
|
schema_extra = {
|
|
"example": {
|
|
"id": 123,
|
|
"timestamp": "2026-03-05T12:34:56.789Z",
|
|
"iface": "eth0",
|
|
"src_mac": "aa:bb:cc:dd:ee:ff",
|
|
"dst_mac": "11:22:33:44:55:66",
|
|
"eth_type": "IPv4",
|
|
"ip_proto": "TCP",
|
|
"src_ip": "192.168.1.10",
|
|
"dst_ip": "192.168.1.1",
|
|
"src_port": 54321,
|
|
"dst_port": 80,
|
|
"vlan_id": None,
|
|
"length": 128,
|
|
"raw_b64": "BASE64...",
|
|
"app_protocol": "HTTP",
|
|
"app_master_protocol": "HTTP",
|
|
"app_category": "Web",
|
|
"app_confidence": "high",
|
|
"app_hostname": "example.org",
|
|
"app_is_encrypted": False,
|
|
"app_risk_score": 0,
|
|
"dpi_metadata": {"method": "GET"},
|
|
}
|
|
}
|