p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Common: Add filename validation to CustomPBAConfigurationSchema

This commit is contained in:
Shreya Malviya 2022-07-26 16:10:20 +05:30
parent 4116ebd24d
commit 626720ff9f
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/common/agent_configuration/agent_sub_configuration_schemas.py
View File

@ -1,4 +1,6 @@
from marshmallow import Schema, fields, post_load
import re
from marshmallow import Schema, fields, post_load, validate
from .agent_sub_configurations import (
CustomPBAConfiguration,
@ -13,12 +15,14 @@ from .agent_sub_configurations import (
)
from .utils import freeze_lists
valid_custom_pba_filename_regex = re.compile(r"^([a-zA-Z0-9\ \._-]+)$")
class CustomPBAConfigurationSchema(Schema):
linux_command = fields.Str()
linux_filename = fields.Str()
linux_filename = fields.Str(validate=validate.Regexp(regex=valid_custom_pba_filename_regex))
windows_command = fields.Str()
windows_filename = fields.Str()
windows_filename = fields.Str(validate=validate.Regexp(regex=valid_custom_pba_filename_regex))
@post_load
def _make_custom_pba_configuration(self, data, **kwargs):