]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/SpawnPP.pm
spawn_pp: cleanup, error checks and descriptive errors
[public-inbox.git] / lib / PublicInbox / SpawnPP.pm
index 288625d00a9279305bea4e7de7f69fe8f493fa14..7a5793f621ea710cf8c7fbc0b4f7cb37e8a2e087 100644 (file)
@@ -1,31 +1,63 @@
-# Copyright (C) 2016 all contributors <meta@public-inbox.org>
+# Copyright (C) all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
+
+# Pure-Perl implementation of "spawn".  This can't take advantage
+# of vfork, so no speedups under Linux for spawning from large processes.
 package PublicInbox::SpawnPP;
-use strict;
-use warnings;
-use POSIX qw(dup2);
+use v5.12;
+use POSIX qw(dup2 _exit setpgid :signal_h);
 
 # Pure Perl implementation for folks that do not use Inline::C
-sub public_inbox_fork_exec ($$$$$$) {
-       my ($in, $out, $err, $f, $cmd, $env) = @_;
-       my $pid = fork;
+sub pi_fork_exec ($$$$$$$) {
+       my ($redir, $f, $cmd, $env, $rlim, $cd, $pgid) = @_;
+       my $old = POSIX::SigSet->new();
+       my $set = POSIX::SigSet->new();
+       $set->fillset or die "sigfillset: $!";
+       sigprocmask(SIG_SETMASK, $set, $old) or die "SIG_SETMASK(set): $!";
+       my $syserr;
+       pipe(my ($r, $w)) or die "pipe: $!";
+       my $pid = fork // die "fork (+exec) @$cmd: $!\n";
        if ($pid == 0) {
-               if ($in != 0) {
-                       dup2($in, 0) or die "dup2 failed for stdin: $!";
+               close $r;
+               $SIG{__DIE__} = sub {
+                       warn(@_);
+                       syswrite($w, my $num = $! + 0);
+                       _exit(1);
+               };
+               for my $child_fd (0..$#$redir) {
+                       my $parent_fd = $redir->[$child_fd];
+                       next if $parent_fd == $child_fd;
+                       dup2($parent_fd, $child_fd) or
+                               die "dup2($parent_fd, $child_fd): $!";
+               }
+               if ($pgid >= 0 && !defined(setpgid(0, $pgid))) {
+                       die "setpgid(0, $pgid): $!";
                }
-               if ($out != 1) {
-                       dup2($out, 1) or die "dup2 failed for stdout: $!";
+               $SIG{$_} = 'DEFAULT' for grep(!/\A__/, keys %SIG);
+               if ($cd ne '') {
+                       chdir $cd or die "chdir $cd: $!";
                }
-               if ($err != 2) {
-                       dup2($err, 2) or die "dup2 failed for stderr: $!";
+               while (@$rlim) {
+                       my ($r, $soft, $hard) = splice(@$rlim, 0, 3);
+                       BSD::Resource::setrlimit($r, $soft, $hard) or
+                               die "setrlimit($r=[$soft,$hard]: $!)";
                }
-               %ENV = ();
-               foreach my $e (@$env) {
-                       my ($k, $v) = split('=', $e, 2);
-                       $ENV{$k} = $v;
+               $old->delset(POSIX::SIGCHLD) or die "sigdelset CHLD: $!";
+               sigprocmask(SIG_SETMASK, $old) or die "SIG_SETMASK ~CHLD: $!";
+               $cmd->[0] = $f;
+               if ($ENV{MOD_PERL}) {
+                       @$cmd = (which('env'), '-i', @$env, @$cmd);
+               } else {
+                       %ENV = map { split(/=/, $_, 2) } @$env;
                }
-               exec @$cmd;
-               die "exec $cmd->[0] failed: $!\n";
+               exec { $f } @$cmd;
+               die "exec @$cmd failed: $!";
+       }
+       close $w;
+       sigprocmask(SIG_SETMASK, $old) or die "SIG_SETMASK(old): $!";
+       if (my $cerrnum = do { local $/, <$r> }) {
+               $! = $cerrnum;
+               die "forked child $@: $!";
        }
        $pid;
 }