#!/usr/bin/perl
#
# Netprog spring 2004
#
# test client used for grading proxy web servers
# give this perl script the address of the server as a hostname and port number
#
# This client reads a script that contains commands used to test
# project 1 submissions. Supported commands are:
#
# CONNECT        : establish a connection to the proxy server
#                  (forces a close as well)
# SEND string    : send the specified string
# CLOSE          : close the TCP socket the proxy
# READLINE       : read one line from proxy and toss away
# READALL        : read from proxy until EOF (tossing away)
# SAVEALL file   : read from proxy until EOF and save in file
# SAVECONTENT    : skip headers and save content in file
# ECHOLINE       : read one line from proxy and print
# ECHOALL        : read from proxy until EOF and print
# EXPECT file    : read from proxy until EOF and compare to a file
# EXPECTCONTENT file  : read from proxy, until EOF and compare to a file
#                        tosses headers before comparing
# PRINT string   : print the string to STDOUT
# EVAL string   : eval the perl commandprint the string to STDOUT
# QUEUE         : add string to send queue
# SENDQUEUE     : send entire send queue

require 5.002;
use strict;
#use sigtrap;
use Socket;

$SIG{'PIPE'} = 'IGNORE';

# location of documents directory
#my $docdir = "/cs/hollingd/netprog/submissions/hw2/testcode/docs";
#my $docdir = "/Users/hollingd/Desktop/Netprog/proj2/testcode/testcode/docs";
my $docdir = "/cs/hollingd/netprog/hw2/testcode/docs";
my %commands = (
	     "close",1,
	     "connect",1,
	     "echoall",1,
	     "echoline",1,
	     "expect",1,
	     "expectcontent",1,
	     "expecterror",1,
	     "print",1,
	     "queue",1,
	     "readall",1,
	     "readline",1,
	     "send",1,
             "sendqueue",1,		
	     "saveall",1,
             "savecontent",1,
             "eval",1,
	     "sleep",1);

select STDOUT; $|=1;

# send queue initially empty
my $queue = "";
# collect command line parameters

my $hostname = shift || die "No host specified!";
my $port = shift || die "No port specified!";
my $script = shift || die "No script specified";
my($command,@tmp,$perlcom);

my @sc;
my $sname;
@sc = split("\/",$script);
$sname = pop @sc;
printf("Script is $sname\n");

open(SCRIPT,"$script") || die $!;
# process commands from the script file
while (<SCRIPT>) {
    chomp; ($command,@tmp) = split;
    $command =~ tr/A-Z/a-z/; 
    if ($command =~/^\s*$/) {
	# nothing - skip it
    } elsif ($command =~ /^#.*$/) {
	printf "$_\n";
    } elsif (! $commands{$command}) {
	print("Error - unknown command: $command ($_)\n");
    } else {
	$perlcom = $command . "_handler \"$_\"";
#	printf("Trying $perlcom\n");
	eval $perlcom;
    }
}
close(SCRIPT);

sub close_handler {
    close(PROX);
}

sub sleep_handler {
    local($_)=$_[0];
    s/SLEEP//;
    sleep $_;
}

sub eval_handler {
    local($_)=$_[0];
    s/EVAL//;
#    printf("EVAL $_\n");
    eval $_;
}

sub print_handler {
    local($_)=$_[0];
    s/PRINT\s+//;
    print "$_\n";
#    eval $_;
}



sub connect_handler {
#    printf("CONNECT\n");
    np_connect($hostname,$port);
}


sub echoall_handler {
    while (<PROX>) {
	printf("$_");
    }
    close(PROX);
}

sub readall_handler {
    while (<PROX>) {
    }
    close(PROX);
}


sub echoline_handler {
    $_=<PROX>;
    printf("$_");
}

sub readline_handler {
    $_=<PROX>;
}



# 
sub saveall_handler {
    local($_) = $_[0];
    my($cmd,$file) = split;
    open(F,">$file") || die $!;
    my($len,$buf);

    while (<PROX>) {
	print F $_;
    }
    close(F);
    close(PROX);
}


# savecontent assumes we are seeing the entire reply
# and tosses headers
sub savecontent_handler {
    local($_) = $_[0];
    my($save) = $_;

    $_=<PROX>;           # the reply
    while (($_=<PROX>) && (! /^\r\n/)) {
#	print;
    }
    saveall_handler($save);
}

sub expect_handler {
    local($_) = $_[0];    
    my($cmd,$file) = split;
    my($line);
    my($same)=0;
    if (! ($file =~ /^\//)) {
	$file = "$docdir/$file";
    }
    if (! -f $file) {
	printf("Error - can't find the file: $file\n");
    } else {
	printf("CHECKING $file\n");
	open(F,"$file") || die $!;
        open (W, "> page") || die $!;
	# while (($line = <F>) && ($same<10)) {
        while ($line = <F>) {
	    $_=<PROX>;
#	    print "CONTENT: $_\n";
#	    print "FILE: $line\n";
            print W $_;
	    if ( $line ne $_ ) {
		$same++;
	    }
	}
	close(F);
        close(W);
	close(PROX);
	if ($same==0) {
	    printf("== Proj 2 grading script $sname PASSED.\n");
	} else {
	    printf("== Proj 2 grading script $sname FAILED.\n");
	}
    }
}

# expectcontent just reads the headers and
# then calls the expect handler
sub expectcontent_handler {
    my($save) = $_[0];
    $_=<PROX>;    # get The reply line
    while (($_=<PROX>) && (! /^\r\n/)) {
#	printf("Skipping header $_");
    }
    expect_handler( $save );
}


# expectnocontent is supposed to read no file
sub expecterror_handler {
    my($save) = $_[0];

    my $status =<PROX>;    # get The reply line
    
    #print "$status\n";    
    #while (<PROX>) {
    #  print "$_\n";
    #}
    #return;        
    
    if (! $status) {
#      print ("== Proj 2 grading script $sname passed partially.\n");
    }
    else {
      chomp ($status);
      my @tmp = split (/\s+/, $status);
      my $errCode = $tmp[1];
      
      if ($errCode =~ /[45][0-9][0-9]/) {
        printf("== Proj 2 grading script $sname PASSED.\n");
        return;
      }
    }
    printf("== Proj 2 grading script $sname FAILED.\n");
}




sub send_handler {
    local($_) = $_[0];
    my($cmd,$file) = split;
    s/^$cmd[ \t]+(.*)$/\1/;
#    printf("SENDING $cmd <$_>\n");
    printf PROX "$_";
}


# add string to request queue handler
sub queue_handler {
    local($_) = $_[0];
    my($cmd,$file) = split;
    s/^$cmd[ \t]+(.*)$/\1/;
    $queue .= $_;
}

sub sendqueue_handler {
    printf PROX "$queue";
    $queue="";
}


#
# handles the CONNECT command 

sub np_connect {
    my($chost,$cport) = @_;

    # just in case
    close(PROX);           

    # create a socket
    socket(PROX, PF_INET, SOCK_STREAM, getprotobyname('tcp')) or die $!;

    my $addr = gethostbyname($chost);

    # build address of destination
    # Need perl 5.002 to do this 
    my $serv_addr = sockaddr_in($cport,$addr) || die $!;

    # call connect
    connect(PROX,$serv_addr) || die $!;
#    printf("CONNECTED\n");
    select PROX; $|=1;  select STDOUT;

}

sub np_test {
    print(PROX "GET http://127.0.0.1/ HTTP/1.0\r\n\r\n");
    while (<PROX>) {
	print "$_\n";
    }
}