File: //opt/imunify360/venv/lib/python3.11/site-packages/im360/__pycache__/aibolit_job.cpython-311.pyc
�
�؞i| � � � d Z ddlZddlZddlZddlZddlZddlZddlmZm Z ddl
m Z ddl m
Z
g d�ZdZ e d� � Z e d � � Z e d
� � Zd Zd Zd
Zdej d�d�Zedd�d�Zedd�d�Zdd�d�Zefd�Zde de fd�Z dS )zHelper functions to kick off the aibolit scan on file upload.
See {modsec,pureftpd}-on-upload scripts for the example usage.
� N)� ExitStack�suppress)�Path)�NamedTemporaryFile)� RESIDENT_DIR� RESIDENT_IN_DIR_NOTIFY_REL_PATH� RESIDENT_IN_DIR_UPLOAD_REL_PATH�UPLOAD_TIMEOUT�create_notify_job�create_remaining_time_func�create_upload_job�$notify_aibolit_start_it_if_necessaryz./var/run/defence360agent/aibolit-resident.sockz /var/imunify360/aibolit/residentzin/upload-jobszin/notify-jobs� �
g�������?)�
start_time�timerc �. � ��� �� � ��� �fd�}|S )z�Create remaining_time() function.
Start the timer if start_time is None otherwise use given value as
the start time.
remaining_time() raises TimeoutError in *timeout*
seconds after the *start_time* according to the *timer*.
c �B �� � �� � �z
z
} | dk rt �| S )Nr )� TimeoutError)� time_leftr �timeoutr s ����F/opt/imunify360/venv/lib/python3.11/site-packages/im360/aibolit_job.py�remaining_timez2create_remaining_time_func.<locals>.remaining_time8 s- �� ��u�u�w�w�� 3�4� � ��>�>� � ��� � )r r r r s ``` r r r * s= ���� � ����J�� � � � � � � �r )�resident_dir_pathr c � � t | |t z d� t j � � � � z |�� � dS )z-Create PID.upload_job in the resident/in dir.z
{}.upload_job��job_pathr N)�
create_jobr �format�os�getpid��filesr r s r r
r
A sR � ��
�"�
)�*�
�
�
�����
-�
-�.� � � � � � � r c � � t | |t z d� t j � � � � z |�� � dS )z.Create UUID.notify_job in the resident/in dir.z
{}.notify_jobr N)r r r! �uuid�uuid4r$ s r r r L sR � ��
�"�
)�*�
�
�
��� � �
.�
.�/� � � � � � � r )r c � � d� | D � � } t |t j d| i� � � d� � � � d S )Nc � � g | ];}t j t j |� � � � � d � � ��<S )�ascii)�base64� b64encoder" �fsencode�decode)�.0�fns r �
<listcomp>z create_job.<locals>.<listcomp>Z s9 � � O� O� O�2�V�
�b�k�"�o�o�
.�
.�
5�
5�g�
>�
>� O� O� Or r% r+ )� write_atomic�json�dumps�encode)r% r r s r r r W sW � �
P� O�� O� O� O�E��� �
�G�U�#�$�$�+�+�G�4�4�� � � � r c � � t j t j t j � � 5 }|� | � � |� t
� � |� d� � ddd� � dS # 1 swxY w Y dS )zZNotify aibolit that there is a file to scan.
Start/restart aibolit if necessary.
� 1N)�socket�AF_UNIX�
SOCK_DGRAM�
settimeout�connect�AIBOLIT_STARTUP_SOCKET�send)r �socks r r r a s� � �
� �v�~�v�'8� 9� 9� �T� ���� � � � � � �+�,�,�,� � � �$����� � � � � � � � � � � ���� � � � � � s �AA<�<B �B �path�contentc � �� t � � 5 }t t | j � � d�� � 5 ��fd�}|� |� � �� |� � ddd� � n # 1 swxY w Y t
�j � � � | � � |� � � ddd� � dS # 1 swxY w Y dS )z�Write *content* to *path* atomically.
Ignore fsync() issues:
https://stackoverflow.com/questions/12003805/threadsafe-and-fault-tolerant-file-writes
F)�dir�deletec � �� t t � � 5 t j � j � � d d d � � d S # 1 swxY w Y d S )N)r �FileNotFoundErrorr" �remove�name)� output_files �r �cleanupz write_atomic.<locals>.cleanup| s� �� � � /�0�0� 0� 0��I�k� .�/�/�/�0� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0���� 0� 0� 0� 0� 0� 0s �=�A�AN)
r r �str�parent�callback�writer rI �replace�pop_all)rA rB �stackrK rJ s @r r3 r3 q sV �� �
��� ��
��D�K� � ��
�
�
�
'�
�
0�
0�
0�
0�
0�
�N�N�7� #� #� #� � � �g� &� &� &�
'�
'�
'�
'�
'�
'�
'�
'�
'�
'�
'����
'�
'�
'�
'�
�[�
� � �&�&�t�,�,�,�
�
�
���� � � � � � � � � � � � ���� � � � � � s4 �$C�0A0�$ C�0A4 �4C�7A4 �8>C�C�
C) �__doc__r, r4 r" r9 �timer' �
contextlibr r �pathlibr �tempfiler �__all__r> r r r �
STUCK_TIMEOUTr
�STARTUP_SOCKET_TIMEOUT� monotonicr r
r r r �bytesr3 r r r �<module>r] s� ��� �
�
�
�
� � � � � � � � �
�
�
�
� � � � � � � � � *� *� *� *� *� *� *� *� � � � � � � '� '� '� '� '� '�
� � �� J� ��t�6�7�7� �"&�$�'7�"8�"8� �"&�$�'7�"8�"8� ��
��� � � �t�~�� � � � �. 3?�� � � � � � 3?�� � � � � � ,0� � � � � � 2H�
�
�
�
� �t� �e� � � � � � r