27Oۈ d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z m
Z
d dl m Z d dl
mZmZ d dlmZ d dlmZmZ ej Z G d d
ee Z G d d ee Z G d
de ZdS ) N)ABCabstractmethod) defaultdict)heappopheappush)Dict)
ConfigFileCore) MessageType)BaseMessageProcessorexpectc 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_messages1 sc s8}}%% = =A&&wx'8'8';<< < < < < < < < < = =r c
K dS )z Modify hookN r s r r2 z"EventProcessorBase._process_modify5
r c
K dS z Create hookNr? r s r r3 z"EventProcessorBase._process_create9 r@ r c
K dS zchange_package hookNr? r s r r4 z*EventProcessorBase._process_change_package= r@ r c
K dS )z Remove hookNr? r s r r5 z+EventProcessorBase._process_account_removedA r@ r c dS z'Whether the message should be processedNr? r s r r0 z(EventProcessorBase._message_is_relatableE r c
K dS z$Whether messages should be processedNr? r) s r
is_enabledz EventProcessorBase.is_enabledI r@ r N)__name__
__module__ __qualname__r r r* r
r cPanelEventr6 r" r r2 r3 r4 r5 r0 rK 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 r2 z"SettingsChangeBase._process_modifyQ sI "(GL"8"8i
++G]CCCCCCCCCCCr c D K | |dd d {V d S )NrS TrV r s r r3 z"SettingsChangeBase._process_createU s6 ++GVTBBBBBBBBBBBr c D K | |dd d {V d S )Nnew_pkgTrY r s r r4 z*SettingsChangeBase._process_change_packageX s6 ++GYEEEEEEEEEEEr c
K d S r r? r s r r5 z+SettingsChangeBase._process_account_removed[ s r FrW 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 rW r] settingss r rV 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 rS r1 r, c 3 K | ]}|d u V d S r r? )r# values r r% z5SettingsChangeBase._apply_settings..s s& AAeETMAAAAAAr z'No information about package in messagez,Settings specified in hook message %s for %sr ) r` ra getallr( rU _get_package_settingsKeyErrorwarning_default_settingsitemson_settings_change) r r rW r] rd package_namefallback_settingsfeaturerg s r rc z"SettingsChangeBase._apply_settingsj 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 rG r? r s r r0 z(SettingsChangeBase._message_is_relatable rH r c
K dS )z9What to do after settings were changed (e.g. sync the DB)Nr? )r userrr rg s r ro z%SettingsChangeBase.on_settings_change r@ r c dS )z Get default package settingsNr? r? r r rm z$SettingsChangeBase._default_settings rH r c
K dS )z"Retrieve settings from the messageNr? r s r rb z-SettingsChangeBase._get_settings_from_message r@ r rp c
K dS )z Get current package settingsNr? )clsrp r] s r rj z(SettingsChangeBase._get_package_settings r@ r c
K dS rJ r? r) s r rK z SettingsChangeBase.is_enabled r@ r )F)rL rM rN __doc__r2 r3 r4 r5 strboolrV rc r r0 ro staticmethodr rm rb classmethodrj rK r? r r rQ rQ N 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 rQ c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )UserConfigProcessorc dS NTr? r s r r0 z)UserConfigProcessor._message_is_relatable s tr c
K dS r r? r) s r rK z UserConfigProcessor.is_enabled s tr c $ K | d p| d }|ra t j t | j d S # t
$ r Y d S t $ r t d| Y d S w xY wd S )Nru r z0Cannot delete Imunify360 config file for user %s) rh osunlinkr pathFileNotFoundErrorOSErrorr` rl )r r ru s r r5 z,UserConfigProcessor._process_account_removed s {{6""=gkk*&=&=
*T**/00000$
F
s &A
B
%$B
B
c K |j d x}rlt j t j t j | t j t j |j d S d S )N old_username) rU rh r renamer joinr
USER_CONFDIRr )r r r s r r2 z#UserConfigProcessor._process_modify sz "<++N;; ;< I T . == T .0@AA
r c
K dS rB r? r s r r3 z#UserConfigProcessor._process_create r@ r c
K dS rD r? r s r r4 z+UserConfigProcessor._process_change_package r@ r N) rL rM rN r0 rK r5 r2 r3 r4 r? r r r r sn
" " " " "r r )r& loggingr abcr r collectionsr heapqr r typingr defence360agent.contracts.configr r
"defence360agent.contracts.messagesr !defence360agent.contracts.pluginsr r
getLoggerr` r rQ r r? r r r s_ # # # # # # # # # # # # # # # # # # # # # # = = = = = = = = : : : : : : 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 W3t " " " " ", " " " " "r