]> Sergey Matveev's repositories - public-inbox.git/commitdiff
lei_xsearch: cleanup {pkt_op_p} on exceptions
authorEric Wong <e@80x24.org>
Thu, 4 Mar 2021 09:03:13 +0000 (17:03 +0800)
committerEric Wong <e@80x24.org>
Thu, 4 Mar 2021 18:29:39 +0000 (14:29 -0400)
We must ensure pkt_op_p doesn't live beyond the scope of
->do_query in the top-level lei-daemon, otherwise it can leave a
stray socket hanging around in case of exceptions.

lib/PublicInbox/LeiXSearch.pm

index 458151801b8c1eaf6179325cba637501c416f603..059aa2844bfa724c1f765b1653c06fcc6b436429 100644 (file)
@@ -416,6 +416,11 @@ sub ipc_atfork_child {
        $self->SUPER::ipc_atfork_child;
 }
 
+sub delete_pkt_op { # OnDestroy callback
+       my $unclosed_after_die = delete($_[0])->{pkt_op_p} or return;
+       close $unclosed_after_die;
+}
+
 sub do_query {
        my ($self, $lei) = @_;
        my $l2m = $lei->{l2m};
@@ -431,6 +436,7 @@ sub do_query {
                'incr_start_query' => [ \&incr_start_query, $self, $l2m ],
        };
        $lei->{auth}->op_merge($ops, $l2m) if $l2m && $lei->{auth};
+       my $od = PublicInbox::OnDestroy->new($$, \&delete_pkt_op, $lei);
        ($lei->{pkt_op_c}, $lei->{pkt_op_p}) = PublicInbox::PktOp->pair($ops);
        $lei->{1}->autoflush(1);
        $lei->start_pager if delete $lei->{need_pager};