관리-도구
편집 파일: notifier.cpython-311.pyc
� IMgI � � � d Z ddlZddlZddlZdZdZdZdZdZdZ d Z d Z d Z d Z d ZdZdedededefd�Zdeddfd�Zdedededdfd�Zdd�ZdS )z$Send events via Notification service� Nz /opt/imunify360/lib/event.sockg $@� i �CONFIG_UPDATED�USER_SCAN_STARTED�USER_SCAN_FINISHED�USER_SCAN_MALWARE_FOUND�CUSTOM_SCAN_STARTED�CUSTOM_SCAN_FINISHED�CUSTOM_SCAN_MALWARE_FOUND�SCRIPT_BLOCKED�event_id�user�body�returnc �� � t j | |t j t j |� � � d� � � � � d� � d�� � }|� d� � }t |� � t k r5t d� t |� � t � � � � �t |� � � t d�� � |z S )Nzutf-8)r r r z#message size {} exceeds limit of {}�big)� byteorder) �json�dumps�base64� b64encode�encode�decode�len� _MAX_SIZE� Exception�format�to_bytes� _LEN_BYTES)r r r �event�binarys �T/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/subsys/notifier.py�_prepare_eventr" s� � � �J� ���$�T�Z��%5�%5�%<�%<�W�%E�%E�F�F�M�M��� � � � � �E� �\�\�'� "� "�F� �6�{�{�Y� � �� 1� 8� 8��F� � �Y� � � � � � �v�;�;� � � �e� � <� <�v� E�E� r c � � K � t j t � � � d {V ��\ }} |� | � � |� � � � d {V �� |� � � d S # |� � � w xY w)N)�asyncio�open_unix_connection� SOCKET_PATH�write�drain�close)r �_�writers r! � _send_eventr- * s� � � � � �2�;�?�?�?�?�?�?�?�?�I�A�v��� � �U�����l�l�n�n� � � � � � � �� � �������� � �������s �/A+ �+Bc � � K � t | ||� � }t j t |� � t � � � d{V �� dS )z>Send an event with given event_id and user, having given body.N)r" r% �wait_forr- �SOCKET_TIMEOUT)r r r r s r! � trigger_eventr1 3 sJ � � � � �8�T�4� 0� 0�E� � �;�u�-�-�~� >� >�>�>�>�>�>�>�>�>�>r# c � �B K � t t di � � � d{V �� dS )zRSend CONFIG_UPDATED event. This forces imunify-notifier to reread its config.� N)r1 �CONFIG_UPDATED_EVENT_ID� r# r! �config_updatedr6 9 s3 � � � � �/��R� 8� 8�8�8�8�8�8�8�8�8�8r# )r N)�__doc__r% r r r' r0 r r r4 �USER_SCAN_STARTED_EVENT_ID�USER_SCAN_FINISHED_EVENT_ID� USER_SCAN_MALWARE_FOUND_EVENT_ID� CUSTOM_SCAN_STARTED_EVENT_ID� CUSTOM_SCAN_FINISHED_EVENT_ID�"CUSTOM_SCAN_MALWARE_FOUND_EVENT_ID�SCRIPT_BLOCKED_EVENT_ID�str�dict�bytesr" r- r1 r6 r5 r# r! �<module>rB s �� *� *� ���� � � � � � � � �.� ��� � � � �*� � 0� � 2� �#<� � 4� � 6� �%@� "�*� �F�S� F�� F�4� F�E� F� F� F� F�(�U� �t� � � � �?�#� ?�S� ?�� ?�� ?� ?� ?� ?� 9� 9� 9� 9� 9� 9r#