File Coverage

blib/lib/Net/SinFP3.pm

Criterion	Covered	Total	%
statement	33	154	21.4
branch	0	40	0.0
condition	0	15	0.0
subroutine	11	16	68.7
pod	2	2	100.0
total	46	227	20.2

#
# $Id: SinFP3.pm 2234 2014-04-08 13:05:14Z gomor $
#
package Net::SinFP3;
use strict;
use warnings;

our $VERSION = '1.22';

use base qw(Class::Gomor::Array DynaLoader);
our @AS = qw(
   global
);
our @AA = qw(
   db
   mode
   search
   input
   output
);
__PACKAGE__->cgBuildIndices;
__PACKAGE__->cgBuildAccessorsScalar(\@AS);
__PACKAGE__->cgBuildAccessorsArray (\@AA);

our %EXPORT_TAGS = (
   matchType => [qw(
      NS_MATCH_TYPE_S1S2S3
      NS_MATCH_TYPE_S1S2
      NS_MATCH_TYPE_S2
   )],
   matchMask => [qw(
      NS_MATCH_MASK_HEURISTIC0
      NS_MATCH_MASK_HEURISTIC1
      NS_MATCH_MASK_HEURISTIC2
   )],
   functions => [qw(
      sinfp3_tcp_synscan
      sinfp3_geterror
   )],
);

our @EXPORT_OK = (
   @{$EXPORT_TAGS{matchType}},
   @{$EXPORT_TAGS{matchMask}},
   @{$EXPORT_TAGS{functions}},
);

__PACKAGE__->bootstrap($VERSION);

use constant NS_MATCH_TYPE_S1S2S3 => 'S1S2S3';
use constant NS_MATCH_TYPE_S1S2   => 'S1S2';
use constant NS_MATCH_TYPE_S2     => 'S2';

use constant NS_MATCH_MASK_HEURISTIC0 => 'BH0FH0WH0OH0MH0SH0LH0';
use constant NS_MATCH_MASK_HEURISTIC1 => 'BH1FH1WH1OH1MH1SH1LH1';
use constant NS_MATCH_MASK_HEURISTIC2 => 'BH2FH2WH2OH2MH2SH2LH2';

use Net::SinFP3::Worker qw(:consts);

sub new {
   my $class = shift;
   my %param = @_;

   # Sets unbuffered STDOUT
   $|++;

   if (!exists($param{output})
   ||  !exists($param{input})
   ||  !exists($param{mode})
   ||  !exists($param{search})
   ||  !exists($param{global})
   ||  !exists($param{db})) {
      die("[-] ".__PACKAGE__.": You must provide all of the following ".
          "attributes: output, input, mode, search, db, global\n");
   }

   my $self = $class->SUPER::new(
      db     => [],
      input  => [],
      mode   => [],
      search => [],
      output => [],
      @_,
   );

   my $log = $self->global->log;

   {
      no strict 'vars';
      for my $var ('output', 'input', 'db', 'mode', 'search') {
         my $idx = '$__'.$var;
         my $ref = ref($self->[eval($idx)]);
         if ($ref !~ /^ARRAY$/) {
            $log->fatal("$var attribute must be an ARRAYREF and it is [$ref]");
         }
      }
   }

   return $self;
}

sub _do {
   my $self = shift;

   my $global = $self->global;
   my $log = $global->log;
   my $input = $global->input;
   my $next = $global->next;

   $log->info("Starting of job with Next ".$next->print);

   my @db = $self->db;
   my @mode = $self->mode;
   my @search = $self->search;
   my @output = $self->output;

   $input->postRun or return;

   for my $db (@db) {
      $log->verbose("Starting of DB [".ref($db)."]");
      $global->db($db);
      $db->init or $log->fatal("Unable to init [".ref($db)."] module");
      $db->run or next;
      $log->verbose("Running of DB [".ref($db)."]: Done");
      for my $mode (@mode) {
         $global->mode($mode);

         $log->verbose(
            "Running with Next: ".$next->print." with type [".ref($next)."]"
         );
         $log->verbose("Starting of Mode [".ref($mode)."]");
         $mode->init or $log->fatal("Unable to init [".ref($mode)."] module");
         $mode->run or next;
         $log->verbose("Running of Mode [".ref($mode)."]: Done");

         for my $search (@search) {
            $global->search($search);

            $log->verbose("Starting of Search [".ref($search)."]");
            $search->init or $log->fatal("Unable to init [".ref($search).
                                         "] module");
            my $result = $search->run or next;
            $global->result($result);
            $log->verbose("Running of Search [".ref($search)."]: Done");

            $mode->postSearch;

            for my $output (@output) {
               $global->output($output);

               $log->verbose("Starting of Output [".ref($output)."]");
               $output->firstInit;
               $output->init or $log->fatal("Unable to init [".ref($output).
                                            "] module");
               $output->run or next;
               $output->post;
               $log->verbose("Running of Output [".ref($output)."]: Done");
            }
            $search->post;
         }
         $mode->post;
      }
      # To have persistent $dbh, we MUST post() in main process
      #$db->post;
   }

   return 1;
}

sub _getWorkerModel {
   my $self = shift;

   my $global = $self->global;
   my $log    = $global->log;

   my $model;
   if ($global->worker =~ /fork/i) {
      eval "use Net::SinFP3::Worker::Fork";
      if ($@) {
         chomp($@);
         $log->fatal("Unable to use worker model Fork: error [$@]");
      }
      $model = 'Net::SinFP3::Worker::Fork';
   }
   elsif ($global->worker =~ /thread/i) {
      eval "use Net::SinFP3::Worker::Thread";
      if ($@) {
         chomp($@);
         $log->fatal("Unable to use worker model Thread: error [$@]");
      }
      $model = 'Net::SinFP3::Worker::Thread';
   }

   return $model;
}

sub run {
   my $self = shift;

   my $global = $self->global;
   my $log = $global->log;
   my @input = $self->input;
   my @output = $self->output;

   # Beware, recursive loop
   $log->global($global);

   my $worker = $self->_getWorkerModel->new(
      global => $global,
   );

   $log->info("Loaded Input:  ".join(', ', map { ref($_) } $self->input));
   $log->info("Loaded DB:     ".join(', ', map { ref($_) } $self->db));
   $log->info("Loaded Mode:   ".join(', ', map { ref($_) } $self->mode));
   $log->info("Loaded Search: ".join(', ', map { ref($_) } $self->search));
   $log->info("Loaded Output: ".join(', ', map { ref($_) } $self->output));

   for my $output (@output) {
      $output->preInit;
   }

   my $job = 0;
   for my $input (@input) {
      $log->info("Starting of Input [".ref($input)."]");
      $input->init or $log->fatal("Unable to init [".ref($input)."] module");

      $global->input($input);

      while (my $next = $input->run) {
         last unless defined($next);

         my @next = (ref($next) =~ /^ARRAY$/) ? @$next : ( $next );
         for my $next (@next) {
            $global->job(++$job);
            $global->next($next);

            $worker->init(
               callback => sub {
                  $self->_do;
               },
            ) or $log->fatal("Unable to init [".ref($worker)."] module");

            # We are just before fork()ing or thread()ing.
            # Now, all data will be copied to the new process.
            my $r = $worker->run;
            if ($r == NS_WORKER_SUCCESS) {
               $input->postFork;
               next;
            }

            # Father process will skip that part
            $log->verbose("Running of job with Next ".$next->print.": Done");

            $worker->post;
         }
      }
      $global->job(0);
      $input->post;
      $log->verbose("Running of Input [".ref($input)."]: Done");
   }

   $worker->clean;

   for my $db ($self->db) {
      $db->post;
   }

   for my $output (@output) {
      $output->lastPost;
   }

   $log->info("Done: operation successful");

   return 1;
}

1;

__END__