90 lines
2.9 KiB
Plaintext
90 lines
2.9 KiB
Plaintext
|
#!/usr/bin/env python
|
||
|
|
#
|
||
|
|
# process-reboot-cause
|
||
|
|
#
|
||
|
|
# Program designed to run once, soon after system boot which will
|
||
|
|
# determine the cause of the previous reboot and store it to the disk,
|
||
|
|
#
|
||
|
|
|
||
|
|
try:
|
||
|
|
import os
|
||
|
|
import pwd
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import syslog
|
||
|
|
import sonic_platform
|
||
|
|
from sonic_daemon_base.daemon_base import Logger
|
||
|
|
except ImportError as err:
|
||
|
|
raise ImportError("%s - required module not found" % str(err))
|
||
|
|
|
||
|
|
VERSION = "1.0"
|
||
|
|
|
||
|
|
SYSLOG_IDENTIFIER = "process-reboot-cause"
|
||
|
|
|
||
|
|
# Global logger instance
|
||
|
|
logger = Logger(SYSLOG_IDENTIFIER)
|
||
|
|
|
||
|
|
REBOOT_CAUSE_DIR = "/host/reboot-cause/"
|
||
|
|
REBOOT_CAUSE_FILE = REBOOT_CAUSE_DIR + "reboot-cause.txt"
|
||
|
|
PREVIOUS_REBOOT_CAUSE_FILE = REBOOT_CAUSE_DIR + "previous-reboot-cause.txt"
|
||
|
|
|
||
|
|
UNKNOWN_REBOOT_CAUSE = "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
logger.log_info("Starting up...")
|
||
|
|
|
||
|
|
if not os.geteuid() == 0:
|
||
|
|
logger.log_error("User {} does not have permission to execute".format(pwd.getpwuid(os.getuid()).pw_name))
|
||
|
|
sys.exit("This utility must be run as root")
|
||
|
|
|
||
|
|
# Create REBOOT_CAUSE_DIR if it doesn't exist
|
||
|
|
if not os.path.exists(REBOOT_CAUSE_DIR):
|
||
|
|
os.makedirs(REBOOT_CAUSE_DIR)
|
||
|
|
|
||
|
|
# Remove stale PREVIOUS_REBOOT_CAUSE_FILE if it exists
|
||
|
|
if os.path.exists(PREVIOUS_REBOOT_CAUSE_FILE):
|
||
|
|
os.remove(PREVIOUS_REBOOT_CAUSE_FILE)
|
||
|
|
|
||
|
|
# Check if the previous reboot was caused by hardware
|
||
|
|
platform = sonic_platform.platform.Platform()
|
||
|
|
|
||
|
|
chassis = platform.get_chassis()
|
||
|
|
|
||
|
|
hardware_reboot_cause, optional_details = chassis.get_reboot_cause()
|
||
|
|
|
||
|
|
# Set a default previous reboot cause
|
||
|
|
previous_reboot_cause = UNKNOWN_REBOOT_CAUSE
|
||
|
|
|
||
|
|
if hardware_reboot_cause == chassis.REBOOT_CAUSE_NON_HARDWARE:
|
||
|
|
# The reboot was not caused by hardware. If there is a REBOOT_CAUSE_FILE, it will
|
||
|
|
# contain any software-related reboot info. We will use it as the previous cause.
|
||
|
|
if os.path.isfile(REBOOT_CAUSE_FILE):
|
||
|
|
cause_file = open(REBOOT_CAUSE_FILE, "r")
|
||
|
|
previous_reboot_cause = cause_file.readline().rstrip('\n')
|
||
|
|
cause_file.close()
|
||
|
|
elif hardware_reboot_cause == chassis.REBOOT_CAUSE_HARDWARE_OTHER:
|
||
|
|
previous_reboot_cause = "{} ({})".format(hardware_reboot_cause, optional_details)
|
||
|
|
else:
|
||
|
|
previous_reboot_cause = hardware_reboot_cause
|
||
|
|
|
||
|
|
# Write the previous reboot cause to PREVIOUS_REBOOT_CAUSE_FILE
|
||
|
|
prev_cause_file = open(PREVIOUS_REBOOT_CAUSE_FILE, "w")
|
||
|
|
prev_cause_file.write(previous_reboot_cause)
|
||
|
|
prev_cause_file.close()
|
||
|
|
|
||
|
|
# Also log the previous reboot cause to the syslog
|
||
|
|
logger.log_info("Previous reboot cause: {}".format(previous_reboot_cause))
|
||
|
|
|
||
|
|
# Remove the old REBOOT_CAUSE_FILE
|
||
|
|
os.remove(REBOOT_CAUSE_FILE)
|
||
|
|
|
||
|
|
# Write a new default reboot cause file for the next reboot
|
||
|
|
cause_file = open(REBOOT_CAUSE_FILE, "w")
|
||
|
|
cause_file.write(UNKNOWN_REBOOT_CAUSE)
|
||
|
|
cause_file.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|