]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/IMAP.pm
imap: rely on smsg->{bytes} for RFC822.SIZE
[public-inbox.git] / lib / PublicInbox / IMAP.pm
index 0d553da91f9651955234422172facb0ee634bb8a..9cdcba693c751fadcba634d403b6072e6ba68356 100644 (file)
@@ -16,7 +16,7 @@
 package PublicInbox::IMAP;
 use strict;
 use base qw(PublicInbox::DS);
-use fields qw(imapd logged_in ibx long_cb -login_tag
+use fields qw(imapd ibx long_cb -login_tag
        uid_min -idle_tag -idle_max);
 use PublicInbox::Eml;
 use PublicInbox::EmlContentFoo qw(parse_content_disposition);
@@ -27,6 +27,7 @@ use Text::ParseWords qw(parse_line);
 use Errno qw(EAGAIN);
 use Time::Local qw(timegm);
 use POSIX qw(strftime);
+use Hash::Util qw(unlock_hash); # dependency of fields for perl 5.10+, anyways
 
 my $Address;
 for my $mod (qw(Email::Address::XS Mail::Address)) {
@@ -52,7 +53,7 @@ my %FETCH_NEED = (
        'BODY[]' => [ NEED_BLOB, \&emit_rfc822 ],
        'RFC822.HEADER' => [ NEED_EML, \&emit_rfc822_header ],
        'RFC822.TEXT' => [ NEED_EML, \&emit_rfc822_text ],
-       'RFC822.SIZE' => [ NEED_BLOB, \&emit_rfc822_size ],
+       'RFC822.SIZE' => [ NEED_SMSG, \&emit_rfc822_size ],
        RFC822 => [ NEED_BLOB, \&emit_rfc822 ],
        BODY => [ NEED_EML, \&emit_body ],
        BODYSTRUCTURE => [ NEED_EML, \&emit_bodystructure ],
@@ -80,6 +81,9 @@ my @MoY = qw(Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec);
 my %MoY;
 @MoY{@MoY} = (0..11);
 
+# RFC 3501 5.4. Autologout Timer needs to be >= 30min
+$PublicInbox::DS::EXPTIME = 60 * 30;
+
 sub greet ($) {
        my ($self) = @_;
        my $capa = capa($self);
@@ -88,7 +92,8 @@ sub greet ($) {
 
 sub new ($$$) {
        my ($class, $sock, $imapd) = @_;
-       my $self = fields::new($class);
+       my $self = fields::new('PublicInbox::IMAP_preauth');
+       unlock_hash(%$self);
        my $ev = EPOLLIN;
        my $wbuf;
        if ($sock->can('accept_SSL') && !$sock->accept_SSL) {
@@ -107,13 +112,15 @@ sub new ($$$) {
        $self;
 }
 
+sub logged_in { 1 }
+
 sub capa ($) {
        my ($self) = @_;
 
        # dovecot advertises IDLE pre-login; perhaps because some clients
        # depend on it, so we'll do the same
        my $capa = 'CAPABILITY IMAP4rev1 IDLE';
-       if ($self->{logged_in}) {
+       if ($self->logged_in) {
                $capa .= ' COMPRESS=DEFLATE';
        } else {
                if (!($self->{sock} // $self)->can('accept_SSL') &&
@@ -126,7 +133,7 @@ sub capa ($) {
 
 sub login_success ($$) {
        my ($self, $tag) = @_;
-       $self->{logged_in} = 1;
+       bless $self, 'PublicInbox::IMAP';
        my $capa = capa($self);
        "$tag OK [$capa] Logged in\r\n";
 }
@@ -151,7 +158,7 @@ sub cmd_close ($$) {
 
 sub cmd_logout ($$) {
        my ($self, $tag) = @_;
-       delete $self->{logged_in};
+       delete $self->{-idle_tag};
        $self->write(\"* BYE logging out\r\n$tag OK Logout done\r\n");
        $self->shutdn; # PublicInbox::DS::shutdn
        undef;
@@ -174,23 +181,63 @@ sub cmd_noop ($$) { "$_[1] OK Noop done\r\n" }
 sub on_inbox_unlock {
        my ($self, $ibx) = @_;
        my $new = $ibx->mm->max;
+       my $uid_end = ($self->{uid_min} // 1) - 1 + UID_BLOCK;
        defined(my $old = $self->{-idle_max}) or die 'BUG: -idle_max unset';
+       $new = $uid_end if $new > $uid_end;
        if ($new > $old) {
                $self->{-idle_max} = $new;
                $self->msg_more("* $_ EXISTS\r\n") for (($old + 1)..($new - 1));
                $self->write(\"* $new EXISTS\r\n");
+       } elsif ($new == $uid_end) { # max exceeded $uid_end
+               # continue idling w/o inotify
+               delete $self->{-idle_max};
+               my $sock = $self->{sock} or return;
+               $ibx->unsubscribe_unlock(fileno($sock));
        }
 }
 
+# called every X minute(s) or so by PublicInbox::DS::later
+my $IDLERS = {};
+my $idle_timer;
+sub idle_tick_all {
+       my $old = $IDLERS;
+       $IDLERS = {};
+       for my $i (values %$old) {
+               next if ($i->{wbuf} || !exists($i->{-idle_tag}));
+               $i->update_idle_time or next;
+               $IDLERS->{fileno($i->{sock})} = $i;
+               $i->write(\"* OK Still here\r\n");
+       }
+       $idle_timer = scalar keys %$IDLERS ?
+                       PublicInbox::DS::later(\&idle_tick_all) : undef;
+}
+
 sub cmd_idle ($$) {
        my ($self, $tag) = @_;
        # IDLE seems allowed by dovecot w/o a mailbox selected *shrug*
        my $ibx = $self->{ibx} or return "$tag BAD no mailbox selected\r\n";
-       $ibx->subscribe_unlock(fileno($self->{sock}), $self);
-       $self->{imapd}->idler_start;
        $self->{-idle_tag} = $tag;
-       $self->{-idle_max} = $ibx->mm->max // 0;
-       "+ idling\r\n"
+       my $max = $ibx->mm->max // 0;
+       my $uid_end = ($self->{uid_min} // 1) - 1 + UID_BLOCK;
+       my $sock = $self->{sock} or return;
+       my $fd = fileno($sock);
+       # only do inotify on most recent slice
+       if ($max < $uid_end) {
+               $ibx->subscribe_unlock($fd, $self);
+               $self->{imapd}->idler_start;
+               $self->{-idle_max} = $max;
+       }
+       $idle_timer //= PublicInbox::DS::later(\&idle_tick_all);
+       $IDLERS->{$fd} = $self;
+       \"+ idling\r\n"
+}
+
+sub stop_idle ($$) {
+       my ($self, $ibx);
+       my $sock = $self->{sock} or return;
+       my $fd = fileno($sock);
+       delete $IDLERS->{$fd};
+       $ibx->unsubscribe_unlock($fd);
 }
 
 sub cmd_done ($$) {
@@ -201,7 +248,7 @@ sub cmd_done ($$) {
                warn "BUG: idle_tag set w/o inbox";
                return "$tag BAD internal bug\r\n";
        };
-       $ibx->unsubscribe_unlock(fileno($self->{sock}));
+       stop_idle($self, $ibx);
        "$idle_tag OK Idle done\r\n";
 }
 
@@ -453,11 +500,12 @@ sub emit_rfc822 {
        $self->msg_more($$bref);
 }
 
-# Mail::IMAPClient::message_string cares about this by default
-# (->Ignoresizeerrors attribute)
+# Mail::IMAPClient::message_string cares about this by default,
+# (->Ignoresizeerrors attribute).  Admins are encouraged to
+# --reindex for IMAP support, anyways.
 sub emit_rfc822_size {
-       my ($self, $k, undef, $bref) = @_;
-       $self->msg_more(' RFC822.SIZE ' . length($$bref));
+       my ($self, $k, $smsg) = @_;
+       $self->msg_more(' RFC822.SIZE ' . $smsg->{bytes});
 }
 
 sub emit_internaldate {
@@ -1185,9 +1233,7 @@ sub busy {
 sub close {
        my ($self) = @_;
        if (my $ibx = delete $self->{ibx}) {
-               if (my $sock = $self->{sock}) {;
-                       $ibx->unsubscribe_unlock(fileno($sock));
-               }
+               stop_idle($self, $ibx);
        }
        $self->SUPER::close; # PublicInbox::DS::close
 }
@@ -1197,4 +1243,9 @@ no warnings 'once';
 *cmd_select = \&cmd_examine;
 *cmd_fetch = \&cmd_uid_fetch;
 
+package PublicInbox::IMAP_preauth;
+our @ISA = qw(PublicInbox::IMAP);
+
+sub logged_in { 0 }
+
 1;