123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556 |
- #include "u.h"
- #include "lib.h"
- #include "mem.h"
- #include "dat.h"
- #include "fns.h"
- #include "ip.h"
- uchar broadcast[Eaddrlen] = {
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- };
- static ushort tftpport = 5000;
- static int Id = 1;
- static Netaddr myaddr;
- static Netaddr server;
- typedef struct {
- uchar header[4];
- uchar data[Segsize];
- } Tftp;
- static Tftp tftpb;
- int
- etherrxpkt(int ctlrno, Etherpkt *pkt, int timo)
- {
- int n;
- for (;;) {
- n = devread(ctlrno, (uchar*)pkt, sizeof(*pkt), 0);
- if (n >= 0)
- return n;
- if (timo-- < 0)
- return -1;
- }
- }
- int
- ethertxpkt(int ctlrno, Etherpkt *pkt, int len, int timo)
- {
- USED(timo);
- return devwrite(ctlrno, (uchar*)pkt, len, 0);
- }
- static void
- hnputs(uchar *ptr, ushort val)
- {
- ptr[0] = val>>8;
- ptr[1] = val;
- }
- static void
- hnputl(uchar *ptr, ulong val)
- {
- ptr[0] = val>>24;
- ptr[1] = val>>16;
- ptr[2] = val>>8;
- ptr[3] = val;
- }
- static ulong
- nhgetl(uchar *ptr)
- {
- return ((ptr[0]<<24) | (ptr[1]<<16) | (ptr[2]<<8) | ptr[3]);
- }
- static ushort
- nhgets(uchar *ptr)
- {
- return ((ptr[0]<<8) | ptr[1]);
- }
- static short endian = 1;
- static char* aendian = (char*)&endian;
- #define LITTLE *aendian
- static ushort
- ptcl_csum(void *a, int len)
- {
- uchar *addr;
- ulong t1, t2;
- ulong losum, hisum, mdsum, x;
- addr = a;
- losum = 0;
- hisum = 0;
- mdsum = 0;
- x = 0;
- if((ulong)addr & 1) {
- if(len) {
- hisum += addr[0];
- len--;
- addr++;
- }
- x = 1;
- }
- while(len >= 16) {
- t1 = *(ushort*)(addr+0);
- t2 = *(ushort*)(addr+2); mdsum += t1;
- t1 = *(ushort*)(addr+4); mdsum += t2;
- t2 = *(ushort*)(addr+6); mdsum += t1;
- t1 = *(ushort*)(addr+8); mdsum += t2;
- t2 = *(ushort*)(addr+10); mdsum += t1;
- t1 = *(ushort*)(addr+12); mdsum += t2;
- t2 = *(ushort*)(addr+14); mdsum += t1;
- mdsum += t2;
- len -= 16;
- addr += 16;
- }
- while(len >= 2) {
- mdsum += *(ushort*)addr;
- len -= 2;
- addr += 2;
- }
- if(x) {
- if(len)
- losum += addr[0];
- if(LITTLE)
- losum += mdsum;
- else
- hisum += mdsum;
- } else {
- if(len)
- hisum += addr[0];
- if(LITTLE)
- hisum += mdsum;
- else
- losum += mdsum;
- }
- losum += hisum >> 8;
- losum += (hisum & 0xff) << 8;
- while(hisum = losum>>16)
- losum = hisum + (losum & 0xffff);
- return ~losum;
- }
- static ushort
- ip_csum(uchar *addr)
- {
- int len;
- ulong sum = 0;
- len = (addr[0]&0xf)<<2;
- while(len > 0) {
- sum += addr[0]<<8 | addr[1] ;
- len -= 2;
- addr += 2;
- }
- sum = (sum & 0xffff) + (sum >> 16);
- sum = (sum & 0xffff) + (sum >> 16);
- return (sum^0xffff);
- }
- static void
- udpsend(int ctlrno, Netaddr *a, void *data, int dlen)
- {
- Udphdr *uh;
- Etherhdr *ip;
- Etherpkt pkt;
- int len, ptcllen;
- uh = (Udphdr*)&pkt;
- memset(uh, 0, sizeof(Etherpkt));
- memmove(uh->udpcksum+sizeof(uh->udpcksum), data, dlen);
- /*
- * UDP portion
- */
- ptcllen = dlen + (UDP_HDRSIZE-UDP_PHDRSIZE);
- uh->ttl = 0;
- uh->udpproto = IP_UDPPROTO;
- uh->frag[0] = 0;
- uh->frag[1] = 0;
- hnputs(uh->udpplen, ptcllen);
- hnputl(uh->udpsrc, myaddr.ip);
- hnputs(uh->udpsport, myaddr.port);
- hnputl(uh->udpdst, a->ip);
- hnputs(uh->udpdport, a->port);
- hnputs(uh->udplen, ptcllen);
- uh->udpcksum[0] = 0;
- uh->udpcksum[1] = 0;
- dlen = (dlen+1)&~1;
- hnputs(uh->udpcksum, ptcl_csum(&uh->ttl, dlen+UDP_HDRSIZE));
- /*
- * IP portion
- */
- ip = (Etherhdr*)&pkt;
- len = UDP_EHSIZE+UDP_HDRSIZE+dlen; /* non-descriptive names */
- ip->vihl = IP_VER|IP_HLEN;
- ip->tos = 0;
- ip->ttl = 255;
- hnputs(ip->length, len-ETHER_HDR);
- hnputs(ip->id, Id++);
- ip->frag[0] = 0;
- ip->frag[1] = 0;
- ip->cksum[0] = 0;
- ip->cksum[1] = 0;
- hnputs(ip->cksum, ip_csum(&ip->vihl));
- /*
- * Ethernet MAC portion
- */
- hnputs(ip->type, ET_IP);
- memmove(ip->d, a->ea, sizeof(ip->d));
- ethertxpkt(ctlrno, &pkt, len, Timeout);
- }
- static void
- nak(int ctlrno, Netaddr *a, int code, char *msg, int report)
- {
- int n;
- char buf[128];
- buf[0] = 0;
- buf[1] = Tftp_ERROR;
- buf[2] = 0;
- buf[3] = code;
- strcpy(buf+4, msg);
- n = strlen(msg) + 4 + 1;
- udpsend(ctlrno, a, buf, n);
- if(report)
- print("\ntftp: error(%d): %s\n", code, msg);
- }
- static int
- udprecv(int ctlrno, Netaddr *a, void *data, int dlen)
- {
- int n, len;
- ushort csm;
- Udphdr *h;
- ulong addr, timo;
- Etherpkt pkt;
- static int rxactive;
- if(rxactive == 0)
- timo = 1000;
- else
- timo = Timeout;
- timo += msec();
- while(timo > msec()){
- n = etherrxpkt(ctlrno, &pkt, timo-msec());
- if(n <= 0)
- continue;
- h = (Udphdr*)&pkt;
- if(nhgets(h->type) != ET_IP)
- continue;
- if(ip_csum(&h->vihl)) {
- print("ip chksum error\n");
- continue;
- }
- if(h->vihl != (IP_VER|IP_HLEN)) {
- print("ip bad vers/hlen\n");
- continue;
- }
- if(h->udpproto != IP_UDPPROTO)
- continue;
- h->ttl = 0;
- len = nhgets(h->udplen);
- hnputs(h->udpplen, len);
- if(nhgets(h->udpcksum)) {
- csm = ptcl_csum(&h->ttl, len+UDP_PHDRSIZE);
- if(csm != 0) {
- print("udp chksum error csum #%4lux len %d\n", csm, n);
- break;
- }
- }
- if(a->port != 0 && nhgets(h->udpsport) != a->port)
- continue;
- if(myaddr.port != 0 && nhgets(h->udpdport) != myaddr.port)
- continue;
- addr = nhgetl(h->udpsrc);
- if(a->ip != Bcastip && addr != a->ip)
- continue;
- len -= UDP_HDRSIZE-UDP_PHDRSIZE;
- if(len > dlen) {
- print("udp: packet too big\n");
- continue;
- }
- memmove(data, h->udpcksum+sizeof(h->udpcksum), len);
- a->ip = addr;
- a->port = nhgets(h->udpsport);
- memmove(a->ea, pkt.s, sizeof(a->ea));
- rxactive = 1;
- return len;
- }
- return 0;
- }
- static int tftpblockno;
- static int
- tftpopen(int ctlrno, Netaddr *a, char *name, Tftp *tftp)
- {
- int i, len, rlen;
- char buf[Segsize+2];
- buf[0] = 0;
- buf[1] = Tftp_READ;
- len = sprint(buf+2, "%s", name) + 2;
- len += sprint(buf+len+1, "octet") + 2;
- for(i = 0; i < 5; i++){
- udpsend(ctlrno, a, buf, len);
- a->port = 0;
- if((rlen = udprecv(ctlrno, a, tftp, sizeof(Tftp))) < sizeof(tftp->header))
- continue;
- switch((tftp->header[0]<<8)|tftp->header[1]){
- case Tftp_ERROR:
- print("tftpopen: error (%d): %s\n",
- (tftp->header[2]<<8)|tftp->header[3], tftp->data);
- return -1;
- case Tftp_DATA:
- tftpblockno = 1;
- len = (tftp->header[2]<<8)|tftp->header[3];
- if(len != tftpblockno){
- print("tftpopen: block error: %d\n", len);
- nak(ctlrno, a, 1, "block error", 0);
- return -1;
- }
- return rlen-sizeof(tftp->header);
- }
- }
- print("tftpopen: failed to connect to server\n");
- return -1;
- }
- static int
- tftpread(int ctlrno, Netaddr *a, Tftp *tftp, int dlen)
- {
- int blockno, len;
- uchar buf[4];
- buf[0] = 0;
- buf[1] = Tftp_ACK;
- buf[2] = tftpblockno>>8;
- buf[3] = tftpblockno;
- tftpblockno++;
- dlen += sizeof(tftp->header);
- buggery:
- udpsend(ctlrno, a, buf, sizeof(buf));
- if((len = udprecv(ctlrno, a, tftp, dlen)) != dlen){
- print("tftpread: %d != %d\n", len, dlen);
- nak(ctlrno, a, 2, "short read", 0);
- }
- blockno = (tftp->header[2]<<8)|tftp->header[3];
- if(blockno != tftpblockno){
- print("tftpread: block error: %d, expected %d\n", blockno, tftpblockno);
- if(blockno == tftpblockno-1)
- goto buggery;
- nak(ctlrno, a, 1, "block error", 0);
- return -1;
- }
- return len-sizeof(tftp->header);
- }
- // #define BOOT_MAGIC L_MAGIC
- #define BOOT_MAGIC 0x0700e0c3
- void
- getether(char *dev, uchar *ea)
- {
- int i;
- char *p;
- p = dev;
- for (i = 0; i < 8; i++) {
- p = strchr(p, ' ');
- if (p == 0)
- panic("no ether addr");
- p++;
- }
- for (i = 0; i < 6; i++) {
- ea[i] = strtoul(p, &p, 16);
- if (*p != (i == 5 ? ' ' : '-'))
- panic("bad ether addr");
- p++;
- }
- }
- static char inibuf[BOOTARGSLEN];
- int
- bootp(char *dev)
- {
- Bootp req, rep;
- int i, fd, dlen, segsize, text, data, bss, total;
- uchar *addr, *p, ea[6];
- char *cp;
- ulong entry;
- Exec *exec;
- char *filename, confname[32];
- getether(dev, ea);
- fd = devopen(dev);
- if (fd < 0)
- panic("bootp devopen");
- memset(&req, 0, sizeof(req));
- req.op = Bootrequest;
- req.htype = 1; /* ethernet */
- req.hlen = Eaddrlen; /* ethernet */
- memmove(req.chaddr, ea, Eaddrlen);
- myaddr.ip = 0;
- myaddr.port = BPportsrc;
- memmove(myaddr.ea, ea, Eaddrlen);
- for(i = 0; i < 10; i++) {
- server.ip = Bcastip;
- server.port = BPportdst;
- memmove(server.ea, broadcast, sizeof(server.ea));
- udpsend(fd, &server, &req, sizeof(req));
- if(udprecv(fd, &server, &rep, sizeof(rep)) <= 0)
- continue;
- if(memcmp(req.chaddr, rep.chaddr, Eaddrlen))
- continue;
- if(rep.htype != 1 || rep.hlen != Eaddrlen)
- continue;
- break;
- }
- if(i >= 10) {
- print("bootp timed out\n");
- return -1;
- }
- sprint(confname, "/alpha/conf/%d.%d.%d.%d",
- rep.yiaddr[0],
- rep.yiaddr[1],
- rep.yiaddr[2],
- rep.yiaddr[3]);
- if(rep.sname[0] != '\0')
- print("%s ", rep.sname);
- print("(%d.%d.%d.%d!%d): %s...",
- rep.siaddr[0],
- rep.siaddr[1],
- rep.siaddr[2],
- rep.siaddr[3],
- server.port,
- confname);
- myaddr.ip = nhgetl(rep.yiaddr);
- myaddr.port = tftpport++;
- server.ip = nhgetl(rep.siaddr);
- server.port = TFTPport;
- if((dlen = tftpopen(fd, &server, confname, &tftpb)) < 0)
- return -1;
- cp = inibuf;
- while(dlen > 0) {
- if(cp-inibuf+dlen > BOOTARGSLEN)
- panic("conf too large");
- memmove(cp, tftpb.data, dlen);
- cp += dlen;
- if(dlen != Segsize)
- break;
- if((dlen = tftpread(fd, &server, &tftpb, sizeof(tftpb.data))) < 0)
- return -1;
- }
- *cp = 0;
- setconf(inibuf);
- filename = "/alpha/9apc";
- cp = getconf("bootfile");
- if(cp != nil)
- filename = cp;
- print("%s\n", filename);
- myaddr.port = tftpport++;
- server.port = TFTPport;
- if((dlen = tftpopen(fd, &server, filename, &tftpb)) < 0)
- return -1;
- exec = (Exec*)(tftpb.data);
- if(dlen < sizeof(Exec) || GLLONG(exec->magic) != BOOT_MAGIC){
- nak(fd, &server, 0, "bad magic number", 1);
- return -1;
- }
- text = GLLONG(exec->text);
- data = GLLONG(exec->data);
- bss = GLLONG(exec->bss);
- total = text+data+bss;
- entry = GLLONG(exec->entry);
- if (!validrgn(entry, entry+total))
- panic("memory range not available: %lux-%lux\n", entry, entry+total);
- print("%d", text);
- addr = (uchar*)entry;
- p = tftpb.data+sizeof(Exec);
- dlen -= sizeof(Exec);
- segsize = text;
- for(;;){
- if(dlen == 0){
- if((dlen = tftpread(fd, &server, &tftpb, sizeof(tftpb.data))) < 0)
- return -1;
- p = tftpb.data;
- }
- if(segsize <= dlen)
- i = segsize;
- else
- i = dlen;
- memmove(addr, p, i);
- addr += i;
- p += i;
- segsize -= i;
- dlen -= i;
- if(segsize <= 0){
- if(data == 0)
- break;
- print("+%d", data);
- segsize = data;
- data = 0;
- // addr = (uchar*)pground((uvlong)addr);
- }
- }
- nak(fd, &server, 3, "ok", 0); /* tftpclose */
- print("+%d=%d\n", bss, total);
- print("entry: 0x%lux\n", entry);
- kexec(entry);
- return 0;
- }
|