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