setup-armbian-monitoring-environment

#!/bin/bash
#
# setup-armbian-monitoring-environment
#
# This script should be called from armhwinfo (relying on the
# following exported variables -- if not called from armhwinfo
# these should be defined/exported the same way armhwinfo does):
#
# export HARDWARE DISTROID DISTROCODE ARCH KERNELID MACHINE ID
#
# The purpose is to create an environment for rpimonitord so that
# relevant hardware informations can be monitored easily. This
# script starts uses configuration files in /etc/armbianmonitor/,
# collects data sources below /etc/armbianmonitor/datasources/ and
# adjusts templates for RPi-Monitor on the fly. Only if the file
# /etc/armbianmonitor/start-monitoring exists this script relinks
# /etc/rpimonitor/data.conf and starts rpimonitord if not already
# running.
#
# In case the script detects that not all necessary data sources
# are available through sysfs it will act as a daemon to collect
# data on its own and to write it to the approriate file. As an
# example the SoC's temperature: Running on an AXP209 based board
# the script will check /sys/class/thermal/thermal_zone0/temp and
# if existing link it to /etc/armbianmonitor/datasources/soctemp.
# In case it does not exist the script will create a normal file
# at this location and writes the thermal value to it using the
# sunxi_axp209_temp binary in an endless loop.
#
# At least the following files/symlinks will be provided below
# /etc/armbianmonitor/datasources/ depending on SoC/PMIC in question:
#
# soctemp (SoC's internal temp in degree Celsius * 1000)
# pmictemp (PMIC's internal temp in degree Celsius * 1000)
# ac_voltage (DC-IN voltage in V * 1000000)
# usb_voltage (USB OTG voltage in V * 1000000)
# battery_voltage (battery voltage in V * 1000000)
# ac_current (DC-IN current in A * 1000000)
# usb_current (USB OTG current in A * 1000000)
# battery_current (battery current in A * 1000000)
#
# When extended debugging has been chosen (using _armbian-monitoring_
# this can be configured. Then /etc/armbianmonitor/start-monitoring
# contains DEBUG) this script will also provide a few more files:
#
# cpustat (cpu_stat,system_stat,user_stat,nice_stat,iowait_stat,irq_stat
#          collected through /proc/cpustat in daemon mode)
# cpu_count (number of active CPU cores)
# vcorevoltage (Vcore in V * 1000 based on sysfs or script.bin/.dtb)
#
# Disk monitoring: For configured disks the following parameters can be
# monitored: temperature, S.M.A.R.T. health, load cycle count and CRC
# errors indicating connection/cable problems. The config file used is
# /etc/armbianmonitoring/disks.conf
#
# Filesystem monitoring: In /etc/armbianmonitoring/filesystems.conf
# mounpoints an trigger values can be defined that will be used to
# create the template stuff for these fs at startup of this script.
#
# The behaviour of this script can be configured through another tool
# called armbian-monitoring. The latter will also check for connected
# disks, get their name and GUID and let the user choose whether the
# disk should be monitored or not. The information has to be stored in
# /etc/armbianmonitor/disks.conf relying on the GUIDs of the disks.

 # define some variables:
CheckInterval=7.5      # time in seconds between two checks
DiskCheckInterval=60   # time in seconds between disk checks

Main() {
	PreRequisits

	case ${ID} in
		Cubieboard|Cubietruck|Orange|"Lamobo R1"|"Lime 2"|Lime|Banana|Micro)
			DealWithAXP209
			;;
		"Banana M2")
			DealWithNewBoard
			# DealWithAXP221
			;;
		"Cubietruck Plus"|"Banana Pi M3")
			DealWithNewBoard
			# DealWithAXP818
			;;
		Guitar|"Roseapple Pi")
			DealWithNewBoard
			# DealWithS500
			;;
		"Odroid XU4"|"Odroid XU3")
			DealWithNewBoard
			# DealWithExynos4
			;;
		"Odroid C1")
			DealWithNewBoard
			# DealWithS805
			;;
		Clearfog|"Turris Omnia")
			DealWithNewBoard
			# DealWithArmada38x
			;;
		"Cubox i4"|"HB i2eX"|"Cubox i2eX"|"HB i1"|"HB i2"|"Wandboard")
			DealWithNewBoard
			# DealWithiMX6
			;;
		"Orange Pi PC"|"Orange Pi Plus"|"Orange Pi 2"|"Orange Pi One"|"Orange Pi Lite")
			DealWithNewBoard
			# DealWithH3
			;;
		Geekbox)
			DealWithRK3368
			;;
		*)
			# No templates exist now. Combine sane defaults with some guessing
			DealWithNewBoard
			;;
	esac

	# Create the Armbian templates
	CreateTemplates

	exit 0

	# Decide depending on existence of /etc/armbianmonitor/start-monitoring
	ShouldMonitoringBeStarted

	# Provide missing data in daemon mode
	LoopEndlessly
} # Main

DealWithAXP209() {
	# check existence of sysfs nodes
	if [ -f /sys/devices/virtual/thermal/thermal_zone0/temp ]; then
		ln -fs /sys/devices/virtual/thermal/thermal_zone0/temp /etc/armbianmonitor/datasources/soctemp
	else
		if [ -L /etc/armbianmonitor/datasources/soctemp ]; then
			rm -f /etc/armbianmonitor/datasources/soctemp
		fi
		echo -n 25000 >/etc/armbianmonitor/datasources/soctemp
		export GetSoCTemp="${ARCH}-${ID}"
	fi
	if [ -d /sys/devices/platform/soc@01c00000/1c2ac00.i2c/i2c-0/0-0034 ]; then
		# mainline kernel and 'axp209 mainline sysfs interface' patch applied
		ln -fs /sys/power/axp_pmu/ac/voltage /etc/armbianmonitor/datasources/ac_voltage
		ln -fs /sys/power/axp_pmu/ac/amperage /etc/armbianmonitor/datasources/ac_current
		ln -fs /sys/power/axp_pmu/vbus/voltage /etc/armbianmonitor/datasources/usb_voltage
		ln -fs /sys/power/axp_pmu/vbus/amperage /etc/armbianmonitor/datasources/usb_current
		ln -fs /sys/power/axp_pmu/battery/voltage /etc/armbianmonitor/datasources/battery_voltage
		ln -fs /sys/power/axp_pmu/battery/amperage /etc/armbianmonitor/datasources/battery_current
		ln -fs /sys/power/axp_pmu/pmu/internal_temp /etc/armbianmonitor/datasources/pmictemp
	fi
	if [ -d /sys/devices/platform/sunxi-i2c.0/i2c-0/0-0034/axp20-supplyer.28/power_supply ]; then
		# sunxi 3.4 kernel
		SysFSPrefix=/sys/devices/platform/sunxi-i2c.0/i2c-0/0-0034/axp20-supplyer.28/power_supply
		ln -fs "${SysFSPrefix}"/ac/voltage_now /etc/armbianmonitor/datasources/ac_voltage
		ln -fs "${SysFSPrefix}"/ac/current_now /etc/armbianmonitor/datasources/ac_current
		ln -fs "${SysFSPrefix}"/usb/voltage_now /etc/armbianmonitor/datasources/usb_voltage
		ln -fs "${SysFSPrefix}"/usb/current_now /etc/armbianmonitor/datasources/usb_current
		ln -fs "${SysFSPrefix}"/battery/voltage_now /etc/armbianmonitor/datasources/battery_voltage
		ln -fs "${SysFSPrefix}"/battery/current_now /etc/armbianmonitor/datasources/battery_current
		ln -fs /sys/devices/platform/sunxi-i2c.0/i2c-0/0-0034/temp1_input /etc/armbianmonitor/datasources/pmictemp
	fi
	# relink template
	if [ "X${DebugMode}" = "XDEBUG" ]; then
		ln -sf /etc/armbianmonitor/templates/axp209_template_debug.conf /etc/armbianmonitor/cpu_pmic.conf
	else
		ln -sf /etc/armbianmonitor/templates/axp209_template.conf /etc/armbianmonitor/cpu_pmic.conf
	fi
} # DealWithAXP209

DealWithNewBoard() {
	# check existence of sysfs nodes
	if [ -f /sys/devices/virtual/thermal/thermal_zone0/temp ]; then
		ln -fs /sys/devices/virtual/thermal/thermal_zone0/temp /etc/armbianmonitor/datasources/soctemp
	elif [ -f /sys/devices/virtual/thermal/thermal_zone1/temp ]; then
		ln -fs /sys/devices/virtual/thermal/thermal_zone1/temp /etc/armbianmonitor/datasources/soctemp
	fi
	# relink template
	if [ "X${DebugMode}" = "XDEBUG" ]; then
		ln -sf /etc/armbianmonitor/templates/unknown_board_template_debug.conf /etc/armbianmonitor/cpu_pmic.conf
	else
		ln -sf /etc/armbianmonitor/templates/unknown_board_template.conf /etc/armbianmonitor/cpu_pmic.conf
	fi
} # DealWithNewBoard

CreateTemplates() {
	# check whether templates we write aren't symlinks to somewhere else
	for i in armbian.conf uptime.conf version.conf ; do
		if [ -L /etc/armbianmonitor/${i} ]; then
			rm /etc/armbianmonitor/${i}
			touch /etc/armbianmonitor/${i}
			chmod 711 /etc/armbianmonitor/${i}
		fi
	done

	# check whether our logo is available
	if [ ! -f /usr/share/rpimonitor/web/img/armbian.png ]; then
		if [ -L /usr/share/rpimonitor/web/img/armbian.png ]; then
			rm /usr/share/rpimonitor/web/img/armbian.png
		fi
		cp -p /etc/armbianmonitor/templates/armbian.png /usr/share/rpimonitor/web/img/
	fi

	# create main template
	echo "web.page.icon='img/armbian.png'
web.page.menutitle='RPi-Monitor  <sub>('+data.hostname+')</sub>'
web.page.pagetitle='RPi-Monitor ('+data.hostname+')'
web.status.1.name=$ID
web.statistics.1.name=$ID
web.addons.1.name=Addons
web.addons.1.addons=about
include=/etc/armbianmonitor/version.conf
include=/etc/armbianmonitor/uptime.conf
include=/etc/armbianmonitor/cpu_pmic.conf
include=/etc/rpimonitor/template/memory.conf" >/etc/armbianmonitor/armbian.conf

	# remove firmware line in version info:
	grep -v "Firmware" /etc/rpimonitor/template/version.conf | sed 's|line.5|line.4|' >/etc/armbianmonitor/version.conf

	# uptime template with correct machine name:
	sed "s/Raspberry Pi/$ID/" < /etc/rpimonitor/template/uptime.conf >/etc/armbianmonitor/uptime.conf

	# check swap settings. In case we're using swap then add template
	HowManySwapDevices=$(swapon -s | wc -l)
	if [[ ${HowManySwapDevices} -gt 1 ]]; then
		echo "include=/etc/rpimonitor/template/swap.conf" >>/etc/armbianmonitor/armbian.conf
	fi

	echo "include=/etc/armbianmonitor/template/filesystems.conf
include=/etc/armbianmonitor/template/disks.conf
include=/etc/rpimonitor/template/network.conf" >>/etc/armbianmonitor/armbian.conf

	UpdateFileSystems
	UpdateDisks
} #

UpdateFileSystems() {
	# Generates /etc/armbianmonitor/template/filesystems.conf dynamically
	# based on the contents of /etc/armbianmonitoring/filesystems.conf

	# if not existing, create config file with single entry for /
	if [ ! -f /etc/armbianmonitoring/filesystems.conf ]; then
		echo '/' >/etc/armbianmonitoring/filesystems.conf
		chmod 711 /etc/armbianmonitoring/filesystems.conf
	fi

	# Update template:

} # UpdateFileSystems

UpdateDisks() {
	# Generates /etc/armbianmonitor/template/disks.conf dynamically
	# based on the contents of /etc/armbianmonitor/disks.conf
	:
} # UpdateDisks

PreRequisits() {
	export PATH=/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin
	unset LANG
	LastDiskCheck=0

	# we need the informations gathered from armhwinfo. In case we're not called
	# from there, get the variables on our own
	if [ "X${ID}" = "X" ]; then
		. /etc/init.d/armhwinfo start >/dev/null 2>&1
	fi

	# check/create /etc/armbianmonitor and /etc/armbianmonitor/datasources
	if [ ! -d /etc/armbianmonitor ]; then
		mkdir -p -m 755 /etc/armbianmonitor
	else
		chmod 755 /etc/armbianmonitor
	fi
	if [ ! -d /etc/armbianmonitor/datasources ]; then
		mkdir -p -m 755 /etc/armbianmonitor/datasources
	else
		chmod 755 /etc/armbianmonitor/datasources
	fi

	# check whether we should do debug monitoring or normal
	read DebugMode </etc/armbianmonitor/start-monitoring
} # PreRequisits

Main