cs.c 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851
  1. #include <u.h>
  2. #include <libc.h>
  3. #include <auth.h>
  4. #include <fcall.h>
  5. #include <bio.h>
  6. #include <ctype.h>
  7. #include <ndb.h>
  8. #include <ip.h>
  9. #include <String.h>
  10. enum
  11. {
  12. Nreply= 20,
  13. Maxreply= 256,
  14. Maxrequest= 128,
  15. Maxpath= 128,
  16. Maxfdata= 8192,
  17. Maxhost= 64, /* maximum host name size */
  18. Maxservice= 64, /* maximum service name size */
  19. Qdir= 0,
  20. Qcs= 1,
  21. };
  22. typedef struct Mfile Mfile;
  23. typedef struct Mlist Mlist;
  24. typedef struct Network Network;
  25. typedef struct Flushreq Flushreq;
  26. typedef struct Job Job;
  27. int vers; /* incremented each clone/attach */
  28. struct Mfile
  29. {
  30. int busy;
  31. char *user;
  32. Qid qid;
  33. int fid;
  34. /*
  35. * current request
  36. */
  37. char *net;
  38. char *host;
  39. char *serv;
  40. char *rem;
  41. /*
  42. * result of the last lookup
  43. */
  44. Network *nextnet;
  45. int nreply;
  46. char *reply[Nreply];
  47. int replylen[Nreply];
  48. };
  49. struct Mlist
  50. {
  51. Mlist *next;
  52. Mfile mf;
  53. };
  54. //
  55. // active requests
  56. //
  57. struct Job
  58. {
  59. Job *next;
  60. int flushed;
  61. Fcall request;
  62. Fcall reply;
  63. };
  64. Lock joblock;
  65. Job *joblist;
  66. Mlist *mlist;
  67. int mfd[2];
  68. int debug;
  69. int paranoia;
  70. jmp_buf masterjmp; /* return through here after a slave process has been created */
  71. int *isslave; /* *isslave non-zero means this is a slave process */
  72. char *dbfile;
  73. Ndb *db, *netdb;
  74. void rversion(Job*);
  75. void rflush(Job*);
  76. void rattach(Job*, Mfile*);
  77. char* rwalk(Job*, Mfile*);
  78. void ropen(Job*, Mfile*);
  79. void rcreate(Job*, Mfile*);
  80. void rread(Job*, Mfile*);
  81. void rwrite(Job*, Mfile*);
  82. void rclunk(Job*, Mfile*);
  83. void rremove(Job*, Mfile*);
  84. void rstat(Job*, Mfile*);
  85. void rwstat(Job*, Mfile*);
  86. void rauth(Job*);
  87. void sendmsg(Job*, char*);
  88. void error(char*);
  89. void mountinit(char*, char*);
  90. void io(void);
  91. void ndbinit(void);
  92. void netinit(int);
  93. void netadd(char*);
  94. char *genquery(Mfile*, char*);
  95. char* ipinfoquery(Mfile*, char**, int);
  96. int needproto(Network*, Ndbtuple*);
  97. int lookup(Mfile*);
  98. Ndbtuple* reorder(Ndbtuple*, Ndbtuple*);
  99. void ipid(void);
  100. void readipinterfaces(void);
  101. void* emalloc(int);
  102. char* estrdup(char*);
  103. Job* newjob(void);
  104. void freejob(Job*);
  105. void setext(char*, int, char*);
  106. void cleanmf(Mfile*);
  107. extern void paralloc(void);
  108. Lock dblock; /* mutex on database operations */
  109. Lock netlock; /* mutex for netinit() */
  110. char *logfile = "cs";
  111. char *paranoiafile = "cs.paranoia";
  112. char mntpt[Maxpath];
  113. char netndb[Maxpath];
  114. /*
  115. * Network specific translators
  116. */
  117. Ndbtuple* iplookup(Network*, char*, char*, int);
  118. char* iptrans(Ndbtuple*, Network*, char*, char*, int);
  119. Ndbtuple* telcolookup(Network*, char*, char*, int);
  120. char* telcotrans(Ndbtuple*, Network*, char*, char*, int);
  121. Ndbtuple* dnsiplookup(char*, Ndbs*);
  122. struct Network
  123. {
  124. char *net;
  125. Ndbtuple *(*lookup)(Network*, char*, char*, int);
  126. char *(*trans)(Ndbtuple*, Network*, char*, char*, int);
  127. int considered;
  128. int fasttimeouthack;
  129. Network *next;
  130. };
  131. enum
  132. {
  133. Nilfast,
  134. Ntcp,
  135. Nil,
  136. Nudp,
  137. Nicmp,
  138. Nicmpv6,
  139. Nrudp,
  140. Ntelco,
  141. };
  142. /*
  143. * net doesn't apply to (r)udp, icmp(v6), or telco (for speed)
  144. */
  145. Network network[] = {
  146. [Ntcp] { "tcp", iplookup, iptrans, 0, 0 },
  147. [Nilfast] { "il", iplookup, iptrans, 0, 1 },
  148. [Nil] { "il", iplookup, iptrans, 0, 0 },
  149. [Nudp] { "udp", iplookup, iptrans, 1, 0 },
  150. [Nicmp] { "icmp", iplookup, iptrans, 1, 0 },
  151. [Nicmpv6] { "icmpv6", iplookup, iptrans, 1, 0 },
  152. [Nrudp] { "rudp", iplookup, iptrans, 1, 0 },
  153. [Ntelco] { "telco", telcolookup, telcotrans, 1, 0 },
  154. { 0 },
  155. };
  156. Lock ipifclock;
  157. Ipifc *ipifcs;
  158. char eaddr[16]; /* ascii ethernet address */
  159. char ipaddr[64]; /* ascii internet address */
  160. uchar ipa[IPaddrlen]; /* binary internet address */
  161. char *mysysname;
  162. Network *netlist; /* networks ordered by preference */
  163. Network *last;
  164. static void
  165. nstrcpy(char *to, char *from, int len)
  166. {
  167. strncpy(to, from, len);
  168. to[len-1] = 0;
  169. }
  170. void
  171. usage(void)
  172. {
  173. fprint(2, "usage: %s [-d] [-f ndb-file] [-x netmtpt] [-n]\n", argv0);
  174. exits("usage");
  175. }
  176. void
  177. main(int argc, char *argv[])
  178. {
  179. char servefile[Maxpath];
  180. int justsetname;
  181. char *p;
  182. char ext[Maxpath];
  183. justsetname = 0;
  184. setnetmtpt(mntpt, sizeof(mntpt), nil);
  185. ext[0] = 0;
  186. ARGBEGIN{
  187. case 'd':
  188. debug = 1;
  189. break;
  190. case 'f':
  191. p = ARGF();
  192. if(p == nil)
  193. usage();
  194. dbfile = p;
  195. break;
  196. case 'x':
  197. p = ARGF();
  198. if(p == nil)
  199. usage();
  200. setnetmtpt(mntpt, sizeof(mntpt), p);
  201. setext(ext, sizeof(ext), mntpt);
  202. break;
  203. case 'n':
  204. justsetname = 1;
  205. break;
  206. }ARGEND
  207. USED(argc);
  208. USED(argv);
  209. rfork(RFREND|RFNOTEG);
  210. snprint(servefile, sizeof(servefile), "#s/cs%s", ext);
  211. snprint(netndb, sizeof(netndb), "%s/ndb", mntpt);
  212. unmount(servefile, mntpt);
  213. remove(servefile);
  214. fmtinstall('E', eipfmt);
  215. fmtinstall('I', eipfmt);
  216. fmtinstall('M', eipfmt);
  217. fmtinstall('F', fcallfmt);
  218. ndbinit();
  219. netinit(0);
  220. if(!justsetname){
  221. mountinit(servefile, mntpt);
  222. io();
  223. }
  224. exits(0);
  225. }
  226. /*
  227. * if a mount point is specified, set the cs extention to be the mount point
  228. * with '_'s replacing '/'s
  229. */
  230. void
  231. setext(char *ext, int n, char *p)
  232. {
  233. int i, c;
  234. n--;
  235. for(i = 0; i < n; i++){
  236. c = p[i];
  237. if(c == 0)
  238. break;
  239. if(c == '/')
  240. c = '_';
  241. ext[i] = c;
  242. }
  243. ext[i] = 0;
  244. }
  245. void
  246. mountinit(char *service, char *mntpt)
  247. {
  248. int f;
  249. int p[2];
  250. char buf[32];
  251. if(pipe(p) < 0)
  252. error("pipe failed");
  253. /*
  254. * make a /srv/cs
  255. */
  256. f = create(service, OWRITE|ORCLOSE, 0666);
  257. if(f < 0)
  258. error(service);
  259. snprint(buf, sizeof(buf), "%d", p[1]);
  260. if(write(f, buf, strlen(buf)) != strlen(buf))
  261. error("write /srv/cs");
  262. switch(rfork(RFFDG|RFPROC|RFNAMEG)){
  263. case 0:
  264. close(p[1]);
  265. break;
  266. case -1:
  267. error("fork failed\n");
  268. default:
  269. /*
  270. * put ourselves into the file system
  271. */
  272. close(p[0]);
  273. if(mount(p[1], -1, mntpt, MAFTER, "") < 0)
  274. error("mount failed\n");
  275. _exits(0);
  276. }
  277. mfd[0] = mfd[1] = p[0];
  278. }
  279. void
  280. ndbinit(void)
  281. {
  282. db = ndbopen(dbfile);
  283. if(db == nil)
  284. error("can't open network database");
  285. netdb = ndbopen(netndb);
  286. if(netdb != nil){
  287. netdb->nohash = 1;
  288. db = ndbcat(netdb, db);
  289. }
  290. }
  291. Mfile*
  292. newfid(int fid)
  293. {
  294. Mlist *f, *ff;
  295. Mfile *mf;
  296. ff = 0;
  297. for(f = mlist; f; f = f->next)
  298. if(f->mf.busy && f->mf.fid == fid)
  299. return &f->mf;
  300. else if(!ff && !f->mf.busy)
  301. ff = f;
  302. if(ff == 0){
  303. ff = emalloc(sizeof *f);
  304. ff->next = mlist;
  305. mlist = ff;
  306. }
  307. mf = &ff->mf;
  308. memset(mf, 0, sizeof *mf);
  309. mf->fid = fid;
  310. return mf;
  311. }
  312. Job*
  313. newjob(void)
  314. {
  315. Job *job;
  316. job = mallocz(sizeof(Job), 1);
  317. lock(&joblock);
  318. job->next = joblist;
  319. joblist = job;
  320. job->request.tag = -1;
  321. unlock(&joblock);
  322. return job;
  323. }
  324. void
  325. freejob(Job *job)
  326. {
  327. Job **l;
  328. lock(&joblock);
  329. for(l = &joblist; *l; l = &(*l)->next){
  330. if((*l) == job){
  331. *l = job->next;
  332. free(job);
  333. break;
  334. }
  335. }
  336. unlock(&joblock);
  337. }
  338. void
  339. flushjob(int tag)
  340. {
  341. Job *job;
  342. lock(&joblock);
  343. for(job = joblist; job; job = job->next){
  344. if(job->request.tag == tag && job->request.type != Tflush){
  345. job->flushed = 1;
  346. break;
  347. }
  348. }
  349. unlock(&joblock);
  350. }
  351. void
  352. io(void)
  353. {
  354. long n;
  355. Mfile *mf;
  356. int slaveflag;
  357. uchar mdata[IOHDRSZ + Maxfdata];
  358. Job *job;
  359. /*
  360. * if we ask dns to fulfill requests,
  361. * a slave process is created to wait for replies. The
  362. * master process returns immediately via a longjmp
  363. * through 'masterjmp'.
  364. *
  365. * *isslave is a pointer into the call stack to a variable
  366. * that tells whether or not the current process is a slave.
  367. */
  368. slaveflag = 0; /* init slave variable */
  369. isslave = &slaveflag;
  370. setjmp(masterjmp);
  371. for(;;){
  372. n = read9pmsg(mfd[0], mdata, sizeof mdata);
  373. if(n<=0)
  374. error("mount read");
  375. job = newjob();
  376. if(convM2S(mdata, n, &job->request) != n){
  377. syslog(1, logfile, "format error %ux %ux %ux %ux %ux",
  378. mdata[0], mdata[1], mdata[2], mdata[3], mdata[4]);
  379. freejob(job);
  380. continue;
  381. }
  382. lock(&dblock);
  383. mf = newfid(job->request.fid);
  384. if(debug)
  385. syslog(0, logfile, "%F", &job->request);
  386. switch(job->request.type){
  387. default:
  388. syslog(1, logfile, "unknown request type %d", job->request.type);
  389. break;
  390. case Tversion:
  391. rversion(job);
  392. break;
  393. case Tauth:
  394. rauth(job);
  395. break;
  396. case Tflush:
  397. rflush(job);
  398. break;
  399. case Tattach:
  400. rattach(job, mf);
  401. break;
  402. case Twalk:
  403. rwalk(job, mf);
  404. break;
  405. case Topen:
  406. ropen(job, mf);
  407. break;
  408. case Tcreate:
  409. rcreate(job, mf);
  410. break;
  411. case Tread:
  412. rread(job, mf);
  413. break;
  414. case Twrite:
  415. rwrite(job, mf);
  416. break;
  417. case Tclunk:
  418. rclunk(job, mf);
  419. break;
  420. case Tremove:
  421. rremove(job, mf);
  422. break;
  423. case Tstat:
  424. rstat(job, mf);
  425. break;
  426. case Twstat:
  427. rwstat(job, mf);
  428. break;
  429. }
  430. unlock(&dblock);
  431. freejob(job);
  432. /*
  433. * slave processes die after replying
  434. */
  435. if(*isslave){
  436. if(debug)
  437. syslog(0, logfile, "slave death %d", getpid());
  438. _exits(0);
  439. }
  440. }
  441. }
  442. void
  443. rversion(Job *job)
  444. {
  445. if(job->request.msize > IOHDRSZ + Maxfdata)
  446. job->reply.msize = IOHDRSZ + Maxfdata;
  447. else
  448. job->reply.msize = job->request.msize;
  449. if(strncmp(job->request.version, "9P2000", 6) != 0)
  450. sendmsg(job, "unknown 9P version");
  451. else{
  452. job->reply.version = "9P2000";
  453. sendmsg(job, 0);
  454. }
  455. }
  456. void
  457. rauth(Job *job)
  458. {
  459. sendmsg(job, "cs: authentication not required");
  460. }
  461. /*
  462. * don't flush till all the slaves are done
  463. */
  464. void
  465. rflush(Job *job)
  466. {
  467. flushjob(job->request.oldtag);
  468. sendmsg(job, 0);
  469. }
  470. void
  471. rattach(Job *job, Mfile *mf)
  472. {
  473. if(mf->busy == 0){
  474. mf->busy = 1;
  475. mf->user = estrdup(job->request.uname);
  476. }
  477. mf->qid.vers = vers++;
  478. mf->qid.type = QTDIR;
  479. mf->qid.path = 0LL;
  480. job->reply.qid = mf->qid;
  481. sendmsg(job, 0);
  482. }
  483. char*
  484. rwalk(Job *job, Mfile *mf)
  485. {
  486. char *err;
  487. char **elems;
  488. int nelems;
  489. int i;
  490. Mfile *nmf;
  491. Qid qid;
  492. err = 0;
  493. nmf = nil;
  494. elems = job->request.wname;
  495. nelems = job->request.nwname;
  496. job->reply.nwqid = 0;
  497. if(job->request.newfid != job->request.fid){
  498. /* clone fid */
  499. nmf = newfid(job->request.newfid);
  500. if(nmf->busy){
  501. nmf = nil;
  502. err = "clone to used channel";
  503. goto send;
  504. }
  505. *nmf = *mf;
  506. nmf->user = estrdup(mf->user);
  507. nmf->fid = job->request.newfid;
  508. nmf->qid.vers = vers++;
  509. mf = nmf;
  510. }
  511. /* else nmf will be nil */
  512. qid = mf->qid;
  513. if(nelems > 0){
  514. /* walk fid */
  515. for(i=0; i<nelems && i<MAXWELEM; i++){
  516. if((qid.type & QTDIR) == 0){
  517. err = "not a directory";
  518. break;
  519. }
  520. if(strcmp(elems[i], "..") == 0 || strcmp(elems[i], ".") == 0){
  521. qid.type = QTDIR;
  522. qid.path = Qdir;
  523. Found:
  524. job->reply.wqid[i] = qid;
  525. job->reply.nwqid++;
  526. continue;
  527. }
  528. if(strcmp(elems[i], "cs") == 0){
  529. qid.type = QTFILE;
  530. qid.path = Qcs;
  531. goto Found;
  532. }
  533. err = "file does not exist";
  534. break;
  535. }
  536. }
  537. send:
  538. if(nmf != nil && (err!=nil || job->reply.nwqid<nelems)){
  539. cleanmf(nmf);
  540. free(nmf->user);
  541. nmf->user = 0;
  542. nmf->busy = 0;
  543. nmf->fid = 0;
  544. }
  545. if(err == nil)
  546. mf->qid = qid;
  547. sendmsg(job, err);
  548. return err;
  549. }
  550. void
  551. ropen(Job *job, Mfile *mf)
  552. {
  553. int mode;
  554. char *err;
  555. err = 0;
  556. mode = job->request.mode;
  557. if(mf->qid.type & QTDIR){
  558. if(mode)
  559. err = "permission denied";
  560. }
  561. job->reply.qid = mf->qid;
  562. job->reply.iounit = 0;
  563. sendmsg(job, err);
  564. }
  565. void
  566. rcreate(Job *job, Mfile *mf)
  567. {
  568. USED(mf);
  569. sendmsg(job, "creation permission denied");
  570. }
  571. void
  572. rread(Job *job, Mfile *mf)
  573. {
  574. int i, n, cnt;
  575. long off, toff, clock;
  576. Dir dir;
  577. uchar buf[Maxfdata];
  578. char *err;
  579. n = 0;
  580. err = 0;
  581. off = job->request.offset;
  582. cnt = job->request.count;
  583. if(mf->qid.type & QTDIR){
  584. clock = time(0);
  585. if(off == 0){
  586. dir.name = "cs";
  587. dir.qid.type = QTFILE;
  588. dir.qid.vers = vers;
  589. dir.qid.path = Qcs;
  590. dir.mode = 0666;
  591. dir.length = 0;
  592. dir.uid = mf->user;
  593. dir.gid = mf->user;
  594. dir.muid = mf->user;
  595. dir.atime = clock; /* wrong */
  596. dir.mtime = clock; /* wrong */
  597. n = convD2M(&dir, buf, sizeof buf);
  598. }
  599. job->reply.data = (char*)buf;
  600. } else {
  601. for(;;){
  602. /* look for an answer at the right offset */
  603. toff = 0;
  604. for(i = 0; mf->reply[i] && i < mf->nreply; i++){
  605. n = mf->replylen[i];
  606. if(off < toff + n)
  607. break;
  608. toff += n;
  609. }
  610. if(i < mf->nreply)
  611. break; /* got something to return */
  612. /* try looking up more answers */
  613. if(lookup(mf) == 0){
  614. /* no more */
  615. n = 0;
  616. goto send;
  617. }
  618. }
  619. /* give back a single reply (or part of one) */
  620. job->reply.data = mf->reply[i] + (off - toff);
  621. if(cnt > toff - off + n)
  622. n = toff - off + n;
  623. else
  624. n = cnt;
  625. }
  626. send:
  627. job->reply.count = n;
  628. sendmsg(job, err);
  629. }
  630. void
  631. cleanmf(Mfile *mf)
  632. {
  633. int i;
  634. if(mf->net != nil){
  635. free(mf->net);
  636. mf->net = nil;
  637. }
  638. if(mf->host != nil){
  639. free(mf->host);
  640. mf->host = nil;
  641. }
  642. if(mf->serv != nil){
  643. free(mf->serv);
  644. mf->serv = nil;
  645. }
  646. if(mf->rem != nil){
  647. free(mf->rem);
  648. mf->rem = nil;
  649. }
  650. for(i = 0; i < mf->nreply; i++){
  651. free(mf->reply[i]);
  652. mf->reply[i] = nil;
  653. mf->replylen[i] = 0;
  654. }
  655. mf->nreply = 0;
  656. mf->nextnet = netlist;
  657. }
  658. void
  659. rwrite(Job *job, Mfile *mf)
  660. {
  661. int cnt, n;
  662. char *err;
  663. char *field[4];
  664. char curerr[64];
  665. err = 0;
  666. cnt = job->request.count;
  667. if(mf->qid.type & QTDIR){
  668. err = "can't write directory";
  669. goto send;
  670. }
  671. if(cnt >= Maxrequest){
  672. err = "request too long";
  673. goto send;
  674. }
  675. job->request.data[cnt] = 0;
  676. /*
  677. * toggle debugging
  678. */
  679. if(strncmp(job->request.data, "debug", 5)==0){
  680. debug ^= 1;
  681. syslog(1, logfile, "debug %d", debug);
  682. goto send;
  683. }
  684. /*
  685. * toggle debugging
  686. */
  687. if(strncmp(job->request.data, "paranoia", 8)==0){
  688. paranoia ^= 1;
  689. syslog(1, logfile, "paranoia %d", paranoia);
  690. goto send;
  691. }
  692. /*
  693. * add networks to the default list
  694. */
  695. if(strncmp(job->request.data, "add ", 4)==0){
  696. if(job->request.data[cnt-1] == '\n')
  697. job->request.data[cnt-1] = 0;
  698. netadd(job->request.data+4);
  699. readipinterfaces();
  700. goto send;
  701. }
  702. /*
  703. * refresh all state
  704. */
  705. if(strncmp(job->request.data, "refresh", 7)==0){
  706. netinit(1);
  707. goto send;
  708. }
  709. /* start transaction with a clean slate */
  710. cleanmf(mf);
  711. /*
  712. * look for a general query
  713. */
  714. if(*job->request.data == '!'){
  715. err = genquery(mf, job->request.data+1);
  716. goto send;
  717. }
  718. if(debug)
  719. syslog(0, logfile, "write %s", job->request.data);
  720. if(paranoia)
  721. syslog(0, paranoiafile, "write %s by %s", job->request.data, mf->user);
  722. /*
  723. * break up name
  724. */
  725. n = getfields(job->request.data, field, 4, 1, "!");
  726. switch(n){
  727. case 1:
  728. mf->net = strdup("net");
  729. mf->host = strdup(field[0]);
  730. break;
  731. case 4:
  732. mf->rem = strdup(field[3]);
  733. /* fall through */
  734. case 3:
  735. mf->serv = strdup(field[2]);
  736. /* fall through */
  737. case 2:
  738. mf->host = strdup(field[1]);
  739. mf->net = strdup(field[0]);
  740. break;
  741. }
  742. /*
  743. * do the first net worth of lookup
  744. */
  745. if(lookup(mf) == 0){
  746. rerrstr(curerr, sizeof curerr);
  747. err = curerr;
  748. }
  749. send:
  750. job->reply.count = cnt;
  751. sendmsg(job, err);
  752. }
  753. void
  754. rclunk(Job *job, Mfile *mf)
  755. {
  756. cleanmf(mf);
  757. free(mf->user);
  758. mf->user = 0;
  759. mf->busy = 0;
  760. mf->fid = 0;
  761. sendmsg(job, 0);
  762. }
  763. void
  764. rremove(Job *job, Mfile *mf)
  765. {
  766. USED(mf);
  767. sendmsg(job, "remove permission denied");
  768. }
  769. void
  770. rstat(Job *job, Mfile *mf)
  771. {
  772. Dir dir;
  773. uchar buf[IOHDRSZ+Maxfdata];
  774. if(mf->qid.type & QTDIR){
  775. dir.name = ".";
  776. dir.mode = DMDIR|0555;
  777. } else {
  778. dir.name = "cs";
  779. dir.mode = 0666;
  780. }
  781. dir.qid = mf->qid;
  782. dir.length = 0;
  783. dir.uid = mf->user;
  784. dir.gid = mf->user;
  785. dir.muid = mf->user;
  786. dir.atime = dir.mtime = time(0);
  787. job->reply.nstat = convD2M(&dir, buf, sizeof buf);
  788. job->reply.stat = buf;
  789. sendmsg(job, 0);
  790. }
  791. void
  792. rwstat(Job *job, Mfile *mf)
  793. {
  794. USED(mf);
  795. sendmsg(job, "wstat permission denied");
  796. }
  797. void
  798. sendmsg(Job *job, char *err)
  799. {
  800. int n;
  801. uchar mdata[IOHDRSZ + Maxfdata];
  802. char ename[ERRMAX];
  803. if(err){
  804. job->reply.type = Rerror;
  805. snprint(ename, sizeof(ename), "cs: %s", err);
  806. job->reply.ename = ename;
  807. }else{
  808. job->reply.type = job->request.type+1;
  809. }
  810. job->reply.tag = job->request.tag;
  811. n = convS2M(&job->reply, mdata, sizeof mdata);
  812. if(n == 0){
  813. syslog(1, logfile, "sendmsg convS2M of %F returns 0", &job->reply);
  814. abort();
  815. }
  816. lock(&joblock);
  817. if(job->flushed == 0)
  818. if(write(mfd[1], mdata, n)!=n)
  819. error("mount write");
  820. unlock(&joblock);
  821. if(debug)
  822. syslog(0, logfile, "%F %d", &job->reply, n);
  823. }
  824. void
  825. error(char *s)
  826. {
  827. syslog(1, "cs", "%s: %r", s);
  828. _exits(0);
  829. }
  830. static int
  831. isvalidip(uchar *ip)
  832. {
  833. return ipcmp(ip, IPnoaddr) != 0 && ipcmp(ip, v4prefix) != 0;
  834. }
  835. static uchar loopbacknet[IPaddrlen] = {
  836. 0, 0, 0, 0,
  837. 0, 0, 0, 0,
  838. 0, 0, 0xff, 0xff,
  839. 127, 0, 0, 0
  840. };
  841. static uchar loopbackmask[IPaddrlen] = {
  842. 0xff, 0xff, 0xff, 0xff,
  843. 0xff, 0xff, 0xff, 0xff,
  844. 0xff, 0xff, 0xff, 0xff,
  845. 0xff, 0, 0, 0
  846. };
  847. void
  848. readipinterfaces(void)
  849. {
  850. if(myipaddr(ipa, mntpt) != 0)
  851. ipmove(ipa, IPnoaddr);
  852. sprint(ipaddr, "%I", ipa);
  853. if (debug)
  854. syslog(0, "dns", "ipaddr is %s\n", ipaddr);
  855. }
  856. /*
  857. * get the system name
  858. */
  859. void
  860. ipid(void)
  861. {
  862. uchar addr[6];
  863. Ndbtuple *t, *tt;
  864. char *p, *attr;
  865. Ndbs s;
  866. int f;
  867. char buf[Maxpath];
  868. /* use environment, ether addr, or ipaddr to get system name */
  869. if(mysysname == 0){
  870. /*
  871. * environment has priority.
  872. *
  873. * on the sgi power the default system name
  874. * is the ip address. ignore that.
  875. *
  876. */
  877. p = getenv("sysname");
  878. if(p && *p){
  879. attr = ipattr(p);
  880. if(strcmp(attr, "ip") != 0)
  881. mysysname = strdup(p);
  882. }
  883. /*
  884. * the /net/ndb contains what the network
  885. * figured out from DHCP. use that name if
  886. * there is one.
  887. */
  888. if(mysysname == 0 && netdb != nil){
  889. ndbreopen(netdb);
  890. for(tt = t = ndbparse(netdb); t != nil; t = t->entry){
  891. if(strcmp(t->attr, "sys") == 0){
  892. mysysname = strdup(t->val);
  893. break;
  894. }
  895. }
  896. ndbfree(tt);
  897. }
  898. /* next network database, ip address, and ether address to find a name */
  899. if(mysysname == 0){
  900. t = nil;
  901. if(isvalidip(ipa))
  902. free(ndbgetvalue(db, &s, "ip", ipaddr, "sys", &t));
  903. if(t == nil){
  904. for(f = 0; f < 3; f++){
  905. snprint(buf, sizeof buf, "%s/ether%d", mntpt, f);
  906. if(myetheraddr(addr, buf) >= 0){
  907. snprint(eaddr, sizeof(eaddr), "%E", addr);
  908. free(ndbgetvalue(db, &s, "ether", eaddr, "sys", &t));
  909. if(t != nil)
  910. break;
  911. }
  912. }
  913. }
  914. for(tt = t; tt != nil; tt = tt->entry){
  915. if(strcmp(tt->attr, "sys") == 0){
  916. mysysname = strdup(tt->val);
  917. break;
  918. }
  919. }
  920. ndbfree(t);
  921. }
  922. /* nothing else worked, use the ip address */
  923. if(mysysname == 0 && isvalidip(ipa))
  924. mysysname = strdup(ipaddr);
  925. /* set /dev/sysname if we now know it */
  926. if(mysysname){
  927. f = open("/dev/sysname", OWRITE);
  928. if(f >= 0){
  929. write(f, mysysname, strlen(mysysname));
  930. close(f);
  931. }
  932. }
  933. }
  934. }
  935. /*
  936. * Set up a list of default networks by looking for
  937. * /net/ * /clone.
  938. */
  939. void
  940. netinit(int background)
  941. {
  942. char clone[Maxpath];
  943. Network *np;
  944. static int working;
  945. if(background){
  946. switch(rfork(RFPROC|RFNOTEG|RFMEM|RFNOWAIT)){
  947. case 0:
  948. break;
  949. default:
  950. return;
  951. }
  952. lock(&netlock);
  953. }
  954. /* add the mounted networks to the default list */
  955. for(np = network; np->net; np++){
  956. if(np->considered)
  957. continue;
  958. snprint(clone, sizeof(clone), "%s/%s/clone", mntpt, np->net);
  959. if(access(clone, AEXIST) < 0)
  960. continue;
  961. if(netlist)
  962. last->next = np;
  963. else
  964. netlist = np;
  965. last = np;
  966. np->next = 0;
  967. np->considered = 1;
  968. }
  969. /* find out what our ip address is */
  970. readipinterfaces();
  971. /* set the system name if we need to, these says ip is all we have */
  972. ipid();
  973. if(debug)
  974. syslog(0, logfile, "mysysname %s eaddr %s ipaddr %s ipa %I\n",
  975. mysysname?mysysname:"???", eaddr, ipaddr, ipa);
  976. if(background){
  977. unlock(&netlock);
  978. _exits(0);
  979. }
  980. }
  981. /*
  982. * add networks to the standard list
  983. */
  984. void
  985. netadd(char *p)
  986. {
  987. Network *np;
  988. char *field[12];
  989. int i, n;
  990. n = getfields(p, field, 12, 1, " ");
  991. for(i = 0; i < n; i++){
  992. for(np = network; np->net; np++){
  993. if(strcmp(field[i], np->net) != 0)
  994. continue;
  995. if(np->considered)
  996. break;
  997. if(netlist)
  998. last->next = np;
  999. else
  1000. netlist = np;
  1001. last = np;
  1002. np->next = 0;
  1003. np->considered = 1;
  1004. }
  1005. }
  1006. }
  1007. int
  1008. lookforproto(Ndbtuple *t, char *proto)
  1009. {
  1010. for(; t != nil; t = t->entry)
  1011. if(strcmp(t->attr, "proto") == 0 && strcmp(t->val, proto) == 0)
  1012. return 1;
  1013. return 0;
  1014. }
  1015. /*
  1016. * lookup a request. the network "net" means we should pick the
  1017. * best network to get there.
  1018. */
  1019. int
  1020. lookup(Mfile *mf)
  1021. {
  1022. Network *np;
  1023. char *cp;
  1024. Ndbtuple *nt, *t;
  1025. char reply[Maxreply];
  1026. int i, rv;
  1027. int hack;
  1028. /* open up the standard db files */
  1029. if(db == 0)
  1030. ndbinit();
  1031. if(db == 0)
  1032. error("can't open mf->network database\n");
  1033. rv = 0;
  1034. if(mf->net == nil)
  1035. return 0; /* must have been a genquery */
  1036. if(strcmp(mf->net, "net") == 0){
  1037. /*
  1038. * go through set of default nets
  1039. */
  1040. for(np = mf->nextnet; np; np = np->next){
  1041. nt = (*np->lookup)(np, mf->host, mf->serv, 1);
  1042. if(nt == nil)
  1043. continue;
  1044. hack = np->fasttimeouthack && !lookforproto(nt, np->net);
  1045. for(t = nt; mf->nreply < Nreply && t; t = t->entry){
  1046. cp = (*np->trans)(t, np, mf->serv, mf->rem, hack);
  1047. if(cp){
  1048. /* avoid duplicates */
  1049. for(i = 0; i < mf->nreply; i++)
  1050. if(strcmp(mf->reply[i], cp) == 0)
  1051. break;
  1052. if(i == mf->nreply){
  1053. /* save the reply */
  1054. mf->replylen[mf->nreply] = strlen(cp);
  1055. mf->reply[mf->nreply++] = cp;
  1056. rv++;
  1057. }
  1058. }
  1059. }
  1060. ndbfree(nt);
  1061. np = np->next;
  1062. break;
  1063. }
  1064. mf->nextnet = np;
  1065. return rv;
  1066. }
  1067. /*
  1068. * if not /net, we only get one lookup
  1069. */
  1070. if(mf->nreply != 0)
  1071. return 0;
  1072. /*
  1073. * look for a specific network
  1074. */
  1075. for(np = netlist; np && np->net != nil; np++){
  1076. if(np->fasttimeouthack)
  1077. continue;
  1078. if(strcmp(np->net, mf->net) == 0)
  1079. break;
  1080. }
  1081. if(np && np->net != nil){
  1082. /*
  1083. * known network
  1084. */
  1085. nt = (*np->lookup)(np, mf->host, mf->serv, 1);
  1086. for(t = nt; mf->nreply < Nreply && t; t = t->entry){
  1087. cp = (*np->trans)(t, np, mf->serv, mf->rem, 0);
  1088. if(cp){
  1089. mf->replylen[mf->nreply] = strlen(cp);
  1090. mf->reply[mf->nreply++] = cp;
  1091. rv++;
  1092. }
  1093. }
  1094. ndbfree(nt);
  1095. return rv;
  1096. } else {
  1097. /*
  1098. * not a known network, don't translate host or service
  1099. */
  1100. if(mf->serv)
  1101. snprint(reply, sizeof(reply), "%s/%s/clone %s!%s",
  1102. mntpt, mf->net, mf->host, mf->serv);
  1103. else
  1104. snprint(reply, sizeof(reply), "%s/%s/clone %s",
  1105. mntpt, mf->net, mf->host);
  1106. mf->reply[0] = strdup(reply);
  1107. mf->replylen[0] = strlen(reply);
  1108. mf->nreply = 1;
  1109. return 1;
  1110. }
  1111. }
  1112. /*
  1113. * translate an ip service name into a port number. If it's a numeric port
  1114. * number, look for restricted access.
  1115. *
  1116. * the service '*' needs no translation.
  1117. */
  1118. char*
  1119. ipserv(Network *np, char *name, char *buf, int blen)
  1120. {
  1121. char *p;
  1122. int alpha = 0;
  1123. int restr = 0;
  1124. char port[10];
  1125. Ndbtuple *t, *nt;
  1126. Ndbs s;
  1127. /* '*' means any service */
  1128. if(strcmp(name, "*")==0){
  1129. strcpy(buf, name);
  1130. return buf;
  1131. }
  1132. /* see if it's numeric or symbolic */
  1133. port[0] = 0;
  1134. for(p = name; *p; p++){
  1135. if(isdigit(*p))
  1136. {}
  1137. else if(isalpha(*p) || *p == '-' || *p == '$')
  1138. alpha = 1;
  1139. else
  1140. return 0;
  1141. }
  1142. t = nil;
  1143. p = nil;
  1144. if(alpha){
  1145. p = ndbgetvalue(db, &s, np->net, name, "port", &t);
  1146. if(p == nil)
  1147. return 0;
  1148. } else {
  1149. /* look up only for tcp ports < 1024 to get the restricted
  1150. * attribute
  1151. */
  1152. if(atoi(name) < 1024 && strcmp(np->net, "tcp") == 0)
  1153. p = ndbgetvalue(db, &s, "port", name, "port", &t);
  1154. if(p == nil)
  1155. p = strdup(name);
  1156. }
  1157. if(t){
  1158. for(nt = t; nt; nt = nt->entry)
  1159. if(strcmp(nt->attr, "restricted") == 0)
  1160. restr = 1;
  1161. ndbfree(t);
  1162. }
  1163. snprint(buf, blen, "%s%s", p, restr ? "!r" : "");
  1164. free(p);
  1165. return buf;
  1166. }
  1167. /*
  1168. * lookup an ip attribute
  1169. */
  1170. int
  1171. ipattrlookup(Ndb *db, char *ipa, char *attr, char *val, int vlen)
  1172. {
  1173. Ndbtuple *t, *nt;
  1174. char *alist[2];
  1175. alist[0] = attr;
  1176. t = ndbipinfo(db, "ip", ipa, alist, 1);
  1177. if(t == nil)
  1178. return 0;
  1179. for(nt = t; nt != nil; nt = nt->entry){
  1180. if(strcmp(nt->attr, attr) == 0){
  1181. nstrcpy(val, nt->val, vlen);
  1182. ndbfree(t);
  1183. return 1;
  1184. }
  1185. }
  1186. /* we shouldn't get here */
  1187. ndbfree(t);
  1188. return 0;
  1189. }
  1190. /*
  1191. * lookup (and translate) an ip destination
  1192. */
  1193. Ndbtuple*
  1194. iplookup(Network *np, char *host, char *serv, int nolookup)
  1195. {
  1196. char *attr, *dnsname;
  1197. Ndbtuple *t, *nt;
  1198. Ndbs s;
  1199. char ts[Maxservice];
  1200. char dollar[Maxhost];
  1201. uchar ip[IPaddrlen];
  1202. uchar net[IPaddrlen];
  1203. uchar tnet[IPaddrlen];
  1204. Ipifc *ifc;
  1205. Iplifc *lifc;
  1206. USED(nolookup);
  1207. /*
  1208. * start with the service since it's the most likely to fail
  1209. * and costs the least
  1210. */
  1211. werrstr("can't translate address");
  1212. if(serv==0 || ipserv(np, serv, ts, sizeof ts) == 0){
  1213. werrstr("can't translate service");
  1214. return 0;
  1215. }
  1216. /* for dial strings with no host */
  1217. if(strcmp(host, "*") == 0)
  1218. return ndbnew("ip", "*");
  1219. /*
  1220. * hack till we go v6 :: = 0.0.0.0
  1221. */
  1222. if(strcmp("::", host) == 0)
  1223. return ndbnew("ip", "*");
  1224. /*
  1225. * '$' means the rest of the name is an attribute that we
  1226. * need to search for
  1227. */
  1228. if(*host == '$'){
  1229. if(ipattrlookup(db, ipaddr, host+1, dollar, sizeof dollar))
  1230. host = dollar;
  1231. }
  1232. /*
  1233. * turn '[ip address]' into just 'ip address'
  1234. */
  1235. if(*host == '[' && host[strlen(host)-1] == ']'){
  1236. host++;
  1237. host[strlen(host)-1] = 0;
  1238. }
  1239. /*
  1240. * just accept addresses
  1241. */
  1242. attr = ipattr(host);
  1243. if(strcmp(attr, "ip") == 0)
  1244. return ndbnew("ip", host);
  1245. /*
  1246. * give the domain name server the first opportunity to
  1247. * resolve domain names. if that fails try the database.
  1248. */
  1249. t = 0;
  1250. werrstr("can't translate address");
  1251. if(strcmp(attr, "dom") == 0)
  1252. t = dnsiplookup(host, &s);
  1253. if(t == 0)
  1254. free(ndbgetvalue(db, &s, attr, host, "ip", &t));
  1255. if(t == 0){
  1256. dnsname = ndbgetvalue(db, &s, attr, host, "dom", nil);
  1257. if(dnsname){
  1258. t = dnsiplookup(dnsname, &s);
  1259. free(dnsname);
  1260. }
  1261. }
  1262. if(t == 0)
  1263. t = dnsiplookup(host, &s);
  1264. if(t == 0)
  1265. return 0;
  1266. /*
  1267. * reorder the tuple to have the matched line first and
  1268. * save that in the request structure.
  1269. */
  1270. t = reorder(t, s.t);
  1271. /*
  1272. * reorder according to our interfaces
  1273. */
  1274. lock(&ipifclock);
  1275. for(ifc = ipifcs; ifc != nil; ifc = ifc->next){
  1276. for(lifc = ifc->lifc; lifc != nil; lifc = lifc->next){
  1277. maskip(lifc->ip, lifc->mask, net);
  1278. for(nt = t; nt; nt = nt->entry){
  1279. if(strcmp(nt->attr, "ip") != 0)
  1280. continue;
  1281. parseip(ip, nt->val);
  1282. maskip(ip, lifc->mask, tnet);
  1283. if(memcmp(net, tnet, IPaddrlen) == 0){
  1284. t = reorder(t, nt);
  1285. unlock(&ipifclock);
  1286. return t;
  1287. }
  1288. }
  1289. }
  1290. }
  1291. unlock(&ipifclock);
  1292. return t;
  1293. }
  1294. /*
  1295. * translate an ip address
  1296. */
  1297. char*
  1298. iptrans(Ndbtuple *t, Network *np, char *serv, char *rem, int hack)
  1299. {
  1300. char ts[Maxservice];
  1301. char reply[Maxreply];
  1302. char x[Maxservice];
  1303. if(strcmp(t->attr, "ip") != 0)
  1304. return 0;
  1305. if(serv == 0 || ipserv(np, serv, ts, sizeof ts) == 0){
  1306. werrstr("can't translate service");
  1307. return 0;
  1308. }
  1309. if(rem != nil)
  1310. snprint(x, sizeof(x), "!%s", rem);
  1311. else
  1312. *x = 0;
  1313. if(*t->val == '*')
  1314. snprint(reply, sizeof(reply), "%s/%s/clone %s%s",
  1315. mntpt, np->net, ts, x);
  1316. else
  1317. snprint(reply, sizeof(reply), "%s/%s/clone %s!%s%s%s",
  1318. mntpt, np->net, t->val, ts, x, hack?"!fasttimeout":"");
  1319. return strdup(reply);
  1320. }
  1321. /*
  1322. * lookup a telephone number
  1323. */
  1324. Ndbtuple*
  1325. telcolookup(Network *np, char *host, char *serv, int nolookup)
  1326. {
  1327. Ndbtuple *t;
  1328. Ndbs s;
  1329. USED(np, nolookup, serv);
  1330. werrstr("can't translate address");
  1331. free(ndbgetvalue(db, &s, "sys", host, "telco", &t));
  1332. if(t == 0)
  1333. return ndbnew("telco", host);
  1334. return reorder(t, s.t);
  1335. }
  1336. /*
  1337. * translate a telephone address
  1338. */
  1339. char*
  1340. telcotrans(Ndbtuple *t, Network *np, char *serv, char *rem, int)
  1341. {
  1342. char reply[Maxreply];
  1343. char x[Maxservice];
  1344. if(strcmp(t->attr, "telco") != 0)
  1345. return 0;
  1346. if(rem != nil)
  1347. snprint(x, sizeof(x), "!%s", rem);
  1348. else
  1349. *x = 0;
  1350. if(serv)
  1351. snprint(reply, sizeof(reply), "%s/%s/clone %s!%s%s", mntpt, np->net,
  1352. t->val, serv, x);
  1353. else
  1354. snprint(reply, sizeof(reply), "%s/%s/clone %s%s", mntpt, np->net,
  1355. t->val, x);
  1356. return strdup(reply);
  1357. }
  1358. /*
  1359. * reorder the tuple to put x's line first in the entry
  1360. */
  1361. Ndbtuple*
  1362. reorder(Ndbtuple *t, Ndbtuple *x)
  1363. {
  1364. Ndbtuple *nt;
  1365. Ndbtuple *line;
  1366. /* find start of this entry's line */
  1367. for(line = x; line->entry == line->line; line = line->line)
  1368. ;
  1369. line = line->line;
  1370. if(line == t)
  1371. return t; /* already the first line */
  1372. /* remove this line and everything after it from the entry */
  1373. for(nt = t; nt->entry != line; nt = nt->entry)
  1374. ;
  1375. nt->entry = 0;
  1376. /* make that the start of the entry */
  1377. for(nt = line; nt->entry; nt = nt->entry)
  1378. ;
  1379. nt->entry = t;
  1380. return line;
  1381. }
  1382. /*
  1383. * create a slave process to handle a request to avoid one request blocking
  1384. * another. parent returns to job loop.
  1385. */
  1386. void
  1387. slave(void)
  1388. {
  1389. if(*isslave)
  1390. return; /* we're already a slave process */
  1391. switch(rfork(RFPROC|RFNOTEG|RFMEM|RFNOWAIT)){
  1392. case -1:
  1393. break;
  1394. case 0:
  1395. if(debug)
  1396. syslog(0, logfile, "slave %d", getpid());
  1397. *isslave = 1;
  1398. break;
  1399. default:
  1400. longjmp(masterjmp, 1);
  1401. }
  1402. }
  1403. /*
  1404. * call the dns process and have it try to translate a name
  1405. */
  1406. Ndbtuple*
  1407. dnsiplookup(char *host, Ndbs *s)
  1408. {
  1409. char buf[Maxreply];
  1410. Ndbtuple *t;
  1411. unlock(&dblock);
  1412. /* save the name before starting a slave */
  1413. snprint(buf, sizeof(buf), "%s", host);
  1414. slave();
  1415. if(strcmp(ipattr(buf), "ip") == 0)
  1416. t = dnsquery(mntpt, buf, "ptr");
  1417. else
  1418. t = dnsquery(mntpt, buf, "ip");
  1419. s->t = t;
  1420. if(t == nil){
  1421. rerrstr(buf, sizeof buf);
  1422. if(strstr(buf, "exist"))
  1423. werrstr("can't translate address: %s", buf);
  1424. else if(strstr(buf, "dns failure"))
  1425. werrstr("temporary problem: %s", buf);
  1426. }
  1427. lock(&dblock);
  1428. return t;
  1429. }
  1430. int
  1431. qmatch(Ndbtuple *t, char **attr, char **val, int n)
  1432. {
  1433. int i, found;
  1434. Ndbtuple *nt;
  1435. for(i = 1; i < n; i++){
  1436. found = 0;
  1437. for(nt = t; nt; nt = nt->entry)
  1438. if(strcmp(attr[i], nt->attr) == 0)
  1439. if(strcmp(val[i], "*") == 0
  1440. || strcmp(val[i], nt->val) == 0){
  1441. found = 1;
  1442. break;
  1443. }
  1444. if(found == 0)
  1445. break;
  1446. }
  1447. return i == n;
  1448. }
  1449. void
  1450. qreply(Mfile *mf, Ndbtuple *t)
  1451. {
  1452. Ndbtuple *nt;
  1453. String *s;
  1454. s = s_new();
  1455. for(nt = t; mf->nreply < Nreply && nt; nt = nt->entry){
  1456. s_append(s, nt->attr);
  1457. s_append(s, "=");
  1458. s_append(s, nt->val);
  1459. if(nt->line != nt->entry){
  1460. mf->replylen[mf->nreply] = s_len(s);
  1461. mf->reply[mf->nreply++] = strdup(s_to_c(s));
  1462. s_restart(s);
  1463. } else
  1464. s_append(s, " ");
  1465. }
  1466. s_free(s);
  1467. }
  1468. enum
  1469. {
  1470. Maxattr= 32,
  1471. };
  1472. /*
  1473. * generic query lookup. The query is of one of the following
  1474. * forms:
  1475. *
  1476. * attr1=val1 attr2=val2 attr3=val3 ...
  1477. *
  1478. * returns the matching tuple
  1479. *
  1480. * ipinfo attr=val attr1 attr2 attr3 ...
  1481. *
  1482. * is like ipinfo and returns the attr{1-n}
  1483. * associated with the ip address.
  1484. */
  1485. char*
  1486. genquery(Mfile *mf, char *query)
  1487. {
  1488. int i, n;
  1489. char *p;
  1490. char *attr[Maxattr];
  1491. char *val[Maxattr];
  1492. Ndbtuple *t;
  1493. Ndbs s;
  1494. n = getfields(query, attr, 32, 1, " ");
  1495. if(n == 0)
  1496. return "bad query";
  1497. if(strcmp(attr[0], "ipinfo") == 0)
  1498. return ipinfoquery(mf, attr, n);
  1499. /* parse pairs */
  1500. for(i = 0; i < n; i++){
  1501. p = strchr(attr[i], '=');
  1502. if(p == 0)
  1503. return "bad query";
  1504. *p++ = 0;
  1505. val[i] = p;
  1506. }
  1507. /* give dns a chance */
  1508. if((strcmp(attr[0], "dom") == 0 || strcmp(attr[0], "ip") == 0) && val[0]){
  1509. t = dnsiplookup(val[0], &s);
  1510. if(t){
  1511. if(qmatch(t, attr, val, n)){
  1512. qreply(mf, t);
  1513. ndbfree(t);
  1514. return 0;
  1515. }
  1516. ndbfree(t);
  1517. }
  1518. }
  1519. /* first pair is always the key. It can't be a '*' */
  1520. t = ndbsearch(db, &s, attr[0], val[0]);
  1521. /* search is the and of all the pairs */
  1522. while(t){
  1523. if(qmatch(t, attr, val, n)){
  1524. qreply(mf, t);
  1525. ndbfree(t);
  1526. return 0;
  1527. }
  1528. ndbfree(t);
  1529. t = ndbsnext(&s, attr[0], val[0]);
  1530. }
  1531. return "no match";
  1532. }
  1533. /*
  1534. * resolve an ip address
  1535. */
  1536. static Ndbtuple*
  1537. ipresolve(char *attr, char *host)
  1538. {
  1539. Ndbtuple *t, *nt, **l;
  1540. t = iplookup(&network[Ntcp], host, "*", 0);
  1541. for(l = &t; *l != nil; ){
  1542. nt = *l;
  1543. if(strcmp(nt->attr, "ip") != 0){
  1544. *l = nt->entry;
  1545. nt->entry = nil;
  1546. ndbfree(nt);
  1547. continue;
  1548. }
  1549. strcpy(nt->attr, attr);
  1550. l = &nt->entry;
  1551. }
  1552. return t;
  1553. }
  1554. char*
  1555. ipinfoquery(Mfile *mf, char **list, int n)
  1556. {
  1557. int i, nresolve;
  1558. int resolve[Maxattr];
  1559. Ndbtuple *t, *nt, **l;
  1560. char *attr, *val;
  1561. /* skip 'ipinfo' */
  1562. list++; n--;
  1563. if(n < 1)
  1564. return "bad query";
  1565. /* get search attribute=value, or assume ip=myipaddr */
  1566. attr = *list;
  1567. if((val = strchr(attr, '=')) != nil){
  1568. *val++ = 0;
  1569. list++;
  1570. n--;
  1571. }else{
  1572. attr = "ip";
  1573. val = ipaddr;
  1574. }
  1575. if(n < 1)
  1576. return "bad query";
  1577. /*
  1578. * don't let ndbipinfo resolve the addresses, we're
  1579. * better at it.
  1580. */
  1581. nresolve = 0;
  1582. for(i = 0; i < n; i++)
  1583. if(*list[i] == '@'){
  1584. list[i]++;
  1585. resolve[i] = 1;
  1586. nresolve++;
  1587. } else
  1588. resolve[i] = 0;
  1589. t = ndbipinfo(db, attr, val, list, n);
  1590. if(t == nil)
  1591. return "no match";
  1592. if(nresolve != 0){
  1593. for(l = &t; *l != nil;){
  1594. nt = *l;
  1595. /* already an address? */
  1596. if(strcmp(ipattr(nt->val), "ip") == 0){
  1597. l = &(*l)->entry;
  1598. continue;
  1599. }
  1600. /* user wants it resolved? */
  1601. for(i = 0; i < n; i++)
  1602. if(strcmp(list[i], nt->attr) == 0)
  1603. break;
  1604. if(i >= n || resolve[i] == 0){
  1605. l = &(*l)->entry;
  1606. continue;
  1607. }
  1608. /* resolve address and replace entry */
  1609. *l = ipresolve(nt->attr, nt->val);
  1610. while(*l != nil)
  1611. l = &(*l)->entry;
  1612. *l = nt->entry;
  1613. nt->entry = nil;
  1614. ndbfree(nt);
  1615. }
  1616. }
  1617. /* make it all one line */
  1618. for(nt = t; nt != nil; nt = nt->entry){
  1619. if(nt->entry == nil)
  1620. nt->line = t;
  1621. else
  1622. nt->line = nt->entry;
  1623. }
  1624. qreply(mf, t);
  1625. return nil;
  1626. }
  1627. void*
  1628. emalloc(int size)
  1629. {
  1630. void *x;
  1631. x = malloc(size);
  1632. if(x == nil)
  1633. abort();
  1634. memset(x, 0, size);
  1635. return x;
  1636. }
  1637. char*
  1638. estrdup(char *s)
  1639. {
  1640. int size;
  1641. char *p;
  1642. size = strlen(s)+1;
  1643. p = malloc(size);
  1644. if(p == nil)
  1645. abort();
  1646. memmove(p, s, size);
  1647. return p;
  1648. }