Copy input to Scaleway in Interactive.pm

[gruntmaster-daemon.git] / gruntmaster-exec

diff --git a/gruntmaster-exec b/gruntmaster-exec
index 319671e2aa8153e5aa88bca254618616ed2bf491..97ac555311340538bb1271704b6a63fbf1c3b704 100755 (executable)
--- a/gruntmaster-exec
+++ b/gruntmaster-exec
@@ -18,9 +18,6 @@ use constant +{
        DIED => 5,
        REJ => 10,
 };
-# These constants are changed by ex/makevm
-use constant USER => 65534;
-use constant GROUP => 65534;
 
 use BSD::Resource qw/setrlimit RLIMIT_AS RLIMIT_FSIZE RLIMIT_NPROC/;
 use IPC::Signal qw/sig_name sig_num/;
@@ -28,39 +25,74 @@ use sigtrap qw/XFSZ/;
 
 use Getopt::Long;
 use POSIX qw//;
+use Text::ParseWords qw/shellwords/;
 use Time::HiRes qw/alarm/;
 
-my (@fds, $timeout, $mlimit, $olimit, $nobody);
-my $close = 1;
+my (@fds, $timeout, $mlimit, $olimit, $sudo, $keep_stderr);
 
 GetOptions(
-       "fd=s"      => \@fds,
-       "timeout=f" => \$timeout,
-       "mlimit=i"  => \$mlimit,
-       "olimit=i"  => \$olimit,
-       "close!"    => \$close,
-       "nobody!"   => \$nobody,
+       "fd=s"         => \@fds,
+       "timeout=f"    => \$timeout,
+       "mlimit=i"     => \$mlimit,
+       "olimit=i"     => \$olimit,
+       "keep-stderr!" => \$keep_stderr,
+       "sudo!"        => \$sudo,
 );
 
+sub test_pipe_read {
+       my $data = '';
+       sysread STDIN, $data, 4096 and syswrite STDOUT, "recvd\n" until $data =~ /done/;
+       syswrite STDOUT, 'done';
+}
+
+sub test_pipe_write {
+       my ($rin, $rout);
+       vec($rin, fileno STDIN, 1) = 1;
+       syswrite STDOUT, "data\n" until select $rout = $rin, undef, undef, 0.05;
+       syswrite STDOUT, 'done';
+       my $data = '';
+       sysread STDIN, $data, 4096 until $data =~ /done/;
+}
+
+sub test_pipes {
+       $ARGV[0] =~ /prog/ ? test_pipe_read : test_pipe_write
+}
+
+my @sudo;
+@sudo = (shellwords ($ENV{GRUNTMASTER_SUDO}), '--') if $ENV{GRUNTMASTER_SUDO} && $sudo;
+undef $mlimit if @sudo; # sudo wants a lot of address space
+
 my $ret = fork // die 'Cannot fork';
 if ($ret) {
-       my $tle;
-       local $SIG{ALRM} = sub { kill KILL => $ret; $tle = 1};
-       alarm ($timeout || 5);
+       my ($tle, $child_ready);
+       local $SIG{USR1} = sub { $child_ready = 1 };
+       sleep 3; # Wait for ready signal (SIGUSR1)
+       unless ($child_ready) {
+               kill KILL => $ret;
+               exit !say ERR, "\nNo response from gruntmaster-exec child";
+       }
+       local $SIG{ALRM} = sub {
+               kill KILL => $ret;
+               $tle = 1
+       };
+       alarm ($timeout || 10);
        waitpid $ret, 0;
        alarm 0;
+       if (@sudo) {
+               $? = $? >> 8;
+               $? = $? < 128 || $? > 128+32 ? ($? << 8) : $? - 128;
+       }
        my $sig = $? & 127;
        my $signame = sig_name $sig;
        exit !say TLE,  "\nTime Limit Exceeded"   if $tle;
        exit !say OLE,  "\nOutput Limit Exceeded" if $sig && $signame eq 'XFSZ';
-       exit !say DIED, "\nCrash (SIG$signame)"   if $sig && $signame ne 'PIPE';
+       exit !say DIED, "\nCrash (SIG$signame)"   if $sig && ($signame ne 'PIPE' || $ARGV[0] !~ /prog/);
        exit !say NZX,  "\nNon-zero exit status: " . ($? >> 8) if $? >> 8;
        exit !say AC,   "\nAll OK";
 } else {
        $^F = 50;
-       if ($close) {
-               POSIX::close $_ for 0 .. $^F;
-       }
+       POSIX::close 2 unless $keep_stderr;
+       POSIX::close $_ for 0, 1, 3 .. $^F;
        for my $fdstring (@fds) {
                my ($fd, $file) = split ' ', $fdstring, 2;
                open my $fh, $file or die $!;
@@ -70,12 +102,19 @@ if ($ret) {
                        POSIX::close $oldfd or die $!;
                }
        }
+       test_pipes if grep /tty|fifo/, @fds;
+       my $nproc = 15;
+       my $debug = $ENV{TEST_VERBOSE};
        %ENV = (ONLINE_JUDGE => 1, PATH => $ENV{PATH}, HOME => $ENV{HOME});
        setrlimit RLIMIT_AS, $mlimit, $mlimit or die $! if $mlimit;
        setrlimit RLIMIT_FSIZE, $olimit, $olimit or die $! if $olimit;
-       setrlimit RLIMIT_NPROC, 1, 1 or die $! if $nobody;
-       POSIX::setgid $nobody ? 65534 : USER;
-       POSIX::setuid $nobody ? 65534 : GROUP;
+       setrlimit RLIMIT_NPROC, $nproc, $nproc or die $! if $sudo;
+       open my $adj, '>', '/proc/self/oom_score_adj';
+       print $adj 900;
+       close $adj;
+       unshift @ARGV, @sudo;
+       kill USR1 => getppid; # Tell parent process that we're ready
+       say STDERR "Executing: ", join ' ', map { "'$_'" } @ARGV if $debug;
        exec @ARGV;
 }