Page MenuHomeVyOS Platform
Files F118309333

restart_frr.py
No OneTemporary
Actions

Size
4 KB
Referenced Files
None
Subscribers
None

restart_frr.py
View Options

#!/usr/bin/env python3
#
# Copyright (C) 2019 VyOS maintainers and contributors
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 or later as
# published by the Free Software Foundation.
#
# 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 <http://www.gnu.org/licenses/>.
#
import sys
import argparse
import subprocess
import logging
from logging.handlers import SysLogHandler
# some default values
watchfrr = '/usr/lib/frr/watchfrr.sh'
vtysh = '/usr/bin/vtysh'
frrconfig = '/etc/frr/frr.conf'
frrconfig_tmp = '/etc/frr/frr.conf.temporary'
# configure logging
logger = logging.getLogger(__name__)
logs_handler = SysLogHandler('/dev/log')
logs_handler.setFormatter(logging.Formatter('%(filename)s: %(message)s'))
logger.addHandler(logs_handler)
logger.setLevel(logging.INFO)
# save or restore current config file
def _save_and_restore(action):
if action == "save":
command = "sudo mv {} {}".format(frrconfig, frrconfig_tmp)
logmsg = "Permanent configuration saved to {}".format(frrconfig_tmp)
if action == "restore":
command = "sudo mv {} {}".format(frrconfig_tmp, frrconfig)
logmsg = "Permanent configuration restored from {}".format(frrconfig_tmp)
return_code = subprocess.call(command, shell=True)
if not return_code == 0:
logger.error("Failed to rename permanent config: \"{}\" returned exit code: {}".format(command, return_code))
return False
logger.info(logmsg)
return True
# write active config to file
def _write_config():
command = "sudo {} -n -c write ".format(vtysh)
return_code = subprocess.call(command, shell=True)
if not return_code == 0:
logger.error("Failed to save active config: \"{}\" returned exit code: {}".format(command, return_code))
return False
logger.info("Active config saved to {}".format(frrconfig))
return True
# check if daemon is running
def _daemon_check(daemon):
command = "sudo {} print_status {}".format(watchfrr, daemon)
return_code = subprocess.call(command, shell=True)
if not return_code == 0:
logger.error("Daemon \"{}\" is not running".format(daemon))
return False
# return True if all checks were passed
return True
# restart daemon
def _daemon_restart(daemon):
command = "sudo {} restart {}".format(watchfrr, daemon)
return_code = subprocess.call(command, shell=True)
if not return_code == 0:
logger.error("Failed to restart daemon \"{}\"".format(daemon))
return False
# return True if restarted sucessfully
logger.info("Daemon \"{}\" restarted".format(daemon))
return True
# check all daemons if they are running
def _check_args_daemon(daemons):
for daemon in daemons:
if not _daemon_check(daemon):
return False
return True
# define program arguments
cmd_args_parser = argparse.ArgumentParser(description='restart frr daemons')
cmd_args_parser.add_argument('--action', choices=['restart'], required=True, help='action to frr daemons')
cmd_args_parser.add_argument('--daemon', choices=['bfdd', 'bgpd', 'ospfd', 'ospf6d', 'ripd', 'ripngd', 'staticd', 'zebra'], required=False, nargs='*', help='select single or multiple daemons')
# parse arguments
cmd_args = cmd_args_parser.parse_args()
# main logic
# restart daemon
if cmd_args.action == 'restart':
if not _save_and_restore('save'):
logger.error("Failed to rename permanent comfig")
print("Failed to rename permanent comfig")
sys.exit(1)
if not _write_config():
print("Failed to save active config")
_save_and_restore('restore')
sys.exit(1)
if cmd_args.daemon:
# check all daemons if they are running
if not _check_args_daemon(cmd_args.daemon):
print("Warning: some of listed daemons are not running")
# run command to restart daemon
for daemon in cmd_args.daemon:
if not _daemon_restart(daemon):
print("Failed to restart daemon: {}".format(daemon))
_save_and_restore('restore')
sys.exit(1)
else:
# run command to restart FRR
if not _daemon_restart(''):
print("Failed to restart FRRouting")
_save_and_restore('restore')
sys.exit(1)
_save_and_restore('restore')
sys.exit(0)

File Metadata

Mime Type
text/x-script.python
Expires
Sat, Oct 17, 12:46 AM (9 h, 47 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
4352079
Default Alt Text
restart_frr.py (4 KB)

Event Timeline