devbridge.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206
  1. #include "u.h"
  2. #include "../port/lib.h"
  3. #include "mem.h"
  4. #include "dat.h"
  5. #include "fns.h"
  6. #include "../port/netif.h"
  7. #include "../port/error.h"
  8. typedef struct Bridge Bridge;
  9. typedef struct Port Port;
  10. typedef struct Centry Centry;
  11. typedef struct Iphdr Iphdr;
  12. typedef struct Tcphdr Tcphdr;
  13. enum
  14. {
  15. Qtopdir= 1, /* top level directory */
  16. Qbridgedir, /* bridge* directory */
  17. Qbctl,
  18. Qstats,
  19. Qcache,
  20. Qlog,
  21. Qportdir, /* directory for a protocol */
  22. Qpctl,
  23. Qlocal,
  24. Qstatus,
  25. MaxQ,
  26. Maxbridge= 4,
  27. Maxport= 128, // power of 2
  28. CacheHash= 257, // prime
  29. CacheLook= 5, // how many cache entries to examine
  30. CacheSize= (CacheHash+CacheLook-1),
  31. CacheTimeout= 5*60, // timeout for cache entry in seconds
  32. TcpMssMax = 1300, // max desirable Tcp MSS value
  33. TunnelMtu = 1400,
  34. };
  35. static Dirtab bridgedirtab[]={
  36. "ctl", {Qbctl}, 0, 0666,
  37. "stats", {Qstats}, 0, 0444,
  38. "cache", {Qcache}, 0, 0444,
  39. "log", {Qlog}, 0, 0666,
  40. };
  41. static Dirtab portdirtab[]={
  42. "ctl", {Qpctl}, 0, 0666,
  43. "local", {Qlocal}, 0, 0444,
  44. "status", {Qstatus}, 0, 0444,
  45. };
  46. enum {
  47. Logcache= (1<<0),
  48. Logmcast= (1<<1),
  49. };
  50. // types of interfaces
  51. enum
  52. {
  53. Tether,
  54. Ttun,
  55. };
  56. static Logflag logflags[] =
  57. {
  58. { "cache", Logcache, },
  59. { "multicast", Logmcast, },
  60. { nil, 0, },
  61. };
  62. static Dirtab *dirtab[MaxQ];
  63. #define TYPE(x) (((ulong)(x).path) & 0xff)
  64. #define PORT(x) ((((ulong)(x).path) >> 8)&(Maxport-1))
  65. #define QID(x, y) (((x)<<8) | (y))
  66. struct Centry
  67. {
  68. uchar d[Eaddrlen];
  69. int port;
  70. long expire; // entry expires this number of seconds after bootime
  71. long src;
  72. long dst;
  73. };
  74. struct Bridge
  75. {
  76. QLock;
  77. int nport;
  78. Port *port[Maxport];
  79. Centry cache[CacheSize];
  80. ulong hit;
  81. ulong miss;
  82. ulong copy;
  83. long delay0; // constant microsecond delay per packet
  84. long delayn; // microsecond delay per byte
  85. int tcpmss; // modify tcpmss value
  86. Log;
  87. };
  88. struct Port
  89. {
  90. int id;
  91. Bridge *bridge;
  92. int ref;
  93. int closed;
  94. Chan *data[2]; // channel to data
  95. int mcast; // send multi cast packets
  96. Proc *readp; // read proc
  97. // the following uniquely identifies the port
  98. int type;
  99. char name[KNAMELEN];
  100. // owner hash - avoids bind/unbind races
  101. ulong ownhash;
  102. // various stats
  103. int in; // number of packets read
  104. int inmulti; // multicast or broadcast
  105. int inunknown; // unknown address
  106. int out; // number of packets read
  107. int outmulti; // multicast or broadcast
  108. int outunknown; // unknown address
  109. int outfrag; // fragmented the packet
  110. int nentry; // number of cache entries for this port
  111. };
  112. enum {
  113. IP_VER = 0x40, /* Using IP version 4 */
  114. IP_HLEN = 0x05, /* Header length in characters */
  115. IP_DF = 0x4000, /* Don't fragment */
  116. IP_MF = 0x2000, /* More fragments */
  117. IP_MAX = (32*1024), /* Maximum Internet packet size */
  118. IP_TCPPROTO = 6,
  119. EOLOPT = 0,
  120. NOOPOPT = 1,
  121. MSSOPT = 2,
  122. MSS_LENGTH = 4, /* Mean segment size */
  123. SYN = 0x02, /* Pkt. is synchronise */
  124. IPHDR = 20, /* sizeof(Iphdr) */
  125. };
  126. struct Iphdr
  127. {
  128. uchar vihl; /* Version and header length */
  129. uchar tos; /* Type of service */
  130. uchar length[2]; /* packet length */
  131. uchar id[2]; /* ip->identification */
  132. uchar frag[2]; /* Fragment information */
  133. uchar ttl; /* Time to live */
  134. uchar proto; /* Protocol */
  135. uchar cksum[2]; /* Header checksum */
  136. uchar src[4]; /* IP source */
  137. uchar dst[4]; /* IP destination */
  138. };
  139. struct Tcphdr
  140. {
  141. uchar sport[2];
  142. uchar dport[2];
  143. uchar seq[4];
  144. uchar ack[4];
  145. uchar flag[2];
  146. uchar win[2];
  147. uchar cksum[2];
  148. uchar urg[2];
  149. };
  150. static Bridge bridgetab[Maxbridge];
  151. static int m2p[] = {
  152. [OREAD] 4,
  153. [OWRITE] 2,
  154. [ORDWR] 6
  155. };
  156. static int bridgegen(Chan *c, char*, Dirtab*, int, int s, Dir *dp);
  157. static void portbind(Bridge *b, int argc, char *argv[]);
  158. static void portunbind(Bridge *b, int argc, char *argv[]);
  159. static void etherread(void *a);
  160. static char *cachedump(Bridge *b);
  161. static void portfree(Port *port);
  162. static void cacheflushport(Bridge *b, int port);
  163. static void etherwrite(Port *port, Block *bp);
  164. extern ulong parseip(uchar*, char*);
  165. extern ushort ipcsum(uchar *addr);
  166. static void
  167. bridgeinit(void)
  168. {
  169. int i;
  170. Dirtab *dt;
  171. // setup dirtab with non directory entries
  172. for(i=0; i<nelem(bridgedirtab); i++) {
  173. dt = bridgedirtab + i;
  174. dirtab[TYPE(dt->qid)] = dt;
  175. }
  176. for(i=0; i<nelem(portdirtab); i++) {
  177. dt = portdirtab + i;
  178. dirtab[TYPE(dt->qid)] = dt;
  179. }
  180. }
  181. static Chan*
  182. bridgeattach(char* spec)
  183. {
  184. Chan *c;
  185. int dev;
  186. dev = atoi(spec);
  187. if(dev<0 || dev >= Maxbridge)
  188. error("bad specification");
  189. c = devattach('B', spec);
  190. mkqid(&c->qid, QID(0, Qtopdir), 0, QTDIR);
  191. c->dev = dev;
  192. return c;
  193. }
  194. static Walkqid*
  195. bridgewalk(Chan *c, Chan *nc, char **name, int nname)
  196. {
  197. return devwalk(c, nc, name, nname, (Dirtab*)0, 0, bridgegen);
  198. }
  199. static int
  200. bridgestat(Chan* c, uchar* db, int n)
  201. {
  202. return devstat(c, db, n, (Dirtab *)0, 0L, bridgegen);
  203. }
  204. static Chan*
  205. bridgeopen(Chan* c, int omode)
  206. {
  207. int perm;
  208. Bridge *b;
  209. omode &= 3;
  210. perm = m2p[omode];
  211. USED(perm);
  212. b = bridgetab + c->dev;
  213. USED(b);
  214. switch(TYPE(c->qid)) {
  215. default:
  216. break;
  217. case Qlog:
  218. logopen(b);
  219. break;
  220. case Qcache:
  221. c->aux = cachedump(b);
  222. break;
  223. }
  224. c->mode = openmode(omode);
  225. c->flag |= COPEN;
  226. c->offset = 0;
  227. return c;
  228. }
  229. static void
  230. bridgeclose(Chan* c)
  231. {
  232. Bridge *b = bridgetab + c->dev;
  233. switch(TYPE(c->qid)) {
  234. case Qcache:
  235. if(c->flag & COPEN)
  236. free(c->aux);
  237. break;
  238. case Qlog:
  239. if(c->flag & COPEN)
  240. logclose(b);
  241. break;
  242. }
  243. }
  244. static long
  245. bridgeread(Chan *c, void *a, long n, vlong off)
  246. {
  247. char buf[256];
  248. Bridge *b = bridgetab + c->dev;
  249. Port *port;
  250. int i, ingood, outgood;
  251. USED(off);
  252. switch(TYPE(c->qid)) {
  253. default:
  254. error(Eperm);
  255. case Qtopdir:
  256. case Qbridgedir:
  257. case Qportdir:
  258. return devdirread(c, a, n, 0, 0, bridgegen);
  259. case Qlog:
  260. return logread(b, a, off, n);
  261. case Qstatus:
  262. qlock(b);
  263. port = b->port[PORT(c->qid)];
  264. if(port == 0)
  265. strcpy(buf, "unbound\n");
  266. else {
  267. i = 0;
  268. switch(port->type) {
  269. default: panic("bridgeread: unknown port type: %d", port->type);
  270. case Tether:
  271. i += snprint(buf+i, sizeof(buf)-i, "ether %s: ", port->name);
  272. break;
  273. case Ttun:
  274. i += snprint(buf+i, sizeof(buf)-i, "tunnel %s: ", port->name);
  275. break;
  276. }
  277. ingood = port->in-port->inmulti-port->inunknown;
  278. outgood = port->out-port->outmulti-port->outunknown;
  279. i += snprint(buf+i, sizeof(buf)-i, "in=%d(%d:%d:%d) out=%d(%d:%d:%d:%d)\n",
  280. port->in, ingood, port->inmulti, port->inunknown,
  281. port->out, outgood, port->outmulti, port->outunknown, port->outfrag);
  282. USED(i);
  283. }
  284. n = readstr(off, a, n, buf);
  285. qunlock(b);
  286. return n;
  287. case Qbctl:
  288. snprint(buf, sizeof(buf), "%s tcpmss\ndelay %ld %ld\n", b->tcpmss ? "set" : "clear",
  289. b->delay0, b->delayn);
  290. n = readstr(off, a, n, buf);
  291. return n;
  292. case Qcache:
  293. n = readstr(off, a, n, c->aux);
  294. return n;
  295. case Qstats:
  296. snprint(buf, sizeof(buf), "hit=%uld miss=%uld copy=%uld\n",
  297. b->hit, b->miss, b->copy);
  298. n = readstr(off, a, n, buf);
  299. return n;
  300. }
  301. }
  302. static void
  303. bridgeoption(Bridge *b, char *option, int value)
  304. {
  305. if(strcmp(option, "tcpmss") == 0)
  306. b->tcpmss = value;
  307. else
  308. error("unknown bridge option");
  309. }
  310. static long
  311. bridgewrite(Chan *c, void *a, long n, vlong off)
  312. {
  313. Bridge *b = bridgetab + c->dev;
  314. Cmdbuf *cb;
  315. char *arg0;
  316. char *p;
  317. USED(off);
  318. switch(TYPE(c->qid)) {
  319. default:
  320. error(Eperm);
  321. case Qbctl:
  322. cb = parsecmd(a, n);
  323. qlock(b);
  324. if(waserror()) {
  325. qunlock(b);
  326. free(cb);
  327. nexterror();
  328. }
  329. if(cb->nf == 0)
  330. error("short write");
  331. arg0 = cb->f[0];
  332. if(strcmp(arg0, "bind") == 0) {
  333. portbind(b, cb->nf-1, cb->f+1);
  334. } else if(strcmp(arg0, "unbind") == 0) {
  335. portunbind(b, cb->nf-1, cb->f+1);
  336. } else if(strcmp(arg0, "cacheflush") == 0) {
  337. log(b, Logcache, "cache flush\n");
  338. memset(b->cache, 0, CacheSize*sizeof(Centry));
  339. } else if(strcmp(arg0, "set") == 0) {
  340. if(cb->nf != 2)
  341. error("usage: set option");
  342. bridgeoption(b, cb->f[1], 1);
  343. } else if(strcmp(arg0, "clear") == 0) {
  344. if(cb->nf != 2)
  345. error("usage: clear option");
  346. bridgeoption(b, cb->f[1], 0);
  347. } else if(strcmp(arg0, "delay") == 0) {
  348. if(cb->nf != 3)
  349. error("usage: delay delay0 delayn");
  350. b->delay0 = strtol(cb->f[1], nil, 10);
  351. b->delayn = strtol(cb->f[2], nil, 10);
  352. } else
  353. error("unknown control request");
  354. poperror();
  355. qunlock(b);
  356. free(cb);
  357. return n;
  358. case Qlog:
  359. cb = parsecmd(a, n);
  360. p = logctl(b, cb->nf, cb->f, logflags);
  361. free(cb);
  362. if(p != nil)
  363. error(p);
  364. return n;
  365. }
  366. }
  367. static int
  368. bridgegen(Chan *c, char *, Dirtab*, int, int s, Dir *dp)
  369. {
  370. Bridge *b = bridgetab + c->dev;
  371. int type = TYPE(c->qid);
  372. Dirtab *dt;
  373. Qid qid;
  374. if(s == DEVDOTDOT){
  375. switch(TYPE(c->qid)){
  376. case Qtopdir:
  377. case Qbridgedir:
  378. snprint(up->genbuf, sizeof(up->genbuf), "#B%ld", c->dev);
  379. mkqid(&qid, Qtopdir, 0, QTDIR);
  380. devdir(c, qid, up->genbuf, 0, eve, 0555, dp);
  381. break;
  382. case Qportdir:
  383. snprint(up->genbuf, sizeof(up->genbuf), "bridge%ld", c->dev);
  384. mkqid(&qid, Qbridgedir, 0, QTDIR);
  385. devdir(c, qid, up->genbuf, 0, eve, 0555, dp);
  386. break;
  387. default:
  388. panic("bridgewalk %llux", c->qid.path);
  389. }
  390. return 1;
  391. }
  392. switch(type) {
  393. default:
  394. // non directory entries end up here
  395. if(c->qid.type & QTDIR)
  396. panic("bridgegen: unexpected directory");
  397. if(s != 0)
  398. return -1;
  399. dt = dirtab[TYPE(c->qid)];
  400. if(dt == nil)
  401. panic("bridgegen: unknown type: %lud", TYPE(c->qid));
  402. devdir(c, c->qid, dt->name, dt->length, eve, dt->perm, dp);
  403. return 1;
  404. case Qtopdir:
  405. if(s != 0)
  406. return -1;
  407. snprint(up->genbuf, sizeof(up->genbuf), "bridge%ld", c->dev);
  408. mkqid(&qid, QID(0, Qbridgedir), 0, QTDIR);
  409. devdir(c, qid, up->genbuf, 0, eve, 0555, dp);
  410. return 1;
  411. case Qbridgedir:
  412. if(s<nelem(bridgedirtab)) {
  413. dt = bridgedirtab+s;
  414. devdir(c, dt->qid, dt->name, dt->length, eve, dt->perm, dp);
  415. return 1;
  416. }
  417. s -= nelem(bridgedirtab);
  418. if(s >= b->nport)
  419. return -1;
  420. mkqid(&qid, QID(s, Qportdir), 0, QTDIR);
  421. snprint(up->genbuf, sizeof(up->genbuf), "%d", s);
  422. devdir(c, qid, up->genbuf, 0, eve, 0555, dp);
  423. return 1;
  424. case Qportdir:
  425. if(s>=nelem(portdirtab))
  426. return -1;
  427. dt = portdirtab+s;
  428. mkqid(&qid, QID(PORT(c->qid),TYPE(dt->qid)), 0, QTFILE);
  429. devdir(c, qid, dt->name, dt->length, eve, dt->perm, dp);
  430. return 1;
  431. }
  432. }
  433. // also in netif.c
  434. static int
  435. parseaddr(uchar *to, char *from, int alen)
  436. {
  437. char nip[4];
  438. char *p;
  439. int i;
  440. p = from;
  441. for(i = 0; i < alen; i++){
  442. if(*p == 0)
  443. return -1;
  444. nip[0] = *p++;
  445. if(*p == 0)
  446. return -1;
  447. nip[1] = *p++;
  448. nip[2] = 0;
  449. to[i] = strtoul(nip, 0, 16);
  450. if(*p == ':')
  451. p++;
  452. }
  453. return 0;
  454. }
  455. // assumes b is locked
  456. static void
  457. portbind(Bridge *b, int argc, char *argv[])
  458. {
  459. Port *port;
  460. char path[8*KNAMELEN];
  461. char buf[100];
  462. char *dev, *dev2=nil, *p;
  463. Chan *ctl;
  464. int type=0, i, n;
  465. char *usage = "usage: bind ether|tunnel name ownhash dev [dev2]";
  466. char name[KNAMELEN];
  467. ulong ownhash;
  468. memset(name, 0, KNAMELEN);
  469. if(argc < 4)
  470. error(usage);
  471. if(strcmp(argv[0], "ether") == 0) {
  472. if(argc != 4)
  473. error(usage);
  474. type = Tether;
  475. strncpy(name, argv[1], KNAMELEN);
  476. name[KNAMELEN-1] = 0;
  477. // parseaddr(addr, argv[1], Eaddrlen);
  478. } else if(strcmp(argv[0], "tunnel") == 0) {
  479. if(argc != 5)
  480. error(usage);
  481. type = Ttun;
  482. strncpy(name, argv[1], KNAMELEN);
  483. name[KNAMELEN-1] = 0;
  484. // parseip(addr, argv[1]);
  485. dev2 = argv[4];
  486. } else
  487. error(usage);
  488. ownhash = atoi(argv[2]);
  489. dev = argv[3];
  490. for(i=0; i<b->nport; i++) {
  491. port = b->port[i];
  492. if(port != nil)
  493. if(port->type == type)
  494. if(memcmp(port->name, name, KNAMELEN) == 0)
  495. error("port in use");
  496. }
  497. for(i=0; i<Maxport; i++)
  498. if(b->port[i] == nil)
  499. break;
  500. if(i == Maxport)
  501. error("no more ports");
  502. port = smalloc(sizeof(Port));
  503. port->ref = 1;
  504. port->id = i;
  505. port->ownhash = ownhash;
  506. if(waserror()) {
  507. portfree(port);
  508. nexterror();
  509. }
  510. port->type = type;
  511. memmove(port->name, name, KNAMELEN);
  512. switch(port->type) {
  513. default: panic("portbind: unknown port type: %d", type);
  514. case Tether:
  515. snprint(path, sizeof(path), "%s/clone", dev);
  516. ctl = namec(path, Aopen, ORDWR, 0);
  517. if(waserror()) {
  518. cclose(ctl);
  519. nexterror();
  520. }
  521. // check addr?
  522. // get directory name
  523. n = devtab[ctl->type]->read(ctl, buf, sizeof(buf), 0);
  524. buf[n] = 0;
  525. for(p = buf; *p == ' '; p++)
  526. ;
  527. snprint(path, sizeof(path), "%s/%lud/data", dev, strtoul(p, 0, 0));
  528. // setup connection to be promiscuous
  529. snprint(buf, sizeof(buf), "connect -1");
  530. devtab[ctl->type]->write(ctl, buf, strlen(buf), 0);
  531. snprint(buf, sizeof(buf), "promiscuous");
  532. devtab[ctl->type]->write(ctl, buf, strlen(buf), 0);
  533. snprint(buf, sizeof(buf), "bridge");
  534. devtab[ctl->type]->write(ctl, buf, strlen(buf), 0);
  535. // open data port
  536. port->data[0] = namec(path, Aopen, ORDWR, 0);
  537. // dup it
  538. incref(port->data[0]);
  539. port->data[1] = port->data[0];
  540. poperror();
  541. cclose(ctl);
  542. break;
  543. case Ttun:
  544. port->data[0] = namec(dev, Aopen, OREAD, 0);
  545. port->data[1] = namec(dev2, Aopen, OWRITE, 0);
  546. break;
  547. }
  548. poperror();
  549. // commited to binding port
  550. b->port[port->id] = port;
  551. port->bridge = b;
  552. if(b->nport <= port->id)
  553. b->nport = port->id+1;
  554. // assumes kproc always succeeds
  555. kproc("etherread", etherread, port); // poperror must be next
  556. port->ref++;
  557. }
  558. // assumes b is locked
  559. static void
  560. portunbind(Bridge *b, int argc, char *argv[])
  561. {
  562. Port *port=nil;
  563. int type=0, i;
  564. char *usage = "usage: unbind ether|tunnel addr [ownhash]";
  565. char name[KNAMELEN];
  566. ulong ownhash;
  567. memset(name, 0, KNAMELEN);
  568. if(argc < 2 || argc > 3)
  569. error(usage);
  570. if(strcmp(argv[0], "ether") == 0) {
  571. type = Tether;
  572. strncpy(name, argv[1], KNAMELEN);
  573. name[KNAMELEN-1] = 0;
  574. // parseaddr(addr, argv[1], Eaddrlen);
  575. } else if(strcmp(argv[0], "tunnel") == 0) {
  576. type = Ttun;
  577. strncpy(name, argv[1], KNAMELEN);
  578. name[KNAMELEN-1] = 0;
  579. // parseip(addr, argv[1]);
  580. } else
  581. error(usage);
  582. if(argc == 3)
  583. ownhash = atoi(argv[2]);
  584. else
  585. ownhash = 0;
  586. for(i=0; i<b->nport; i++) {
  587. port = b->port[i];
  588. if(port != nil)
  589. if(port->type == type)
  590. if(memcmp(port->name, name, KNAMELEN) == 0)
  591. break;
  592. }
  593. if(i == b->nport)
  594. error("port not found");
  595. if(ownhash != 0 && port->ownhash != 0 && ownhash != port->ownhash)
  596. error("bad owner hash");
  597. port->closed = 1;
  598. b->port[i] = nil; // port is now unbound
  599. cacheflushport(b, i);
  600. // try and stop reader
  601. if(port->readp)
  602. postnote(port->readp, 1, "unbind", 0);
  603. portfree(port);
  604. }
  605. // assumes b is locked
  606. static Centry *
  607. cachelookup(Bridge *b, uchar d[Eaddrlen])
  608. {
  609. int i;
  610. uint h;
  611. Centry *p;
  612. long sec;
  613. // dont cache multicast or broadcast
  614. if(d[0] & 1)
  615. return 0;
  616. h = 0;
  617. for(i=0; i<Eaddrlen; i++) {
  618. h *= 7;
  619. h += d[i];
  620. }
  621. h %= CacheHash;
  622. p = b->cache + h;
  623. sec = TK2SEC(m->ticks);
  624. for(i=0; i<CacheLook; i++,p++) {
  625. if(memcmp(d, p->d, Eaddrlen) == 0) {
  626. p->dst++;
  627. if(sec >= p->expire) {
  628. log(b, Logcache, "expired cache entry: %E %d\n",
  629. d, p->port);
  630. return nil;
  631. }
  632. p->expire = sec + CacheTimeout;
  633. return p;
  634. }
  635. }
  636. log(b, Logcache, "cache miss: %E\n", d);
  637. return nil;
  638. }
  639. // assumes b is locked
  640. static void
  641. cacheupdate(Bridge *b, uchar d[Eaddrlen], int port)
  642. {
  643. int i;
  644. uint h;
  645. Centry *p, *pp;
  646. long sec;
  647. // dont cache multicast or broadcast
  648. if(d[0] & 1) {
  649. log(b, Logcache, "bad source address: %E\n", d);
  650. return;
  651. }
  652. h = 0;
  653. for(i=0; i<Eaddrlen; i++) {
  654. h *= 7;
  655. h += d[i];
  656. }
  657. h %= CacheHash;
  658. p = b->cache + h;
  659. pp = p;
  660. sec = p->expire;
  661. // look for oldest entry
  662. for(i=0; i<CacheLook; i++,p++) {
  663. if(memcmp(p->d, d, Eaddrlen) == 0) {
  664. p->expire = TK2SEC(m->ticks) + CacheTimeout;
  665. if(p->port != port) {
  666. log(b, Logcache, "NIC changed port %d->%d: %E\n",
  667. p->port, port, d);
  668. p->port = port;
  669. }
  670. p->src++;
  671. return;
  672. }
  673. if(p->expire < sec) {
  674. sec = p->expire;
  675. pp = p;
  676. }
  677. }
  678. if(pp->expire != 0)
  679. log(b, Logcache, "bumping from cache: %E %d\n", pp->d, pp->port);
  680. pp->expire = TK2SEC(m->ticks) + CacheTimeout;
  681. memmove(pp->d, d, Eaddrlen);
  682. pp->port = port;
  683. pp->src = 1;
  684. pp->dst = 0;
  685. log(b, Logcache, "adding to cache: %E %d\n", pp->d, pp->port);
  686. }
  687. // assumes b is locked
  688. static void
  689. cacheflushport(Bridge *b, int port)
  690. {
  691. Centry *ce;
  692. int i;
  693. ce = b->cache;
  694. for(i=0; i<CacheSize; i++,ce++) {
  695. if(ce->port != port)
  696. continue;
  697. memset(ce, 0, sizeof(Centry));
  698. }
  699. }
  700. static char *
  701. cachedump(Bridge *b)
  702. {
  703. int i, n;
  704. long sec, off;
  705. char *buf, *p, *ep;
  706. Centry *ce;
  707. char c;
  708. qlock(b);
  709. if(waserror()) {
  710. qunlock(b);
  711. nexterror();
  712. }
  713. sec = TK2SEC(m->ticks);
  714. n = 0;
  715. for(i=0; i<CacheSize; i++)
  716. if(b->cache[i].expire != 0)
  717. n++;
  718. n *= 51; // change if print format is changed
  719. n += 10; // some slop at the end
  720. buf = malloc(n);
  721. p = buf;
  722. ep = buf + n;
  723. ce = b->cache;
  724. off = seconds() - sec;
  725. for(i=0; i<CacheSize; i++,ce++) {
  726. if(ce->expire == 0)
  727. continue;
  728. c = (sec < ce->expire)?'v':'e';
  729. p += snprint(p, ep-p, "%E %2d %10ld %10ld %10ld %c\n", ce->d,
  730. ce->port, ce->src, ce->dst, ce->expire+off, c);
  731. }
  732. *p = 0;
  733. poperror();
  734. qunlock(b);
  735. return buf;
  736. }
  737. // assumes b is locked
  738. static void
  739. ethermultiwrite(Bridge *b, Block *bp, Port *port)
  740. {
  741. Port *oport;
  742. Block *bp2;
  743. Etherpkt *ep;
  744. int i, mcast, bcast;
  745. static uchar bcastaddr[Eaddrlen] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
  746. if(waserror()) {
  747. if(bp)
  748. freeb(bp);
  749. nexterror();
  750. }
  751. ep = (Etherpkt*)bp->rp;
  752. mcast = ep->d[0] & 1;
  753. if(mcast)
  754. bcast = memcmp(ep->d, bcastaddr, Eaddrlen) == 0;
  755. else
  756. bcast = 0;
  757. oport = nil;
  758. for(i=0; i<b->nport; i++) {
  759. if(i == port->id || b->port[i] == nil)
  760. continue;
  761. if(mcast && !bcast && !b->port[i]->mcast)
  762. continue;
  763. if(mcast)
  764. b->port[i]->outmulti++;
  765. else
  766. b->port[i]->outunknown++;
  767. // delay one so that the last write does not copy
  768. if(oport != nil) {
  769. b->copy++;
  770. bp2 = copyblock(bp, blocklen(bp));
  771. if(!waserror()) {
  772. etherwrite(oport, bp2);
  773. poperror();
  774. }
  775. }
  776. oport = b->port[i];
  777. }
  778. // last write free block
  779. if(oport) {
  780. bp2 = bp; bp = nil; USED(bp);
  781. if(!waserror()) {
  782. etherwrite(oport, bp2);
  783. poperror();
  784. }
  785. } else
  786. freeb(bp);
  787. poperror();
  788. }
  789. static void
  790. tcpmsshack(Etherpkt *epkt, int n)
  791. {
  792. int hl;
  793. Iphdr *iphdr;
  794. Tcphdr *tcphdr;
  795. ulong mss;
  796. ulong cksum;
  797. int optlen;
  798. uchar *optr;
  799. // check it is an ip packet
  800. if(nhgets(epkt->type) != 0x800)
  801. return;
  802. iphdr = (Iphdr*)(epkt->data);
  803. n -= ETHERHDRSIZE;
  804. if(n < IPHDR)
  805. return;
  806. // check it is ok IP packet
  807. if(iphdr->vihl != (IP_VER|IP_HLEN)) {
  808. hl = (iphdr->vihl&0xF)<<2;
  809. if((iphdr->vihl&0xF0) != IP_VER || hl < (IP_HLEN<<2))
  810. return;
  811. } else
  812. hl = IP_HLEN<<2;
  813. // check TCP
  814. if(iphdr->proto != IP_TCPPROTO)
  815. return;
  816. n -= hl;
  817. if(n < sizeof(Tcphdr))
  818. return;
  819. tcphdr = (Tcphdr*)((uchar*)(iphdr) + hl);
  820. // MSS can only appear in SYN packet
  821. if(!(tcphdr->flag[1] & SYN))
  822. return;
  823. hl = (tcphdr->flag[0] & 0xf0)>>2;
  824. if(n < hl)
  825. return;
  826. // check for MSS option
  827. optr = (uchar*)(tcphdr) + sizeof(Tcphdr);
  828. n = hl - sizeof(Tcphdr);
  829. for(;;) {
  830. if(n <= 0 || *optr == EOLOPT)
  831. return;
  832. if(*optr == NOOPOPT) {
  833. n--;
  834. optr++;
  835. continue;
  836. }
  837. optlen = optr[1];
  838. if(optlen < 2 || optlen > n)
  839. return;
  840. if(*optr == MSSOPT && optlen == MSS_LENGTH)
  841. break;
  842. n -= optlen;
  843. optr += optlen;
  844. }
  845. mss = nhgets(optr+2);
  846. if(mss <= TcpMssMax)
  847. return;
  848. // fit checksum
  849. cksum = nhgets(tcphdr->cksum);
  850. if(optr-(uchar*)tcphdr & 1) {
  851. print("tcpmsshack: odd alignment!\n");
  852. // odd alignments are a pain
  853. cksum += nhgets(optr+1);
  854. cksum -= (optr[1]<<8)|(TcpMssMax>>8);
  855. cksum += (cksum>>16);
  856. cksum &= 0xffff;
  857. cksum += nhgets(optr+3);
  858. cksum -= ((TcpMssMax&0xff)<<8)|optr[4];
  859. cksum += (cksum>>16);
  860. } else {
  861. cksum += mss;
  862. cksum -= TcpMssMax;
  863. cksum += (cksum>>16);
  864. }
  865. hnputs(tcphdr->cksum, cksum);
  866. hnputs(optr+2, TcpMssMax);
  867. }
  868. /*
  869. * process to read from the ethernet
  870. */
  871. static void
  872. etherread(void *a)
  873. {
  874. Port *port = a;
  875. Bridge *b = port->bridge;
  876. Block *bp, *bp2;
  877. Etherpkt *ep;
  878. Centry *ce;
  879. long md;
  880. qlock(b);
  881. port->readp = up; /* hide identity under a rock for unbind */
  882. while(!port->closed){
  883. // release lock to read - error means it is time to quit
  884. qunlock(b);
  885. if(waserror()) {
  886. print("etherread read error: %s\n", up->errstr);
  887. qlock(b);
  888. break;
  889. }
  890. if(0)print("devbridge: etherread: reading\n");
  891. bp = devtab[port->data[0]->type]->bread(port->data[0], ETHERMAXTU, 0);
  892. if(0)print("devbridge: etherread: blocklen = %d\n", blocklen(bp));
  893. poperror();
  894. qlock(b);
  895. if(bp == nil || port->closed)
  896. break;
  897. if(waserror()) {
  898. //print("etherread bridge error\n");
  899. if(bp)
  900. freeb(bp);
  901. continue;
  902. }
  903. if(blocklen(bp) < ETHERMINTU)
  904. error("short packet");
  905. port->in++;
  906. ep = (Etherpkt*)bp->rp;
  907. cacheupdate(b, ep->s, port->id);
  908. if(b->tcpmss)
  909. tcpmsshack(ep, BLEN(bp));
  910. /*
  911. * delay packets to simulate a slow link
  912. */
  913. if(b->delay0 || b->delayn){
  914. md = b->delay0 + b->delayn * BLEN(bp);
  915. if(md > 0)
  916. microdelay(md);
  917. }
  918. if(ep->d[0] & 1) {
  919. log(b, Logmcast, "mulitcast: port=%d src=%E dst=%E type=%#.4ux\n",
  920. port->id, ep->s, ep->d, (ep->type[0]<<8)|ep->type[1] );
  921. port->inmulti++;
  922. bp2 = bp; bp = nil;
  923. ethermultiwrite(b, bp2, port);
  924. } else {
  925. ce = cachelookup(b, ep->d);
  926. if(ce == nil) {
  927. b->miss++;
  928. port->inunknown++;
  929. bp2 = bp; bp = nil;
  930. ethermultiwrite(b, bp2, port);
  931. }else if(ce->port != port->id){
  932. b->hit++;
  933. bp2 = bp; bp = nil;
  934. etherwrite(b->port[ce->port], bp2);
  935. }
  936. }
  937. poperror();
  938. if(bp)
  939. freeb(bp);
  940. }
  941. //print("etherread: trying to exit\n");
  942. port->readp = nil;
  943. portfree(port);
  944. qunlock(b);
  945. pexit("hangup", 1);
  946. }
  947. static int
  948. fragment(Etherpkt *epkt, int n)
  949. {
  950. Iphdr *iphdr;
  951. if(n <= TunnelMtu)
  952. return 0;
  953. // check it is an ip packet
  954. if(nhgets(epkt->type) != 0x800)
  955. return 0;
  956. iphdr = (Iphdr*)(epkt->data);
  957. n -= ETHERHDRSIZE;
  958. if(n < IPHDR)
  959. return 0;
  960. // check it is ok IP packet - I don't handle IP options for the momment
  961. if(iphdr->vihl != (IP_VER|IP_HLEN))
  962. return 0;
  963. // check for don't fragment
  964. if(iphdr->frag[0] & (IP_DF>>8))
  965. return 0;
  966. // check for short block
  967. if(nhgets(iphdr->length) > n)
  968. return 0;
  969. return 1;
  970. }
  971. static void
  972. etherwrite(Port *port, Block *bp)
  973. {
  974. Iphdr *eh, *feh;
  975. Etherpkt *epkt;
  976. int n, lid, len, seglen, chunk, dlen, blklen, offset, mf;
  977. Block *xp, *nb;
  978. ushort fragoff, frag;
  979. port->out++;
  980. epkt = (Etherpkt*)bp->rp;
  981. n = blocklen(bp);
  982. if(port->type != Ttun || !fragment(epkt, n)) {
  983. devtab[port->data[1]->type]->bwrite(port->data[1], bp, 0);
  984. return;
  985. }
  986. port->outfrag++;
  987. if(waserror()){
  988. freeblist(bp);
  989. nexterror();
  990. }
  991. seglen = (TunnelMtu - ETHERHDRSIZE - IPHDR) & ~7;
  992. eh = (Iphdr*)(epkt->data);
  993. len = nhgets(eh->length);
  994. frag = nhgets(eh->frag);
  995. mf = frag & IP_MF;
  996. frag <<= 3;
  997. dlen = len - IPHDR;
  998. xp = bp;
  999. lid = nhgets(eh->id);
  1000. offset = ETHERHDRSIZE+IPHDR;
  1001. while(xp != nil && offset && offset >= BLEN(xp)) {
  1002. offset -= BLEN(xp);
  1003. xp = xp->next;
  1004. }
  1005. xp->rp += offset;
  1006. if(0) print("seglen=%d, dlen=%d, mf=%x, frag=%d\n", seglen, dlen, mf, frag);
  1007. for(fragoff = 0; fragoff < dlen; fragoff += seglen) {
  1008. nb = allocb(ETHERHDRSIZE+IPHDR+seglen);
  1009. feh = (Iphdr*)(nb->wp+ETHERHDRSIZE);
  1010. memmove(nb->wp, epkt, ETHERHDRSIZE+IPHDR);
  1011. nb->wp += ETHERHDRSIZE+IPHDR;
  1012. if((fragoff + seglen) >= dlen) {
  1013. seglen = dlen - fragoff;
  1014. hnputs(feh->frag, (frag+fragoff)>>3 | mf);
  1015. }
  1016. else
  1017. hnputs(feh->frag, (frag+fragoff>>3) | IP_MF);
  1018. hnputs(feh->length, seglen + IPHDR);
  1019. hnputs(feh->id, lid);
  1020. /* Copy up the data area */
  1021. chunk = seglen;
  1022. while(chunk) {
  1023. blklen = chunk;
  1024. if(BLEN(xp) < chunk)
  1025. blklen = BLEN(xp);
  1026. memmove(nb->wp, xp->rp, blklen);
  1027. nb->wp += blklen;
  1028. xp->rp += blklen;
  1029. chunk -= blklen;
  1030. if(xp->rp == xp->wp)
  1031. xp = xp->next;
  1032. }
  1033. feh->cksum[0] = 0;
  1034. feh->cksum[1] = 0;
  1035. hnputs(feh->cksum, ipcsum(&feh->vihl));
  1036. // don't generate small packets
  1037. if(BLEN(nb) < ETHERMINTU)
  1038. nb->wp = nb->rp + ETHERMINTU;
  1039. devtab[port->data[1]->type]->bwrite(port->data[1], nb, 0);
  1040. }
  1041. poperror();
  1042. freeblist(bp);
  1043. }
  1044. // hold b lock
  1045. static void
  1046. portfree(Port *port)
  1047. {
  1048. port->ref--;
  1049. if(port->ref < 0)
  1050. panic("portfree: bad ref");
  1051. if(port->ref > 0)
  1052. return;
  1053. if(port->data[0])
  1054. cclose(port->data[0]);
  1055. if(port->data[1])
  1056. cclose(port->data[1]);
  1057. memset(port, 0, sizeof(Port));
  1058. free(port);
  1059. }
  1060. Dev bridgedevtab = {
  1061. 'B',
  1062. "bridge",
  1063. devreset,
  1064. bridgeinit,
  1065. devshutdown,
  1066. bridgeattach,
  1067. bridgewalk,
  1068. bridgestat,
  1069. bridgeopen,
  1070. devcreate,
  1071. bridgeclose,
  1072. bridgeread,
  1073. devbread,
  1074. bridgewrite,
  1075. devbwrite,
  1076. devremove,
  1077. devwstat,
  1078. };