return iso path

[pxelator] / lib / PXElator / dnsd.pm

diff --git a/lib/PXElator/dnsd.pm b/lib/PXElator/dnsd.pm
index b82664f..2a68c71 100644 (file)
--- a/lib/PXElator/dnsd.pm
+++ b/lib/PXElator/dnsd.pm
@@ -6,9 +6,10 @@ use strict;
 use Net::DNS::Nameserver;
 use Net::DNS::Resolver;
 use Data::Dump qw/dump/;
-use CouchDB;
+use store;
 
 use server;
+use client;
 our $debug = server::debug;
 
 my $res = Net::DNS::Resolver->new(
@@ -17,31 +18,40 @@ my $res = Net::DNS::Resolver->new(
        debug => $debug,
 );
 
-our $ptr_cache;
+our ( $ptr_cache, $a_cache );
 sub name_ip {
        my ( $name, $ip ) = @_;
        $ptr_cache->{ join('.', reverse split(/\./, $ip)) } = $name;
+       $a_cache->{$name} = $ip;
        return $ip;
 }
 
+name_ip 'server' => $server::ip;
+
+foreach my $ip ( client::all_ips ) {
+       if ( my $name = client::conf( $ip => 'hostname' ) ) {
+               name_ip $name => $ip;
+       }
+}
+
 sub reply_handler {
        my ($qname, $qclass, $qtype, $peerhost,$query,$conn) = @_;
        my ($rcode, @ans, @auth, @add);
 
-       server->refresh;
        $debug = server::debug;
 
-       CouchDB::audit( 'request', {
+       my $audit = {
                qname => $qname,
                qclass => $qclass,
                qtype => $qtype,
                peerhost =>  $peerhost,
-               sockhost => $conn->{"sockhost"}
-       });
+               sockhost => $conn->{"sockhost"},
+               source => 'unknown',
+       };
 
        $query->print if $debug;
 
-       my $local = $1     if $qname =~ m{^(.+)\.\Q$server::domain_name\E$};
+       my $local = $1     if $qname =~ m{^(.+)\.\Q$server::domain\E$};
           $local = $qname if $qname !~ m{\.};
 
        my $ttl = 3600;
@@ -50,27 +60,32 @@ sub reply_handler {
                warn "local[$local] $qname $qtype";
                $rcode = "NOERROR";
                my $rdata;
-               if ( $qtype eq "A" && $local eq "server" ) {
-                       $rdata = name_ip( $local, $server::ip );
-                       CouchDB::audit('local', $rdata);
+               if ( $qtype eq "A" ) {
+                       if ( $rdata = $a_cache->{$local} ) {
+                               $audit->{source} = 'local';
+                       } else {
+                               $rcode = "NXDOMAIN";
+warn "## no $local in ",dump( $a_cache );
+                       }
+               } elsif ( $qtype eq 'PTR' ) {
+                       $qname =~ s{\.in-addr\.arpa$}{} || warn "W: can't strip suffix from $qtype $qname";
+                       if ( my $rdata = $ptr_cache->{$qname} ) {
+                               $rdata .= '.' . $server::domain;
+                               push @ans, Net::DNS::RR->new("$qname $ttl $qclass $qtype $rdata");
+                               $audit->{source} = 'PTR';
+                       } else {
+warn "## no $qname in ",dump( $ptr_cache );
+                               $rcode = "NXDOMAIN";
+                       }
                } else {
-                       $rcode = "NXDOMAIN";
+                       $audit->{warn} = "qtype $qtype not supported";
                }
 
                push @ans, Net::DNS::RR->new("$qname $ttl $qclass $qtype $rdata") if $ttl;
 
-       } elsif ( $qtype eq 'PTR' && $qname =~ m{^([0-9\.]*)\.in-addr\.arpa$} ) {
-                       if ( my $rdata = $ptr_cache->{$1} ) {
-                               $rdata .= '.' . $server::domain_name;
-                               push @ans, Net::DNS::RR->new("$qname $ttl $qclass $qtype $rdata");
-                               CouchDB::audit('ptr', $rdata );
-                       } else {
-warn "## ",dump( $ptr_cache );
-                               $rcode = "NXDOMAIN";
-                       }
        } elsif ( my $packet = $res->query( $qname, $qtype ) ) {
 
-               CouchDB::audit( 'upstream', $packet );
+               $audit->{source} = 'upstream';
                $packet->print;
                push @ans, $_ foreach $packet->answer;
                $rcode = "NOERROR";
@@ -82,7 +97,16 @@ warn "## ",dump( $ptr_cache );
 
        warn "rcode: $rcode ",dump( @ans );
 
-       CouchDB::audit('response', { rcode => $rcode, ans => [ @ans ], auth => [ @auth ], add => [ @add ] });
+       $audit->{rcode} = $rcode;
+       $audit->{ans} = [ map {
+               my $data;
+               foreach my $n ( keys %$_ ) {
+                       $data->{$n} = $_->{$n};
+               }
+               $data;
+       } @ans ];
+
+       store::audit( 'response', $audit );
 
        # mark the answer as authoritive (by setting the 'aa' flag
        return ($rcode, \@ans, \@auth, \@add, { aa => 1 });
@@ -91,12 +115,15 @@ warn "## ",dump( $ptr_cache );
 sub start {
        my $ns = Net::DNS::Nameserver->new(
                LocalPort    => 53,
-               ReplyHandler => \&reply_handler,
+               ReplyHandler => sub {
+                       server->refresh;
+                       reply_handler(@_);
+               },
                Verbose      => $debug,
        ) || die "couldn't create nameserver object\n";
 
-       CouchDB::audit('start', { listen => { port => 53, domain_name => $server::domain_name } });
-       warn "DNS $server::domain_name";
+       store::audit('start', { port => 53, domain => $server::domain });
+       warn "DNS $server::domain";
 
        $ns->main_loop;
 }