From 3ab109da956268e5cb5e23a30f9a7ee63ca1385f Mon Sep 17 00:00:00 2001 From: Steven Barth Date: Wed, 17 Sep 2008 20:44:12 +0000 Subject: [PATCH] Initial implementation of OpenWrt mtd writer --- libs/sys/luasrc/sys/mtdow.lua | 192 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 192 insertions(+) create mode 100644 libs/sys/luasrc/sys/mtdow.lua diff --git a/libs/sys/luasrc/sys/mtdow.lua b/libs/sys/luasrc/sys/mtdow.lua new file mode 100644 index 000000000..851c1bee2 --- /dev/null +++ b/libs/sys/luasrc/sys/mtdow.lua @@ -0,0 +1,192 @@ +--[[ +LuCI - Lua Configuration Interface + +Copyright 2008 Steven Barth +Copyright 2008 Jo-Philipp Wich + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + +http://www.apache.org/licenses/LICENSE-2.0 + +$Id$ +]]-- + +local io = require "io" +local os = require "os" +local fs = require "luci.fs" +local util = require "luci.util" +local ltn12 = require "luci.ltn12" +local posix = require "posix" + +local type, assert, error = type, assert, error + +module "luci.sys.mtdow" + +local WRITE_IMAGE = 0 +local WRITE_COMBINED = 1 +local WRITE_EMULATED = 2 + +Writer = util.class() + +-- x86 +EmulatedWriter = util.class(Writer) +EmulatedWriter.blocks = { + image = { + magic = "eb48", + device = "/dev/hda", + write = WRITE_SEPARATELY + } +} + +-- Broadcom +TRXWriter = util.class(Writer) +TRXWriter.blocks = { + image = { + magic = {"4844", "5735"}, + device = "linux", + write = WRITE_COMBINED + } +} + +-- Magicbox +CommonWriter = util.class(Writer) +CommonWriter.blocks = { + image = { + device = "linux", + write = WRITE_COMBINED + } +} + +-- Atheros +RedWriter = util.class(Writer) +RedWriter.blocks = { + kernel = { + device = "vmlinux.bin.l7", + write = WRITE_IMAGE + }, + rootfs = { + device = "rootfs", + write = WRITE_COMBINED + } +} + +function EmulatedWriter.write_block(self, name, imagestream, appendpattern) + if appendpattern then + os.execute("grep rootfs /proc/mtd >/dev/null || " + .. "{ echo /dev/hda2,65536,rootfs > " + .. "/sys/module/block2mtd/parameters/block2mtd }") + end + return Writer.write_block(self, name, imagestream, appendpattern) +end + + +Writer.MTD = "/sbin/mtd" +Writer.SAFEMTD = "/tmp/mtd" +Writer.IMAGEFIFO = "/tmp/mtdimage.fifo" + +function Writer.write_block(self, name, imagestream, appendfile) + assert(self.blocks[name], "Undefined block: " % name) + local block = self.blocks[name] + local device = block.device + device = fs.stat(device) and device or self:_find_mtdblock(device) + assert(device, "Unable to determine device file") + if block.magic then + imagestream = self:_check_magic(imagestream, block.magic) + end + assert(imagestream, "Invalid image file") + + if appendfile then + if block.write == WRITE_COMBINED then + return (self:_write_combined(device, imagestream, appendfile) == 0) + elseif block.write == WRITE_EMULATED then + return (self:_write_emulated(device, imagestream, appendfile) == 0) + else + error("Appending is not supported for selected platform.") + end + else + return (self:_write_memory(device, imagestream) == 0) + end +end + +function Writer._check_magic(self, imagestream, magic) + magic = type(magic) == "table" and magic or {magic} + + local block = imagestream() + assert(block, "Invalid image stream") + local cm = "%x%x" % {block:byte(1), block:byte(2)} + + if util.contains(magic, cm) then + return ltn12.source.cat(ltn12.source.string(block), imagestream) + end +end + +function Writer._find_mtdblock(self, mtdname) + local k + local prefix = "/dev/mtdblock" + prefix = prefix .. (fs.stat(prefix) and "/" or "") + + for l in io.lines("/proc/mtd") do + local k = l:match('([%w-_]+):.*-"%s"' % mtdname) + if k then return prefix..k end + end +end + +function Write._write_emulated(self, devicename, imagestream, appendfile) + local stat = (self:_write_memory(device, imagestream) == 0) + stat = stat and (self:_refresh_block("rootfs") == 0) + local squash = self:_find_mtdblock("rootfs_data") + if squash then + stat = stat and (self:_append("rootfs_data", imagestream, true) == 0) + else + stat = stat and (self:_append("rootfs", imagestream) == 0) + end + return stat +end + +function Writer._write_memory(self, devicename, imagestream) + local devicestream = ltn12.sink.file(io.open(devicename, "w")) + local stat, err = ltn12.pump.all(imagestream, devicestream) + if stat then + return os.execute("sync") + end +end + +function Writer._write_combined(self, devicename, imagestream, appendfile) + assert(fs.copy(self.MTD, self.SAFEMTD), "Unable to copy mtd writer") + assert(posix.mkfifo(self.IMAGEFIFO), "Unable to create image pipe") + + local imagefifo = io.open(self.IMAGEFIFO, "w") + + assert(imagefifo, "Unable to open image pipe") + + local imageproc = posix.fork() + assert(imageproc ~= -1, "Unable to fork()") + if imageproc == 0 then + ltn12.pump.all(imagestream, ltn12.sink.file(imagefifo)) + os.exit(0) + end + + return os.execute( + "%s -j '%s' write '%s' '%s'" % { + self.SAFEMTD, appendfile, devicename, self.IMAGEFIFO + } + ) +end + +function Writer._refresh_block(self, devicename) + assert(fs.copy(self.MTD, self.SAFEMTD), "Unable to copy mtd writer") + return os.execute("%s refresh '%s'" % {self.SAFEMTD, devicename}) +end + +function Writer._append(self, devicename, appendfile, erase) + assert(fs.copy(self.MTD, self.SAFEMTD), "Unable to copy mtd writer") + erase = erase and ("-e '%s' " % devicename) or '' + + return os.execute( + "%s %s jffs2write '%s' '%s'" % { + self.SAFEMTD, erase, appendfile, devicename + } + ) +end \ No newline at end of file -- 2.11.0