관리-도구
편집 파일: defence360.cpython-311.pyc
� IMg� � � � d dl Z d dlZd dlZd dlZd dlZd dlmZ d dl m Z m Z d dl m Z d dlmZmZmZmZmZ d dlmZmZ d dlmZ ej e� � Zd� Z d � Z dS ) � N)�Core)�SUCCESS� SocketError)� is_root_user)� EXIT_CODES�EXITCODE_GENERAL_ERROR� print_error�print_response�print_warnings)� EnvParser�create_cli_parser)� flush_sentryc �$ � t j t j � � t j j � � � | � � t � � }|� |�� � }|j s t j � d� � rHt j j � |j p t j � d� � � � |j r)t j j � |j � � t! |d� � �rPt! |d� � �r? |� |� � }t% j t j |j |j |�� � } |j di |�|��\ }}t/ |� � t1 � � |t2 k r"t5 |j ||j |j � � n=t; |||j |j � � t= j t@ | � � d S d S # tB $ rU}t5 d dd�" |� � i|j |j � � t= j tF � � Y d }~d S d }~ww xY wtI |�% � � � � d S ) N)�args� IMUNIFY360_LOGGING_CONFIG_FILE�endpoint�generate_endpoint_params)�exclude�itemsz ERROR: {}� )&�os�umask�Config� FILE_UMASK�defence360agent� internals�logger� reconfigurer � parse_args� log_config�environ�get� update_logging_config_from_file�console_log_level�setConsoleLogLevel�hasattrr r �parse�command�envvar_parameter_optionsr r r r r �json�verboser �sys�exitr r �formatr �print� format_help) �rpc_handlers_init�cli_args�parserr � cli_kwargs� envvar_kwargs�result�data�es �O/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/defence360.py�mainr: s� � ��H�V� � � � �� �$�0�0�2�2�2������ � � �F� � � �(� � +� +�D� �� �"�*�.�.�)I�J�J� ��!�(�H�H� �O� O�r�z�~�~�.N� O� O� � � � � � ��!�(�;�;� � "� � � � �t�Z� � � $�W�T�3M�%N�%N� $� -� �6�6�t�<�<�J� %�O�� �� ��-�"� � � �M� )�4�=�G�G�=�G�J�G�G�L�F�D� �4� � � � �N�N�N��� � � �t�|�T�4�9�d�l�K�K�K�K��F�D�$�)�T�\�B�B�B����F�+�,�,�,�,�,� L�K�� � -� -� -� ��w� � 2� 2�1� 5� 5�6�� �4�<� � � � �H�+� ,� ,� ,� ,� ,� ,� ,� ,� ,����� -���� �f� � �"�"�#�#�#�#�#s �1CH � I,�A I'�'I,c �j � t � � sYt � dt j � � t dt j �� � t j t � � t | t j dd � � � t j � � � � � d S # t $ r7 t � d� � t j t � � Y d S t" $ r7 t � d� � t j t � � Y d S w xY w)Nz'%s could be used by the root user only!z/Imunify360 CLI is unavailable for non-root user)�file� z User pressed Ctrl+C, exiting...z5Unknown error happened. See logs for more information)r r �infor �NAMEr/ r, �stderrr- r r: �argv�asyncio�get_event_loop�close�KeyboardInterrupt�warning� Exception� exception)r1 s r9 � entrypointrI F s* � � �>�>� )�� � �=�v�{�K�K�K� � =�C�J� � � � � ��'�(�(�(� )� � ������ �-�-�-� � � � �&�&�(�(�(�(�(�� � )� )� )����8�9�9�9� ��'�(�(�(�(�(�(� � )� )� )���� C� � � � ��'�(�(�(�(�(�(� )���s �)AB2 �2=D2�2<D2�1D2) rB �loggingr r, � defence360agent.internals.loggerr � defence360agent.contracts.configr r �defence360agent.simple_rpcr r �defence360agent.utilsr �defence360agent.utils.clir r r r r � defence360agent.utils.parsersr r �defence360agent.sentryr � getLogger�__name__r r: rI r � r9 �<module>rU s! �� ���� ���� � � � � � � � � '� '� '� '� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� .� .� .� .� .� .�� � � � � � � � � � � � � � G� F� F� F� F� F� F� F� /� /� /� /� /� /� �� �8� $� $��+$� +$� +$�\)� )� )� )� )rT