]> Sergey Matveev's repositories - public-inbox.git/blob - lib/PublicInbox/ProcessPipe.pm
ds: support greeting protocols
[public-inbox.git] / lib / PublicInbox / ProcessPipe.pm
1 # Copyright (C) 2016-2021 all contributors <meta@public-inbox.org>
2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
3
4 # a tied handle for auto reaping of children tied to a pipe, see perltie(1)
5 package PublicInbox::ProcessPipe;
6 use strict;
7 use v5.10.1;
8 use Carp qw(carp);
9
10 sub TIEHANDLE {
11         my ($class, $pid, $fh, $cb, $arg) = @_;
12         bless { pid => $pid, fh => $fh, ppid => $$, cb => $cb, arg => $arg },
13                 $class;
14 }
15
16 sub BINMODE { binmode(shift->{fh}) } # for IO::Uncompress::Gunzip
17
18 sub READ { read($_[0]->{fh}, $_[1], $_[2], $_[3] || 0) }
19
20 sub READLINE { readline($_[0]->{fh}) }
21
22 sub WRITE {
23         use bytes qw(length);
24         syswrite($_[0]->{fh}, $_[1], $_[2] // length($_[1]), $_[3] // 0);
25 }
26
27 sub PRINT {
28         my $self = shift;
29         print { $self->{fh} } @_;
30 }
31
32 sub FILENO { fileno($_[0]->{fh}) }
33
34 sub _close ($;$) {
35         my ($self, $wait) = @_;
36         my $fh = delete $self->{fh};
37         my $ret = defined($fh) ? close($fh) : '';
38         my ($pid, $cb, $arg) = delete @$self{qw(pid cb arg)};
39         return $ret unless defined($pid) && $self->{ppid} == $$;
40         if ($wait) { # caller cares about the exit status:
41                 my $wp = waitpid($pid, 0);
42                 if ($wp == $pid) {
43                         $ret = '' if $?;
44                         if ($cb) {
45                                 eval { $cb->($arg, $pid) };
46                                 carp "E: cb(arg, $pid): $@" if $@;
47                         }
48                 } else {
49                         carp "waitpid($pid, 0) = $wp, \$!=$!, \$?=$?";
50                 }
51         } else { # caller just undef-ed it, let event loop deal with it
52                 require PublicInbox::DS;
53                 PublicInbox::DS::dwaitpid($pid, $cb, $arg);
54         }
55         $ret;
56 }
57
58 # if caller uses close(), assume they want to check $? immediately so
59 # we'll waitpid() synchronously.  n.b. wantarray doesn't seem to
60 # propagate `undef' down to tied methods, otherwise I'd rely on that.
61 sub CLOSE { _close($_[0], 1) }
62
63 # if relying on DESTROY, assume the caller doesn't care about $? and
64 # we can let the event loop call waitpid() whenever it gets SIGCHLD
65 sub DESTROY {
66         _close($_[0]);
67         undef;
68 }
69
70 1;