git.rot13.org Git - pxelator/blob - lib/PXElator/dnsd.pm

   1 package dnsd;
   2
   3 use warnings;
   4 use strict;
   5
   6 use Net::DNS::Nameserver;
   7 use Net::DNS::Resolver;
   8 use Data::Dump qw/dump/;
   9 use CouchDB;
  10
  11 use server;
  12 our $debug = server::debug;
  13
  14 my $res = Net::DNS::Resolver->new(
  15 #       nameserver => [ '10.60.0.1' ],
  16         recurse => 1,
  17         debug => $debug,
  18 );
  19
  20 our $ptr_cache;
  21 sub name_ip {
  22         my ( $name, $ip ) = @_;
  23         $ptr_cache->{ join('.', reverse split(/\./, $ip)) } = $name;
  24         return $ip;
  25 }
  26
  27 sub reply_handler {
  28         my ($qname, $qclass, $qtype, $peerhost,$query,$conn) = @_;
  29         my ($rcode, @ans, @auth, @add);
  30
  31         server->refresh;
  32         $debug = server::debug;
  33
  34         CouchDB::audit( 'request', {
  35                 qname => $qname,
  36                 qclass => $qclass,
  37                 qtype => $qtype,
  38                 peerhost =>  $peerhost,
  39                 sockhost => $conn->{"sockhost"}
  40         });
  41
  42         $query->print if $debug;
  43
  44         my $local = $1     if $qname =~ m{^(.+)\.\Q$server::domain_name\E$};
  45            $local = $qname if $qname !~ m{\.};
  46
  47         my $ttl = 3600;
  48
  49         my $audit = { source => 'unknown' };
  50
  51         if ( $local ) {
  52                 warn "local[$local] $qname $qtype";
  53                 $rcode = "NOERROR";
  54                 my $rdata;
  55                 if ( $qtype eq "A" && $local eq "server" ) {
  56                         $rdata = name_ip( $local, $server::ip );
  57                         $audit->{source} = 'local';
  58                 } else {
  59                         $rcode = "NXDOMAIN";
  60                 }
  61
  62                 push @ans, Net::DNS::RR->new("$qname $ttl $qclass $qtype $rdata") if $ttl;
  63
  64         } elsif ( $qtype eq 'PTR' && $qname =~ m{^([0-9\.]*)\.in-addr\.arpa$} ) {
  65                         if ( my $rdata = $ptr_cache->{$1} ) {
  66                                 $rdata .= '.' . $server::domain_name;
  67                                 push @ans, Net::DNS::RR->new("$qname $ttl $qclass $qtype $rdata");
  68                                 $audit->{source} = 'PTR';
  69                         } else {
  70 warn "## ",dump( $ptr_cache );
  71                                 $rcode = "NXDOMAIN";
  72                         }
  73         } elsif ( my $packet = $res->query( $qname, $qtype ) ) {
  74
  75                 $audit->{source} = 'upstream';
  76                 $packet->print;
  77                 push @ans, $_ foreach $packet->answer;
  78                 $rcode = "NOERROR";
  79
  80         } else {
  81                 # not found
  82                 $rcode = "NXDOMAIN";
  83         }
  84
  85         warn "rcode: $rcode ",dump( @ans );
  86
  87         $audit->{rcode} = $rcode;
  88         $audit->{ans} = [ map {
  89                 my $data;
  90                 foreach my $n ( keys %$_ ) {
  91                         $data->{$n} = $_->{$n};
  92                 }
  93                 $data;
  94         } @ans ];
  95
  96         CouchDB::audit( 'response', $audit );
  97
  98         # mark the answer as authoritive (by setting the 'aa' flag
  99         return ($rcode, \@ans, \@auth, \@add, { aa => 1 });
 100 }
 101
 102 sub start {
 103         my $ns = Net::DNS::Nameserver->new(
 104                 LocalPort    => 53,
 105                 ReplyHandler => \&reply_handler,
 106                 Verbose      => $debug,
 107         ) || die "couldn't create nameserver object\n";
 108
 109         CouchDB::audit('start', { port => 53, domain_name => $server::domain_name });
 110         warn "DNS $server::domain_name";
 111
 112         $ns->main_loop;
 113 }
 114
 115 1;