{xN D d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ d dlmZmZ ej e Z ej d Z G d d
e Z ed d d
g Z G d de Z G d de
Z d Z d Z d Z dS ) N)
namedtuplewraps) Validator)ANTIVIRUS_MODE
BackupRestoreMalware)
LicenseCLN) BackupSystem get_backendz^[A-Fa-f0-9]{64}$c e Zd ZddZdS )ValidationErrorNc b t |t r |g| _ n|| _ |pi | _ d S N)
isinstancestrerrors
extra_data)selfr r s W/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/rpc_tools/validate.py__init__zValidationError.__init__ s6 fc " " !!(DKK DK$* r )__name__
__module__ __qualname__r r r r r s( + + + + + +r r OrderByBase column_namedescc 4 e Zd Z fdZed Z xZS )OrderByc J t | || S r )super__new__)clsr r __class__s r r$ zOrderBy.__new__$ s wwsK666r c t j d | dd \ }} | ||dk S # t $ r5}t d t | | d}~ww xY w)zP
:param ob_string: for example: 'user+', 'id-'
:return:
z ^(.+)([+|-]) -z Incorrect order_by: ({}): {}N)recompilesplit
ValueErrorformatr )r% ob_stringcol_namesignes r
fromstringzOrderBy.fromstring' s Z 7 7 = =i H H2 NNHd3x-- - .55c!ffiHH
s AA
B0A>>B)r r r r$ classmethodr4
__classcell__r& s @r r! r! # sS 7 7 7 7 7 [ r r! c e Zd ZdZ fdZd Zd Zd Zd Zde fdZ
d Z d
Z d Z
d efd
Zd Zd Zd Zd Zd Zd Zd Zde fdZd Zd Zde ded dfdZ xZS )SchemaValidatorz%Y-%m-%dc H t j |i | i | _ d S r )r# r r )r argskwargsr& s r r zSchemaValidator.__init__9 s* $)&))) r c d t |t r|S t | S r )r r! r4 r values r _normalize_coerce_order_byz*SchemaValidator._normalize_coerce_order_by= s- eW % % L!!%(((r c h t | S r )r striplowerr> s r _normalize_coerce_sha256hashz,SchemaValidator._normalize_coerce_sha256hashB s& 5zz !!'')))r c 4 t rdS | t j S |S )NF)r r DATABASE_SCAN_ENABLEDr> s r _normalize_coerce_scan_dbz)SchemaValidator._normalize_coerce_scan_dbE s" 5 =0 0 r c 4 t |t rdS dS )NTF)r r! r> s r _validate_type_order_byz'SchemaValidator._validate_type_order_byL s eW % % 4ur r? c t t t | S r )
SHA256_REGEXPmatchr rB r> s r _validate_type_sha256hashz)SchemaValidator._validate_type_sha256hashQ s( ""3u::#3#3#5#5666r c |rJt j | s-| |d | dS dS dS )z#{'type': 'boolean', 'empty': False}zPath {} should be absoluteN)ospathisabs_errorr/ )r is_absolute_pathfieldr? s r _validate_is_absolute_pathz*SchemaValidator._validate_is_absolute_pathT sc O7==''
O E#?#F#Fu#M#MNNNNN O O
O
Or c |r? | d dS # t $ r | |d Y dS w xY wdS )z{'type': 'boolean'}asciiz Must only contain ascii symbolsN)encodeUnicodeEncodeErrorrR )r isasciirT r? s r _validate_isasciiz!SchemaValidator._validate_isasciiZ sr F
F W%%%%%%
F
F
F E#DEEEEEE
F F Fs ??c t | S r )intr> s r _normalize_coerce_intz%SchemaValidator._normalize_coerce_intb s 5zzr returnc t j t j S r )mathceildatetimenow timestamp)r documents r _normalize_default_setter_nowz-SchemaValidator._normalize_default_setter_nowe s- y*..00::<<===r c dS )a; {'type': 'dict', 'empty': False, 'schema': {
'users': {'type': 'list', 'allowed': ['non-root', 'root'],
'empty': False},
'require_rpc': {'type': 'string', 'empty': True, 'default': 'running',
'allowed': ['running', 'stopped', 'any', 'direct']}
}}
Nr r r; r< s r
_validate_cliz SchemaValidator._validate_clii r c dS )z"{'type': 'string', 'empty': False}Nr ri s r _validate_helpz SchemaValidator._validate_helps rk r c dS )z4{'type': 'boolean', 'empty': True, 'default': False}Nr ri s r _validate_positionalz$SchemaValidator._validate_positionalw rk r c dS )z!{'type': 'string', 'empty': True}Nr ri s r _validate_return_typez%SchemaValidator._validate_return_type{ rk r c dS )z5{'type': 'boolean', 'empty': False, 'default': False}Nr ri s r _validate_cli_onlyz"SchemaValidator._validate_cli_only~ rk r c dS )z
Parameter can be passed via the specified environment variable.
The value specified via a CLI argument takes precedence.
The rule's arguments are validated against this schema:
{'type': 'string', 'empty': False}
Nr ri s r _validate_envvarz SchemaValidator._validate_envvar rk r c dS )a
Parameter will only be accepted if provided via environment
variable specified by `envvar`. It will be rejected if passed as
a CLI argument.
The rule's arguments are validated against this schema:
{'type': 'boolean', 'default': False}
Nr ri s r _validate_envvar_onlyz%SchemaValidator._validate_envvar_only rk r c H |r t j | S |S r )rO rP abspathr> s r _normalize_coerce_pathz&SchemaValidator._normalize_coerce_path s# *7??5)) ) r c N t |t r|S t | S r )r r r r> s r _normalize_coerce_backup_systemz/SchemaValidator._normalize_coerce_backup_system s' e\ * * L5!!!r c t t j rt j s| |d d S d S )NzBackup is not enabled!)r ENABLED
backup_systemrR )r rT r? s r _validator_backup_is_enabledz,SchemaValidator._validator_backup_is_enabled sB % 9-*E*G*G 9 KK 7 8 8 8 8 8 9 9r rT Nc
t j | d S # t $ r5}| |d| dt | d Y d }~d S d }~ww xY w)NzIncorrect timestamp: z ())rc
fromtimestampr. rR r )r rT r? r3 s r _validator_timestampz$SchemaValidator._validator_timestamp s K + +E 2 2 2 2 2 K K K KK Iu I IA I I I J J J J J J J J J Ks #
A"*A A") r r r _DATE_FORMATr r@ rD rG rI r rM rU r[ r^ r] rg rj rm ro rq rs ru rw rz r| r r r6 r7 s @r r9 r9 6 s L ) ) )
* * *
7s 7 7 7 7O O O F F F > > > > > 1 1 1C C C0 0 0D D D C " " " 9 9 9K# Kc Kd K K K K K K K Kr r9 c $ | ||id }| ||| i sNt d ||| j t
| j | j | j | S )NT)always_return_documentz6Validation error with command {}, params {}, errors {})
normalizedvalidateloggerwarningr/ r r r rf ) validatorhashableparamsvaluess r r r s
!
! 64 " F x)9 : ; ; F D K K&)"2
i . 0DEEE h ''r c fd}|S )Nc @ t fd }|S )Nc K t | d }t || d | d< | g|R i | d {V }|S )Ncommandr )tupler )requestr; r< r resultfr s r wrapperz5validate_middleware..wrapped..wrapper sq WY /00H (8WX%6! !GH 1W6t666v66666666FMr r )r r r s ` r wrappedz$validate_middleware..wrapped s9 q
r r )r r s ` r validate_middlewarer s#
Nr c p t d t fd }t r|S S )zz
Decorator for CLI commands methods that ensures that the AV+ license
is valid.
:raises ValidationError:
zImunifyAV+ license requiredc N K t j r | i | d {V S r )r
is_valid_av_plus)r; r< exceptionfuncs r
async_wrapperz/validate_av_plus_license..async_wrapper sD & ( ( / t.v........ .r )r r r )r r r s ` @r validate_av_plus_licenser sW =>>I
4[[ [
Kr )!rc loggingra rO r+ collectionsr functoolsr cerberus.validatorr defence360agent.contracts.configr r r !defence360agent.contracts.licenser
%defence360agent.subsys.backup_systemsr r getLoggerr r r, rK Exceptionr r r! r9 r r r r r r r s " " " " " " ( ( ( ( ( (
9 8 8 8 8 8 K K K K K K K K 8 $ $
.//
+ + + + +i + + + j (?@@ k &rK rK rK rK rKi rK rK rKj ( ( (
r