]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/Watch.pm
net_reader: use and accept URIimap objects in more places
[public-inbox.git] / lib / PublicInbox / Watch.pm
index bc296e01fc3ac7baec4e50dc082e7aef59457a2e..8d13ea352fb54804bbc7208e67970bd646761223 100644 (file)
@@ -1,17 +1,19 @@
-# Copyright (C) 2016-2020 all contributors <meta@public-inbox.org>
+# Copyright (C) 2016-2021 all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 #
 # ref: https://cr.yp.to/proto/maildir.html
-#      http://wiki2.dovecot.org/MailboxFormat/Maildir
+#      httsp://wiki2.dovecot.org/MailboxFormat/Maildir
 package PublicInbox::Watch;
 use strict;
 use v5.10.1;
 use PublicInbox::Eml;
-use PublicInbox::InboxWritable qw(eml_from_path warn_ignore_cb);
+use PublicInbox::InboxWritable qw(eml_from_path);
+use PublicInbox::MdirReader;
+use PublicInbox::NetReader;
 use PublicInbox::Filter::Base qw(REJECT);
 use PublicInbox::Spamcheck;
 use PublicInbox::Sigfd;
-use PublicInbox::DS qw(now);
+use PublicInbox::DS qw(now add_timer);
 use PublicInbox::MID qw(mids);
 use PublicInbox::ContentHash qw(content_hash);
 use PublicInbox::EOFpipe;
@@ -44,6 +46,7 @@ sub new {
        my ($class, $cfg) = @_;
        my (%mdmap, $spamc);
        my (%imap, %nntp); # url => [inbox objects] or 'watchspam'
+       my (@imap, @nntp);
 
        # "publicinboxwatch" is the documented namespace
        # "publicinboxlearn" is legacy but may be supported
@@ -57,10 +60,12 @@ sub new {
                        if (is_maildir($dir)) {
                                # skip "new", no MUA has seen it, yet.
                                $mdmap{"$dir/cur"} = 'watchspam';
-                       } elsif ($url = imap_url($dir)) {
-                               $imap{$url} = 'watchspam';
+                       } elsif (my $uri = imap_uri($dir)) {
+                               $imap{$$uri} = 'watchspam';
+                               push @imap, $uri;
                        } elsif ($url = nntp_url($dir)) {
                                $nntp{$url} = 'watchspam';
+                               push @nntp, $url;
                        } else {
                                warn "unsupported $k=$dir\n";
                        }
@@ -87,14 +92,17 @@ sub new {
                                return if is_watchspam($cur, $cur_dst, $ibx);
                                push @{$mdmap{$new} //= []}, $ibx;
                                push @$cur_dst, $ibx;
-                       } elsif ($url = imap_url($watch)) {
+                       } elsif (my $uri = imap_uri($watch)) {
+                               my $url = $$uri;
                                return if is_watchspam($url, $imap{$url}, $ibx);
                                compile_watchheaders($ibx);
-                               push @{$imap{$url} ||= []}, $ibx;
+                               my $n = push @{$imap{$url} ||= []}, $ibx;
+                               push @imap, $uri if $n == 1;
                        } elsif ($url = nntp_url($watch)) {
                                return if is_watchspam($url, $nntp{$url}, $ibx);
                                compile_watchheaders($ibx);
-                               push @{$nntp{$url} ||= []}, $ibx;
+                               my $n = push @{$nntp{$url} ||= []}, $ibx;
+                               push @nntp, $url if $n == 1;
                        } else {
                                warn "watch unsupported: $k=$watch\n";
                        }
@@ -116,6 +124,8 @@ sub new {
                pi_cfg => $cfg,
                imap => scalar keys %imap ? \%imap : undef,
                nntp => scalar keys %nntp? \%nntp : undef,
+               imap_order => scalar(@imap) ? \@imap : undef,
+               nntp_order => scalar(@nntp) ? \@nntp: undef,
                importers => {},
                opendirs => {}, # dirname => dirhandle (in progress scans)
                ops => [], # 'quit', 'full'
@@ -174,7 +184,7 @@ sub _remove_spam {
        # path must be marked as (S)een
        $path =~ /:2,[A-R]*S[T-Za-z]*\z/ or return;
        my $eml = eml_from_path($path) or return;
-       local $SIG{__WARN__} = warn_ignore_cb();
+       local $SIG{__WARN__} = PublicInbox::Eml::warn_ignore_cb();
        $self->{pi_cfg}->each_inbox(\&remove_eml_i, $self, $eml, $path);
 }
 
@@ -207,7 +217,8 @@ sub import_eml ($$$) {
 
 sub _try_path {
        my ($self, $path) = @_;
-       return unless PublicInbox::InboxWritable::is_maildir_path($path);
+       my $fl = PublicInbox::MdirReader::maildir_path_flags($path) // return;
+       return if $fl =~ /[DT]/; # no Drafts or Trash
        if ($path !~ $self->{mdre}) {
                warn "unrecognized path: $path\n";
                return;
@@ -277,131 +288,6 @@ sub watch_fs_init ($) {
        PublicInbox::DirIdle->new([keys %{$self->{mdmap}}], $cb);
 }
 
-# avoid exposing deprecated "snews" to users.
-my %SCHEME_MAP = ('snews' => 'nntps');
-
-sub uri_scheme ($) {
-       my ($uri) = @_;
-       my $scheme = $uri->scheme;
-       $SCHEME_MAP{$scheme} // $scheme;
-}
-
-# returns the git config section name, e.g [imap "imaps://user@example.com"]
-# without the mailbox, so we can share connections between different inboxes
-sub uri_section ($) {
-       my ($uri) = @_;
-       uri_scheme($uri) . '://' . $uri->authority;
-}
-
-sub cfg_intvl ($$$) {
-       my ($cfg, $key, $url) = @_;
-       my $v = $cfg->urlmatch($key, $url) // return;
-       $v =~ /\A[0-9]+(?:\.[0-9]+)?\z/s and return $v + 0;
-       if (ref($v) eq 'ARRAY') {
-               $v = join(', ', @$v);
-               warn "W: $key has multiple values: $v\nW: $key ignored\n";
-       } else {
-               warn "W: $key=$v is not a numeric value in seconds\n";
-       }
-}
-
-sub cfg_bool ($$$) {
-       my ($cfg, $key, $url) = @_;
-       my $orig = $cfg->urlmatch($key, $url) // return;
-       my $bool = $cfg->git_bool($orig);
-       warn "W: $key=$orig for $url is not boolean\n" unless defined($bool);
-       $bool;
-}
-
-# flesh out common IMAP-specific data structures
-sub imap_common_init ($) {
-       my ($self) = @_;
-       my $cfg = $self->{pi_cfg};
-       my $mic_args = {}; # scheme://authority => Mail:IMAPClient arg
-       for my $url (sort keys %{$self->{imap}}) {
-               my $uri = PublicInbox::URIimap->new($url);
-               my $sec = uri_section($uri);
-               for my $k (qw(Starttls Debug Compress)) {
-                       my $bool = cfg_bool($cfg, "imap.$k", $url) // next;
-                       $mic_args->{$sec}->{$k} = $bool;
-               }
-               my $to = cfg_intvl($cfg, 'imap.timeout', $url);
-               $mic_args->{$sec}->{Timeout} = $to if $to;
-               for my $k (qw(pollInterval idleInterval)) {
-                       $to = cfg_intvl($cfg, "imap.$k", $url) // next;
-                       $self->{imap_opt}->{$sec}->{$k} = $to;
-               }
-               my $k = 'imap.fetchBatchSize';
-               my $bs = $cfg->urlmatch($k, $url) // next;
-               if ($bs =~ /\A([0-9]+)\z/) {
-                       $self->{imap_opt}->{$sec}->{batch_size} = $bs;
-               } else {
-                       warn "$k=$bs is not an integer\n";
-               }
-       }
-       $mic_args;
-}
-
-sub auth_anon_cb { '' }; # for Mail::IMAPClient::Authcallback
-
-sub mic_for ($$$) { # mic = Mail::IMAPClient
-       my ($self, $url, $mic_args) = @_;
-       my $uri = PublicInbox::URIimap->new($url);
-       require PublicInbox::GitCredential;
-       my $cred = bless {
-               url => $url,
-               protocol => $uri->scheme,
-               host => $uri->host,
-               username => $uri->user,
-               password => $uri->password,
-       }, 'PublicInbox::GitCredential';
-       my $common = $mic_args->{uri_section($uri)} // {};
-       # IMAPClient and Net::Netrc both mishandles `0', so we pass `127.0.0.1'
-       my $host = $cred->{host};
-       $host = '127.0.0.1' if $host eq '0';
-       my $mic_arg = {
-               Port => $uri->port,
-               Server => $host,
-               Ssl => $uri->scheme eq 'imaps',
-               Keepalive => 1, # SO_KEEPALIVE
-               %$common, # may set Starttls, Compress, Debug ....
-       };
-       my $mic = PublicInbox::IMAPClient->new(%$mic_arg) or
-               die "E: <$url> new: $@\n";
-
-       # default to using STARTTLS if it's available, but allow
-       # it to be disabled since I usually connect to localhost
-       if (!$mic_arg->{Ssl} && !defined($mic_arg->{Starttls}) &&
-                       $mic->has_capability('STARTTLS') &&
-                       $mic->can('starttls')) {
-               $mic->starttls or die "E: <$url> STARTTLS: $@\n";
-       }
-
-       # do we even need credentials?
-       if (!defined($cred->{username}) &&
-                       $mic->has_capability('AUTH=ANONYMOUS')) {
-               $cred = undef;
-       }
-       if ($cred) {
-               $cred->check_netrc unless defined $cred->{password};
-               $cred->fill; # may prompt user here
-               $mic->User($mic_arg->{User} = $cred->{username});
-               $mic->Password($mic_arg->{Password} = $cred->{password});
-       } else { # AUTH=ANONYMOUS
-               $mic->Authmechanism($mic_arg->{Authmechanism} = 'ANONYMOUS');
-               $mic->Authcallback($mic_arg->{Authcallback} = \&auth_anon_cb);
-       }
-       if ($mic->login && $mic->IsAuthenticated) {
-               # success! keep IMAPClient->new arg in case we get disconnected
-               $self->{mic_arg}->{uri_section($uri)} = $mic_arg;
-       } else {
-               warn "E: <$url> LOGIN: $@\n";
-               $mic = undef;
-       }
-       $cred->run($mic ? 'approve' : 'reject') if $cred;
-       $mic;
-}
-
 sub imap_import_msg ($$$$$) {
        my ($self, $url, $uid, $raw, $flags) = @_;
        # our target audience expects LF-only, save storage
@@ -414,13 +300,11 @@ sub imap_import_msg ($$$$$) {
                        import_eml($self, $ibx, $eml);
                }
        } elsif ($inboxes eq 'watchspam') {
-               # we don't remove unseen messages
-               if ($flags =~ /\\Seen\b/) {
-                       local $SIG{__WARN__} = warn_ignore_cb();
-                       my $eml = PublicInbox::Eml->new($raw);
-                       $self->{pi_cfg}->each_inbox(\&remove_eml_i,
+               return if $flags !~ /\\Seen\b/; # don't remove unseen messages
+               local $SIG{__WARN__} = PublicInbox::Eml::warn_ignore_cb();
+               my $eml = PublicInbox::Eml->new($raw);
+               $self->{pi_cfg}->each_inbox(\&remove_eml_i,
                                                $self, $eml, "$url UID:$uid");
-               }
        } else {
                die "BUG: destination unknown $inboxes";
        }
@@ -583,17 +467,17 @@ sub watch_atfork_child ($) {
        delete $self->{opendirs};
        PublicInbox::DS->Reset;
        %SIG = (%SIG, %{$self->{sig}}, CHLD => 'DEFAULT');
-       PublicInbox::Sigfd::sig_setmask($self->{oldset});
+       PublicInbox::DS::sig_setmask($self->{oldset});
 }
 
 sub watch_atfork_parent ($) {
        my ($self) = @_;
        _done_for_now($self);
-       PublicInbox::Sigfd::block_signals();
+       PublicInbox::DS::block_signals();
 }
 
-sub imap_idle_requeue ($) { # DS::add_timer callback
-       my ($self, $url_intvl) = @{$_[0]};
+sub imap_idle_requeue { # DS::add_timer callback
+       my ($self, $url_intvl) = @_;
        return if $self->{quit};
        push @{$self->{idle_todo}}, $url_intvl;
        event_step($self);
@@ -607,8 +491,7 @@ sub imap_idle_reap { # PublicInbox::DS::dwaitpid callback
        my ($url, $intvl) = @$url_intvl;
        return if $self->{quit};
        warn "W: PID=$pid on $url died: \$?=$?\n" if $?;
-       PublicInbox::DS::add_timer(60,
-                               \&imap_idle_requeue, [ $self, $url_intvl ]);
+       add_timer(60, \&imap_idle_requeue, $self, $url_intvl);
 }
 
 sub reap { # callback for EOFpipe
@@ -625,8 +508,11 @@ sub imap_idle_fork ($$) {
        my ($self, $url_intvl) = @_;
        my ($url, $intvl) = @$url_intvl;
        pipe(my ($r, $w)) or die "pipe: $!";
-       defined(my $pid = fork) or die "fork: $!";
+       my $seed = rand(0xffffffff);
+       my $pid = fork // die "fork: $!";
        if ($pid == 0) {
+               srand($seed);
+               eval { Net::SSLeay::randomize() };
                close $r;
                watch_atfork_child($self);
                watch_imap_idle_1($self, $url, $intvl);
@@ -648,7 +534,7 @@ sub event_step {
                                imap_idle_fork($self, $url_intvl);
                        }
                };
-               PublicInbox::Sigfd::sig_setmask($oldset);
+               PublicInbox::DS::sig_setmask($oldset);
                die $@ if $@;
        }
        fs_scan_step($self) if $self->{mdre};
@@ -699,13 +585,16 @@ sub watch_nntp_fetch_all ($$) {
        }
 }
 
-sub poll_fetch_fork ($) { # DS::add_timer callback
-       my ($self, $intvl, $urls) = @{$_[0]};
+sub poll_fetch_fork { # DS::add_timer callback
+       my ($self, $intvl, $urls) = @_;
        return if $self->{quit};
        pipe(my ($r, $w)) or die "pipe: $!";
        my $oldset = watch_atfork_parent($self);
+       my $seed = rand(0xffffffff);
        my $pid = fork;
        if (defined($pid) && $pid == 0) {
+               srand($seed);
+               eval { Net::SSLeay::randomize() };
                close $r;
                watch_atfork_child($self);
                if ($urls->[0] =~ m!\Aimaps?://!i) {
@@ -716,7 +605,7 @@ sub poll_fetch_fork ($) { # DS::add_timer callback
                close $w;
                _exit(0);
        }
-       PublicInbox::Sigfd::sig_setmask($oldset);
+       PublicInbox::DS::sig_setmask($oldset);
        die "fork: $!"  unless defined $pid;
        $self->{poll_pids}->{$pid} = [ $intvl, $urls ];
        PublicInbox::EOFpipe->new($r, \&reap, [$pid, \&poll_fetch_reap, $self]);
@@ -732,27 +621,12 @@ sub poll_fetch_reap {
                warn "W: PID=$pid died: \$?=$?\n", map { "$_\n" } @$urls;
        }
        warn("I: will check $_ in ${intvl}s\n") for @$urls;
-       PublicInbox::DS::add_timer($intvl, \&poll_fetch_fork,
-                                       [$self, $intvl, $urls]);
+       add_timer($intvl, \&poll_fetch_fork, $self, $intvl, $urls);
 }
 
 sub watch_imap_init ($$) {
        my ($self, $poll) = @_;
-       eval { require PublicInbox::IMAPClient } or
-               die "Mail::IMAPClient is required for IMAP:\n$@\n";
-       eval { require PublicInbox::IMAPTracker } or
-               die "DBD::SQLite is required for IMAP\n:$@\n";
-
-       my $mic_args = imap_common_init($self); # read args from config
-
-       # make sure we can connect and cache the credentials in memory
-       $self->{mic_arg} = {}; # schema://authority => IMAPClient->new args
-       my $mics = {}; # schema://authority => IMAPClient obj
-       for my $url (sort keys %{$self->{imap}}) {
-               my $uri = PublicInbox::URIimap->new($url);
-               $mics->{uri_section($uri)} //= mic_for($self, $url, $mic_args);
-       }
-
+       my $mics = imap_common_init($self); # read args from config
        my $idle = []; # [ [ url1, intvl1 ], [url2, intvl2] ]
        for my $url (keys %{$self->{imap}}) {
                my $uri = PublicInbox::URIimap->new($url);
@@ -777,7 +651,7 @@ sub nntp_common_init ($) {
        my ($self) = @_;
        my $cfg = $self->{pi_cfg};
        my $nn_args = {}; # scheme://authority => Net::NNTP->new arg
-       for my $url (sort keys %{$self->{nntp}}) {
+       for my $url (@{$self->{nntp_order}}) {
                my $sec = uri_section(uri_new($url));
 
                # Debug and Timeout are passed to Net::NNTP->new
@@ -801,106 +675,6 @@ sub nntp_common_init ($) {
        $nn_args;
 }
 
-# Net::NNTP doesn't support CAPABILITIES, yet
-sub try_starttls ($) {
-       my ($host) = @_;
-       return if $host =~ /\.onion\z/s;
-       return if $host =~ /\A127\.[0-9]+\.[0-9]+\.[0-9]+\z/s;
-       return if $host eq '::1';
-       1;
-}
-
-sub nn_new ($$$) {
-       my ($nn_arg, $nntp_opt, $url) = @_;
-       my $nn = Net::NNTP->new(%$nn_arg) or die "E: <$url> new: $!\n";
-
-       # default to using STARTTLS if it's available, but allow
-       # it to be disabled for localhost/VPN users
-       if (!$nn_arg->{SSL} && $nn->can('starttls')) {
-               if (!defined($nntp_opt->{starttls}) &&
-                               try_starttls($nn_arg->{Host})) {
-                       # soft fail by default
-                       $nn->starttls or warn <<"";
-W: <$url> STARTTLS tried and failed (not requested)
-
-               } elsif ($nntp_opt->{starttls}) {
-                       # hard fail if explicitly configured
-                       $nn->starttls or die <<"";
-E: <$url> STARTTLS requested and failed
-
-               }
-       } elsif ($nntp_opt->{starttls}) {
-               $nn->can('starttls') or
-                       die "E: <$url> Net::NNTP too old for STARTTLS\n";
-               $nn->starttls or die <<"";
-E: <$url> STARTTLS requested and failed
-
-       }
-       $nn;
-}
-
-sub nn_for ($$$) { # nn = Net::NNTP
-       my ($self, $url, $nn_args) = @_;
-       my $uri = uri_new($url);
-       my $sec = uri_section($uri);
-       my $nntp_opt = $self->{nntp_opt}->{$sec} //= {};
-       my $host = $uri->host;
-       # Net::NNTP and Net::Netrc both mishandle `0', so we pass `127.0.0.1'
-       $host = '127.0.0.1' if $host eq '0';
-       my $cred;
-       my ($u, $p);
-       if (defined(my $ui = $uri->userinfo)) {
-               require PublicInbox::GitCredential;
-               $cred = bless {
-                       url => $sec,
-                       protocol => uri_scheme($uri),
-                       host => $host,
-               }, 'PublicInbox::GitCredential';
-               ($u, $p) = split(/:/, $ui, 2);
-               ($cred->{username}, $cred->{password}) = ($u, $p);
-               $cred->check_netrc unless defined $p;
-       }
-       my $common = $nn_args->{$sec} // {};
-       my $nn_arg = {
-               Port => $uri->port,
-               Host => $host,
-               SSL => $uri->secure, # snews == nntps
-               %$common, # may Debug ....
-       };
-       my $nn = nn_new($nn_arg, $nntp_opt, $url);
-
-       if ($cred) {
-               $cred->fill; # may prompt user here
-               if ($nn->authinfo($u, $p)) {
-                       push @{$nntp_opt->{-postconn}}, [ 'authinfo', $u, $p ];
-               } else {
-                       warn "E: <$url> AUTHINFO $u XXXX failed\n";
-                       $nn = undef;
-               }
-       }
-
-       if ($nntp_opt->{compress}) {
-               # https://rt.cpan.org/Ticket/Display.html?id=129967
-               if ($nn->can('compress')) {
-                       if ($nn->compress) {
-                               push @{$nntp_opt->{-postconn}}, [ 'compress' ];
-                       } else {
-                               warn "W: <$url> COMPRESS failed\n";
-                       }
-               } else {
-                       delete $nntp_opt->{compress};
-                       warn <<"";
-W: <$url> COMPRESS not supported by Net::NNTP
-W: see https://rt.cpan.org/Ticket/Display.html?id=129967 for updates
-
-               }
-       }
-
-       $self->{nn_arg}->{$sec} = $nn_arg;
-       $cred->run($nn ? 'approve' : 'reject') if $cred;
-       $nn;
-}
-
 sub nntp_fetch_all ($$$) {
        my ($self, $nn, $url) = @_;
        my $uri = uri_new($url);
@@ -989,10 +763,10 @@ sub watch_nntp_init ($$) {
 
        # make sure we can connect and cache the credentials in memory
        $self->{nn_arg} = {}; # schema://authority => Net::NNTP->new args
-       for my $url (sort keys %{$self->{nntp}}) {
+       for my $url (@{$self->{nntp_order}}) {
                nn_for($self, $url, $nn_args);
        }
-       for my $url (keys %{$self->{nntp}}) {
+       for my $url (@{$self->{nntp_order}}) {
                my $uri = uri_new($url);
                my $sec = uri_section($uri);
                my $intvl = $self->{nntp_opt}->{$sec}->{pollInterval};
@@ -1009,8 +783,7 @@ sub watch { # main entry point
        watch_nntp_init($self, $poll) if $self->{nntp};
        while (my ($intvl, $urls) = each %$poll) {
                # poll all URLs for a given interval sequentially
-               PublicInbox::DS::add_timer(0, \&poll_fetch_fork,
-                                               [$self, $intvl, $urls]);
+               add_timer(0, \&poll_fetch_fork, $self, $intvl, $urls);
        }
        watch_fs_init($self) if $self->{mdre};
        PublicInbox::DS->SetPostLoopCallback(sub { !$self->quit_done });
@@ -1129,31 +902,4 @@ EOF
        undef;
 }
 
-sub uri_new {
-       my ($url) = @_;
-
-       # URI::snews exists, URI::nntps does not, so use URI::snews
-       $url =~ s!\Anntps://!snews://!i;
-       URI->new($url);
-}
-
-sub imap_url {
-       my ($url) = @_;
-       require PublicInbox::URIimap;
-       my $uri = PublicInbox::URIimap->new($url);
-       $uri ? $uri->canonical->as_string : undef;
-}
-
-my %IS_NNTP = (news => 1, snews => 1, nntp => 1);
-sub nntp_url {
-       my ($url) = @_;
-       require URI;
-       my $uri = uri_new($url);
-       return unless $uri && $IS_NNTP{$uri->scheme} && $uri->group;
-       $url = $uri->canonical->as_string;
-       # nntps is IANA registered, snews is deprecated
-       $url =~ s!\Asnews://!nntps://!;
-       $url;
-}
-
 1;