forked from p15670423/monkey
Common: Add maximum tunnel open time validation to AgentConfigurationSchema
This commit is contained in:
parent
85a11b62bd
commit
a603b16573
monkey/common/agent_configuration
|
@ -3,7 +3,7 @@ from __future__ import annotations
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import Any, Mapping, Tuple
|
from typing import Any, Mapping, Tuple
|
||||||
|
|
||||||
from marshmallow import Schema, fields
|
from marshmallow import Schema, fields, validate
|
||||||
from marshmallow.exceptions import MarshmallowError
|
from marshmallow.exceptions import MarshmallowError
|
||||||
|
|
||||||
from ..utils.code_utils import freeze_lists_in_mapping
|
from ..utils.code_utils import freeze_lists_in_mapping
|
||||||
|
@ -117,7 +117,7 @@ class AgentConfiguration:
|
||||||
|
|
||||||
|
|
||||||
class AgentConfigurationSchema(Schema):
|
class AgentConfigurationSchema(Schema):
|
||||||
keep_tunnel_open_time = fields.Float()
|
keep_tunnel_open_time = fields.Float(validate=validate.Range(min=0))
|
||||||
custom_pbas = fields.Nested(CustomPBAConfigurationSchema)
|
custom_pbas = fields.Nested(CustomPBAConfigurationSchema)
|
||||||
post_breach_actions = fields.List(fields.Nested(PluginConfigurationSchema))
|
post_breach_actions = fields.List(fields.Nested(PluginConfigurationSchema))
|
||||||
credential_collectors = fields.List(fields.Nested(PluginConfigurationSchema))
|
credential_collectors = fields.List(fields.Nested(PluginConfigurationSchema))
|
||||||
|
|
Loading…
Reference in New Issue