#!/usr/local/bin/perl

use strict;
use warnings;

use IO::Socket::INET;
use threads;
use threads::shared;
use Net::DNS;
use Thread::Semaphore;

my %resolved : shared;
my %to_resolve;
my @list : shared;
my $count = 0;
my $count_array : shared;
my @threads;

my $semaphore = new Thread::Semaphore;

use constant THREAD_COUNT => 20;

open ( OUTPUT, ">cache.txt" ) or die $!;

#no error checking, just assumes that all the files listed are correct.
foreach my $file ( @ARGV )
{
  print "starting with file $file\n";

  open ( INPUT, $file ) or die $!;
  print "reading in IP addresses...\n";
  my $start_read = time();
  $count = 0;
  while (my $line =  <INPUT> )
  {
    $count++;
    my ( $address, @stuff ) = split(" ", $line);

    $to_resolve{$address} = 1;

  if ( $count % 32768 == 0 ) { print "$count\n" };
  }

  print "$count lines read in ", time() - $start_read, "s\n";
  my $start_resolve = time();
  @list = keys ( %to_resolve );
  print "$#list ips to resolve\n";

  $count_array=0;

  for ( my $loop = 0; ($loop < THREAD_COUNT and $loop < $#list / 20); $loop ++ )
  {
    print "starting thread $loop...";
    $threads[$loop] = threads -> new ( \&run_resolver, $loop, $start_resolve );
    print "done\n";
  }

  for ( my $loop = 0; ( $loop < THREAD_COUNT and $loop < $#list / 20 ); $loop++ )
  {
    print "waiting for thread $loop...";
    $threads[$loop] -> join;
    print "done.\n";
  }
  print "resolve of $#list ip addresses complete in ", time() - $start_resolve,"s\n";
  print "writing file...\n";

  my $start_write = time();

  foreach my $address ( keys ( %resolved ) )
  {
    print OUTPUT time(), " ", $address, " ", $resolved{$address},"\n";
  }

  print "output file written in ", time() - $start_write, "s\n";


  close ( INPUT );
}#foreach

close ( OUTPUT );

sub run_resolver
{
  my ( $number, $start_time ) = @_;
  my $resolver = new Net::DNS::Resolver;
  my $query;
  my $addr;
  my $finished = 0;
  my $count = 0;
  sleep 1;
  
  #while ( $addr = (keys ( %to_resolve ) )[0] )
  while ( not $finished )
  {
    $count++;
    #print "$number waiting for lock\n";
    $semaphore -> down;
    #print "$number got lock\n";
    unless ( $addr = $list[$count_array++] )  
        { $finished = 1; $semaphore -> up; next;}
    #print "$number checking for ip validity..\n";
    unless ( $addr =~ m/\d+\.\d+\.\d+\.\d+/ )
    {
      print "\'$addr\' is invalid, releasing lock and skipping.\n";
      $semaphore -> up;
      next;
    }
    #print "$number got addr of $addr $count_array / $#list\n";
    $semaphore -> up;
    #print "$number releasing lock\n";
    #print "$number finding $addr...\n";
    $query = $resolver -> search ( $addr );
    if ( $query ) 
    {
      $resolved{$addr} = $addr;
      foreach my $RR ( $query -> answer ) 
      {
        next unless $RR -> type eq "PTR";
        $resolved{$addr} = $RR -> rdatastr;
      }
   }
   else
   {
     $resolved{$addr} = $addr;
   }

   #if ( $resolved{$addr} ) 
   #   { print "-> $number $addr = $resolved{$addr}, $count ($count_array / $#list )\n";}

   if ( $count_array % 1024 == 0 ) { print "$number: ", time() - $start_time , "s :$count_array / $#list resolved.\n"; }

 } # while

}