devbridge.c 23 KB

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