#!/usr/bin/perl -w

use strict;

use FindBin;		    # for FindBin::Bin

use lib $FindBin::Bin;  # for BldMachineDir (if in same dir. as this .PL file)

use Getopt::Long; # Get long options

use AntiVirus qw{Start Stop Scan WaitTillAntiVirusStarts};

####my $gTimeNow = time();

# Get the command line args

my ($gCommand,$gOutFilesDir ,$gDirsToScan,$gWaitingTime, $gRetries) = &ProcessCommandLine();

# If the command is to stop the services/processes, then do so.

if($gCommand eq 'STOP')

{    	

	Stop($gWaitingTime, $gRetries);	

}

elsif($gCommand eq 'START')

{

    Start();

}

else    # ($gCommand eq 'SCAN') NB Any unknown command has been caught by ProcessCommandLine()

{

    Scan($gOutFilesDir, $gDirsToScan);

}

####printf ("DEBUG MSG: Elapsed time = %d seconds", time() - $gTimeNow);   ####???? For debugging!

################################################################################

# ProcessCommandLine                                                           #

# Inputs: None (Gets data from @ARGV)                                          #

# Outputs: ($iCommand, $iOutFilesDir, \@iDirsToScan)                             #

# Remarks: None                                                                #

################################################################################

sub ProcessCommandLine

{

    my ($iHelp, $iCommand,$iOutFilesDir,@iDirsToScan,$iWaitingTime, $iRetries );

    unless (GetOptions('h' => \$iHelp, 'o=s' => \$iOutFilesDir, 'c=s' => \$iCommand,'w=s' => \$iWaitingTime,'r=s' => \$iRetries, 'd=s' => \@iDirsToScan))

    {

        Usage('Command Line error(s), as above.');

    }

    if (scalar @ARGV)

    {

        Usage("Redundant data on Command Line: @ARGV");

    }

    if ($iHelp)

    {

        Usage();

    }

    unless(defined($iCommand))

    {

        Usage('No command given');

    }

    # Uppercase $iCommand once and for all. Then we can check using 'eq'.

    # NB: uppercasing undef results in a defined, but empty, string!

    $iCommand = uc $iCommand;

    if($iCommand eq 'SCAN')

    {

        unless((scalar @iDirsToScan) and ($iOutFilesDir))

        {   # Make sure there are some directories to scan!

            # It is an error to ask for a scan and to not

            # supply directories, so print usage information

            Usage('With SCAN command, must specify directory(ies) and output file');

        }

        return ($iCommand, $iOutFilesDir, \@iDirsToScan);

    }

    if(($iCommand eq 'START')or ($iCommand eq 'STOP'))

        {

            if((scalar @iDirsToScan) or ($iOutFilesDir))

            {       # Can't specify directories when starting and stopping

            # the AV processes and services.

            Usage('With START/STOP command, cannot specify directories &/or output file');

        }

        # Only valid to start/stop if no directories have been given 

	   return ($iCommand, $iOutFilesDir, \@iDirsToScan,$iWaitingTime, $iRetries);

        }

    # Something else has gone wrong. So print usage.

    Usage("Unknown command $iCommand");

}

################################################################################

# Usage                                                                        #

# Inputs: Optional error message                                               #

# Outputs: Usage information for the user.                                     #

# Remarks: None                                                                #

################################################################################

sub Usage

{

    my $iErrorMsg = shift;

    if ($iErrorMsg)

    {

        print STDERR "\nERROR: $iErrorMsg.\n";

    }

    print <<USAGE_EOF;

Usage: AntiVirus.pl -c STOP [-w waitTime -r retries]

	   AntiVirus.pl -c START

       AntiVirus.pl -c SCAN -d scandir1 [-d scandir2 [-d scandir3]] -o outdir

Parameters:

 -c  command to perform, limited to "START", "STOP", "SCAN"

 -d  used ONLY in combination with "-c SCAN" to specify the directories to 

     scan. Multiple allowed.

 -o  used ONLY in combination with "-c SCAN" to specify the full name 

     of the directory in which the output file is to be written.

 -w  used ONLY in combination with "-c STOP" .If Antivirus is inactive at

     the time of STOP attempt,this argument specifies the waiting time in

     seconds before next attempt is made to check AV service status

 -r  used ONLY in combination with "-c STOP" .If Antivirus is inactive at

     the time of STOP attempt,this argument specifies the maximum number 

     of attempts to check  AntiVirus service status.

Optional Parameters:

 -h  help - print this information and exit

Examples:

 1) AntiVirus.pl -c STOP

 2) AntiVirus.pl -c START

 3) AntiVirus.pl -c SCAN -d M:\\epoc32\\ -o M:\\logs\\cedar\\

 4) AntiVirus.pl -c STOP -w 30 -r 5

Description and Remarks:

 This script controls operation of the installed anti-virus software. The 

 script stops and starts anti-virus services &/or programs. Reports of 

 missing anti-virus programs or failures to start/stop a service are 

 reported with keyword "WARNING:" or "REMARK:" prepended.

 Scanning is carried out on all the directories listed on the command line,

 each one being specified with the -d option.

 Currently only McAfee anti-virus software is supported.

USAGE_EOF

    exit 1;

}