134 lines
3.7 KiB
134 lines
3.7 KiB
#!/usr/bin/python3
|
|
"""
|
|
Copyright: 2014 Sipwise Development Team <support@sipwise.com>
|
|
|
|
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 package 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 <http://www.gnu.org/licenses/>.
|
|
|
|
On Debian systems, the complete text of the GNU General
|
|
Public License version 3 can be found in "/usr/share/common-licenses/GPL-3".
|
|
"""
|
|
import logging
|
|
import os
|
|
import os.path
|
|
import pyinotify
|
|
import sys
|
|
import argparse
|
|
import signal
|
|
|
|
BASE_DIR = "/usr/share/kamailio-config-tests"
|
|
if 'BASE_DIR' in os.environ:
|
|
BASE_DIR = os.environ['BASE_DIR']
|
|
filelog = os.path.join(BASE_DIR, 'log', 'pid_watcher.log')
|
|
logging.basicConfig(
|
|
filename=filelog, level=logging.DEBUG, format='%(asctime)s %(message)s')
|
|
|
|
base_path = "/run"
|
|
|
|
"""
|
|
files we want to watch
|
|
"""
|
|
services = [
|
|
'kamailio/kamailio.lb.pid',
|
|
'kamailio/kamailio.proxy.pid',
|
|
'sems-b2b/sems-b2b.pid',
|
|
'ngcp-witnessd.pid'
|
|
]
|
|
|
|
watched_dirs = [
|
|
'/run/kamailio',
|
|
'/run/sems-b2b',
|
|
'/run'
|
|
]
|
|
|
|
watched = {}
|
|
|
|
|
|
def sigterm_handler(_signo, _stack_frame):
|
|
logging.info("Process aborted")
|
|
sys.exit(2)
|
|
|
|
|
|
class Handler(pyinotify.ProcessEvent):
|
|
|
|
def my_init(self, watched):
|
|
self.watched = watched
|
|
|
|
def check_done(self):
|
|
logging.info("All OK, done")
|
|
sys.exit(0)
|
|
|
|
def check_all(self):
|
|
all = True
|
|
for k, v in self.watched.items():
|
|
all = (all and (v['created'] or v['modified']))
|
|
logging.info("checking: %s[%s] all:%s" % (k, v, all))
|
|
return all
|
|
|
|
def process_IN_CREATE(self, event):
|
|
if event.pathname in watched:
|
|
watched[event.pathname]['created'] = True
|
|
logging.info("created %s" % event.pathname)
|
|
if self.check_all():
|
|
self.check_done()
|
|
|
|
def process_IN_IGNORED(self, event):
|
|
if event.pathname in watched:
|
|
watched[event.pathname]['deleted'] = True
|
|
logging.info("deleted %s" % event.pathname)
|
|
|
|
def process_IN_MODIFY(self, event):
|
|
if event.pathname in watched:
|
|
watched[event.pathname]['modified'] = True
|
|
logging.info("modified %s" % event.pathname)
|
|
if self.check_all():
|
|
self.check_done()
|
|
# for debug
|
|
# def process_default(self, event):
|
|
# if watched.has_key(event.pathname):
|
|
# print event
|
|
|
|
# catch SIGTERM signal
|
|
signal.signal(signal.SIGTERM, sigterm_handler)
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description='watch some pids to detect restarts')
|
|
parser.add_argument('--pbx', dest='pbx', action='store_true',
|
|
help='pbx is enabled')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.pbx:
|
|
watched_dirs += [
|
|
'/run/fastcgi',
|
|
]
|
|
services += [
|
|
'fastcgi/ngcp-panel.pid',
|
|
]
|
|
|
|
logging.info("PID watcher started")
|
|
wm = pyinotify.WatchManager()
|
|
handler = Handler(watched=watched)
|
|
notifier = pyinotify.Notifier(wm, default_proc_fun=handler)
|
|
for service in services:
|
|
service_pid = os.path.join(base_path, service)
|
|
logging.info("Watching %s" % service_pid)
|
|
watched[service_pid] = {
|
|
'deleted': False, 'created': False, 'modified': False}
|
|
wm.add_watch(service_pid, pyinotify.IN_IGNORED)
|
|
for d in watched_dirs:
|
|
wm.add_watch(d, pyinotify.ALL_EVENTS)
|
|
|
|
logging.info("start loop")
|
|
notifier.loop()
|