#! /usr/bin/perl

###########################################################################
# This script executes a series of ping transfers with varying
# test packet sizes
###########################################################################
#
# Author: Yee-Ting Li, UCL HEP
# 
# Change History:
#  19-07-02: initial release
#  31-07-02: added argument mask parsing
#
###########################################################################

require 'ping-cook.pl';

$DEBUG = 1;

if ( $#ARGV < 0 ) {
    &printHelp;
}

# parse through the args
for ( $i=0; $i<$#ARGV+1; $i++ ) {

    if( $ARGV[$i] =~ /^-(.*)/ ) {

	$switch = $1;

	# connection host
	if ( $switch eq 'd' ) {
	    $destip = $ARGV[$i+1];
	}
	elsif ( $switch eq 'n' ) {
#	    @pktss = split( /,/, $ARGV[$i+1] );
	    $pkts = $ARGV[$i+1];
	}
	elsif ( $switch eq 's' ) {
	    @sizes = split( /,/, &grab_input( $ARGV[$i+1]) );
#	    $size = $ARGV[$i+1] ;
	}
	elsif ( $switch eq 'f' ) {
	    $outfile = $ARGV[$i+1];
	}
	elsif ( $switch eq 'i' ) {
#	    @intervals = split( /,/, $ARGV[$i+1] );
	    $interval = $ARGV[$i+1];
	}
	elsif ( $switch eq 'c' ) {
	    $primeCacheNumber = $ARGV[$i+1];
	}
	elsif ( $switch eq 'h' ) {
	    &printHelp;
	}


    }


}


# go through an put default values
if ( $destip eq '' ) { &printHelp; }
if ( $pkts eq '' ) { $pkts = 10; }
if ( $interval eq '' ) { $interval = 500; }
if ( $primeCacheNumber eq '' ) { $primeCacheNumber = 0; }
if ( $#sizes eq -1  ) { 
    @sizes = ( '64', '128', '192', '256', '512', '1024' , '1500', '2000', '3000' ); 
}


if ( $DEBUG ) { print "\ndest: $destip, packet number: $pkts, pkt interval: $interval\npacket sizes: @sizes\n" ; }




$sourceip = `hostname`; chomp($sourceip);
$PING = `which ping`; chomp( $PING );

### Setup to go

# Get the time and date
my ($sec, $minutes, $hour, $mday, $month, $year, $wday, $yday, $isdst) = gmtime();

# create a directory based on dd-mm-yy in working directory
$dir = sprintf "%2.2d-%2.2d-%4d", $mday, $month+1, $year+1900;
`mkdir $dir \&> /dev/null`;

$count = 1;
$total = $#sizes + 1;

@files;

print STDERR '=' x 72 . "\n";

   $interval = $interval / 1000;

foreach my $size (@sizes) {

   print STDERR "Conducting test size of ping packet $size ($count/$total)\n";
   
    # get time again
    ($sec, $minutes, $hour, $mday, $month, $year, $wday, $yday, $isdst) = gmtime();

    # create a filename to store output
    $filename = sprintf "$dir\/ping_%4d%2.2d%2.2d-%2.2d%2.2d%2.2d_" . "$sourceip" . "_" . "$destip" . "_i" . $interval . '_s' . $size . "_n" . "$pkts" .".log", $year+1900, $month+1, $mday, $hour, $minutes, $sec;


    # add filename to array
    push ( @files, $filename );
    $ping_cmd = "echo \"interval: $interval\" \&> $filename;$PING $destip -s $size -c $primeCacheNumber -i $interval \&>/dev/null;$PING $destip -s $size -c $pkts -i $interval >> $filename";

#    print $ping_cmd;

    # execute iperf
    system( $ping_cmd );

    $count++ ;
}

print STDERR '=' x 72 . "\n";
print STDERR "Tests complete!\n";
print STDERR "\nCooking...\n";
print STDERR '=' x 72 . "\n";



if ( $outfile eq '' ) {
    &pingCook( @files );
}
else {
    
    open(SAVEOUT, ">&STDOUT");
    open (STDOUT,">$outfile") || warn "SSH_Open: Could not redirect STDOUT to $filename: $!\n";
    select(STDOUT); $| = 1;	
    &pingCook( @files );   
    # redirect stdout to stdout
    open(STDOUT, ">&SAVEOUT");
    open ( OUT, "<$outfile" );
    while ( <OUT> ) {
	print $_;
    }

}
print STDERR '=' x 72 . "\n";

print STDERR "\nComplete!\n";


sub printHelp{
    print "Usage: do_packetsize.pl -d <sink> [-c <prime cache number>] [-i <interval>(msec)] [-n <number of packets>] [-s <s1,s2,s3...>(bytes)] \n";
    exit;
}