luci-app-advanced-reboot: convert to client side

Rework the advanced-reboot app to client side implementation.
Change devices file to json type.

Signed-off-by: Ansuel Smith <ansuelsmth@gmail.com>
Tested-by: Stefan Lippers-Hollmann <s.l-h@gmx.de>

1 files changed, 362 insertions, 0 deletions

diff --git a/applications/luci-app-advanced-reboot/root/usr/libexec/rpcd/luci.advanced_reboot b/applications/luci-app-advanced-reboot/root/usr/libexec/rpcd/luci.advanced_reboot
new file mode 100755
index 0000000000..a6c7c0e540
--- /dev/null
+++ b/applications/luci-app-advanced-reboot/root/usr/libexec/rpcd/luci.advanced_reboot
@@ -0,0 +1,362 @@
+#!/usr/bin/env lua
+
+local json = require "luci.jsonc"
+local nixio = require "nixio"
+local fs   = require "nixio.fs"
+local sys  = require "luci.sys"
+local util = require "luci.util"
+
+local devices_dir = "/usr/share/advanced-reboot/devices/"
+
+local function logger(t)
+	util.exec("logger -t advanced-reboot '" .. tostring(t) .. "'")
+end
+
+local function is_alt_mountable(p1_mtd, p2_mtd)
+	if p1_mtd:sub(1,3) == "mtd" and
+				 p2_mtd:sub(1,3) == "mtd" and
+				 fs.access("/usr/sbin/ubiattach") and
+				 fs.access("/usr/sbin/ubiblock") and
+				 fs.access("/bin/mount") then
+		return true
+	else
+		return false
+	end
+end
+
+local function alt_partition_mount(op_ubi)
+	local ubi_dev
+	util.exec('for i in alt_rom alt_overlay firmware; do [ ! -d "$i" ] && mkdir -p "/alt/${i}"; done')
+	ubi_dev = tostring(util.exec("ubiattach -m " .. tostring(op_ubi)))
+	_, _, ubi_dev = ubi_dev:find("UBI device number (%d+)")
+	if not ubi_dev then
+		util.exec("ubidetach -m " .. tostring(op_ubi))
+		return
+	end
+	util.exec("ubiblock --create /dev/ubi" .. ubi_dev .. "_0")
+	util.exec("mount -t squashfs -o ro /dev/ubiblock" .. ubi_dev .. "_0 /alt/alt_rom")
+	util.exec("mount -t ubifs /dev/ubi" .. ubi_dev .. "_1 /alt/alt_overlay")
+--	util.exec("mount -t overlay overlay -o noatime,lowerdir=/alt/rom,upperdir=/alt/overlay/upper,workdir=/alt/overlay/work /alt/firmware")
+end
+
+local function alt_partition_unmount(op_ubi)
+	local i
+	local mtdCount = tonumber(util.exec("ubinfo | grep 'Present UBI devices' | grep -c ','"))
+	mtdCount = mtdCount and mtdCount + 1 or 10
+--	util.exec("[ -d /alt/firmware ] && umount /alt/firmware")
+	util.exec("[ -d /alt/alt_overlay ] && umount /alt/alt_overlay")
+	util.exec("[ -d /alt/alt_rom ] && umount /alt/alt_rom")
+	for i = 0, mtdCount do
+		if not fs.access("/sys/devices/virtual/ubi/ubi" .. tostring(i) .. "/mtd_num") then break end
+		ubi_mtd =  tonumber(util.trim(util.exec("cat /sys/devices/virtual/ubi/ubi" .. i .. "/mtd_num")))
+		if ubi_mtd and ubi_mtd == op_ubi then
+			util.exec("ubiblock --remove /dev/ubi" .. tostring(i) .. "_0")
+			util.exec("ubidetach -m " .. tostring(op_ubi))
+			util.exec('rm -rf /alt')
+		end
+	end
+end
+
+local function get_partition_os_info(op_ubi)
+	local cp_info, op_info
+	if fs.access("/etc/os-release") then
+		cp_info = util.trim(util.exec('. /etc/os-release && echo "$PRETTY_NAME"'))
+		if cp_info:find("SNAPSHOT") then
+			cp_info = util.trim(util.exec('. /etc/os-release && echo "$OPENWRT_RELEASE"'))
+		end
+	end
+	logger(string.format("attempting to mount alternative partition (mtd%s)", tostring(op_ubi)))
+	alt_partition_unmount(op_ubi)
+	alt_partition_mount(op_ubi)
+	if fs.access("/alt/alt_rom/etc/os-release") then
+		op_info = util.trim(util.exec('. /alt/alt_rom/etc/os-release && echo "$PRETTY_NAME"'))
+		if op_info:find("SNAPSHOT") then
+			op_info = util.trim(util.exec('. /alt/alt_rom/etc/os-release && echo "$OPENWRT_RELEASE"'))
+		end
+	end
+	logger(string.format("attempting to unmount alternative partition (mtd%s)", tostring(op_ubi)))
+	alt_partition_unmount(op_ubi)
+	return cp_info, op_info
+end
+
+local function find_device_data(romBoardName)
+	local filename
+
+	for filename in fs.dir(devices_dir) do
+		local filedata = fs.readfile(devices_dir .. filename)
+
+		local p = json.parse(filedata or {})
+
+		if p then
+			local boardName
+
+			if p.boardName then
+				boardName = p.boardName:gsub('%p','')
+				if romBoardName:match(boardName) then return p end
+			end
+			if p.boardNames then
+				for i, v in pairs(p.boardNames) do
+					boardName = v:gsub('%p','')
+					if romBoardName:match(boardName) then return p end
+				end
+			end
+		end
+	end
+
+	return nil
+end
+
+local methods = {
+	obtain_device_info = {
+		call = function()
+
+			local ret = {}
+			local romBoardName = fs.readfile('/tmp/sysinfo/board_name')
+
+			if not romBoardName then
+				ret.error = 'NO_BOARD_NAME'
+				return ret
+			end
+
+			romBoardName = romBoardName:gsub('\n','')
+
+			ret.rom_board_name = romBoardName
+
+			romBoardName = romBoardName:gsub('%p','')
+
+			local p, boardName, n, p1_label, p1_version, p2_label, p2_version, p1_os, p2_os
+			local current_partition
+			local op_ubi, cp_info, op_info, zyxelFlagPartition
+
+			p = find_device_data(romBoardName)
+
+			if p then
+				if p.labelOffset then
+					if p.partition1MTD then
+						p1_label = util.trim(util.exec("dd if=/dev/" .. p.partition1MTD .. " bs=1 skip=" .. p.labelOffset .. " count=128" .. " 2>/dev/null"))
+						n, p1_version = p1_label:match('(Linux)-([%d|.]+)')
+					end
+
+					if p1_label then
+						if p1_label:find("LEDE") then p1_os = "LEDE" end
+						if p1_label:find("OpenWrt") then p1_os = "OpenWrt" end
+						if p.vendorName and p1_label:find(p.vendorName) then p1_os = p.vendorName end
+					end
+
+					if not p1_os then
+						p1_os = (p.vendorName and p.vendorName or 'Unknown') .. "/" .. "Unknown"
+					end
+					if p1_os and p1_version then p1_os = p1_os .. " (Linux " .. p1_version .. ")" end
+
+					if p.partition2MTD then
+						p2_label = util.trim(util.exec("dd if=/dev/" .. p.partition2MTD .. " bs=1 skip=" .. p.labelOffset .. " count=128" .. " 2>/dev/null"))
+						n, p2_version = p2_label:match('(Linux)-([%d|.]+)')
+					end
+
+					if p2_label then
+						if p2_label:find("LEDE") then p2_os = "LEDE" end
+						if p2_label:find("OpenWrt") then p2_os = "OpenWrt" end
+						if p.vendorName and p2_label:find(p.vendorName) then p2_os = p.vendorName end
+					end
+
+					if not p2_os then
+						p2_os = (p.vendorName and p.vendorName or 'Unknown') .. "/" .. "Unknown"
+					end
+					if p2_os and p2_version then p2_os = p2_os .. " (Linux " .. p2_version .. ")" end
+				else
+					p1_os = p.vendorName .. "/" .. "Unknown" .. " (" .. "Compressed" .. ")"
+					p2_os = p.vendorName .. "/" .. "Unknown" .. " (" .. "Compressed" .. ")"
+				end
+
+				if p.bootEnv1 then
+					if fs.access("/usr/sbin/fw_printenv") and fs.access("/usr/sbin/fw_setenv") then
+						current_partition = tonumber(util.trim(util.exec("fw_printenv -n " .. p.bootEnv1)))
+					end
+				else
+					if not zyxelFlagPartition then zyxelFlagPartition = util.trim(util.exec(". /lib/functions.sh; find_mtd_part 0:DUAL_FLAG")) end
+					if zyxelFlagPartition then
+						current_partition = tonumber(util.exec("dd if=" .. zyxelFlagPartition .. " bs=1 count=1 2>/dev/null | hexdump -n 1 -e '1/1 \"%d\"'"))
+					else
+						ret.error = 'NO_DUAL_FLAG'
+						logger("Unable to find Dual Boot Flag Partition.")
+						return ret
+					end
+				end
+
+				ret.current_partition = current_partition
+
+				if is_alt_mountable(p.partition1MTD, p.partition2MTD) then
+					if current_partition == p.bootEnv1Partition1Value then
+						op_ubi = tonumber(p.partition2MTD:sub(4)) + 1
+					else
+						op_ubi = tonumber(p.partition1MTD:sub(4)) + 1
+					end
+					local cp_info, op_info = get_partition_os_info(op_ubi)
+					if current_partition == p.bootEnv1Partition1Value then
+						p1_os = cp_info or p1_os
+						p2_os = op_info or p2_os
+					else
+						p1_os = op_info or p1_os
+						p2_os = cp_info or p2_os
+					end
+				end
+
+				ret.device_name = (p.vendorName and p.vendorName or "") .. " " .. p.deviceName
+
+				ret.partitions = {
+					{
+						os = p1_os,
+						state = p.bootEnv1Partition1Value == current_partition and 'Current' or 'Alternative',
+						number = p.bootEnv1Partition1Value
+					},
+					{
+						os = p2_os,
+						state = p.bootEnv1Partition2Value == current_partition and 'Current' or 'Alternative',
+						number = p.bootEnv1Partition2Value
+					}
+				}
+			end
+
+			return ret
+		end
+	},
+	toggle_boot_partition = {
+		call = function()
+			local ret = {}
+			local zyxelFlagPartition, zyxelBootFlag, zyxelNewBootFlag, errorCode, curEnvSetting, newEnvSetting
+
+			local romBoardName = fs.readfile('/tmp/sysinfo/board_name')
+
+			if not romBoardName then
+				ret.error = 'NO_BOARD_NAME'
+				return ret
+			end
+
+			romBoardName = romBoardName:gsub('\n',''):gsub('%p','')
+			p = find_device_data(romBoardName)
+			local bev1, bev2 = p.bootEnv1, p.bootEnv2
+
+			if bev1 or bev2 then -- Linksys devices
+				if bev1 then
+					curEnvSetting = tonumber(util.trim(util.exec("fw_printenv -n " .. bev1)))
+					if not curEnvSetting then
+						logger(string.format("Unable to obtain firmware environment variable: %s.", bev1))
+						ret.error = 'NO_FIRM_ENV'
+						ret.args = { bev1 }
+						return ret
+					else
+						local bev1p1, bev1p2 = p.bootEnv1Partition1Value, p.bootEnv1Partition2Value
+						newEnvSetting = curEnvSetting == bev1p1 and bev1p2 or bev1p1
+						errorCode = sys.call("fw_setenv " .. bev1 .. " " .. newEnvSetting)
+						if errorCode ~= 0 then
+							logger(string.format("Unable to set firmware environment variable: %s to %s.", bev1, newEnvSetting))
+							ret.error = 'ERR_SET_ENV'
+							ret.args = { bev1, newEnvSetting }
+							return ret
+						end
+					end
+				end
+				if bev2 then
+					curEnvSetting = util.trim(util.exec("fw_printenv -n " .. bev2))
+					if not curEnvSetting then
+						logger(string.format("Unable to obtain firmware environment variable: %s.", bev2))
+						ret.error = 'NO_FIRM_ENV'
+						ret.args = { bev2 }
+						return ret
+					else
+						local bev2p1, bev2p2 = p.bootEnv2Partition1Value, p.bootEnv2Partition1Value
+						newEnvSetting = curEnvSetting == bev2p1 and bev2p2 or bev2p1
+						errorCode = sys.call("fw_setenv " .. bev2 .. " '" .. newEnvSetting .. "'")
+						if errorCode ~= 0 then
+							logger(string.format("Unable to set firmware environment variable: %s to %s.", bev2, newEnvSetting))
+							ret.error = 'ERR_SET_ENV'
+							ret.args = { bev2, newEnvSetting }
+							return ret
+						end
+					end
+				end
+			else -- NetGear device
+				if not zyxelFlagPartition then zyxelFlagPartition = util.trim(util.exec(". /lib/functions.sh; find_mtd_part 0:DUAL_FLAG")) end
+				if not zyxelFlagPartition then
+					logger("Unable to find Dual Boot Flag Partition.")
+					ret.error = 'NO_DUAL_FLAG'
+				else
+					zyxelBootFlag = tonumber(util.exec("dd if=" .. zyxelFlagPartition .. " bs=1 count=1 2>/dev/null | hexdump -n 1 -e '1/1 \"%d\"'"))
+					zyxelNewBootFlag = zyxelBootFlag and zyxelBootFlag == 1 and "\\xff" or "\\x01"
+					if zyxelNewBootFlag then
+						errorCode = sys.call("printf \"" .. zyxelNewBootFlag .. "\" >" .. zyxelFlagPartition )
+						if errorCode ~= 0 then
+							logger(string.format("Unable to set Dual Boot Flag Partition entry for partition: %s.", zyxelFlagPartition))
+							ret.error = 'ERR_SET_DUAL_FLAG'
+							ret.args = { zyxelFlagPartition }
+							return ret
+						end
+					end
+				end
+			end
+
+			return ret
+		end
+	}
+}
+
+local function parseInput()
+	local parse = json.new()
+	local done, err
+
+	while true do
+		local chunk = io.read(4096)
+		if not chunk then
+			break
+		elseif not done and not err then
+			done, err = parse:parse(chunk)
+		end
+	end
+
+	if not done then
+		print(json.stringify({ error = err or "Incomplete input" }))
+		os.exit(1)
+	end
+
+	return parse:get()
+end
+
+local function validateArgs(func, uargs)
+	local method = methods[func]
+	if not method then
+		print(json.stringify({ error = "Method not found" }))
+		os.exit(1)
+	end
+
+	if type(uargs) ~= "table" then
+		print(json.stringify({ error = "Invalid arguments" }))
+		os.exit(1)
+	end
+
+	uargs.ubus_rpc_session = nil
+
+	local k, v
+	local margs = method.args or {}
+	for k, v in pairs(uargs) do
+		if margs[k] == nil or
+		   (v ~= nil and type(v) ~= type(margs[k]))
+		then
+			print(json.stringify({ error = "Invalid arguments" }))
+			os.exit(1)
+		end
+	end
+
+	return method
+end
+
+if arg[1] == "list" then
+	local _, method, rv = nil, nil, {}
+	for _, method in pairs(methods) do rv[_] = method.args or {} end
+	print((json.stringify(rv):gsub(":%[%]", ":{}")))
+elseif arg[1] == "call" then
+	local args = parseInput()
+	local method = validateArgs(arg[2], args)
+	local result, code = method.call(args)
+	print((json.stringify(result):gsub("^%[%]$", "{}")))
+	os.exit(code or 0)
+end