devip.c 24 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406
  1. #include "u.h"
  2. #include "../port/lib.h"
  3. #include "mem.h"
  4. #include "dat.h"
  5. #include "fns.h"
  6. #include "../port/error.h"
  7. #include "../ip/ip.h"
  8. enum
  9. {
  10. Qtopdir= 1, /* top level directory */
  11. Qtopbase,
  12. Qarp= Qtopbase,
  13. Qbootp,
  14. Qndb,
  15. Qiproute,
  16. Qipselftab,
  17. Qipgate6,
  18. Qlog,
  19. Qprotodir, /* directory for a protocol */
  20. Qprotobase,
  21. Qclone= Qprotobase,
  22. Qstats,
  23. Qconvdir, /* directory for a conversation */
  24. Qconvbase,
  25. Qctl= Qconvbase,
  26. Qdata,
  27. Qerr,
  28. Qlisten,
  29. Qlocal,
  30. Qremote,
  31. Qstatus,
  32. Qsnoop,
  33. Logtype= 5,
  34. Masktype= (1<<Logtype)-1,
  35. Logconv= 12,
  36. Maskconv= (1<<Logconv)-1,
  37. Shiftconv= Logtype,
  38. Logproto= 8,
  39. Maskproto= (1<<Logproto)-1,
  40. Shiftproto= Logtype + Logconv,
  41. Nfs= 16,
  42. };
  43. #define TYPE(x) ( ((ulong)(x).path) & Masktype )
  44. #define CONV(x) ( (((ulong)(x).path) >> Shiftconv) & Maskconv )
  45. #define PROTO(x) ( (((ulong)(x).path) >> Shiftproto) & Maskproto )
  46. #define QID(p, c, y) ( ((p)<<(Shiftproto)) | ((c)<<Shiftconv) | (y) )
  47. static char network[] = "network";
  48. QLock fslock;
  49. Fs *ipfs[Nfs]; /* attached fs's */
  50. Queue *qlog;
  51. extern void nullmediumlink(void);
  52. extern void pktmediumlink(void);
  53. long ndbwrite(Fs *f, char *a, ulong off, int n);
  54. static int
  55. ip3gen(Chan *c, int i, Dir *dp)
  56. {
  57. Qid q;
  58. Conv *cv;
  59. char *p;
  60. cv = ipfs[c->dev]->p[PROTO(c->qid)]->conv[CONV(c->qid)];
  61. if(cv->owner == nil)
  62. kstrdup(&cv->owner, eve);
  63. mkqid(&q, QID(PROTO(c->qid), CONV(c->qid), i), 0, QTFILE);
  64. switch(i) {
  65. default:
  66. return -1;
  67. case Qctl:
  68. devdir(c, q, "ctl", 0, cv->owner, cv->perm, dp);
  69. return 1;
  70. case Qdata:
  71. devdir(c, q, "data", qlen(cv->rq), cv->owner, cv->perm, dp);
  72. return 1;
  73. case Qerr:
  74. devdir(c, q, "err", qlen(cv->eq), cv->owner, cv->perm, dp);
  75. return 1;
  76. case Qlisten:
  77. devdir(c, q, "listen", 0, cv->owner, cv->perm, dp);
  78. return 1;
  79. case Qlocal:
  80. p = "local";
  81. break;
  82. case Qremote:
  83. p = "remote";
  84. break;
  85. case Qsnoop:
  86. if(strcmp(cv->p->name, "ipifc") != 0)
  87. return -1;
  88. devdir(c, q, "snoop", qlen(cv->sq), cv->owner, 0400, dp);
  89. return 1;
  90. case Qstatus:
  91. p = "status";
  92. break;
  93. }
  94. devdir(c, q, p, 0, cv->owner, 0444, dp);
  95. return 1;
  96. }
  97. static int
  98. ip2gen(Chan *c, int i, Dir *dp)
  99. {
  100. Qid q;
  101. switch(i) {
  102. case Qclone:
  103. mkqid(&q, QID(PROTO(c->qid), 0, Qclone), 0, QTFILE);
  104. devdir(c, q, "clone", 0, network, 0666, dp);
  105. return 1;
  106. case Qstats:
  107. mkqid(&q, QID(PROTO(c->qid), 0, Qstats), 0, QTFILE);
  108. devdir(c, q, "stats", 0, network, 0444, dp);
  109. return 1;
  110. }
  111. return -1;
  112. }
  113. static int
  114. ip1gen(Chan *c, int i, Dir *dp)
  115. {
  116. Qid q;
  117. char *p;
  118. int prot;
  119. int len = 0;
  120. Fs *f;
  121. extern ulong kerndate;
  122. f = ipfs[c->dev];
  123. prot = 0666;
  124. mkqid(&q, QID(0, 0, i), 0, QTFILE);
  125. switch(i) {
  126. default:
  127. return -1;
  128. case Qarp:
  129. p = "arp";
  130. break;
  131. case Qbootp:
  132. p = "bootp";
  133. break;
  134. case Qndb:
  135. p = "ndb";
  136. len = strlen(f->ndb);
  137. q.vers = f->ndbvers;
  138. break;
  139. case Qiproute:
  140. p = "iproute";
  141. break;
  142. case Qipselftab:
  143. p = "ipselftab";
  144. prot = 0444;
  145. break;
  146. case Qipgate6:
  147. p = "ipgate6";
  148. prot = 0444;
  149. break;
  150. case Qlog:
  151. p = "log";
  152. break;
  153. }
  154. devdir(c, q, p, len, network, prot, dp);
  155. if(i == Qndb && f->ndbmtime > kerndate)
  156. dp->mtime = f->ndbmtime;
  157. return 1;
  158. }
  159. static int
  160. ipgen(Chan *c, char*, Dirtab*, int, int s, Dir *dp)
  161. {
  162. Qid q;
  163. Conv *cv;
  164. Fs *f;
  165. f = ipfs[c->dev];
  166. switch(TYPE(c->qid)) {
  167. case Qtopdir:
  168. if(s == DEVDOTDOT){
  169. mkqid(&q, QID(0, 0, Qtopdir), 0, QTDIR);
  170. sprint(up->genbuf, "#I%lud", c->dev);
  171. devdir(c, q, up->genbuf, 0, network, 0555, dp);
  172. return 1;
  173. }
  174. if(s < f->np) {
  175. if(f->p[s]->connect == nil)
  176. return 0; /* protocol with no user interface */
  177. mkqid(&q, QID(s, 0, Qprotodir), 0, QTDIR);
  178. devdir(c, q, f->p[s]->name, 0, network, 0555, dp);
  179. return 1;
  180. }
  181. s -= f->np;
  182. return ip1gen(c, s+Qtopbase, dp);
  183. case Qarp:
  184. case Qbootp:
  185. case Qndb:
  186. case Qlog:
  187. case Qiproute:
  188. case Qipselftab:
  189. case Qipgate6:
  190. return ip1gen(c, TYPE(c->qid), dp);
  191. case Qprotodir:
  192. if(s == DEVDOTDOT){
  193. mkqid(&q, QID(0, 0, Qtopdir), 0, QTDIR);
  194. sprint(up->genbuf, "#I%lud", c->dev);
  195. devdir(c, q, up->genbuf, 0, network, 0555, dp);
  196. return 1;
  197. }
  198. if(s < f->p[PROTO(c->qid)]->ac) {
  199. cv = f->p[PROTO(c->qid)]->conv[s];
  200. sprint(up->genbuf, "%d", s);
  201. mkqid(&q, QID(PROTO(c->qid), s, Qconvdir), 0, QTDIR);
  202. devdir(c, q, up->genbuf, 0, cv->owner, 0555, dp);
  203. return 1;
  204. }
  205. s -= f->p[PROTO(c->qid)]->ac;
  206. return ip2gen(c, s+Qprotobase, dp);
  207. case Qclone:
  208. case Qstats:
  209. return ip2gen(c, TYPE(c->qid), dp);
  210. case Qconvdir:
  211. if(s == DEVDOTDOT){
  212. s = PROTO(c->qid);
  213. mkqid(&q, QID(s, 0, Qprotodir), 0, QTDIR);
  214. devdir(c, q, f->p[s]->name, 0, network, 0555, dp);
  215. return 1;
  216. }
  217. return ip3gen(c, s+Qconvbase, dp);
  218. case Qctl:
  219. case Qdata:
  220. case Qerr:
  221. case Qlisten:
  222. case Qlocal:
  223. case Qremote:
  224. case Qstatus:
  225. case Qsnoop:
  226. return ip3gen(c, TYPE(c->qid), dp);
  227. }
  228. return -1;
  229. }
  230. static void
  231. ipreset(void)
  232. {
  233. nullmediumlink();
  234. pktmediumlink();
  235. fmtinstall('i', eipfmt);
  236. fmtinstall('I', eipfmt);
  237. fmtinstall('E', eipfmt);
  238. fmtinstall('V', eipfmt);
  239. fmtinstall('M', eipfmt);
  240. }
  241. static Fs*
  242. ipgetfs(int dev)
  243. {
  244. extern void (*ipprotoinit[])(Fs*);
  245. Fs *f;
  246. int i;
  247. if(dev >= Nfs)
  248. return nil;
  249. qlock(&fslock);
  250. if(ipfs[dev] == nil){
  251. f = smalloc(sizeof(Fs));
  252. ip_init(f);
  253. arpinit(f);
  254. netloginit(f);
  255. for(i = 0; ipprotoinit[i]; i++)
  256. ipprotoinit[i](f);
  257. f->dev = dev;
  258. ipfs[dev] = f;
  259. }
  260. qunlock(&fslock);
  261. return ipfs[dev];
  262. }
  263. IPaux*
  264. newipaux(char *owner, char *tag)
  265. {
  266. IPaux *a;
  267. int n;
  268. a = smalloc(sizeof(*a));
  269. kstrdup(&a->owner, owner);
  270. memset(a->tag, ' ', sizeof(a->tag));
  271. n = strlen(tag);
  272. if(n > sizeof(a->tag))
  273. n = sizeof(a->tag);
  274. memmove(a->tag, tag, n);
  275. return a;
  276. }
  277. #define ATTACHER(c) (((IPaux*)((c)->aux))->owner)
  278. static Chan*
  279. ipattach(char* spec)
  280. {
  281. Chan *c;
  282. int dev;
  283. dev = atoi(spec);
  284. if(dev >= 16)
  285. error("bad specification");
  286. ipgetfs(dev);
  287. c = devattach('I', spec);
  288. mkqid(&c->qid, QID(0, 0, Qtopdir), 0, QTDIR);
  289. c->dev = dev;
  290. c->aux = newipaux(commonuser(), "none");
  291. return c;
  292. }
  293. static Walkqid*
  294. ipwalk(Chan* c, Chan *nc, char **name, int nname)
  295. {
  296. IPaux *a = c->aux;
  297. Walkqid* w;
  298. w = devwalk(c, nc, name, nname, nil, 0, ipgen);
  299. if(w != nil && w->clone != nil)
  300. w->clone->aux = newipaux(a->owner, a->tag);
  301. return w;
  302. }
  303. static int
  304. ipstat(Chan* c, uchar* db, int n)
  305. {
  306. return devstat(c, db, n, nil, 0, ipgen);
  307. }
  308. static int
  309. incoming(void* arg)
  310. {
  311. Conv *conv;
  312. conv = arg;
  313. return conv->incall != nil;
  314. }
  315. static int m2p[] = {
  316. [OREAD] 4,
  317. [OWRITE] 2,
  318. [ORDWR] 6
  319. };
  320. static Chan*
  321. ipopen(Chan* c, int omode)
  322. {
  323. Conv *cv, *nc;
  324. Proto *p;
  325. int perm;
  326. Fs *f;
  327. omode &= 3;
  328. perm = m2p[omode];
  329. f = ipfs[c->dev];
  330. switch(TYPE(c->qid)) {
  331. default:
  332. break;
  333. case Qndb:
  334. if(omode & (OWRITE|OTRUNC) && !iseve())
  335. error(Eperm);
  336. if((omode & (OWRITE|OTRUNC)) == (OWRITE|OTRUNC))
  337. f->ndb[0] = 0;
  338. break;
  339. case Qlog:
  340. netlogopen(f);
  341. break;
  342. case Qiproute:
  343. break;
  344. case Qtopdir:
  345. case Qprotodir:
  346. case Qconvdir:
  347. case Qstatus:
  348. case Qremote:
  349. case Qlocal:
  350. case Qstats:
  351. case Qbootp:
  352. case Qipselftab:
  353. case Qipgate6:
  354. if(omode != OREAD)
  355. error(Eperm);
  356. break;
  357. case Qsnoop:
  358. if(omode != OREAD)
  359. error(Eperm);
  360. p = f->p[PROTO(c->qid)];
  361. cv = p->conv[CONV(c->qid)];
  362. if(strcmp(ATTACHER(c), cv->owner) != 0 && !iseve())
  363. error(Eperm);
  364. incref(&cv->snoopers);
  365. break;
  366. case Qclone:
  367. p = f->p[PROTO(c->qid)];
  368. qlock(p);
  369. if(waserror()){
  370. qunlock(p);
  371. nexterror();
  372. }
  373. cv = Fsprotoclone(p, ATTACHER(c));
  374. qunlock(p);
  375. poperror();
  376. if(cv == nil) {
  377. error(Enodev);
  378. break;
  379. }
  380. mkqid(&c->qid, QID(p->x, cv->x, Qctl), 0, QTFILE);
  381. break;
  382. case Qdata:
  383. case Qctl:
  384. case Qerr:
  385. p = f->p[PROTO(c->qid)];
  386. qlock(p);
  387. cv = p->conv[CONV(c->qid)];
  388. qlock(cv);
  389. if(waserror()) {
  390. qunlock(cv);
  391. qunlock(p);
  392. nexterror();
  393. }
  394. if((perm & (cv->perm>>6)) != perm) {
  395. if(strcmp(ATTACHER(c), cv->owner) != 0)
  396. error(Eperm);
  397. if((perm & cv->perm) != perm)
  398. error(Eperm);
  399. }
  400. cv->inuse++;
  401. if(cv->inuse == 1){
  402. kstrdup(&cv->owner, ATTACHER(c));
  403. cv->perm = 0660;
  404. }
  405. qunlock(cv);
  406. qunlock(p);
  407. poperror();
  408. break;
  409. case Qlisten:
  410. cv = f->p[PROTO(c->qid)]->conv[CONV(c->qid)];
  411. if((perm & (cv->perm>>6)) != perm) {
  412. if(strcmp(ATTACHER(c), cv->owner) != 0)
  413. error(Eperm);
  414. if((perm & cv->perm) != perm)
  415. error(Eperm);
  416. }
  417. if(cv->state != Announced)
  418. error("not announced");
  419. nc = nil;
  420. while(nc == nil) {
  421. /* give up if we got a hangup */
  422. if(qisclosed(cv->rq))
  423. error("listen hungup");
  424. qlock(&cv->listenq);
  425. if(waserror()) {
  426. qunlock(&cv->listenq);
  427. nexterror();
  428. }
  429. /* wait for a connect */
  430. sleep(&cv->listenr, incoming, cv);
  431. qlock(cv);
  432. nc = cv->incall;
  433. if(nc != nil){
  434. cv->incall = nc->next;
  435. mkqid(&c->qid, QID(PROTO(c->qid), nc->x, Qctl), 0, QTFILE);
  436. kstrdup(&cv->owner, ATTACHER(c));
  437. }
  438. qunlock(cv);
  439. qunlock(&cv->listenq);
  440. poperror();
  441. }
  442. break;
  443. }
  444. c->mode = openmode(omode);
  445. c->flag |= COPEN;
  446. c->offset = 0;
  447. return c;
  448. }
  449. static void
  450. ipcreate(Chan*, char*, int, ulong)
  451. {
  452. error(Eperm);
  453. }
  454. static void
  455. ipremove(Chan*)
  456. {
  457. error(Eperm);
  458. }
  459. static int
  460. ipwstat(Chan *c, uchar *dp, int n)
  461. {
  462. Dir d;
  463. Conv *cv;
  464. Fs *f;
  465. Proto *p;
  466. f = ipfs[c->dev];
  467. switch(TYPE(c->qid)) {
  468. default:
  469. error(Eperm);
  470. break;
  471. case Qctl:
  472. case Qdata:
  473. break;
  474. }
  475. n = convM2D(dp, n, &d, nil);
  476. if(n > 0){
  477. p = f->p[PROTO(c->qid)];
  478. cv = p->conv[CONV(c->qid)];
  479. if(!iseve() && strcmp(ATTACHER(c), cv->owner) != 0)
  480. error(Eperm);
  481. if(d.uid[0])
  482. kstrdup(&cv->owner, d.uid);
  483. cv->perm = d.mode & 0777;
  484. }
  485. return n;
  486. }
  487. void
  488. closeconv(Conv *cv)
  489. {
  490. Conv *nc;
  491. Ipmulti *mp;
  492. qlock(cv);
  493. if(--cv->inuse > 0) {
  494. qunlock(cv);
  495. return;
  496. }
  497. /* close all incoming calls since no listen will ever happen */
  498. for(nc = cv->incall; nc; nc = cv->incall){
  499. cv->incall = nc->next;
  500. closeconv(nc);
  501. }
  502. cv->incall = nil;
  503. kstrdup(&cv->owner, network);
  504. cv->perm = 0660;
  505. while((mp = cv->multi) != nil)
  506. ipifcremmulti(cv, mp->ma, mp->ia);
  507. cv->r = nil;
  508. cv->rgen = 0;
  509. cv->p->close(cv);
  510. cv->state = Idle;
  511. qunlock(cv);
  512. }
  513. static void
  514. ipclose(Chan* c)
  515. {
  516. Fs *f;
  517. f = ipfs[c->dev];
  518. switch(TYPE(c->qid)) {
  519. default:
  520. break;
  521. case Qlog:
  522. if(c->flag & COPEN)
  523. netlogclose(f);
  524. break;
  525. case Qdata:
  526. case Qctl:
  527. case Qerr:
  528. if(c->flag & COPEN)
  529. closeconv(f->p[PROTO(c->qid)]->conv[CONV(c->qid)]);
  530. break;
  531. case Qsnoop:
  532. if(c->flag & COPEN)
  533. decref(&f->p[PROTO(c->qid)]->conv[CONV(c->qid)]->snoopers);
  534. break;
  535. }
  536. free(((IPaux*)c->aux)->owner);
  537. free(c->aux);
  538. }
  539. enum
  540. {
  541. Statelen= 32*1024,
  542. };
  543. static long
  544. ipread(Chan *ch, void *a, long n, vlong off)
  545. {
  546. Conv *c;
  547. Proto *x;
  548. char *buf, *p;
  549. long rv;
  550. Fs *f;
  551. ulong offset = off;
  552. f = ipfs[ch->dev];
  553. p = a;
  554. switch(TYPE(ch->qid)) {
  555. default:
  556. error(Eperm);
  557. case Qtopdir:
  558. case Qprotodir:
  559. case Qconvdir:
  560. return devdirread(ch, a, n, 0, 0, ipgen);
  561. case Qarp:
  562. return arpread(f->arp, a, offset, n);
  563. case Qbootp:
  564. return bootpread(a, offset, n);
  565. case Qndb:
  566. return readstr(offset, a, n, f->ndb);
  567. case Qiproute:
  568. return routeread(f, a, offset, n);
  569. case Qipselftab:
  570. return ipselftabread(f, a, offset, n);
  571. case Qipgate6:
  572. return ipgateread6(f, a, offset, n);
  573. case Qlog:
  574. return netlogread(f, a, offset, n);
  575. case Qctl:
  576. buf = smalloc(16);
  577. sprint(buf, "%lud", CONV(ch->qid));
  578. rv = readstr(offset, p, n, buf);
  579. free(buf);
  580. return rv;
  581. case Qremote:
  582. buf = smalloc(Statelen);
  583. x = f->p[PROTO(ch->qid)];
  584. c = x->conv[CONV(ch->qid)];
  585. if(x->remote == nil) {
  586. sprint(buf, "%I!%d\n", c->raddr, c->rport);
  587. } else {
  588. (*x->remote)(c, buf, Statelen-2);
  589. }
  590. rv = readstr(offset, p, n, buf);
  591. free(buf);
  592. return rv;
  593. case Qlocal:
  594. buf = smalloc(Statelen);
  595. x = f->p[PROTO(ch->qid)];
  596. c = x->conv[CONV(ch->qid)];
  597. if(x->local == nil) {
  598. sprint(buf, "%I!%d\n", c->laddr, c->lport);
  599. } else {
  600. (*x->local)(c, buf, Statelen-2);
  601. }
  602. rv = readstr(offset, p, n, buf);
  603. free(buf);
  604. return rv;
  605. case Qstatus:
  606. buf = smalloc(Statelen);
  607. x = f->p[PROTO(ch->qid)];
  608. c = x->conv[CONV(ch->qid)];
  609. (*x->state)(c, buf, Statelen-2);
  610. rv = readstr(offset, p, n, buf);
  611. free(buf);
  612. return rv;
  613. case Qdata:
  614. c = f->p[PROTO(ch->qid)]->conv[CONV(ch->qid)];
  615. return qread(c->rq, a, n);
  616. case Qerr:
  617. c = f->p[PROTO(ch->qid)]->conv[CONV(ch->qid)];
  618. return qread(c->eq, a, n);
  619. case Qsnoop:
  620. c = f->p[PROTO(ch->qid)]->conv[CONV(ch->qid)];
  621. return qread(c->sq, a, n);
  622. case Qstats:
  623. x = f->p[PROTO(ch->qid)];
  624. if(x->stats == nil)
  625. error("stats not implemented");
  626. buf = smalloc(Statelen);
  627. (*x->stats)(x, buf, Statelen);
  628. rv = readstr(offset, p, n, buf);
  629. free(buf);
  630. return rv;
  631. }
  632. }
  633. static Block*
  634. ipbread(Chan* ch, long n, ulong offset)
  635. {
  636. Conv *c;
  637. Proto *x;
  638. Fs *f;
  639. switch(TYPE(ch->qid)){
  640. case Qdata:
  641. f = ipfs[ch->dev];
  642. x = f->p[PROTO(ch->qid)];
  643. c = x->conv[CONV(ch->qid)];
  644. return qbread(c->rq, n);
  645. default:
  646. return devbread(ch, n, offset);
  647. }
  648. }
  649. /*
  650. * set local address to be that of the ifc closest to remote address
  651. */
  652. static void
  653. setladdr(Conv* c)
  654. {
  655. findlocalip(c->p->f, c->laddr, c->raddr);
  656. }
  657. /*
  658. * set a local port making sure the quad of raddr,rport,laddr,lport is unique
  659. */
  660. static char*
  661. setluniqueport(Conv* c, int lport)
  662. {
  663. Proto *p;
  664. Conv *xp;
  665. int x;
  666. p = c->p;
  667. qlock(p);
  668. for(x = 0; x < p->nc; x++){
  669. xp = p->conv[x];
  670. if(xp == nil)
  671. break;
  672. if(xp == c)
  673. continue;
  674. if((xp->state == Connected || xp->state == Announced)
  675. && xp->lport == lport
  676. && xp->rport == c->rport
  677. && ipcmp(xp->raddr, c->raddr) == 0
  678. && ipcmp(xp->laddr, c->laddr) == 0){
  679. qunlock(p);
  680. return "address in use";
  681. }
  682. }
  683. c->lport = lport;
  684. qunlock(p);
  685. return nil;
  686. }
  687. /*
  688. * pick a local port and set it
  689. */
  690. static void
  691. setlport(Conv* c)
  692. {
  693. Proto *p;
  694. ushort *pp;
  695. int x, found;
  696. p = c->p;
  697. if(c->restricted)
  698. pp = &p->nextrport;
  699. else
  700. pp = &p->nextport;
  701. qlock(p);
  702. for(;;(*pp)++){
  703. /*
  704. * Fsproto initialises p->nextport to 0 and the restricted
  705. * ports (p->nextrport) to 600.
  706. * Restricted ports must lie between 600 and 1024.
  707. * For the initial condition or if the unrestricted port number
  708. * has wrapped round, select a random port between 5000 and 1<<15
  709. * to start at.
  710. */
  711. if(c->restricted){
  712. if(*pp >= 1024)
  713. *pp = 600;
  714. }
  715. else while(*pp < 5000)
  716. *pp = nrand(1<<15);
  717. found = 0;
  718. for(x = 0; x < p->nc; x++){
  719. if(p->conv[x] == nil)
  720. break;
  721. if(p->conv[x]->lport == *pp){
  722. found = 1;
  723. break;
  724. }
  725. }
  726. if(!found)
  727. break;
  728. }
  729. c->lport = (*pp)++;
  730. qunlock(p);
  731. }
  732. /*
  733. * set a local address and port from a string of the form
  734. * [address!]port[!r]
  735. */
  736. static char*
  737. setladdrport(Conv* c, char* str, int announcing)
  738. {
  739. char *p;
  740. char *rv;
  741. ushort lport;
  742. rv = nil;
  743. /*
  744. * ignore restricted part if it exists. it's
  745. * meaningless on local ports.
  746. */
  747. p = strchr(str, '!');
  748. if(p != nil){
  749. *p++ = 0;
  750. if(strcmp(p, "r") == 0)
  751. p = nil;
  752. }
  753. c->lport = 0;
  754. if(p == nil){
  755. if(announcing)
  756. ipmove(c->laddr, IPnoaddr);
  757. else
  758. setladdr(c);
  759. p = str;
  760. } else {
  761. if(strcmp(str, "*") == 0)
  762. ipmove(c->laddr, IPnoaddr);
  763. else
  764. parseip(c->laddr, str);
  765. }
  766. /* one process can get all connections */
  767. if(announcing && strcmp(p, "*") == 0){
  768. if(!iseve())
  769. error(Eperm);
  770. return setluniqueport(c, 0);
  771. }
  772. lport = atoi(p);
  773. if(lport <= 0)
  774. setlport(c);
  775. else
  776. rv = setluniqueport(c, lport);
  777. return rv;
  778. }
  779. static char*
  780. setraddrport(Conv* c, char* str)
  781. {
  782. char *p;
  783. p = strchr(str, '!');
  784. if(p == nil)
  785. return "malformed address";
  786. *p++ = 0;
  787. parseip(c->raddr, str);
  788. c->rport = atoi(p);
  789. p = strchr(p, '!');
  790. if(p){
  791. if(strstr(p, "!r") != nil)
  792. c->restricted = 1;
  793. }
  794. return nil;
  795. }
  796. /*
  797. * called by protocol connect routine to set addresses
  798. */
  799. char*
  800. Fsstdconnect(Conv *c, char *argv[], int argc)
  801. {
  802. char *p;
  803. switch(argc) {
  804. default:
  805. return "bad args to connect";
  806. case 2:
  807. p = setraddrport(c, argv[1]);
  808. if(p != nil)
  809. return p;
  810. setladdr(c);
  811. setlport(c);
  812. break;
  813. case 3:
  814. p = setraddrport(c, argv[1]);
  815. if(p != nil)
  816. return p;
  817. p = setladdrport(c, argv[2], 0);
  818. if(p != nil)
  819. return p;
  820. }
  821. if( (memcmp(c->raddr, v4prefix, IPv4off) == 0 &&
  822. memcmp(c->laddr, v4prefix, IPv4off) == 0)
  823. || ipcmp(c->raddr, IPnoaddr) == 0)
  824. c->ipversion = V4;
  825. else
  826. c->ipversion = V6;
  827. return nil;
  828. }
  829. /*
  830. * initiate connection and sleep till its set up
  831. */
  832. static int
  833. connected(void* a)
  834. {
  835. return ((Conv*)a)->state == Connected;
  836. }
  837. static void
  838. connectctlmsg(Proto *x, Conv *c, Cmdbuf *cb)
  839. {
  840. char *p;
  841. if(c->state != 0)
  842. error(Econinuse);
  843. c->state = Connecting;
  844. c->cerr[0] = '\0';
  845. if(x->connect == nil)
  846. error("connect not supported");
  847. p = x->connect(c, cb->f, cb->nf);
  848. if(p != nil)
  849. error(p);
  850. qunlock(c);
  851. if(waserror()){
  852. qlock(c);
  853. nexterror();
  854. }
  855. sleep(&c->cr, connected, c);
  856. qlock(c);
  857. poperror();
  858. if(c->cerr[0] != '\0')
  859. error(c->cerr);
  860. }
  861. /*
  862. * called by protocol announce routine to set addresses
  863. */
  864. char*
  865. Fsstdannounce(Conv* c, char* argv[], int argc)
  866. {
  867. memset(c->raddr, 0, sizeof(c->raddr));
  868. c->rport = 0;
  869. switch(argc){
  870. default:
  871. return "bad args to announce";
  872. case 2:
  873. return setladdrport(c, argv[1], 1);
  874. }
  875. return nil;
  876. }
  877. /*
  878. * initiate announcement and sleep till its set up
  879. */
  880. static int
  881. announced(void* a)
  882. {
  883. return ((Conv*)a)->state == Announced;
  884. }
  885. static void
  886. announcectlmsg(Proto *x, Conv *c, Cmdbuf *cb)
  887. {
  888. char *p;
  889. if(c->state != 0)
  890. error(Econinuse);
  891. c->state = Announcing;
  892. c->cerr[0] = '\0';
  893. if(x->announce == nil)
  894. error("announce not supported");
  895. p = x->announce(c, cb->f, cb->nf);
  896. if(p != nil)
  897. error(p);
  898. qunlock(c);
  899. if(waserror()){
  900. qlock(c);
  901. nexterror();
  902. }
  903. sleep(&c->cr, announced, c);
  904. qlock(c);
  905. poperror();
  906. if(c->cerr[0] != '\0')
  907. error(c->cerr);
  908. }
  909. /*
  910. * called by protocol bind routine to set addresses
  911. */
  912. char*
  913. Fsstdbind(Conv* c, char* argv[], int argc)
  914. {
  915. switch(argc){
  916. default:
  917. return "bad args to bind";
  918. case 2:
  919. return setladdrport(c, argv[1], 0);
  920. }
  921. return nil;
  922. }
  923. static void
  924. bindctlmsg(Proto *x, Conv *c, Cmdbuf *cb)
  925. {
  926. char *p;
  927. if(x->bind == nil)
  928. p = Fsstdbind(c, cb->f, cb->nf);
  929. else
  930. p = x->bind(c, cb->f, cb->nf);
  931. if(p != nil)
  932. error(p);
  933. }
  934. static void
  935. tosctlmsg(Conv *c, Cmdbuf *cb)
  936. {
  937. if(cb->nf < 2)
  938. c->tos = 0;
  939. else
  940. c->tos = atoi(cb->f[1]);
  941. }
  942. static void
  943. ttlctlmsg(Conv *c, Cmdbuf *cb)
  944. {
  945. if(cb->nf < 2)
  946. c->ttl = MAXTTL;
  947. else
  948. c->ttl = atoi(cb->f[1]);
  949. }
  950. static long
  951. ipwrite(Chan* ch, void *v, long n, vlong off)
  952. {
  953. Conv *c;
  954. Proto *x;
  955. char *p;
  956. Cmdbuf *cb;
  957. uchar ia[IPaddrlen], ma[IPaddrlen];
  958. Fs *f;
  959. char *a;
  960. ulong offset = off;
  961. a = v;
  962. f = ipfs[ch->dev];
  963. switch(TYPE(ch->qid)){
  964. default:
  965. error(Eperm);
  966. case Qdata:
  967. x = f->p[PROTO(ch->qid)];
  968. c = x->conv[CONV(ch->qid)];
  969. if(c->wq == nil)
  970. error(Eperm);
  971. qwrite(c->wq, a, n);
  972. break;
  973. case Qarp:
  974. return arpwrite(f, a, n);
  975. case Qiproute:
  976. return routewrite(f, ch, a, n);
  977. case Qlog:
  978. netlogctl(f, a, n);
  979. return n;
  980. case Qndb:
  981. return ndbwrite(f, a, offset, n);
  982. break;
  983. case Qctl:
  984. x = f->p[PROTO(ch->qid)];
  985. c = x->conv[CONV(ch->qid)];
  986. cb = parsecmd(a, n);
  987. qlock(c);
  988. if(waserror()) {
  989. qunlock(c);
  990. free(cb);
  991. nexterror();
  992. }
  993. if(cb->nf < 1)
  994. error("short control request");
  995. if(strcmp(cb->f[0], "connect") == 0)
  996. connectctlmsg(x, c, cb);
  997. else if(strcmp(cb->f[0], "announce") == 0)
  998. announcectlmsg(x, c, cb);
  999. else if(strcmp(cb->f[0], "bind") == 0)
  1000. bindctlmsg(x, c, cb);
  1001. else if(strcmp(cb->f[0], "ttl") == 0)
  1002. ttlctlmsg(c, cb);
  1003. else if(strcmp(cb->f[0], "tos") == 0)
  1004. tosctlmsg(c, cb);
  1005. else if(strcmp(cb->f[0], "ignoreadvice") == 0)
  1006. c->ignoreadvice = 1;
  1007. else if(strcmp(cb->f[0], "addmulti") == 0){
  1008. if(cb->nf < 2)
  1009. error("addmulti needs interface address");
  1010. if(cb->nf == 2){
  1011. if(!ipismulticast(c->raddr))
  1012. error("addmulti for a non multicast address");
  1013. parseip(ia, cb->f[1]);
  1014. ipifcaddmulti(c, c->raddr, ia);
  1015. } else {
  1016. parseip(ma, cb->f[2]);
  1017. if(!ipismulticast(ma))
  1018. error("addmulti for a non multicast address");
  1019. parseip(ia, cb->f[1]);
  1020. ipifcaddmulti(c, ma, ia);
  1021. }
  1022. } else if(strcmp(cb->f[0], "remmulti") == 0){
  1023. if(cb->nf < 2)
  1024. error("remmulti needs interface address");
  1025. if(!ipismulticast(c->raddr))
  1026. error("remmulti for a non multicast address");
  1027. parseip(ia, cb->f[1]);
  1028. ipifcremmulti(c, c->raddr, ia);
  1029. } else if(x->ctl != nil) {
  1030. p = x->ctl(c, cb->f, cb->nf);
  1031. if(p != nil)
  1032. error(p);
  1033. } else
  1034. error("unknown control request");
  1035. qunlock(c);
  1036. free(cb);
  1037. poperror();
  1038. }
  1039. return n;
  1040. }
  1041. static long
  1042. ipbwrite(Chan* ch, Block* bp, ulong offset)
  1043. {
  1044. Conv *c;
  1045. Proto *x;
  1046. Fs *f;
  1047. int n;
  1048. switch(TYPE(ch->qid)){
  1049. case Qdata:
  1050. f = ipfs[ch->dev];
  1051. x = f->p[PROTO(ch->qid)];
  1052. c = x->conv[CONV(ch->qid)];
  1053. if(c->wq == nil)
  1054. error(Eperm);
  1055. if(bp->next)
  1056. bp = concatblock(bp);
  1057. n = BLEN(bp);
  1058. qbwrite(c->wq, bp);
  1059. return n;
  1060. default:
  1061. return devbwrite(ch, bp, offset);
  1062. }
  1063. }
  1064. Dev ipdevtab = {
  1065. 'I',
  1066. "ip",
  1067. ipreset,
  1068. devinit,
  1069. devshutdown,
  1070. ipattach,
  1071. ipwalk,
  1072. ipstat,
  1073. ipopen,
  1074. ipcreate,
  1075. ipclose,
  1076. ipread,
  1077. ipbread,
  1078. ipwrite,
  1079. ipbwrite,
  1080. ipremove,
  1081. ipwstat,
  1082. };
  1083. int
  1084. Fsproto(Fs *f, Proto *p)
  1085. {
  1086. if(f->np >= Maxproto)
  1087. return -1;
  1088. p->f = f;
  1089. if(p->ipproto > 0){
  1090. if(f->t2p[p->ipproto] != nil)
  1091. return -1;
  1092. f->t2p[p->ipproto] = p;
  1093. }
  1094. p->qid.type = QTDIR;
  1095. p->qid.path = QID(f->np, 0, Qprotodir);
  1096. p->conv = malloc(sizeof(Conv*)*(p->nc+1));
  1097. if(p->conv == nil)
  1098. panic("Fsproto");
  1099. p->x = f->np;
  1100. p->nextport = 0;
  1101. p->nextrport = 600;
  1102. f->p[f->np++] = p;
  1103. return 0;
  1104. }
  1105. /*
  1106. * return true if this protocol is
  1107. * built in
  1108. */
  1109. int
  1110. Fsbuiltinproto(Fs* f, uchar proto)
  1111. {
  1112. return f->t2p[proto] != nil;
  1113. }
  1114. /*
  1115. * called with protocol locked
  1116. */
  1117. Conv*
  1118. Fsprotoclone(Proto *p, char *user)
  1119. {
  1120. Conv *c, **pp, **ep;
  1121. retry:
  1122. c = nil;
  1123. ep = &p->conv[p->nc];
  1124. for(pp = p->conv; pp < ep; pp++) {
  1125. c = *pp;
  1126. if(c == nil){
  1127. c = malloc(sizeof(Conv));
  1128. if(c == nil)
  1129. error(Enomem);
  1130. qlock(c);
  1131. c->p = p;
  1132. c->x = pp - p->conv;
  1133. if(p->ptclsize != 0){
  1134. c->ptcl = malloc(p->ptclsize);
  1135. if(c->ptcl == nil) {
  1136. free(c);
  1137. error(Enomem);
  1138. }
  1139. }
  1140. *pp = c;
  1141. p->ac++;
  1142. c->eq = qopen(1024, Qmsg, 0, 0);
  1143. (*p->create)(c);
  1144. break;
  1145. }
  1146. if(canqlock(c)){
  1147. /*
  1148. * make sure both processes and protocol
  1149. * are done with this Conv
  1150. */
  1151. if(c->inuse == 0 && (p->inuse == nil || (*p->inuse)(c) == 0))
  1152. break;
  1153. qunlock(c);
  1154. }
  1155. }
  1156. if(pp >= ep) {
  1157. if(p->gc != nil && (*p->gc)(p))
  1158. goto retry;
  1159. return nil;
  1160. }
  1161. c->inuse = 1;
  1162. kstrdup(&c->owner, user);
  1163. c->perm = 0660;
  1164. c->state = Idle;
  1165. ipmove(c->laddr, IPnoaddr);
  1166. ipmove(c->raddr, IPnoaddr);
  1167. c->r = nil;
  1168. c->rgen = 0;
  1169. c->lport = 0;
  1170. c->rport = 0;
  1171. c->restricted = 0;
  1172. c->ttl = MAXTTL;
  1173. qreopen(c->rq);
  1174. qreopen(c->wq);
  1175. qreopen(c->eq);
  1176. qunlock(c);
  1177. return c;
  1178. }
  1179. int
  1180. Fsconnected(Conv* c, char* msg)
  1181. {
  1182. if(msg != nil && *msg != '\0')
  1183. strncpy(c->cerr, msg, ERRMAX-1);
  1184. switch(c->state){
  1185. case Announcing:
  1186. c->state = Announced;
  1187. break;
  1188. case Connecting:
  1189. c->state = Connected;
  1190. break;
  1191. }
  1192. wakeup(&c->cr);
  1193. return 0;
  1194. }
  1195. Proto*
  1196. Fsrcvpcol(Fs* f, uchar proto)
  1197. {
  1198. if(f->ipmux)
  1199. return f->ipmux;
  1200. else
  1201. return f->t2p[proto];
  1202. }
  1203. Proto*
  1204. Fsrcvpcolx(Fs *f, uchar proto)
  1205. {
  1206. return f->t2p[proto];
  1207. }
  1208. /*
  1209. * called with protocol locked
  1210. */
  1211. Conv*
  1212. Fsnewcall(Conv *c, uchar *raddr, ushort rport, uchar *laddr, ushort lport, uchar version)
  1213. {
  1214. Conv *nc;
  1215. Conv **l;
  1216. int i;
  1217. qlock(c);
  1218. i = 0;
  1219. for(l = &c->incall; *l; l = &(*l)->next)
  1220. i++;
  1221. if(i >= Maxincall) {
  1222. qunlock(c);
  1223. return nil;
  1224. }
  1225. /* find a free conversation */
  1226. nc = Fsprotoclone(c->p, network);
  1227. if(nc == nil) {
  1228. qunlock(c);
  1229. return nil;
  1230. }
  1231. ipmove(nc->raddr, raddr);
  1232. nc->rport = rport;
  1233. ipmove(nc->laddr, laddr);
  1234. nc->lport = lport;
  1235. nc->next = nil;
  1236. *l = nc;
  1237. nc->state = Connected;
  1238. nc->ipversion = version;
  1239. qunlock(c);
  1240. wakeup(&c->listenr);
  1241. return nc;
  1242. }
  1243. long
  1244. ndbwrite(Fs *f, char *a, ulong off, int n)
  1245. {
  1246. if(off > strlen(f->ndb))
  1247. error(Eio);
  1248. if(off+n >= sizeof(f->ndb))
  1249. error(Eio);
  1250. memmove(f->ndb+off, a, n);
  1251. f->ndb[off+n] = 0;
  1252. f->ndbvers++;
  1253. f->ndbmtime = seconds();
  1254. return n;
  1255. }
  1256. ulong
  1257. scalednconv(void)
  1258. {
  1259. if(cpuserver && conf.npage*BY2PG >= 128*MB)
  1260. return Nchans*4;
  1261. return Nchans;
  1262. }