관리-도구
편집 파일: model.cpython-311.pyc
� IMg� � �( � d dl Z d dlZd dlZd dlmZmZ d dlmZmZ d dl m c m c m Z d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZ d d lmZ ej e � � Z G d � d e� � Z d � Z d ee de!fd�Z"de!fd�Z#d ee de!fd�Z$dS )� N)�List�Optional)� BooleanField� CharField)� MessageType)�Model�instance)�run_in_executor)�execute_iterable_expression�� update_config)� MalwareHitc � � e Zd ZdZ G d� d� � Z ed�� � Z edd�� � Ze de e d e fd �� � Z e d ee d e fd �� � ZdS )� MyImunifyz Secure-site related settingsc � � e Zd Zej ZdZdS )�MyImunify.Meta� myimunifyN)�__name__� __module__� __qualname__r �db�database�db_table� � �T/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/myimunify/model.py�Metar s � � � � � ��;�� ���r r T)�uniqueF)�null�default�user�returnc �\ � |�|dk rdS | � |ddi�� � \ }}|j S )z%Get SecureSite protection by usernameN�rootT� protectionF)r! �defaults)� get_or_creater% )�clsr! �perm�_s r �get_protectionzMyImunify.get_protection! s? � � �<�4�6�>�>��4��#�#�� �u�8M�#�N�N���a��� r �users�statusc � �� | � �fd�|D � � � � � | j gg | j �i�� � � � � d S )Nc � �� g | ]}|�d ���S ))r! r% r )�.0r! r- s �r � <listcomp>z5MyImunify.update_users_protection.<locals>.<listcomp>. s! �� � D� D� D�d�d�&� 1� 1� D� D� Dr )�conflict_target�preserve�update)� insert_many� on_conflictr! r% �execute)r( r, r- s `r �update_users_protectionz!MyImunify.update_users_protection+ sd �� � ��� D� D� D� D�e� D� D� D� � � �+� �X�J���N�F�+� � � � �'�)�)�)�)�)r N)r r r �__doc__r r r! r r% � classmethodr �str�boolr+ r r8 r r r r r s� � � � � � �&�&� � � � � � � � � �9�D� !� !� !�D� � �5�%�8�8�8�J�� �(�3�-� �D� � � � �[� � ��D��I� �t� � � � �[�� � r r c � � K � t j |d�� � }|r0| � t j |�� � � � � d {V �� d S d S )NT)r! �cleanup)�hits)r �malicious_select�process_messager �MalwareCleanupTask)�sinkr! r? s r �malware_cleanuprD 6 sj � � � � � &�D�$� ?� ?� ?�D� � N��"�"�;�#A�t�#L�#L�#L�M�M�M�M�M�M�M�M�M�M�M�N� Nr r, r- c � � � ���K � t d ��fd�� � � d {V �� d ��sd��� fd��D � � }�r|� fd��D � � z }t j |� � d {V �� d S )Nc �: �� t � �� � � S �N)r r8 )r- r, s ��r �<lambda>z)update_users_protection.<locals>.<lambda>? s �� � �1�1�%��@�@� r �LOGc �: �� g | ]}t �d d�ii|� � ��S )�PROACTIVE_DEFENCE�moder )r0 r! �proactive_moderC s ��r r1 z+update_users_protection.<locals>.<listcomp>D sH �� � � � � � � � �6�>�":� ;� � � � � � r c �0 �� g | ]}t �|� � ��S r )rD )r0 r! rC s �r r1 z+update_users_protection.<locals>.<listcomp>N s# �� �@�@�@�$�/�$��-�-�@�@�@r )r �asyncio�gather)rC r, r- �tasksrM s ``` @r r8 r8 < s� ����� � � � � �@�@�@�@�@� � � � � � � � � � �N� � � �� � � � � � � � � �E� � A� �@�@�@�@�%�@�@�@�@�� �.�%� � � � � � � � � � r c � � K � t j � � � � � � d{V ��}t | ||� � � d{V �� dS )z#Set protection status for all usersN)�hp� HostingPanel� get_usersr8 )rC r- � panel_userss r �#set_protection_status_for_all_usersrW S s[ � � � ���)�)�3�3�5�5�5�5�5�5�5�5�K� !�$� �V� <� <�<�<�<�<�<�<�<�<�<r r" c � � K � t |� � }t � t j � � }t t j |� � � � � � }||z }|r;t � d|� � d� }t |t |� � � � ||z }|r2t � d|� � t | |d� � � d{V �� t |� � pt |� � S )z5Synchronize existing permissions with myimunify userszRemove myimunify users %sc � � t � � � � t j � | � � � � S rG )r �delete�wherer! �in_)�users_to_removes r � expressionz sync_users.<locals>.expressione s9 � � �#�#�%�%�+�+���"�"�?�3�3�� � r zAdd permissions to users %sFN) �setr �selectr! � itertools�chain�tuples�logger�infor �listr8 r< )rC r, rV �myimunify_usersr] r^ � users_to_adds r � sync_usersri Y s� � � � ��e�*�*�K� �&�&�y�~�6�6�O��)�/�?�+A�+A�+C�+C�D�E�E�O�%� �3�O�� G�� � �/� �A�A�A� � � � $�J��_�0E�0E�F�F�F� � �0�L�� :�� � �1�<�@�@�@�%�d�E�5�9�9�9�9�9�9�9�9�9� � � � � 6��o�!6�!6�6r )%rO ra �logging�typingr r �peeweer r �+defence360agent.subsys.panels.hosting_panel�subsys�panels� hosting_panelrS �"defence360agent.contracts.messagesr �defence360agent.modelr r �$defence360agent.model.simplificationr �defence360agent.utilsr � defence360agent.utils.configr �imav.malwarelib.modelr � getLoggerr rd r rD r; r< r8 rW ri r r r �<module>rx s� �� ���� � � � � ���� !� !� !� !� !� !� !� !� *� *� *� *� *� *� *� *� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� :� :� :� :� :� :� 1� 1� 1� 1� 1� 1� 1� 1� @� @� @� @� @� @� =� =� =� =� =� =� 6� 6� 6� 6� 6� 6� ,� ,� ,� ,� ,� ,� �� �8� $� $�� � � � � �� � � �FN� N� N� !�t�C�y� !�$� !� !� !� !�.=�D� =� =� =� =� 7�$�s�)� 7�� 7� 7� 7� 7� 7� 7r