Edit File: restore.cpython-311.pyc
� ���h� � �� � d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m Z ddl m Z ddl mZmZ ddlmZ dd lmZ dd lmZ dd lmZ ej e� � Z G d � d � � Zde de fd�Z de fd�Z dS )u This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>. Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see <https://www.imunify360.com/legal/eula> � N)�Malware�MalwareScanIntensity)� MessageType)�TheSink)�resource_limits)�MalwareScanResourceType� RESTORE_ORIGINAL_INTENSITY_KEY)� MalwareHit)�MDS_PATH)�MDSDetachedRestoreDir)� hash_pathc �, � e Zd Zdd�d�Zdefd�Zd� ZdS )�MalwareDatabaseRestoreN)� restore_idc �` � || _ || _ |pt j � � j | _ d S )N)�app_name�path�uuid�uuid4�hexr )�selfr r r s �U/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/scan/mds/restore.py�__init__z MalwareDatabaseRestore.__init__+ s) � � �� ��� �$�8�� � � �(8����� �work_dirc �� � dt dt | j | j � � d| j d| j dt |j � � d| j dt |j � � dt t j � � gS ) Nz/opt/ai-bolit/wrapperz --restorez--pathz --app-namez --report-filez --detachedz --progressz --db-timeout) r � storage_pathr r �str� report_filer � progress_filer �MDS_DB_TIMEOUT)r r s r �cmdzMalwareDatabaseRestore.cmd0 sq � � #� � � ���D�M� 2� 2� � �I� � �M� � ��$� %� %� � �O� � ��&� '� '� � ��&� '� '�! � r c � � K � t | j � � 5 }| � |� � }|j � d�� � 5 }|j � d�� � 5 }t � d|� � t j |t j t j d||t |� � t �� � � d {V �� d d d � � n # 1 swxY w Y d d d � � n # 1 swxY w Y d d d � � d S # 1 swxY w Y d S )N�w)�modez Running MDS Restore with: %sT)� intensity_cpu� intensity_io�start_new_session�stdout�stderr�cwd�key)r r r" �log_file�open�err_file�logger�infor �create_subprocessr �CPU�IOr r )r r r" �l_f�e_fs r �restorez MalwareDatabaseRestore.restoreD s� � � � � "�4�?� 3� 3� �x��(�(�8�$�$�C��"�'�'�� (� � � ��h�'�,�,�#�,�6�6� �:=�� � � :�C�@�@�@�%�7��"6�":�!5�!8�&*� � ��H� � �6� � � � � � � � � � � � � � � � � � � � � ���� � � � � � � � � � � � � � � ���� � � � � � � � � � � � � � � � ���� � � � � � sZ �1D� C+�$A$C� C+�C �C+�C � C+� D�+C/ �/D�2C/ �3D�D� D)�__name__� __module__� __qualname__r r r" r7 � r r r r * s[ � � � � � �59� 9� 9� 9� 9� 9� � 1� � � � �(� � � � r r r r c �D � dt | |z t j �� � z S )Nz /var/imunify360/cleanup_storage/)� hash_provider)r �hashlib�sha384�r r s r r r X s+ � � -� � �x��w�~�1� 1� 1� � r �sinkc � � K � d� | D � � }|D ];}|� t j |j |j �� � � � � d {V �� �<d S )Nc �F � g | ] }|j t j j k � |�� S r; )� resource_typer �DB�value)�.0�hits r � <listcomp>z restore_hits.<locals>.<listcomp>_ s7 � � � � � � � � 7� :� @� @� @� � @� @� @r r@ )�process_messager �MalwareDatabaseRestoreTask� orig_filer )�hitsrA �db_hitsrH s r � restore_hitsrO ^ s� � � � �� ��� � �G� � � ���"�"� � 2��]�S�\� � � � � � � � � � � � � � � r ) �__doc__r> �loggingr � defence360agent.contracts.configr r �"defence360agent.contracts.messagesr �"defence360agent.internals.the_sinkr �defence360agent.utilsr �imav.malwarelib.configr r �imav.malwarelib.modelr �imav.malwarelib.scan.mdsr �!imav.malwarelib.scan.mds.detachedr �imav.malwarelib.utilsr � getLoggerr8 r0 r r r rO r; r r �<module>r\ sn ��� �* ���� ���� � � � � J� J� J� J� J� J� J� J� :� :� :� :� :� :� 6� 6� 6� 6� 6� 6� 1� 1� 1� 1� 1� 1�� � � � � � � � -� ,� ,� ,� ,� ,� -� -� -� -� -� -� C� C� C� C� C� C� +� +� +� +� +� +� �� �8� $� $��+� +� +� +� +� +� +� +�\�s� �c� � � � � �7� � � � � � r