관리-도구
편집 파일: event_monitor.cpython-311.pyc
� IMgL � �� � d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m Z m Z m Z d dl mZ d dlmZ d dlmZ d d lmZ d d lmZmZ d d lmZ ee� � Z G d � d ee� � Z dS )� N)�ABC)� getLogger)�Path)�Dict�List�Optional)�Core)� MessageType)� MessageSource)�%NativeFeatureManagementSettingsChange)�EventProcessorBase�UserConfigProcessor)�recurring_checkc � � e Zd Zej ZdZd� Zd� Zd� Z e de fd�� � Z e de de fd�� � Zdeej fd �Z ed � � d � � � Zd S ) � EventMonitorz *.*.*.*.jsonc �> � d | _ d | _ g | _ d | _ d S �N)�_loop�_sink� _processors�_processing_task��selfs �Z/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/event_monitor.py�__init__zEventMonitor.__init__ s& � ��� ��� �57��� $�� � � � c � �$ K � || _ || _ | j � t |� � � � | j � t |� � � � | j � | � � � � � | _ d S r ) r r r �appendr r � create_task�#_check_inbox_folder_generate_eventsr )r �loop�sinks r � create_sourcezEventMonitor.create_source" s� � � � ��� ��� � �� � � E�d� K� K�L�L�L� �� � � 3�D� 9� 9�:�:�:� $� � 6� 6� � 4� 4� 6� 6�! � ! �� � � r c � �� K � | j � � � t j t j � � 5 | j � d {V �� d d d � � d S # 1 swxY w Y d S r )r �cancel� contextlib�suppress�asyncio�CancelledErrorr s r �shutdownzEventMonitor.shutdown+ s� � � � � � �$�$�&�&�&� � ��!7� 8� 8� (� (��'� '� '� '� '� '� '� '� (� (� (� (� (� (� (� (� (� (� (� (���� (� (� (� (� (� (s �A�A� A�filec � � | � � � d S # t $ r Y d S t $ r'}t � d| |� � Y d }~d S d }~ww xY w)NzCouldn't remove file %s %s)�unlink�FileNotFoundError� Exception�logger�warning)r+ �es r �_rmfilezEventMonitor._rmfile0 s~ � � B� �K�K�M�M�M�M�M�� � � � � �D�D�� B� B� B� �N�N�7��q� A� A� A� A� A� A� A� A� A����� B���s � � A� A� A�A�returnc �N � t j | � � � � � S r )�json�loads� read_text)r+ s r � _from_jsonzEventMonitor._from_json9 s � ��z�$�.�.�*�*�+�+�+r c �� � |j � d� � ^}}}}}t |dz |z � � }n,# t $ r t � d|� � Y d S w xY w t j � |||| � |� � �� � S # t $ r t � d|� � Y n/t j $ r t � d|� � Y nw xY wd S )N�.z+hook-event-file detected with wrong name %s)�username�hook�ts�fieldszhook file disappeared %sz hook file have broken json %s) �name�split�float� ValueErrorr0 r1 r � cPanelEvent�from_hook_eventr9 r. r6 �JSONDecodeError)r r+ r< r= �ts1�ts2�_r> s r �_event_to_messagez EventMonitor._event_to_message= s$ � � �+/�9�?�?�3�+?�+?� (�H�d�C��q��s�S�y�3��'�'�B�B��� � � � �N�N�H�$� O� O� O��4�4� ���� B� �*�:�:�!������t�,�,� ;� � � �� !� =� =� =� �N�N�5�t� <� <� <� <� <��#� B� B� B� �N�N�:�D� A� A� A� A� A� B���� �ts' �47 �%A � A �$5B �%C-�)C-�,C-� c � � K � t | j � � � d� � D ]�} | � |� � }|�9| j D ]1}|� � � � d {V ��r|� |� � �2n2# t $ r%}t � d|� � Y d }~nd }~ww xY w| � |� � ��# | � |� � w xY w| j D ] }|� � � � d {V �� � d S )Nz *.*.*.jsonz Failed to process %s hook event) r � EVENT_DIR�globrJ r � is_enabled� add_messager/ r0 �errorr3 �process_messages)r r+ �message� processor�excs r r z0EventMonitor._check_inbox_folder_generate_eventsS sR � � � ��� �(�(�-�-�l�;�;� #� #�D� #� �0�0��6�6���&�%)�%5� ;� ;� �!*�!5�!5�!7�!7�7�7�7�7�7�7� ;� %� 1� 1�'� :� :� :���� � E� E� E�� � � >��D�D�D�D�D�D�D�D����� E���� � � �T�"�"�"�"��� � �T�"�"�"�"���� �)� /� /�I��,�,�.�.� .� .� .� .� .� .� .� .� /� /s0 �AA>�=C�> B-�B(�#C�(B-�-C�C N)�__name__� __module__� __qualname__r �INBOX_HOOKS_DIRrM �PATTERNr r# r* � staticmethodr r3 r r9 r r rD rJ r r � r r r r s� � � � � � ��$�I� �G�%� %� %� � � �(� (� (� �B�d� B� B� B� �\�B� �,�� ,�$� ,� ,� ,� �\�,���+�2I�)J� � � � �, �_�R��� /� /� �� /� /� /r r ) r( r& r6 �abcr �loggingr �pathlibr �typingr r r � defence360agent.contracts.configr �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr �1defence360agent.feature_management.plugins.nativer �7defence360agent.plugins.event_monitor_message_processorr r �defence360agent.utilsr rV r0 r r\ r r �<module>rg sV �� ���� � � � � � � � � � � � � � � � � � � � � � � � � � � '� '� '� '� '� '� '� '� '� '� 1� 1� 1� 1� 1� 1� :� :� :� :� :� :� ;� ;� ;� ;� ;� ;�� � � � � �� � � � � � � � 2� 1� 1� 1� 1� 1� ��8� � ��I/� I/� I/� I/� I/�=�#� I/� I/� I/� I/� I/r