X-Git-Url: https://git.archive.openwrt.org/?p=project%2Fluci.git;a=blobdiff_plain;f=modules%2Fadmin-full%2Fluasrc%2Fcontroller%2Fadmin%2Fsystem.lua;h=d3afead402e81b0d2ee66594fa56651f84b33a3f;hp=862a741cded4d150cb8f977d717aaa83d040fe88;hb=cde6e129d4759e7a5089b8779c416b1a85206217;hpb=97ff4a156ff1aa6a4d2ab65ab13d12b4eb2651e2

diff --git a/modules/admin-full/luasrc/controller/admin/system.lua b/modules/admin-full/luasrc/controller/admin/system.lua
index 862a741cd..d3afead40 100644
--- a/modules/admin-full/luasrc/controller/admin/system.lua
+++ b/modules/admin-full/luasrc/controller/admin/system.lua
@@ -2,6 +2,7 @@
 LuCI - Lua Configuration Interface
 
 Copyright 2008 Steven Barth <steven@midlink.org>
+Copyright 2008-2009 Jo-Philipp Wich <xm@subsignal.org>
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -14,143 +15,134 @@ $Id$
 module("luci.controller.admin.system", package.seeall)
 
 function index()
-	luci.i18n.loadc("admin-core")
+	luci.i18n.loadc("base")
 	local i18n = luci.i18n.translate
 	
-	entry({"admin", "system"}, template("admin_system/index"), i18n("system", "System"), 30)
-	entry({"admin", "system", "packages"}, call("action_packages"), i18n("a_s_packages", "Paketverwaltung"), 10)
-	entry({"admin", "system", "packages", "ipkg"}, call("action_ipkg"), i18n("a_s_p_ipkg", "IPKG-Konfiguration"))
-	entry({"admin", "system", "passwd"}, call("action_passwd"), i18n("a_s_changepw", "Passwort ändern"), 20)
-	entry({"admin", "system", "sshkeys"}, call("action_sshkeys"), i18n("a_s_sshkeys", "SSH-Schlüssel"), 30)
-	entry({"admin", "system", "system"}, cbi("admin_system/system"), i18n("system", "System"), 40)
-	entry({"admin", "system", "fstab"}, cbi("admin_system/fstab"), i18n("a_s_fstab", "Einhängepunkte"), 50)
-	entry({"admin", "system", "upgrade"}, call("action_upgrade"), i18n("a_s_flash", "Firmwareupgrade"), 60)
-	entry({"admin", "system", "reboot"}, call("action_reboot"), i18n("reboot", "Neu starten"), 70)
-end
+	entry({"admin", "system"}, alias("admin", "system", "system"), i18n("System"), 30).index = true
+	entry({"admin", "system", "system"}, cbi("admin_system/system"), i18n("System"), 1)
+	entry({"admin", "system", "packages"}, call("action_packages"), i18n("Software"), 10)
+	entry({"admin", "system", "packages", "ipkg"}, form("admin_system/ipkg"))
+	entry({"admin", "system", "passwd"}, form("admin_system/passwd"), i18n("Admin Password"), 20)
+	entry({"admin", "system", "sshkeys"}, form("admin_system/sshkeys"), i18n("<abbr title=\"Secure Shell\">SSH</abbr>-Keys"), 30)
+	entry({"admin", "system", "processes"}, form("admin_system/processes"), i18n("Processes"), 45)
 
-function action_editor()
-	local file = luci.http.formvalue("file", "")
-	local data = luci.http.formvalue("data")
-	local err  = nil
-	local msg  = nil
-	local stat = true
-	
-	if file and data then
-		stat, err = luci.fs.writefile(file, data)
-	end
-	
-	if not stat then
-		err = luci.util.split(err, " ")
-		table.remove(err, 1)
-		msg = table.concat(err, " ")
-	end
-	
-	local cnt, err = luci.fs.readfile(file)
-	if cnt then
-		cnt = luci.util.pcdata(cnt)
+	if nixio.fs.access("/etc/config/fstab") then
+		entry({"admin", "system", "fstab"}, cbi("admin_system/fstab"), i18n("Mount Points"), 50)
 	end
-	luci.template.render("admin_system/editor", {fn=file, cnt=cnt, msg=msg})	
-end
 
-function action_ipkg()
-	local file = "/etc/ipkg.conf"
-	local data = luci.http.formvalue("data")
-	local stat = nil
-	local err  = nil
-	
-	if data then
-		stat, err = luci.fs.writefile(file, data)
-	end	
-	
-	local cnt  = luci.fs.readfile(file)	
-	if cnt then
-		cnt = luci.util.pcdata(cnt)
+	if nixio.fs.access("/sys/class/leds") then
+		entry({"admin", "system", "leds"}, cbi("admin_system/leds"), i18n("<abbr title=\"Light Emitting Diode\">LED</abbr> Configuration"), 60)
 	end
-	
-	luci.template.render("admin_system/ipkg", {cnt=cnt, msg=err})	
+
+	entry({"admin", "system", "backup"}, call("action_backup"), i18n("Backup / Restore"), 70)
+	entry({"admin", "system", "upgrade"}, call("action_upgrade"), i18n("Flash Firmware"), 80)
+	entry({"admin", "system", "reboot"}, call("action_reboot"), i18n("Reboot"), 90)
 end
 
 function action_packages()
 	local ipkg = require("luci.model.ipkg")
-	local void = nil
 	local submit = luci.http.formvalue("submit")
-	
-	
+	local changes = false
+	local install = { }
+	local remove  = { }
+
 	-- Search query
 	local query = luci.http.formvalue("query")
 	query = (query ~= '') and query or nil
 	
 	
 	-- Packets to be installed
-	local install = submit and luci.http.formvaluetable("install")
-	
+	local ninst = submit and luci.http.formvalue("install")
+	local uinst = nil	
+
 	-- Install from URL
 	local url = luci.http.formvalue("url")
 	if url and url ~= '' and submit then
-		if not install then
-			install = {}
-		end
-		install[url] = 1
+		uinst = url
 	end
-	
+
 	-- Do install
-	if install then
-		for k, v in pairs(install) do
-			void, install[k] = ipkg.install(k)
-		end
+	if ninst then
+		_, install[ninst] = ipkg.install(ninst)
+		changes = true
 	end
-	
-	
+
+	if uinst then
+		_, install[uinst] = ipkg.install(uinst)
+		changes = true
+	end
+
 	-- Remove packets
-	local remove = submit and luci.http.formvaluetable("remove")
-	if remove then	
-		for k, v in pairs(remove) do
-			void, remove[k] = ipkg.remove(k)
-		end	
+	local rem = submit and luci.http.formvalue("remove")
+	if rem then	
+		_, remove[rem] = ipkg.remove(rem)
+		changes = true
 	end
-	
+
 	
 	-- Update all packets
 	local update = luci.http.formvalue("update")
 	if update then
-		void, update = ipkg.update()
+		_, update = ipkg.update()
 	end
 	
 	
 	-- Upgrade all packets
 	local upgrade = luci.http.formvalue("upgrade")
 	if upgrade then
-		void, upgrade = ipkg.upgrade()
+		_, upgrade = ipkg.upgrade()
 	end
+
 	
-	
-	-- Package info
-	local info = luci.model.ipkg.info(query)
-	info = info or {}
-	local pkgs = {}
-	
-	-- Sort after status and name
-	for k, v in pairs(info) do
-		local x = 0
-		for i, j in pairs(pkgs) do
-			local vins = (v.Status and v.Status.installed)
-			local jins = (j.Status and j.Status.installed)
-			if vins ~= jins then
-				if vins then
-					break
-				end
-			else
-				if j.Package > v.Package then
-					break
-				end
+	luci.template.render("admin_system/packages", {
+		query=query, install=install, remove=remove, update=update, upgrade=upgrade
+	})
+ 
+	-- Remove index cache
+	if changes then
+		nixio.fs.unlink("/tmp/luci-indexcache")
+	end	
+end
+
+function action_backup()
+	local reset_avail = os.execute([[grep '"rootfs_data"' /proc/mtd >/dev/null 2>&1]]) == 0
+	local restore_cmd = "tar -xzC/ >/dev/null 2>&1"
+	local backup_cmd  = "tar -cz %s 2>/dev/null"
+
+	local restore_fpi 
+	luci.http.setfilehandler(
+		function(meta, chunk, eof)
+			if not restore_fpi then
+				restore_fpi = io.popen(restore_cmd, "w")
+			end
+			if chunk then
+				restore_fpi:write(chunk)
+			end
+			if eof then
+				restore_fpi:close()
 			end
-			x = i
 		end
-		table.insert(pkgs, x+1, v)
-	end 
+	)
+		  
+	local upload = luci.http.formvalue("archive")
+	local backup = luci.http.formvalue("backup")
+	local reset  = reset_avail and luci.http.formvalue("reset")
 	
-	luci.template.render("admin_system/packages", {pkgs=pkgs, query=query,
-	 install=install, remove=remove, update=update, upgrade=upgrade})	
+	if upload and #upload > 0 then
+		luci.template.render("admin_system/applyreboot")
+		luci.sys.reboot()
+	elseif backup then
+		local reader = ltn12_popen(backup_cmd:format(_keep_pattern()))
+		luci.http.header('Content-Disposition', 'attachment; filename="backup-%s-%s.tar.gz"' % {
+			luci.sys.hostname(), os.date("%Y-%m-%d")})
+		luci.http.prepare_content("application/x-targz")
+		luci.ltn12.pump.all(reader, luci.http.write)
+	elseif reset then
+		luci.template.render("admin_system/applyreboot")
+		luci.util.exec("mtd -r erase rootfs_data")
+	else
+		luci.template.render("admin_system/backup", {reset_avail = reset_avail})
+	end
 end
 
 function action_passwd()
@@ -177,45 +169,174 @@ function action_reboot()
 	end
 end
 
-function action_sshkeys()
-	local file = "/etc/dropbear/authorized_keys"
-	local data = luci.http.formvalue("data")
-	local stat = nil
-	local err  = nil
+function action_upgrade()
+	require("luci.model.uci")
+
+	local tmpfile = "/tmp/firmware.img"
 	
-	if data then
-		stat, err = luci.fs.writefile(file, data)
-	end	
+	local function image_supported()
+		-- XXX: yay...
+		return ( 0 == os.execute(
+			". /etc/functions.sh; " ..
+			"include /lib/upgrade; " ..
+			"platform_check_image %q >/dev/null"
+				% tmpfile
+		) )
+	end
 	
-	local cnt  = luci.fs.readfile(file)	
-	if cnt then
-		cnt = luci.util.pcdata(cnt)
+	local function image_checksum()
+		return (luci.sys.exec("md5sum %q" % tmpfile):match("^([^%s]+)"))
 	end
 	
-	luci.template.render("admin_system/sshkeys", {cnt=cnt, msg=err})	
-end
+	local function storage_size()
+		local size = 0
+		if nixio.fs.access("/proc/mtd") then
+			for l in io.lines("/proc/mtd") do
+				local d, s, e, n = l:match('^([^%s]+)%s+([^%s]+)%s+([^%s]+)%s+"([^%s]+)"')
+				if n == "linux" then
+					size = tonumber(s, 16)
+					break
+				end
+			end
+		elseif nixio.fs.access("/proc/partitions") then
+			for l in io.lines("/proc/partitions") do
+				local x, y, b, n = l:match('^%s*(%d+)%s+(%d+)%s+([^%s]+)%s+([^%s]+)')
+				if b and n and not n:match('[0-9]') then
+					size = tonumber(b) * 1024
+					break
+				end
+			end
+		end
+		return size
+	end
 
-function action_upgrade()
-	require("luci.model.uci")
-	local ret  = nil
-	local plat = luci.fs.mtime("/lib/upgrade/platform.sh")
+
+	-- Install upload handler
+	local file
+	luci.http.setfilehandler(
+		function(meta, chunk, eof)
+			if not nixio.fs.access(tmpfile) and not file and chunk and #chunk > 0 then
+				file = io.open(tmpfile, "w")
+			end
+			if file and chunk then
+				file:write(chunk)
+			end
+			if file and eof then
+				file:close()
+			end
+		end
+	)
+
+
+	-- Determine state
+	local keep_avail   = true
+	local step         = tonumber(luci.http.formvalue("step") or 1)
+	local has_image    = nixio.fs.access(tmpfile)
+	local has_support  = image_supported()
+	local has_platform = nixio.fs.access("/lib/upgrade/platform.sh")
+	local has_upload   = luci.http.formvalue("image")
 	
-	local image   = luci.http.upload("image")
-	local keepcfg = luci.http.formvalue("keepcfg")
+	-- This does the actual flashing which is invoked inside an iframe
+	-- so don't produce meaningful errors here because the the 
+	-- previous pages should arrange the stuff as required.
+	if step == 4 then
+		if has_platform and has_image and has_support then
+			-- Mimetype text/plain
+			luci.http.prepare_content("text/plain")
+			luci.http.write("Starting luci-flash...\n")
+
+			-- Now invoke sysupgrade
+			local keepcfg = keep_avail and luci.http.formvalue("keepcfg") == "1"
+			local flash = ltn12_popen("/sbin/luci-flash %s %q" %{
+				keepcfg and "-k %q" % _keep_pattern() or "", tmpfile
+			})
+
+			luci.ltn12.pump.all(flash, luci.http.write)
+
+			-- Make sure the device is rebooted
+			luci.sys.reboot()
+		end
+
+
+	--
+	-- This is step 1-3, which does the user interaction and
+	-- image upload.
+	--
+
+	-- Step 1: file upload, error on unsupported image format
+	elseif not has_image or not has_support or step == 1 then
+		-- If there is an image but user has requested step 1
+		-- or type is not supported, then remove it.
+		if has_image then
+			nixio.fs.unlink(tmpfile)
+		end
+			
+		luci.template.render("admin_system/upgrade", {
+			step=1,
+			bad_image=(has_image and not has_support or false),
+			keepavail=keep_avail,
+			supported=has_platform
+		} )
+
+	-- Step 2: present uploaded file, show checksum, confirmation
+	elseif step == 2 then
+		luci.template.render("admin_system/upgrade", {
+			step=2,
+			checksum=image_checksum(),
+			filesize=nixio.fs.stat(tmpfile).size,
+			flashsize=storage_size(),
+			keepconfig=(keep_avail and luci.http.formvalue("keepcfg") == "1")
+		} )
 	
-	if plat and image then
-		local kpattern = nil
-		if keepcfg then
-			local files = luci.model.uci.get_all("luci", "flash_keep")
-			if files.luci and files.luci.flash_keep then
-				kpattern = ""
-				for k,v in pairs(files.luci.flash_keep) do
-					kpattern = kpattern .. " " ..  v
-				end
+	-- Step 3: load iframe which calls the actual flash procedure
+	elseif step == 3 then
+		luci.template.render("admin_system/upgrade", {
+			step=3,
+			keepconfig=(keep_avail and luci.http.formvalue("keepcfg") == "1")
+		} )
+	end	
+end
+
+function _keep_pattern()
+	local kpattern = ""
+	local files = luci.model.uci.cursor():get_all("luci", "flash_keep")
+	if files then
+		kpattern = ""
+		for k, v in pairs(files) do
+			if k:sub(1,1) ~= "." and nixio.fs.glob(v)() then
+				kpattern = kpattern .. " " ..  v
 			end
 		end
-		ret = luci.sys.flash(image, kpattern)
 	end
-	
-	luci.template.render("admin_system/upgrade", {sysupgrade=plat, ret=ret})
-end
\ No newline at end of file
+	return kpattern
+end
+
+function ltn12_popen(command)
+
+	local fdi, fdo = nixio.pipe()
+	local pid = nixio.fork()
+
+	if pid > 0 then
+		fdo:close()
+		local close
+		return function()
+			local buffer = fdi:read(2048)
+			local wpid, stat = nixio.waitpid(pid, "nohang")
+			if not close and wpid and stat == "exited" then
+				close = true
+			end
+
+			if buffer and #buffer > 0 then
+				return buffer
+			elseif close then
+				fdi:close()
+				return nil
+			end
+		end
+	elseif pid == 0 then
+		nixio.dup(fdo, nixio.stdout)
+		fdi:close()
+		fdo:close()
+		nixio.exec("/bin/sh", "-c", command)
+	end
+end