Edit File: strategy_getter.cpython-311.pyc
� P 6h � � � d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m Z ddl m Z ej e� � Z G d� dej � � ZdS ) zDPlugin that detects whether IDS has set on non-resident agent start.� N)�utils)�plugins�messages��strategy)�configc �l � � e Zd ZdZ� fd�Zd� Zd� Z ej e � � d� � � Z d� Z d� Z � xZ S )�StrategyGetterz8Send StrategyChange message on non-resident agent start.c �H �� t � � j di |�� d | _ d S )N� )�super�__init__�_StrategyGetter__task)�self�kwargs� __class__s ��R/opt/imunify360/venv/lib/python3.11/site-packages/im360/plugins/strategy_getter.pyr zStrategyGetter.__init__ s+ �� ������"�"�6�"�"�"��� � � � c � �~ K � || _ || _ |� | � � � � � | _ d S �N)�_loop�_sink� create_task�_detect_strategy_changer )r �loop�sinks r � create_sourcez StrategyGetter.create_source s9 � � � ��� ��� ��&�&�t�'C�'C�'E�'E�F�F�� � � r c � � K � | j �| j � � � rd S | j � � � t j t j � � 5 | j � d {V �� d d d � � d S # 1 swxY w Y d S r )r �done�cancel� contextlib�suppress�asyncio�CancelledError�r s r �shutdownzStrategyGetter.shutdown s� � � � � �;� �$�+�"2�"2�"4�"4� � �F� � ��� � � � � ��!7� 8� 8� � ��+� � � � � � � � � � � � � � � � � � � ���� � � � � � s � A7�7A;�>A;c � �� K � t j t j � � 5 | � � � � d{V �� t j t j j � � � d{V �� �D# 1 swxY w Y dS )z+Check whether IDS has changed periodically.TN) r! r" r# r$ �_refresh_strategy�sleepr �Subsys� THIRD_PARTY_IDS_CHECK_TIMEOUTr% s r r z&StrategyGetter._detect_strategy_change" s� � � � � � ��!7� 8� 8� � � ��,�,�.�.�.�.�.�.�.�.�.� �m��M�?�� � � � � � � � � � � � � ���� � � � � � s �AA'�'A+�.A+c � �4 K � t j � � � }|t j j k r | � |� � � d {V �� d S d S # t j $ r Y d S t $ r&}t � d|� � Y d }~d S d }~ww xY w)NzFailed to get strategy, %s) r �Strategy�get�current�_on_strategy_changedr# r$ � Exception�logger�error)r � new_strategy�es r r( z StrategyGetter._refresh_strategy, s� � � � � >�#�,�0�0�2�2�L� �x� 0� 8�8�8��/�/� �=�=�=�=�=�=�=�=�=�=�=� 9�8�� �%� � � � �D�D�� :� :� :� �L�L�5�q� 9� 9� 9� 9� 9� 9� 9� 9� 9����� :���s � A �B�( B�1B�Bc � � K � t � dt j j |� � |t j _ | j � t j � |�� � � � � d {V �� d S )NzGot new strategy: %s -> %sr ) r2 �infor r- r/ r �process_messager � MessageType�StrategyChange)r r4 s r r0 z#StrategyGetter._on_strategy_changed7 s� � � � �� � � (� � � %� � � � � %1���!��j�(�(� � � /� /� � /� F� F� � � � � � � � � � � r )�__name__� __module__� __qualname__�__doc__r r r& r �abort_agent_onr1 r r( r0 � __classcell__)r s @r r r s� �� � � � � �B�B�� � � � �G� G� G� � � � �U��)�$�$�� � %�$�� >� >� >� � � � � � � r r )r> r# r! �logging�defence360agentr �defence360agent.contractsr r �im360.internalsr �im360.contractsr � getLoggerr; r2 � MessageSourcer r r r �<module>rH s� �� J� J� ���� � � � � ���� !� !� !� !� !� !� 7� 7� 7� 7� 7� 7� 7� 7� $� $� $� $� $� $� "� "� "� "� "� "� �� �8� $� $��2 � 2 � 2 � 2 � 2 �W�*� 2 � 2 � 2 � 2 � 2 r