X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;ds=sidebyside;f=submit%2FMO%2FSubmit.pm;h=18441dc791cc360ecff490debb6e498a6ca4f8c9;hb=9306a593e097b30b55196f0fb7dd7c189389a0d3;hp=5852b80cd972f39a3b347dd76e51d06cb122211f;hpb=c05f79a5d7b6d14d938d25198846c173f27e8b39;p=eval.git

diff --git a/submit/MO/Submit.pm b/submit/MO/Submit.pm
index 5852b80..18441dc 100644
--- a/submit/MO/Submit.pm
+++ b/submit/MO/Submit.pm
@@ -9,30 +9,44 @@ use warnings;
 use IO::Socket::INET;
 use IO::Socket::SSL; # qw(debug3);
 use Sherlock::Object;
+use POSIX;
 
 sub new($) {
+	my $user = $ENV{"USER"} or die "Environment variable USER not set\n";
+	my $home = $ENV{"HOME"} or die "Environment variable HOME not set\n";
+	my $mo = "$home/.mo";
+	my $root = $ENV{"MO_ROOT"} or die "Environment variable MO_ROOT not set\n";
 	my $self = {
+		"Contest" => "CPSPC 2007",
 		"Server" => "localhost:8888",
-		"Key" => "client-key.pem",
-		"Cert" => "client-cert.pem",
-		"CACert" => "ca-cert.pem",
-		"user" => "testuser",
+		"Key" => "$mo/key.pem",		# Keys and certificates
+		"Cert" => "$mo/cert.pem",
+		"CACert" => "$mo/ca-cert.pem",
+		"Trace" => defined $ENV{"MO_SUBMIT_TRACE"},
+		"History" => "$home/.history",	# Keep submission history in this directory
+		"RefreshTimer" => 5000,		# How often GUI sends STATUS commands [ms]
+		"user" => $user,
 		"sk" => undef,
 		"error" => undef,
 	};
-	# FIXME: Read config file
 	return bless $self;
 }
 
+sub DESTROY($) {
+	my $self = shift @_;
+	$self->disconnect;
+}
+
 sub log($$) {
 	my ($self, $msg) = @_;
-	print STDERR "LOG: $msg\n";
+	print STDERR "SUBMIT: $msg\n" if $self->{"Trace"};
 }
 
 sub err($$) {
 	my ($self, $msg) = @_;
-	print STDERR "ERROR: $msg\n";
+	print STDERR "ERROR: $msg\n" if $self->{"Trace"};
 	$self->{"error"} = $msg;
+	$self->disconnect;
 }
 
 sub is_connected($) {
@@ -51,7 +65,7 @@ sub disconnect($) {
 
 sub connect($) {
 	my $self = shift @_;
-	!defined $self->{"sk"} or close $self->{"sk"};
+	$self->disconnect;
 	$self->log("Connecting to submit server");
 	my $sk = new IO::Socket::INET(
 		PeerAddr => $self->{"Server"},
@@ -79,9 +93,9 @@ sub connect($) {
 			$sk,
 			SSL_version => 'TLSv1',
 			SSL_use_cert => 1,
-			SSL_key_file => "client-key.pem",
-			SSL_cert_file => "client-cert.pem",
-			SSL_ca_file => "ca-cert.pem",
+			SSL_key_file => $self->{"Key"},
+			SSL_cert_file => $self->{"Cert"},
+			SSL_ca_file => $self->{"CACert"},
 			SSL_verify_mode => 3,
 		);
 		if (!defined $sk) {
@@ -97,7 +111,6 @@ sub connect($) {
 	my $err = $reply->get("-");
 	if (defined $err) {
 		$self->err("Cannot log in: $err");
-		$self->disconnect;
 		return undef;
 	}
 
@@ -109,6 +122,10 @@ sub request($$) {
 	my ($self, $obj) = @_;
 	my $sk = $self->{"sk"};
 	$obj->write($sk);	### FIXME: Flushing
+	if ($sk->error) {
+		$self->err("Connection broken");
+		return undef;
+	}
 	print $sk "\n";
 	return $self->reply;
 }
@@ -125,4 +142,31 @@ sub reply($) {
 	}
 }
 
+sub send_file($$$) {
+	my ($self, $fh, $size) = @_;
+	my $sk = $self->{"sk"};
+	while ($size) {
+		my $l = ($size < 4096 ? $size : 4096);
+		my $buf = "";
+		if ($fh->read($buf, $l) != $l) {
+			$self->err("File shrunk during upload");
+			return undef;
+		}
+		$sk->write($buf, $l);
+		if ($sk->error) {
+			$self->err("Connection broken");
+			return undef;
+		}
+		$size -= $l;
+	}
+	return $self->reply;
+}
+
+sub local_submit($$$$$) {
+	my ($self, $task, $part, $ext, $filename) = @_;
+	my $hist = $self->{"History"};
+	-d $hist or mkdir $hist or return "Unable to create $hist: $!";
+	### FIXME: Unfinished
+}
+
 1;