p6 =H d dl Z d dlmZ d dlmZ d dlmZ d dlm Z m
Z
G d de Z G d d Z
G d
d Z G d d
Z e Z G d de Z G d de Z G d de Z G d de Z G d dee Z G d de Z G d d Z G d d e Z G d d e Z G d d!e Z G d" d#ee Z G d$ d%ee Z G d& d'e Z G d( d)ee Z G d* d+ee Z G d, d-e Z! G d. d/ee Z" G d0 d1e Z# G d2 d3e Z$ G d4 d5e Z% G d6 d7ee Z& G d8 d9ee Z' G d: d;ee Z( G d< d=e Z) G d> d?e Z* G d@ dAee Z+dBe,dCe-dDe,fdEZ. G dF dGee Z/ G dH dIeee Z0 G dJ dKe Z1 G dL dM Z2 G dN dOe Z3 G dP dQee Z4 G dR dSe Z5 G dT dUee Z6 G dV dWee Z7 G dX dYee Z8dS )Z N)Enum)List)Core)batched batched_dictc e Zd ZdS )MessageNotFoundErrorN__name__
__module__ __qualname__ W/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/contracts/messages.pyr r Dr r c e Zd ZdZd Zd ZdS )UnknownMessagez&
Used as stub for MessageType
c t d )NzMessage class is not found.)r selfs r __init__zUnknownMessage.__init__ s "#@AAAr c dS )NUnknownr )r names r __getattr__zUnknownMessage.__getattr__ s yr N)r r r
__doc__r r r r r r r
s? B B B r r c 8 e Zd Zg Z fdZed Z xZS )MessageTc n t j di | | j | d S )Nr )super__init_subclass__ _subclassesappend)clskwargs __class__s r r! zMessageT.__init_subclass__ s< !!++F+++ s#####r c * t | j S N)tupler" r$ s r get_subclasseszMessageT.get_subclasses s S_%%%r )r r r
r" r! classmethodr+
__classcell__r& s @r r r sX K$ $ $ $ $ & & [& & & & &r r c e Zd ZdZd ZdS ) _MessageTypea
Used to get specific message class. For example,
>>> _MessageType().ConfigUpdate
>>> _MessageType().NotExistMessage
>>>
c f t D ]}|j |k r|c S t S r( )Messager+ r r )r r subclss r r z_MessageType.__getattr__/ s> ,,.. F$&&
' r N)r r r
r r r r r r0 r0 % s- r r0 c e Zd ZdZdZdS ) ReportTargetapiconnN)r r r
APIPERSISTENT_CONNECTIONr r r r5 r5 : s
C "r r5 c < e Zd ZdZej Zedefd Z dS )
ReportablezD
Mixin class for messages that should be sent to the server
methodc d | D ]}|t |d k r|c S dS )ao
Return a subclass with the same DEFAULT_METHOD as *method*.
It can be used to detect report target from message method.
NOTE: it is not guaranteed that the class with the *method* is unique,
in this case the first subclass found is returned, but
it is tested that all such subclasses have the same TARGET.
DEFAULT_METHODN)__subclasses__getattr)r$ r< subclasss r get_subclass_with_methodz#Reportable.get_subclass_with_methodF sH **,, H+;<<<< =tr N)
r r r
r r5 r9 TARGETr, strrB r r r r; r; ? sM
/F c [ r r; c . e Zd ZdZedefd ZdS )Receivedz
Mixin class for messages received from the server.
These messages are created in the client360 plugin when receiving a
request from imunify360.cloudlinux.com.
actionc | D ],}t |dg pt |d g}||v r|c S -t d | )NRECEIVED_ACTIONSr> z*Message class is not found for "{}" action)r? r@ r format)r$ rG rA received_actionss r get_subclass_with_actionz!Received.get_subclass_with_action] s **,, H &x1CR H H "233M ))) *" 8 ? ? G G
r N)r r r
r r, rD rL r r r rF rF U sE
c
[
r rF c ^ e Zd ZdZedd Zedefd Zedd ZdS )LockableNreturnc K | j t j | _ | j d {V d S r( )_lockasyncioLockacquirer* s r rT zLockable.acquirem sG 9 CIi !!!!!!!!!!!r c F | j d uo| j S r( )rQ lockedr* s r rV zLockable.lockeds s" y$;)9)9););;r c J | j | j d S d S r( )rQ releaser* s r rX zLockable.releasew s, 9 I ! r rO N) r r r
rQ r, rT boolrV rX r r r rN rN j s~ E" " " ["
r r2 r )r argsr% r& s r r zMessage.__init__ sB 1 ! 0DN%gt %t6v66666r c > d | D S )Nc & i | ]\ }}|d k ||S )r< r ).0kvs r
z#Message.payload.. s# ???Ah 1 r itemsr s r payloadzMessage.payload s ?? ????r c X | | S # t $ r}t | |d}~ww xY w)z
Called when an attribute lookup has not found the attribute
in the usual places
A shortcut to access an item from dict
N)KeyErrorAttributeError)r r excs r r zMessage.__getattr__ s@ 0: 0 0 0 &&C / 0s
)$)c fd D } }| | d j j | S )a.
Do not flood console.log with large sequences If there is a list
of more than _FOLD_LIST_THRESHOLD items inside the message,
then this list will be collapsed and the number of items in
the list will be shown. The message itself will not be
collapsed.
c i | ]k\ }}t |d rVt | j k -|t |t rt |j ndt | dlS )__len__)hasattrlen_FOLD_LIST_THRESHOLD
isinstancerD _shorten_str_SHORTEN_STR_THRESHOLD)rd re colxnr s r rg z$Message.__repr__.. s
5ui((
E
T 666
%%%+|E4#>???*SZZ*** 766r {}({}))ri copyupdaterJ r& r
)r
items_to_fold
folded_msgs ` r __repr__zMessage.__repr__ sq
!JJLL
YY[[
-(((t~ :JGGGr c * | S r( )r r s r __str__zMessage.__str__ s }} r rY )r r r
r r> PRIORITYPROCESSING_TIME_THRESHOLDru rx r propertyrj r r r r- r. s @r r2 r2 } s NH " 7 7 7 7 7 7
@ @ X@
0
0
0H H H, r r2 c 4 e Zd Z fdZed Z xZS ) MessageListc L t | d S )Nlist)r r )r msg_listr& s r r zMessageList.__init__ s$
h'''''r c | j S r( r r s r rj zMessageList.payload s
yr )r r r
r r rj r- r. s @r r r sS ( ( ( ( ( X r r c e Zd ZdZdefdZdS )ShortenReprListMixinz
Do not flood console.log with large sequences
The method collapses messages that are a list.
Instead of showing all the elements of the message,
their number will be displayed.
r c d | j j d t | dg S )Nrz z <{} item(s)>ri )rJ r& r
rt getr s r r z ShortenReprListMixin.__repr__ sG N ' ! !#dhhw&;&;"<"< = =
r N)r r r
r dictr r r r r r s9
t
r r c " e Zd ZdZeZdefdZdS )
AccumulatablezMessages of this class will be grouped into a list of LIST_CLASS
message instance by Accumulate plugin. Messages whose do_accumulate()
call returns False will not be added to list.rO c dS )zAReturn True if this message is worth collecting, False otherwise.Tr r s r
do_accumulatezAccumulatable.do_accumulate s tr N)r r r
r r
LIST_CLASSrZ r r r r r r s@ 5 5 Jt r r c e Zd ZdS )ServerConnectedNr
r r r r r r r r c e Zd ZdS )ServerReconnectedNr
r r r r r r r r c * e Zd ZdZdZdZ fdZ xZS )Pingzr
Will send this message on connected, reconnected events
to provide central server with agent version
PINGr c f t t j | d< d S )Nversion)r r
CoreConfigVERSION)r r& s r r z
Ping.__init__ s)
$,Yr )r r r
r r> r r r- r. s @r r r sN
NH- - - - - - - - -r r c & e Zd ZdZdZ fdZ xZS )Ackzd
Notify Server that a persistent message with *seq_number* has been
received by Agent.
ACKc ` t j di | t | | d< d S )N)per_seq_metar )r r r )r
seq_numberr% r& s r r z Ack.__init__ s8 ""6""" Z000W
r )r r r
r r> r r- r. s @r r r sI N1 1 1 1 1 1 1 1 1r r c e Zd ZdZdZdS )NoopzG
Sending NOOP to the agent to track the message in agent logs.
NOOPNr r r
r r> r r r r r s NNNr r c * e Zd ZdZdZej Zd ZdS ) ServerConfigz.
Information about server environment
SERVER_CONFIGc @ d | j j S Nz{}()rJ r& r
r s r r zServerConfig.__repr__ }}T^ 8999r N r r r
r r> r5 r8 rC r r r r r r s< %N
F: : : : :r r c * e Zd ZdZdZej Zd ZdS )
DomainListz*
Information about server domains
DOMAIN_LISTc @ d | j j S r r r s r r zDomainList.__repr__ r r Nr r r r r r s< #N
F: : : : :r r c e Zd ZdZd Zd ZdS ) FilesUpdatedz/
To consume products of files.update()
c || d< || d< dS )z\
:param files_type: files.Type
:param files_index: files.LocalIndex
files_type files_indexNr )r r r s r r zFilesUpdated.__init__) s (\ )]r c \ d | j j | d | d S )z?
Do not flood console.log with large sequences
z+{}({{'files_type':'{}', 'files_index':{}}})r r r r s r r zFilesUpdated.__repr__2 s4 =CC N '
r N)r r r
r r r r r r r r $ s< * * *
r r c e Zd ZdZdZdS ) UpdateFilesz9
Update files by getting message from the server
UPDATENr r r r r r = s NNNr r c e Zd ZdZdS ) ConfigUpdate
CONFIG_UPDATENr r r
r> r r r r r E $NNNr r c e Zd ZdZdS )Rejectz
Kinda message filtering facility.
Raised in order to stop message processing through plugins.
Takes reason of reject as argument.
N)r r r
r r r r r r I s Dr r c e Zd ZdZdS )HealthHEALTHNr r r r r r S s NNNr r c e Zd ZdZdS )
CommandInvokeCOMMAND_INVOKENr r r r r r W %NNNr r c e Zd ZdZdS )
ScanFailed SCAN_FAILEDNr r r r r r [ s "NNNr r c e Zd ZdZdS )
CleanupFailedCLEANUP_FAILEDNr r r r r r _ r r r c e Zd ZdZdZdS )RestoreFromBackupTaskz5
Creates a task to restore files from backup
MALWARE_RESTORE_FROM_BACKUPNr r r r r r c s 3NNNr r c B e Zd ZdZh dZededededefd Z dS ) cPanelEvent PANEL_EVENT> planexcludenew_pkg
imunify360_avimunify360_proactiveusernamehooktsfieldsc fd| D }|dk r%d|v r!d|v r |d |d k r |d |d< ||||d S )Nc v i | ]5\ }}| j v | |6S r )lowerALLOWED_FIELDS)rd re rf r$ s r rg z/cPanelEvent.from_hook_event..y sG
1wwyyC ...
GGIIq...r Modifyusernewuser old_username)r r data timestamprh )r$ r r r r r s ` r from_hook_eventzcPanelEvent.from_hook_eventu s
H & V##v&"333#)&>D s $
r N)
r r r
r> r r, rD floatr r r r r r r k sj "N N
"%
+0
:>
[
r r c e Zd ZdZdS ) IContactSent
ICONTACT_SENTNr r r r r r r r r slimitrO c |dk sJ t | |k r$| d|dz dz
d| | dz dz d n| S )z1Shorten *s* string if its length exceeds *limit*. N z...)rt )r r s r rw rw sf 19999 q66E>>
uzA~
991eVq[1_%6%6#7999
r c e Zd ZdZdZdS )
BackupInfoz(Information about enabled backup backend BACKUP_INFONr r r r r r s 22"NNNr r c e Zd ZdZdS )
MDSReportList
MDS_SCAN_LISTNr r r r r r r r r c e Zd ZeZdS ) MDSReportN)r r r
r r r r r r r s JJJr r c n e Zd ZdZdZdZdZedee fd Z
edee fd Z dS )
Splittablez
A message list could be split into multiple batches.
The split is possible for a list itself along with internal resources.
Nmessagesc # K | j r| j r|D ]}| | j x}|V #t | }t |t
rt nt } ||| j D ]/}| }||| j < || }|V 0dS t | E d{V dS )z
Split messages' internal lists of things into batches.
A field that is meant to split is defined by `BATCH_FIELD`.
N)
BATCH_FIELD
BATCH_SIZEr typerv r r r r{ iter) r$ r messageri
message_classbatcherbatchr new_messages r _split_itemszSplittable._split_items s ? &s~ &#
*
* $[[ 9 99EB!MMMM$(MMM(25$(?(?L W ")!?!? * * &||~~05S_-&3mD&9&9 ))))) *
*
* H~~ % % % % % % % % %r c # K | j pt | }| | }t || E d {V d S r( ) LIST_SIZErt r r )r$ r list_sizesplits r r zSplittable.batched sS M2S]] **5),,,,,,,,,,,r ) r r r
r r r r r, r r r r r r r r r s
IJK&D $7 & & & [&, -tM 2 - - - [- - -r r c e Zd ZdZdZdS )EnsureServiceStatez-Ensure the service has the appropriate statusENSURE_SERVICE_STATENr r r r r r s 77+NNNr r c e Zd ZdZdZdS )SensorWordpressIncidentListzAggregated incident list
INCIDENT_LISTNr r r r r r s ""$NNNr r c e Zd ZdZdS )WordpressPluginActionWP_SECURITY_PLUGIN_ACTIONNr r r r r r s 0NNNr r c * e Zd ZdZdZej Zd ZdS )WordpressPluginTelemetryzX
Information about telemetry event related to Imunify Security WordPress plugin
WP_SECURITY_PLUGIN_EVENTc @ d | j j S r r r s r r z!WordpressPluginTelemetry.__repr__ r r Nr r r r r r s< 0N
F: : : : :r r c e Zd ZdZdZdS )WPRuleDisabledz#WordPress protection rule disabled.
RULE_DISABLEDNr r r r r! r! s --$NNNr r! c e Zd ZdZdZdS )
WPRuleEnabledz%WordPress protection rule re-enabled. RULE_ENABLEDNr r r r r$ r$ s //#NNNr r$ )9rR enumr typingr defence360agent.contracts.configr r defence360agent.utilsr r Exceptionr r r r0 MessageTyper5 r; rF rN r r2 r r r r r r r r r r r r r r r r r r r r r rD intrw r r r r r r r r r! r$ r r r r- sH ? ? ? ? ? ? 7 7 7 7 7 7 7 7 9 & & & & & & & & $ lnn # # # # #4 # # #
,
x
* x &; ; ; ; ; dH ; ; ; | '
G g
- - - - -7J - - - 1 1 1 1 1': 1 1 1 7 : : : : :7J : : : : : : : :* : : :
7
2 '8 % % % % %7 % % % Y W & & & & &GZ & & &# # # # #* # # #& & & & &GZ & & &3 3 3 3 3G 3 3 3"
"
"
"
"
' "
"
"
J% % % % %7J % % %C # # # # #* # # # % % % % %(': % % %
&- &- &- &- &- &- &- &-R, , , , , , , , % % % % %+z % % % 1 1 1 1 1G 1 1 1 : : : : :w
: : :% % % % %Wj % % % $ $ $ $ $GZ $ $ $ $ $r