modules/luci-mod-admin-full/luasrc/controller/admin/system.lua

   1 -- Copyright 2008 Steven Barth <steven@midlink.org>
   2 -- Copyright 2008-2011 Jo-Philipp Wich <jow@openwrt.org>
   3 -- Licensed to the public under the Apache License 2.0.
   4
   5 module("luci.controller.admin.system", package.seeall)
   6
   7 function index()
   8         local fs = require "nixio.fs"
   9
  10         entry({"admin", "system"}, alias("admin", "system", "system"), _("System"), 30).index = true
  11         entry({"admin", "system", "system"}, cbi("admin_system/system"), _("System"), 1)
  12         entry({"admin", "system", "clock_status"}, post_on({ set = true }, "action_clock_status"))
  13
  14         entry({"admin", "system", "admin"}, cbi("admin_system/admin"), _("Administration"), 2)
  15
  16         if fs.access("/bin/opkg") then
  17                 entry({"admin", "system", "packages"}, post_on({ exec = "1" }, "action_packages"), _("Software"), 10)
  18                 entry({"admin", "system", "packages", "ipkg"}, form("admin_system/ipkg"))
  19         end
  20
  21         entry({"admin", "system", "startup"}, form("admin_system/startup"), _("Startup"), 45)
  22         entry({"admin", "system", "crontab"}, form("admin_system/crontab"), _("Scheduled Tasks"), 46)
  23
  24         if fs.access("/sbin/block") and fs.access("/etc/config/fstab") then
  25                 entry({"admin", "system", "fstab"}, cbi("admin_system/fstab"), _("Mount Points"), 50)
  26                 entry({"admin", "system", "fstab", "mount"}, cbi("admin_system/fstab/mount"), nil).leaf = true
  27                 entry({"admin", "system", "fstab", "swap"},  cbi("admin_system/fstab/swap"),  nil).leaf = true
  28         end
  29
  30         local nodes, number = fs.glob("/sys/class/leds/*")
  31         if number > 0 then
  32                 entry({"admin", "system", "leds"}, cbi("admin_system/leds"), _("<abbr title=\"Light Emitting Diode\">LED</abbr> Configuration"), 60)
  33         end
  34
  35         entry({"admin", "system", "flashops"}, call("action_flashops"), _("Backup / Flash Firmware"), 70)
  36         entry({"admin", "system", "flashops", "reset"}, post("action_reset"))
  37         entry({"admin", "system", "flashops", "backup"}, post("action_backup"))
  38         entry({"admin", "system", "flashops", "backupfiles"}, form("admin_system/backupfiles"))
  39
  40         -- call() instead of post() due to upload handling!
  41         entry({"admin", "system", "flashops", "restore"}, call("action_restore"))
  42         entry({"admin", "system", "flashops", "sysupgrade"}, call("action_sysupgrade"))
  43
  44         entry({"admin", "system", "reboot"}, template("admin_system/reboot"), _("Reboot"), 90)
  45         entry({"admin", "system", "reboot", "call"}, post("action_reboot"))
  46 end
  47
  48 function action_clock_status()
  49         local set = tonumber(luci.http.formvalue("set"))
  50         if set ~= nil and set > 0 then
  51                 local date = os.date("*t", set)
  52                 if date then
  53                         luci.sys.call("date -s '%04d-%02d-%02d %02d:%02d:%02d'" %{
  54                                 date.year, date.month, date.day, date.hour, date.min, date.sec
  55                         })
  56                         luci.sys.call("/etc/init.d/sysfixtime restart")
  57                 end
  58         end
  59
  60         luci.http.prepare_content("application/json")
  61         luci.http.write_json({ timestring = os.date("%c") })
  62 end
  63
  64 function action_packages()
  65         local fs = require "nixio.fs"
  66         local ipkg = require "luci.model.ipkg"
  67         local submit = (luci.http.formvalue("exec") == "1")
  68         local update, upgrade
  69         local changes = false
  70         local install = { }
  71         local remove  = { }
  72         local stdout  = { "" }
  73         local stderr  = { "" }
  74         local out, err
  75
  76         -- Display
  77         local display = luci.http.formvalue("display") or "installed"
  78
  79         -- Letter
  80         local letter = string.byte(luci.http.formvalue("letter") or "A", 1)
  81         letter = (letter == 35 or (letter >= 65 and letter <= 90)) and letter or 65
  82
  83         -- Search query
  84         local query = luci.http.formvalue("query")
  85         query = (query ~= '') and query or nil
  86
  87
  88         -- Modifying actions
  89         if submit then
  90                 -- Packets to be installed
  91                 local ninst = luci.http.formvalue("install")
  92                 local uinst = nil
  93
  94                 -- Install from URL
  95                 local url = luci.http.formvalue("url")
  96                 if url and url ~= '' then
  97                         uinst = url
  98                 end
  99
 100                 -- Do install
 101                 if ninst then
 102                         install[ninst], out, err = ipkg.install(ninst)
 103                         stdout[#stdout+1] = out
 104                         stderr[#stderr+1] = err
 105                         changes = true
 106                 end
 107
 108                 if uinst then
 109                         local pkg
 110                         for pkg in luci.util.imatch(uinst) do
 111                                 install[uinst], out, err = ipkg.install(pkg)
 112                                 stdout[#stdout+1] = out
 113                                 stderr[#stderr+1] = err
 114                                 changes = true
 115                         end
 116                 end
 117
 118                 -- Remove packets
 119                 local rem = luci.http.formvalue("remove")
 120                 if rem then
 121                         remove[rem], out, err = ipkg.remove(rem)
 122                         stdout[#stdout+1] = out
 123                         stderr[#stderr+1] = err
 124                         changes = true
 125                 end
 126
 127
 128                 -- Update all packets
 129                 update = luci.http.formvalue("update")
 130                 if update then
 131                         update, out, err = ipkg.update()
 132                         stdout[#stdout+1] = out
 133                         stderr[#stderr+1] = err
 134                 end
 135
 136
 137                 -- Upgrade all packets
 138                 upgrade = luci.http.formvalue("upgrade")
 139                 if upgrade then
 140                         upgrade, out, err = ipkg.upgrade()
 141                         stdout[#stdout+1] = out
 142                         stderr[#stderr+1] = err
 143                 end
 144         end
 145
 146
 147         -- List state
 148         local no_lists = true
 149         local old_lists = false
 150         if fs.access("/var/opkg-lists/") then
 151                 local list
 152                 for list in fs.dir("/var/opkg-lists/") do
 153                         no_lists = false
 154                         if (fs.stat("/var/opkg-lists/"..list, "mtime") or 0) < (os.time() - (24 * 60 * 60)) then
 155                                 old_lists = true
 156                                 break
 157                         end
 158                 end
 159         end
 160
 161
 162         luci.template.render("admin_system/packages", {
 163                 display   = display,
 164                 letter    = letter,
 165                 query     = query,
 166                 install   = install,
 167                 remove    = remove,
 168                 update    = update,
 169                 upgrade   = upgrade,
 170                 no_lists  = no_lists,
 171                 old_lists = old_lists,
 172                 stdout    = table.concat(stdout, ""),
 173                 stderr    = table.concat(stderr, "")
 174         })
 175
 176         -- Remove index cache
 177         if changes then
 178                 fs.unlink("/tmp/luci-indexcache")
 179         end
 180 end
 181
 182 local function image_supported(image)
 183         return (os.execute("sysupgrade -T %q >/dev/null" % image) == 0)
 184 end
 185
 186 local function image_checksum(image)
 187         return (luci.sys.exec("md5sum %q" % image):match("^([^%s]+)"))
 188 end
 189
 190 local function image_sha256_checksum(image)
 191         return (luci.sys.exec("sha256sum %q" % image):match("^([^%s]+)"))
 192 end
 193
 194 local function supports_sysupgrade()
 195         return nixio.fs.access("/lib/upgrade/platform.sh")
 196 end
 197
 198 local function supports_reset()
 199         return (os.execute([[grep -sq "^overlayfs:/overlay / overlay " /proc/mounts]]) == 0)
 200 end
 201
 202 local function storage_size()
 203         local size = 0
 204         if nixio.fs.access("/proc/mtd") then
 205                 for l in io.lines("/proc/mtd") do
 206                         local d, s, e, n = l:match('^([^%s]+)%s+([^%s]+)%s+([^%s]+)%s+"([^%s]+)"')
 207                         if n == "linux" or n == "firmware" then
 208                                 size = tonumber(s, 16)
 209                                 break
 210                         end
 211                 end
 212         elseif nixio.fs.access("/proc/partitions") then
 213                 for l in io.lines("/proc/partitions") do
 214                         local x, y, b, n = l:match('^%s*(%d+)%s+(%d+)%s+([^%s]+)%s+([^%s]+)')
 215                         if b and n and not n:match('[0-9]') then
 216                                 size = tonumber(b) * 1024
 217                                 break
 218                         end
 219                 end
 220         end
 221         return size
 222 end
 223
 224
 225 function action_flashops()
 226         --
 227         -- Overview
 228         --
 229         luci.template.render("admin_system/flashops", {
 230                 reset_avail   = supports_reset(),
 231                 upgrade_avail = supports_sysupgrade()
 232         })
 233 end
 234
 235 function action_sysupgrade()
 236         local fs = require "nixio.fs"
 237         local http = require "luci.http"
 238         local image_tmp = "/tmp/firmware.img"
 239
 240         local fp
 241         http.setfilehandler(
 242                 function(meta, chunk, eof)
 243                         if not fp and meta and meta.name == "image" then
 244                                 fp = io.open(image_tmp, "w")
 245                         end
 246                         if fp and chunk then
 247                                 fp:write(chunk)
 248                         end
 249                         if fp and eof then
 250                                 fp:close()
 251                         end
 252                 end
 253         )
 254
 255         if not luci.dispatcher.test_post_security() then
 256                 fs.unlink(image_tmp)
 257                 return
 258         end
 259
 260         --
 261         -- Cancel firmware flash
 262         --
 263         if http.formvalue("cancel") then
 264                 fs.unlink(image_tmp)
 265                 http.redirect(luci.dispatcher.build_url('admin/system/flashops'))
 266                 return
 267         end
 268
 269         --
 270         -- Initiate firmware flash
 271         --
 272         local step = tonumber(http.formvalue("step") or 1)
 273         if step == 1 then
 274                 if image_supported(image_tmp) then
 275                         luci.template.render("admin_system/upgrade", {
 276                                 checksum = image_checksum(image_tmp),
 277                                 sha256ch = image_sha256_checksum(image_tmp),
 278                                 storage  = storage_size(),
 279                                 size     = (fs.stat(image_tmp, "size") or 0),
 280                                 keep     = (not not http.formvalue("keep"))
 281                         })
 282                 else
 283                         fs.unlink(image_tmp)
 284                         luci.template.render("admin_system/flashops", {
 285                                 reset_avail   = supports_reset(),
 286                                 upgrade_avail = supports_sysupgrade(),
 287                                 image_invalid = true
 288                         })
 289                 end
 290         --
 291         -- Start sysupgrade flash
 292         --
 293         elseif step == 2 then
 294                 local keep = (http.formvalue("keep") == "1") and "" or "-n"
 295                 luci.template.render("admin_system/applyreboot", {
 296                         title = luci.i18n.translate("Flashing..."),
 297                         msg   = luci.i18n.translate("The system is flashing 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."),
 298                         addr  = (#keep > 0) and "192.168.1.1" or nil
 299                 })
 300                 fork_exec("sleep 1; killall dropbear uhttpd; sleep 1; /sbin/sysupgrade %s %q" %{ keep, image_tmp })
 301         end
 302 end
 303
 304 function action_backup()
 305         local reader = ltn12_popen("sysupgrade --create-backup - 2>/dev/null")
 306
 307         luci.http.header(
 308                 'Content-Disposition', 'attachment; filename="backup-%s-%s.tar.gz"' %{
 309                         luci.sys.hostname(),
 310                         os.date("%Y-%m-%d")
 311                 })
 312
 313         luci.http.prepare_content("application/x-targz")
 314         luci.ltn12.pump.all(reader, luci.http.write)
 315 end
 316
 317 function action_restore()
 318         local fs = require "nixio.fs"
 319         local http = require "luci.http"
 320         local archive_tmp = "/tmp/restore.tar.gz"
 321
 322         local fp
 323         http.setfilehandler(
 324                 function(meta, chunk, eof)
 325                         if not fp and meta and meta.name == "archive" then
 326                                 fp = io.open(archive_tmp, "w")
 327                         end
 328                         if fp and chunk then
 329                                 fp:write(chunk)
 330                         end
 331                         if fp and eof then
 332                                 fp:close()
 333                         end
 334                 end
 335         )
 336
 337         if not luci.dispatcher.test_post_security() then
 338                 fs.unlink(archive_tmp)
 339                 return
 340         end
 341
 342         local upload = http.formvalue("archive")
 343         if upload and #upload > 0 then
 344                 luci.template.render("admin_system/applyreboot")
 345                 os.execute("tar -C / -xzf %q >/dev/null 2>&1" % archive_tmp)
 346                 luci.sys.reboot()
 347                 return
 348         end
 349
 350         http.redirect(luci.dispatcher.build_url('admin/system/flashops'))
 351 end
 352
 353 function action_reset()
 354         if supports_reset() then
 355                 luci.template.render("admin_system/applyreboot", {
 356                         title = luci.i18n.translate("Erasing..."),
 357                         msg   = luci.i18n.translate("The system is erasing the configuration partition now and will reboot itself when finished."),
 358                         addr  = "192.168.1.1"
 359                 })
 360
 361                 fork_exec("sleep 1; killall dropbear uhttpd; sleep 1; jffs2reset -y && reboot")
 362                 return
 363         end
 364
 365         http.redirect(luci.dispatcher.build_url('admin/system/flashops'))
 366 end
 367
 368 function action_passwd()
 369         local p1 = luci.http.formvalue("pwd1")
 370         local p2 = luci.http.formvalue("pwd2")
 371         local stat = nil
 372
 373         if p1 or p2 then
 374                 if p1 == p2 then
 375                         stat = luci.sys.user.setpasswd("root", p1)
 376                 else
 377                         stat = 10
 378                 end
 379         end
 380
 381         luci.template.render("admin_system/passwd", {stat=stat})
 382 end
 383
 384 function action_reboot()
 385         luci.sys.reboot()
 386 end
 387
 388 function fork_exec(command)
 389         local pid = nixio.fork()
 390         if pid > 0 then
 391                 return
 392         elseif pid == 0 then
 393                 -- change to root dir
 394                 nixio.chdir("/")
 395
 396                 -- patch stdin, out, err to /dev/null
 397                 local null = nixio.open("/dev/null", "w+")
 398                 if null then
 399                         nixio.dup(null, nixio.stderr)
 400                         nixio.dup(null, nixio.stdout)
 401                         nixio.dup(null, nixio.stdin)
 402                         if null:fileno() > 2 then
 403                                 null:close()
 404                         end
 405                 end
 406
 407                 -- replace with target command
 408                 nixio.exec("/bin/sh", "-c", command)
 409         end
 410 end
 411
 412 function ltn12_popen(command)
 413
 414         local fdi, fdo = nixio.pipe()
 415         local pid = nixio.fork()
 416
 417         if pid > 0 then
 418                 fdo:close()
 419                 local close
 420                 return function()
 421                         local buffer = fdi:read(2048)
 422                         local wpid, stat = nixio.waitpid(pid, "nohang")
 423                         if not close and wpid and stat == "exited" then
 424                                 close = true
 425                         end
 426
 427                         if buffer and #buffer > 0 then
 428                                 return buffer
 429                         elseif close then
 430                                 fdi:close()
 431                                 return nil
 432                         end
 433                 end
 434         elseif pid == 0 then
 435                 nixio.dup(fdo, nixio.stdout)
 436                 fdi:close()
 437                 fdo:close()
 438                 nixio.exec("/bin/sh", "-c", command)
 439         end
 440 end