Edit File: schedule_watcher.cpython-311.pyc
� i �h. � �B � 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! ed� � Z"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_schedulez-/var/imunify360/premium_revisium_license.flagc � � t j � � } t � � � }t p| p|}|rt nt S �N)r �is_valid_av_plus�REVISIUM_PREMIUM_MARKER�existsr �AVAILABLE_INTERVALS� AVP_INTERVALS)� valid_avp�revisium_license_exists� conditions �]/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/plugins/schedule_watcher.py�allowed_schedule_intervalr ? sF � ��+�-�-�I� 5� <� <� >� >��#�#�L� �L�5L�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__G s0 � ��_�_�&�&�� � �������� � � r c � � K � d S r � )r& �loops r � create_sinkzScheduleWatcher.create_sinkL s � � � � �r c � � K � || _ d S r )r$ )r& r* �sinks r � create_sourcez ScheduleWatcher.create_sourceO 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_cronR 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 r1 r2 r r0 �unlink�touch� write_text)�job�paths r � _write_cronzScheduleWatcher._write_cronW 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)r1 �INTERVAL�Interval�NONEr �logger�errorr" r �infor �DAY�HOUR�WEEK� DAY_OF_WEEK�MONTH� DAY_OF_MONTH� CRON_STRING�format�CMD)r& �schedule_schema�scheduler9 s r �_get_jobzScheduleWatcher._get_jobb 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)rP r" rB rD r; )r& r9 s r r# z ScheduleWatcher._update_cronv 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" rU )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! r; rP r# rU r r � ConfigUpdater �log_error_and_ignorerZ r) r r r r F s� � � � � � �� � � � � �� � � �6� 6� �\�6� �!� !� �\�!�� � �( � � � � � � �V�K� $�%�%� �U� �!�!�3� 3� "�!� &�%�3� 3� 3r r N)%�__doc__� contextlibr �loggingr �pathlibr �defence360agentr � defence360agent.contracts.configr r r1 r r@ r �!defence360agent.contracts.licenser �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr r r �imav.malwarelib.utilsr r[ rB rA rE rG rI r r r r r r) r r �<module>rk 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�� � �$� N�O�O� �?� ?� ?�F3� F3� F3� F3� F3�k�=� F3� F3� F3� F3� F3r