#!/usr/bin/perl -w
use strict;
use IO::Socket;
my ($host, $port, $kidpid, $handle, $line);

unless (@ARGV == 2) { die "usage: $0 host port" }
($host, $port) = @ARGV;

# create a tcp connection to the specified host and port
$handle = IO::Socket::INET->new(Proto     => "tcp",
   PeerAddr  => $host,
   PeerPort  => $port)
   or die "can't connect to port $port on $host: $!";

$handle->autoflush(1);    # so output gets there right away
print STDERR "[Connected to $host:$port]\n";


# Get/display handshake message from server
my $phandshake = get_from_server($handle);
map { print $_, "\n" } @$phandshake;


# Socket loop
while (1) {
    # Get user input and send it to the server
    $line = <STDIN>;
    print $handle $line;

    # Get and display server response
    my $plines = get_from_server($handle);
    map { print "$_\n" } @$plines;

}


sub get_from_server {
    my $handle = shift;
    my @lines;
    while (1) {
        my $line = <$handle>;
        defined($line) or die "Server closed connection\n";
        chomp $line;
        ($line =~ s/^([-+])//) or die "Invalid server response:  '$line'\n";
        my $char = $1;
        push @lines, $line;
        last if ($char eq '+');
    }
    return [ @lines ];
}