355 lines
10 KiB
Bash
Executable File
355 lines
10 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Install motd scripts
|
|
# Bryan C. Roessler
|
|
|
|
parent="${BASH_SOURCE[0]}"
|
|
parent=${parent%/*}
|
|
|
|
[[ -f "$parent"/script-functions ]] && . "$parent"/script-functions || exit 1
|
|
|
|
is_root
|
|
|
|
DEBUG=0
|
|
[[ "$#" -gt 0 ]] && echo "Debug on" && DEBUG=1
|
|
|
|
installdir="/usr/local/bin"
|
|
|
|
[[ -d "$installdir" ]] || mkdir -p "$installdir"
|
|
|
|
if (( DEBUG )); then
|
|
script="generate-motd.sh"
|
|
else
|
|
script="$installdir/generate-motd.sh"
|
|
fi
|
|
|
|
cat <<- 'EOF' > "$script"
|
|
#!/usr/bin/env bash
|
|
|
|
echo -n '
|
|
_ _ _ _ _
|
|
| | | | | | | | | |
|
|
| |__| | __ _ _ __| |_ _ __ ___ __ _ _ __ | | __ _| |__
|
|
| __ |/ _` | `__| __| `_ ` _ \ / _` | `_ \ | | / _` | `_ \
|
|
| | | | (_| | | | |_| | | | | | (_| | | | | | |___| (_| | |_) |
|
|
|_| |_|\__,_|_| \__|_| |_| |_|\__,_|_| |_| |______\__,_|_.__/
|
|
'
|
|
EOF
|
|
|
|
# System info
|
|
cat <<- 'EOF' >> "$script"
|
|
# get load averages
|
|
IFS=" " read LOAD1 LOAD5 LOAD15 <<<$(cat /proc/loadavg | awk '{ print $1,$2,$3 }')
|
|
# get free memory
|
|
IFS=" " read USED AVAIL TOTAL <<<$(free -htm | grep "Mem" | awk {'print $3,$7,$2'})
|
|
# get processes
|
|
PROCESS=`ps -eo user=|sort|uniq -c | awk '{ print $2 " " $1 }'`
|
|
PROCESS_ALL=`echo "$PROCESS"| awk {'print $2'} | awk '{ SUM += $1} END { print SUM }'`
|
|
PROCESS_ROOT=`echo "$PROCESS"| grep root | awk {'print $2'}`
|
|
PROCESS_USER=`echo "$PROCESS"| grep -v root | awk {'print $2'} | awk '{ SUM += $1} END { print SUM }'`
|
|
# get processors
|
|
PROCESSOR_NAME=`grep "model name" /proc/cpuinfo | cut -d ' ' -f3- | awk {'print $0'} | head -1`
|
|
PROCESSOR_COUNT=`grep -ioP 'processor\t:' /proc/cpuinfo | wc -l`
|
|
|
|
W="\e[0;39m"
|
|
G="\e[1;32m"
|
|
|
|
echo -e "
|
|
$W Distro......: $W`cat /etc/*release | grep "PRETTY_NAME" | cut -d "=" -f 2- | sed 's/"//g'`
|
|
$W Kernel......: $W`uname -sr`
|
|
$W Uptime......: $W`uptime -p`
|
|
$W Load........: $G$LOAD1$W (1m), $G$LOAD5$W (5m), $G$LOAD15$W (15m)
|
|
$W Processes...: $W$G$PROCESS_ROOT$W (root), $G$PROCESS_USER$W (user), $G$PROCESS_ALL$W (total)
|
|
$W CPU.........: $W$PROCESSOR_NAME ($G$PROCESSOR_COUNT$W vCPU)
|
|
$W Memory......: $G$USED$W used, $G$AVAIL$W avail, $G$TOTAL$W total$W"
|
|
EOF
|
|
|
|
# Disk usage
|
|
cat <<- 'EOF' >> "$script"
|
|
# config
|
|
max_usage=90
|
|
bar_width=50
|
|
# colors
|
|
white="\e[39m"
|
|
green="\e[1;32m"
|
|
red="\e[1;31m"
|
|
dim="\e[2m"
|
|
undim="\e[0m"
|
|
|
|
# disk usage: ignore zfs, squashfs & tmpfs
|
|
while IFS= read -r line; do dfs+=("$line"); done < <(df -H -x zfs -x squashfs -x tmpfs -x devtmpfs -x overlay --output=target,pcent,size | tail -n+2)
|
|
printf "\nDisk usage\n"
|
|
|
|
for line in "${dfs[@]}"; do
|
|
# get disk usage
|
|
usage=$(echo "$line" | awk '{print $2}' | sed 's/%//')
|
|
used_width=$((($usage*$bar_width)/100))
|
|
# color is green if usage < max_usage, else red
|
|
if [ "${usage}" -ge "${max_usage}" ]; then
|
|
color=$red
|
|
else
|
|
color=$green
|
|
fi
|
|
# print green/red bar until used_width
|
|
bar="[${color}"
|
|
for ((i=0; i<$used_width; i++)); do
|
|
bar+="="
|
|
done
|
|
# print dimmmed bar until end
|
|
bar+="${white}${dim}"
|
|
for ((i=$used_width; i<$bar_width; i++)); do
|
|
bar+="="
|
|
done
|
|
bar+="${undim}]"
|
|
# print usage line & bar
|
|
echo "${line}" | awk '{ printf("%-31s%+3s used out of %+4s\n", $1, $2, $3); }' | sed -e 's/^/ /'
|
|
echo -e "${bar}" | sed -e 's/^/ /'
|
|
done
|
|
EOF
|
|
|
|
# # Disk health
|
|
# cat <<- 'EOF' >> "$script"
|
|
# # config
|
|
# MAX_TEMP=40
|
|
# # set column width
|
|
# COLUMNS=2
|
|
# # colors
|
|
# white="\e[39m"
|
|
# green="\e[1;32m"
|
|
# red="\e[1;31m"
|
|
# dim="\e[2m"
|
|
# undim="\e[0m"
|
|
|
|
# # disks to check
|
|
# disks=(sda sdb sdc sdd sde sdf sdg sdi)
|
|
# disknames=(sda sdb sdc sdd sde sdf sdg sdi)
|
|
|
|
# # hddtemp
|
|
# hddtemp_host=localhost
|
|
# hddtemp_port=7634
|
|
|
|
# # logfiles to check
|
|
# logfiles='/var/log/syslog /var/log/syslog.1'
|
|
|
|
# # get all lines with smartd entries from syslog
|
|
# lines=$(tac $logfiles | grep -hiP 'smartd\[[[:digit:]]+\]:' | grep -iP "previous self-test")
|
|
# # use nc to query temps from hddtemp daemon
|
|
# hddtemp=$(timeout 0.01 nc $hddtemp_host $hddtemp_port | sed 's/|//m' | sed 's/||/ \n/g')
|
|
|
|
# out=""
|
|
# for i in "${!disks[@]}"; do
|
|
# disk=${disks[$i]}
|
|
# # use disknames if given
|
|
# diskname=${disknames[$i]}
|
|
# if [ -z "${diskname}" ]; then
|
|
# diskname=$disk
|
|
# fi
|
|
# uuid=$(blkid -s UUID -o value "/dev/${disk}")
|
|
# status=$( (grep "${uuid}" <<< "${lines}") | grep -m 1 -oP "previous self-test.*" | awk '{ print $4 " " $5 }')
|
|
# temp=$( (grep "${disk}" <<< "${hddtemp}") | awk -F'|' '{ print $3 }')
|
|
|
|
# # color green if temp <= MAX_TEMP, else red
|
|
# if [[ "${temp}" -gt "${MAX_TEMP}" ]]; then
|
|
# color=$red
|
|
# else
|
|
# color=$green
|
|
# fi
|
|
# # add "C" if temp is numeric
|
|
# if [[ "$temp" =~ ^[0-9]+$ ]]; then
|
|
# temp="${temp}C"
|
|
# fi
|
|
# # color green if status is "without error", else red
|
|
# if [[ "${status}" == "without error" ]]; then
|
|
# status_color=$green
|
|
# else
|
|
# status_color=$red
|
|
# fi
|
|
|
|
# # print temp & smartd error
|
|
# out+="${diskname}:,${color}${temp}${undim} | ${status_color}${status}${undim},"
|
|
# # insert \n every $COLUMNS column
|
|
# if [ $((($i+1) % $COLUMNS)) -eq 0 ]; then
|
|
# out+="\n"
|
|
# fi
|
|
# done
|
|
# out+="\n"
|
|
|
|
# printf "\ndisk status:\n"
|
|
# printf "$out" | column -ts $',' | sed -e 's/^/ /'
|
|
# EOF
|
|
|
|
|
|
# Services
|
|
cat <<- 'EOF' >> "$script"
|
|
# set column width
|
|
COLUMNS=2
|
|
# colors
|
|
green="\e[1;32m"
|
|
red="\e[1;31m"
|
|
undim="\e[0m"
|
|
|
|
services=("fail2ban" "firewalld" "nmb" "motion" "btrfs-balance.timer" "btrfs-scrub.timer" "smb" "backup.timer" "btrbk.timer" "fstrim.timer" "dnf-automatic.timer" "smartd" "cockpit.socket" "generate-motd.timer")
|
|
# sort services
|
|
IFS=$'\n' services=($(sort <<<"${services[*]}"))
|
|
unset IFS
|
|
|
|
service_status=()
|
|
# get status of all services
|
|
for service in "${services[@]}"; do
|
|
service_status+=($(systemctl is-active "$service"))
|
|
done
|
|
|
|
out=""
|
|
for i in ${!services[@]}; do
|
|
# color green if service is active, else red
|
|
if [[ "${service_status[$i]}" == "active" ]]; then
|
|
out+="${services[$i]}:,${green}${service_status[$i]}${undim},"
|
|
else
|
|
out+="${services[$i]}:,${red}${service_status[$i]}${undim},"
|
|
fi
|
|
# insert \n every $COLUMNS column
|
|
if [ $((($i+1) % $COLUMNS)) -eq 0 ]; then
|
|
out+="\n"
|
|
fi
|
|
done
|
|
out+="\n"
|
|
|
|
printf "\nServices\n"
|
|
printf "$out" | column -ts $',' | sed -e 's/^/ /'
|
|
EOF
|
|
|
|
|
|
# Fail2Ban
|
|
cat <<- 'EOF' >> "$script"
|
|
# fail2ban-client status to get all jails, takes about ~70ms
|
|
jails=($(fail2ban-client status | grep "Jail list:" | sed "s/ //g" | awk '{split($2,a,",");for(i in a) print a[i]}'))
|
|
|
|
out="jail,failed,total,banned,total\n"
|
|
|
|
for jail in ${jails[@]}; do
|
|
# slow because fail2ban-client has to be called for every jail (~70ms per jail)
|
|
status=$(fail2ban-client status ${jail})
|
|
failed=$(echo "$status" | grep -ioP '(?<=Currently failed:\t)[[:digit:]]+')
|
|
totalfailed=$(echo "$status" | grep -ioP '(?<=Total failed:\t)[[:digit:]]+')
|
|
banned=$(echo "$status" | grep -ioP '(?<=Currently banned:\t)[[:digit:]]+')
|
|
totalbanned=$(echo "$status" | grep -ioP '(?<=Total banned:\t)[[:digit:]]+')
|
|
out+="$jail,$failed,$totalfailed,$banned,$totalbanned\n"
|
|
done
|
|
|
|
printf "\nfail2ban status\n"
|
|
printf $out | column -ts $',' | sed -e 's/^/ /'
|
|
EOF
|
|
|
|
# Help links
|
|
cat <<- 'EOF' >> "$script"
|
|
echo "\
|
|
|
|
Links (ctrl+click to follow)
|
|
Server Manual.........: https://tinyurl.com/jjz9h6fr
|
|
Cockpit (for admins)..: http://localhost:9090
|
|
Robot Camera..........: http://localhost:9999
|
|
JupyterLab............: http://localhost:8888
|
|
RStudio Server........: http://localhost:8787
|
|
Robot Computer........: vnc://192.168.16.101:5900
|
|
Windows 10 VM.........: vnc://localhost:5900 (pw: hartman)
|
|
"
|
|
EOF
|
|
|
|
# Scheduled reboot
|
|
cat <<- "EOF" >> "$script"
|
|
if systemctl is-active scheduled-reboot.timer &>/dev/null; then
|
|
echo -n "Next scheduled reboot: "
|
|
time=$(systemctl cat scheduled-reboot.timer | grep OnCalendar=)
|
|
time=${time#*=}
|
|
echo "$time"
|
|
fi
|
|
EOF
|
|
|
|
# cat <<- 'EOF' > "$script"
|
|
# #!/usr/bin/env bash
|
|
|
|
# [[ -v NO_MOTD ]] && exit 0
|
|
|
|
# USER=$(whoami)
|
|
# HOSTNAME=$(uname -n)
|
|
# ARRAY=$(df -Ph | grep array | awk '{print $4}' | tr -d '\n')
|
|
# ROOT=$(df -Ph | grep sdb3 | awk '{print $4}' | tr -d '\n')
|
|
# BACKUP=$(df -Ph | grep backup | awk '{print $4}' | tr -d '\n')
|
|
|
|
# MEMUSED=$(free -t -m | grep "Mem:" | awk '{print $3" MB";}')
|
|
# MEMTOTAL=$(free -t -m | grep "Mem:" | awk '{print $2" MB";}')
|
|
# PSA=$(ps -Afl | wc -l)
|
|
|
|
# SWAPMEM=$(free -m | tail -n 1 | awk '{print $3}')
|
|
|
|
# #System uptime
|
|
# uptime=$(cut -f1 -d. /proc/uptime)
|
|
# upDays=$((uptime/60/60/24))
|
|
# upHours=$((uptime/60/60%24))
|
|
# upMins=$((uptime/60%60))
|
|
# upSecs=$((uptime%60))
|
|
|
|
# #System load
|
|
# LOAD1=$(awk {'print $1'} /proc/loadavg)
|
|
# LOAD5=$(awk {'print $2'} /proc/loadavg)
|
|
# LOAD15=$(awk {'print $3'} /proc/loadavg)
|
|
|
|
|
|
|
|
# echo "
|
|
# ===========================================================================
|
|
# - Hostname............: $HOSTNAME
|
|
# - Release.............: $(cat /etc/redhat-release)
|
|
# - Users...............: Currently $(users | wc -w) user(s) logged on
|
|
# ===========================================================================
|
|
# - Current user........: $USER
|
|
# - CPU usage...........: $LOAD1, $LOAD5, $LOAD15 (1, 5, 15 min)
|
|
# - Memory used.........: $MEMUSED / $MEMTOTAL
|
|
# - Processes...........: $PSA running
|
|
# - System uptime.......: $upDays days $upHours hours $upMins minutes $upSecs seconds
|
|
# - Disk space HOME|ROOT: $ROOT remaining
|
|
# - Disk space ARRAY....: $ARRAY remaining
|
|
# - Disk space BACKUP...: $BACKUP remaining
|
|
# ===========================================================================
|
|
# "
|
|
# EOF
|
|
|
|
generate-services() {
|
|
service="/usr/lib/systemd/system/generate-motd.service"
|
|
timer="/usr/lib/systemd/system/generate-motd.timer"
|
|
cat <<-EOF > "$service"
|
|
[Unit]
|
|
Description=Generate MoTD
|
|
|
|
[Service]
|
|
Type=simple
|
|
ExecStart=/usr/bin/bash -c '$script > /etc/motd'
|
|
|
|
[Install]
|
|
WantedBy=default.target
|
|
EOF
|
|
cat <<-'EOF' > "$timer"
|
|
[Unit]
|
|
Description=Generate MoTD every minute on a timer
|
|
|
|
[Timer]
|
|
OnCalendar=*:0/1
|
|
OnBootSec=10s
|
|
|
|
[Install]
|
|
WantedBy=timers.target
|
|
EOF
|
|
}
|
|
|
|
chmod +x "$script"
|
|
|
|
if (( DEBUG )); then
|
|
bash generate-motd.sh
|
|
cat -n generate-motd.sh
|
|
rm generate-motd.sh
|
|
else
|
|
generate-services && \
|
|
systemctl daemon-reload && \
|
|
systemctl enable --now generate-motd.timer
|
|
fi
|
|
|
|
exit $? |