jrPV. | d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZmZm Z d dl
m Z d dl m
Z
mZmZ d dlmZ d dlZd dlmZ d dlmZ d d lmZ d d
lmZmZ d d l m Z ej e Z d Z!d
Z" edddg Z# edddg Z$ G d de# Z% G d de$ Z& G d de Zd Z'dS ) N)
namedtuple)datedatetime timedeltawraps) IPv4Network IPv6Network
ip_network)Path)SchemaValidator)validate) Webshield) IPSetPortlibipset)IPtoday yesterday
PortProtoBaseportproto
PeriodBasesincetoc 4 e Zd Z fdZed Z xZS ) PortProtoc 4 |t j vr"t d | t j |cxk rt j k s$n t d | t
| || S )Nz Protocol {} is not supportedzPort {} is incorrect)r PROTOS
ValueErrorformatMIN_PORTMAX_PORTsuper__new__)clsr r __class__s N/opt/imunify360/venv/lib/python3.11/site-packages/im360/simple_rpc/validate.pyr$ zPortProto.__new__ s ( ( ( ; B B5 I IJJ J"T>>>>I,>>>>> 3 : :4 @ @AA AwwsD%000 c | d \ }}t | } | || S # t $ r5}t d t | | d }~ww xY w)N:z Incorrect port_proto ({}): {})splitintr r str)r% pp_stringr r es r'
fromstringzPortProto.fromstring& s #//#..KD%t99D3tU## # /66s1vvyII
s 25
A40A//A4__name__
__module__ __qualname__r$ classmethodr0
__classcell__r& s @r' r r sS 1 1 1 1 1 [ r( r c 4 e Zd Z fdZed Z xZS )Periodc t j | t j | f nA# t $ r4}t d t | d }~ww xY wt | || S )Nz Incorrect value for period: {})r
fromtimestampr r r- r# r$ )r% r r r/ r& s r' r$ zPeriod.__new__3 s N "5 ) )8+A"+E+E E E E N N N = D DSVV L LMM M N wwsE2...s (,
A*/A%%A*c
t j }|| dddd z
}|t k r+t j |z
t j }}n|t k rUt j |z
t d z
}t j |z
}||}}n!d} t |d d |dd c}
nI# t t f$ r5}
t d |t |
d }
~
ww xY w
t d |D s"t d
t
fd
|D } | | i} t j t d i | z
t j }} | || S ) Nr )hourminutesecond microsecond days)weeksrC hoursminutessecondsz#Invalid string from period: {} ({})c 3 & K | ] }|d V
dS r N ).0p_names r' z$Period.fromstring..X s& %K%KFfQi%K%K%K%K%K%Kr( zInvalid suffix: {}c 3 4 K | ]}|d k |V dS rJ rK )rL xpsfxs r' rN z$Period.fromstring..\ s+ J JrSBqE\\\\\\ J Jr( rK )r nowreplace
total_secondsTODAYtime YESTERDAYr r, r
IndexErrorr r- endswithtuplenext timestamp)r%
period_stringrR seconds_since_midnightr r from_dateto_date period_namesvalr/ sfx_expanded real_argsrQ s @r' r0 zPeriod.fromstring; s lnn #++1Qqa+HH H
-// E ! ! &<8=9 9
@ ?r( c b |sd S |}t j }||k rN| |d t j dt j | || j z }||k r1| |d || j d S d S )Nz)Expiration contains expired timestamp {}!z%x %X %ZzRExpiration time {} is too far into the future. It is more than {} seconds from now)rV r r strftimegmtimeMAX_IPSET_TIMEOUT)ri r rj expiration_timerR max_expiration_times r' _validator_expirationz%SchemaValidator._validator_expiration s F ikk c ! ! KK;BBM*dk/.J.JKK
"D$: : 0 0 0 KK;& $*@AA
1 0r( c N t j s| |d d S d S )Nz8This command is not supported when webshield is disabled)r ENABLEr r s r' _validator_webshield_is_enabledz/SchemaValidator._validator_webshield_is_enabled s= KKJ
r( N)r2 r3 r4 r IPSET_TIMEOUT_MAXr rk ro rr rt rw rz r| r~ r r r r rK r( r' r
r
g s 2A A A% % % 3 3 3 > > >
= = =
. r( r
c t t j t dz }|dz |dz |dz fdfd fd fd}|S ) Nz../..z3defence360agent/simple_rpc/schema_responses/anotherz(imav/simple_rpc/schema_responses/anotherz)im360/simple_rpc/schema_responses/anotherc | }| r|S | }| r|S | S rm ) with_nameexists)filenamecore_schema_pathimav_schema_path core_schemas
im360_schemas imav_schemass r' #get_file_from_schema_responses_dirsz@validate_middleware..get_file_from_schema_responses_dirs ss '11(;; " " $ $ $# #'11(;; " " $ $ $# # &&x000r( c | dz } | d 5 }t j | cd d d S # 1 swxY w Y d S # t $ r t d| Y d S w xY w)Nz.jsonrzSchema file not found: %s)openjsonloadFileNotFoundErrorloggerwarning) return_type schema_pathfr s r' get_response_schemaz0validate_middleware..get_response_schema s 99 ' !
!!#&&
$!y||
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$ NN6 D D D44 s3 A A A A A A A %A=<A=c > K j | dd }|d S | }|d S dg |d} t j || d S # t j $ r*}t
d|||| Y d }~d S d }~ww xY w)Nr success)resultmessagesdataz5Validating %r using schema %r failed with error "%s".)exc_info)schemaget
jsonschemar ValidationErrorr critical)hashabler r r targeterrorr validators r' validate_responsez.validate_middleware..validate_response s &**84488 MM F$$[11 > F
&2vFF / / / / /) OOG
s A# #B 2 BB c B t fd }|S )Nc K t | d }t || d | d< | g|R i | d {V }t j || |S )Ncommandparams)rZ r asyncio
ensure_future)requestargskwargsr r r r r s r' wrapperz5validate_middleware..wrapped..wrapper s WY /00H (8WX%6! !GH 1W6t666v66666666F
!"3"3Hf"E"E F F FMr( r )r r r r s ` r' wrappedz$validate_middleware..wrapped s> q
r( )r ospathdirname__file__) r baser r r r r r r s ` @@@@@@r' validate_middlewarer s )) * *W 4DOOLDDLFFM1 1 1 1 1 1 1 . $ Nr( )(r r loggingr rV collectionsr r r r functoolsr ipaddressr r
r pathlibr r "defence360agent.rpc_tools.validater
SchemaValidatorBaser im360.contracts.configr im360.internals.corer r defence360agent.utils.validater getLoggerr2 r rU rW r r r r9 r rK r( r' r s " " " " " " . . . . . . . . . . : : : : : : : : : : 8 7 7 7 7 7 , , , , , , 4 4 4 4 4 4 4 4 - - - - - - 8 $ $
?VW,=>>
Z wo
6
6
(2 2 2 2 2 Z 2 2 2 ja a a a a) a a aHC C C C Cr(