{ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z d dlm
Z
mZmZmZmZ d dlZd dlmZ d dlmZ d dlmZ d dlmZmZm Z d d l m Z d
Z e d Z ej! e" Z#d Z$ ed
de%fd Z&d Z' G d de( Z) G d de( Z* G d de) Z+d dZ, G d d Z- G d d Z.d Z/dS ) N) defaultdict)Path) parse_qsl urlencodeurljoinurlparse
urlunparse)ANTIVIRUS_MODE)
LicenseCLN) HostingPanel)
CheckRunErrorasync_lru_cache check_run) get_hostnamei, z/usr/sbin/ie-configzpwget -qq -O - https://repo.imunify360.cloudlinux.com/defence360/imunifyemail-deploy.sh | bash -s 'is-supported' )maxsizereturnc K t t d d {V nM# t $ r@} | j dk r*t dt
| Y d } ~ dS d } ~ ww xY wdS )NT)shelld zimunify-email check failed F)r IE_SUPPORTED_CMDr
returncodeloggererrorstr)es R/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/internals/cln.pyis_imunify_email_supportedr s (55555555555 <3 LL?s1vv?? @ @ @uuuuu 4s !
A+5A&&A+c K t rdS t sdS t t t dg d{V } n# t
$ r Y dS w xY wd| v S )z Try to get imunify-email statusFstatusNz&spamfilter exim configuration: enabled)r
_IMUNIFY_EMAIL_CONFIG_EXECUTABLEexistsr r r
decode)outputs r get_imunify_email_statusr% * s u + 2 2 4 4 u
1
2
2H =
uu 3v}} FFs )A
A A c e Zd ZddZd ZdS )CLNErrorNc " || _ || _ d S Nmessager )selfr r+ s r __init__zCLNError.__init__: s c R | j r| j S d | j S )Nz#Unexpected status code from CLN: {})r+ formatr r, s r __str__zCLNError.__str__> s* < < 4;;DKHHHr. )NN)__name__
__module__ __qualname__r- r2 r. r r' r' 9 s< I I I I Ir. r' c e Zd ZdS )InvalidLicenseErrorN)r3 r4 r5 r6 r. r r8 r8 E s Dr. r8 c * e Zd ZdZd Zd Zd Zd ZdS )BackupNotFoundi @c || _ d S r) url)r, r= s r r- zBackupNotFound.__init__L s
r. c dS )NzBackup not found in CLNr6 r1 s r r2 zBackupNotFound.__str__O s ((r. c " | j d S t | j }t t |j }| |d< t
|j |j |j |j
t | |j f S )N
used_space)
r= r dictr query _disk_usager schemenetlocpathparamsr fragment)r, purB s r add_used_spacez BackupNotFound.add_used_spaceR s 8 F
dh
Yrx(()) " . . 0 0l %
r. c N d}t j }t }|D ]i}|j |vr^d|j vrU|j d s;|t j |j j z
}| |j jt || j z S )Nr noautoz /dev/loop) psutildisk_partitionssetdeviceopts
startswith
disk_usage
mountpointusedaddroundGB)r,
total_used
partitions processedps r rC zBackupNotFound._disk_usagee s
+--
EE ( (A** QV++,,[99 , f / = = BB
ah'''Z$')***r. N)r3 r4 r5 rX r- r2 rJ rC r6 r. r r: r: I sU B ) ) )
& + + + + +r. r: c i }|||d< |t |t r| dddi nt |t r.| d }| dddi nJt
j | d }| dddi ||d < t
j t j j
| fi ||
}|5 |j d k r|j dfcddd S |j d v r | } |j t j | fcddd S # t j $ r'}t# d
| d|j |j |d}~ww xY w# t$ j $ r t) d w xY wt# |j # 1 swxY w Y dS # t
j j $ r}|j dk rt# |j |d}|j wt0 d| |||j |j | } n"# t$ j $ r t) d w xY w| d }t# ||j |d}~wt
j j $ r#}t# t | |d}~wt$ j $ r t) d t8 $ r%}t0 d| |||| d}~ww xY w)z!To be used by RestCLN._request().Nheaders Content-typezapplication/octet-streamzutf-8ztext/plain; charset=utf-8asciiz!application/x-www-form-urlencodeddata)timeout ) zNon-json data from CLN: z
for code=r* zTimed out reading responsei z,CLN.post(url=%r, data=%r, headers=%r): %d %sz Timed out reading error messagereplace)errors)r+ z Timed out receiving responsez5CLN.post(url=%r, data=%r, headers=%r, timeout=%r): %s)
isinstancebytes
setdefaultr encodeurllibparser requesturlopenRequestcodereadjsonloadsr# JSONDecodeErrorr' socketrb TimeoutErrorr HTTPErrorfpr warningreasonURLErrorOSError)
r= ra r^ rb kwargsrespcontentr r+ resp_datas
r
_post_requestr t sX
F #y dE " " N,FG
c
"
"
;;w''D N,GH
<))$//66w??D !DE
v?*~%% N "3 1 1& 1 17 &
R * *yC y$ * * * * * * * * j((! "iikkG ! #y$*W^^5E5E*F*F F * * * * * * * * / ! ! ! &!57 !5 !5)-!5 !5 $(9 !
!! ~ E E E&'CDDDE ty)))+ * * * * * * * * * *M < ! > > > 6C<<16"" ) 4 NN>
F FFHH >
F
F
F"#DEEE
F &&i&88Gwqv666A= < . . .s1vv&&&A- > ; ; ;9::: C
s
6G: G-& G-0F7,E>>F4
"F//F44G-7 GG--G14G1:M AK!I65K6 J0KM K::*M $ MM c j e Zd ZdZdZej dd Zej dd Z e
ej de Z e e d Z
e e d Z e e d Z e e d
Z e e d Z e e d Zd
ZdZeddedd Zed Zededefd Ze d dede defd Z ededefd Z edefd Z ededefd Z ed d Z!dS ) RestCLNzhttps://{domain}/api/im/zcln.cloudlinux.comIM360_CLN_API_BASE_URLzipv6.cln.cloudlinux.comzipv4.cln.cloudlinux.comdomainregister
unregistercheckinzab/credentialsz ab/removezab/checkokzok-trialN)ra r^ rb c r K t j d t |||| d {V S r) )asyncioget_event_looprun_in_executorr )clsr= ra r^ rb s r _requestzRestCLN._request sR +--== -dGW
r. c K t | j | j d }dt d} | || d {V \ }}ns# t $ rf}|j dk rOt | j | j d }| || d {V \ }}n|Y d }~nd }~ww xY w|S )Nr r IPLkeyhostnamera i ) r _URL_PATH_TEMPLATEr0 _IPV4_DOMAIN_NAMEr r r' r _IPV6_DOMAIN_NAME)r v4_license_urlra _token cln_errorv6_license_urls r process_ipl_licencezRestCLN.process_ipl_licence s! " ) )1F ) G G
,..99
\\.t\ D DDDDDDDHAuu 3&&!(*11 " 4 2 " " "% n4 !H!HHHHHHH55 5555 s A$ $
C.A CCr r c K |dk r| d{V S | | j |t d d{V \ }}|S )z
Register server with key
:param key: registration key
:return: license token in case of success
r Nr r )r r
_REGISTER_URLr )r r r r s r r zRestCLN.register s %<< 0022222222 2 ,..99 &
5 r. server_id users_countr c " K |p
t }t d{V }t } | d{V }nH# t $ r;}t
dt | d |j }Y d}~nd}~ww xY w|||||dt d{V idd}t j | } t
d| |
| j | d d
i d{V \ }
} | S ) z
Update license token
:param str server_id: server id
:param int users_count: users count
:param str hostname: current server hostname
:return: dict new license token
Nz Failed to get panel version: %sT)exc_infoIM_EMAIL)userspanel imunifyEmailsupported_features)idr imzCLN checkin: %sr_ zapplication/json)ra r^ )r r% r name Exceptionr r r NAMEr rs dumpsinfor _CHECKIN_URL) r r r r imunify_email_statusr
panel_namer reqra r r s r r zRestCLN.checkin s~ -|~~%=%?%? ? ? ? ? ? ? $ $zz||++++++JJ $ $ $ LL13q66D
JJJJJJ $ $# 4 &@&B&B B B B B B B'
z# %t,,, #%78 &
5
s A
B1BBc X K | | j d|i d{V \ }}|S )zl
Creates Acronis Backup account and get user & password
:param server_id: server id
r r N)r _ACRONIS_CREDENTIALS_URL)r r r credss r acronis_credentialszRestCLN.acronis_credentials< sV (i/@ &
5 r. c R K | | j d|i d{V dS )zT
Removes Acronis Backup account
:param server_id: server id
r r N)r _ACRONIS_REMOVE_URLr r s r acronis_removezRestCLN.acronis_removeG s> ll32$ 9JlKKKKKKKKKKKr. c K | | j d|i d{V \ }}|dk rt d |S )z
If Acronis account exists return backup size in GB or if backups
not exists URL for backups
:param server_id: server id
r r Nre r< )r _ACRONIS_CHECK_URLr: )r r r responses r
acronis_checkzRestCLN.acronis_checkO sq "% "$ ): ". "
"
S== T*** *r. c | K |pt j }| | j d|i d{V dS )z<
Unregister server id
:return: None
r r N)r
get_server_idr _UNREGISTER_URLr s r r zRestCLN.unregister] sQ ;!9!;!; ll3.dI5FlGGGGGGGGGGGr. r) )"r3 r4 r5 r _BASE_DOMAIN_NAMEosenvirongetr r r0 _BASE_URLr r r r r r r STATUS_OK_PAID_LICENSESTATUS_OK_TRIAL_LICENSE classmethod_TIMEOUTr r r rA r intr r r r r r6 r. r r r sk 3,
";
"; #))z~~ 68IJJ * I GIz22M gi 66O79i00L &wy2B C C !')[99 J77 ! ()-tX
[
[, [
+ ++ + + + + [+Z # $ [ LS L L L [L C D [ H H H [H H Hr. r c e Zd Z ee Zed Zed Zed Z ed Z
ed Z ed Z dS )CLNc F | j | | d S r) )
_CALLBACKSrV )r method_name
coro_callbacks r add_callback_forzCLN.add_callback_forj s# {#''
66666r. c K | j | D ]h} | d {V # t j $ r t $ r9}t d ||| Y d }~ad }~ww xY wd S )Nz;Error '{!r}' happened when run callback {} forCLN {} method)r r CancelledErrorr r exceptionr0 )r r callbackr s r run_callbacks_forzCLN.run_callbacks_forn s {3 H
hjj )
$$*F1h $D$D
s $A6/A11A6c , | d S )NIMAVP)rR )r r s r
is_avp_keyzCLN.is_avp_key{ s ~~g&&&r. c K | | rt st d t | d {V }t j | s5t |d d {V t d t j | | d d {V d S )Nz4Imunify360 can not be registered with ImunifyAV+ keyr z"License is invalid for this serverr )
r r
r8 r r r is_validr updater )r r licenses r r z CLN.register s >># ~ %F
((-------- "7++ L$$WT]33 3 3 3 3 3 3 3%&JKK K'"""##J///////////r. c K t d {V t j | d d {V d S )Nr )r r r deleter )r s r r zCLN.unregister se """""""""##L11111111111r. c $ K t j rt j S t j d rt j S t |d t j d{V }t d| | t
d{V nt j | | d d{V t j S )z>Refreshes token and returns new one on success, None otherwiseis_alternativer NzGot new token from CLN: %s
refresh_token)
r is_free get_tokenr r r r r r r r r r )r r new_tokens r r zCLN.refresh_token s * ')) ) ! ! % %&6 7 7 * ')) )!//%+z7MNNNNNNNN 0)<<< .."" " " " " " " " " i ( ( (##O444444444#%%%r. N)
r3 r4 r5 r rO r r r r r r r r r6 r. r r r g s S!!J7 7 [7
[
' ' ['
0
0 [
0 2 2 [2
& & [& & &r. r c H dD ] }t || d S )N)r r r )r )r r )coror s r subscribe_to_license_changesr s7 B > > [ ====> >r. )NNN)0r rs loggingr rv urllib.errorrl urllib.parseurllib.request collectionsr pathlibr r r r r r rM defence360agent.contracts.configr
!defence360agent.contracts.licenser +defence360agent.subsys.panels.hosting_panelr defence360agent.utilsr
r r defence360agent.utils.commonr r r! getLoggerr3 r r boolr r% r r' r8 r: r r r r r6 r. r r s
# # # # # # L L L L L L L L L L L L L L
; ; ; ; ; ; 8 8 8 8 8 8 D D D D D D K K K K K K K K K K 5 5 5 5 5 5 #'4(=#>#> 8 $ $ $ G G G I I I I Iy I I I ) (+ (+ (+ (+ (+X (+ (+ (+VU* U* U* U*pXH XH XH XH XH XH XH XHv@& @& @& @& @& @& @& @&F> > > > >r.