관리-도구
편집 파일: imunify_patch_subscription.cpython-311.pyc
� �]��Q�L� �� � d Z ddlZddlmZ ddlmZ ddlmZmZm Z ddl m Z ddl m Z ddlmZ dd lmZ dd lmZ dd lmZmZ dd lmZ erdd l mZ G d� dee� � ZdS )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)� TYPE_CHECKING)� MessageType)� MessageSink� MessageSource�expect)�Scope)�VulnerabilityPatchTask��ensure_id_file)�ImunifyPatchSubscriptionAPI)�VulnerabilityHitStatus)�ImunifyPatchSubscription�VulnerabilityHit)� user_list)� RefreshImunifyPatchSubscriptionc � � e Zd Zej Zd� Zd� Zd d�Z e e j � � dd�� � Z d e ddfd �Zd ee ddfd �ZdS )� ImunifyPatchSubscriptionPluginc � � K � d S �N� )�self�loops �g/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/plugins/imunify_patch_subscription.py� create_sinkz*ImunifyPatchSubscriptionPlugin.create_sink0 s � � � � �� c � � K � || _ d S r )�_sink)r r �sinks r � create_sourcez,ImunifyPatchSubscriptionPlugin.create_source3 s � � � ��� � � r �returnNc �6 � t j � � | _ d S r )�asyncio�Lock�_lock)r s r �__init__z'ImunifyPatchSubscriptionPlugin.__init__6 s � � �\�^�^�� � � r �_r c � � ��K � | j 4 �d {V �� d� t j � � � d {V ��D � � � d {V ���t j t �� � � � � � � � d {V ����� d d d � � �d {V �� d S �D ]'}|�v r!| � �| � � � d {V �� �(��fd��D � � }| � |� � d d d � � �d {V �� d S # 1 �d {V ��swxY w Y d S )Nc � �V K � i | ]$}t |d � � � d{V ��|d ��%S )�userNr ��.0�xs r � <dictcomp>zGImunifyPatchSubscriptionPlugin.refresh_subscription.<locals>.<dictcomp>> sP � � � � � � �� %�Q�v�Y�/�/�/�/�/�/�/�/��6��� � r c �( �� g | ]}|�v��| ��S r r )r+ �user_id�subscribed_users_ids�userss ��r � <listcomp>zGImunifyPatchSubscriptionPlugin.refresh_subscription.<locals>.<listcomp>N s3 �� � "� "� "���"6�6�6� �g��6�6�6r ) r$ r � panel_usersr �get_subscriptions�list�keys�_set_subscription�_unset_subscription)r r& r/ �unsubscribed_usersr0 r1 s @@r �refresh_subscriptionz3ImunifyPatchSubscriptionPlugin.refresh_subscription9 sN ��� � � � �:� 9� 9� 9� 9� 9� 9� 9� 9�� � (� 4� 6� 6�6�6�6�6�6�6�� � � � � � � � �E� 2�C����� � �&�&�� � � � � � � � !� $�+�� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 0� A� A���e�#�#� �0�0��w� �@�@�@�@�@�@�@�@�@��"� "� "� "� "� $�"� "� "� � � $� $�%7� 8� 8� 8�- 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9���� 9� 9� 9� 9� 9� 9s �A&C,� AC,�, C6�9C6r) c � �l K � t j |�� � \ }}|r�d� t j t j � � � t j |k t j t j k � � D � � }|r3| j � t ||�� � � � � d {V �� d S d S d S )N)r/ c � � g | ] }|j �� S r )� orig_filer* s r r2 zDImunifyPatchSubscriptionPlugin._set_subscription.<locals>.<listcomp>X s* � � � � �� � � � � r )�filelistr) ) r � get_or_creater �selectr= �wherer) �statusr � VULNERABLEr �process_messager )r r) r& �created�pathss r r7 z0ImunifyPatchSubscriptionPlugin._set_subscriptionU s� � � � �-�;�D�I�I�I� ��7� � � � �)�0�$�.�� ��%�$�)�T�1�$�+�-�8�9�� � � � �E� � ��j�0�0�*�E��E�E�E�� � � � � � � � � � � � � � r r1 c � � t j � � � t j � |� � � � � � � d S r )r �deleterA r/ �in_�execute)r r1 s r r8 z2ImunifyPatchSubscriptionPlugin._unset_subscriptiong sA � � �'�)�)�/�/� $� ,� 0� 0�� 7� 7� � � �'�)�)�)�)�)r )r N)r& r r N)�__name__� __module__� __qualname__r �AV�SCOPEr r r% r r r r: �strr7 r5 r8 r r r r r - s� � � � � � � �H�E� � � �� � �$� $� $� $� �V�K� 7�8�8�9�2�9� �9� 9� 9� 9�8�9�6�C� �D� � � � �$��c�� �t� � � � � � r r )�__doc__r" �typingr �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr r r �defence360agent.utilsr �imav.contracts.messagesr � imav.contracts.imunify_patch_idr �.imav.malwarelib.api.imunify_patch_subscriptionr �imav.malwarelib.configr �imav.malwarelib.modelr r �imav.malwarelib.utilsr r r r r r �<module>r\ sU ��� �* ���� � � � � � � :� :� :� :� :� :�� � � � � � � � � � (� '� '� '� '� '� :� :� :� :� :� :� :� :� :� :� :� :�� � � � � � :� 9� 9� 9� 9� 9� L� L� L� L� L� L� L� L� +� +� +� +� +� +�� H�G�G�G�G�G�G�=� =� =� =� =�[�-� =� =� =� =� =r