L ^ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z m Z d dl
mZ ej e Z G d de Z G d de Zd e e d
e
d e fd Z G d
de Zdej ej ej ej ffdZdddZ d e fdZ d e fdZ dS ) N)Path)AnyCallableType)LazyLockc e Zd ZdS ) RulesLockN__name__
__module__ __qualname__ I/opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/__init__.pyr r Dr r c e Zd ZdS )
ModSecLockNr
r r r r r r r r errorvaluereturnc fd}|S )a.
Decorator that causes coroutine to return *value* if it raises *error*.
For example:
return await return_value_on_error(FileNotFoundError, X)(coro)(*args)
is equivalent to:
try:
return await coro(*args)
except FileNotFoundError:
return X
c L t j fd }|S )Nc K | i | d {V S # $ r)}t d| cY d }~S d }~ww xY w)Nz Replacing %s from %s with %s)loggerinfo)argskwargsecoror r s r wrapperz9return_value_on_error..decorator..wrapper) sz
!T4 26 2 22222222
:AtUKKK
s
A <AA) functoolswraps)r r r r s ` r decoratorz(return_value_on_error..decorator( sC
r r )r r r# s `` r return_value_on_errorr$ s* r c e Zd Z fdZ xZS )PasswordChangeErrorc V t d| d| d S )Nz)Password change process exited with code z: )super__init__)selfstderr
returncode __class__s r r) z PasswordChangeError.__init__7 s<
N
N Nf N N
r )r r r
r)
__classcell__)r- s @r r& r& 6 s8
r r& c | t | k r"t d| dt | t t t | s
J d d | d fdt | D t fd|D rS L)z
Return password with *length* char.
It is guaranteed that password have at least one character from
each given *required_char_groups*
z
Given length=zR is too short. Can't get chars from all required groups len(required_char_groups)=zgot empty char group Tc 8 g | ]}t j S r )secretschoice).0_alphabets r
z,generate_strong_password..] s# LLLGN8 4 4LLLr c 3 P K | ] t fd D V dS )c 3 K | ]}|v V d S )Nr )r5 cgroups r z5generate_strong_password...` s' --qU
------r N)any)r5 r< passwords @r r= z+generate_strong_password.._ sR
27C----H--- - -
r )len
ValueErrorallmapjoinrange)lengthrequired_char_groupsr7 r? s @@r generate_strong_passwordrH = s ())))
,V
,
,'((
,
,
ww+,,H 77LLLLeFmmLLLMM
;O
O r )
passwd_cmdc h |!t j dt j g}t |t
sJ t
j g || d | gdz dd d S # t j
$ r }t |j |j
d }~ww xY w)Npasswd)path
T)inputcapture_outputcheck)shutilwhichosdefpath
isinstancelist
subprocessrunrD encodeCalledProcessErrorr& r+ r, )username new_passwordrI r s r change_system_passwordr^ e s l8"*===>
j$ ' ''' ': #j #( #**l 1 1 3 3 4q 899
( : : :!!(AL999:s AB B1B,,B1c t d } | rW | }|dk S # t $ r t
d Y nw xY wdS )zHCheck if apache2nginx is enabled by reading /var/lib/apache2nginx/state.z/var/lib/apache2nginx/stateonz&Failed to read apache2nginx state fileF)r is_file read_textstripOSErrorr exception)
state_filecontents r is_apache2nginx_enabledrh t s 344J G G **,,2244Gd? " G G G E F F F F F G 5s +A $A87A8c t d } | dk S # t $ r Y n1t $ r%}t
d| Y d}~nd}~ww xY wdS )z7Check if coraza4cpanel is enabled by reading flag file.z./usr/share/imunify360-webshield/standalone_wafcorazaz*Failed to read coraza4cpanel flag file: %sNF)r rb rc FileNotFoundError Exceptionr r )rL r s r use_coraza4cpanelrm s @ A ADJ~~ %%''833
J J J A5IIIIIIIIJ 5s );
A5 A5A00A5) r! loggingrT r3 rR stringrX pathlibr typingr r r defence360agent.utilsr getLoggerr r r r rl r$ r& ascii_lowercaseascii_uppercasedigits punctuationrH r^ boolrh rm r r r ry s
& & & & & & & & & & * * * * * * 8 $ $ i <
)
% % % % P BF : : : : : 4 r