x^w d dl Z d dlZd dlmZ d dlmZmZmZ ej e Z
dxZ \ Z Z
Z G d de Zed Zeeeeee f f Z G d d ZdS ) N)Path)ListLiteralUniontcpudpallc e Zd ZdS )IpPortDenyModeErrorN)__name__
__module__ __qualname__ S/opt/imunify360/venv/lib/python3.11/site-packages/im360/model/port_ips_deny_mode.pyr r s Dr r c e Zd ZdZedefd Zedededefd Zede e
d e e
d f de fd
Z
d S ) WhitelistPortIPsDenyModez /etc/imunify360/whitelist/portsreturnc t i t i i}t | j d D ]*}| | | }+|S )z Read port:networks from the filez*.txt)TCPUDPr
_LIST_PATHglob
_load_fileabsolute)clsport_to_networkspaths r loadz WhitelistPortIPsDenyMode.load sg S"- ((--g66 D "~~
!1 r r r c t | D ]\ }}| d d }|s6g }d}| d \ }} }
t
t g} |
d \ } }
}| } | t v r| t k r| g} n|
}| sd} t | }| d D ]U}| }|r= t j
| | | C# t $ r d}Y Qw xY wV| D ]g} |rc|| |g }| | t% t' t) | || |< hn# t $ r d}Y nw xY w|s t* d|| |S )N#r T:F,z1Wrong IP/Subnet format in %s:%s. Use CIDR format.) enumerateopen readlines partitionstripr r lowerPROTOSALLintsplit ipaddress
ip_networkappend
ValueError
setdefaultextendsortedlistsetloggererror)r r r line_numlineitem network_listvalidport semicolontailprotosproto_networksnetworkexisting_portss r r z#WhitelistPortIPsDenyMode._load_file! sY ( (=(=(?(? @ @ 3 3 NHd>>#&&q)//11D
LE$(NN3$7$7 !D)T#&*F!% !4!4 E1h ;;==..00EC<< #WF
4yy (~~c 2 2 * *G %mmooG ** % 0 9 9 9 ( / / 8 8 8 8 ) * * *$)EEE* * $ E# )9%)@)K)K "* * '--l;;;8> ^!4!4 5 59 9(/5
,
G s$ 3G0)E E)(E) G'&G'rC r r r c . |x t |t r t d |D r |}n1xxdk rn xdk rn n |g}ndk rddg}n t d | t fd|D S )Nc 3 K | ]}|d v V dS )rH Nr ).0ps r z1WhitelistPortIPsDenyMode.count..d s8 6 6()^#6 6 6 6 6 6r r r r
z
Invalid inputc 3 z K | ]5}t d | D V 6dS )c 3 4 K | ]}t | V d S )N)len)rK rE s r rM z;WhitelistPortIPsDenyMode.count...p s( II(H
IIIIIIr N)sumvalues)rK rL proto_to_portss r rM z1WhitelistPortIPsDenyMode.count..o s_
IInQ.?.F.F.H.HIII I I
r )
isinstancer6 r
r2 r rQ )r rC rB rS s @r countz WhitelistPortIPsDenyMode.count\ s j--# 6 6-26 6 6 3 3 111
r N)r
r r r classmethodPortNetworkMapr r r r r r r- rU r r r r r s 2J ^ [ 8 8 +98 8 8 8 [8 t
' ($w|/D*E E
[
r r )r/ loggingpathlibr typingr r r getLoggerr
r8 r+ r r r, Exceptionr Protodictr- strrW r r r r r` s ' ' ' ' ' ' ' ' ' ' 8 $ $ 1 0 #sC ) #$eT#tCy. 112^
^
^
^
^
^
^
^
^
^
r