관리-도구
편집 파일: utils.cpython-311.pyc
� IMg! � �V � d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m Z d dl m Z d dl mZmZmZ d dlZd dlZd dlmZ d dlmZ d d lmZ d d lmZm Z m Z m Z m Z m Z d � Z!d � Z"dd�Z# ddee ddf fd�Z$ddeee fd�Z% ed� � d� � � Z&d� Z' dd�Z(dS )� N)�suppress)� lru_cache�wraps)�chain)�Path)�Optional�Tuple� Generator)� SimpleRpc)�run_in_executor)�ValidationError)� AV_PID_PATH�IM360_NON_RESIDENT_PID_PATH�IM360_RESIDENT_PID_PATH�antivirus_mode�is_centos6_or_cloudlinux6� is_root_userc �� � t j rt nt } | � � � r�t j � � }t t � � 5 t | � � � � � }t � � r1t j |� � � � � d j }||k ot j |� � cddd� � S # 1 swxY w Y dS )z/Check if non-resident agent instance is runningr NF)r �enabledr r �exists�os�getpidr � Exception�int� read_textr �psutil�Process�children�pid� pid_exists)�rpc_process_pid_path� current_pidr s �T/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/rpc_tools/utils.py�rpc_is_runningr$ s � � &�-�N� � �3N� � �"�"�$�$� A��i�k�k� � �i� � � A� A��*�4�4�6�6�7�7�C�(�*�*� <� �n�S�)�)�2�2�4�4�Q�7�;���+�%�@�&�*;�C�*@�*@� A� A� A� A� A� A� A� A� A� A� A� A���� A� A� A� A� �5s �A:C � C!�$C!c � � t j rt � � S t j � � rMt j � � } t t j � � � � }|| k ot j |� � S dS )z&Check if the agent instance is runningF) �Config�SOCKET_ACTIVATIONr$ r r r r r r r r )r" r s r# � is_runningr( - sl � � � � � � � � � �%�'�'� =��i�k�k� ��)�3�5�5�6�6���k�!�<�f�&7��&<�&<�<� �5� �schemac # � K � t | |� � D ]�}|� d� � x}� � � r't j |� � � � � }n&t j |� � � � � }|� � � D ]+\ }}t |� d� � � � |fV � �,��d S )Nz.pickle� ) �find_schema_files� with_suffixr �pickle�loads� read_bytes�yaml� safe_loadr �items�tuple�split)�base� schema_dir�path�p�document�k�vs r# � _find_schemar> 9 s� � � � �!�$� �3�3� )� )���!�!�)�,�,� ,�A� 4� 4� 6� 6� 8� �|�A�L�L�N�N�3�3�H�H��~�d�n�n�&6�&6�7�7�H� �N�N�$�$� )� )�D�A�q������ � �%�%�q�(� (� (� (� (� )� )� )r) �returnc # �z K � | |z }g |� d� � �|� d� � �D ]}|V � �d S )Nz*.yamlz*.yml)�rglob)r7 r8 r: r9 s r# r- r- E sX � � � � �z��A�7�!�'�'�(�#�#�7�a�g�g�g�&6�&6�7� � ��� � � � �� r) �pathsc �� � | rt | � � d d � ng }t t � � j j dz }|� ||j dz dz g� � |S )N� simple_rpc�feature_management�rpc)�listr �__file__�parent�extend)rB �resultr9 s r# �get_schema_pathsrL M si � � $� ,�T�%�[�[����^�^�"�F� ��>�>� � '�,� 6�D� �M�M� � �K�.� .�� 6� �� � � �Mr) � c � � g }t | � � D ]$}|� t |� � � � �%t t |� � � S �N)rL �appendr> �dictr )rB r* � base_paths r# �prepare_schemarS Y sM � � �F�%�e�,�,� /� /� �� � �l�9�-�-�.�.�.�.� ��v�� � � r) c �< � � t � � � � fd�� � }|S )Nc � �d �� �K � t t j � � � ��fd�� � � d {V ��S )Nc � �� �� i ���S rO � )�args�f�kwargss ���r# �<lambda>z<run_in_executor_decorator.<locals>.wrapper.<locals>.<lambda>e s �� �a�a��.@��.@�.@� r) )r �asyncio�get_event_loop)rX rZ rY s ``�r# �wrapperz*run_in_executor_decorator.<locals>.wrapperb sW ���� � � �$� � "� $� $�&@�&@�&@�&@�&@�&@� � � � � � � � � r) )r )rY r^ s ` r# �run_in_executor_decoratorr_ a s3 �� � �1�X�X� � � � � �X� � �Nr) c �� � g }|D ]w}|d |� d|� � } } | r,|r*|� |� | | � � � � �N|� |� | |� � � � �xt | � � } t |� � } | dk r/| | k r)|� |� | | � � � � |rt |� � �i S )aP :param list affected: IPs that were changed during operation :param list of tuples || list of str not_affected: IPs & it's listnames that weren't changed during operation :param list all_list: list of all IPs that take place in operation :param str success_warning: msg if IP was changed :param str failure_warning: msg if IPs wasn't changed and it's absent in any other lists :param str in_another_list_warning: msg if IPs wasn't changed , however it present in another list :return list of st warnings: msg to be printed �rec�listnamerM )�getrP �format�lenr ) �affected� not_affected� dest_listname�all_list�success_warning�failure_warning�in_another_list_warning�warnings�item�recordrb � num_deleted� total_nums r# �generate_warningsrr k s� � �* �H� � K� K�� ��;�����]�(K�(K��� � K�/� K� �O�O� 3� :� :�6�8� L� L� M� M� M� M� �O�O�O� 2� 2�6�=� I� I� J� J� J� J��h�-�-�K��H� � �I��1�}�}��k�1�1�����.�.�{�I�F�F�G�G�G�� (� �h�'�'�'� �Ir) )Nr* rO ))r/ r\ r �socket� contextlibr � functoolsr r � itertoolsr �pathlibr �typingr r r r2 r � defence360agent.contracts.configr r& �$defence360agent.model.simplificationr �"defence360agent.rpc_tools.validater �defence360agent.utilsr r r r r r r$ r( r> r- rL rS r_ rr rW r) r# �<module>r} s �� � � � � ���� � � � � � � � � � � � � � � &� &� &� &� &� &� &� &� � � � � � � � � � � � � -� -� -� -� -� -� -� -� -� -� � � � � � � � � @� @� @� @� @� @� @� @� @� @� @� @� >� >� >� >� >� >�� � � � � � � � � � � � � � � �� � �$ � � � )� )� )� )� #�� ��t�T�4� � �� � � � � �H�U�4�[� 1� � � � � ��1� � � � � � � �� � �" !�&� &� &� &� &� &r)