]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/CmdIPC4.pm
No ext_urls
[public-inbox.git] / lib / PublicInbox / CmdIPC4.pm
index 90fca62d0301648a17e664e0ba8213f0066b7230..e368d0324199d1db9617838d8b7d0f4ed874f4bf 100644 (file)
@@ -1,32 +1,41 @@
-# Copyright (C) 2020-2021 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>
 
 # callers should use PublicInbox::CmdIPC4->can('send_cmd4') (or recv_cmd4)
 # first choice for script/lei front-end and 2nd choice for lei backend
-# libsocket-msghdr-perl is in Debian but many other distros as of 2021.
+# libsocket-msghdr-perl is in Debian but not many other distros as of 2021.
 package PublicInbox::CmdIPC4;
-use strict;
-use v5.10.1;
+use v5.12;
 use Socket qw(SOL_SOCKET SCM_RIGHTS);
 BEGIN { eval {
 require Socket::MsgHdr; # XS
 no warnings 'once';
 
-# 3 FDs per-sendmsg(2) + buffer
-*send_cmd4 = sub ($$$$$$) { # (sock, in, out, err, buf, flags) = @_;
-       my $mh = Socket::MsgHdr->new(buf => $_[4]);
-       $mh->cmsghdr(SOL_SOCKET, SCM_RIGHTS, pack('iii', @_[1,2,3]));
-       Socket::MsgHdr::sendmsg($_[0], $mh, $_[5]) or die "sendmsg: $!";
+# any number of FDs per-sendmsg(2) + buffer
+*send_cmd4 = sub ($$$$) { # (sock, fds, buf, flags) = @_;
+       my ($sock, $fds, undef, $flags) = @_;
+       my $mh = Socket::MsgHdr->new(buf => $_[2]);
+       $mh->cmsghdr(SOL_SOCKET, SCM_RIGHTS, pack('i' x scalar(@$fds), @$fds));
+       my $s;
+       my $try = 0;
+       do {
+               $s = Socket::MsgHdr::sendmsg($sock, $mh, $flags);
+       } while (!defined($s) &&
+                       ($!{ENOBUFS} || $!{ENOMEM} || $!{ETOOMANYREFS}) &&
+                       (++$try < 50) &&
+                       warn "sleeping on sendmsg: $! (#$try)\n" &&
+                       select(undef, undef, undef, 0.1) == 0);
+       $s;
 };
 
 *recv_cmd4 = sub ($$$) {
        my ($s, undef, $len) = @_; # $_[1] = destination buffer
        my $mh = Socket::MsgHdr->new(buflen => $len, controllen => 256);
-       my $r = Socket::MsgHdr::recvmsg($s, $mh, 0) // die "recvmsg: $!";
+       my $r = Socket::MsgHdr::recvmsg($s, $mh, 0) // return (undef);
        $_[1] = $mh->buf;
        return () if $r == 0;
        my (undef, undef, $data) = $mh->cmsghdr;
-       unpack('iii', $data);
+       defined($data) ? unpack('i' x (length($data) / 4), $data) : ();
 };
 
 } } # /eval /BEGIN