Run user programs as nobody:nogroup
[gruntmaster-daemon.git] / lib / Gruntmaster / Daemon / Format.pm
CommitLineData
ddceb393
MG
1package Gruntmaster::Daemon::Format;
2
3use 5.014000;
4use strict;
5use warnings;
6use parent qw/Exporter/;
9577371b 7no if $] > 5.017011, warnings => 'experimental::smartmatch';
ddceb393 8
fdb99417 9use Expect::Simple;
ddceb393 10use File::Basename qw/fileparse/;
c40a2dd0 11use File::Slurp qw/read_file write_file/;
ddceb393 12use List::MoreUtils qw/natatime/;
f5e29130 13use Log::Log4perl qw/get_logger/;
fdb99417
MG
14use POSIX qw/mkfifo/;
15use String::ShellQuote qw/shell_quote/;
c40a2dd0 16use Try::Tiny;
ddceb393 17
99d37110 18our $VERSION = '5999.000_004';
fdb99417 19our @EXPORT_OK = qw/prepare_files stopvms/;
ddceb393
MG
20
21##################################################
22
fdb99417
MG
23our (%vm);
24
25sub runvm {
26 my ($name) = @_;
27 return unless $ENV{GRUNTMASTER_VM};
28 mkfifo "$name.in", 0600;
29 mkfifo "$name.out", 0600;
30 get_logger->trace("Starting VM $name");
31 $vm{$name} = Expect::Simple->new({
32 Cmd => "$ENV{GRUNTMASTER_VM} $name",
33 Prompt => '# ',
34 DisconnectCmd => 'exit',
35 RawPty => 1,
36 Timeout => 10,
37 });
38}
39
40sub stopvms { %vm = () }
41
c40a2dd0 42sub execlist {
fdb99417
MG
43 my ($vm, @args) = @_;
44 my $er = "exec-result-$vm";
45 if ($vm{$vm}) {
46 my $cmd = ">$er " . shell_quote 'gruntmaster-exec', @args;
47 get_logger->trace("Running in VM $vm: $cmd");
48 $vm{$vm}->send($cmd);
c40a2dd0 49 } else {
99d37110 50 my $ret = fork // die "Cannot fork\n";
fdb99417
MG
51 if ($ret) {
52 waitpid $ret, 0;
53 } else {
99d37110 54 open STDOUT, '>', $er or die "Cannot open $er\n";
e61f1a82 55 get_logger->trace("Running: gruntmaster-exec @args");
fdb99417
MG
56 exec 'gruntmaster-exec', @args;
57 }
c40a2dd0 58 }
fdb99417
MG
59
60 die "gruntmaster-exec died\n" if -z $er;
61 my ($excode, $exmsg) = read_file $er;
62 unlink $er;
99d37110 63 chomp ($excode, $exmsg); ## no critic (ProhibitParensWithBuiltins)
fdb99417 64 get_logger->trace("Exec result: $excode $exmsg");
99d37110 65 die [$excode, $exmsg] if $excode > 0; ## no critic (RequireCarping)
c40a2dd0
MG
66}
67
ddceb393
MG
68sub command_and_args{
69 my ($format, $basename) = @_;
9577371b
MG
70
71 given($format) {
5a4392d5 72 "./$basename" when [qw/C CPP GCCGO GOLANG HASKELL PASCAL/];
9577371b 73 "./$basename.exe" when 'MONO';
cdd811c0
MG
74 java => $basename when 'JAVA';
75 perl => $basename when 'PERL';
76 python => $basename when 'PYTHON';
c40a2dd0 77 default { die "Don't know how to execute format $format\n" }
9577371b 78 }
ddceb393
MG
79}
80
81sub mkrun{
82 my $format = shift;
83 sub{
d16a999c 84 local *__ANON__ = 'mkrun_runner';
ddceb393 85 my ($name, %args) = @_;
d16a999c 86 get_logger->trace("Running $name...");
99d37110 87 my $basename = fileparse $name, qr/[.][^.]*/s;
4e08f696 88 my @args = ('--nobody');
c40a2dd0
MG
89 push @args, '--timeout', $args{timeout} if $args{timeout};
90 push @args, '--mlimit', $args{mlimit} if $args{mlimit};
91 push @args, '--olimit', $args{olimit} if $args{olimit};
92 my @fds = exists $args{fds} ? @{$args{fds}} : ();
93 my $it = natatime 2, @fds;
94 while (my ($fd, $file) = $it->()) {
95 push @args, "--fd=$fd $file";
ddceb393 96 }
fdb99417 97 execlist $basename, @args, command_and_args($format, $basename);
ddceb393
MG
98 }
99}
100
ad77b7d3
MG
101sub prepare{
102 my ($name, $format) = @_;
ad77b7d3
MG
103 get_logger->trace("Preparing file $name...");
104
c40a2dd0 105 try {
fdb99417 106 execlist prog => '--fd=1 >>errors', '--fd=2 >>errors', 'gruntmaster-compile', $format, $name;
c40a2dd0
MG
107 } catch {
108 die "Compile error\n"
109 } finally {
110 $Gruntmaster::Daemon::errors .= read_file 'errors';
111 $Gruntmaster::Daemon::errors .= "\n";
112 unlink 'errors';
113 };
ad77b7d3
MG
114}
115
116sub prepare_files{
117 my $meta = shift;
fdb99417 118 runvm $_ for keys %{$meta->{files}};
ad77b7d3 119
759d3ca5 120 for my $file (values %{$meta->{files}}) {
ad77b7d3
MG
121 my ($format, $name, $content) = @{$file}{qw/format name content/};
122
123 $file->{run} = mkrun($format);
124 write_file $name, $content;
125 prepare $name, $format;
126 }
127}
128
bc372959
MG
1291;
130__END__
131
132=encoding utf-8
133
134=head1 NAME
135
136Gruntmaster::Daemon::Format - Utility functions for handling source files
137
138=head1 SYNOPSIS
139
140 use Gruntmaster::Daemon::Format qw/prepare_files/;
141 prepare_files { files => {
142 prog => {
143 name => 'prog.pl',
144 format => 'PERL',
145 content => 'print "Hello, world!"'
146 },
147 ver => {
148 name => 'ver.cpp',
149 format => 'CPP',
150 content => ...
151 },
152 }};
153
154=head1 DESCRIPTION
155
156Gruntmaster::Daemon::Format exports utility functions for handling source files.
157
158=over
159
160=item B<prepare_files> I<$meta>
161
162Compiles all the source files in C<< $meta->{files} >>.
163
164=back
165
166=head1 AUTHOR
167
168Marius Gavrilescu E<lt>marius@ieval.roE<gt>
169
170=head1 COPYRIGHT AND LICENSE
171
172Copyright (C) 2014 by Marius Gavrilescu
173
174This library is free software: you can redistribute it and/or modify
175it under the terms of the GNU Affero General Public License as published by
176the Free Software Foundation, either version 3 of the License, or
177(at your option) any later version.
178
179
180=cut
This page took 0.079612 seconds and 4 git commands to generate.