From 6940fd3b6d15c7c4adf6b266bcaca7c668600abc Mon Sep 17 00:00:00 2001
From: Pavel Shirshov <pavelsh@microsoft.com>
Date: Wed, 21 Feb 2018 13:52:09 -0800
Subject: [PATCH] Refactor fast-reboot script. Generate fast-reboot-dumps into
 configurable directory

---
 scripts/fast-reboot         | 27 +++++++++++----------------
 scripts/fast-reboot-dump.py | 18 +++++++++---------
 2 files changed, 20 insertions(+), 25 deletions(-)

diff --git a/scripts/fast-reboot b/scripts/fast-reboot
index ba42a11acd..1873dfe39f 100755
--- a/scripts/fast-reboot
+++ b/scripts/fast-reboot
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # Check root privileges
-if [ "$EUID" -ne 0 ]
+if [[ "$EUID" -ne 0 ]]
 then
   echo "Please run as root"
   exit
@@ -9,7 +9,7 @@ fi
 
 
 # Unload the previously loaded kernel if any loaded
-if [ "$(cat /sys/kernel/kexec_loaded)" -eq 1 ]
+if [[ "$(cat /sys/kernel/kexec_loaded)" -eq 1 ]]
 then
   /sbin/kexec -u
 fi
@@ -27,33 +27,28 @@ sonic_asic_type=$(sonic-cfggen -y /etc/sonic/sonic_version.yml -v asic_type)
 /sbin/kexec -l "$KERNEL_IMAGE" --initrd="$INITRD" --append="$BOOT_OPTIONS"
 
 # Dump the ARP and FDB tables to files also as default routes for both IPv4 and IPv6
-/usr/bin/fast-reboot-dump.py
-docker cp /tmp/fdb.json swss:/
-docker cp /tmp/arp.json swss:/
-if [ -e /tmp/default_routes.json ]
-then
-  docker cp /tmp/default_routes.json swss:/
-fi
+# into /host/fast-reboot
+mkdir -p /host/fast-reboot
+/usr/bin/fast-reboot-dump.py /host/fast-reboot
 
-# Kill bgpd to enable graceful restart of BGP
-docker exec -ti bgp killall -9 watchquagga
+# Kill bgpd to start the bgp graceful restart procedure
 docker exec -ti bgp killall -9 zebra
 docker exec -ti bgp killall -9 bgpd
 
 # Kill lldp, otherwise it sends informotion about reboot
-docker kill lldp
+docker kill lldp > /dev/null
 
 # Kill teamd, otherwise it gets down all LAGs
-docker kill teamd
+docker kill teamd > /dev/null
 
-# Kill other containers to make reboot faster
-docker ps -qa | xargs docker kill
+# Kill other containers to make the reboot faster
+docker ps -q | xargs docker kill > /dev/null
 
 # Stop the docker container engine. Otherwise we will have a broken docker storage
 systemctl stop docker.service
 
 # Stop opennsl modules for Broadcom platform
-if [ "$sonic_asic_type" = 'broadcom' ];
+if [[ "$sonic_asic_type" = 'broadcom' ]];
 then
   service_name=$(systemctl list-units --plain --no-pager --no-legend --type=service | grep opennsl | cut -f 1 -d' ')
   systemctl stop "$service_name"
diff --git a/scripts/fast-reboot-dump.py b/scripts/fast-reboot-dump.py
index e112b3f52e..f5a32ebc16 100644
--- a/scripts/fast-reboot-dump.py
+++ b/scripts/fast-reboot-dump.py
@@ -4,6 +4,7 @@
 import json
 import socket
 import struct
+import sys
 import os
 from fcntl import ioctl
 import binascii
@@ -244,18 +245,17 @@ def generate_default_route_entries(filename):
 
     db.close(db.APPL_DB)
 
-    if len(default_routes_output) > 0:
-        with open(filename, 'w') as fp:
-            json.dump(default_routes_output, fp, indent=2, separators=(',', ': '))
-    else:
-        if os.path.isfile(filename):
-            os.unlink(filename)
+    with open(filename, 'w') as fp:
+        json.dump(default_routes_output, fp, indent=2, separators=(',', ': '))
 
 
 def main():
-    all_available_macs, map_mac_ip_per_vlan = generate_fdb_entries('/tmp/fdb.json')
-    arp_entries = generate_arp_entries('/tmp/arp.json', all_available_macs)
-    generate_default_route_entries('/tmp/default_routes.json')
+    root_dir = '/tmp'
+    if len(sys.argv) > 1 and os.path.isdir(sys.argv[1]):
+        root_dir = sys.argv[1]
+    all_available_macs, map_mac_ip_per_vlan = generate_fdb_entries(root_dir + '/fdb.json')
+    arp_entries = generate_arp_entries(root_dir + '/arp.json', all_available_macs)
+    generate_default_route_entries(root_dir + '/default_routes.json')
     garp_send(arp_entries, map_mac_ip_per_vlan)
 
     return