Edit File: lfd.cpython-311.pyc
� P 6h� � � � 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#