#!/bin/bash stty eof undef stty intr undef hostsystem="$( cat /etc/bmchost )" bmcd_cmdpipe="/run/bmcd/bmcd.cmd" bmcd_statepipe="/run/bmcd/bmcd.state" help() { echo -e "Available commands:" echo -e " \e[1mstate\e[0m - Show the system power state" echo -e " \e[1mconsole\e[0m - Connect to host via serial console; ^A+D to disconnect" echo -e " \e[1mpower\e[0m - Press power switch on host" echo -e " \e[1mreset\e[0m - Press reset switch on host" echo -e " \e[1mkill\e[0m - Forcibly power off host" echo -e " \e[1mlocate\e[0m - Enable locator (flash power LED)" echo -e " \e[1munlocate\e[0m - Disable locator" echo -e " \e[1mhelp\e[0m - This help menu" echo -e " \e[1mbmc\e[0m - Show BMC information" echo -e " \e[1mhostname\e[0m - Set BMC hostname" echo -e " \e[1mpassword\e[0m - Set BMC password" echo -e " \e[1mshell\e[0m - Start a BMC bash shell" echo -e " \e[1mexit/logout\e[0m - Disconnect from the BMC" } bmcinfo() { echo -e "BMC information:" echo -e " IP address: $( ip -4 addr list eth0 | grep inet | awk '{ print $2 }' )" echo -e " Hostname: $( hostname )" echo -e " BMC temperature: $( sudo /opt/vc/bin/vcgencmd measure_temp | awk -F'=' '{ print $2 }' )" } sethostname() { echo -n "Enter new hostname: " read newhostname sudo sed -i '/^127.0.1.1/d' /etc/hosts &>/dev/null sudo tee -a /etc/hosts <<<"127.0.1.1 $newhostname" &>/dev/null sudo hostname $newhostname &>/dev/null sudo tee /etc/hostname <<<"$newhostname" &>/dev/null echo "Hostname set to $newhostname" } setpassword() { echo -n "Enter new BMC password: " read -s password_1 echo echo -n "Reenter new BMC password: " read -s password_2 echo if [ "${password_1}" == "${password_2}" ]; then echo -n "Setting BMC password... " sudo chpasswd <<<"bmc:${password_1}" echo "done." else echo "Passwords to not match!" fi } resetsw_press() { echo "Pressing reset switch." echo "resetsw_press" > ${bmcd_cmdpipe} } powersw_press() { echo "Pressing power switch." echo "powersw_press" > ${bmcd_cmdpipe} sleep 2 } powersw_hold() { echo "Holding power switch." echo "powersw_hold" > ${bmcd_cmdpipe} sleep 2 } locate_on() { echo "Enabling locator - host power LED will flash." echo "locate_on" > ${bmcd_cmdpipe} } locate_off() { echo "Disabling locator." echo "locate_off" > ${bmcd_cmdpipe} } readpower() { powerstate_raw=$(cat ${bmcd_statepipe}) if [[ "${powerstate_raw}" -eq 1 ]]; then powerstate="\e[32mOn\e[0m" else powerstate="\e[31mOff\e[0m" fi } readpower echo echo -e "--------------------" echo -e "| Raspberry Pi BMC |" echo -e "--------------------" echo echo -e "Host system: \e[1m${hostsystem}\e[0m" echo -e "Host state: $powerstate" echo help echo while true; do stty eof undef stty intr undef echo -en "\e[1m\e[34m[$(hostname)]>\e[0m " read input case $input in 'state') readpower echo -e "Host state: $powerstate" echo ;; 'console') echo "Starting console..." # Connect to screen, or start it sudo screen -r serialconsole &>/dev/null || sudo screen -S serialconsole /dev/ttyUSB0 115200 # If the user killed screen, restart it - just in case pgrep screen &>/dev/null || sudo screen -S serialconsole /dev/ttyUSB0 115200 echo ;; 'power') powersw_press readpower echo -e "Host state: $powerstate" echo ;; 'reset') resetsw_press readpower echo -e "Host state: $powerstate" echo ;; 'kill') powersw_hold readpower echo -e "Host state: $powerstate" echo ;; 'locate') locate_on echo ;; 'unlocate') locate_off echo ;; 'help') help echo ;; 'bmc') bmcinfo echo ;; 'hostname') sethostname echo ;; 'password') setpassword echo ;; 'shell') stty sane /bin/bash help echo ;; 'exit'|'logout') exit 0 ;; '') continue ;; *) echo "Invalid command." echo ;; esac done