d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z m
Z
d dl m Z d dl
mZ d dlmZ d dlmZmZ d d lmZmZ ej Z G d
d ee Z G d d
ee Z G d de ZdS ) N)ABCabstractmethod) defaultdict)heappopheappush)Dict)Core) MessageType)BaseMessageProcessorexpect)is_safe_subdir_namermtreec e Zd Zd Zd Zd Z eej d Z d Z
e d Z e d Z
e d Ze d Ze d
Ze d Zd S )
EventProcessorBasec F t t | _ || _ d S N)r list_msg_buf_loop)selfloops l/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/event_monitor_message_processor.py__init__zEventProcessorBase.__init__ s #D))
c X t | j |d |d |f d S )Nusername timestamp)r r r messages r add_messagez EventProcessorBase.add_message s9 M'*- .1Ew0O
r c z K t j fd j D d {V d S )Nc 3 B K | ]} | V d S r )process_user_messages).0
user_messagesr s r z6EventProcessorBase.process_messages.. sE ! **=99 r )asynciogatherr valuesr s `r process_messagesz#EventProcessorBase.process_messages sr n %)]%9%9%;%;
r c x K | | sd S |j dk r | | d {V d S |j dk r | | d {V d S |j dk r | | d {V d S |j dk r | | d {V d S d S )NModifyCreatechange_packageRemove)_message_is_relatablehook_process_modify_process_create_process_change_package_process_account_removedr s r
process_eventz EventProcessorBase.process_event$ s ))'22 F <8 # #&&w// / / / / / / / / /
\X
%
%&&w// / / / / / / / / /
\ -
-
-..w77 7 7 7 7 7 7 7 7 7
\X
%
%//88 8 8 8 8 8 8 8 8 8 &
%r c K t t | D ]0}| t | d d {V 1d S )N )rangelenprocess_messager )r messages_s r r# z(EventProcessorBase.process_user_messages2 sc s8}}%% = =A&&wx'8'8';<< < < < < < < < < = =r c
K dS )z Modify hookN r s r r3 z"EventProcessorBase._process_modify6
r c
K dS z Create hookNr@ r s r r4 z"EventProcessorBase._process_create: rA r c
K dS zchange_package hookNr@ r s r r5 z*EventProcessorBase._process_change_package> rA r c
K dS )z Remove hookNr@ r s r r6 z+EventProcessorBase._process_account_removedB rA r c dS z'Whether the message should be processedNr@ r s r r1 z(EventProcessorBase._message_is_relatableF r c
K dS z$Whether messages should be processedNr@ r* s r
is_enabledz EventProcessorBase.is_enabledJ rA r N)__name__
__module__ __qualname__r r r+ r r
cPanelEventr7 r# r r3 r4 r5 r6 r1 rL r@ r r r r s
VK #$$ 9 9 %$ 9= = = ^ ^ " " ^" ^ 6 6 ^6 3 3 ^3 3 3r r c
e Zd ZdZd Zd Zd Zd Z ddede d d
fd Z
d Z e d
Z
e d Zee d eeef fd Ze d Zee dede d eeef fd Ze d Zd
S )SettingsChangeBasez'Process hook event messages from cPanelc \ K d|j v rdnd}| || d {V d S )Nplanexclude)data_get_settings_and_update)r r
package_fields r r3 z"SettingsChangeBase._process_modifyR sI "(GL"8"8i
++G]CCCCCCCCCCCr c D K | |dd d {V d S )NrT TrW r s r r4 z"SettingsChangeBase._process_createV s6 ++GVTBBBBBBBBBBBr c D K | |dd d {V d S )Nnew_pkgTrZ r s r r5 z*SettingsChangeBase._process_change_packageY s6 ++GYEEEEEEEEEEEr c
K d S r r@ r s r r6 z+SettingsChangeBase._process_account_removed\ s r FrX add_to_packagereturnNc K t d| | | d {V }| |||| d {V d S )NzGet settings from %s)loggerinfo_get_settings_from_message_apply_settings)r r rX r^ settingss r rW z+SettingsChangeBase._get_settings_and_update_ s *G444 88AAAAAAAA"" ]NH
r c K t d| | d 9|d dk r-t d | D rd S t | s |j | }| || d {V }n># t $ r1 t d | }Y nw xY w|
D ]\ }}| || ||< t d||d |
D ](\ }}| |d || d {V )d S ) Nz
Step 1 %s rT r2 r- c 3 K | ]}|d u V d S r r@ )r$ values r r& z5SettingsChangeBase._apply_settings..t s& AAeETMAAAAAAr z'No information about package in messagez,Settings specified in hook message %s for %sr ) ra rb getallr) rV _get_package_settingsKeyErrorwarning_default_settingsitemson_settings_change) r r rX r^ re package_namefallback_settingsfeaturerh s r rd z"SettingsChangeBase._apply_settingsk s L(+++
KK '8++AAx/@/@AAAAA ,
F8??$$%% C
&|M :
+/*D*D .+ + % % % % % %!!
=
=
= HIII$($:$:$<$<!!!
= #+.."2"2
C
C=(9'(BHW% : J
'nn . . O ONGU))'**=wNN N N N N N N N N O Os
B8 88C32C3c dS rH r@ r s r r1 z(SettingsChangeBase._message_is_relatable rI r c
K dS )z9What to do after settings were changed (e.g. sync the DB)Nr@ )r userrs rh s r rp z%SettingsChangeBase.on_settings_change rA r c dS )z Get default package settingsNr@ r@ r r rn z$SettingsChangeBase._default_settings rI r c
K dS )z"Retrieve settings from the messageNr@ r s r rc z-SettingsChangeBase._get_settings_from_message rA r rq c
K dS )z Get current package settingsNr@ )clsrq r^ s r rk z(SettingsChangeBase._get_package_settings rA r c
K dS rK r@ r* s r rL z SettingsChangeBase.is_enabled rA r )F)rM rN rO __doc__r3 r4 r5 r6 strboolrW rd r r1 rp staticmethodr rn rc classmethodrk rL r@ r r rR rR O s 11D D DC C CF F F
%
O O OB 6 6 ^6 H H ^H +tCH~ + + + ^ \+ 1 1 ^1 + +04+
c3h + + + ^ [+
3 3 ^3 3 3r rR c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )UserConfigProcessorc dS NTr@ r s r r1 z)UserConfigProcessor._message_is_relatable s tr c
K dS r r@ r* s r rL z UserConfigProcessor.is_enabled s tr c t K | d p| d }t | sd S t j t
j | } t | d S # t $ r Y d S t $ r'}t d|| Y d }~d S d }~ww xY w)Nrv r z'Failed to remove user_config dir %s: %s) ri r
ospathjoinr USER_CONFDIRr FileNotFoundErrorOSErrorra rm )r r rv targetes r r6 z,UserConfigProcessor._process_account_removed s {{6""=gkk*&=&="4(( F d /66 6NNNNN DD NN961
s )A: :
B7 B7 B22B7c K |j d }|j }|r t | rt | sd S t j t j t j | t j t j | d S # t $ r Y d S t $ r(}t
d||| Y d }~d S d }~ww xY w)N old_usernamez)Failed to rename user_config %s -> %s: %s)rV ri r r
r renamer r r r r r ra rm )r r r new_usernamer s r r3 z#UserConfigProcessor._process_modify s |''77 ' #L11 $L11
F
I T . == T . ==
! DD NN;
s A%B. .
C,; C, C''C,c
K dS rC r@ r s r r4 z#UserConfigProcessor._process_create rA r c
K dS rE r@ r s r r5 z+UserConfigProcessor._process_change_package rA r N) rM rN rO r1 rL r6 r3 r4 r5 r@ r r r r sn 0 " " " " "r r ) r' loggingr abcr r collectionsr heapqr r typingr defence360agent.contracts.configr "defence360agent.contracts.messagesr
!defence360agent.contracts.pluginsr r defence360agent.utilsr
r getLoggerra r rR r r@ r r r sq # # # # # # # # # # # # # # # # # # # # # # 1 1 1 1 1 1 : : : : : : J J J J J J J J = = = = = = = = ;3 ;3 ;3 ;3 ;3-s ;3 ;3 ;3|W3 W3 W3 W3 W3+S W3 W3 W3t1" 1" 1" 1" 1", 1" 1" 1" 1" 1"r