1 -- Copyright 2017 Stan Grishin <stangri@melmac.net>
2 -- Licensed to the public under the Apache License 2.0.
4 module("luci.controller.advanced_reboot", package.seeall)
6 -- device_name, board_name, part1, part2, offset, env_var_1, value_1_1, value_1_2, env_var_2, value_2_1, value_2_2
8 {"Linksys EA3500", "linksys-audi", "mtd3", "mtd5", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
9 {"Linksys E4200v2/EA4500", "linksys-viper", "mtd3", "mtd5", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
10 {"Linksys EA8500", "ea8500", "mtd13", "mtd15", 32, "boot_part", 1, 2},
11 {"Linksys WRT1200AC", "armada-385-linksys-caiman", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
12 {"Linksys WRT1900AC", "armada-xp-linksys-mamba", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
13 {"Linksys WRT1900ACv2", "armada-385-linksys-cobra", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
14 {"Linksys WRT1900ACS", "armada-385-linksys-shelby", "mtd4", "mtd6", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
15 {"Linksys WRT3200ACM", "armada-385-linksys-rango", "mtd5", "mtd7", 32, "boot_part", 1, 2, "bootcmd", "run nandboot", "run altnandboot"},
16 {"ZyXEL NBG6817","nbg6817","mmcblk0p4","mmcblk0p7",32,nil,255,1}
20 device_board_name = luci.util.trim(luci.sys.exec("cat /tmp/sysinfo/board_name"))
22 table_board_name = devices[i][2]:gsub('-','')
23 if device_board_name and device_board_name:gsub('-',''):match(table_board_name) then
24 device_name = devices[i][1]
25 partition_one_mtd = devices[i][3] or nil
26 partition_two_mtd = devices[i][4] or nil
27 partition_skip = devices[i][5] or nil
28 boot_envvar1 = devices[i][6] or nil
29 boot_envvar1_partition_one = tonumber(devices[i][7]) or nil
30 boot_envvar1_partition_two = tonumber(devices[i][8]) or nil
31 boot_envvar2 = devices[i][9] or nil
32 boot_envvar2_partition_one = devices[i][10] or nil
33 boot_envvar2_partition_two = devices[i][11] or nil
34 if partition_one_mtd and partition_skip then
35 partition_one_label = luci.util.trim(luci.sys.exec("dd if=/dev/" .. partition_one_mtd .. " bs=1 skip=" .. partition_skip .. " count=25" .. " 2>/dev/null"))
36 n, partition_one_version = string.match(partition_one_label, '(Linux)-([%d|.]+)')
38 if partition_two_mtd and partition_skip then
39 partition_two_label = luci.util.trim(luci.sys.exec("dd if=/dev/" .. partition_two_mtd .. " bs=1 skip=" .. partition_skip .. " count=25" .. " 2>/dev/null"))
40 n, partition_two_version = string.match(partition_two_label, '(Linux)-([%d|.]+)')
42 if string.find(partition_one_label, "LEDE") then partition_one_os = "LEDE" end
43 if string.find(partition_one_label, "OpenWrt") then partition_one_os = "OpenWrt" end
44 if string.find(partition_one_label, "Linksys") then partition_one_os = "Linksys" end
45 if string.find(partition_two_label, "LEDE") then partition_two_os = "LEDE" end
46 if string.find(partition_two_label, "OpenWrt") then partition_two_os = "OpenWrt" end
47 if string.find(partition_two_label, "Linksys") then partition_two_os = "Linksys" end
48 if device_name and device_name == "ZyXEL NBG6817" then
49 if not partition_one_os then partition_one_os = "ZyXEL" end
50 if not partition_two_os then partition_two_os = "ZyXEL" end
52 if not partition_one_os then partition_one_os = "Unknown" end
53 if not partition_two_os then partition_two_os = "Unknown" end
54 if partition_one_os and partition_one_version then partition_one_os = partition_one_os .. " (Linux " .. partition_one_version .. ")" end
55 if partition_two_os and partition_two_version then partition_two_os = partition_two_os .. " (Linux " .. partition_two_version .. ")" end
57 if device_name and device_name == "ZyXEL NBG6817" then
58 if not zyxelFlagPartition then zyxelFlagPartition = luci.util.trim(luci.sys.exec("source /lib/functions.sh; find_mtd_part 0:DUAL_FLAG")) end
59 if not zyxelFlagPartition then
60 errorMessage = errorMessage .. luci.i18n.translate("Unable to find Dual Boot Flag Partition." .. " ")
61 luci.util.perror(luci.i18n.translate("Unable to find Dual Boot Flag Partition."))
63 current_partition = tonumber(luci.sys.exec("dd if=" .. zyxelFlagPartition .. " bs=1 count=1 2>/dev/null | hexdump -n 1 -e '1/1 \"%d\"'"))
66 if nixio.fs.access("/usr/sbin/fw_printenv") and nixio.fs.access("/usr/sbin/fw_setenv") then
67 current_partition = tonumber(luci.util.trim(luci.sys.exec("/usr/sbin/fw_printenv -n " .. boot_envvar1)))
70 other_partition = current_partition == boot_envvar1_partition_two and boot_envvar1_partition_one or boot_envvar1_partition_two
75 entry({"admin", "system", "advanced_reboot"}, template("advanced_reboot/advanced_reboot"), _("Advanced Reboot"), 90)
76 entry({"admin", "system", "advanced_reboot", "reboot"}, post("action_reboot"))
77 -- if device_name then entry({"admin", "system", "advanced_reboot", "altreboot"}, post("action_altreboot")) end
78 entry({"admin", "system", "advanced_reboot", "alternative_reboot"}, post("action_altreboot"))
79 entry({"admin", "system", "advanced_reboot", "power_off"}, post("action_poweroff"))
82 function action_reboot()
83 luci.template.render("admin_system/applyreboot", {
84 title = luci.i18n.translate("Rebooting..."),
85 msg = luci.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."),
86 addr = luci.ip.new(uci.cursor():get("network", "lan", "ipaddr")) or "192.168.1.1"
91 function action_altreboot()
92 local zyxelFlagPartition, zyxelBootFlag, zyxelNewBootFlag, errorCode, curEnvSetting, newEnvSetting
95 if luci.http.formvalue("cancel") then
96 luci.http.redirect(luci.dispatcher.build_url('admin/system/advanced_reboot'))
97 -- luci.template.render("advanced_reboot/advanced_reboot",{
98 -- device_name=device_name,
99 -- boot_envvar1_partition_one=boot_envvar1_partition_one,
100 -- partition_one_os=partition_one_os,
101 -- boot_envvar1_partition_two=boot_envvar1_partition_two,
102 -- partition_two_os=partition_two_os,
103 -- current_partition=current_partition,
104 -- errorMessage = luci.i18n.translate("Alternative reboot cancelled.")})
107 local step = tonumber(luci.http.formvalue("step") or 1)
109 if device_name and nixio.fs.access("/usr/sbin/fw_printenv") and nixio.fs.access("/usr/sbin/fw_setenv") then
110 luci.template.render("advanced_reboot/alternative_reboot",{})
112 luci.template.render("advanced_reboot/advanced_reboot",{errorMessage = luci.i18n.translate("No access to fw_printenv or fw_printenv!")})
114 elseif step == 2 then
115 if boot_envvar1 or boot_envvar2 then -- Linksys devices
117 curEnvSetting = tonumber(luci.util.trim(luci.sys.exec("/usr/sbin/fw_printenv -n " .. boot_envvar1)))
118 if not curEnvSetting then
119 errorMessage = errorMessage .. luci.i18n.translate("Unable to obtain firmware environment variable") .. ": " .. boot_envvar1 .. ". "
120 luci.util.perror(luci.i18n.translate("Unable to obtain firmware environment variable") .. ": " .. boot_envvar1 .. ".")
122 newEnvSetting = curEnvSetting == boot_envvar1_partition_one and boot_envvar1_partition_two or boot_envvar1_partition_one
123 errorCode = luci.sys.call("/usr/sbin/fw_setenv " .. boot_envvar1 .. " " .. newEnvSetting)
124 if errorCode ~= 0 then
125 errorMessage = errorMessage .. luci.i18n.translate("Unable to set firmware environment variable") .. ": " .. boot_envvar1 .. " " .. luci.i18n.translate("to") .. " " .. newEnvSetting .. ". "
126 luci.util.perror(luci.i18n.translate("Unable to set firmware environment variable") .. ": " .. boot_envvar1 .. " " .. luci.i18n.translate("to") .. " " .. newEnvSetting .. ".")
131 curEnvSetting = luci.util.trim(luci.sys.exec("/usr/sbin/fw_printenv -n " .. boot_envvar2))
132 if not curEnvSetting then
133 errorMessage = errorMessage .. luci.i18n.translate("Unable to obtain firmware environment variable") .. ": " .. boot_envvar2 .. ". "
134 luci.util.perror(luci.i18n.translate("Unable to obtain firmware environment variable") .. ": " .. boot_envvar2 .. ".")
136 newEnvSetting = curEnvSetting == boot_envvar2_partition_one and boot_envvar2_partition_two or boot_envvar2_partition_one
137 errorCode = luci.sys.call("/usr/sbin/fw_setenv " .. boot_envvar2 .. " '" .. newEnvSetting .. "'")
138 if errorCode ~= 0 then
139 errorMessage = errorMessage .. luci.i18n.translate("Unable to set firmware environment variable") .. ": " .. boot_envvar2 .. " " .. luci.i18n.translate("to") .. " " .. newEnvSetting .. ". "
140 luci.util.perror(luci.i18n.translate("Unable to set firmware environment variable") .. ": " .. boot_envvar2 .. " " .. luci.i18n.translate("to") .. " " .. newEnvSetting .. ".")
144 else -- NetGear device
145 if not zyxelFlagPartition then zyxelFlagPartition = luci.util.trim(luci.sys.exec("source /lib/functions.sh; find_mtd_part 0:DUAL_FLAG")) end
146 if not zyxelFlagPartition then
147 errorMessage = errorMessage .. luci.i18n.translate("Unable to find Dual Boot Flag Partition." .. " ")
148 luci.util.perror(luci.i18n.translate("Unable to find Dual Boot Flag Partition."))
150 zyxelBootFlag = tonumber(luci.sys.exec("dd if=" .. zyxelFlagPartition .. " bs=1 count=1 2>/dev/null | hexdump -n 1 -e '1/1 \"%d\"'"))
151 zyxelNewBootFlag = zyxelBootFlag and zyxelBootFlag == 1 and "\\xff" or "\\x01"
152 if zyxelNewBootFlag then
153 errorCode = luci.sys.call("printf \"" .. zyxelNewBootFlag .. "\" >" .. zyxelFlagPartition )
154 if errorCode ~= 0 then
155 errorMessage = errorMessage .. luci.i18n.translate("Unable to set Dual Boot Flag Partition entry for partition") .. ": " .. zyxelFlagPartition .. ". "
156 luci.util.perror(luci.i18n.translate("Unable to set Dual Boot Flag Partition entry for partition") .. ": " .. zyxelFlagPartition .. ".")
161 if errorMessage == "" then
162 luci.template.render("admin_system/applyreboot", {
163 title = luci.i18n.translate("Rebooting..."),
164 msg = luci.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."),
165 addr = luci.ip.new(uci.cursor():get("network", "lan", "ipaddr")) or "192.168.1.1"
169 luci.template.render("advanced_reboot/advanced_reboot",{
170 device_name=device_name,
171 boot_envvar1_partition_one=boot_envvar1_partition_one,
172 partition_one_os=partition_one_os,
173 boot_envvar1_partition_two=boot_envvar1_partition_two,
174 partition_two_os=partition_two_os,
175 current_partition=current_partition,
176 errorMessage = errorMessage})
181 function action_poweroff()
182 if luci.http.formvalue("cancel") then
183 luci.http.redirect(luci.dispatcher.build_url('admin/system/advanced_reboot'))
186 local step = tonumber(luci.http.formvalue("step") or 1)
188 if nixio.fs.access("/sbin/poweroff") then
189 luci.template.render("advanced_reboot/power_off",{})
191 luci.template.render("advanced_reboot/advanced_reboot",{})
193 elseif step == 2 then
194 luci.template.render("admin_system/applyreboot", {
195 title = luci.i18n.translate("Shutting down..."),
196 msg = luci.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."),
197 addr = luci.ip.new(uci.cursor():get("network", "lan", "ipaddr")) or "192.168.1.1"
199 luci.sys.call("/sbin/poweroff")