| Current File : //opt/imunify360/venv/lib/python3.11/site-packages/im360/plugins/__pycache__/lfd.cpython-311.pyc |
�
�*(i�
� � � d dl Z d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ e j e� � Z G d� d e
e � � ZdS )
� N)�utils)�config)� MessageType)� MessageSink�
MessageSource�expect)�strategy)�csfc � � e Zd Zej j Zej j Z
dZ e j
� ej j dd� � Ze j
� ej j dd� � Zed� � � Zd� Zd� Zd� Z eej � � d � � � Z eej � � d
� � � Z e j � � d � � � Z!d � Z"d
� Z#dS )�LFDF�scriptsz lfd_block.py�block_report_userc �
� t j d� � � � � }t j � |� � t j � | j � � k S # t t f$ r Y dS w xY w)N� BLOCK_REPORTF) r
�Config�get�os�path�realpath�BLOCK_REPORT_SCRIPT�FileNotFoundError�NotADirectoryError)�self�current_scripts �F/opt/imunify360/venv/lib/python3.11/site-packages/im360/plugins/lfd.py�_script_installedzLFD._script_installed s� � � � �Z�� 7� 7� ;� ;� =� =�N� �7�#�#�N�3�3�r�w�7G�7G��(�8� 8� �
�� "�#5�6� � � ��5�5� ���s �&A- �-B�Bc � � K � || _ d S �N)�_loop)r �loops r � create_sinkzLFD.create_sink* � � � � ���
�
�
� c � � K � || _ d S r )�_sink)r r �sinks r �
create_sourcezLFD.create_source- r" r# c � �
K � d S r � )r s r �shutdownz LFD.shutdown0 s
� � � �
�r# c � �v K � | � |j t j j | j � � � d {V �� d S r )�
_switch_stater r �CSFIntegration�ENABLEDr )r �messages r �on_strategy_changezLFD.on_strategy_change5 s[ � � � �� � � � � � !� )� � "�
�
�
�
�
�
�
�
�
�
�
r# c � � K � | � t j j t j j | j � � � d {V �� d S r )r, r �Strategy�currentr r- r. r )r �_s r �on_config_changezLFD.on_config_change= s^ � � � �� � � � � %� � !� )� � "�
�
�
�
�
�
�
�
�
�
�
r# c � � K � |o
|| j k }|r |s | � � � � d {V �� d S d S |r | � � � � d {V �� d S d S r )�STRATEGY�_revert_script�
_setup_script)r �current_strategy�
is_enabled�is_script_installed�should_script_be_installeds r r, zLFD._switch_stateE s� � � � � &0� &
� ��
� -� #� � '�-�
,��)�)�+�+�+�+�+�+�+�+�+�+�+�
,�
,�
'� '��$�$�&�&� &� &� &� &� &� &� &� &� &� '� 'r# c � � K � t � d� � t j � t j d� � � | j � � � � }|rF|| j k r;t � d| j |� � t j
|| j � � t j � � � d {V �� d S )Nz+Setup lfd BLOCK_REPORT to Imunify360 scriptr zCreating a symlink %s to %s) �logger�infor r r r
r �setr �USER_SCRIPT_LINK�symlink� lfd_restart)r �
old_scripts r r9 zLFD._setup_scriptR s� � � � �� � �A�B�B�B��W�%�%� �J�~� &� &� *� *�4�+C� D� D�
�
�
� � :�:��)A�A�A� �K�K�-��%��
�
�
�
�J�z�4�#8� 9� 9� 9��o� � � � � � � � � � � r# c � � K � t � d� � t j � | j � � r3t j | j � � }t j | j � � nd} t j d� � �
|� � t j � � � d {V ��rt j � � � d {V �� d S d S # t $ r&}t � d|� � Y d }~d S d }~ww xY w)NzReverting lfd BLOCK_REPORT� r z'Failed to revert BLOCK_REPORT value: %s)r? r@ r r �islinkrB �readlink�remover
r rA �
is_runningrD r �warning)r � user_script�es r r8 zLFD._revert_scripta s" � � � �� � �0�1�1�1�
�7�>�>�$�/� 0� 0� � �+�d�&;�<�<�K� �I�d�+� ,� ,� ,� ,� �K� (� �J�~� &� &� *� *�;� 7� 7� 7� �^�%�%�%�%�%�%�%�%�
(��o�'�'�'�'�'�'�'�'�'�'�'�
(�
(�� !� I� I� I� �N�N�D�a� H� H� H� H� H� H� H� H� H����� I���s �7'C �
D� C?�?DN)$�__name__�
__module__� __qualname__r �ProcessingOrderr �PROCESSING_ORDERr r2 �CSF_COOP_STRATEGYr7 �AVAILABLE_ON_FREEMIUMr r �join�
common_config� Packaging�DATADIRr rB �propertyr r! r' r* r r �StrategyChanger0 � ConfigUpdater5 r �log_error_and_ignorer, r9 r8 r) r# r r r sY � � � � � �"�2�6��� �2�H� !�� �'�,�,�� �'��N�� �� �w�|�|�� �'��4G�� �� �� � �X��� � �� � �
�
�
�
�V�K� &�'�'�
�
� (�'�
� �V�K� $�%�%�
�
� &�%�
� �U� �!�!�
'�
'� "�!�
'�
�
�
� (� (� (� (� (r# r )�loggingr �defence360agentr �defence360agent.contractsr rW �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr r r �im360.contracts�im360.internalsr � im360.subsysr
� getLoggerrO r? r r) r# r �<module>rg s �� ���� � � � � !� !� !� !� !� !� =� =� =� =� =� =� :� :� :� :� :� :�� � � � � � � � � �
#� "� "� "� "� "� $� $� $� $� $� $� � � � � � � �� �8� $� $��\(� \(� \(� \(� \(�+�}� \(� \(� \(� \(� \(r#