*{RZ d dl Z d dlZd dlmZ d dlmZmZmZmZm Z m
Z
m Z d dl m
Z
mZmZmZ d dlZd dlmZ d dlmZ G d de Z G d
d e Z G d d
e ZdS ) N)ABCMeta)
IPV4LENGTH
IPV6LENGTH IPv4Address IPv4Network IPv6Address IPv6Network
ip_network)IteratorListOptionalUnion)Signal)IPc d e Zd Z e Z e Z e Z e Zd ZdS )SourceInterfacec " K t )z":rtype: iterable[(ip, expiration)]NotImplementedErrorselfs P/opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/tree_cache/core.py fetch_allzSourceInterface.fetch_all" s !### N) __name__
__module__ __qualname__r addeddeletedclearedupdatedr r r r r sU FHHE fhhG fhhG fhhG$ $ $ $ $r r ) metaclassc x e Zd Zd Zd Zdeeeee e
f fdZ deeeee e
f fdZ d Z
dS )TreeCacheInterfacec " K t )zICheck if the cache contains specified ip.
:type ip: str
r r ips r containszTreeCacheInterface.contains(
"###r c " K t )dCheck if the cache contains exactly specified ip, not parent subnet.
:type ip: str
r r' s r contains_exactlyz#TreeCacheInterface.contains_exactly/ r* r ipsc " K t )z'Returns ips that is presented in cache.r r r. s r filter_containedz#TreeCacheInterface.filter_contained6 r* r c " K t )z+Returns ips that is NOT presented in cache.r r0 s r filter_not_containedz'TreeCacheInterface.filter_not_contained= r* r c t )*
re-fill on the next call
r r s r resetzTreeCacheInterface.resetD s "###r N)r r r r) r- r r r r r r r1 r3 r6 r" r r r% r% ' s $ $ $$ $ $$
% [+{JK
L$ $ $ $$
% [+{JK
L$ $ $ $$ $ $ $ $r r% c h e Zd ZddZd Zd Zd Zd Zd Zd Z e
d e e e
eef fd
Zed e e e
eef fd Zed Zd
Zd Zd Zdee e e
eef fdZdee e e
eef fdZd edee fdZ d Z ed Z dede fdZ d Z!dS ) TreeCache c L || _ || _ d| _ d| _ d| _ |j | j | |j | j | |j
| j | |j | j
| dS )z :type source: SourceInterfacer NF)_source_full_sync_period
_last_sync_tree_expiredr connect _on_addedr _on_deletedr _on_clearedr! _on_updated)r sourcefull_sync_periods r __init__zTreeCache.__init__L s !1
T^V444 t /888 t /888 t /88888r c r | j d S | j t j | | d S N)r> insertr adopt_to_ipvX_networkr senderr(
expirations r rA zTreeCache._on_added\ s8 : F
"2266
CCCCCr c | j d S | j t j | d S # t $ r Y d S w xY wrI )r> deleter rK KeyError)r rM r( s r rB zTreeCache._on_deleteda s] : F J b 6r : : ; ; ; ; ; DD s ,9
AAc B t j t | _ d S rI )pytriciaPyTriciar r> )r rM s r rC zTreeCache._on_clearedj s &z22
r c 4 | ||| d S rI )rA rL s r rD zTreeCache._on_updatedm s vr:.....r c K t j t | _ | j }| j d {V D ]-\ }}| t j | | .d S rI ) rS rT r r> r; r rJ r rK )r treer( rN s r
_init_treezTreeCache._init_treep s &z22
z$(L$:$:$<$< < < < < < < B BNB
KK044j A A A A B Br c K | j $t j | j | j z k r_| d {V t j | _ | j 9t j | j | j z k ]d S d S rI )r> timer= r< rX r s r _sync_if_neededzTreeCache._sync_if_neededv s j DIKK Od 4 4%
%
//## # # # # # # # "ikkDO j DIKK Od 4 4%
%
%
%
%
%
r r( c R | || \ }}t | S )zM
:return bool: if 'tree' contains ip as is or by subnet mask
)_lookupbool)clsrW r( ip_nwk_s r _containszTreeCache._contains} s' KKb)) F||r c 2 t j | }t d D ]e}| | }|s dS | | }|r-|t j k r| | `||fc S t d )zLookup specified ip or parent subnet considering expiration.
:return tuple: str(ip_network(contains that ip)), expiration
)NNzDToo deep recursion. Something goes wrong. Please contact developers.)r rK rangeget_keygetrZ rP RuntimeError)rW r( ra r` rN s r r] zTreeCache._lookup s
%b
)
) s . .A\\"%%F
.!zz "XXf - -
.* ";";KK''' !:----
)
r c t j | }| | r0t | | r| | S dS )NF)r rK has_keyr8 rb )rW r( s r _contains_exactlyzTreeCache._contains_exactly sU
%b
)
) << (""4,,
(||B'''ur c p K | d{V | | j | S )zCheck if the cache contains specified ip or parent subnet.
:type ip: str
:return tuple: str(ip_network(contains that ip)), expiration
N)r[ r] r> r' s r lookupzTreeCache.lookup sB ""$$$$$$$$$||DJ+++r c p K | d{V | | j | S )zZCheck if the cache contains specified ip or parent subnet.
:type ip: str
N)r[ rb r> r' s r r) zTreeCache.contains sB
""$$$$$$$$$~~dj"---r c p K | d{V | | j | S )r, N)r[ rk r> r' s r r- zTreeCache.contains_exactly sD
""$$$$$$$$$%%dj"555r r. c ` K |s|S d{V fd|D S )z+Returns ips that is presented in the cache.Nc J g | ] } j | | S r" rb r> .0r( r s r
z.TreeCache.filter_contained.. s. CCCrDNN4:r$B$BCCCCr r[ r0 s ` r r1 zTreeCache.filter_contained sT
J""$$$$$$$$$CCCCSCCCCr c ` K |s|S d{V fd|D S )z/Returns ips that is NOT presented in the cache.Nc J g | ] } j | | S r" rr rs s r ru z2TreeCache.filter_not_contained.. s. GGGrtz2(F(FGGGGr rv r0 s ` r r3 z TreeCache.filter_not_contained sT
J""$$$$$$$$$GGGGSGGGGr returnc | K | | d{V }t j |d r||k r|S dS )z%Returns IP network that contains `ip`NTstrict)_get_keyr is_valid_ip_network)r r( keys r
get_subnetzTreeCache.get_subnet sU MM"%%%%%%%% !#d 3 3 3 r Jtr c K t j | s
J d|z | d {V | j | S )Nz%s is not valid IP!)r r~ r[ r> rf )r ip_ntws r r} zTreeCache._get_key se %f--MM/Dv/MMM-""$$$$$$$$$z!!&)))r c $ t d | S )Nc t | j t k r!t t | j n| S rI )r
prefixlenr strnetwork_address)r( s r z5TreeCache._fix_addresses_in_network.. s7 b>>+z99 JrNN2333 r )map) network_lists r _fix_addresses_in_networkz#TreeCache._fix_addresses_in_network s%
r
target_subnetc z K t j d st g S | d{V | j }|st g S fd| j | D }t j | | g S )z
Returns a iterator of IP address and networks that located in cache
and being members of `target_subnet` and `target_subnet` itself
Tr{ Nc 3 b K | ])}t | j t v %|V *d S rI )r
r )rt netr s r z0TreeCache.get_ips_from_subnet.. sK &
&
#.*]2K2KKK
KKKK&
&
r )
r r~ iterr[ r> rf children itertoolschainr )r r
parent_subnetitems_in_subnet_from_caches ` r get_ips_from_subnetz TreeCache.get_ips_from_subnet s
%mDAAA 88O""$$$$$$$$$
**=99
88O&
&
&
&
z**=99&
&
&
"
* *+E F F
O
r c d| _ dS )r5 N)r> r s r r6 zTreeCache.reset s
r N)r9 )"r r r rG rA rB rC rD rX r[ classmethodr r r r r rb staticmethodr] rk rm r) r- r r1 r3 r r
r r} r r r r6 r" r r r8 r8 K s( 9 9 9 9 D D D
3 3 3/ / /B B B * * *
+{KD E [
[+{JK
\
@ \, , ,. . .6 6 6
D
% [+{JK
L
D
D
D
D
H
% [+{JK
L
H
H
H
H 3 8C= * * *
\
s
x
. r r8 )r rZ abcr ipaddressr r r r r r r
typingr r r
r rS blinkerr defence360agent.utils.validater r r% r8 r" r r r sw 3 2 2 2 2 2 2 2 2 2 2 2 - - - - - -$ $ $ $ $ $ $ $ $$!$ !$ !$ !$ !$7 !$ !$ !$ !$HS S S S S" S S S S Sr