mirror of
https://github.com/saymrwulf/uhd.git
synced 2026-05-16 21:10:10 +00:00
e9288a439c
- Fix typo in company name (missing 'a') - Updated SPDX license identifier to version 3.0
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
#
|
|
# Copyright 2018 Ettus Research, a National Instruments Company
|
|
#
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
"""
|
|
systemd watchdog control module
|
|
"""
|
|
|
|
import os
|
|
import time
|
|
import threading
|
|
from systemd import daemon
|
|
|
|
MPM_WATCHDOG_DEFAULT_TIMEOUT = 30
|
|
# How often per watchdog interval we send a ping
|
|
MPM_WATCHDOG_TIMEOUT_FRAC = 3.0
|
|
|
|
def has_watchdog():
|
|
"""Check if the system has a watchdog checking on us.
|
|
|
|
We do this by checking on a set value for WATCHDOG_USEC.
|
|
"""
|
|
return bool(os.environ.get('WATCHDOG_USEC', False))
|
|
|
|
def transfer_control(pid):
|
|
"""
|
|
Transfer control of watchdog notifications to new PID.
|
|
"""
|
|
daemon.notify("MAINPID={:d}".format(int(pid)))
|
|
|
|
def _watchdog_task(shared_state, log):
|
|
"""
|
|
Continuously ping the watchdog to tell him that we're still alive.
|
|
|
|
This will keep running until the parent thread dies, or
|
|
shared_state.system_ready gets set to False by someone.
|
|
"""
|
|
watchdog_timeout = \
|
|
float(os.environ.get(
|
|
'WATCHDOG_USEC',
|
|
MPM_WATCHDOG_DEFAULT_TIMEOUT
|
|
)) / 1e6
|
|
watchdog_interval = watchdog_timeout / MPM_WATCHDOG_TIMEOUT_FRAC
|
|
daemon.notify("READY=1")
|
|
log.debug("Watchdog primed, going into watchdog loop (Interval: %s s)",
|
|
watchdog_interval)
|
|
while shared_state.system_ready.value:
|
|
# Sleep first, then ping, that avoids the case where transfer_control()
|
|
# is not yet complete before we call this for the first time, which
|
|
# would lead in error messages popping up in the systemd journal.
|
|
time.sleep(watchdog_interval)
|
|
log.trace("Pinging watchdog....")
|
|
daemon.notify("WATCHDOG=1")
|
|
log.error("Terminating watchdog thread!")
|
|
return
|
|
|
|
def spawn_watchdog_task(shared_state, log):
|
|
"""Spawn and return watchdog thread.
|
|
|
|
Creates a daemonic thread, because we don't want the watchdog task to
|
|
outlive the main thread.
|
|
"""
|
|
task = threading.Thread(
|
|
target=_watchdog_task,
|
|
args=[shared_state, log],
|
|
name="MPMWatchdogTask",
|
|
daemon=True,
|
|
)
|
|
task.start()
|
|
return task
|
|
|