303 lines
14 KiB
Lua
303 lines
14 KiB
Lua
-- Copyright 2017-2018 Stan Grishin <stangri@melmac.net>
|
|
-- Licensed to the public under the Apache License 2.0.
|
|
|
|
module("luci.controller.advanced_reboot", package.seeall)
|
|
|
|
local util = require "luci.util"
|
|
local fs = require "nixio.fs"
|
|
local sys = require "luci.sys"
|
|
local http = require "luci.http"
|
|
local dispatcher = require "luci.dispatcher"
|
|
local i18n = require "luci.i18n"
|
|
local ltemplate = require "luci.template"
|
|
local ip = require "luci.ip"
|
|
local http = require "luci.http"
|
|
local sys = require "luci.sys"
|
|
local dispatcher = require "luci.dispatcher"
|
|
local uci = require "luci.model.uci".cursor()
|
|
local packageName = "luci-app-advanced-reboot"
|
|
|
|
function logger(t)
|
|
util.exec("logger -t " .. packageName .. " '" .. tostring(t) .. "'")
|
|
end
|
|
|
|
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
|
|
|
|
function get_partition_os_info(op_ubi)
|
|
local cp_info, ap_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(i18n.translate("attempting to mount alternative partition") .. " (mtd" .. tostring(op_ubi) .. ")")
|
|
alt_partition_unmount(op_ubi)
|
|
alt_partition_mount(op_ubi)
|
|
if fs.access("/alt/rom/etc/os-release") then
|
|
ap_info = util.trim(util.exec('. /alt/rom/etc/os-release && echo "$PRETTY_NAME"'))
|
|
if ap_info:find("SNAPSHOT") then
|
|
ap_info = util.trim(util.exec('. /alt/rom/etc/os-release && echo "$OPENWRT_RELEASE"'))
|
|
end
|
|
end
|
|
logger(i18n.translate("attempting to unmount alternative partition") .. " (mtd" .. tostring(op_ubi) .. ")")
|
|
alt_partition_unmount(op_ubi)
|
|
return cp_info, ap_info
|
|
end
|
|
|
|
function alt_partition_mount(op_ubi)
|
|
local ubi_dev
|
|
util.exec('for i in rom 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/rom")
|
|
util.exec("mount -t ubifs /dev/ubi" .. ubi_dev .. "_1 /alt/overlay")
|
|
-- util.exec("mount -t overlay overlay -o noatime,lowerdir=/alt/rom,upperdir=/alt/overlay/upper,workdir=/alt/overlay/work /alt/firmware")
|
|
end
|
|
|
|
function alt_partition_unmount(op_ubi)
|
|
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/overlay ] && umount /alt/overlay")
|
|
util.exec("[ -d /alt/rom ] && umount /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
|
|
|
|
devices = {
|
|
-- deviceName, boardName, part1MTD, part2MTD, offset, envVar1, envVar1Value1, envVar1Value2, envVar2, envVar2Value1, envVar2Value2
|
|
{"Linksys EA3500", "linksys-audi", "mtd3", "mtd5", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys E4200v2/EA4500", "linksys-viper", "mtd3", "mtd5", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys EA6350v3", "linksys-ea6350v3", "mtd10", "mtd12", 192, "boot_part", 1, 2},
|
|
{"Linksys EA8300", "linksys-ea8300", "mtd10", "mtd12", 192, "boot_part", 1, 2},
|
|
{"Linksys EA8500", "ea8500", "mtd13", "mtd15", 32, "boot_part", 1, 2},
|
|
-- {"Linksys EA9500", "linksys-panamera", "mtd3", "mtd6", 28, "boot_part", 1, 2},
|
|
{"Linksys WRT1200AC", "linksys-caiman", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys WRT1900AC", "linksys-mamba", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys WRT1900ACv2", "linksys-cobra", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys WRT1900ACS", "linksys-shelby", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys WRT3200ACM", "linksys-rango", "mtd5", "mtd7", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"Linksys WRT32X", "linksys-venom", "mtd5", "mtd7", nil, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
|
|
{"ZyXEL NBG6817", "nbg6817", "mmcblk0p4", "mmcblk0p7", 32, nil, 255, 1}
|
|
}
|
|
|
|
-- local errorMessage, d
|
|
-- local device_name, p1_mtd, p2_mtd, offset, bev1, bev1p1, bev1p2, bev2, bev2p1, bev2p2
|
|
romBoardName = util.trim(util.exec("cat /tmp/sysinfo/board_name"))
|
|
|
|
for i=1, #devices do
|
|
d = devices[i][2]:gsub('%p','')
|
|
if romBoardName and romBoardName:gsub('%p',''):match(d) then
|
|
device_name = devices[i][1]
|
|
p1_mtd = devices[i][3] or nil
|
|
p2_mtd = devices[i][4] or nil
|
|
offset = devices[i][5] or nil
|
|
bev1 = devices[i][6] or nil
|
|
bev1p1 = tonumber(devices[i][7]) or nil
|
|
bev1p2 = tonumber(devices[i][8]) or nil
|
|
bev2 = devices[i][9] or nil
|
|
bev2p1 = devices[i][10] or nil
|
|
bev2p2 = devices[i][11] or nil
|
|
if p1_mtd and offset then
|
|
p1_label = util.trim(util.exec("dd if=/dev/" .. p1_mtd .. " bs=1 skip=" .. offset .. " count=128" .. " 2>/dev/null"))
|
|
n, p1_version = p1_label:match('(Linux)-([%d|.]+)')
|
|
end
|
|
if p2_mtd and offset then
|
|
p2_label = util.trim(util.exec("dd if=/dev/" .. p2_mtd .. " bs=1 skip=" .. offset .. " count=128" .. " 2>/dev/null"))
|
|
n, p2_version = p2_label:match('(Linux)-([%d|.]+)')
|
|
end
|
|
if p1_label and p1_label:find("LEDE") then p1_os = "LEDE" end
|
|
if p1_label and p1_label:find("OpenWrt") then p1_os = "OpenWrt" end
|
|
if p1_label and p1_label:find("Linksys") then p1_os = "Linksys" end
|
|
if p2_label and p2_label:find("LEDE") then p2_os = "LEDE" end
|
|
if p2_label and p2_label:find("OpenWrt") then p2_os = "OpenWrt" end
|
|
if p2_label and p2_label:find("Linksys") then p2_os = "Linksys" end
|
|
if device_name == "ZyXEL NBG6817" then
|
|
if not p1_os then p1_os = "ZyXEL" end
|
|
if not p2_os then p2_os = "ZyXEL" end
|
|
end
|
|
if device_name == "Linksys WRT32X" then
|
|
if not p1_os then p1_os = "Unknown/Compressed" end
|
|
if not p2_os then p2_os = "Unknown/Compressed" end
|
|
end
|
|
if not p1_os then p1_os = "Unknown" end
|
|
if not p2_os then p2_os = "Unknown" end
|
|
if p1_os and p1_version then p1_os = p1_os .. " (Linux " .. p1_version .. ")" end
|
|
if p2_os and p2_version then p2_os = p2_os .. " (Linux " .. p2_version .. ")" end
|
|
|
|
if device_name == "ZyXEL NBG6817" then
|
|
if not zyxelFlagPartition then zyxelFlagPartition = util.trim(util.exec(". /lib/functions.sh; find_mtd_part 0:DUAL_FLAG")) end
|
|
if not zyxelFlagPartition then
|
|
errorMessage = errorMessage or "" .. i18n.translate("Unable to find Dual Boot Flag Partition." .. " ")
|
|
util.perror(i18n.translate("Unable to find Dual Boot Flag Partition."))
|
|
else
|
|
current_partition = tonumber(util.exec("dd if=" .. zyxelFlagPartition .. " bs=1 count=1 2>/dev/null | hexdump -n 1 -e '1/1 \"%d\"'"))
|
|
end
|
|
else
|
|
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 " .. bev1)))
|
|
end
|
|
end
|
|
other_partition = current_partition == bev1p2 and bev1p1 or bev1p2
|
|
|
|
if is_alt_mountable(p1_mtd, p2_mtd) then
|
|
if current_partition == bev1p1 then
|
|
op_ubi = tonumber(p2_mtd:sub(4)) + 1
|
|
else
|
|
op_ubi = tonumber(p1_mtd:sub(4)) + 1
|
|
end
|
|
local cp_info, ap_info = get_partition_os_info(op_ubi)
|
|
if current_partition == bev1p1 then
|
|
p1_os = cp_info or p1_os
|
|
p2_os = ap_info or p2_os
|
|
else
|
|
p1_os = ap_info or p1_os
|
|
p2_os = cp_info or p2_os
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function index()
|
|
entry({"admin", "system", "advanced_reboot"}, template("advanced_reboot/advanced_reboot"), _("Advanced Reboot"), 90)
|
|
entry({"admin", "system", "advanced_reboot", "reboot"}, post("action_reboot"))
|
|
entry({"admin", "system", "advanced_reboot", "alternative_reboot"}, post("action_altreboot"))
|
|
entry({"admin", "system", "advanced_reboot", "power_off"}, post("action_poweroff"))
|
|
end
|
|
|
|
function action_reboot()
|
|
ltemplate.render("advanced_reboot/applyreboot", {
|
|
title = i18n.translate("Rebooting..."),
|
|
msg = i18n.translate("The system is rebooting now.<br /> DO NOT POWER OFF THE DEVICE!<br /> Wait a few minutes before you try to reconnect. It might be necessary to renew the address of your computer to reach the device again, depending on your settings."),
|
|
addr = ip.new(type(ip) == "string" and ip or "192.168.1.1") or "192.168.1.1"
|
|
})
|
|
sys.reboot()
|
|
end
|
|
|
|
function action_altreboot()
|
|
local zyxelFlagPartition, zyxelBootFlag, zyxelNewBootFlag, errorCode, curEnvSetting, newEnvSetting
|
|
errorMessage = nil
|
|
errorCode = 0
|
|
if http.formvalue("cancel") then
|
|
http.redirect(dispatcher.build_url('admin/system/advanced_reboot'))
|
|
return
|
|
end
|
|
local step = tonumber(http.formvalue("step") or 1)
|
|
if step == 1 then
|
|
if fs.access("/usr/sbin/fw_printenv") and fs.access("/usr/sbin/fw_setenv") then
|
|
ltemplate.render("advanced_reboot/alternative_reboot",{})
|
|
else
|
|
ltemplate.render("advanced_reboot/advanced_reboot",{errorMessage = i18n.translate("No access to fw_printenv or fw_printenv!")})
|
|
end
|
|
elseif step == 2 then
|
|
if bev1 or bev2 then -- Linksys devices
|
|
if bev1 then
|
|
curEnvSetting = tonumber(util.trim(util.exec("fw_printenv -n " .. bev1)))
|
|
if not curEnvSetting then
|
|
errorMessage = errorMessage .. i18n.translate("Unable to obtain firmware environment variable") .. ": " .. bev1 .. ". "
|
|
util.perror(i18n.translate("Unable to obtain firmware environment variable") .. ": " .. bev1 .. ".")
|
|
else
|
|
newEnvSetting = curEnvSetting == bev1p1 and bev1p2 or bev1p1
|
|
errorCode = sys.call("fw_setenv " .. bev1 .. " " .. newEnvSetting)
|
|
if errorCode ~= 0 then
|
|
errorMessage = errorMessage or "" .. i18n.translate("Unable to set firmware environment variable") .. ": " .. bev1 .. " " .. i18n.translate("to") .. " " .. newEnvSetting .. ". "
|
|
util.perror(i18n.translate("Unable to set firmware environment variable") .. ": " .. bev1 .. " " .. i18n.translate("to") .. " " .. newEnvSetting .. ".")
|
|
end
|
|
end
|
|
end
|
|
if bev2 then
|
|
curEnvSetting = util.trim(util.exec("fw_printenv -n " .. bev2))
|
|
if not curEnvSetting then
|
|
errorMessage = errorMessage or "" .. i18n.translate("Unable to obtain firmware environment variable") .. ": " .. bev2 .. ". "
|
|
util.perror(i18n.translate("Unable to obtain firmware environment variable") .. ": " .. bev2 .. ".")
|
|
else
|
|
newEnvSetting = curEnvSetting == bev2p1 and bev2p2 or bev2p1
|
|
errorCode = sys.call("fw_setenv " .. bev2 .. " '" .. newEnvSetting .. "'")
|
|
if errorCode ~= 0 then
|
|
errorMessage = errorMessage or "" .. i18n.translate("Unable to set firmware environment variable") .. ": " .. bev2 .. " " .. i18n.translate("to") .. " " .. newEnvSetting .. ". "
|
|
util.perror(i18n.translate("Unable to set firmware environment variable") .. ": " .. bev2 .. " " .. i18n.translate("to") .. " " .. newEnvSetting .. ".")
|
|
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
|
|
errorMessage = errorMessage .. i18n.translate("Unable to find Dual Boot Flag Partition." .. " ")
|
|
util.perror(i18n.translate("Unable to find Dual Boot Flag Partition."))
|
|
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
|
|
errorMessage = errorMessage or "" .. i18n.translate("Unable to set Dual Boot Flag Partition entry for partition") .. ": " .. zyxelFlagPartition .. ". "
|
|
util.perror(i18n.translate("Unable to set Dual Boot Flag Partition entry for partition") .. ": " .. zyxelFlagPartition .. ".")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
if not errorMessage then
|
|
ltemplate.render("advanced_reboot/applyreboot", {
|
|
title = i18n.translate("Rebooting..."),
|
|
msg = i18n.translate("The system is rebooting to an alternative partition now.<br /> DO NOT POWER OFF THE DEVICE!<br /> Wait a few minutes before you try to reconnect. It might be necessary to renew the address of your computer to reach the device again, depending on your settings."),
|
|
addr = ip.new(uci:get("network", "lan", "ipaddr")) or "192.168.1.1"
|
|
})
|
|
sys.reboot()
|
|
else
|
|
ltemplate.render("advanced_reboot/advanced_reboot",{
|
|
romBoardName=romBoardName,
|
|
device_name=device_name,
|
|
bev1p1=bev1p1,
|
|
p1_os=p1_os,
|
|
bev1p2=bev1p2,
|
|
p2_os=p2_os,
|
|
current_partition=current_partition,
|
|
errorMessage = errorMessage})
|
|
end
|
|
end
|
|
end
|
|
|
|
function action_poweroff()
|
|
if http.formvalue("cancel") then
|
|
http.redirect(dispatcher.build_url('admin/system/advanced_reboot'))
|
|
return
|
|
end
|
|
local step = tonumber(http.formvalue("step") or 1)
|
|
if step == 1 then
|
|
if fs.access("/sbin/poweroff") then
|
|
ltemplate.render("advanced_reboot/power_off",{})
|
|
else
|
|
ltemplate.render("advanced_reboot/advanced_reboot",{})
|
|
end
|
|
elseif step == 2 then
|
|
ltemplate.render("advanced_reboot/applyreboot", {
|
|
title = i18n.translate("Shutting down..."),
|
|
msg = i18n.translate("The system is shutting down now.<br /> DO NOT POWER OFF THE DEVICE!<br /> It might be necessary to renew the address of your computer to reach the device again, depending on your settings."),
|
|
addr = ip.new(uci:get("network", "lan", "ipaddr")) or "192.168.1.1"
|
|
})
|
|
sys.call("/sbin/poweroff")
|
|
end
|
|
end
|