use strict;
use warnings;
use parent qw/Exporter/;
+no if $] > 5.017011, warnings => 'experimental::smartmatch';
-use POSIX qw//;
+use Expect::Simple;
use File::Basename qw/fileparse/;
-use Gruntmaster::Daemon::Constants qw/TLE OLE DIED NZX/;
-use Log::Log4perl qw/get_logger/;
-use Time::HiRes qw/alarm/;
+use File::Slurp qw/read_file write_file/;
use List::MoreUtils qw/natatime/;
-use IPC::Signal qw/sig_name sig_num/;
-use IPC::Open3 qw/open3/;
-use File::Spec::Functions qw/devnull/;
-use Fcntl qw/F_GETFD F_SETFD FD_CLOEXEC/;
+use Log::Log4perl qw/get_logger/;
+use POSIX qw/mkfifo/;
+use String::ShellQuote qw/shell_quote/;
+use Try::Tiny;
-our $VERSION = '0.001';
-our @EXPORT_OK = qw/mkrun/;
+our $VERSION = '5999.000_004';
+our @EXPORT_OK = qw/prepare_files stopvms/;
##################################################
-sub command_and_args{
- my ($format, $basename) = @_;
- return "./$basename" if $format eq 'CPP';
- die
+our (%vm);
+
+sub runvm {
+ my ($name) = @_;
+ return unless $ENV{GRUNTMASTER_VM};
+ mkfifo "$name.in", 0600;
+ mkfifo "$name.out", 0600;
+ get_logger->trace("Starting VM $name");
+ $vm{$name} = Expect::Simple->new({
+ Cmd => "$ENV{GRUNTMASTER_VM} $name",
+ Prompt => '# ',
+ DisconnectCmd => 'exit',
+ RawPty => 1,
+ Timeout => 10,
+ });
+}
+
+sub stopvms { %vm = () }
+
+sub execlist {
+ my ($vm, @args) = @_;
+ my $er = "exec-result-$vm";
+ if ($vm{$vm}) {
+ my $cmd = ">$er " . shell_quote 'gruntmaster-exec', @args;
+ get_logger->trace("Running in VM $vm: $cmd");
+ $vm{$vm}->send($cmd);
+ } else {
+ my $ret = fork // die "Cannot fork\n";
+ if ($ret) {
+ waitpid $ret, 0;
+ } else {
+ open STDOUT, '>', $er or die "Cannot open $er\n";
+ get_logger->trace("Running: gruntmaster-exec @args");
+ exec 'gruntmaster-exec', @args;
+ }
+ }
+
+ die "gruntmaster-exec died\n" if -z $er;
+ my ($excode, $exmsg) = read_file $er;
+ unlink $er;
+ chomp ($excode, $exmsg); ## no critic (ProhibitParensWithBuiltins)
+ get_logger->trace("Exec result: $excode $exmsg");
+ die [$excode, $exmsg] if $excode > 0; ## no critic (RequireCarping)
}
sub mkrun{
my $format = shift;
sub{
+ local *__ANON__ = 'mkrun_runner';
my ($name, %args) = @_;
- my $basename = fileparse $name, qr/\.[^.]*/;
- my $ret = fork // die 'Cannot fork';
- if ($ret) {
- my $tle;
- local $SIG{ALRM} = sub { kill KILL => $ret; $tle = 1};
- alarm $args{timeout} if exists $args{timeout};
- wait;
- alarm 0;
- my $sig = $? & 127;
- my $signame = sig_name $sig;
- die [TLE, "Time Limit Exceeded"] if $tle;
- die [OLE, 'Output Limit Exceeded'] if $sig && $signame eq 'XFSZ';
- die [DIED, "Crash (SIG$signame)"] if $sig;
- die [NZX, "Non-zero exit status: " . ($? >> 8)] if $?;
- } else {
- my @fds = exists $args{fds} ? @{$args{fds}} : ();
- $^F = 50;
- POSIX::close $_ for 0 .. $^F;
- my $it = natatime 2, @fds;
- while (my ($fd, $file) = $it->()) {
- open my $fh, $file or die $!;
- my $oldfd = fileno $fh;
- if ($oldfd != $fd) {
- POSIX::dup2 $oldfd, $fd or die $!;
- POSIX::close $oldfd or die $!;
- }
- }
- exec 'gruntmaster-exec', $args{mlimit} // 0, $args{olimit} // 0, command_and_args($format, $basename), exists $args{args} ? @{$args{args}} : ();
+ get_logger->trace("Running $name...");
+ my $basename = fileparse $name, qr/[.][^.]*/s;
+ my @args = ('--sudo');
+ push @args, '--no-close' if $ENV{TEST_VERBOSE};
+ push @args, '--timeout', $args{timeout} if $args{timeout};
+ push @args, '--mlimit', $args{mlimit} if $args{mlimit};
+ push @args, '--olimit', $args{olimit} if $args{olimit};
+ my @fds = exists $args{fds} ? @{$args{fds}} : ();
+ my $it = natatime 2, @fds;
+ while (my ($fd, $file) = $it->()) {
+ push @args, "--fd=$fd $file";
}
+ execlist $basename, @args, '--', "./$basename", @{$args{args}}
}
}
-1
+sub prepare{
+ my ($name, $format) = @_;
+ get_logger->trace("Preparing file $name...");
+
+ try {
+ execlist prog => '--fd=1 >>errors', '--fd=2 >>errors', 'gruntmaster-compile', $format, $name;
+ } catch {
+ my $exmsg = $_->[1];
+ die "Compile error ($exmsg)\n"
+ } finally {
+ $Gruntmaster::Daemon::errors .= read_file 'errors';
+ $Gruntmaster::Daemon::errors .= "\n" if -s 'errors';
+ unlink 'errors';
+ };
+}
+
+sub prepare_files{
+ my $meta = shift;
+ runvm $_ for keys %{$meta->{files}};
+
+ for my $file (values %{$meta->{files}}) {
+ my ($format, $name, $content) = @{$file}{qw/format name content/};
+
+ $file->{run} = mkrun($format);
+ write_file $name, $content;
+ prepare $name, $format;
+ }
+}
+
+1;
+__END__
+
+=encoding utf-8
+
+=head1 NAME
+
+Gruntmaster::Daemon::Format - Utility functions for handling source files
+
+=head1 SYNOPSIS
+
+ use Gruntmaster::Daemon::Format qw/prepare_files/;
+ prepare_files { files => {
+ prog => {
+ name => 'prog.pl',
+ format => 'PERL',
+ content => 'print "Hello, world!"'
+ },
+ ver => {
+ name => 'ver.cpp',
+ format => 'CPP',
+ content => ...
+ },
+ }};
+
+=head1 DESCRIPTION
+
+Gruntmaster::Daemon::Format exports utility functions for handling source files.
+
+=over
+
+=item B<prepare_files> I<$meta>
+
+Compiles all the source files in C<< $meta->{files} >>.
+
+=back
+
+=head1 AUTHOR
+
+Marius Gavrilescu E<lt>marius@ieval.roE<gt>
+
+=head1 COPYRIGHT AND LICENSE
+
+Copyright (C) 2014 by Marius Gavrilescu
+
+This library is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+
+=cut
projects / gruntmaster-daemon.git / blobdiff