forked from p15670423/monkey
Common: Create PropagationConfiguration using pydantic
This commit is contained in:
parent
602604e408
commit
501f752248
|
@ -1,7 +1,7 @@
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import Dict, Tuple
|
from typing import Dict, Tuple
|
||||||
|
|
||||||
from pydantic import PositiveFloat, conint, validator
|
from pydantic import PositiveFloat, PositiveInt, conint, validator
|
||||||
|
|
||||||
from common.base_models import MutableInfectionMonkeyBaseModel
|
from common.base_models import MutableInfectionMonkeyBaseModel
|
||||||
|
|
||||||
|
@ -337,3 +337,20 @@ class PropagationConfiguration:
|
||||||
maximum_depth: int
|
maximum_depth: int
|
||||||
network_scan: NetworkScanConfiguration
|
network_scan: NetworkScanConfiguration
|
||||||
exploitation: ExploitationConfiguration
|
exploitation: ExploitationConfiguration
|
||||||
|
|
||||||
|
|
||||||
|
class Pydantic___PropagationConfiguration:
|
||||||
|
"""
|
||||||
|
A configuration for propagation
|
||||||
|
|
||||||
|
Attributes:
|
||||||
|
:param maximum_depth: Maximum number of hops allowed to spread from the machine where
|
||||||
|
the attack started i.e. how far to propagate in the network from the
|
||||||
|
first machine
|
||||||
|
:param network_scan: Configuration for network scanning
|
||||||
|
:param exploitation: Configuration for exploitation
|
||||||
|
"""
|
||||||
|
|
||||||
|
maximum_depth: PositiveInt
|
||||||
|
network_scan: Pydantic___NetworkScanConfiguration
|
||||||
|
exploitation: Pydantic___ExploitationConfiguration
|
||||||
|
|
Loading…
Reference in New Issue