Run user programs as nobody:nogroup
[gruntmaster-daemon.git] / lib / Gruntmaster / Daemon / Format.pm
1 package Gruntmaster::Daemon::Format;
2
3 use 5.014000;
4 use strict;
5 use warnings;
6 use parent qw/Exporter/;
7 no if $] > 5.017011, warnings => 'experimental::smartmatch';
8
9 use Expect::Simple;
10 use File::Basename qw/fileparse/;
11 use File::Slurp qw/read_file write_file/;
12 use List::MoreUtils qw/natatime/;
13 use Log::Log4perl qw/get_logger/;
14 use POSIX qw/mkfifo/;
15 use String::ShellQuote qw/shell_quote/;
16 use Try::Tiny;
17
18 our $VERSION = '5999.000_004';
19 our @EXPORT_OK = qw/prepare_files stopvms/;
20
21 ##################################################
22
23 our (%vm);
24
25 sub 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
40 sub stopvms { %vm = () }
41
42 sub execlist {
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);
49 } else {
50 my $ret = fork // die "Cannot fork\n";
51 if ($ret) {
52 waitpid $ret, 0;
53 } else {
54 open STDOUT, '>', $er or die "Cannot open $er\n";
55 get_logger->trace("Running: gruntmaster-exec @args");
56 exec 'gruntmaster-exec', @args;
57 }
58 }
59
60 die "gruntmaster-exec died\n" if -z $er;
61 my ($excode, $exmsg) = read_file $er;
62 unlink $er;
63 chomp ($excode, $exmsg); ## no critic (ProhibitParensWithBuiltins)
64 get_logger->trace("Exec result: $excode $exmsg");
65 die [$excode, $exmsg] if $excode > 0; ## no critic (RequireCarping)
66 }
67
68 sub command_and_args{
69 my ($format, $basename) = @_;
70
71 given($format) {
72 "./$basename" when [qw/C CPP GCCGO GOLANG HASKELL PASCAL/];
73 "./$basename.exe" when 'MONO';
74 java => $basename when 'JAVA';
75 perl => $basename when 'PERL';
76 python => $basename when 'PYTHON';
77 default { die "Don't know how to execute format $format\n" }
78 }
79 }
80
81 sub mkrun{
82 my $format = shift;
83 sub{
84 local *__ANON__ = 'mkrun_runner';
85 my ($name, %args) = @_;
86 get_logger->trace("Running $name...");
87 my $basename = fileparse $name, qr/[.][^.]*/s;
88 my @args = ('--nobody');
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";
96 }
97 execlist $basename, @args, command_and_args($format, $basename);
98 }
99 }
100
101 sub prepare{
102 my ($name, $format) = @_;
103 get_logger->trace("Preparing file $name...");
104
105 try {
106 execlist prog => '--fd=1 >>errors', '--fd=2 >>errors', 'gruntmaster-compile', $format, $name;
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 };
114 }
115
116 sub prepare_files{
117 my $meta = shift;
118 runvm $_ for keys %{$meta->{files}};
119
120 for my $file (values %{$meta->{files}}) {
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
129 1;
130 __END__
131
132 =encoding utf-8
133
134 =head1 NAME
135
136 Gruntmaster::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
156 Gruntmaster::Daemon::Format exports utility functions for handling source files.
157
158 =over
159
160 =item B<prepare_files> I<$meta>
161
162 Compiles all the source files in C<< $meta->{files} >>.
163
164 =back
165
166 =head1 AUTHOR
167
168 Marius Gavrilescu E<lt>marius@ieval.roE<gt>
169
170 =head1 COPYRIGHT AND LICENSE
171
172 Copyright (C) 2014 by Marius Gavrilescu
173
174 This library is free software: you can redistribute it and/or modify
175 it under the terms of the GNU Affero General Public License as published by
176 the 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.058249 seconds and 4 git commands to generate.