summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorChristian Poessinger <christian@poessinger.com>2019-12-30 23:49:45 +0100
committerGitHub <noreply@github.com>2019-12-30 23:49:45 +0100
commit68e9026dc33f53ef105aa2f20b777d126a0e3072 (patch)
tree65f8e865c0d9f424efb6ebd56c0b93f13ce1c5e3 /src
parent3c233a8952b069892298662c4c757b5a6ffe3561 (diff)
parent452a9651f215205b34268c723beec5b2f964ad09 (diff)
downloadvyos-1x-68e9026dc33f53ef105aa2f20b777d126a0e3072.tar.gz
vyos-1x-68e9026dc33f53ef105aa2f20b777d126a0e3072.zip
Merge pull request #195 from zdc/T1514
FRRouting: T1514: Extended FRR restarting functionality and fixed some bugs
Diffstat (limited to 'src')
-rwxr-xr-xsrc/op_mode/restart_frr.py135
1 files changed, 98 insertions, 37 deletions
diff --git a/src/op_mode/restart_frr.py b/src/op_mode/restart_frr.py
index 085b8c355..da6407e23 100755
--- a/src/op_mode/restart_frr.py
+++ b/src/op_mode/restart_frr.py
@@ -20,12 +20,13 @@ import argparse
import subprocess
import logging
from logging.handlers import SysLogHandler
+from pathlib import Path
+import psutil
# 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'
+frrconfig_tmp = '/tmp/frr_restart'
# configure logging
logger = logging.getLogger(__name__)
@@ -34,36 +35,79 @@ 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)
+# check if it is safe to restart FRR
+def _check_safety():
+ try:
+ # print warning
+ answer = input("WARNING: This is a potentially unsafe function! You may lose the connection to the router or active configuration after running this command. Use it at your own risk! Continue? [y/N]: ")
+ if not answer.lower() == "y":
+ logger.error("User aborted command")
+ return False
- 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))
+ # check if another restart process already running
+ if len([process for process in psutil.process_iter(attrs=['pid', 'name', 'cmdline']) if 'python' in process.info['name'] and 'restart_frr.py' in process.info['cmdline'][1]]) > 1:
+ logger.error("Another restart_frr.py already running")
+ answer = input("Another restart_frr.py process is already running. It is unsafe to continue. Do you want to process anyway? [y/N]: ")
+ if not answer.lower() == "y":
+ return False
+
+ # check if watchfrr.sh is running
+ for process in psutil.process_iter(attrs=['pid', 'name', 'cmdline']):
+ if 'bash' in process.info['name'] and watchfrr in process.info['cmdline']:
+ logger.error("Another {} already running".format(watchfrr))
+ answer = input("Another {} process is already running. It is unsafe to continue. Do you want to process anyway? [y/N]: ".format(watchfrr))
+ if not answer.lower() == "y":
+ return False
+
+ # check if vtysh is running
+ for process in psutil.process_iter(attrs=['pid', 'name', 'cmdline']):
+ if 'vtysh' in process.info['name']:
+ logger.error("The vtysh is running by another task")
+ answer = input("The vtysh is running by another task. It is unsafe to continue. Do you want to process anyway? [y/N]: ")
+ if not answer.lower() == "y":
+ return False
+
+ # check if temporary directory exists
+ if Path(frrconfig_tmp).exists():
+ logger.error("The temporary directory \"{}\" already exists".format(frrconfig_tmp))
+ answer = input("The temporary directory \"{}\" already exists. It is unsafe to continue. Do you want to process anyway? [y/N]: ".format(frrconfig_tmp))
+ if not answer.lower() == "y":
+ return False
+ except:
+ logger.error("Something goes wrong in _check_safety()")
return False
- logger.info(logmsg)
+ # return True if all check was passed or user confirmed to ignore they results
return True
# write active config to file
def _write_config():
- command = "sudo {} -n -c write ".format(vtysh)
+ # create temporary directory
+ Path(frrconfig_tmp).mkdir(parents=False, exist_ok=True)
+ # save frr.conf to it
+ command = "{} -n -w --config_dir {} 2> /dev/null".format(vtysh, frrconfig_tmp)
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))
+ logger.info("Active config saved to {}".format(frrconfig_tmp))
return True
+# clear and remove temporary directory
+def _cleanup():
+ tmpdir = Path(frrconfig_tmp)
+ try:
+ if tmpdir.exists():
+ for file in tmpdir.iterdir():
+ file.unlink()
+ tmpdir.rmdir()
+ except:
+ logger.error("Failed to remove temporary directory {}".format(frrconfig_tmp))
+ print("Failed to remove temporary directory {}".format(frrconfig_tmp))
+
# check if daemon is running
def _daemon_check(daemon):
- command = "sudo {} print_status {}".format(watchfrr, daemon)
+ command = "{} 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))
@@ -74,16 +118,32 @@ def _daemon_check(daemon):
# restart daemon
def _daemon_restart(daemon):
- command = "sudo {} restart {}".format(watchfrr, daemon)
+ command = "{} 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
+ # return True if restarted successfully
logger.info("Daemon \"{}\" restarted".format(daemon))
return True
+# reload old config
+def _reload_config(daemon):
+ if daemon != '':
+ command = "{} -n -b --config_dir {} -d {} 2> /dev/null".format(vtysh, frrconfig_tmp, daemon)
+ else:
+ command = "{} -n -b --config_dir {} 2> /dev/null".format(vtysh, frrconfig_tmp)
+
+ return_code = subprocess.call(command, shell=True)
+ if not return_code == 0:
+ logger.error("Failed to reinstall configuration")
+ return False
+
+ # return True if restarted successfully
+ logger.info("Configuration reinstalled successfully")
+ return True
+
# check all daemons if they are running
def _check_args_daemon(daemons):
for daemon in daemons:
@@ -102,34 +162,35 @@ 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")
+ # check if it is safe to restart FRR
+ if not _check_safety():
+ print("\nOne of the safety checks was failed or user aborted command. Exiting.")
sys.exit(1)
if not _write_config():
print("Failed to save active config")
- _save_and_restore('restore')
+ _cleanup()
sys.exit(1)
- if cmd_args.daemon:
- # check all daemons if they are running
+ # a little trick to make further commands more clear
+ if not cmd_args.daemon:
+ cmd_args.daemon = ['']
+
+ # check all daemons if they are running
+ if cmd_args.daemon != ['']:
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')
+ # run command to restart daemon
+ for daemon in cmd_args.daemon:
+ if not _daemon_restart(daemon):
+ print("Failed to restart daemon: {}".format(daemon))
+ _cleanup()
sys.exit(1)
+ # reinstall old configuration
+ _reload_config(daemon)
- _save_and_restore('restore')
+ # cleanup after all actions
+ _cleanup()
sys.exit(0)