You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
3.7 KiB
139 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()
|