관리-도구
편집 파일: 160_remove_quarantine.cpython-311.pyc
� ��O��n�C� � � d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZm Z d dl m Z m Z d dl mZ d dlmZ e j e� � ZdZd Zd Zed d gZd � Zdd�Zd� Zde eef fd�Z dedeeef fd�Z dS )� N)�glob)�Path)�Tuple�Union)� CharField�Model)� FilenameField)� HostingPanelz.imunify.quarantinedz/var/imunify360� quarantinedz/var/wwwz/home*c �2 � � G � fd�dt � � }|S )zl Model stub for migration because we can't use migrator.orm[] due to custom field FilenameField c �Z �� e Zd Z G � fd�d� � Z ed�� � Z e� � ZdS )� get_model.<locals>.MalwareHitc � �� e Zd ZdZ� ZdS )�"get_model.<locals>.MalwareHit.Meta� malware_hitsN)�__name__� __module__� __qualname__�db_table�database��dbs ��e/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/160_remove_quarantine.py�Metar s �� � � � � �%�H��H�H�H� r F)�nullN)r r r r r � orig_filer �statusr s �r � MalwareHitr s` �� � � � � � � � � � � � � � � � "�M�u�-�-�-� �������r r )r )r r s ` r � get_modelr s? �� � � � � � � � �U� � � � �r Fc � � |rd S |pt }t |� � }|� � � � |j t k � � }|D ]8}t |j � � \ } } || � � |� � � �9t D ]B} t t j � | t � � � � D ] } || � � ��Cd S �N)�delete_quarantine_folderr �select�wherer � QUARANTINED� find_quarr �delete_instance�QUARANTINE_PARENTSr �os�path�join� QUAR_NAME) � _migratorr �fake�delete_function�_�__�modelr �hit�path_to_delete�parents r �migrater7 ) s� � � � ��� &�A�)A�O� �h� � �E��,�,�.�.�&�&�u�|�{�'B�C�C�K� � � �� %�c�m� 4� 4�������'�'�'� ��� � � � � %� ,� ,�� "�2�7�<�<�� �#B�#B� C� C� ,� ,�N� �O�N� +� +� +� +� ,�,� ,r c � � d S r" � )r1 r2 s r �rollbackr: ? s � ��Dr �quarantine_pathc �� � t | � � } | j t k rK| | � � � k r5t � d| � � t j | d�� � d S d S d S )Nz Deleting quarantine folder %sT)� ignore_errors)r �namer- �resolve�logger�info�shutil�rmtree)r; s r r# r# C sr � ��?�+�+�O�� � �)�)� �� 6� 6� 8� 8� 8� 8�� � �3�_�E�E�E�� �o�T�:�:�:�:�:�:� *�)� 8� 8r �source�returnc � � t | � � }t t � � t z |� t d� � � � f}d}d}|j D ]P} t j |� � � j � � }|} n # t $ r Y �>t $ r |cY c S w xY w|� |j dk r|S |� � � |� |� � z } t � � � |j � � }n# t t f$ r |cY S w xY w |� |� � }n# t" $ r |cY S w xY w|t z |fS )zy Find file in quarantine by source path. This function is copied from agent code since it is to be removed. �/N�root)r �DEF_QUARr- � relative_to�parents�pwd�getpwuid�stat�st_uid�FileNotFoundError�KeyError�pw_namer? r � base_home_dir�pw_dir� RuntimeError� ValueError) rD �file�default_result�userr6 r+ �resolved_place�base_dir�relatives r r' r' M s� � � ��<�<�D��(�^�^�i�/��1A�1A�$�s�)�)�1L�1L�L�N� �D� �F�� � � �� ��<�� � � � � 2�3�3�D� �F� �E�� !� � � � �H�� "� "� "�!� !� !� !� !� !� "���� �|�t�|�v�-�-� � ��^�^�%�%��(8�(8��(@�(@�@�N� � �>�>�/�/�� �<�<���� �|� ,� � � � � � � � ���� �!�-�-�h�7�7���� � � � � � � � � ���� �i� �� )�)s< � +B � B'� B'�&B'�$&D � D!� D!�%D; �; E � E )FN) �loggingr* rL rB r �pathlibr �typingr r �peeweer r �$defence360agent.model.simplificationr �+defence360agent.subsys.panels.hosting_panelr � getLoggerr r@ r- rI r& r) r r7 r: �strr# r' r9 r r �<module>re s\ �� ���� � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � #� #� #� #� #� #� #� #� ?� >� >� >� >� >� D� D� D� D� D� D� �� �8� $� $�� "� � ��� � � �H�5� �� � �",� ,� ,� ,�, � � �;�e�C��I�.>� ;� ;� ;� ;�'*�c� '*�e�D�$�J� /� '*� '*� '*� '*� '*� '*r