ref: e59ffed426f628794d4669f152eff9a6239b99db
dir: /os/ip/bootp.c/
#include "u.h" #include "../port/lib.h" #include "mem.h" #include "dat.h" #include "fns.h" #include "../port/error.h" #include "kernel.h" #include "ip.h" static ulong fsip; static ulong auip; static ulong gwip; static ulong ipmask; static ulong ipaddr; enum { Bootrequest = 1, Bootreply = 2, }; typedef struct Bootp { /* udp.c oldheader */ uchar raddr[IPaddrlen]; uchar laddr[IPaddrlen]; uchar rport[2]; uchar lport[2]; /* bootp itself */ uchar op; /* opcode */ uchar htype; /* hardware type */ uchar hlen; /* hardware address len */ uchar hops; /* hops */ uchar xid[4]; /* a random number */ uchar secs[2]; /* elapsed snce client started booting */ uchar pad[2]; uchar ciaddr[4]; /* client IP address (client tells server) */ uchar yiaddr[4]; /* client IP address (server tells client) */ uchar siaddr[4]; /* server IP address */ uchar giaddr[4]; /* gateway IP address */ uchar chaddr[16]; /* client hardware address */ uchar sname[64]; /* server host name (optional) */ uchar file[128]; /* boot file name */ uchar vend[128]; /* vendor-specific goo */ } Bootp; /* * bootp returns: * * "fsip d.d.d.d * auip d.d.d.d * gwip d.d.d.d * ipmask d.d.d.d * ipaddr d.d.d.d" * * where d.d.d.d is the IP address in dotted decimal notation, and each * address is followed by a newline. */ static Bootp req; static Proc* rcvprocp; static int recv; static int done; static Rendez bootpr; static char rcvbuf[512+2*IPaddrlen+2*2]; static void rcvbootp(void *a) { int n, fd; Bootp *rp; char *field[4]; uchar ip[IPaddrlen]; if(waserror()) pexit("", 0); rcvprocp = up; /* store for postnote below */ fd = (int)a; while(done == 0) { n = kread(fd, rcvbuf, sizeof(rcvbuf)); if(n <= 0) break; rp = (Bootp*)rcvbuf; /* currently ignore udp's header */ if(memcmp(req.chaddr, rp->chaddr, 6) == 0 && rp->htype == 1 && rp->hlen == 6 && getfields((char*)rp->vend+4, field, 4, 1, " ") == 4 && strncmp((char*)rp->vend, "p9 ", 4) == 0){ if(ipaddr == 0) ipaddr = nhgetl(rp->yiaddr); if(ipmask == 0) ipmask = parseip(ip, field[0]); if(fsip == 0) fsip = parseip(ip, field[1]); if(auip == 0) auip = parseip(ip, field[2]); if(gwip == 0) gwip = parseip(ip, field[3]); break; } } poperror(); rcvprocp = nil; recv = 1; wakeup(&bootpr); pexit("", 0); } static char* rbootp(Ipifc *ifc) { int cfd, dfd, tries, n; char ia[5+3*24], im[16], *av[3]; uchar nipaddr[4], ngwip[4], nipmask[4]; char dir[Maxpath]; av[1] = "0.0.0.0"; av[2] = "0.0.0.0"; ipifcadd(ifc, av, 3, 0, nil); cfd = kannounce("udp!*!68", dir); if(cfd < 0) return "bootp announce failed"; strcat(dir, "/data"); if(kwrite(cfd, "headers", 7) < 0){ kclose(cfd); return "bootp ctl headers failed"; } kwrite(cfd, "oldheaders", 10); dfd = kopen(dir, ORDWR); if(dfd < 0){ kclose(cfd); return "bootp open data failed"; } kclose(cfd); /* create request */ memset(&req, 0, sizeof(req)); ipmove(req.raddr, IPv4bcast); hnputs(req.rport, 67); req.op = Bootrequest; req.htype = 1; /* ethernet (all we know) */ req.hlen = 6; /* ethernet (all we know) */ /* Hardware MAC address */ memmove(req.chaddr, ifc->mac, 6); /* Fill in the local IP address if we know it */ ipv4local(ifc, req.ciaddr); memset(req.file, 0, sizeof(req.file)); strcpy((char*)req.vend, "p9 "); done = 0; recv = 0; kproc("rcvbootp", rcvbootp, (void*)dfd, KPDUPFDG); /* * broadcast bootp's till we get a reply, * or fixed number of tries */ tries = 0; while(recv == 0) { if(kwrite(dfd, &req, sizeof(req)) < 0) print("bootp: write: %s\n", commonerror()); tsleep(&bootpr, return0, 0, 1000); if(++tries > 10) { print("bootp: timed out\n"); break; } } kclose(dfd); done = 1; if(rcvprocp != nil){ postnote(rcvprocp, 1, "timeout", 0); rcvprocp = nil; } av[1] = "0.0.0.0"; av[2] = "0.0.0.0"; ipifcrem(ifc, av, 3); hnputl(nipaddr, ipaddr); sprint(ia, "%V", nipaddr); hnputl(nipmask, ipmask); sprint(im, "%V", nipmask); av[1] = ia; av[2] = im; ipifcadd(ifc, av, 3, 0, nil); if(gwip != 0) { hnputl(ngwip, gwip); n = snprint(ia, sizeof(ia), "add 0.0.0.0 0.0.0.0 %V", ngwip); routewrite(ifc->conv->p->f, nil, ia, n); } return nil; } static int rbootpread(char *bp, ulong offset, int len) { int n; char *buf; uchar a[4]; buf = smalloc(READSTR); if(waserror()){ free(buf); nexterror(); } hnputl(a, fsip); n = snprint(buf, READSTR, "fsip %15V\n", a); hnputl(a, auip); n += snprint(buf + n, READSTR-n, "auip %15V\n", a); hnputl(a, gwip); n += snprint(buf + n, READSTR-n, "gwip %15V\n", a); hnputl(a, ipmask); n += snprint(buf + n, READSTR-n, "ipmask %15V\n", a); hnputl(a, ipaddr); snprint(buf + n, READSTR-n, "ipaddr %15V\n", a); len = readstr(offset, bp, len, buf); poperror(); free(buf); return len; } char* (*bootp)(Ipifc*) = rbootp; int (*bootpread)(char*, ulong, int) = rbootpread;