t/cmd_ipc.t

   1 #!perl -w
   2 # Copyright (C) 2021 all contributors <meta@public-inbox.org>
   3 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   4 use strict;
   5 use v5.10.1;
   6 use Test::More;
   7 use PublicInbox::TestCommon;
   8 use Socket qw(AF_UNIX SOCK_STREAM MSG_EOR);
   9 pipe(my ($r, $w)) or BAIL_OUT;
  10 my ($send, $recv);
  11 require_ok 'PublicInbox::Spawn';
  12 my $SOCK_SEQPACKET = eval { Socket::SOCK_SEQPACKET() } // undef;
  13
  14 my $do_test = sub { SKIP: {
  15         my ($type, $flag, $desc) = @_;
  16         defined $type or skip 'SOCK_SEQPACKET missing', 7;
  17         my ($s1, $s2);
  18         my $src = 'some payload' x 40;
  19         socketpair($s1, $s2, AF_UNIX, $type, 0) or BAIL_OUT $!;
  20         $send->($s1, fileno($r), fileno($w), fileno($s1), $src, $flag);
  21         my (@fds) = $recv->($s2, my $buf, length($src) + 1);
  22         is($buf, $src, 'got buffer payload '.$desc);
  23         my ($r1, $w1, $s1a);
  24         my $opens = sub {
  25                 ok(open($r1, '<&=', $fds[0]), 'opened received $r');
  26                 ok(open($w1, '>&=', $fds[1]), 'opened received $w');
  27                 ok(open($s1a, '+>&=', $fds[2]), 'opened received $s1');
  28         };
  29         $opens->();
  30         my @exp = stat $r;
  31         my @cur = stat $r1;
  32         is("$exp[0]\0$exp[1]", "$cur[0]\0$cur[1]", '$r dev/ino matches');
  33         @exp = stat $w;
  34         @cur = stat $w1;
  35         is("$exp[0]\0$exp[1]", "$cur[0]\0$cur[1]", '$w dev/ino matches');
  36         @exp = stat $s1;
  37         @cur = stat $s1a;
  38         is("$exp[0]\0$exp[1]", "$cur[0]\0$cur[1]", '$s1 dev/ino matches');
  39         if (defined($SOCK_SEQPACKET) && $type == $SOCK_SEQPACKET) {
  40                 $r1 = $w1 = $s1a = undef;
  41                 $src = (',' x 1023) . '-' .('.' x 1024);
  42                 $send->($s1, fileno($r), fileno($w), fileno($s1), $src, $flag);
  43                 (@fds) = $recv->($s2, $buf, 1024);
  44                 is($buf, (',' x 1023) . '-', 'silently truncated buf');
  45                 $opens->();
  46                 $r1 = $w1 = $s1a = undef;
  47                 close $s1;
  48                 @fds = $recv->($s2, $buf, length($src) + 1);
  49                 is_deeply(\@fds, [], "no FDs on EOF $desc");
  50                 is($buf, '', "buffer cleared on EOF ($desc)");
  51
  52         }
  53 } };
  54
  55 my $send_ic = PublicInbox::Spawn->can('send_cmd4');
  56 my $recv_ic = PublicInbox::Spawn->can('recv_cmd4');
  57 SKIP: {
  58         ($send_ic && $recv_ic) or skip 'Inline::C not installed/enabled', 12;
  59         $send = $send_ic;
  60         $recv = $recv_ic;
  61         $do_test->(SOCK_STREAM, 0, 'Inline::C stream');
  62         $do_test->($SOCK_SEQPACKET, MSG_EOR, 'Inline::C seqpacket');
  63 }
  64
  65 SKIP: {
  66         require_mods('Socket::MsgHdr', 13);
  67         require_ok 'PublicInbox::CmdIPC4';
  68         $send = PublicInbox::CmdIPC4->can('send_cmd4');
  69         $recv = PublicInbox::CmdIPC4->can('recv_cmd4');
  70         $do_test->(SOCK_STREAM, 0, 'MsgHdr stream');
  71         $do_test->($SOCK_SEQPACKET, MSG_EOR, 'MsgHdr seqpacket');
  72         SKIP: {
  73                 ($send_ic && $recv_ic) or
  74                         skip 'Inline::C not installed/enabled', 12;
  75                 $recv = $recv_ic;
  76                 $do_test->(SOCK_STREAM, 0, 'Inline::C -> MsgHdr stream');
  77                 $do_test->($SOCK_SEQPACKET, 0, 'Inline::C -> MsgHdr seqpacket');
  78         }
  79 }
  80
  81 SKIP: {
  82         require_mods('IO::FDPass', 13);
  83         require_ok 'PublicInbox::CmdIPC1';
  84         $send = PublicInbox::CmdIPC1->can('send_cmd1');
  85         $recv = PublicInbox::CmdIPC1->can('recv_cmd1');
  86         $do_test->(SOCK_STREAM, 0, 'IO::FDPass stream');
  87         $do_test->($SOCK_SEQPACKET, MSG_EOR, 'IO::FDPass seqpacket');
  88 }
  89
  90 done_testing;