path: root/appl/cmd/auxi/mangaload.b

implement Mangaload;

# to do:
#	- set arp entry based on /lib/ndb if necessary

include "sys.m";
	sys: Sys;

include "draw.m";

include "ip.m";
	ip: IP;
	IPaddr: import ip;

include "timers.m";
	timers: Timers;
	Timer: import timers;

include "dial.m";
	dial: Dial;

include "arg.m";

Mangaload: module
{
	init:	fn(nil: ref Draw->Context, nil: list of string);
};

# manga parameters
FlashBlocksize: con 16r10000;
FlashSize: con 16r400000;	# 4meg for now
FlashUserArea: con 16r3C0000;

# magic values
FooterOffset: con 16rFFEC;
FooterSig: con 16rA0FFFF9F;	# ARM flash library
FileInfosize: con 64;
FileNamesize: con FileInfosize - 3*4;	# x, y, z
Packetdatasize: con 1500-28;	# ether data less IP + ICMP header
RequestTimeout: con 500;
Probecount: con 10;	# query unit every so many packets

# manga uses extended TFTP ops in ICMP InfoRequest packets
Tftp_Req: con 0;
Tftp_Read: con 1;
Tftp_Write: con 2;
Tftp_Data: con 3;
Tftp_Ack: con 4;
Tftp_Error: con 5;
Tftp_Last: con 6;

Icmp: adt
{
	ttl:	int;	# time to live
	src:	IPaddr;
	dst:	IPaddr;
	ptype:	int;
	code:	int;
	id:	int;
	seq:	int;
	data:	array of byte;
	munged:	int;	# packet received but corrupt

	unpack:	fn(b: array of byte): ref Icmp;
};

# ICMP packet types
EchoReply: con 0;
Unreachable: con 3;
SrcQuench: con 4;
EchoRequest: con 8;
TimeExceed: con 11;
Timestamp: con 13;
TimestampReply: con 14;
InfoRequest: con 15;
InfoReply: con 16;

Nmsg: con 32;
Interval: con 1000;	# ms

debug := 0;
flashblock := 1;	# never 0, that's the boot firmware
maxfilesize := 8*FlashBlocksize;
flashlim := FlashSize/FlashBlocksize;
loadinitrd := 0;
maxlen := 512*1024;
mypid := 0;
Datablocksize: con 4096;

init(nil: ref Draw->Context, args: list of string)
{
	sys = load Sys Sys->PATH;
	timers = load Timers Timers->PATH;
	dial = load Dial Dial->PATH;
	ip = load IP IP->PATH;
	ip->init();


	arg := load Arg Arg->PATH;
	arg->init(args);
	arg->setusage("mangaload [-48dr] destination file");
	while((o := arg->opt()) != 0)
		case o {
		'4' =>
			flashlim = 4*1024*1024/FlashBlocksize;
		'8' =>
			flashlim = 8*1024*1024/FlashBlocksize;
		'r' =>
			loadinitrd = 1;
			flashblock = 9;
			if(flashlim > 4*1024*1024/FlashBlocksize)
				maxfilesize = 113*FlashBlocksize;
			else
				maxfilesize = 50*FlashBlocksize;
		'd' =>
			debug++;
		}
	args = arg->argv();
	if(len args != 2)
		arg->usage();
	arg = nil;

	sys->pctl(Sys->NEWPGRP|Sys->FORKFD, nil);

	filename := hd tl args;
	fd := sys->open(filename, Sys->OREAD);
	if(fd == nil){
		sys->fprint(sys->fildes(2), "mangaload: can't open %s: %r\n", filename);
		raise "fail:open";
	}
	(ok, d) := sys->fstat(fd);
	if(ok < 0){
		sys->fprint(sys->fildes(2), "mangaload: can't stat %s: %r\n", filename);
		raise "fail:stat";
	}
	if(d.length > big maxfilesize){
		sys->fprint(sys->fildes(2), "mangaload: file %s too long (must not exceed %d bytes)\n",
			filename, maxfilesize);
		raise "fail:size";
	}
	filesize := int d.length;

	port := sys->sprint("%d", 16r8695);
	addr := dial->netmkaddr(hd args, "icmp", port);
	c := dial->dial(addr, port);
	if(c == nil){
		sys->fprint(sys->fildes(2), "mangaload: can't dial %s: %r\n", addr);
		raise "fail:dial";
	}
	
	tpid := timers->init(20);

	pids := chan of int;
	replies := chan [2] of ref Icmp;
	spawn reader(c.dfd, replies, pids);
	rpid := <-pids;

	flashoffset := flashblock * FlashBlocksize;

	# file name first
	bname := array of byte filename;
	l := len bname;
	buf := array[Packetdatasize] of byte;
	ip->put4(buf, 0, filesize);
	ip->put4(buf, 4, l);
	buf[8:] = bname;
	l += 2*4;
	buf[l++] = byte 0;
	ip->put4(buf, l, flashoffset);
	l += 4;
	{
		if(send(c.dfd, buf[0:l], Tftp_Write, 0) < 0)
			senderr();
		(op, iseq, data) := recv(replies, 400);
		sys->print("initial reply: %d %d\n", op, iseq);
		if(op != Tftp_Ack){
			why := "no response";
			if(op == Tftp_Error)
				why = "manga cannot receive file";
			sys->fprint(sys->fildes(2), "mangaload: %s\n", why);
			raise "fail:error";
		}
		sys->print("sending %s size %d at address %d (0x%x)\n", filename, filesize, flashoffset, flashoffset);
		seq := 1;
		nsent := 0;
		last := 0;
		while((n := sys->read(fd, buf, len buf)) >= 0 && !last){
			last = n != len buf;
		  Retry:
			for(;;){
				if(++nsent%10 == 0){	# probe
					o = Tftp_Req;
					send(c.dfd, array[0] of byte, Tftp_Req, seq);
					(op, iseq, data) = recv(replies, 500);
					if(debug || op != Tftp_Ack)
						sys->print("ack reply: %d %d\n", op, iseq);
					if(op == Tftp_Last || op == Tftp_Error){
						if(op == Tftp_Last)
							sys->print("timed out\n");
						else
							sys->print("error reply\n");
						raise "disaster";
					}
					if(debug)
						sys->print("ok\n");
					continue Retry;
				}
				send(c.dfd, buf[0:n], Tftp_Data, seq);
				(op, iseq, data) = recv(replies, 40);
				case op {
				Tftp_Error =>
					sys->fprint(sys->fildes(2), "mangaload: manga refused data\n");
					raise "disaster";
				Tftp_Ack =>
					if(seq == iseq){
						seq++;
						break Retry;
					}
					sys->print("sequence error: rcvd %d expected %d\n", iseq, seq);
					if(iseq > seq){
						sys->print("unrecoverable sequence error\n");
						send(c.dfd, array[0] of byte, Tftp_Data, ++seq);	# stop manga
						raise "disaster";
					}
					# resend
					sys->seek(fd, -big ((seq-iseq)*len buf), 1);
					seq = iseq;
				Tftp_Last =>
					seq++;
					break Retry;	# timeout ok: manga doesn't usually reply unless packet lost
				}
			}
		}
	}exception{
	* =>
		;
	}
	kill(rpid);
	kill(tpid);
	sys->print("ok?\n");
}

kill(pid: int)
{
	if(pid)
		sys->fprint(sys->open("#p/"+string pid+"/ctl", Sys->OWRITE), "kill");
}

senderr()
{
	sys->fprint(sys->fildes(2), "mangaload: icmp write failed: %r\n");
	raise "disaster";
}

send(fd: ref Sys->FD, data: array of byte, op: int, seq: int): int
{
	buf := array[64*1024+512] of {* => byte 0};
	buf[Odata:] = data;
	ip->put2(buf, Oseq, seq);
	buf[Otype] = byte InfoRequest;
	buf[Ocode] = byte op;
	if(sys->write(fd, buf, Odata+len data) < Odata+len data)
		return -1;
	if(debug)
		sys->print("sent op=%d seq=%d ld=%d\n", op, seq, len data);
	return 0;
}

flush(input: chan of ref Icmp)
{
	for(;;)alt{
	<-input =>
		;
	* =>
		return;
	}
}

recv(input: chan of ref Icmp, msec: int): (int, int, array of byte)
{
	t := Timer.start(msec);
	alt{
	<-t.timeout =>
		return (Tftp_Last, 0, nil);
	ic := <-input =>
		t.stop();
		if(ic.ptype == InfoReply)
			return (ic.code, ic.seq, ic.data);
		return (Tftp_Last, 0, nil);
	}
}

reader(fd: ref Sys->FD, out: chan of ref Icmp, pid: chan of int)
{
	pid <-= sys->pctl(0, nil);
	for(;;){
		buf := array[64*1024+512] of byte;
		n := sys->read(fd, buf, len buf);
		if(n <= 0){
			if(n == 0)
				sys->werrstr("unexpected eof");
			break;
		}
		ic := Icmp.unpack(buf[0:n]);
		if(ic != nil){
			if(debug)
				sys->print("recv type=%d op=%d seq=%d id=%d\n", ic.ptype, ic.code, ic.seq, ic.id);
			out <-= ic;
		}else
			sys->fprint(sys->fildes(2), "mangaload: corrupt icmp packet rcvd\n");
	}
	sys->print("read: %r\n");
	out <-= nil;
}

# IP and ICMP packet header
Ovihl: con 0;
Otos: con 1;
Olength: con 2;
Oid: con Olength+2;
Ofrag: con Oid+2;
Ottl: con Ofrag+2;
Oproto: con Ottl+1;
Oipcksum: con Oproto+1;
Osrc: con Oipcksum+2;
Odst: con Osrc+4;
Otype: con Odst+4;
Ocode: con Otype+1;
Ocksum: con Ocode+1;
Oicmpid: con Ocksum+2;
Oseq: con Oicmpid+2;
Odata: con Oseq+2;

Icmp.unpack(b: array of byte): ref Icmp
{
	if(len b < Odata)
		return nil;
	ic := ref Icmp;
	ic.ttl = int b[Ottl];
	ic.src = IPaddr.newv4(b[Osrc:]);
	ic.dst = IPaddr.newv4(b[Odst:]);
	ic.ptype = int b[Otype];
	ic.code = int b[Ocode];
	ic.seq = ip->get2(b, Oseq);
	ic.id = ip->get2(b, Oicmpid);
	ic.munged = 0;
	if(len b > Odata)
		ic.data = b[Odata:];
	return ic;
}