Edit File: server_pull.cpython-311.opt-1.pyc
� Gh= � �� � d Z ddlZddlZddlZddlmZ ddlZddlmZ ddl m Z ddl m Z ddl mZ ddlmZmZ dd lmZ dd lmZ ej e� � Z G d � d e� � 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)�Path)� ExitStack)�utils)�Core)� MessageType)� MessageSink�expect)�Scope)�malware_responsec �X � e Zd Zej ZdZd� Z ee j � � d� � � Z dS )�ServerPullFileSenderNc � � K � || _ d S )N)�loop)�selfr s �M/opt/imunify360/venv/lib/python3.11/site-packages/imav/plugins/server_pull.py� create_sinkz ServerPullFileSender.create_sink+ s � � � ��� � � � c � �n K � |d }|d D �]�} t � � 5 }t j � |� � r�|� t j t j �� � � � }|� t j dd|g� � � d {V ��� � |� d� � t j |j |� � }t! j t% |j � � � � � � � � � � }t* � d|j |� � t j ||�� � � d {V �� d d d � � n # 1 swxY w Y ��c# t0 $ r2}t* � t5 |� � � � Y d }~���d }~ww xY wd S ) N�reason�files)�dir�lsz-alr z)Uploading temporary file %s with hash: %s)� upload_reason)r �os�path�isdir� enter_context�tempfile�NamedTemporaryFiler �TMPDIR�writer � check_run�seekr �MalwareHitPath�name�hashlib�sha256r � read_bytes� hexdigest�logger�info� upload_file� Exception�warning�str)r �messager �file�stack�tf�hash_�es r � upload_filesz!ServerPullFileSender.upload_files. s � � � ���"���G�$� '� '�D� '� �[�[� �E��w�}�}�T�*�*� � "� 0� 0� $� 7�D�K� H� H� H� � �� ���u���e�T�7J�'K�'K�!K�!K�!K�!K�!K�!K�L�L�L����� � � � /� >�r�w�� M� M�� '�� ���M�M� 4� 4� 6� 6�!� !�#�)�+�+� � � � � G� �G� !�� � � +�6� �F�� � � � � � � � � �#� � � � � � � � � � ���� � � � ���( � '� '� '����s�1�v�v�&�&�&�&�&�&�&�&����� '����- '� 's; �E6�D8E)� E6�)E- �-E6�0E- �1E6�6 F2� 'F-�-F2) �__name__� __module__� __qualname__r �IM360�SCOPEr r r r �MalwareSendFilesr6 � r r r r ' sX � � � � � � �K�E� �D�� � � �V�K� (�)�)�'� '� *�)�'� '� 'r r )�__doc__r& �loggingr �pathlibr r � contextlibr �defence360agentr � defence360agent.contracts.configr �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr r �defence360agent.utilsr �imav.malwarelib.utilsr � getLoggerr7 r* r r= r r �<module>rI s ��� �* ���� ���� � � � � � � � � � � ���� � � � � � � !� !� !� !� !� !� 1� 1� 1� 1� 1� 1� :� :� :� :� :� :� A� A� A� A� A� A� A� A� '� '� '� '� '� '� 2� 2� 2� 2� 2� 2� �� �8� $� $��!'� !'� !'� !'� !'�;� !'� !'� !'� !'� !'r