t/spawn.t

   1 # Copyright (C) 2015-2021 all contributors <meta@public-inbox.org>
   2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   3 use strict;
   4 use warnings;
   5 use Test::More;
   6 use PublicInbox::Spawn qw(which spawn popen_rd);
   7 use PublicInbox::Sigfd;
   8
   9 {
  10         my $true = which('true');
  11         ok($true, "'true' command found with which()");
  12 }
  13
  14 {
  15         my $pid = spawn(['true']);
  16         ok($pid, 'spawned process');
  17         is(waitpid($pid, 0), $pid, 'waitpid succeeds on spawned process');
  18         is($?, 0, 'true exited successfully');
  19 }
  20
  21 { # ensure waitpid(-1, 0) and SIGCHLD works in spawned process
  22         my $script = <<'EOF';
  23 $| = 1; # unbuffer stdout
  24 defined(my $pid = fork) or die "fork: $!";
  25 if ($pid == 0) { exit }
  26 elsif ($pid > 0) {
  27         my $waited = waitpid(-1, 0);
  28         $waited == $pid or die "mismatched child $pid != $waited";
  29         $? == 0 or die "child err: $>";
  30         $SIG{CHLD} = sub { print "HI\n"; exit };
  31         print "RDY $$\n";
  32         select(undef, undef, undef, 0.01) while 1;
  33 }
  34 EOF
  35         my $oldset = PublicInbox::Sigfd::block_signals();
  36         my $rd = popen_rd([$^X, '-e', $script]);
  37         diag 'waiting for child to reap grandchild...';
  38         chomp(my $line = readline($rd));
  39         my ($rdy, $pid) = split(' ', $line);
  40         is($rdy, 'RDY', 'got ready signal, waitpid(-1) works in child');
  41         ok(kill('CHLD', $pid), 'sent SIGCHLD to child');
  42         is(readline($rd), "HI\n", '$SIG{CHLD} works in child');
  43         ok(close $rd, 'popen_rd close works');
  44         PublicInbox::Sigfd::sig_setmask($oldset);
  45 }
  46
  47 {
  48         my ($r, $w);
  49         pipe $r, $w or die "pipe failed: $!";
  50         my $pid = spawn(['echo', 'hello world'], undef, { 1 => fileno($w) });
  51         close $w or die "close pipe[1] failed: $!";
  52         is(<$r>, "hello world\n", 'read stdout of spawned from pipe');
  53         is(waitpid($pid, 0), $pid, 'waitpid succeeds on spawned process');
  54         is($?, 0, 'true exited successfully');
  55 }
  56
  57 {
  58         my ($r, $w);
  59         pipe $r, $w or die "pipe failed: $!";
  60         my $pid = spawn(['sh', '-c', 'echo $HELLO'],
  61                 { 'HELLO' => 'world' }, { 1 => $w });
  62         close $w or die "close pipe[1] failed: $!";
  63         is(<$r>, "world\n", 'read stdout of spawned from pipe');
  64         is(waitpid($pid, 0), $pid, 'waitpid succeeds on spawned process');
  65         is($?, 0, 'sh exited successfully');
  66 }
  67
  68 {
  69         my $fh = popen_rd([qw(echo hello)]);
  70         ok(fileno($fh) >= 0, 'tied fileno works');
  71         my $l = <$fh>;
  72         is($l, "hello\n", 'tied readline works');
  73         $l = <$fh>;
  74         ok(!$l, 'tied readline works for EOF');
  75 }
  76
  77 {
  78         my $fh = popen_rd([qw(printf foo\nbar)]);
  79         ok(fileno($fh) >= 0, 'tied fileno works');
  80         my @line = <$fh>;
  81         is_deeply(\@line, [ "foo\n", 'bar' ], 'wantarray works on readline');
  82 }
  83
  84 {
  85         my $fh = popen_rd([qw(echo hello)]);
  86         my $buf;
  87         is(sysread($fh, $buf, 6), 6, 'sysread got 6 bytes');
  88         is($buf, "hello\n", 'tied gets works');
  89         is(sysread($fh, $buf, 6), 0, 'sysread got EOF');
  90         $? = 1;
  91         ok(close($fh), 'close succeeds');
  92         is($?, 0, '$? set properly');
  93 }
  94
  95 {
  96         my $fh = popen_rd([qw(false)]);
  97         ok(!close($fh), 'close fails on false');
  98         isnt($?, 0, '$? set properly: '.$?);
  99 }
 100
 101 SKIP: {
 102         eval {
 103                 require BSD::Resource;
 104                 defined(BSD::Resource::RLIMIT_CPU())
 105         } or skip 'BSD::Resource::RLIMIT_CPU missing', 3;
 106         my ($r, $w);
 107         pipe($r, $w) or die "pipe: $!";
 108         my $cmd = ['sh', '-c', 'while true; do :; done'];
 109         my $fd = fileno($w);
 110         my $opt = { RLIMIT_CPU => [ 1, 1 ], RLIMIT_CORE => [ 0, 0 ], 1 => $fd };
 111         my $pid = spawn($cmd, undef, $opt);
 112         close $w or die "close(w): $!";
 113         my $rset = '';
 114         vec($rset, fileno($r), 1) = 1;
 115         ok(select($rset, undef, undef, 5), 'child died before timeout');
 116         is(waitpid($pid, 0), $pid, 'XCPU child process reaped');
 117         isnt($?, 0, 'non-zero exit status');
 118 }
 119
 120 done_testing();
 121
 122 1;