diff options
Diffstat (limited to 'lib/PublicInbox/LeiAuth.pm')
-rw-r--r-- | lib/PublicInbox/LeiAuth.pm | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/lib/PublicInbox/LeiAuth.pm b/lib/PublicInbox/LeiAuth.pm new file mode 100644 index 00000000..6593ba51 --- /dev/null +++ b/lib/PublicInbox/LeiAuth.pm @@ -0,0 +1,81 @@ +# Copyright (C) 2021 all contributors <meta@public-inbox.org> +# License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt> + +# Authentication worker for anything that needs auth for read/write IMAP +# (eventually for read-only NNTP access) +package PublicInbox::LeiAuth; +use strict; +use v5.10.1; +use parent qw(PublicInbox::IPC); +use PublicInbox::PktOp qw(pkt_do); +use PublicInbox::NetReader; + +sub nrd_merge { + my ($lei, $nrd_new) = @_; + if ($lei->{pkt_op_p}) { # from lei_convert worker + pkt_do($lei->{pkt_op_p}, 'nrd_merge', $nrd_new); + } else { # single lei-daemon consumer + my $self = $lei->{auth} or return; # client disconnected + my $nrd = $self->{nrd}; + %$nrd = (%$nrd, %$nrd_new); + } +} + +sub do_auth { # called via wq_io_do + my ($self) = @_; + my ($lei, $nrd) = @$self{qw(lei nrd)}; + $nrd->imap_common_init($lei); + nrd_merge($lei, $nrd); # tell lei-daemon updated auth info +} + +sub do_finish_auth { # dwaitpid callback + my ($arg, $pid) = @_; + my ($self, $lei, $post_auth_cb, @args) = @$arg; + $? ? $lei->dclose : $post_auth_cb->(@args); +} + +sub auth_eof { + my ($lei, $post_auth_cb, @args) = @_; + my $self = delete $lei->{auth} or return; + $self->wq_wait_old(\&do_finish_auth, $lei, $post_auth_cb, @args); +} + +sub auth_start { + my ($self, $lei, $post_auth_cb, @args) = @_; + $lei->_lei_cfg(1); # workers may need to read config + my $ops = { + '!' => [ $lei->can('fail_handler'), $lei ], + '|' => [ $lei->can('sigpipe_handler'), $lei ], + 'x_it' => [ $lei->can('x_it'), $lei ], + 'child_error' => [ $lei->can('child_error'), $lei ], + 'nrd_merge' => [ \&nrd_merge, $lei ], + '' => [ \&auth_eof, $lei, $post_auth_cb, @args ], + }; + ($lei->{pkt_op_c}, $lei->{pkt_op_p}) = PublicInbox::PktOp->pair($ops); + $self->wq_workers_start('lei_auth', 1, $lei->oldset, {lei => $lei}); + my $op = delete $lei->{pkt_op_c}; + delete $lei->{pkt_op_p}; + $self->wq_io_do('do_auth', []); + $self->wq_close(1); + $lei->event_step_init; # wait for shutdowns + if ($lei->{oneshot}) { + while ($op->{sock}) { $op->event_step } + } +} + +sub ipc_atfork_child { + my ($self) = @_; + # prevent {sock} from being closed in lei_atfork_child: + my $s = delete $self->{lei}->{sock}; + delete $self->{lei}->{auth}; # drop circular ref + $self->{lei}->lei_atfork_child; + $self->{lei}->{sock} = $s if $s; + $self->SUPER::ipc_atfork_child; +} + +sub new { + my ($cls, $nrd) = @_; + bless { nrd => $nrd }, $cls; +} + +1; |