p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 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 ( from .agent_sub_configurations import (
CustomPBAConfiguration, CustomPBAConfiguration,
@ -13,12 +15,14 @@ from .agent_sub_configurations import (
) )
from .utils import freeze_lists from .utils import freeze_lists
valid_custom_pba_filename_regex = re.compile(r"^([a-zA-Z0-9\ \._-]+)$")
class CustomPBAConfigurationSchema(Schema): class CustomPBAConfigurationSchema(Schema):
linux_command = fields.Str() 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_command = fields.Str()
windows_filename = fields.Str() windows_filename = fields.Str(validate=validate.Regexp(regex=valid_custom_pba_filename_regex))
@post_load @post_load
def _make_custom_pba_configuration(self, data, **kwargs): def _make_custom_pba_configuration(self, data, **kwargs):