Zڑ7 x d dl Z d dlZd dlmZ d dlmZmZ ej e Z G d dej
Z dS ) N)utils)messagespluginsc e Zd ZdZdZ ej ej j d Z
fdZ fdZ d Z
d Zd Zd Zd
Z xZS ) IDSAwareMessageSinkz^A message sink that accepts messages taking into account the
current IDS (strategy).
Nc K | j |j | j k r| s | d{V t d| dS # t j $ r t $ r2 t d| t j Y dS w xY wdS | r | d{V t d| dS # t t j f$ r2 t d| t j Y dS w xY wdS )z7De/activate the plugin on strategy change if necessary.NzPlugin %r is activatedz Failed to activate %r pluginPlugin %r is deactivated Failed to deactivate %r plugin)
STRATEGYstrategy is_activeactivateloggerinfoasyncioCancelledError Exception exceptionr fail_agent_service
deactivate)selfmessages L/opt/imunify360/venv/lib/python3.11/site-packages/im360/contracts/plugins.pyon_strategy_changez&IDSAwareMessageSink.on_strategy_change s = G$4
$E$E>>##
@@ --//))))))) KK 8$?????
- / / /$$%CTJJJ,....../
@
@ ^^
>
>oo'''''''''
6===== w 56
+
+
+ !A4HHH(******
+ > >s% A& &AB10B1 D AE
E
c K | s t |t j j r't | d {V S d S )N)r
isinstancer MessageTypeStrategyChangesuperprocess_message)r r __class__s r r z#IDSAwareMessageSink.process_message- sk >> :z X)8
: 0099999999 9 : : c X K | rn | d {V t d| n7# t t
j f$ r t d| Y nw xY wt d {V S )Nr r
)
r
r r r r r r r r shutdown)r r! s r r$ z IDSAwareMessageSink.shutdown3 s >> >
>oo'''''''''
6==== w 56
I
I
I !A4HHHHH
I
WW%%'''''''''s A 1BBc 2 K | dS )zrThis method will be called for plugins with specified strategy
when this strategy becomes active.
N)_mark_as_activer s r r z IDSAwareMessageSink.activate> s
r" c 2 K | dS )ztThis method will be called for plugins with specified strategy
when this strategy becomes inactive.
N)_mark_as_inactiver' s r r z IDSAwareMessageSink.deactivateD s
r" c $ t | dd S )z Indicates if plugin is active.
_is_activeF)getattrr' s r r
z IDSAwareMessageSink.is_activeJ s t\5111r" c d| _ d S )NTr+ r' s r r& z#IDSAwareMessageSink._mark_as_activeN s
r" c d| _ d S )NFr. r' s r r) z%IDSAwareMessageSink._mark_as_inactiveQ s
r" )__name__
__module__ __qualname____doc__r r expectr r r r r r$ r r r
r& r)
__classcell__)r! s @r r r
s
HW^H(788> > 98>4: : : : : ( ( ( ( ( ! ! ! 2 2 2 r" r ) r loggingdefence360agentr defence360agent.contractsr r getLoggerr0 r MessageSinkr r" r r< s ! ! ! ! ! ! 7 7 7 7 7 7 7 7 8 $ $H H H H H '- H H H H H r"