sys/src/cmd/ndb/inform.c

   1 /* RFC2136 DNS inform - necessary for Win2k3 DNS servers */
   2 #include <u.h>
   3 #include <libc.h>
   4 #include <bio.h>
   5 #include <ndb.h>
   6 #include <ip.h>
   7 #include "dns.h"
   8
   9 enum {
  10         FQDNMAX = 255,
  11 };
  12
  13 char *errmsgs[] = {
  14         [0]  "ok",
  15         [1]  "request format error",
  16         [2]  "internal server error",
  17         [3]  "domain name does not exist",
  18         [4]  "request not supported",
  19         [5]  "permission denied",
  20         [6]  "domain name already exists",
  21         [7]  "resource record already exists",
  22         [8]  "resource record does not exist",
  23         [9]  "server not authoritative",
  24         [10] "domain name not in zone",
  25 };
  26
  27 char *dnsrch[] = {
  28         "dnsdomain",
  29         "dom",
  30 };
  31
  32 void
  33 usage(void)
  34 {
  35         fprint(2, "usage: %s [-x netmtpt]\n", argv0);
  36         exits("usage");
  37 }
  38
  39 void
  40 ding(void *, char *msg)
  41 {
  42         if(strstr(msg, "alarm") != nil)
  43                 noted(NCONT);
  44         noted(NDFLT);
  45 }
  46
  47 int
  48 g16(uchar **p)
  49 {
  50         int n;
  51
  52         n  = *(*p)++ << 8;
  53         n |= *(*p)++;
  54         return n;
  55 }
  56
  57 void
  58 p16(uchar **p, int n)
  59 {
  60         *(*p)++ = n >> 8;
  61         *(*p)++ = n;
  62 }
  63
  64 void
  65 p32(uchar **p, int n)
  66 {
  67         *(*p)++ = n >> 24;
  68         *(*p)++ = n >> 16;
  69         *(*p)++ = n >> 8;
  70         *(*p)++ = n;
  71 }
  72
  73 void
  74 pmem(uchar **p, void *v, int len)
  75 {
  76         memmove(*p, v, len);
  77         *p += len;
  78 }
  79
  80 void
  81 pname(uchar **p, char *s)
  82 {
  83         uchar *len;
  84
  85         while (*s){
  86                 len = (*p)++;
  87                 while(*s && *s != '.')
  88                         *(*p)++ = *s++;
  89                 *len = *p - len - 1;
  90                 if(*s == '.')
  91                         s++;
  92         }
  93         *(*p)++ = 0;
  94 }
  95
  96 void
  97 main(int argc, char *argv[])
  98 {
  99         int debug, len, fd;
 100         uint err;
 101         char *sysname, *dnsdomain, *dom, *ns, net[32];
 102         uchar *p, buf[4096], addr[IPv4addrlen], v6addr[IPaddrlen];
 103         ushort txid;
 104         Ndb *db;
 105         Ndbtuple *t, *tt;
 106         static char *query[] = { "dom", "dnsdomain", "ns", };
 107
 108         fmtinstall('I', eipfmt);
 109         fmtinstall('V', eipfmt);
 110         setnetmtpt(net, sizeof net, nil);
 111
 112         debug = 0;
 113         ns = nil;
 114         dom = nil;
 115         dnsdomain = nil;
 116         ARGBEGIN{
 117         case 'd':
 118                 debug = 1;
 119                 break;
 120         case 'x':
 121                 setnetmtpt(net, sizeof net, EARGF(usage()));
 122                 break;
 123         default:
 124                 usage();
 125         }ARGEND;
 126
 127         if(argc != 0)
 128                 usage();
 129
 130         if((sysname = getenv("sysname")) == nil)
 131                 sysfatal("$sysname not set");
 132
 133         if((db = ndbopen(nil)) == nil)
 134                 sysfatal("can't open ndb: %r");
 135         tt = ndbipinfo(db, "sys", sysname, query, nelem(query));
 136         for(t = tt; t; t = t->entry)
 137                 if(strcmp(t->attr, "ns") == 0)
 138                         ns = t->val;
 139                 else if(strcmp(t->attr, "dom") == 0)
 140                         dom = t->val;
 141                 else if(strcmp(t->attr, "dnsdomain") == 0)
 142                         dnsdomain = t->val;
 143         ndbfree(tt);
 144         ndbclose(db);
 145
 146         if(!ns)
 147                 sysfatal("no relevant ns=");
 148         if(!dom)
 149                 sysfatal("no relevant dom=");
 150         if(!dnsdomain)
 151                 sysfatal("no relevant dnsdomain=");
 152
 153         myipaddr(v6addr, net);
 154         memmove(addr, v6addr + IPaddrlen - IPv4addrlen, IPv4addrlen);
 155
 156         if(debug){
 157                 print("ip=%V\n", addr);
 158                 print("ns=%s\n", ns);
 159                 print("dnsdomain=%s\n", dnsdomain);
 160                 print("dom=%s\n", dom);
 161         }
 162
 163         if((fd = dial(netmkaddr(ns, "udp", "dns"), 0, 0, 0)) < 0)
 164                 sysfatal("can't dial %s: %r", ns);
 165
 166         txid = time(nil) + getpid();
 167
 168         p = buf;
 169         p16(&p, txid);          /* ID */
 170         p16(&p, 5<<11);         /* flags */
 171         p16(&p, 1);             /* # Zones */
 172         p16(&p, 0);             /* # prerequisites */
 173         p16(&p, 2);             /* # updates */
 174         p16(&p, 0);             /* # additionals */
 175
 176         pname(&p, dnsdomain);   /* zone */
 177         p16(&p, Tsoa);          /* zone type */
 178         p16(&p, Cin);           /* zone class */
 179
 180         /* delete old name */
 181         pname(&p, dom);         /* name */
 182         p16(&p, Ta);            /* type: v4 addr */
 183         p16(&p, Call);          /* class */
 184         p32(&p, 0);             /* TTL */
 185         p16(&p, 0);             /* data len */
 186
 187         /* add new A record */
 188         pname(&p, dom);         /* name */
 189         p16(&p, Ta);            /* type: v4 addr */
 190         p16(&p, Cin);           /* class */
 191         p32(&p, 60*60*25);      /* TTL (25 hours) */
 192         p16(&p, IPv4addrlen);   /* data len */
 193         pmem(&p, addr, IPv4addrlen);    /* v4 address */
 194
 195         len = p - buf;
 196         if(write(fd, buf, len) != len)
 197                 sysfatal("write failed: %r");
 198
 199         notify(ding);
 200         alarm(3000);
 201         do{
 202                 if(read(fd, buf, sizeof buf) < 0)
 203                         sysfatal("timeout");
 204                 p = buf;
 205         }while(g16(&p) != txid);
 206         alarm(0);
 207
 208         close(fd);
 209
 210         err = g16(&p) & 7;
 211         if(err != 0 && err != 7)        /* err==7 is just a "yes, I know" warning */
 212                 if(err < nelem(errmsgs))
 213                         sysfatal("%s", errmsgs[err]);
 214                 else
 215                         sysfatal("unknown dns server error %d", err);
 216         exits(0);
 217 }