20060303a

1 files changed, 188 insertions, 0 deletions

diff --git a/appl/cmd/mpc/qflash.b b/appl/cmd/mpc/qflash.b
new file mode 100644
index 00000000..13c77ce8
--- /dev/null
+++ b/appl/cmd/mpc/qflash.b
@@ -0,0 +1,188 @@
+implement Writeflash;
+
+include "sys.m";
+	sys: Sys;
+
+include "draw.m";
+
+include "string.m";
+	str: String;
+
+Writeflash: module
+{
+	init:	fn(nil: ref Draw->Context, args: list of string);
+};
+
+Region: adt {
+	base:	int;
+	limit:	int;
+};
+
+# could come from file or #F/flash/flashctl
+FLASHSEG: con 256*1024;
+kernelregion := Region(FLASHSEG, FLASHSEG+2*FLASHSEG);
+bootregion := Region(0, FLASHSEG);
+
+stderr: ref Sys->FD;
+prog := "qflash";
+damaged := 0;
+
+usage()
+{
+	sys->fprint(stderr, "Usage: %s [-b] [-o offset] [-f flashdev] file\n", prog);
+	exit;
+}
+
+err(s: string)
+{
+	sys->fprint(stderr, "%s: %s", prog, s);
+	if(!damaged)
+		sys->fprint(stderr, "; flash not modified\n");
+	else
+		sys->fprint(stderr, "; flash might now be invalid\n");
+	exit;
+}
+
+init(nil: ref Draw->Context, args: list of string)
+{
+	sys = load Sys Sys->PATH;
+	sys->pctl(Sys->FORKFD|Sys->NEWPGRP, nil);
+	stderr = sys->fildes(2);
+	if(args != nil){
+		prog = hd args;
+		args = tl args;
+	}
+	str = load String String->PATH;
+	if(str == nil)
+		err(sys->sprint("can't load %s: %r", String->PATH));
+	region := kernelregion;
+	flash := "#F/flash/flash";
+	offset := 0;
+	save := 0;
+	
+	for(; args != nil && (hd args)[0] == '-'; args = tl args)
+		case hd args {
+		"-b" =>
+			region = bootregion;
+			offset = 16r100 - 8*4;	# size of exec header
+			save = 1;
+		"-h" =>
+			region.limit += FLASHSEG;
+		"-f" =>
+			if(tl args == nil)
+				usage();
+			flash = hd args;
+			args = tl args;
+		"-o" =>
+			if(tl args == nil)
+				usage();
+			args = tl args;
+			s := hd args;
+			v: int;
+			rs: string;
+			if(str->prefix("16r", s))
+				(v, rs) = str->toint(s[3:], 16);
+			else if(str->prefix("0x", s))
+				(v, rs) = str->toint(s[2:], 16);
+			else if(str->prefix("0", s))
+				(v, rs) = str->toint(s[1:], 8);
+			else
+				(v, rs) = str->toint(s, 10);
+			if(v < 0 || len rs != 0)
+				err(sys->sprint("bad offset: %s", s));
+			offset = v;
+		"-s" =>
+			save = 1;
+		* =>
+			usage();
+		}
+	if(args == nil)
+		usage();
+	fname := hd args;
+	fd := sys->open(fname, Sys->OREAD);
+	if(fd == nil)
+		err(sys->sprint("can't open %s: %r", fname));
+	(r, dir) := sys->fstat(fd);
+	if(r < 0)
+		err(sys->sprint("can't stat %s: %r", fname));
+	length := int dir.length;
+	avail := region.limit - (region.base+offset);
+	if(length > avail)
+		err(sys->sprint("%s contents %ud bytes, exceeds flash region %ud bytes", fname, length, avail));
+	# check fname's contents...
+	where := region.base+offset;
+	saved: list of (int, array of byte);
+	if(save){
+		saved = saveflash(flash, region.base, where) :: saved;
+		saved = saveflash(flash, where+length, region.limit) :: saved;
+	}
+	for(i := (region.base+offset)/FLASHSEG; i < region.limit/FLASHSEG; i++)
+		erase(flash, i);
+	out := sys->open(flash, Sys->OWRITE);
+	if(out == nil)
+		err(sys->sprint("can't open %s for writing: %r", flash));
+	if(sys->seek(out, big where, 0) != big where)
+		err(sys->sprint("can't seek to #%6.6ux on flash: %r", where));
+	if(length)
+		sys->print("writing %ud bytes to %s at #%6.6ux\n", length, flash, where);
+	buf := array[Sys->ATOMICIO] of byte;
+	total := 0;
+	while((n := sys->read(fd, buf, len buf)) > 0) {
+		if(total+n > avail)
+			err(sys->sprint("file %s too big for region of %ud bytes", fname, avail));
+		r = sys->write(out, buf, n);
+		damaged = 1;
+		if(r != n){
+			if(r < 0)
+				err(sys->sprint("error writing %s at byte %ud: %r", flash, total));
+			else
+				err(sys->sprint("short write on %s at byte %ud", flash, total));
+		}
+		total += n;
+	}
+	if(n < 0)
+		err(sys->sprint("error reading %s: %r", fname));
+	sys->print("wrote %ud bytes from %s to flash %s (#%6.6ux-#%6.6ux)\n", total, fname, flash, region.base, region.base+total);
+	for(l := saved; l != nil; l = tl l){
+		(addr, data) := hd l;
+		n = len data;
+		if(n == 0)
+			continue;
+		sys->print("restoring %ud bytes at #%6.6ux\n", n, addr);
+		if(sys->seek(out, big addr, 0) != big addr)
+			err(sys->sprint("can't seek to #%6.6ux on %s: %r", addr, flash));
+		r = sys->write(out, data, n);
+		if(r < 0)
+			err(sys->sprint("error writing %s: %r", flash));
+		else if(r != n)
+			err(sys->sprint("short write on %s at byte %ud/%ud", flash, r, n));
+		else
+			sys->print("restored %ud bytes at #%6.6ux\n", n, addr);
+	}
+}
+
+erase(flash: string, seg: int)
+{
+	ctl := sys->open(flash+"ctl", Sys->OWRITE);
+	if(ctl == nil)
+		err(sys->sprint("can't open %sctl: %r\n", flash));
+	if(sys->fprint(ctl, "erase %ud", seg*FLASHSEG) < 0)
+		err(sys->sprint("can't erase flash %s segment %d: %r\n", flash, seg));
+}
+
+saveflash(flash: string, base: int, limit: int): (int, array of byte)
+{
+	fd := sys->open(flash, Sys->OREAD);
+	if(fd == nil)
+		err(sys->sprint("can't open %s for reading: %r", flash));
+	nb := limit - base;
+	if(nb <= 0)
+		return (base, nil);
+	if(sys->seek(fd, big base, 0) != big base)
+		err(sys->sprint("can't seek to #%6.6ux to save flash contents: %r", base));
+	saved := array[nb] of byte;
+	if(sys->read(fd, saved, len saved) != len saved)
+		err(sys->sprint("can't read flash #%6.6ux to #%6.6ux: %r", base, limit));
+	sys->print("saved %ud bytes at #%6.6ux\n", len saved, base);
+	return (base, saved);
+}