123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233 |
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <ndb.h>
- #include <ip.h>
- typedef struct x
- {
- Ndbtuple *t;
- Ndbtuple *it;
- Ndbtuple *nt;
- } X;
- X x[4096];
- int nx;
- char *domname = "research.att.com";
- int domnamlen;
- char*
- upper(char *x)
- {
- char *p;
- int c;
- for(p = x; c = *p; p++)
- *p = toupper(c);
- return x;
- }
- void
- printArecord(int fd, X *p)
- {
- Ndbtuple *nt;
- char *c;
- char *dom = 0;
- char *curdom = 0;
- int i, cdlen = 0;
- int mxweight = 0;
- if(p->nt) {
- return;
- }
- for(nt=p->t; nt; nt = nt->entry) {
- /* we are only going to handle things in the specified domain */
- c = strchr(nt->val, '.');
- if (c==0 || strcmp(++c, domname)!=0)
- continue;
- i = c - nt->val - 1;
- if(strcmp(nt->attr, "dom") == 0) {
- curdom = nt->val;
- cdlen = i;
- if (dom == 0) {
- dom = curdom;
- fprint(fd, "%-.*s%.*s IN A %s\n", i, nt->val, 15-i, " ", p->it->val);
- } else
- fprint(fd, "%-.*s%.*s IN CNAME %s.\n", i, nt->val, 15-i, " ", dom);
- } else if(strcmp(nt->attr, "mx") == 0) {
- if (curdom != 0)
- fprint(fd, "%-.*s%.*s MX %d %s.\n", cdlen, curdom, 15-cdlen, " ", mxweight++, nt->val);
- }
- }
- }
- void
- printentry(int fd, X *p)
- {
- Ndbtuple *nt;
- if(p->nt)
- return;
- fprint(fd, "%s ", p->it->val);
- for(nt = p->t; nt; nt = nt->entry)
- if(strcmp(nt->attr, "dom") == 0)
- fprint(fd, " %s", nt->val);
- for(nt = p->t; nt; nt = nt->entry)
- if(strcmp(nt->attr, "sys") == 0)
- fprint(fd, " %s", nt->val);
- fprint(fd, "\n");
- }
- void
- printsys(int fd, X *p)
- {
- Ndbtuple *nt;
- for(nt = p->t; nt; nt = nt->entry)
- if(strcmp(nt->attr, "dom") == 0)
- fprint(fd, "%s\n", nt->val);
- }
- void
- printtxt(int fd, X *p)
- {
- int i;
- Ndbtuple *nt;
- if(p->nt){
- for(;;){
- i = strlen(p->it->val);
- if(strcmp(p->it->val+i-2, ".0") == 0)
- p->it->val[i-2] = 0;
- else
- break;
- }
- fprint(fd, "\nNET : %s : %s\n", p->it->val, upper(p->nt->val));
- return;
- }
- fprint(fd, "HOST : %s :", p->it->val);
- i = 0;
- for(nt = p->t; nt; nt = nt->entry)
- if(strcmp(nt->attr, "dom") == 0){
- if(i++ == 0)
- fprint(fd, " %s", upper(nt->val));
- else
- fprint(fd, ", %s", upper(nt->val));
- }
- fprint(fd, "\n");
- }
- void
- parse(char *file)
- {
- int i;
- Ndb *db;
- Ndbtuple *t, *nt, *tt, *ipnett;
- char *p;
- db = ndbopen(file);
- if(db == 0)
- exits("no database");
- while(t = ndbparse(db)){
- for(nt = t; nt; nt = nt->entry){
- if(strcmp(nt->attr, "ip") == 0)
- break;
- if(strcmp(nt->attr, "flavor") == 0
- && strcmp(nt->val, "console") == 0)
- return;
- }
- if(nt == 0){
- ndbfree(t);
- continue;
- }
- /* dump anything not on our nets */
- ipnett = 0;
- for(tt = t; tt; tt = tt->entry){
- if(strcmp(tt->attr, "ipnet") == 0){
- ipnett = tt;
- break;
- }
- if(strcmp(tt->attr, "dom") == 0){
- i = strlen(tt->val);
- p = tt->val+i-domnamlen;
- if(p >= tt->val && strcmp(p, domname) == 0)
- break;
- }
- }
- if(tt == 0){
- ndbfree(t);
- continue;
- }
- for(; nt; nt = nt->entry){
- if(strcmp(nt->attr, "ip") != 0)
- continue;
- x[nx].it = nt;
- x[nx].nt = ipnett;
- x[nx++].t = t;
- }
- }
- }
- void
- main(int argc, char *argv[])
- {
- int i, fd;
- char fn[128];
- if (argc>1)
- domname = argv[1];
- domnamlen = strlen(domname);
- if(argc > 2){
- for(i = 2; i < argc; i++)
- parse(argv[i]);
- } else {
- parse("/lib/ndb/local");
- parse("/lib/ndb/friends");
- }
-
- // sprint(fn, "/lib/ndb/hosts.%-.21s", domname);
- // fd = create(fn, OWRITE, 0664);
- // if(fd < 0){
- // fprint(2, "can't create %s: %r\n", fn);
- // exits("boom");
- // }
- // for(i = 0; i < nx; i++)
- // printentry(fd, &x[i]);
- // close(fd);
- //
- sprint(fn, "/lib/ndb/db.%-.24s", domname);
- fd = create(fn, OWRITE, 0664);
- if(fd < 0){
- fprint(2, "can't create %s: %r\n", fn);
- exits("boom");
- }
- fprint(fd, "; This file is generated automatically, do not edit!\n");
- for(i = 0; i < nx; i++)
- printArecord(fd, &x[i]);
- close(fd);
- sprint(fn, "/lib/ndb/equiv.%-.21s", domname);
- fd = create(fn, OWRITE, 0664);
- if(fd < 0){
- fprint(2, "can't create %s: %r\n", fn);
- exits("boom");
- }
- for(i = 0; i < nx; i++)
- printsys(fd, &x[i]);
- close(fd);
- sprint(fn, "/lib/ndb/txt.%-.23s", domname);
- fd = create(fn, OWRITE, 0664);
- if(fd < 0){
- fprint(2, "can't create %s: %r\n", fn);
- exits("boom");
- }
- for(i = 0; i < nx; i++)
- printtxt(fd, &x[i]);
- close(fd);
- exits(0);
- }
|