Changeset 35 for trunk/DNSDB.pm

Timestamp:

11/13/09 18:05:34 (15 years ago)

Author:

Kris Deugau

Message:

/trunk

checkpoint - my Net::DNS vs DBD::Pg weirdness went away

• clean up return-message handling for AXFR
• check AXFR host to remove taint flag so IO::Socket doesn't whine
• rearrange statement handles in DNSDB::importAXFR
• add some flags to fail on no records at all, no SOA, or no NS records
• tweak template to put error in a row at the top of the import form vs an uncontrolled div

File:

• trunk/DNSDB.pm (modified) (9 diffs)

trunk/DNSDB.pm

@@ -816 +816 @@
 sub importAXFR {
   my $dbh = shift;
-  my $ifrom = shift;
+  my $ifrom_in = shift;
   my $domain = shift;
   my $group = shift;
@@ -824 +824 @@
 ##fixme:  add mode to delete&replace, merge+overwrite, merge new?
 
-
 my $nrecs = 0;
-my $flags = 0;
-my $warnmsg;
+my $soaflag = 0;
+my $nsflag = 0;
+my $warnmsg = '';
+my $ifrom;
+
+  # choke on possible bad setting in ifrom
+  # IPv4 and v6!
+  ($ifrom) = ($ifrom_in =~ /^([0-9a-f\:.]+|[0-9a-z_.-]+)$/i);
+  return ('FAIL', "Bad AXFR source host $ifrom")
+        unless ($ifrom) = ($ifrom_in =~ /^([0-9a-f\:.]+|[0-9a-z_.-]+)$/i);
 
   # Allow transactions, and raise an exception on errors so we can catch it later.
@@ -834 +841 @@
   local $dbh->{RaiseError} = 1;
 
+  my $sth_domin = $dbh->prepare("INSERT INTO domains (domain,group_id,status) VALUES (?,?,?)");
+  my $sth_getid = $dbh->prepare("SELECT domain_id FROM domains WHERE domain=?");
   my $dom_id;
+
+# quick check to start to see if we've already got one
+  $sth_getid->execute($domain);
+  ($dom_id) = $sth_getid->fetchrow_array;
+
+  return ('FAIL', "Domain already exists") if $dom_id;
 
   eval {
     # can't do this, can't nest transactions.  sigh.
-    #my ($dcode, $dmsg) = addDomain($dbh, $domain, $group, $status);
+    #my ($dcode, $dmsg) = addDomain(dbh, domain, group, status);
 
 ##fixme:  serial
-    my $sth = $dbh->prepare("INSERT INTO domains (domain,group_id,status) VALUES (?,?,?)");
-$warnmsg = "trying lokido... (".$dbh->{AutoCommit}."), (".$dbh->{RaiseError}.")" if $domain eq 'waslokido.com';
-    $sth->execute($domain,$group,$status);
+    $sth_domin->execute($domain,$group,$status);
+
+## bizarre DBI<->Net::DNS interaction bug:
+## sometimes a zone will cause an immediate commit-and-exit (sort of) of the while()
 
     # get domain id so we can do the records
-    $sth = $dbh->prepare("select domain_id from domains where domain='$domain'");
-    $sth->execute;
-    ($dom_id) = $sth->fetchrow_array();
-$warnmsg .= " [domid $dom_id]";
+    $sth_getid->execute($domain);
+    ($dom_id) = $sth_getid->fetchrow_array();
 
     my $res = Net::DNS::Resolver->new;
-    unless ($res->axfr_start($domain)) {
-      $dbh->rollback;
-      die "Couldn't begin AXFR\n";
-    }
-
-#die "just started AXFR\n";
-
-    while (my $rr = $res->axfr_next) {
-$warnmsg = $rr->string;
+    $res->nameservers($ifrom);
+    $res->axfr_start($domain)
+        or die "Couldn't begin AXFR\n";
+
+#my $foobar = $res->axfr_next() if $domain =~ /loki/;
+#die "outside the loop with ".$foobar->type."\n" if $domain =~ /loki/;
+
+    while (my $rr = $res->axfr_next()) {
       my $type = $rr->type;
-# nasty big ugly case-like thing here, since we have to do *some* different
-# processing depending on the record.  le sigh.
+
+#die "first record! $type\n" if $domain =~ /lok/;
+
       my $sql = "INSERT INTO records (domain_id,host,type,ttl,val";
       my $vallen = "?,?,?,?,?";
@@ -870 +884 @@
 #      my $val;
 
+$soaflag = 1 if $type eq 'SOA';
+$nsflag = 1 if $type eq 'NS';
+
 ##work
-# gnnnnnh.  going to need to (rr->string)  ->  split  ->  <localvars>  ?
-my @vallist = ($dom_id, $rr->name, $reverse_typemap{$type}, $rr->ttl);
+      my @vallist = ($dom_id, $rr->name, $reverse_typemap{$type}, $rr->ttl);
 
 # "Primary" types:
 # A, NS, CNAME, SOA, PTR(warn in forward), MX, TXT, AAAA, SRV, A6(ob), SPF
 # maybe KEY
+
+# nasty big ugly case-like thing here, since we have to do *some* different
+# processing depending on the record.  le sigh.
 
       if ($type eq 'A') {
@@ -882 +901 @@
       } elsif ($type eq 'NS') {
         push @vallist, $rr->nsdname;
+        $nsflag = 1;
       } elsif ($type eq 'CNAME') {
         push @vallist, $rr->cname;
@@ -887 +907 @@
         $vallist[1] = $rr->mname.":".$rr->rname;
         push @vallist, ($rr->refresh.":".$rr->retry.":".$rr->expire.":".$rr->minimum);
+        $soaflag = 1;
       } elsif ($type eq 'PTR') {
         # hmm.  PTR records should not be in forward zones.
@@ -910 +931 @@
         push @vallist, $rr->port;
       } elsif ($type eq 'KEY') {
+        # we don't actually know what to do with these...
         push @vallist, ($rr->flags.":".$rr->protocol.":".$rr->algorithm.":".$rr->key.":".$rr->keytag.":".$rr->privatekeyname);
+      } else {
+        # Finding a different record type is not fatal.... just problematic.
+        $warnmsg .= "Unusual record ".$rr->name." ($type) found\n";
       }
-
-$warnmsg = $rr->string;
-$dbh->rollback if $domain eq 'waslokido.com';
-die "first record: ".$rr->string."\n" if $domain eq 'waslokido.com';
 
 # BIND supports:
@@ -934 +955 @@
       }
 
-      $sth = $dbh->prepare($sql.") VALUES (".$vallen.")");
-      $sth->execute(@vallist) or die "failed to insert ".$rr->string.": ".$sth->errstr."\n" if $sth->err;
+      my $sth_rr = $dbh->prepare($sql.") VALUES (".$vallen.")");
+      $sth_rr->execute(@vallist) or die "failed to insert ".$rr->string.": ".$sth_rr->errstr."\n" if $sth_rr->err;
 
       if ($type eq 'SOA') {
       }
 
+#die "die.die.die!\n";
+
+      $nrecs++;
+
+    } # while axfr_next
+
+    die "No records found;  either $ifrom is not authoritative or doesn't allow transfers\n" if !$nrecs;
+    die "Bad zone:  No SOA record!\n" if !$soaflag;
+    die "Bad zone:  No NS records!\n" if !$nsflag;
+
 $dbh->rollback;
-#die "die.die.die!\n";
-
-      #print $rr->rdata."\n";
-    }
+#    $dbh->commit;
+
+#    die "Bad zone.  BAD zone!  No cookie!\n";
+#die "don't insert me!  I'm nasty, I am!\n";
   };
 
@@ -950 +981 @@
     my $msg = $@;
     eval { $dbh->rollback; };
+#    $dbh->do("delete from domains where domain='$domain'");
     return ('FAIL',$msg." $warnmsg");
   } else {
-    return ('WARN', "OOOK!  Ooooook.  Ook. ($warnmsg)") if $domain eq 'deepnet.cx';
-    return ('WARN', "Funky Things Happened: $warnmsg") if $domain eq 'waslokido.com';
+#    $dbh->do("delete from domains where domain='$domain'");
+    return ('WARN', $warnmsg) if $warnmsg;
     return ('OK',"ook");
   }