123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341 |
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <ip.h>
- #define LOG "pptpd"
- typedef struct Call Call;
- typedef struct Event Event;
- #define SDB if(debug) fprint(2,
- #define EDB );
- enum {
- Magic = 0x1a2b3c4d,
- Nhash = 17,
- Nchan = 10, /* maximum number of channels */
- Window = 8, /* default window size */
- Timeout = 60, /* timeout in seconds for control channel */
- Pktsize = 2000, /* maximum packet size */
- Tick = 500, /* tick length in milliseconds */
- Sendtimeout = 4, /* in ticks */
- };
- enum {
- Syncframe = 0x1,
- Asyncframe = 0x2,
- Analog = 0x1,
- Digital = 0x2,
- Version = 0x100,
- };
- enum {
- Tstart = 1,
- Rstart = 2,
- Tstop = 3,
- Rstop = 4,
- Techo = 5,
- Recho = 6,
- Tcallout = 7,
- Rcallout = 8,
- Tcallreq = 9,
- Rcallreq = 10,
- Acallcon = 11,
- Tcallclear = 12,
- Acalldis = 13,
- Awaninfo = 14,
- Alinkinfo = 15,
- };
- struct Event {
- QLock;
- QLock waitlk;
- int wait;
- int ready;
- };
- struct Call {
- int ref;
- QLock lk;
- int id;
- int serial;
- int pppfd;
- int closed;
- int pac; /* server is acting as a PAC */
- int recvwindow; /* recv windows */
- int sendwindow; /* send windows */
- int delay;
- int sendaccm;
- int recvaccm;
- uint seq; /* current seq number - for send */
- uint ack; /* current acked mesg - for send */
- uint rseq; /* highest recv seq number for in order packet */
- uint rack; /* highest ack sent */
- Event eack; /* recved ack - for send */
- ulong tick;
- uchar remoteip[IPaddrlen]; /* remote ip address */
- int dhcpfd[2]; /* pipe to dhcpclient */
- /* error stats */
- struct {
- int crc;
- int frame;
- int hardware;
- int overrun;
- int timeout;
- int align;
- } err;
- struct {
- int send;
- int sendack;
- int recv;
- int recvack;
- int dropped;
- int missing;
- int sendwait;
- int sendtimeout;
- } stat;
- Call *next;
- };
- struct {
- QLock lk;
- int start;
- int grefd;
- int grecfd;
- uchar local[IPaddrlen];
- uchar remote[IPaddrlen];
- char *tcpdir;
- uchar ipaddr[IPaddrlen]; /* starting ip addresss to allocate */
- int recvwindow;
- char *pppdir;
- char *pppexec;
- double rcvtime; /* time at which last request was received */
- int echoid; /* id of last echo request */
- Call *hash[Nhash];
- } srv;
- /* GRE flag bits */
- enum {
- GRE_chksum = (1<<15),
- GRE_routing = (1<<14),
- GRE_key = (1<<13),
- GRE_seq = (1<<12),
- GRE_srcrt = (1<<11),
- GRE_recur = (7<<8),
- GRE_ack = (1<<7),
- GRE_ver = 0x7,
- };
- /* GRE protocols */
- enum {
- GRE_ppp = 0x880b,
- };
- int debug;
- double drop;
- void myfatal(char *fmt, ...);
- #define PSHORT(p, v) ((p)[0]=((v)>>8), (p)[1]=(v))
- #define PLONG(p, v) (PSHORT(p, (v)>>16), PSHORT(p+2, (v)))
- #define PSTRING(d,s,n) strncpy((char*)(d), s, n)
- #define GSHORT(p) (((p)[0]<<8) | ((p)[1]<<0))
- #define GLONG(p) ((GSHORT((p))<<16) | ((GSHORT((p)+2))<<0))
- #define GSTRING(d,s,n) strncpy(d, (char*)(s), n), d[(n)-1] = 0
- void serve(void);
- int sstart(uchar*, int);
- int sstop(uchar*, int);
- int secho(uchar*, int);
- int scallout(uchar*, int);
- int scallreq(uchar*, int);
- int scallcon(uchar*, int);
- int scallclear(uchar*, int);
- int scalldis(uchar*, int);
- int swaninfo(uchar*, int);
- int slinkinfo(uchar*, int);
- Call *callalloc(int id);
- void callclose(Call*);
- void callfree(Call*);
- Call *calllookup(int id);
- void gretimeout(void*);
- void pppread(void*);
- void srvinit(void);
- void greinit(void);
- void greread(void*);
- void greack(Call *c);
- void timeoutthread(void*);
- int argatoi(char *p);
- void usage(void);
- int ipaddralloc(Call *c);
- void *emallocz(int size);
- void esignal(Event *e);
- void ewait(Event *e);
- int proc(char **argv, int fd0, int fd1, int fd2);
- double realtime(void);
- ulong thread(void(*f)(void*), void *a);
- void
- main(int argc, char *argv[])
- {
- ARGBEGIN{
- case 'd': debug++; break;
- case 'p': srv.pppdir = ARGF(); break;
- case 'P': srv.pppexec = ARGF(); break;
- case 'w': srv.recvwindow = argatoi(ARGF()); break;
- case 'D': drop = atof(ARGF()); break;
- default:
- usage();
- }ARGEND
- fmtinstall('I', eipfmt);
- fmtinstall('E', eipfmt);
- fmtinstall('V', eipfmt);
- fmtinstall('M', eipfmt);
- rfork(RFNOTEG|RFREND);
- if(argc != 1)
- usage();
- srv.tcpdir = argv[0];
- srvinit();
- syslog(0, LOG, ": src=%I: pptp started: %d", srv.remote, getpid());
- SDB "\n\n\n%I: pptp started\n", srv.remote EDB
- greinit();
- thread(timeoutthread, 0);
- serve();
- syslog(0, LOG, ": src=%I: server exits", srv.remote);
- exits(0);
- }
- void
- usage(void)
- {
- fprint(2, "usage: pptpd [-dD] [-p ppp-net] [-w window] tcpdir\n");
- exits("usage");
- }
- void
- serve(void)
- {
- uchar buf[2000], *p;
- int n, n2, len;
- int magic;
- int op, type;
- n = 0;
- for(;;) {
- n2 = read(0, buf+n, sizeof(buf)-n);
- if(n2 < 0)
- myfatal("bad read on ctl channel: %r");
- if(n2 == 0)
- break;
- n += n2;
- p = buf;
- for(;;) {
- if(n < 12)
- break;
- qlock(&srv.lk);
- srv.rcvtime = realtime();
- qunlock(&srv.lk);
- len = GSHORT(p);
- type = GSHORT(p+2);
- magic = GLONG(p+4);
- op = GSHORT(p+8);
- if(magic != Magic)
- myfatal("bad magic number: got %x", magic);
- if(type != 1)
- myfatal("bad message type: %d", type);
- switch(op) {
- default:
- myfatal("unknown control op: %d", op);
- case Tstart: /* start-control-connection-request */
- n2 = sstart(p, n);
- break;
- case Tstop:
- n2 = sstop(p, n);
- if(n2 > 0)
- return;
- break;
- case Techo:
- n2 = secho(p, n);
- break;
- case Tcallout:
- n2 = scallout(p, n);
- break;
- case Tcallreq:
- n2 = scallreq(p, n);
- break;
- case Acallcon:
- n2 = scallcon(p, n);
- break;
- case Tcallclear:
- n2 = scallclear(p, n);
- break;
- case Acalldis:
- n2 = scalldis(p, n);
- break;
- case Awaninfo:
- n2 = swaninfo(p, n);
- break;
- case Alinkinfo:
- n2 = slinkinfo(p, n);
- break;
- }
- if(n2 == 0)
- break;
- if(n2 != len)
- myfatal("op=%d: bad length: got %d expected %d", op, len, n2);
- n -= n2;
- p += n2;
-
- }
- /* move down partial message */
- if(p != buf && n != 0)
- memmove(buf, p, n);
- }
- }
- int
- sstart(uchar *p, int n)
- {
- int ver, frame, bearer, maxchan, firm;
- char host[64], vendor[64], *sysname;
- uchar buf[156];
- if(n < 156)
- return 0;
- ver = GSHORT(p+12);
- frame = GLONG(p+16);
- bearer = GLONG(p+20);
- maxchan = GSHORT(p+24);
- firm = GSHORT(p+26);
- GSTRING(host, p+28, 64);
- GSTRING(vendor, p+92, 64);
- SDB "%I: start ver = %x f = %d b = %d maxchan = %d firm = %d host = %s vendor = %s\n",
- srv.remote, ver, frame, bearer, maxchan, firm, host, vendor EDB
- if(ver != Version)
- myfatal("bad version: got %x expected %x", ver, Version);
- if(srv.start)
- myfatal("multiple start messages");
- srv.start = 1;
- sysname = getenv("sysname");
- if(sysname == 0)
- strcpy(host, "gnot");
- else
- strncpy(host, sysname, 64);
- free(sysname);
- memset(buf, 0, sizeof(buf));
- PSHORT(buf+0, sizeof(buf)); /* length */
- PSHORT(buf+2, 1); /* message type */
- PLONG(buf+4, Magic); /* magic */
- PSHORT(buf+8, Rstart); /* op */
- PSHORT(buf+12, Version); /* version */
- buf[14] = 1; /* result = ok */
- PLONG(buf+16, Syncframe|Asyncframe); /* frameing */
- PLONG(buf+20, Digital|Analog); /* berear capabilities */
- PSHORT(buf+24, Nchan); /* max channels */
- PSHORT(buf+26, 1); /* driver version */
- PSTRING(buf+28, host, 64); /* host name */
- PSTRING(buf+92, "plan 9", 64); /* vendor */
- if(write(1, buf, sizeof(buf)) < sizeof(buf))
- myfatal("write failed: %r");
- return 156;
- }
- int
- sstop(uchar *p, int n)
- {
- int reason;
- uchar buf[16];
- if(n < 16)
- return 0;
- reason = p[12];
-
- SDB "%I: stop %d\n", srv.remote, reason EDB
- memset(buf, 0, sizeof(buf));
- PSHORT(buf+0, sizeof(buf)); /* length */
- PSHORT(buf+2, 1); /* message type */
- PLONG(buf+4, Magic); /* magic */
- PSHORT(buf+8, Rstop); /* op */
- buf[12] = 1; /* ok */
- if(write(1, buf, sizeof(buf)) < sizeof(buf))
- myfatal("write failed: %r");
- return 16;
- }
- int
- secho(uchar *p, int n)
- {
- int id;
- uchar buf[20];
- if(n < 16)
- return 0;
- id = GLONG(p+12);
-
- SDB "%I: echo %d\n", srv.remote, id EDB
- memset(buf, 0, sizeof(buf));
- PSHORT(buf+0, sizeof(buf)); /* length */
- PSHORT(buf+2, 1); /* message type */
- PLONG(buf+4, Magic); /* magic */
- PSHORT(buf+8, Recho); /* op */
- PLONG(buf+12, id); /* id */
- p[16] = 1; /* ok */
- if(write(1, buf, sizeof(buf)) < sizeof(buf))
- myfatal("write failed: %r");
- return 16;
- }
- int
- scallout(uchar *p, int n)
- {
- int id, serial;
- int minbps, maxbps, bearer, frame;
- int window, delay;
- int nphone;
- char phone[64], sub[64], buf[32];
- Call *c;
- if(n < 168)
- return 0;
- if(!srv.start)
- myfatal("%I: did not recieve start message", srv.remote);
-
- id = GSHORT(p+12);
- serial = GSHORT(p+14);
- minbps = GLONG(p+16);
- maxbps = GLONG(p+20);
- bearer = GLONG(p+24);
- frame = GLONG(p+28);
- window = GSHORT(p+32);
- delay = GSHORT(p+34);
- nphone = GSHORT(p+36);
- GSTRING(phone, p+40, 64);
- GSTRING(sub, p+104, 64);
- SDB "%I: callout id = %d serial = %d bps=[%d,%d] b=%x f=%x win = %d delay = %d np=%d phone=%s sub=%s\n",
- srv.remote, id, serial, minbps, maxbps, bearer, frame, window, delay, nphone, phone, sub EDB
- c = callalloc(id);
- c->sendwindow = window;
- c->delay = delay;
- c->pac = 1;
- c->recvwindow = srv.recvwindow;
- memset(buf, 0, sizeof(buf));
- PSHORT(buf+0, sizeof(buf)); /* length */
- PSHORT(buf+2, 1); /* message type */
- PLONG(buf+4, Magic); /* magic */
- PSHORT(buf+8, Rcallout); /* op */
- PSHORT(buf+12, id); /* call id */
- PSHORT(buf+14, id); /* peer id */
- buf[16] = 1; /* ok */
- PLONG(buf+20, 10000000); /* speed */
- PSHORT(buf+24, c->recvwindow); /* window size */
- PSHORT(buf+26, 0); /* delay */
- PLONG(buf+28, 0); /* channel id */
-
- if(write(1, buf, sizeof(buf)) < sizeof(buf))
- myfatal("write failed: %r");
- return 168;
- }
- int
- scallreq(uchar *p, int n)
- {
- USED(p);
- USED(n);
- myfatal("callreq: not done yet");
- return 0;
- }
- int
- scallcon(uchar *p, int n)
- {
- USED(p);
- USED(n);
- myfatal("callcon: not done yet");
- return 0;
- }
- int
- scallclear(uchar *p, int n)
- {
- Call *c;
- int id;
- uchar buf[148];
- if(n < 16)
- return 0;
- id = GSHORT(p+12);
-
- SDB "%I: callclear id=%d\n", srv.remote, id EDB
-
- if(c = calllookup(id)) {
- callclose(c);
- callfree(c);
- }
- memset(buf, 0, sizeof(buf));
- PSHORT(buf+0, sizeof(buf)); /* length */
- PSHORT(buf+2, 1); /* message type */
- PLONG(buf+4, Magic); /* magic */
- PSHORT(buf+8, Acalldis); /* op */
- PSHORT(buf+12, id); /* id */
- buf[14] = 3; /* reply to callclear */
- if(write(1, buf, sizeof(buf)) < sizeof(buf))
- myfatal("write failed: %r");
- return 16;
- }
- int
- scalldis(uchar *p, int n)
- {
- Call *c;
- int id, res;
- if(n < 148)
- return 0;
- id = GSHORT(p+12);
- res = p[14];
- SDB "%I: calldis id=%d res=%d\n", srv.remote, id, res EDB
- if(c = calllookup(id)) {
- callclose(c);
- callfree(c);
- }
- return 148;
- }
- int
- swaninfo(uchar *p, int n)
- {
- Call *c;
- int id;
- if(n < 40)
- return 0;
-
- id = GSHORT(p+12);
- SDB "%I: waninfo id = %d\n", srv.remote, id EDB
-
- c = calllookup(id);
- if(c != 0) {
- c->err.crc = GLONG(p+16);
- c->err.frame = GLONG(p+20);
- c->err.hardware = GLONG(p+24);
- c->err.overrun = GLONG(p+28);
- c->err.timeout = GLONG(p+32);
- c->err.align = GLONG(p+36);
- callfree(c);
- }
-
- return 40;
- }
- int
- slinkinfo(uchar *p, int n)
- {
- Call *c;
- int id;
- int sendaccm, recvaccm;
- if(n < 24)
- return 0;
- id = GSHORT(p+12);
- sendaccm = GLONG(p+16);
- recvaccm = GLONG(p+20);
- SDB "%I: linkinfo id=%d saccm=%ux raccm=%ux\n", srv.remote, id, sendaccm, recvaccm EDB
- if(c = calllookup(id)) {
- c->sendaccm = sendaccm;
- c->recvaccm = recvaccm;
- callfree(c);
- }
-
- return 24;
- }
- Call*
- callalloc(int id)
- {
- uint h;
- Call *c;
- char buf[300], *argv[30], local[20], remote[20], **p;
- int fd, pfd[2], n;
- h = id%Nhash;
- qlock(&srv.lk);
- for(c=srv.hash[h]; c; c=c->next)
- if(c->id == id)
- myfatal("callalloc: duplicate id: %d", id);
- c = emallocz(sizeof(Call));
- c->ref = 1;
- c->id = id;
- c->sendaccm = ~0;
- c->recvaccm = ~0;
- if(!ipaddralloc(c))
- myfatal("callalloc: could not alloc remote ip address");
- if(pipe(pfd) < 0)
- myfatal("callalloc: pipe failed: %r");
- sprint(buf, "%s/ipifc/clone", srv.pppdir);
- fd = open(buf, OWRITE);
- if(fd < 0)
- myfatal("callalloc: could not open %s: %r", buf);
- n = sprint(buf, "iprouting");
- if(write(fd, buf, n) < n)
- myfatal("callalloc: write to ifc failed: %r");
- close(fd);
- p = argv;
- *p++ = srv.pppexec;
- *p++ = "-SC";
- *p++ = "-x";
- *p++ = srv.pppdir;
- if(debug)
- *p++ = "-d";
- sprint(local, "%I", srv.ipaddr);
- *p++ = local;
- sprint(remote, "%I", c->remoteip);
- *p++ = remote;
- *p = 0;
- proc(argv, pfd[0], pfd[0], 2);
- close(pfd[0]);
- c->pppfd = pfd[1];
- c->next = srv.hash[h];
- srv.hash[h] = c;
- qunlock(&srv.lk);
- c->ref++;
- thread(pppread, c);
- c->ref++;
- thread(gretimeout, c);
- syslog(0, LOG, ": src=%I: call started: id=%d: remote ip=%I", srv.remote, id, c->remoteip);
- return c;
- }
- void
- callclose(Call *c)
- {
- Call *oc;
- int id;
- uint h;
- syslog(0, LOG, ": src=%I: call closed: id=%d: send=%d sendack=%d recv=%d recvack=%d dropped=%d missing=%d sendwait=%d sendtimeout=%d",
- srv.remote, c->id, c->stat.send, c->stat.sendack, c->stat.recv, c->stat.recvack,
- c->stat.dropped, c->stat.missing, c->stat.sendwait, c->stat.sendtimeout);
- qlock(&srv.lk);
- if(c->closed) {
- qunlock(&srv.lk);
- return;
- }
- c->closed = 1;
- close(c->dhcpfd[0]);
- close(c->dhcpfd[1]);
- close(c->pppfd);
- c->pppfd = -1;
- h = c->id%Nhash;
- id = c->id;
- for(c=srv.hash[h],oc=0; c; oc=c,c=c->next)
- if(c->id == id)
- break;
- if(oc == 0)
- srv.hash[h] = c->next;
- else
- oc->next = c->next;
- c->next = 0;
- qunlock(&srv.lk);
- callfree(c);
- }
- void
- callfree(Call *c)
- {
- int ref;
-
- qlock(&srv.lk);
- ref = --c->ref;
- qunlock(&srv.lk);
- if(ref > 0)
- return;
-
- /* already unhooked from hash list - see callclose */
- assert(c->closed == 1);
- assert(ref == 0);
- assert(c->next == 0);
- SDB "call free\n" EDB
- free(c);
- }
- Call*
- calllookup(int id)
- {
- uint h;
- Call *c;
- h = id%Nhash;
- qlock(&srv.lk);
- for(c=srv.hash[h]; c; c=c->next)
- if(c->id == id)
- break;
- if(c != 0)
- c->ref++;
- qunlock(&srv.lk);
- return c;
- }
- void
- srvinit(void)
- {
- char buf[100];
- int fd, n;
- sprint(buf, "%s/local", srv.tcpdir);
- if((fd = open(buf, OREAD)) < 0)
- myfatal("could not open %s: %r", buf);
- if((n = read(fd, buf, sizeof(buf))) < 0)
- myfatal("could not read %s: %r", buf);
- buf[n] = 0;
- parseip(srv.local, buf);
- close(fd);
- sprint(buf, "%s/remote", srv.tcpdir);
- if((fd = open(buf, OREAD)) < 0)
- myfatal("could not open %s: %r", buf);
- if((n = read(fd, buf, sizeof(buf))) < 0)
- myfatal("could not read %s: %r", buf);
- buf[n] = 0;
- parseip(srv.remote, buf);
- close(fd);
- if(srv.pppdir == 0)
- srv.pppdir = "/net";
- if(srv.pppexec == 0)
- srv.pppexec = "/bin/ip/ppp";
- if(myipaddr(srv.ipaddr, srv.pppdir) < 0)
- myfatal("could not read local ip addr: %r");
- if(srv.recvwindow == 0)
- srv.recvwindow = Window;
- }
- void
- greinit(void)
- {
- char addr[100], *p;
- int fd, cfd;
- SDB "srv.tcpdir = %s\n", srv.tcpdir EDB
- strcpy(addr, srv.tcpdir);
- p = strrchr(addr, '/');
- if(p == 0)
- myfatal("bad tcp dir: %s", srv.tcpdir);
- *p = 0;
- p = strrchr(addr, '/');
- if(p == 0)
- myfatal("bad tcp dir: %s", srv.tcpdir);
- sprint(p, "/gre!%I!34827", srv.remote);
- SDB "addr = %s\n", addr EDB
- fd = dial(addr, 0, 0, &cfd);
- if(fd < 0)
- myfatal("%I: dial %s failed: %r", srv.remote, addr);
- srv.grefd = fd;
- srv.grecfd = cfd;
- thread(greread, 0);
- }
- void
- greread(void *)
- {
- uchar buf[Pktsize], *p;
- int n, i;
- int flag, prot, len, callid;
- uchar src[IPaddrlen], dst[IPaddrlen];
- uint rseq, ack;
- Call *c;
- static double t, last;
- for(;;) {
- n = read(srv.grefd, buf, sizeof(buf));
- if(n < 0)
- myfatal("%I: bad read on gre: %r", srv.remote);
- if(n == sizeof(buf))
- myfatal("%I: gre read: buf too small", srv.remote);
- p = buf;
- v4tov6(src, p);
- v4tov6(dst, p+4);
- flag = GSHORT(p+8);
- prot = GSHORT(p+10);
- p += 12; n -= 12;
- if(ipcmp(src, srv.remote) != 0 || ipcmp(dst, srv.local) != 0)
- myfatal("%I: gre read bad address src=%I dst=%I", srv.remote, src, dst);
- if(prot != GRE_ppp)
- myfatal("%I: gre read gave bad protocol", srv.remote);
- if(flag & (GRE_chksum|GRE_routing)){
- p += 4; n -= 4;
- }
- if(!(flag&GRE_key))
- myfatal("%I: gre packet does not contain a key: f=%ux",
- srv.remote, flag);
- len = GSHORT(p);
- callid = GSHORT(p+2);
- p += 4; n -= 4;
- c = calllookup(callid);
- if(c == 0) {
- SDB "%I: unknown callid: %d\n", srv.remote, callid EDB
- continue;
- }
- qlock(&c->lk);
- c->stat.recv++;
- if(flag&GRE_seq) {
- rseq = GLONG(p);
- p += 4; n -= 4;
- } else
- rseq = c->rseq;
- if(flag&GRE_ack){
- ack = GLONG(p);
- p += 4; n -= 4;
- } else
- ack = c->ack;
- /* skip routing if present */
- if(flag&GRE_routing) {
- while((i=p[3]) != 0) {
- n -= i;
- p += i;
- }
- }
- if(len > n)
- myfatal("%I: bad len in gre packet", srv.remote);
- if((int)(ack-c->ack) > 0) {
- c->ack = ack;
- esignal(&c->eack);
- }
- if(debug)
- t = realtime();
- if(len == 0) {
- /* ack packet */
- c->stat.recvack++;
- SDB "%I: %.3f (%.3f): gre %d: recv ack a=%ux n=%d flag=%ux\n", srv.remote, t, t-last,
- c->id, ack, n, flag EDB
- } else {
- SDB "%I: %.3f (%.3f): gre %d: recv s=%ux a=%ux len=%d\n", srv.remote, t, t-last,
- c->id, rseq, ack, len EDB
- /*
- * the following handles the case of a single pair of packets
- * received out of order
- */
- n = rseq-c->rseq;
- if(n > 0 && (drop == 0. || frand() > drop)) {
- c->stat.missing += n-1;
- /* current packet */
- write(c->pppfd, p, len);
- } else {
- /* out of sequence - drop on the floor */
- c->stat.dropped++;
- SDB "%I: %.3f: gre %d: recv out of order or dup packet: seq=%ux len=%d\n",
- srv.remote, realtime(), c->id, rseq, len EDB
- }
- }
- if((int)(rseq-c->rseq) > 0)
- c->rseq = rseq;
- if(debug)
- last=t;
- /* open up client window */
- if((int)(c->rseq-c->rack) > (c->recvwindow>>1))
- greack(c);
- qunlock(&c->lk);
- callfree(c);
- }
- }
- void
- greack(Call *c)
- {
- uchar buf[20];
- c->stat.sendack++;
- SDB "%I: %.3f: gre %d: send ack %ux\n", srv.remote, realtime(), c->id, c->rseq EDB
- v6tov4(buf+0, srv.local); /* source */
- v6tov4(buf+4, srv.remote); /* source */
- PSHORT(buf+8, GRE_key|GRE_ack|1);
- PSHORT(buf+10, GRE_ppp);
- PSHORT(buf+12, 0);
- PSHORT(buf+14, c->id);
- PLONG(buf+16, c->rseq);
- write(srv.grefd, buf, sizeof(buf));
- c->rack = c->rseq;
- }
- void
- gretimeout(void *a)
- {
- Call *c;
- c = a;
- while(!c->closed) {
- sleep(Tick);
- qlock(&c->lk);
- c->tick++;
- qunlock(&c->lk);
- esignal(&c->eack);
- }
- callfree(c);
- exits(0);
- }
- void
- pppread(void *a)
- {
- Call *c;
- uchar buf[2000], *p;
- int n;
- ulong tick;
- c = a;
- for(;;) {
- p = buf+24;
- n = read(c->pppfd, p, sizeof(buf)-24);
- if(n <= 0)
- break;
-
- qlock(&c->lk);
- /* add gre header */
- c->seq++;
- tick = c->tick;
- while(c->seq-c->ack>c->sendwindow && c->tick-tick<Sendtimeout && !c->closed) {
- c->stat.sendwait++;
- SDB "window full seq = %d ack = %ux window = %ux\n", c->seq, c->ack, c->sendwindow EDB
- qunlock(&c->lk);
- ewait(&c->eack);
- qlock(&c->lk);
- }
-
- if(c->tick-tick >= Sendtimeout) {
- c->stat.sendtimeout++;
- SDB "send timeout = %d ack = %ux window = %ux\n", c->seq, c->ack, c->sendwindow EDB
- }
- v6tov4(buf+0, srv.local); /* source */
- v6tov4(buf+4, srv.remote); /* source */
- PSHORT(buf+8, GRE_key|GRE_seq|GRE_ack|1);
- PSHORT(buf+10, GRE_ppp);
- PSHORT(buf+12, n);
- PSHORT(buf+14, c->id);
- PLONG(buf+16, c->seq);
- PLONG(buf+20, c->rseq);
- c->stat.send++;
- c->rack = c->rseq;
- SDB "%I: %.3f: gre %d: send s=%ux a=%ux len=%d\n", srv.remote, realtime(),
- c->id, c->seq, c->rseq, n EDB
- if(drop == 0. || frand() > drop)
- if(write(srv.grefd, buf, n+24)<n+24)
- myfatal("pppread: write failed: %r");
- qunlock(&c->lk);
- }
- SDB "pppread exit: %d\n", c->id);
-
- callfree(c);
- exits(0);
- }
- void
- timeoutthread(void*)
- {
- for(;;) {
- sleep(30*1000);
- qlock(&srv.lk);
- if(realtime() - srv.rcvtime > 5*60)
- myfatal("server timedout");
- qunlock(&srv.lk);
- }
- }
- /* use syslog() rather than fprint(2, ...) */
- void
- myfatal(char *fmt, ...)
- {
- char sbuf[512];
- va_list arg;
- uchar buf[16];
- /* NT don't seem to like us just going away */
- memset(buf, 0, sizeof(buf));
- PSHORT(buf+0, sizeof(buf)); /* length */
- PSHORT(buf+2, 1); /* message type */
- PLONG(buf+4, Magic); /* magic */
- PSHORT(buf+8, Tstop); /* op */
- buf[12] = 3; /* local shutdown */
- write(1, buf, sizeof(buf));
- va_start(arg, fmt);
- vseprint(sbuf, sbuf+sizeof(sbuf), fmt, arg);
- va_end(arg);
- SDB "%I: fatal: %s\n", srv.remote, sbuf EDB
- syslog(0, LOG, ": src=%I: fatal: %s", srv.remote, sbuf);
- close(0);
- close(1);
- close(srv.grefd);
- close(srv.grecfd);
- postnote(PNGROUP, getpid(), "die");
- exits(sbuf);
- }
- int
- argatoi(char *p)
- {
- char *q;
- int i;
- if(p == 0)
- usage();
- i = strtol(p, &q, 0);
- if(q == p)
- usage();
- return i;
- }
- void
- dhcpclientwatch(void *a)
- {
- Call *c = a;
- uchar buf[1];
- for(;;) {
- if(read(c->dhcpfd[0], buf, sizeof(buf)) <= 0)
- break;
- }
- if(!c->closed)
- myfatal("dhcpclient terminated");
- callfree(c);
- exits(0);
- }
- int
- ipaddralloc(Call *c)
- {
- int pfd[2][2];
- char *argv[4], *p;
- Biobuf bio;
- argv[0] = "/bin/ip/dhcpclient";
- argv[1] = "-x";
- argv[2] = srv.pppdir;
- argv[3] = 0;
- if(pipe(pfd[0])<0)
- myfatal("ipaddralloc: pipe failed: %r");
- if(pipe(pfd[1])<0)
- myfatal("ipaddralloc: pipe failed: %r");
- if(proc(argv, pfd[0][0], pfd[1][1], 2) < 0)
- myfatal("ipaddralloc: proc failed: %r");
- close(pfd[0][0]);
- close(pfd[1][1]);
- c->dhcpfd[0] = pfd[1][0];
- c->dhcpfd[1] = pfd[0][1];
- Binit(&bio, pfd[1][0], OREAD);
- for(;;) {
- p = Brdline(&bio, '\n');
- if(p == 0)
- break;
- if(strncmp(p, "ip=", 3) == 0) {
- p += 3;
- parseip(c->remoteip, p);
- } else if(strncmp(p, "end\n", 4) == 0)
- break;
- }
- Bterm(&bio);
- c->ref++;
- thread(dhcpclientwatch, c);
- return ipcmp(c->remoteip, IPnoaddr) != 0;
- }
- void
- esignal(Event *e)
- {
- qlock(e);
- if(e->wait == 0) {
- e->ready = 1;
- qunlock(e);
- return;
- }
- assert(e->ready == 0);
- e->wait = 0;
- rendezvous(e, (void*)1);
- qunlock(e);
- }
- void
- ewait(Event *e)
- {
- qlock(&e->waitlk);
- qlock(e);
- assert(e->wait == 0);
- if(e->ready) {
- e->ready = 0;
- } else {
- e->wait = 1;
- qunlock(e);
- rendezvous(e, (void*)2);
- qlock(e);
- }
- qunlock(e);
- qunlock(&e->waitlk);
- }
- ulong
- thread(void(*f)(void*), void *a)
- {
- int pid;
- pid=rfork(RFNOWAIT|RFMEM|RFPROC);
- if(pid < 0)
- myfatal("rfork failed: %r");
- if(pid != 0)
- return pid;
- (*f)(a);
- return 0; // never reaches here
- }
- double
- realtime(void)
- {
- long times(long*);
- return times(0) / 1000.0;
- }
- void *
- emallocz(int size)
- {
- void *p;
- p = malloc(size);
- if(p == 0)
- myfatal("malloc failed: %r");
- memset(p, 0, size);
- return p;
- }
- static void
- fdclose(void)
- {
- int fd, n, i;
- Dir *d, *p;
- if((fd = open("#d", OREAD)) < 0)
- return;
- n = dirreadall(fd, &d);
- for(p = d; n > 0; n--, p++) {
- i = atoi(p->name);
- if(i > 2)
- close(i);
- }
- free(d);
- }
- int
- proc(char **argv, int fd0, int fd1, int fd2)
- {
- int r, flag;
- char *arg0, file[200];
- arg0 = argv[0];
- strcpy(file, arg0);
- if(access(file, 1) < 0) {
- if(strncmp(arg0, "/", 1)==0
- || strncmp(arg0, "#", 1)==0
- || strncmp(arg0, "./", 2)==0
- || strncmp(arg0, "../", 3)==0)
- return 0;
- sprint(file, "/bin/%s", arg0);
- if(access(file, 1) < 0)
- return 0;
- }
- flag = RFPROC|RFFDG|RFENVG|RFNOWAIT;
- if((r = rfork(flag)) != 0) {
- if(r < 0)
- return 0;
- return r;
- }
- if(fd0 != 0) {
- if(fd1 == 0)
- fd1 = dup(0, -1);
- if(fd2 == 0)
- fd2 = dup(0, -1);
- close(0);
- if(fd0 >= 0)
- dup(fd0, 0);
- }
- if(fd1 != 1) {
- if(fd2 == 1)
- fd2 = dup(1, -1);
- close(1);
- if(fd1 >= 0)
- dup(fd1, 1);
- }
- if(fd2 != 2) {
- close(2);
- if(fd2 >= 0)
- dup(fd2, 2);
- }
- fdclose();
- exec(file, argv);
- myfatal("proc: exec failed: %r");
- return 0;
- }
|