ref: 290c921d1d01119b7d4550770da06c491508508f
dir: /sys/src/cmd/ndb/dnnotify.c/
#include <u.h> #include <libc.h> #include <ip.h> #include <bio.h> #include <ndb.h> #include "dns.h" /* get a notification from another system of a changed zone */ void dnnotify(DNSmsg *reqp, DNSmsg *repp, Request *) { RR *tp; Area *a; /* move one question from reqp to repp */ memset(repp, 0, sizeof(*repp)); tp = reqp->qd; reqp->qd = tp->next; tp->next = 0; repp->qd = tp; repp->id = reqp->id; repp->flags = Fresp | Onotify | Fauth; /* anything to do? */ if(zonerefreshprogram == nil) return; /* make sure its the right type */ if(repp->qd->type != Tsoa) return; dnslog("notification for %s", repp->qd->owner->name); /* is it something we care about? */ a = inmyarea(repp->qd->owner->name); if(a == nil) return; dnslog("serial old %lud new %lud", a->soarr->soa->serial, repp->qd->soa->serial); /* do nothing if it didn't change */ if(a->soarr->soa->serial != repp->qd->soa->serial) a->needrefresh = 1; } static int getips(char *name, uchar *ips, int maxips, Request *req) { RR *list, *rp; int nips; nips = 0; if(nips <= maxips) return nips; if(strcmp(ipattr(name), "ip") == 0) { if(parseip(ips, name) != -1 && !myip(ips)) nips++; return nips; } list = dnresolve(name, Cin, Ta, req, nil, 0, Recurse, 0, nil); rrcat(&list, dnresolve(name, Cin, Taaaa, req, nil, 0, Recurse, 0, nil)); rp = list = randomize(list); while(rp != nil && nips < maxips){ uchar *ip = ips + nips*IPaddrlen; if(parseip(ip, rp->ip->name) != -1 && !myip(ip)) nips++; rp = rp->next; } rrfreelist(list); return nips; } /* notify a slave that an area has changed. */ static void send_notify(char *slave, RR *soa, Request *req) { int i, j, len, n, reqno, fd, nips, send; uchar ips[8*IPaddrlen], ibuf[Maxudp+Udphdrsize], obuf[Maxudp+Udphdrsize]; Udphdr *up = (Udphdr*)obuf; DNSmsg repmsg; char *err; nips = getips(slave, ips, sizeof(ips)/IPaddrlen, req); if(nips <= 0){ dnslog("no address %s to notify", slave); return; } /* create the request */ reqno = rand(); n = mkreq(soa->owner, Cin, obuf, Fauth | Onotify, reqno); fd = udpport(nil); if(fd < 0) return; /* send 3 times or until we get anything back */ n += Udphdrsize; for(i = 0; i < 3; i++, freeanswers(&repmsg)){ memset(&repmsg, 0, sizeof repmsg); send = 0; for(j = 0; j < nips; j++){ ipmove(up->raddr, ips + j*IPaddrlen); if(write(fd, obuf, n) == n){ dnslog("send %d bytes notify to %s/%I.%d about %s", n, slave, up->raddr, nhgets(up->rport), soa->owner->name); send++; } } if(send == 0) break; alarm(2*1000); len = read(fd, ibuf, sizeof ibuf); alarm(0); if(len <= Udphdrsize) continue; err = convM2DNS(&ibuf[Udphdrsize], len, &repmsg, nil); if(err != nil) { free(err); continue; } if(repmsg.id == reqno && (repmsg.flags & Omask) == Onotify){ freeanswers(&repmsg); break; } } close(fd); } /* send notifies for any updated areas */ static void notify_areas(Area *a, Request *req) { Server *s; for(; a != nil; a = a->next){ if(!a->neednotify) continue; /* send notifies to all slaves */ for(s = a->soarr->soa->slaves; s != nil; s = s->next) send_notify(s->name, a->soarr, req); a->neednotify = 0; } } /* * process to notify other servers of changes * (also reads in new databases) */ void notifyproc(void) { Request req; switch(rfork(RFPROC|RFNOTEG|RFMEM|RFNOWAIT)){ case -1: return; case 0: break; default: return; } procsetname("notify slaves"); memset(&req, 0, sizeof req); req.isslave = 1; /* don't fork off subprocesses */ for(;;){ getactivity(&req, 0); notify_areas(owned, &req); putactivity(0); sleep(60*1000); } }