관리-도구
편집 파일: schedule_watcher.cpython-311.pyc
� ��g� � �, � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl m Z dd l mZ dd lmZ dd lmZ dd lmZmZmZ dd lmZ ee� � Zej ej ej ej gZ ej ej gZ!d� Z" G d� dee� � 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> � )�suppress)� getLogger)�Path)�utils)�ANTIVIRUS_MODE)�MalwareScanSchedule)�MalwareScanScheduleInterval)� SystemConfig)� LicenseCLN)� MessageType)� MessageSink� MessageSource�expect)�reset_malware_schedulec �\ � t j � � } t p| }|rt nt S �N)r �is_valid_av_plusr �AVAILABLE_INTERVALS� AVP_INTERVALS)� valid_avp� conditions �]/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/plugins/schedule_watcher.py�allowed_schedule_intervalr 9 s. � ��+�-�-�I�#�#�1� �I�"+� >� � � �>� c � � e Zd Zd� Zd� Zd� Zed� � � Zed� � � Zd� Z d� Z d� Z e e j � � ej � � d � � � � � Zd S ) �ScheduleWatcherc �n � | � � � | _ | � � � d | _ d S r )� _read_cron�_cron� _update_cron�_sink��selfs r �__init__zScheduleWatcher.__init__@ s0 � ��_�_�&�&�� � �������� � � r c � � K � d S r � )r# �loops r � create_sinkzScheduleWatcher.create_sinkE s � � � � �r c � � K � || _ d S r )r! )r# r' �sinks r � create_sourcez ScheduleWatcher.create_sourceH s � � � ��� � � r c � � t t � � 5 t t j � � � � � cd d d � � S # 1 swxY w Y d S r )r �FileNotFoundErrorr �Config� CRON_PATH� read_textr& r r r zScheduleWatcher._read_cronK s� � � �'� (� (� 6� 6���(�)�)�3�3�5�5� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6���� 6� 6� 6� 6� 6� 6s �+A � A�Ac � � t t j � � }t t � � 5 |� � � d d d � � n # 1 swxY w Y | r-|� d�� � |� | � � d S d S )Ni� )�mode)r r. r/ r r- �unlink�touch� write_text)�job�paths r � _write_cronzScheduleWatcher._write_cronP s� � ��F�$�%�%�� �'� (� (� � � �K�K�M�M�M� � � � � � � � � � � ���� � � � � � !� �J�J�E�J� "� "� "� �O�O�C� � � � � � !� !s �A�A�Ac �f � t j t j k rd S t j t vr,t � dt j � � | j S t j t � � vr(t � d� � t � � t j t j ddft j t j dt j ft j t j t j dfi}|t j }t j j |dt j i�}|S )Nz Unsupported interval value: %sz4Malware schedule interval is being reset to defaults�*�cmd)r. �INTERVAL�Interval�NONEr �logger�errorr r �infor �DAY�HOUR�WEEK� DAY_OF_WEEK�MONTH� DAY_OF_MONTH� CRON_STRING�format�CMD)r# �schedule_schema�scheduler6 s r �_get_jobzScheduleWatcher._get_job[ s� � � �?�h�m� +� +��4� �_�$7� 7� 7� �L�L�9�6�?� K� K� K��:� � �_�$=�$?�$?� ?� ?� �K�K�N� O� O� O� "� $� $� $� �L�6�;��S�1� �M�F�K��f�.@�A� �N�V�[�&�*=�s� C� �� #�6�?�3��� �'��B�v�z�B�B��� r c � � | � � � }| j |k r8t � d� � | � |� � || _ dS dS )Nz Update background scan scheduleTF)rM r r? rA r8 )r# r6 s r r z ScheduleWatcher._update_crono sT � ��m�m�o�o�� �:�� � � �K�K�9� :� :� :� � � �S� !� !� !� �D�J��4��ur c � �l K � | j � t j � � � � � d {V �� d S r )r! �process_messager � MalwareScanQueueStopBackgroundr" s r �_stop_background_scanz%ScheduleWatcher._stop_background_scan{ sT � � � ��j�(�(� � 6� 8� 8� � � � � � � � � � � r c � � K � t |d t � � r7| � � � r%| j s | � � � � d {V �� d S d S d S d S )N�conf)� isinstancer r r rR )r# �messages r �schedule_config_updatedz'ScheduleWatcher.schedule_config_updated� s� � � � � �g�f�o�|� 4� 4� 3�� � �"�"� 3�4�:� 3��0�0�2�2�2�2�2�2�2�2�2�2�2� 3� 3� 3� 3� 3� 3r N)�__name__� __module__� __qualname__r$ r( r+ � staticmethodr r8 rM r rR r r � ConfigUpdater �log_error_and_ignorerW r& r r r r ? s� � � � � � �� � � � � �� � � �6� 6� �\�6� �!� !� �\�!�� � �( � � � � � � �V�K� $�%�%� �U� �!�!�3� 3� "�!� &�%�3� 3� 3r r N)$�__doc__� contextlibr �loggingr �pathlibr �defence360agentr � defence360agent.contracts.configr r r. r r= r �!defence360agent.contracts.licenser �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr r r �imav.malwarelib.utilsr rX r? r> rB rD rF r r r r r& r r �<module>rh s� ��� �* � � � � � � � � � � � � � � � � � � !� !� !� !� !� !� ;� ;� ;� ;� ;� ;� J� J� J� J� J� J�� � � � � � :� 9� 9� 9� 9� 9� 8� 8� 8� 8� 8� 8� :� :� :� :� :� :�� � � � � � � � � � 9� 8� 8� 8� 8� 8� ��8� � �� �M� �L� �M� �N� � � �M� �N�� � ?� ?� ?� F3� F3� F3� F3� F3�k�=� F3� F3� F3� F3� F3r