#~/BF V d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZmZ d d lmZ d d
lmZ d d lmZmZ d d lmZ G d
de Z G d de Z G d de Z G d de Z G d de Z G d de Z!dS ) N)ListTuple) DoesNotExist)run_in_executor)ips_for_country)csf)SourceInterface)Country CountryList)IPList)GlobalWhitelist)local_dns_from_resolv_conflocal_ip_addresses)validatec 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )DBIPListCacheSourcec d || _ t j j | j | t j j | j | t j j | j | t j j
| j | d S N) listnamer Signalsaddedconnect _on_addeddeleted _on_deletedcleared _on_clearedupdated _on_updated)selfr s N/opt/imunify360/venv/lib/python3.11/site-packages/im360/model/cache_sources.py__init__z DBIPListCacheSource.__init__ s
$$T^X>>> &&t'7BBB &&t'7BBB &&t'7BBBBB c T | j | |j |j d S Nip
expiration)r sendr' r( r senderr' s r! r z DBIPListCacheSource._on_added s&
2=AAAAAr# c > | j | | d S N)r' r r) r* s r! r z DBIPListCacheSource._on_deleted" s# $2&&&&&r# c : | j | d S r )r r) )r r+ s r! r z DBIPListCacheSource._on_cleared% s $ r# c T | j | |j |j d S r% )r r) r' r( r* s r! r z DBIPListCacheSource._on_updated( s) $25R]CCCCCr# c b K fd}t t j | d {V S )Nc N t t j t j t j t j j k t j S r )
listr selectr' r( wherer
is_expiredtuplesexecuter s r! r4 z-DBIPListCacheSource.fetch_all..select, sj
fi):;;v$-788)+++,,
r# )loopcb)r asyncioget_event_loop)r r4 s ` r! fetch_allz DBIPListCacheSource.fetch_all+ sP %'*@*B*BvNNNNNNNNNNr# N) __name__
__module__ __qualname__r" r r r r r> r# r! r r s| C C CB B B' ' ' D D D
O
O
O
O
Or# r c h e Zd Z fdZed d Zed Zd Zd Z d Z
fdZ xZ S )
CountryIPListCacheSourcec t | t j j | j | t j j | j | d S r ) superr" r r r r #_CountryIPListCacheSource__on_addedr %_CountryIPListCacheSource__on_deleted)r r __class__s r! r" z!CountryIPListCacheSource.__init__9 s^
"""!))$/8DDD#++D,=xHHHHHr# Nc < t j t j t j | k }n# t $ r g cY S w xY w|sg S t t |d S )z+
Return country's ip list.
r )
r
r4 coder5 idr7 firstr r3 r )
country_idr country_codes r! __selectz!CountryIPListCacheSource.__select? s
w|,,wzZ/00
L III IOLO44555s A#A& & A54A5c L | | D ]
} || d S r )!_CountryIPListCacheSource__select)clsrN actionsubnets r! __on_signalz$CountryIPListCacheSource.__on_signalQ s8 ll:.. F F6NNNN r# c F |d fd d S )NrN c > j | d S )Nr r& )r r) xr s r! z5CountryIPListCacheSource.__on_added..[ s djoodqQo?? r# $_CountryIPListCacheSource__on_signal)r _senderkwargss ` r!
__on_addedz#CountryIPListCacheSource.__on_addedX s9 < ? ? ? ?
r# c F |d fd d S )NrN c < j | S r- r. rY s r! r[ z7CountryIPListCacheSource.__on_deleted..` s DL,=,=dq,=,I,I r# r\ )r r+ r_ s ` r! __on_deletedz%CountryIPListCacheSource.__on_deleted^ s9 < "I"I"I"I
r# c g }t j t j t t j t j k t j | j k
}|D ] \ }|d t | D z
}!|S )N)onc g | ]}|d fS )r rB ).0rZ s r!
z8CountryIPListCacheSource.__fetch_all..n s BBBqQFBBBr# ) r r4 r
rK joinrL rN r5 r r7 r8 r )r ips
country_codesrO s r! __fetch_allz$CountryIPListCacheSource.__fetch_allc s
w| , ,
T'wz[-C CT
E
E
U;'4=8
9
9
VXX
WYY - C CO\ BBOL$A$ABBB BCC
r# c K t t j | j d {V }|t d {V z S r )r r< r= $_CountryIPListCacheSource__fetch_allrF r> )r country_subnetsrI s r! r> z"CountryIPListCacheSource.fetch_allq st / " $ $d&6!
!
(9(9(;(;";";";";";";<
__classcell__)rI s @r! rD rD 8 s I I I I I 6 6 6 \6" [
= = = = = = = = =r# rD c e Zd Zd Zd ZdS )ConstantCacheSourcec ( d |D | _ d S )Nc g | ]}|d fS r rB rg r' s r! rh z0ConstantCacheSource.__init__..{ s ...Bb$Z...r# _ips)r rj s r! r" z ConstantCacheSource.__init__z s ..#... r# c K | j S r rx r9 s r! r> z ConstantCacheSource.fetch_all} s yr# N)r? r@ rA r" r> rB r# r! rt rt y s2 / / / r# rt c e Zd Zd ZdS )GlobalwhitelistCacheSourcec f K t d t j d{V D S )z":rtype: iterable[(ip, expiration)]c 3 K | ]}|d fV d S r rB rw s r! z7GlobalwhitelistCacheSource.fetch_all.. s& FF2RJFFFFFFr# N)r3 r
loadr9 s r! r> z$GlobalwhitelistCacheSource.fetch_all s> FF/C/E/E)E)E)E)E)E)EFFFFFFr# N)r? r@ rA r> rB r# r! r| r| s( G G G G Gr# r| c z e Zd Ze ej d deeedf fd Z d Z
dS )&WhitelistCacheSourceFromSystemSettings )maxsizereturnNc t t t t j j t }d t
t j t | D S )Nc g | ]}|d fS r rB rw s r! rh zEWhitelistCacheSourceFromSystemSettings._fetch_all.. s,
J
r# )
mapstrr IPipv6_to_64networkr set itertoolschainr )rS local_ipss r!
_fetch_allz1WhitelistCacheSourceFromSystemSettings._fetch_all so X[24F4H4HII
: < z0WhitelistCacheSourceFromSystemSettings.fetch_all s r# ) r? r@ rA rq functools lru_cacher r r r r> rB r# r! r r sq Y###
4c4i 01
$# [
! ! ! ! !r# r c e Zd Zd Zd Zd ZdS )WhitelistCacheSourceFromCSFc || _ d S r ) csf_files)r r s r! r" z$WhitelistCacheSourceFromCSF.__init__ s
"r# c t t d t j d | j D D }d |D S )Nc 3 K | ] \ }}|V
d S r rB )rg r' _s r! r z5WhitelistCacheSourceFromCSF._fetch.. s: B r# c 3 > K | ]}t j | V d S r )r
ips_from_file)rg fs r! r z5WhitelistCacheSourceFromCSF._fetch.. s- CCqc'**CCCCCCr# c g | ]}|d fS r rB rw s r! rh z6WhitelistCacheSourceFromCSF._fetch.. s 444rT
444r# )r3 r r r r )r ips_from_filess r! _fetchz"WhitelistCacheSourceFromCSF._fetch sj &_CCDNCCC
54^4444r# c . K | S r )r r9 s r! r> z%WhitelistCacheSourceFromCSF.fetch_all s {{}} r# N)r? r@ rA r" r r> rB r# r! r r sA # # # 5 5 5 r# r )"r< r r typingr r peeweer $defence360agent.model.simplificationr "im360.internals.core.ipset.countryr im360.subsysr im360.utils.tree_cacher im360.model.countryr
r im360.model.firewallr im360.model.global_whitelistr
im360.utils.netr r defence360agent.utilsr r rD rt r| r r rB r# r! r s) @ @ @ @ @ @ > > > > > > 2 2 2 2 2 2 4 4 4 4 4 4 4 4 ' ' ' ' ' ' 8 8 8 8 8 8 + * * * * * O O O O O/ O O OD>= >= >= >= >= 2 >= >= >=B / G G G G G G G G ! ! ! ! !_ ! ! !$ / r#