chiark / gitweb /
new subfile stuff
[chiark-utils.git] / scripts / named-conf
1 #!/usr/bin/perl -w
2
3 use strict;
4 use IO::File;
5 use Data::Dumper;
6 use POSIX;
7
8 use vars qw($quis
9             $mode $doall
10             $etcfile $where
11             $debug $needglue $localonly $verbosity);
12
13 $quis= $0; $quis =~ s,.*/,,;
14
15 $mode= '';
16 $doall= 0;
17 $etcfile= "/etc/bind/chiark-conf-gen.zones";
18 $where= '<built-in>';
19 $debug= 0;
20 $needglue= 2;
21 $localonly= 0;
22 $verbosity= 1;
23
24 use vars qw($dig_owner $dig_type $dig_rdata);
25
26 while (@ARGV && $ARGV[0] =~ m/^\-/) {
27     $_= shift @ARGV;
28     if (s/^\-\-//) {
29         last if m/^$/;
30         if (m/^(yes|no|force)$/) { m/^./; $mode= $&; }
31         elsif (m/^all$/) { $doall=1; }
32         elsif (m/^config$/) { $etcfile= loarg(); $where= '--config option'; }
33         elsif (m/^glueless$/) { $needglue--; }
34         elsif (m/^localonly$/) { $localonly=1; }
35         elsif (m/^quiet$/) { $verbosity=0; }
36         elsif (m/^verbose$/) { $verbosity=2; }
37         else { usageerr("unknown option --$_"); }
38     } else {
39         s/^\-//;
40         last if m/^$/;
41         while (m/^./) {
42             if (s/^[ynf]//) { $mode=$&; }
43             elsif (s/^A//) { $doall=1; }
44             elsif (s/^C//) { $etcfile= soarg(); $where= '-C option'; }
45             elsif (s/^D//) { $debug++; }
46             elsif (s/^g//) { $needglue--; }
47             elsif (s/^l//) { $localonly=1; }
48             elsif (s/^q//) { $verbosity=0; }
49             elsif (s/^v//) { $verbosity=2; }
50             else { usageerr("unknown option -$&"); }
51         }
52     }
53 }
54
55 sub loarg() { usageerr("missing option value") if !@ARGV; return shift @ARGV; }
56 sub soarg() { my ($rv); $rv=$_; $_=''; return length $rv ? $rv : loarg(); }
57
58 usageerr("-g may be specified at most twice") if $needglue<0;
59 usageerr("-D may be specified at most twice") if $debug>2;
60 usageerr("must specify either -f|-y|-n or zones (and not both)")
61     if !!$mode == !!@ARGV;
62
63 sub usageerr ($) {
64     die
65 "$_[0]
66 usage: named-conf-regen [-rvq] -f|-y|-n|<zone>...\n".
67 "operation modes:\n".
68 " -f --force   install without checking\n".
69 " -y --yes     check and install\n".
70 " -n --no      check only\n".
71 "additional options:\n".
72 " -q --quiet   no output for OK zones\n".
73 " -v --verbose extra verbose\n";
74 }
75
76 cfg_fail("config filename $etcfile should not be directory")
77     if $etcfile =~ m,/$,;
78
79 use vars qw($default_dir);
80 $default_dir= $etcfile =~ m,^.*/, ? $& : './';
81
82 use vars qw($slave_dir $slave_prefix $slave_suffix);
83 $slave_dir= 'slave';
84 $slave_prefix= '';
85 $slave_suffix= '';
86
87 use vars qw(@self_ns @self_soa @self_addr);
88 @self_ns= @self_soa= @self_addr= ();
89
90 use vars qw(%zone_cfg @zone_cfg_list);
91 %zone_cfg= ();
92 @zone_cfg_list= ();
93
94 use vars qw($output $default_output %output_contents);
95 $output= '';
96 $default_output= '';
97 %output_contents= ();
98
99 use vars qw($check $install);
100 $check= $mode !~ m/^f/;
101 $install= $mode =~ m/^[yf]/;
102
103 read_config($etcfile);
104 debug_dump('@zone_cfg_list %zone_cfg');
105 process_zones($mode ? @zone_cfg_list : @ARGV);
106 debug_dump('%output_contents');
107 output_files() if $install;
108
109 exit 0;
110
111 #-------------------- configuration reading
112
113 sub cfg_fail ($) { die "$quis: $where:\n $_[0]\n"; }
114
115 sub read_config ($) {
116     my ($if) = @_;
117     my ($fh,$z,@self,$before,
118         $mod,$dir,$prefix,$suffix,$subfile,$lprefix,$lsuffix,$zf);
119     local ($_);
120
121     $fh= new IO::File $if,'r' or cfg_fail("open $if:\n $!");
122     $before= '';
123     for (;;) {
124         if (!defined($_= <$fh>)) {
125             cfg_fail("read config file $if:\n $!") if $fh->error();
126             last;
127         }
128         chomp; s/\s+$//;
129         if (m/\\$/) { $before.= $_; next; }
130         $_= $before.$_;
131         $before= '';
132         s/^\s+//;
133         $where= "$if:$.";
134         next if m/^\#/;
135         last if m/^end$/;
136         next unless m/\S/;
137         if (m/^self(\-ns|\-soa|)\s+(\S.*\S)/) {
138             @self= split /\s+/, $2;
139             @self_ns= @self if $1 ne '-soa';
140             @self_soa= @self if $1 ne '-ns';
141         } elsif (m/^self\-addr\s+([0-9. \t]+)/) {
142             @self_addr= split /\s+/, $1;
143         } elsif (m,^
144                  primary\-dir ([*?]?)
145                  \s+ (\S+)/([^/ \t]*)
146                  (?: \s+ ([^/ \t]*) (?: (/.+) )?
147                   )?
148                  $,x) {
149             ($mod, $dir, $prefix, $suffix, $subfile) =
150                 ($1,qualify($2),$3,$4,$5);
151             $suffix= '' if !defined $suffix;
152             $subfile= '' if !defined $subfile;
153             $suffix= '_db' if !length $suffix && !length $subfile;
154             if (-d "$dir$prefix") { $dir.=$prefix; $prefix=''; }
155             opendir D, $dir or cfg_fail("open primary-dir $dir:\n $!");
156             $lprefix= length $prefix; $lsuffix= length $suffix;
157             while (defined($_= readdir D)) {
158                 next if m/^\./ && !$lprefix;
159                 next unless length > $lprefix+$lsuffix;
160                 next unless substr($_,0,$lprefix) eq $prefix;
161                 next unless substr($_,length($_)-$lsuffix) eq $suffix;
162                 $z= substr($_,$lprefix,length($_)-($lprefix+$lsuffix));
163                 $zf= $dir.'/'.$prefix.$z.$suffix.$subfile;
164                 if (!stat $zf) {
165                     next if length $subfile && $! == &ENOENT;
166                     cfg_fail("cannot stat zonefile $zf:\n $!");
167                 }
168                 -f _ or cfg_fail("zonefile $zf is not a plain file");
169                 zone_conf($z,'primary','p',$mod,$zf);
170             }
171             closedir D or cfg_fail("close primary-dir $dir:\n $!");
172         } elsif (m/^primary([*?]?)\s+(\S+)\s+(\S+)$/) {
173             zone_conf($2,'primary','p',$1,qualify($3));
174         } elsif (m/^published([*?]?)\s+(\S+)\s+([0-9.\t]+)$/) {
175             zone_conf($2,'published','s',$1,'',$3);
176         } elsif (m/^stealth([*?]?)\s+(\S+)\s+([0-9. \t]+)$/) {
177             zone_conf($2,'stealth','u',$1,'',split /\s+/, $3);
178         } elsif (m/^slave\-dir\s+(\S+)(?:(?:\s+(\S+))\s+(\S+))?$/) {
179             ($slave_dir, $slave_prefix, $slave_suffix) = (qualify($1),$2,$3);
180         } elsif (m/^output\s+bind8\+(\S+)$/) {
181             cfg_fail("default output may not apply to only some zones")
182                 if @zone_cfg_list && length $default_output;
183             set_output(qualify($1));
184         } elsif (m/^include\s+(\S+)$/) {
185             read_config($1);
186         } else {
187             cfg_fail("unknown configuration directive".
188                      " or incorrect syntax or arguments");
189         }
190     }
191     $fh->close or cfg_fail("close config file $if:\n $!");
192 }
193
194 sub qualify ($) {
195     my ($i) = @_;
196     $i= "$default_dir$i" unless $i =~ m,^/,;
197     return $i;
198 }
199
200 sub zone_conf ($$$$$@) {
201     my ($zone,$style,$sabbr,$mod,$file,@servers) = @_;
202     my ($sfx,$aref);
203     $file= qualify("$slave_dir/$slave_prefix".$zone.$slave_suffix)
204         unless length $file;
205     if (!length $output) {
206         $default_output= qualify('chiark-conf-gen.bind8')
207             unless length $default_output;
208         set_output($default_output);
209     }
210     cfg_fail("redefined zone $zone") if exists $zone_cfg{$zone};
211     $zone_cfg{$zone}{'file'}= $file;
212     $zone_cfg{$zone}{'style_p'}= $style.$mod;
213     $zone_cfg{$zone}{'s'}= $sabbr.$mod; # p)rimary s)econdary u)npub f)oreign
214     $zone_cfg{$zone}{'servers'}= [ @servers ];
215     foreach $sfx (qw(soa ns addr)) {
216         { no strict 'refs'; $aref= [ @{ "self_$sfx" } ]; }
217         @$aref or cfg_fail("failed to specify self-$sfx before zone");
218         $zone_cfg{$zone}{"self_$sfx"}= $aref;
219     }
220     $zone_cfg{$zone}{'output'}= $output;
221     push @zone_cfg_list, $zone;
222 }
223
224 sub set_output($) {
225     my ($newout) = @_;
226     $output= $newout;
227     $output_contents{$output}= '';
228 }
229
230
231 #-------------------- checking
232
233 use vars qw($zone $cfg $warnings);
234 $warnings= 0;
235
236 sub progress ($) {
237     return if !$verbosity;
238     print "$_[0]\n";
239 }
240
241 sub verbose ($) {
242     return if $verbosity<2;
243     print "    $_[0]\n";
244 }
245
246 sub process_zones (@) {
247     my (@zones) = @_;
248     local ($zone,$cfg);
249
250     foreach $zone (@zones) {
251         $cfg= $zone_cfg{$zone} || {
252             'style_p' => 'foreign',
253             's' => 'f',
254             'servers' => [ ],
255             };
256         progress(sprintf "%-40s %s", $zone, $$cfg{'style_p'});
257         if ($check && ($doall || $cfg->{'s'} !~ m/\?/)) {
258             eval {
259                 if ($localonly && $cfg->{'s'} =~ m/f/) {
260                     zone_warning("foreign zone specified with -l");
261                 } elsif ($cfg->{'s'} =~ m/\*/ ||
262                          ($localonly && $cfg->{'s'} !~ m/p/)) {
263                     zone_check_local();
264                 } else {
265                     zone_check_full();
266                 }
267             };
268             zone_warning("checks failed: $@") if length $@;
269         }
270         zone_output() if $install;
271     }
272     print STDERR "$quis: $warnings warnings\n" or die $!
273         if $warnings;
274 }
275
276 sub zone_warning ($) {
277     my ($w) = @_;
278     $w =~ s/\n$//;
279     $w =~ s,\n, // ,g;
280     print STDERR "$zone: warning: $w\n" or die $!;
281     $warnings++;
282 }
283
284 sub zone_warnmore ($) {
285     print STDERR "$zone:  $_[0]\n" or die $!;
286 }
287
288 use vars qw(%delgs); # $delgs{$nameserver_list} = [ $whosaidandwhy ]
289 use vars qw(%auths); # $auths{$nameserver_list} = [ $whosaidandwhy ]
290 use vars qw(%glue);  # $glue{$name}{$addr_list} = [ $whosaidandwhy ]
291 use vars qw(%soas);  # $soa{"$origin $serial"} = [ $whosaidandwhy ]
292 use vars qw(%addr_is_ok %warned_glueless %warned_mynameaddr);
293 use vars qw($delg_to_us);
294 use vars qw(@to_check); # ($addr,$whyask,$is_auth,$glueless_ok, ...)
295 use vars qw(@to_check_soa); # ($addr,$whyask, ...)
296
297 sub zone_check_full () {
298     zone_reset();
299     zone_investigate();
300     zone_consistency();
301     zone_servers_ok();
302 }
303
304 sub zone_reset() {
305     %delgs= %auths= %glue= %soas=
306         %warned_glueless= %warned_mynameaddr=
307             %addr_is_ok= ();
308     $delg_to_us= 0;
309     @to_check= @to_check_soa= ();
310 }
311
312 sub zone_investigate() {
313     my ($super_zone, @super_nsnames,
314         $super_ns, @super_ns_addrs, $s, $wa, $is_auth,
315         %nsrrset_checked, %soa_checked, $addr, $glueless_ok, $rcode);
316
317     $super_zone= $zone;
318     for (;;) {
319         debug_trace("zone $zone superzone $super_zone");
320         $super_zone =~ s/^[^.]+\.// or die "no superzone ? ($super_zone)\n";
321         ($rcode,@super_nsnames)= lookup($super_zone,'ns-','06');
322         last if !$rcode;
323     }
324     for $super_ns (@super_nsnames) {
325         $super_ns= lc $super_ns;
326         ($rcode,@super_ns_addrs)= lookup($super_ns,'a','0');
327         foreach $addr (@super_ns_addrs) {
328             push @to_check,
329                  $addr,
330                  "$super_ns, server for $super_zone",
331                  0, 0;
332         }
333     }
334     for (;;) {
335         # We do these in order so that we always do NS RRset checks on
336         # nameservers that came from other NS RRsets first; otherwise
337         # we might set nsrrset_checked due to a glueless_ok check,
338         # and then not check for gluefulness later.
339         debug_dump('@to_check @to_check_soa');
340         if (($addr,$wa,$is_auth,$glueless_ok,@to_check) = @to_check) {
341             push @to_check_soa, $addr, $wa if $is_auth;
342             next if $nsrrset_checked{$addr}++;
343             zone_check_nsrrset($addr, "[$addr] $wa",
344                                $is_auth, $glueless_ok);
345         } elsif (($addr,$wa,@to_check_soa) = @to_check_soa) {
346             next if $soa_checked{$addr}++;
347             zone_check_soa($addr,"[$addr] $wa","[$addr] NS");
348         } else {
349             last;
350         }
351     }
352 }
353
354 sub zone_check_nsrrset ($$$$) {
355     my ($uaddr,$ww, $is_auth, $glueless_ok) = @_;
356     my (@s, $s, %s2g, @glue, $glue, $delgs_or_auths, $wwn);
357     verbose("checking delegation by $ww");
358     dig(sub {
359         if ($dig_type eq 'ns' && $dig_owner eq $zone) {
360             $s2g{lc $dig_rdata} = [ ];
361         } elsif ($dig_type eq 'a' && exists $s2g{$dig_owner}) {
362             $wwn= "in glue from $ww";
363             push @to_check, $dig_rdata, "$dig_owner, $wwn", 1, 0;
364             zone_server_addr($dig_rdata,$dig_owner,$wwn,"NS [$uaddr]",0);
365             push @{ $s2g{$dig_owner} }, $dig_rdata;
366         }
367     },
368              $zone,'ns',$uaddr);
369     if (!%s2g) { zone_warning("unable to find NS RRset at $ww"); return; }
370     elsif (keys %s2g == 1) { zone_warning("only one nameserver at $ww"); }
371     @s= sort keys %s2g;
372     foreach $s (@s) {
373         @glue= @{ $s2g{$s} };
374         if (!@glue) {
375             zone_warning("glueless NS $s,".
376                          ($needglue<=1 ? " (eg)" : "").
377                          " from $ww")
378                 unless $glueless_ok || !$needglue ||
379                        ($needglue<=1 && $warned_glueless{$s}++);
380             next;
381         }
382         $glue= join ' ', sort @glue;
383         push @{ $glue{$s}{$glue} }, $ww;
384     }
385     $s= join ' ', @s;
386     $delgs_or_auths= $is_auth ? \%auths : \%delgs;
387     push @{ $delgs_or_auths->{$s} }, $ww;
388 }
389
390 sub zone_server_addr ($$$$$) {
391     my ($addr,$name,$ww,$wwq,$is_soa) = @_;
392     $addr_is_ok{$addr}= "$name ($wwq)"
393         if $is_soa || $cfg->{'s'} =~ m/u/;
394     zone_warning("configured as stealth but we [$addr]".
395                  " are published ($name $wwq)")
396         if $cfg->{'s'} =~ m/u/ && grep { $_ eq $addr } @self_addr;
397
398     my ($name_is_self, $addr_is_self);
399     $name_is_self= grep { $_ eq $name }
400         @{ $cfg->{$is_soa ? 'self_soa' : 'self_ns'} };
401     $addr_is_self= grep { $_ eq $addr }
402         @{ $cfg->{'self_addr'} };
403     if ($name_is_self && !$addr_is_self) {
404         zone_warning("our name $name with wrong address [$addr], (eg) $ww")
405             unless $warned_mynameaddr{$name}{$addr}++;
406     } elsif (!$name_is_self && $addr_is_self) {
407         zone_warning(($is_soa ? "SOA ORIGIN maps to" : "allegedly served by").
408                      " us [$addr] with wrong name $name, (eg) $ww")
409             unless $warned_mynameaddr{$name}{$addr}++;
410     }
411     $delg_to_us=1 if $name_is_self;
412 }
413
414 sub zone_check_soa ($$$) {
415     my ($uaddr,$ww,$wwq) = @_;
416     my ($lame,$origin,$got,$rcode,@soa_addrs,$soa_addr,$wwn);
417     verbose("checking service at $wwq");
418     $lame= 'dead or lame';
419     dig(sub {
420         if ($dig_type eq 'flags:') {
421             $lame= $dig_rdata =~ m/ aa / ? '' : 'lame';
422         } elsif ($dig_type eq 'soa' && $dig_owner eq $zone && !$lame) {
423             die "several SOAs ? $ww" if defined $origin;
424             $got= $dig_rdata;
425             $got =~ m/^(\S+) \d+/ or die "$got ?";
426             $origin= $1;
427         }
428     },
429              $zone,'soa',$uaddr);
430     $lame= 'broken' if !$lame && !defined $origin;
431     if ($lame) { zone_warning("$lame server $ww"); return; }
432     push @{ $soas{$got} }, $ww;
433     ($rcode,@soa_addrs)= lookup($origin,'a','0');
434     foreach $soa_addr (@soa_addrs) {
435         $wwn= "SOA ORIGIN from $ww";
436         zone_server_addr($soa_addr,$origin,$wwn,"SOA [$uaddr]",1);
437         push @to_check, $soa_addr, "$origin, $wwn";
438     }
439 }
440
441 sub zone_consistency() {
442     my ($d, $org_ser, $origin, $a, $h, $self_soa);
443     zone_consistency_set('delegations',\%delgs);
444     foreach $d (keys %delgs) { delete $auths{$d}; }
445     zone_consistency_set('zone nameserver rrset',\%auths);
446     foreach $h (keys %glue) {
447         zone_consistency_set("glue for $h", $glue{$h});
448     }
449     zone_consistency_set("SOA ORIGIN and SERIAL",\%soas);
450     $self_soa= $cfg->{'self_soa'};
451     if ($cfg->{'s'} =~ m/p/) {
452         foreach $org_ser (keys %soas) {
453             $org_ser =~ m/^(\S+) \d+$/ or die "$org_ser ?";
454             $origin= $1;
455             next if grep { $_ eq $origin } @$self_soa;
456             zone_warning("SOA ORIGIN $origin is not our name (@$self_soa),".
457                          " eg from ".($soas{$org_ser}[0]));
458         }
459     }
460 }
461
462 sub zone_servers_ok () {
463     my ($showok);
464     if (%addr_is_ok) {
465         $showok= 0;
466         foreach $a (@{ $cfg->{'servers'} }) {
467             next if exists $addr_is_ok{$a};
468             zone_warning("we slave from $a"); $showok=1;
469         }
470         if ($showok) {
471             foreach $a (keys %addr_is_ok) {
472                 zone_warnmore("permitted master [$a] $addr_is_ok{$a}");
473             }
474         }
475     }
476     if ($cfg->{'s'} =~ m/s/ && !$delg_to_us) {
477         zone_warnmore("we are supposedly published secondary,".
478                       " but not listed as a nameserver");
479     }
480 }
481
482 sub zone_consistency_set ($%) {
483     my ($msg,$set) = @_;
484     my ($d,$o);
485     if (keys(%$set) > 1) {
486         zone_warning("inconsistent $msg:");
487         foreach $d (keys %$set) {
488             foreach $o (@{ $set->{$d} }) { zone_warnmore(" $d from $o"); }
489         }
490     }
491 }
492
493 sub zone_check_local () {
494     zone_reset();
495     zone_servers_simplefind();
496     zone_servers_ok();
497 }
498
499 sub zone_servers_simplefind () {
500     my ($rcode,@nsnames,$ns,@soas,$origin);
501
502     ($rcode,@nsnames)= lookup($zone,'ns-','0');
503     foreach $ns (@nsnames) { zone_server_simple($ns,'NS',0); }
504
505     ($rcode,@soas)= lookup($zone,'soa','0');
506     die "multiple SOA RRs in set!  @soas ?" if @soas!=1;
507     $soas[0] =~ m/^(\S+)\s/ or die "SOA ? $_";
508     zone_server_simple(domain_canon($1),'SOA',1);
509 }
510
511 sub zone_server_simple ($$$) {
512     my ($name,$ww,$is_soa) = @_;
513     my ($rcode,@addrs,$addr);
514     ($rcode,@addrs)= lookup($name,'a','0');
515     foreach $addr (@addrs) { zone_server_addr($addr,$name,$ww,$ww,$is_soa); }
516 }
517
518 #-------------------- outputting
519
520 sub zone_output () {
521     $output_contents{$$cfg{'output'}}.=
522         sprintf(<<'END',
523 zone "%s" {
524     type %s;
525     file "%s";
526 };
527 END
528                 $zone,
529                 $$cfg{'s'} =~ m/p/ ? 'master' : 'slave',
530                 $$cfg{'file'});
531 }
532
533 sub output_files () {
534     my ($fn,$ofn,$mfn,$l,$dir, $maxmode,$h,@to_install);
535     
536     foreach $ofn (keys %output_contents) {
537         $fn= $ofn; $mfn= "output file $fn";
538         for (;;) {
539             if (!lstat $fn) {
540                 $! == &ENOENT or die "$quis: stat $mfn:\n $!\n";
541                 $maxmode= 0666;
542                 last;
543             } elsif (-f _) {
544                 $maxmode= (stat _)[2];
545                 last;
546             } elsif (-l _) {
547                 defined($l= readlink $fn)
548                     or die "$quis: readlink $mfn:\n $!\n";
549                 $dir= $fn =~ m,^.*/, ? $& : './';
550                 $fn= "$dir$l" unless $l =~ m,^/,;
551                 $mfn= "output file $fn (symlink target of $ofn)";
552             } else {
553                 die "$quis: output file $mfn exists but is not a file".
554                     " (or symlink to one)";
555             }
556         }
557         unlink "$fn.new" or $! == &ENOENT or
558             die "$quis: cannot clear out old .new version of $mfn:\n $!";
559         $h= new IO::File "$fn.new",'w',$maxmode
560             or die("$quis: create .new version of $mfn:\n $!");
561         print $h
562             "# generated by $quis, do not edit\n",
563             $output_contents{$ofn}
564             or die "$quis: write data to .new version of $mfn:\n $!";
565         $h->close
566             or die "$quis: close .new version of $mfn:\n $!";
567         push @to_install, $fn,$mfn;
568     }
569
570     while (($fn,$mfn, @to_install) = @to_install) {
571         rename "$fn.new",$fn
572             or die "$quis: install new version of $mfn:\n $!";
573     }
574 }
575
576 #-------------------- general utilities
577
578 sub debug_dump ($) {
579     my ($vn);
580     return unless $debug>1;
581     local $Data::Dumper::Terse=1;
582     foreach $vn (split /\s+/, $_[0]) {
583         print "$vn := ", eval "Dumper(\\$vn)";
584     }
585 }
586
587 sub debug_trace ($) {
588     return unless $debug;
589     print "D $_[0]\n";
590 }
591
592 sub lookup ($$$) {
593     my ($domain,$type,$okrcodes) = @_;
594     my ($c,$h,@result);
595     debug_trace("lookup ==> (->$okrcodes) $domain $type");
596     $h= new IO::Handle;
597
598     defined($c= open $h, "-|") or die "$quis: fork adnshost:\n $!\n";
599     if (!$c) {
600         exec 'adnshost','-Fi','+Do','+Dt','+Dc','-Cf',"-t$type",
601              '-',"$domain.";
602         die "$quis: exec adnshost:\n $!\n";
603     }
604     @result= $h->getlines();
605     $h->error and die "$quis: read from adnshost:\n $!\n";
606     chomp @result;
607     $!=0; $h->close;
608     die "$quis: lookup -t$type $domain $okrcodes failed $? $! @result\n"
609         if $! or $?>6 or index($okrcodes,$?)<0;
610     debug_trace("lookup <== $? @result");
611     return ($?,@result);
612 }
613
614
615 sub dig (&$$$) {
616     my ($eachrr, $qowner,$qtype,$qaddr) = @_;
617     # also pseudo-rr with type `flags:'
618     my ($h,$inmid,$irdata,$c);
619     local ($_);
620
621     debug_trace("dig ==> \@$qaddr $qowner $qtype");
622
623     $h= new IO::Handle;
624     defined($c= open $h, "-|") or die "$quis: fork dig:\n $!\n";
625     if (!$c) {
626         open STDERR, ">&STDOUT" or die $!;
627         exec ('dig',
628               '+nodef','+nosea','+nodebug','+norecurse',
629               "\@$qaddr",'-t',$qtype,$qowner);
630         die "$quis: exec dig:\n $!\n";
631     }
632     $inmid='';
633     for (;;) {
634         if (!defined($_= $h->getline())) {
635             $h->error() and die "$quis: read from dig:\n $!\n";
636             last;
637         }
638         chomp;
639         if (length $inmid) {
640             s/^\s+/ / or die "$inmid // $_ ?";
641             s/\;.*$//;
642             $_= $inmid.$_;
643             $inmid='';
644             s/$/ \(/ unless s/\s*\)\s*$//;
645         }
646         if (s/\s*\(\s*$//) { $inmid= $_; next; }
647         if (m/^\;\; flags\:( [-0-9a-z ]+)\;/) {
648             $dig_owner=''; $dig_type='flags:'; $dig_rdata= "$1 ";
649             debug_trace("dig  f: $dig_rdata");
650             &$eachrr;
651         } elsif (m/^\;/) {
652         } elsif (!m/\S/) {
653         } elsif (m/^([-.0-9a-z]+)\s+\d\w+\s+in\s+([a-z]+)\s+(\S.*)/i) {
654             $dig_owner=domain_canon($1); $dig_type=lc $2; $irdata=$3;
655             if ($dig_type eq 'a') {
656                 $irdata =~ m/^[.0-9]+$/ or die "$irdata ?";
657                 $dig_rdata= $&;
658             } elsif ($dig_type eq 'ns') {
659                 $irdata =~ m/^[-.0-9a-z]+$/i or die "bad nameserver $irdata ?";
660                 $dig_rdata= domain_canon($irdata);
661             } elsif ($dig_type eq 'soa') {
662                 $irdata =~ m/^([-.0-9a-z]+)\s+.*\s+(\d+)(?:\s+\d\w+){4}$/i
663                     or die "bad SOA $irdata ?";
664                 $dig_rdata= domain_canon($1).' '.$2;
665             } else {
666                 debug_trace("ignoring uknown RR type $dig_type");
667                 next;
668             }
669             debug_trace("dig  $dig_owner $dig_type $dig_rdata");
670             &$eachrr;
671         } else {
672             debug_trace("ignoring unknown dig output $_");
673         }
674     }
675     $h->close;
676     debug_trace("dig <== gave $?");
677 }
678
679 sub domain_canon ($) {
680     local ($_) = @_;
681     s/(.)\.$/$1/;
682     die "domain $_ ?" unless m/^[0-9a-z]/i;
683     return lc $_;
684 }