Changeset 33 for trunk/DNSDB.pm

Timestamp:

11/10/09 17:51:55 (15 years ago)

Author:

Kris Deugau

Message:

/trunk

checkpoint, fleshing out axfr import

File:

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

trunk/DNSDB.pm

@@ -15 +15 @@
 use Exporter;
 use DBI;
+use Net::DNS;
 #use Net::SMTP;
 #use NetAddr::IP qw( Compact );
@@ -811 +812 @@
 
 
+## DNSDB::importAXFR
+# Import a domain via AXFR
+sub importAXFR {
+  my $dbh = shift;
+  my $ifrom = shift;
+  my $domain = shift;
+  my $group = shift;
+  my $status = shift || 1;
+  my $rwsoa = shift || 0;
+  my $rwns = shift || 0;
+##fixme:  add mode to delete&replace, merge+overwrite, merge new?
+
+  my $res = Net::DNS::Resolver->new;
+$res->axfr_start($domain);
+my $nrecs = 0;
+my $flags = 0;
+
+  # Allow transactions, and raise an exception on errors so we can catch it later.
+  # Use local to make sure these get "reset" properly on exiting this block
+  local $dbh->{AutoCommit} = 0;
+  local $dbh->{RaiseError} = 1;
+
+  eval {
+    # can't do this, can't nest transactions.  sigh.
+    #my ($dcode, $dmsg) = addDomain($dbh, $domain, $group, $status);
+
+##fixme:  serial
+    my $sth = $dbh->prepare("INSERT INTO domains (domain,group_id,status) VALUES (?,?,?)");
+    $sth->execute($domain,$group,$state);
+
+    # 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();
+
+    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.
+      my $sql = "INSERT INTO records (domain_id,host,type,val,ttl";
+      my $vallen = "?,?,?,?,?";
+      my $host,$val;
+
+##work
+# gnnnnnh.  going to need to (rr->string)  ->  split  ->  <localvars>
+my @vallist;
+      if ($type eq 'SOA') {
+        $host = $rr->mname.":".$rr->rname;
+        $val = $rr->refresh.":".$rr->retry.":".$rr->expire.":".$rr->minimum;
+      }
+      if ($type eq 'MX' || $type eq 'SRV') {
+        $sql .= ",distance";
+        $vallen .= ",?";
+        $sql .= ",weight,port" if $type eq 'SRV';
+        $vallen .= ",?,?" if $type eq 'SRV';
+      }
+$sth = $dbh->prepare($sql.") VALUES (".$vallen.")");
+$sth->execute(@vallist);
+
+      #print $rr->rdata."\n";
+    }
+  };
+
+  if ($@) {
+    my $msg = $@;
+    eval { $dbh->rollback; };
+    return ('FAIL',$msg);
+  } else {
+    return ('OK',"ook");
+  }
+
+  return ('WARN',"
+} # end importAXFR()
+
+
 # shut Perl up
 1;