From feb8d823e59f8c9bf1bdadce6c7fc2a88966f12e Mon Sep 17 00:00:00 2001 From: Ansuel Smith Date: Wed, 7 Oct 2020 14:42:39 +0200 Subject: 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 Tested-by: Stefan Lippers-Hollmann --- .../root/usr/libexec/rpcd/luci.advanced_reboot | 362 +++++++++++++++++++++ 1 file changed, 362 insertions(+) create mode 100755 applications/luci-app-advanced-reboot/root/usr/libexec/rpcd/luci.advanced_reboot (limited to 'applications/luci-app-advanced-reboot/root/usr/libexec/rpcd/luci.advanced_reboot') 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 -- cgit v1.2.3