OL d dl mZ d dlmZmZmZmZmZmZm Z m
Z
m Z d dl m
Z
mZmZ e
d Z G d dee Z G d dee Zd Zdd Z G d d
ZdS ) )Enum)
IPV4LENGTH
IPV6LENGTHAddressValueError IPv4Address IPv4Network IPv6Address IPv6Network
ip_address
ip_network)LiteralOptionalUnion)ipv4ipv6c e Zd ZdZdZd ZdS ) LocalhostIPz 127.0.0.1z::1c | j S N)valueselfs S/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/validate.py__str__zLocalhostIP.__str__ s
z N)__name__
__module__ __qualname__r r r r r r r s- D D r r c X e Zd ZdZdZdZd Zedee ded fd Z
dS ) NumericIPVersionz=Example: (IPListRecord.version==NumericIPVersion[ip_version]) c * t | j S r )strr r s r r zNumericIPVersion.__str__ s 4: r
ip_versionreturnc F |d S |t j k r| j n| j S r )IPV4r r )clsr& s r from_ip_versionz NumericIPVersion.from_ip_version# s) 4%..sxxCHz"IP.check_ip_ver..; s' >>c7c>>>>>>>r )anyr* r7 )r+ r< s `r check_ip_verzIP.check_ip_ver9 s/ >>>>cfcf-=>>>>>>r c J t | n# t $ r Y dS w xY wdS NFT)r
ValueErrorr+ r2 s r is_valid_ipzIP.is_valid_ip= s? t 55 t
c 6 | j |i |p | j |i |S r )r4 is_valid_ipv6_network)r+ argskwargss r is_valid_ip_networkzIP.is_valid_ip_networkE s> (s(
8
&S
&
7
7
7 8r c J t | n# t $ r Y dS w xY wdS rA )r r rC s r r1 zIP.is_valid_ipv4_addrK ? 55 trE c J t | n# t $ r Y dS w xY wdS rA )r r rC s r is_valid_ipv6_addrzIP.is_valid_ipv6_addrS rL rE Fr2 c n t | }n# t $ r Y dS w xY w|r|j t k S dS rA )r rB prefixlenr r+ r2 r5 ips r r4 zIP.is_valid_ipv4_network[ T T""BB 55 .<:- -trE c n t | }n# t $ r Y dS w xY w|r|j t k S dS rA )r
rB rP r rQ s r rG zIP.is_valid_ipv6_networki rS rE c | | r t j S | | r t j S t d )NzInvalid ip address)r4 r) r* rG r7 rB rC s r type_ofz
IP.type_ofw sK $ $T * * 5L
&
&t
,
, 5L-...r /64c H t ||z d }t | S )zConver ipv6 addr to ipv6 network with mask
:param str ip: ip for converting
:param str mask: ip network mask
F)r5 )r
r% )r+ rR masknetworks r convert_to_ipv6_networkzIP.convert_to_ipv6_network s& b4i6667||r ip_argr' c t | t t f r| S t | t t f r7| j dk rt nt }t t | |f S t | S )zt
Eliminate str from the Union
:raise ValueError: if cannot convert ip_arg str to ip network
r" )
isinstancer r
r r r< r r r int)r\ rP s r adopt_to_ipvX_networkzIP.adopt_to_ipvX_network ss f{K 8 9 9 8M
k :
;
; 8&,n&9&9
zI s6{{I 677 7&!!!r netc p t |j st |j S t | S )zz
IPv4Network('192.168.1.1/32') -> '192.168.1.1'
IPv4Network('192.168.1.0/24') -> '192.168.1.0/24'
)r_ hostmaskr% network_address)r+ ra s r ip_net_to_stringzIP.ip_net_to_string s2 3< ,s*++ +3xxr rR c t |t r/t | t | S |S r )r^ r r
r[ r% )r+ rR s r ipv6_to_64networkzIP.ipv6_to_64network s> b+ & & E s : :3r77 C CDD D r NF)rW )r r r r* r/ __annotations__r7 r. r? rD rJ r1 rN r r% r r
r4 rG rV r[ staticmethodr r r` re rg r r r r) r) 5 s6 B B ? ? [? [ 8 8 [8
[ [ @E k;67 [ @E k;67 [ / / [/ [
"c; [+MN
" {K' (
"
"
" \
" 5k)A#B s [ {K45 {K' ( [ r r) Nrh )enumr ipaddressr r r r r r r
r r typingr
r r r/ r% r r_ r! r1 r4 r) r r r rn sf
, + + + + + + + + + N # #t = = = = =sD = = =&' ' '2 2 2 2u u u u u u u u u ur