cs.c 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845
  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. [Nilfast] { "il", iplookup, iptrans, 0, 1 },
  147. [Ntcp] { "tcp", iplookup, iptrans, 0, 0 },
  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. if(job->request.fid<0)
  383. error("fid out of range");
  384. lock(&dblock);
  385. mf = newfid(job->request.fid);
  386. if(debug)
  387. syslog(0, logfile, "%F", &job->request);
  388. switch(job->request.type){
  389. default:
  390. syslog(1, logfile, "unknown request type %d", job->request.type);
  391. break;
  392. case Tversion:
  393. rversion(job);
  394. break;
  395. case Tauth:
  396. rauth(job);
  397. break;
  398. case Tflush:
  399. rflush(job);
  400. break;
  401. case Tattach:
  402. rattach(job, mf);
  403. break;
  404. case Twalk:
  405. rwalk(job, mf);
  406. break;
  407. case Topen:
  408. ropen(job, mf);
  409. break;
  410. case Tcreate:
  411. rcreate(job, mf);
  412. break;
  413. case Tread:
  414. rread(job, mf);
  415. break;
  416. case Twrite:
  417. rwrite(job, mf);
  418. break;
  419. case Tclunk:
  420. rclunk(job, mf);
  421. break;
  422. case Tremove:
  423. rremove(job, mf);
  424. break;
  425. case Tstat:
  426. rstat(job, mf);
  427. break;
  428. case Twstat:
  429. rwstat(job, mf);
  430. break;
  431. }
  432. unlock(&dblock);
  433. freejob(job);
  434. /*
  435. * slave processes die after replying
  436. */
  437. if(*isslave){
  438. if(debug)
  439. syslog(0, logfile, "slave death %d", getpid());
  440. _exits(0);
  441. }
  442. }
  443. }
  444. void
  445. rversion(Job *job)
  446. {
  447. if(job->request.msize > IOHDRSZ + Maxfdata)
  448. job->reply.msize = IOHDRSZ + Maxfdata;
  449. else
  450. job->reply.msize = job->request.msize;
  451. if(strncmp(job->request.version, "9P2000", 6) != 0)
  452. sendmsg(job, "unknown 9P version");
  453. else{
  454. job->reply.version = "9P2000";
  455. sendmsg(job, 0);
  456. }
  457. }
  458. void
  459. rauth(Job *job)
  460. {
  461. sendmsg(job, "cs: authentication not required");
  462. }
  463. /*
  464. * don't flush till all the slaves are done
  465. */
  466. void
  467. rflush(Job *job)
  468. {
  469. flushjob(job->request.oldtag);
  470. sendmsg(job, 0);
  471. }
  472. void
  473. rattach(Job *job, Mfile *mf)
  474. {
  475. if(mf->busy == 0){
  476. mf->busy = 1;
  477. mf->user = estrdup(job->request.uname);
  478. }
  479. mf->qid.vers = vers++;
  480. mf->qid.type = QTDIR;
  481. mf->qid.path = 0LL;
  482. job->reply.qid = mf->qid;
  483. sendmsg(job, 0);
  484. }
  485. char*
  486. rwalk(Job *job, Mfile *mf)
  487. {
  488. char *err;
  489. char **elems;
  490. int nelems;
  491. int i;
  492. Mfile *nmf;
  493. Qid qid;
  494. err = 0;
  495. nmf = nil;
  496. elems = job->request.wname;
  497. nelems = job->request.nwname;
  498. job->reply.nwqid = 0;
  499. if(job->request.newfid != job->request.fid){
  500. /* clone fid */
  501. if(job->request.newfid<0){
  502. err = "clone newfid out of range";
  503. goto send;
  504. }
  505. nmf = newfid(job->request.newfid);
  506. if(nmf->busy){
  507. nmf = nil;
  508. err = "clone to used channel";
  509. goto send;
  510. }
  511. *nmf = *mf;
  512. nmf->user = estrdup(mf->user);
  513. nmf->fid = job->request.newfid;
  514. nmf->qid.vers = vers++;
  515. mf = nmf;
  516. }
  517. /* else nmf will be nil */
  518. qid = mf->qid;
  519. if(nelems > 0){
  520. /* walk fid */
  521. for(i=0; i<nelems && i<MAXWELEM; i++){
  522. if((qid.type & QTDIR) == 0){
  523. err = "not a directory";
  524. break;
  525. }
  526. if(strcmp(elems[i], "..") == 0 || strcmp(elems[i], ".") == 0){
  527. qid.type = QTDIR;
  528. qid.path = Qdir;
  529. Found:
  530. job->reply.wqid[i] = qid;
  531. job->reply.nwqid++;
  532. continue;
  533. }
  534. if(strcmp(elems[i], "cs") == 0){
  535. qid.type = QTFILE;
  536. qid.path = Qcs;
  537. goto Found;
  538. }
  539. err = "file does not exist";
  540. break;
  541. }
  542. }
  543. send:
  544. if(nmf != nil && (err!=nil || job->reply.nwqid<nelems)){
  545. cleanmf(nmf);
  546. free(nmf->user);
  547. nmf->user = 0;
  548. nmf->busy = 0;
  549. nmf->fid = 0;
  550. }
  551. if(err == nil)
  552. mf->qid = qid;
  553. sendmsg(job, err);
  554. return err;
  555. }
  556. void
  557. ropen(Job *job, Mfile *mf)
  558. {
  559. int mode;
  560. char *err;
  561. err = 0;
  562. mode = job->request.mode;
  563. if(mf->qid.type & QTDIR){
  564. if(mode)
  565. err = "permission denied";
  566. }
  567. job->reply.qid = mf->qid;
  568. job->reply.iounit = 0;
  569. sendmsg(job, err);
  570. }
  571. void
  572. rcreate(Job *job, Mfile *mf)
  573. {
  574. USED(mf);
  575. sendmsg(job, "creation permission denied");
  576. }
  577. void
  578. rread(Job *job, Mfile *mf)
  579. {
  580. int i, n, cnt;
  581. long off, toff, clock;
  582. Dir dir;
  583. uchar buf[Maxfdata];
  584. char *err;
  585. n = 0;
  586. err = 0;
  587. off = job->request.offset;
  588. cnt = job->request.count;
  589. if(mf->qid.type & QTDIR){
  590. clock = time(0);
  591. if(off == 0){
  592. dir.name = "cs";
  593. dir.qid.type = QTFILE;
  594. dir.qid.vers = vers;
  595. dir.qid.path = Qcs;
  596. dir.mode = 0666;
  597. dir.length = 0;
  598. dir.uid = mf->user;
  599. dir.gid = mf->user;
  600. dir.muid = mf->user;
  601. dir.atime = clock; /* wrong */
  602. dir.mtime = clock; /* wrong */
  603. n = convD2M(&dir, buf, sizeof buf);
  604. }
  605. job->reply.data = (char*)buf;
  606. } else {
  607. for(;;){
  608. /* look for an answer at the right offset */
  609. toff = 0;
  610. for(i = 0; mf->reply[i] && i < mf->nreply; i++){
  611. n = mf->replylen[i];
  612. if(off < toff + n)
  613. break;
  614. toff += n;
  615. }
  616. if(i < mf->nreply)
  617. break; /* got something to return */
  618. /* try looking up more answers */
  619. if(lookup(mf) == 0){
  620. /* no more */
  621. n = 0;
  622. goto send;
  623. }
  624. }
  625. /* give back a single reply (or part of one) */
  626. job->reply.data = mf->reply[i] + (off - toff);
  627. if(cnt > toff - off + n)
  628. n = toff - off + n;
  629. else
  630. n = cnt;
  631. }
  632. send:
  633. job->reply.count = n;
  634. sendmsg(job, err);
  635. }
  636. void
  637. cleanmf(Mfile *mf)
  638. {
  639. int i;
  640. if(mf->net != nil){
  641. free(mf->net);
  642. mf->net = nil;
  643. }
  644. if(mf->host != nil){
  645. free(mf->host);
  646. mf->host = nil;
  647. }
  648. if(mf->serv != nil){
  649. free(mf->serv);
  650. mf->serv = nil;
  651. }
  652. if(mf->rem != nil){
  653. free(mf->rem);
  654. mf->rem = nil;
  655. }
  656. for(i = 0; i < mf->nreply; i++){
  657. free(mf->reply[i]);
  658. mf->reply[i] = nil;
  659. mf->replylen[i] = 0;
  660. }
  661. mf->nreply = 0;
  662. mf->nextnet = netlist;
  663. }
  664. void
  665. rwrite(Job *job, Mfile *mf)
  666. {
  667. int cnt, n;
  668. char *err;
  669. char *field[4];
  670. char curerr[64];
  671. err = 0;
  672. cnt = job->request.count;
  673. if(mf->qid.type & QTDIR){
  674. err = "can't write directory";
  675. goto send;
  676. }
  677. if(cnt >= Maxrequest){
  678. err = "request too long";
  679. goto send;
  680. }
  681. job->request.data[cnt] = 0;
  682. /*
  683. * toggle debugging
  684. */
  685. if(strncmp(job->request.data, "debug", 5)==0){
  686. debug ^= 1;
  687. syslog(1, logfile, "debug %d", debug);
  688. goto send;
  689. }
  690. /*
  691. * toggle debugging
  692. */
  693. if(strncmp(job->request.data, "paranoia", 8)==0){
  694. paranoia ^= 1;
  695. syslog(1, logfile, "paranoia %d", paranoia);
  696. goto send;
  697. }
  698. /*
  699. * add networks to the default list
  700. */
  701. if(strncmp(job->request.data, "add ", 4)==0){
  702. if(job->request.data[cnt-1] == '\n')
  703. job->request.data[cnt-1] = 0;
  704. netadd(job->request.data+4);
  705. readipinterfaces();
  706. goto send;
  707. }
  708. /*
  709. * refresh all state
  710. */
  711. if(strncmp(job->request.data, "refresh", 7)==0){
  712. netinit(1);
  713. goto send;
  714. }
  715. /* start transaction with a clean slate */
  716. cleanmf(mf);
  717. /*
  718. * look for a general query
  719. */
  720. if(*job->request.data == '!'){
  721. err = genquery(mf, job->request.data+1);
  722. goto send;
  723. }
  724. if(debug)
  725. syslog(0, logfile, "write %s", job->request.data);
  726. if(paranoia)
  727. syslog(0, paranoiafile, "write %s by %s", job->request.data, mf->user);
  728. /*
  729. * break up name
  730. */
  731. n = getfields(job->request.data, field, 4, 1, "!");
  732. switch(n){
  733. case 1:
  734. mf->net = strdup("net");
  735. mf->host = strdup(field[0]);
  736. break;
  737. case 4:
  738. mf->rem = strdup(field[3]);
  739. /* fall through */
  740. case 3:
  741. mf->serv = strdup(field[2]);
  742. /* fall through */
  743. case 2:
  744. mf->host = strdup(field[1]);
  745. mf->net = strdup(field[0]);
  746. break;
  747. }
  748. /*
  749. * do the first net worth of lookup
  750. */
  751. if(lookup(mf) == 0){
  752. rerrstr(curerr, sizeof curerr);
  753. err = curerr;
  754. }
  755. send:
  756. job->reply.count = cnt;
  757. sendmsg(job, err);
  758. }
  759. void
  760. rclunk(Job *job, Mfile *mf)
  761. {
  762. cleanmf(mf);
  763. free(mf->user);
  764. mf->user = 0;
  765. mf->busy = 0;
  766. mf->fid = 0;
  767. sendmsg(job, 0);
  768. }
  769. void
  770. rremove(Job *job, Mfile *mf)
  771. {
  772. USED(mf);
  773. sendmsg(job, "remove permission denied");
  774. }
  775. void
  776. rstat(Job *job, Mfile *mf)
  777. {
  778. Dir dir;
  779. uchar buf[IOHDRSZ+Maxfdata];
  780. if(mf->qid.type & QTDIR){
  781. dir.name = ".";
  782. dir.mode = DMDIR|0555;
  783. } else {
  784. dir.name = "cs";
  785. dir.mode = 0666;
  786. }
  787. dir.qid = mf->qid;
  788. dir.length = 0;
  789. dir.uid = mf->user;
  790. dir.gid = mf->user;
  791. dir.muid = mf->user;
  792. dir.atime = dir.mtime = time(0);
  793. job->reply.nstat = convD2M(&dir, buf, sizeof buf);
  794. job->reply.stat = buf;
  795. sendmsg(job, 0);
  796. }
  797. void
  798. rwstat(Job *job, Mfile *mf)
  799. {
  800. USED(mf);
  801. sendmsg(job, "wstat permission denied");
  802. }
  803. void
  804. sendmsg(Job *job, char *err)
  805. {
  806. int n;
  807. uchar mdata[IOHDRSZ + Maxfdata];
  808. char ename[ERRMAX];
  809. if(err){
  810. job->reply.type = Rerror;
  811. snprint(ename, sizeof(ename), "cs: %s", err);
  812. job->reply.ename = ename;
  813. }else{
  814. job->reply.type = job->request.type+1;
  815. }
  816. job->reply.tag = job->request.tag;
  817. n = convS2M(&job->reply, mdata, sizeof mdata);
  818. if(n == 0){
  819. syslog(1, logfile, "sendmsg convS2M of %F returns 0", &job->reply);
  820. abort();
  821. }
  822. lock(&joblock);
  823. if(job->flushed == 0)
  824. if(write(mfd[1], mdata, n)!=n)
  825. error("mount write");
  826. unlock(&joblock);
  827. if(debug)
  828. syslog(0, logfile, "%F %d", &job->reply, n);
  829. }
  830. void
  831. error(char *s)
  832. {
  833. syslog(1, "cs", "%s: %r", s);
  834. _exits(0);
  835. }
  836. static int
  837. isvalidip(uchar *ip)
  838. {
  839. return ipcmp(ip, IPnoaddr) != 0 && ipcmp(ip, v4prefix) != 0;
  840. }
  841. static uchar loopbacknet[IPaddrlen] = {
  842. 0, 0, 0, 0,
  843. 0, 0, 0, 0,
  844. 0, 0, 0xff, 0xff,
  845. 127, 0, 0, 0
  846. };
  847. static uchar loopbackmask[IPaddrlen] = {
  848. 0xff, 0xff, 0xff, 0xff,
  849. 0xff, 0xff, 0xff, 0xff,
  850. 0xff, 0xff, 0xff, 0xff,
  851. 0xff, 0, 0, 0
  852. };
  853. void
  854. readipinterfaces(void)
  855. {
  856. if(myipaddr(ipa, mntpt) != 0)
  857. ipmove(ipa, IPnoaddr);
  858. sprint(ipaddr, "%I", ipa);
  859. if (debug)
  860. syslog(0, "dns", "ipaddr is %s\n", ipaddr);
  861. }
  862. /*
  863. * get the system name
  864. */
  865. void
  866. ipid(void)
  867. {
  868. uchar addr[6];
  869. Ndbtuple *t, *tt;
  870. char *p, *attr;
  871. Ndbs s;
  872. int f;
  873. char buf[Maxpath];
  874. /* use environment, ether addr, or ipaddr to get system name */
  875. if(mysysname == 0){
  876. /*
  877. * environment has priority.
  878. *
  879. * on the sgi power the default system name
  880. * is the ip address. ignore that.
  881. *
  882. */
  883. p = getenv("sysname");
  884. if(p){
  885. attr = ipattr(p);
  886. if(strcmp(attr, "ip") != 0)
  887. mysysname = strdup(p);
  888. }
  889. /*
  890. * the /net/ndb contains what the network
  891. * figured out from DHCP. use that name if
  892. * there is one.
  893. */
  894. if(mysysname == 0 && netdb != nil){
  895. ndbreopen(netdb);
  896. for(tt = t = ndbparse(netdb); t != nil; t = t->entry){
  897. if(strcmp(t->attr, "sys") == 0){
  898. mysysname = strdup(t->val);
  899. break;
  900. }
  901. }
  902. ndbfree(tt);
  903. }
  904. /* next network database, ip address, and ether address to find a name */
  905. if(mysysname == 0){
  906. t = nil;
  907. if(isvalidip(ipa))
  908. t = ndbgetvalue(db, &s, "ip", ipaddr, "sys", nil, 0);
  909. if(t == nil){
  910. for(f = 0; f < 3; f++){
  911. snprint(buf, sizeof buf, "%s/ether%d", mntpt, f);
  912. if(myetheraddr(addr, buf) >= 0){
  913. snprint(eaddr, sizeof(eaddr), "%E", addr);
  914. t = ndbgetvalue(db, &s, "ether", eaddr, "sys",
  915. nil, 0);
  916. if(t != nil)
  917. break;
  918. }
  919. }
  920. }
  921. for(tt = t; tt != nil; tt = tt->entry)
  922. if(strcmp(t->attr, "sys") == 0){
  923. mysysname = strdup(t->val);
  924. break;
  925. }
  926. ndbfree(t);
  927. }
  928. /* nothing else worked, use the ip address */
  929. if(mysysname == 0 && isvalidip(ipa))
  930. mysysname = strdup(ipaddr);
  931. /* set /dev/sysname if we now know it */
  932. if(mysysname){
  933. f = open("/dev/sysname", OWRITE);
  934. if(f >= 0){
  935. write(f, mysysname, strlen(mysysname));
  936. close(f);
  937. }
  938. }
  939. }
  940. }
  941. /*
  942. * Set up a list of default networks by looking for
  943. * /net/ * /clone.
  944. */
  945. void
  946. netinit(int background)
  947. {
  948. char clone[Maxpath];
  949. Network *np;
  950. static int working;
  951. if(background){
  952. switch(rfork(RFPROC|RFNOTEG|RFMEM|RFNOWAIT)){
  953. case 0:
  954. break;
  955. default:
  956. return;
  957. }
  958. lock(&netlock);
  959. }
  960. /* add the mounted networks to the default list */
  961. for(np = network; np->net; np++){
  962. if(np->considered)
  963. continue;
  964. snprint(clone, sizeof(clone), "%s/%s/clone", mntpt, np->net);
  965. if(access(clone, AEXIST) < 0)
  966. continue;
  967. if(netlist)
  968. last->next = np;
  969. else
  970. netlist = np;
  971. last = np;
  972. np->next = 0;
  973. np->considered = 1;
  974. }
  975. /* find out what our ip address is */
  976. readipinterfaces();
  977. /* set the system name if we need to, these says ip is all we have */
  978. ipid();
  979. if(debug)
  980. syslog(0, logfile, "mysysname %s eaddr %s ipaddr %s ipa %I\n",
  981. mysysname?mysysname:"???", eaddr, ipaddr, ipa);
  982. if(background){
  983. unlock(&netlock);
  984. _exits(0);
  985. }
  986. }
  987. /*
  988. * add networks to the standard list
  989. */
  990. void
  991. netadd(char *p)
  992. {
  993. Network *np;
  994. char *field[12];
  995. int i, n;
  996. n = getfields(p, field, 12, 1, " ");
  997. for(i = 0; i < n; i++){
  998. for(np = network; np->net; np++){
  999. if(strcmp(field[i], np->net) != 0)
  1000. continue;
  1001. if(np->considered)
  1002. break;
  1003. if(netlist)
  1004. last->next = np;
  1005. else
  1006. netlist = np;
  1007. last = np;
  1008. np->next = 0;
  1009. np->considered = 1;
  1010. }
  1011. }
  1012. }
  1013. int
  1014. lookforproto(Ndbtuple *t, char *proto)
  1015. {
  1016. for(; t != nil; t = t->entry)
  1017. if(strcmp(t->attr, "proto") == 0 && strcmp(t->val, proto) == 0)
  1018. return 1;
  1019. return 0;
  1020. }
  1021. /*
  1022. * lookup a request. the network "net" means we should pick the
  1023. * best network to get there.
  1024. */
  1025. int
  1026. lookup(Mfile *mf)
  1027. {
  1028. Network *np;
  1029. char *cp;
  1030. Ndbtuple *nt, *t;
  1031. char reply[Maxreply];
  1032. int i, rv;
  1033. int hack;
  1034. /* open up the standard db files */
  1035. if(db == 0)
  1036. ndbinit();
  1037. if(db == 0)
  1038. error("can't open mf->network database\n");
  1039. rv = 0;
  1040. if(mf->net == nil)
  1041. return 0; /* must have been a genquery */
  1042. if(strcmp(mf->net, "net") == 0){
  1043. /*
  1044. * go through set of default nets
  1045. */
  1046. for(np = mf->nextnet; np; np = np->next){
  1047. nt = (*np->lookup)(np, mf->host, mf->serv, 1);
  1048. if(nt == nil)
  1049. continue;
  1050. hack = np->fasttimeouthack && !lookforproto(nt, np->net);
  1051. for(t = nt; mf->nreply < Nreply && t; t = t->entry){
  1052. cp = (*np->trans)(t, np, mf->serv, mf->rem, hack);
  1053. if(cp){
  1054. /* avoid duplicates */
  1055. for(i = 0; i < mf->nreply; i++)
  1056. if(strcmp(mf->reply[i], cp) == 0)
  1057. break;
  1058. if(i == mf->nreply){
  1059. /* save the reply */
  1060. mf->replylen[mf->nreply] = strlen(cp);
  1061. mf->reply[mf->nreply++] = cp;
  1062. rv++;
  1063. }
  1064. }
  1065. }
  1066. ndbfree(nt);
  1067. np = np->next;
  1068. break;
  1069. }
  1070. mf->nextnet = np;
  1071. return rv;
  1072. }
  1073. /*
  1074. * if not /net, we only get one lookup
  1075. */
  1076. if(mf->nreply != 0)
  1077. return 0;
  1078. /*
  1079. * look for a specific network
  1080. */
  1081. for(np = netlist; np && np->net != nil; np++){
  1082. if(np->fasttimeouthack)
  1083. continue;
  1084. if(strcmp(np->net, mf->net) == 0)
  1085. break;
  1086. }
  1087. if(np && np->net != nil){
  1088. /*
  1089. * known network
  1090. */
  1091. syslog(0, logfile, "specific lookup %s", np->net);
  1092. nt = (*np->lookup)(np, mf->host, mf->serv, 1);
  1093. syslog(0, logfile, "returned %p", nt);
  1094. for(t = nt; mf->nreply < Nreply && t; t = t->entry){
  1095. cp = (*np->trans)(t, np, mf->serv, mf->rem, 0);
  1096. if(cp){
  1097. mf->replylen[mf->nreply] = strlen(cp);
  1098. mf->reply[mf->nreply++] = cp;
  1099. rv++;
  1100. }
  1101. }
  1102. ndbfree(nt);
  1103. return rv;
  1104. } else {
  1105. /*
  1106. * not a known network, don't translate host or service
  1107. */
  1108. if(mf->serv)
  1109. snprint(reply, sizeof(reply), "%s/%s/clone %s!%s",
  1110. mntpt, mf->net, mf->host, mf->serv);
  1111. else
  1112. snprint(reply, sizeof(reply), "%s/%s/clone %s",
  1113. mntpt, mf->net, mf->host);
  1114. mf->reply[0] = strdup(reply);
  1115. mf->replylen[0] = strlen(reply);
  1116. mf->nreply = 1;
  1117. return 1;
  1118. }
  1119. }
  1120. /*
  1121. * translate an ip service name into a port number. If it's a numeric port
  1122. * number, look for restricted access.
  1123. *
  1124. * the service '*' needs no translation.
  1125. */
  1126. char*
  1127. ipserv(Network *np, char *name, char *buf, int blen)
  1128. {
  1129. char *p;
  1130. int alpha = 0;
  1131. int restr = 0;
  1132. char port[10];
  1133. Ndbtuple *t, *nt;
  1134. Ndbs s;
  1135. /* '*' means any service */
  1136. if(strcmp(name, "*")==0){
  1137. strcpy(buf, name);
  1138. return buf;
  1139. }
  1140. /* see if it's numeric or symbolic */
  1141. port[0] = 0;
  1142. for(p = name; *p; p++){
  1143. if(isdigit(*p))
  1144. {}
  1145. else if(isalpha(*p) || *p == '-' || *p == '$')
  1146. alpha = 1;
  1147. else
  1148. return 0;
  1149. }
  1150. if(alpha){
  1151. t = ndbgetvalue(db, &s, np->net, name, "port", port, sizeof(port));
  1152. if(t == 0)
  1153. return 0;
  1154. } else {
  1155. /* look up only for tcp ports < 1024 to get the restricted
  1156. * attribute
  1157. */
  1158. t = nil;
  1159. if(atoi(name) < 1024 && strcmp(np->net, "tcp") == 0)
  1160. t = ndbgetvalue(db, &s, "port", name, "port", port, sizeof(port));
  1161. if(t == nil)
  1162. nstrcpy(port, name, sizeof(port));
  1163. }
  1164. if(t){
  1165. for(nt = t; nt; nt = nt->entry)
  1166. if(strcmp(nt->attr, "restricted") == 0)
  1167. restr = 1;
  1168. ndbfree(t);
  1169. }
  1170. snprint(buf, blen, "%s%s", port, restr ? "!r" : "");
  1171. return buf;
  1172. }
  1173. /*
  1174. * lookup an ip attribute
  1175. */
  1176. int
  1177. ipattrlookup(Ndb *db, char *ipa, char *attr, char *val, int vlen)
  1178. {
  1179. Ndbtuple *t, *nt;
  1180. char *alist[2];
  1181. alist[0] = attr;
  1182. t = ndbipinfo(db, "ip", ipa, alist, 1);
  1183. if(t == nil)
  1184. return 0;
  1185. for(nt = t; nt != nil; nt = nt->entry){
  1186. if(strcmp(nt->attr, attr) == 0){
  1187. nstrcpy(val, nt->val, vlen);
  1188. ndbfree(t);
  1189. return 1;
  1190. }
  1191. }
  1192. /* we shouldn't get here */
  1193. ndbfree(t);
  1194. return 0;
  1195. }
  1196. /*
  1197. * lookup (and translate) an ip destination
  1198. */
  1199. Ndbtuple*
  1200. iplookup(Network *np, char *host, char *serv, int nolookup)
  1201. {
  1202. char *attr;
  1203. Ndbtuple *t, *nt;
  1204. Ndbs s;
  1205. char ts[Maxservice];
  1206. char th[Maxhost];
  1207. char dollar[Maxhost];
  1208. uchar ip[IPaddrlen];
  1209. uchar net[IPaddrlen];
  1210. uchar tnet[IPaddrlen];
  1211. Ipifc *ifc;
  1212. Iplifc *lifc;
  1213. USED(nolookup);
  1214. /*
  1215. * start with the service since it's the most likely to fail
  1216. * and costs the least
  1217. */
  1218. werrstr("can't translate address");
  1219. if(serv==0 || ipserv(np, serv, ts, sizeof ts) == 0){
  1220. werrstr("can't translate service");
  1221. return 0;
  1222. }
  1223. /* for dial strings with no host */
  1224. if(strcmp(host, "*") == 0)
  1225. return ndbnew("ip", "*");
  1226. /*
  1227. * hack till we go v6 :: = 0.0.0.0
  1228. */
  1229. if(strcmp("::", host) == 0)
  1230. return ndbnew("ip", "*");
  1231. /*
  1232. * '$' means the rest of the name is an attribute that we
  1233. * need to search for
  1234. */
  1235. if(*host == '$'){
  1236. if(ipattrlookup(db, ipaddr, host+1, dollar, sizeof dollar))
  1237. host = dollar;
  1238. }
  1239. /*
  1240. * turn '[ip address]' into just 'ip address'
  1241. */
  1242. if(*host == '[' && host[strlen(host)-1] == ']'){
  1243. host++;
  1244. host[strlen(host)-1] = 0;
  1245. }
  1246. /*
  1247. * just accept addresses
  1248. */
  1249. attr = ipattr(host);
  1250. if(strcmp(attr, "ip") == 0)
  1251. return ndbnew("ip", host);
  1252. /*
  1253. * give the domain name server the first opportunity to
  1254. * resolve domain names. if that fails try the database.
  1255. */
  1256. t = 0;
  1257. werrstr("can't translate address");
  1258. if(strcmp(attr, "dom") == 0)
  1259. t = dnsiplookup(host, &s);
  1260. if(t == 0)
  1261. t = ndbgetvalue(db, &s, attr, host, "ip", th, sizeof(th));
  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. char th[Maxhost];
  1330. USED(np, nolookup, serv);
  1331. werrstr("can't translate address");
  1332. t = ndbgetvalue(db, &s, "sys", host, "telco", th, sizeof(th));
  1333. if(t == 0)
  1334. return ndbnew("telco", host);
  1335. return reorder(t, s.t);
  1336. }
  1337. /*
  1338. * translate a telephone address
  1339. */
  1340. char*
  1341. telcotrans(Ndbtuple *t, Network *np, char *serv, char *rem, int)
  1342. {
  1343. char reply[Maxreply];
  1344. char x[Maxservice];
  1345. if(strcmp(t->attr, "telco") != 0)
  1346. return 0;
  1347. if(rem != nil)
  1348. snprint(x, sizeof(x), "!%s", rem);
  1349. else
  1350. *x = 0;
  1351. if(serv)
  1352. snprint(reply, sizeof(reply), "%s/%s/clone %s!%s%s", mntpt, np->net,
  1353. t->val, serv, x);
  1354. else
  1355. snprint(reply, sizeof(reply), "%s/%s/clone %s%s", mntpt, np->net,
  1356. t->val, x);
  1357. return strdup(reply);
  1358. }
  1359. /*
  1360. * reorder the tuple to put x's line first in the entry
  1361. */
  1362. Ndbtuple*
  1363. reorder(Ndbtuple *t, Ndbtuple *x)
  1364. {
  1365. Ndbtuple *nt;
  1366. Ndbtuple *line;
  1367. /* find start of this entry's line */
  1368. for(line = x; line->entry == line->line; line = line->line)
  1369. ;
  1370. line = line->line;
  1371. if(line == t)
  1372. return t; /* already the first line */
  1373. /* remove this line and everything after it from the entry */
  1374. for(nt = t; nt->entry != line; nt = nt->entry)
  1375. ;
  1376. nt->entry = 0;
  1377. /* make that the start of the entry */
  1378. for(nt = line; nt->entry; nt = nt->entry)
  1379. ;
  1380. nt->entry = t;
  1381. return line;
  1382. }
  1383. /*
  1384. * create a slave process to handle a request to avoid one request blocking
  1385. * another. parent returns to job loop.
  1386. */
  1387. void
  1388. slave(void)
  1389. {
  1390. if(*isslave)
  1391. return; /* we're already a slave process */
  1392. switch(rfork(RFPROC|RFNOTEG|RFMEM|RFNOWAIT)){
  1393. case -1:
  1394. break;
  1395. case 0:
  1396. if(debug)
  1397. syslog(0, logfile, "slave %d", getpid());
  1398. *isslave = 1;
  1399. break;
  1400. default:
  1401. longjmp(masterjmp, 1);
  1402. }
  1403. }
  1404. /*
  1405. * call the dns process and have it try to translate a name
  1406. */
  1407. Ndbtuple*
  1408. dnsiplookup(char *host, Ndbs *s)
  1409. {
  1410. char buf[Maxreply];
  1411. Ndbtuple *t;
  1412. unlock(&dblock);
  1413. /* save the name before starting a slave */
  1414. snprint(buf, sizeof(buf), "%s", host);
  1415. slave();
  1416. if(strcmp(ipattr(buf), "ip") == 0)
  1417. t = dnsquery(mntpt, buf, "ptr");
  1418. else
  1419. t = dnsquery(mntpt, buf, "ip");
  1420. s->t = t;
  1421. if(t == nil){
  1422. rerrstr(buf, sizeof buf);
  1423. if(strstr(buf, "exist"))
  1424. werrstr("can't translate address: %s", buf);
  1425. else if(strstr(buf, "dns failure"))
  1426. werrstr("temporary problem: %s", buf);
  1427. }
  1428. lock(&dblock);
  1429. return t;
  1430. }
  1431. int
  1432. qmatch(Ndbtuple *t, char **attr, char **val, int n)
  1433. {
  1434. int i, found;
  1435. Ndbtuple *nt;
  1436. for(i = 1; i < n; i++){
  1437. found = 0;
  1438. for(nt = t; nt; nt = nt->entry)
  1439. if(strcmp(attr[i], nt->attr) == 0)
  1440. if(strcmp(val[i], "*") == 0
  1441. || strcmp(val[i], nt->val) == 0){
  1442. found = 1;
  1443. break;
  1444. }
  1445. if(found == 0)
  1446. break;
  1447. }
  1448. return i == n;
  1449. }
  1450. void
  1451. qreply(Mfile *mf, Ndbtuple *t)
  1452. {
  1453. Ndbtuple *nt;
  1454. String *s;
  1455. s = s_new();
  1456. for(nt = t; mf->nreply < Nreply && nt; nt = nt->entry){
  1457. s_append(s, nt->attr);
  1458. s_append(s, "=");
  1459. s_append(s, nt->val);
  1460. if(nt->line != nt->entry){
  1461. mf->replylen[mf->nreply] = s_len(s);
  1462. mf->reply[mf->nreply++] = strdup(s_to_c(s));
  1463. s_restart(s);
  1464. } else
  1465. s_append(s, " ");
  1466. }
  1467. s_free(s);
  1468. }
  1469. enum
  1470. {
  1471. Maxattr= 32,
  1472. };
  1473. /*
  1474. * generic query lookup. The query is of one of the following
  1475. * forms:
  1476. *
  1477. * attr1=val1 attr2=val2 attr3=val3 ...
  1478. *
  1479. * returns the matching tuple
  1480. *
  1481. * ipinfo attr=val attr1 attr2 attr3 ...
  1482. *
  1483. * is like ipinfo and returns the attr{1-n}
  1484. * associated with the ip address.
  1485. */
  1486. char*
  1487. genquery(Mfile *mf, char *query)
  1488. {
  1489. int i, n;
  1490. char *p;
  1491. char *attr[Maxattr];
  1492. char *val[Maxattr];
  1493. Ndbtuple *t;
  1494. Ndbs s;
  1495. n = getfields(query, attr, 32, 1, " ");
  1496. if(n == 0)
  1497. return "bad query";
  1498. if(strcmp(attr[0], "ipinfo") == 0)
  1499. return ipinfoquery(mf, attr, n);
  1500. /* parse pairs */
  1501. for(i = 0; i < n; i++){
  1502. p = strchr(attr[i], '=');
  1503. if(p == 0)
  1504. return "bad query";
  1505. *p++ = 0;
  1506. val[i] = p;
  1507. }
  1508. /* give dns a chance */
  1509. if((strcmp(attr[0], "dom") == 0 || strcmp(attr[0], "ip") == 0) && val[0]){
  1510. t = dnsiplookup(val[0], &s);
  1511. if(t){
  1512. if(qmatch(t, attr, val, n)){
  1513. qreply(mf, t);
  1514. ndbfree(t);
  1515. return 0;
  1516. }
  1517. ndbfree(t);
  1518. }
  1519. }
  1520. /* first pair is always the key. It can't be a '*' */
  1521. t = ndbsearch(db, &s, attr[0], val[0]);
  1522. /* search is the and of all the pairs */
  1523. while(t){
  1524. if(qmatch(t, attr, val, n)){
  1525. qreply(mf, t);
  1526. ndbfree(t);
  1527. return 0;
  1528. }
  1529. ndbfree(t);
  1530. t = ndbsnext(&s, attr[0], val[0]);
  1531. }
  1532. return "no match";
  1533. }
  1534. /*
  1535. * resolve an ip address
  1536. */
  1537. static Ndbtuple*
  1538. ipresolve(char *attr, char *host)
  1539. {
  1540. Ndbtuple *t, *nt, **l;
  1541. t = iplookup(&network[Ntcp], host, "*", 0);
  1542. for(l = &t; *l != nil; ){
  1543. nt = *l;
  1544. if(strcmp(nt->attr, "ip") != 0){
  1545. *l = nt->entry;
  1546. nt->entry = nil;
  1547. ndbfree(nt);
  1548. continue;
  1549. }
  1550. strcpy(nt->attr, attr);
  1551. l = &nt->entry;
  1552. }
  1553. return t;
  1554. }
  1555. char*
  1556. ipinfoquery(Mfile *mf, char **list, int n)
  1557. {
  1558. int i, nresolve;
  1559. int resolve[Maxattr];
  1560. Ndbtuple *t, *nt, **l;
  1561. char *attr, *val;
  1562. /* skip 'ipinfo' */
  1563. list++; n--;
  1564. if(n < 2)
  1565. return "bad query";
  1566. /* get search attribute=value */
  1567. attr = *list++; n--;
  1568. val = strchr(attr, '=');
  1569. if(val == nil)
  1570. return "bad query";
  1571. *val++ = 0;
  1572. /*
  1573. * don't let ndbipinfo resolve the addresses, we're
  1574. * better at it.
  1575. */
  1576. nresolve = 0;
  1577. for(i = 0; i < n; i++)
  1578. if(*list[i] == '@'){
  1579. list[i]++;
  1580. resolve[i] = 1;
  1581. nresolve++;
  1582. } else
  1583. resolve[i] = 0;
  1584. t = ndbipinfo(db, attr, val, list, n);
  1585. if(t == nil)
  1586. return "no match";
  1587. if(nresolve != 0){
  1588. for(l = &t; *l != nil;){
  1589. nt = *l;
  1590. /* already an address? */
  1591. if(strcmp(ipattr(nt->val), "ip") == 0){
  1592. l = &(*l)->entry;
  1593. continue;
  1594. }
  1595. /* user wants it resolved? */
  1596. for(i = 0; i < n; i++)
  1597. if(strcmp(list[i], nt->attr) == 0)
  1598. break;
  1599. if(i >= n || resolve[i] == 0){
  1600. l = &(*l)->entry;
  1601. continue;
  1602. }
  1603. /* resolve address and replace entry */
  1604. *l = ipresolve(nt->attr, nt->val);
  1605. while(*l != nil)
  1606. l = &(*l)->entry;
  1607. *l = nt->entry;
  1608. nt->entry = nil;
  1609. ndbfree(nt);
  1610. }
  1611. }
  1612. /* make it all one line */
  1613. for(nt = t; nt != nil; nt = nt->entry){
  1614. if(nt->entry == nil)
  1615. nt->line = t;
  1616. else
  1617. nt->line = nt->entry;
  1618. }
  1619. qreply(mf, t);
  1620. return nil;
  1621. }
  1622. void*
  1623. emalloc(int size)
  1624. {
  1625. void *x;
  1626. x = malloc(size);
  1627. if(x == nil)
  1628. abort();
  1629. memset(x, 0, size);
  1630. return x;
  1631. }
  1632. char*
  1633. estrdup(char *s)
  1634. {
  1635. int size;
  1636. char *p;
  1637. size = strlen(s)+1;
  1638. p = malloc(size);
  1639. if(p == nil)
  1640. abort();
  1641. memmove(p, s, size);
  1642. return p;
  1643. }