OwlCyberSecurity - MANAGER
Edit File: imunify_patch_subscription.cpython-311.opt-1.pyc
� E�\hq � � � d Z ddlZddlmZ ddlmZmZ ddlmZ ddl m Z m Z m Z ddl mZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZmZ dd lmZ er ddlm Z ddlm Z ee � � Z G d� de e � � 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)� getLogger)� TYPE_CHECKING�Any)� MessageType)� MessageSink� MessageSource�expect)�Scope)�get_imunify_patch_id)�VulnerabilityPatchTask)�ImunifyPatchSubscriptionAPI)�VulnerabilityHitStatus)�ImunifyPatchSubscription�VulnerabilityHit)� user_list)�ImunifyPatchUserId)� RefreshImunifyPatchSubscriptionc � � e Zd Zej Zd� Zd� Zdd�Zde de f fd�Z e e j � � dd �� � Zd eddfd �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_sink6 s � � � � �� c � � K � || _ d S r )�_sink)r r �sinks r � create_sourcez,ImunifyPatchSubscriptionPlugin.create_source9 s � � � ��� � � r �returnNc �6 � t j � � | _ d S r )�asyncio�Lock�_lock)r s r �__init__z'ImunifyPatchSubscriptionPlugin.__init__< s � � �\�^�^�� � � r r c � � K � i }t j � � � d {V ��D ]*}t |d � � � d {V ��}|r |d ||< �+|S )N�user)r � panel_usersr )r � users_mapr) �patch_ids r �_get_users_mapz-ImunifyPatchSubscriptionPlugin._get_users_map? su � � � �� �#�/�1�1�1�1�1�1�1�1� 3� 3�D� 1�$�v�,� ?� ?�?�?�?�?�?�?�H�� 3�&*�6�l� �(�#���r �_r c � �@ ��K � | j 4 �d {V �� | � � � � d {V ����s.t � d� � d d d � � �d {V �� d S 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 )Nz,No users found while refreshing subscriptionc �( �� g | ]}|�v��| ��S r r )�.0�user_id�subscribed_users_ids�userss ��r � <listcomp>zGImunifyPatchSubscriptionPlugin.refresh_subscription.<locals>.<listcomp>] s3 �� � "� "� "���"6�6�6� �g��6�6�6r ) r& r- �logger�warningr �get_subscriptions�list�keys�_set_subscription�_unset_subscription)r r. r2 �unsubscribed_usersr3 r4 s @@r �refresh_subscriptionz3ImunifyPatchSubscriptionPlugin.refresh_subscriptionG s~ ��� � � � �:� 9� 9� 9� 9� 9� 9� 9� 9� �-�-�/�/�/�/�/�/�/�/�E�� ���� M�N�N�N�� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 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 �8D � <D �,AD � D�Dr) 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)r2 c � � g | ] }|j �� S r )� orig_file)r1 �xs r r5 zDImunifyPatchSubscriptionPlugin._set_subscription.<locals>.<listcomp>g s* � � � � �� � � � � r )�filelistr) ) r � get_or_creater �selectrA �wherer) �statusr � VULNERABLEr �process_messager )r r) r. �created�pathss r r; z0ImunifyPatchSubscriptionPlugin._set_subscriptiond s� � � � �-�;�D�I�I�I� ��7� � � � �)�0�$�.�� ��%�$�)�T�1�$�+�-�8�9�� � � � �E� � ��j�0�0�*�E��E�E�E�� � � � � � � � � � � � � � r r4 c � � t j � � � t j � |� � � � � � � d S r )r �deleterF r2 �in_�execute)r r4 s r r< z2ImunifyPatchSubscriptionPlugin._unset_subscriptionv sA � � �'�)�)�/�/� $� ,� 0� 0�� 7� 7� � � �'�)�)�)�)�)r )r" N)r. r r" N)�__name__� __module__� __qualname__r �AV�SCOPEr r! r' �dictr r- r r r r> �strr; r9 r< r r r r r 3 s� � � � � � � �H�E� � � �� � �$� $� $� $��d�+?��+D�&E� � � � � �V�K� 7�8�8�9�2�9� �9� 9� 9� 9�8�9�8�C� �D� � � � �$��c�� �t� � � � � � r r )!�__doc__r$ �loggingr �typingr r �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr r r �defence360agent.utilsr � imav.contracts.imunify_patch_idr �imav.contracts.messagesr �.imav.malwarelib.api.imunify_patch_subscriptionr �imav.malwarelib.configr �imav.malwarelib.modelr r �imav.malwarelib.utilsr r r rP r6 r r r r �<module>rc s� ��� �* ���� � � � � � � %� %� %� %� %� %� %� %� :� :� :� :� :� :�� � � � � � � � � � (� '� '� '� '� '�� � � � � � ;� :� :� :� :� :�� � � � � � :� 9� 9� 9� 9� 9� L� L� L� L� L� L� L� L� +� +� +� +� +� +�� H�B�B�B�B�B�B�G�G�G�G�G�G� ��8� � ��F� F� F� F� F�[�-� F� F� F� F� Fr