#!/opt/imunify360/venv/bin/python import socket import json import time from pathlib import Path from logging import getLogger from defence360agent.internals import logger as lg from defence360agent.contracts.config import SimpleRpc from defence360agent.subsys.persistent_state import PERSISTENT_STATE_DIR logger = getLogger("check_recurrent") def is_need_to_run(lock_file: Path): try: content = lock_file.read_text() next_run = float(content) if content else 0 current_time = time.time() if current_time >= next_run: logger.info( "Need to wakeup agent to process scheduled task %s " "(scheduled time: %d, current time: %d, next run in: %d)", lock_file.name, current_time, next_run, next_run - current_time, ) return True return False except (FileNotFoundError, ValueError) as e: logger.error( "Error while reading scheduled-lock file %s %s", lock_file, e ) return False def main(): for lock_file in PERSISTENT_STATE_DIR.glob("*.lock"): if is_need_to_run(lock_file): try: with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as sock: sock.settimeout(30) sock.connect(SimpleRpc.SOCKET_PATH) # it's doesn't matter what we send, we just need to wake up the server msg = ( json.dumps({"command": ["wakeup"], "params": {}}) + "\n" ) sock.sendall(msg.encode()) except Exception as e: logger.error("Failed to connect to rpc socket %s", e) else: return if __name__ == "__main__": lg.reconfigure() main()