1 # Copyright (C) 2021 all contributors <meta@public-inbox.org>
2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
4 # Authentication worker for anything that needs auth for read/write IMAP
5 # (eventually for read-only NNTP access)
6 package PublicInbox::LeiAuth;
9 use parent qw(PublicInbox::IPC);
10 use PublicInbox::PktOp qw(pkt_do);
13 my ($lei, $net_new) = @_;
14 if ($lei->{pkt_op_p}) { # from lei_convert worker
15 pkt_do($lei->{pkt_op_p}, 'net_merge', $net_new);
16 } else { # single lei-daemon consumer
17 my $self = $lei->{auth} or return; # client disconnected
18 my $net = $self->{net};
19 %$net = (%$net, %$net_new);
23 sub do_auth_atfork { # used by IPC WQ workers
25 return if $wq->{-wq_worker_nr} != 0;
27 my $net = $self->{net};
28 my $mics = $net->imap_common_init($lei);
29 net_merge($lei, $net);
30 $net->{mics_cached} = $mics;
33 sub net_merge_done1 { # bump merge-count in top-level lei-daemon
35 return if ++$wq->{nr_net_merge_done} != $wq->{-wq_nr_workers};
36 $wq->net_merge_complete; # defined per wq-class (e.g. LeiImport)
39 sub net_merge_all { # called via wq_broadcast
40 my ($wq, $net_new) = @_;
41 my $net = $wq->{lei}->{net};
42 %$net = (%$net, %$net_new);
43 pkt_do($wq->{lei}->{pkt_op_p}, 'net_merge_done1') or
44 die "pkt_op_do net_merge_done1: $!";
47 # called by top-level lei-daemon when first worker is done with auth
48 sub net_merge_continue {
49 my ($wq, $net_new) = @_;
50 $wq->wq_broadcast('net_merge_all', $net_new);
53 sub op_merge { # prepares PktOp->pair ops
54 my ($self, $ops, $wq) = @_;
55 $ops->{net_merge} = [ \&net_merge_continue, $wq ];
56 $ops->{net_merge_done1} = [ \&net_merge_done1, $wq ];
60 my ($cls, $net) = @_; # net may be NetReader or descendant (NetWriter)
61 bless { net => $net }, $cls;