]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/NNTP.pm
nntp: fix XOVER command
[public-inbox.git] / lib / PublicInbox / NNTP.pm
index bc8d6eda63b972b781ce2e05b897e4c38be294a8..0d0de976f562454c0c3443eed3c95cbbf6a9f6b7 100644 (file)
@@ -14,6 +14,9 @@ use POSIX qw(strftime);
 use Time::HiRes qw(clock_gettime ualarm CLOCK_MONOTONIC);
 use constant {
        r501 => '501 command syntax error',
+       r221 => '221 Header follows',
+       r224 => '224 Overview information follows (multi-line)',
+       r430 => '430 No article with that message-id',
        long_response_limit => 0xffffffff,
 };
 
@@ -346,7 +349,7 @@ sub art_lookup ($$$) {
                        goto find_mid;
                } elsif ($art =~ /\A<([^>]+)>\z/) {
                        $mid = $1;
-                       $err = '430 no such article found';
+                       $err = r430;
                        $n = $ng->mm->num_for($mid) if $ng;
                        goto found if defined $n;
                        foreach my $g (values %{$self->{nntpd}->{groups}}) {
@@ -374,18 +377,16 @@ found:
        my $bytes;
        my $s = eval { Email::MIME->new($ng->gcf->cat_file($o, \$bytes)) };
        return $err unless $s;
+       my $lines;
        if ($set_headers) {
                $s->header_set('Newsgroups', $ng->{name});
-               $s->header_set('Lines', $s->body =~ tr!\n!\n!);
                $s->header_set('Xref', xref($ng, $n));
+               $lines = $s->body =~ tr!\n!\n!;
 
                # must be last
-               if ($set_headers == 2) {
-                       $s->header_set('Bytes', $bytes);
-                       $s->body_set('');
-               }
+               $s->body_set('') if ($set_headers == 2);
        }
-       [ $n, $mid, $s ];
+       [ $n, $mid, $s, $bytes, $lines, $ng ];
 }
 
 sub simple_body_write ($$) {
@@ -393,17 +394,12 @@ sub simple_body_write ($$) {
        my $body = $s->body;
        $s->body_set('');
        $body =~ s/^\./../smg;
+       $body =~ s/(?<!\r)\n/\r\n/sg;
        do_more($self, $body);
+       do_more($self, "\r\n") unless $body =~ /\r\n\z/s;
        '.'
 }
 
-sub header_str ($) {
-       my ($s) = @_;
-       my $h = $s->header_obj;
-       $h->header_set('Bytes');
-       $h->as_string
-}
-
 sub set_art {
        my ($self, $art) = @_;
        $self->{article} = $art if defined $art && $art =~ /\A\d+\z/;
@@ -416,7 +412,7 @@ sub cmd_article ($;$) {
        my ($n, $mid, $s) = @$r;
        set_art($self, $art);
        more($self, "220 $n <$mid> article retrieved - head and body follow");
-       do_more($self, header_str($s));
+       do_more($self, $s->header_obj->as_string);
        do_more($self, "\r\n");
        simple_body_write($self, $s);
 }
@@ -428,7 +424,7 @@ sub cmd_head ($;$) {
        my ($n, $mid, $s) = @$r;
        set_art($self, $art);
        more($self, "221 $n <$mid> article retrieved - head follows");
-       do_more($self, header_str($s));
+       do_more($self, $s->header_obj->as_string);
        '.'
 }
 
@@ -482,8 +478,11 @@ sub get_range ($$) {
        [ $beg, $end ];
 }
 
-sub xhdr ($$) {
+sub hdr_val ($$) {
        my ($r, $header) = @_;
+       $header = lc $header;
+       return $r->[3] if ($header eq 'bytes');
+       return $r->[4] if ($header eq 'lines');
        $r = $r->[2]->header_obj->header($header);
        defined $r or return;
        $r =~ s/[\r\n\t]+/ /sg;
@@ -545,19 +544,18 @@ sub long_response ($$$$) {
        undef;
 }
 
-sub xhdr_message_id ($$) { # optimize XHDR Message-ID [range] for slrnpull.
-       my ($self, $range) = @_;
+sub hdr_message_id ($$$) { # optimize XHDR Message-ID [range] for slrnpull.
+       my ($self, $xhdr, $range) = @_;
 
-       my $mm = $self->{ng}->mm;
        if (defined $range && $range =~ /\A<(.+)>\z/) { # Message-ID
-               my $n = $mm->num_for($range);
-               more($self, '221 Header follows');
-               more($self, "<$range> <$range>") if defined $n;
-               '.';
+               my ($ng, $n) = mid_lookup($self, $1);
+               return r430 unless $n;
+               hdr_mid_response($self, $xhdr, $ng, $n, $range, $range);
        } else { # numeric range
                $range = $self->{article} unless defined $range;
                my $r = get_range($self, $range);
                return $r unless ref $r;
+               my $mm = $self->{ng}->mm;
                my ($beg, $end) = @$r;
                more($self, '221 Header follows');
                $self->long_response($beg, $end, sub {
@@ -573,20 +571,34 @@ sub xref ($$) {
        "$ng->{domain} $ng->{name}:$n"
 }
 
-sub xhdr_xref ($$) { # optimize XHDR Xref [range] for rtin
-       my ($self, $range) = @_;
+sub mid_lookup ($$) {
+       my ($self, $mid) = @_;
+       my $self_ng = $self->{ng};
+       if ($self_ng) {
+               my $n = $self_ng->mm->num_for($mid);
+               return ($self_ng, $n) if defined $n;
+       }
+       foreach my $ng (values %{$self->{nntpd}->{groups}}) {
+               next if defined $self_ng && $ng eq $self_ng;
+               my $n = $ng->mm->num_for($mid);
+               return ($ng, $n) if defined $n;
+       }
+       (undef, undef);
+}
+
+sub hdr_xref ($$$) { # optimize XHDR Xref [range] for rtin
+       my ($self, $xhdr, $range) = @_;
 
-       my $ng = $self->{ng};
-       my $mm = $ng->mm;
        if (defined $range && $range =~ /\A<(.+)>\z/) { # Message-ID
-               my $n = $mm->num_for($range);
-               more($self, '221 Header follows');
-               more($self, "<$range> ".xref($ng, $n)) if defined $n;
-               '.';
+               my ($ng, $n) = mid_lookup($self, $1);
+               return r430 unless $n;
+               hdr_mid_response($self, $xhdr, $ng, $n, $range, xref($ng, $n));
        } else { # numeric range
                $range = $self->{article} unless defined $range;
                my $r = get_range($self, $range);
                return $r unless ref $r;
+               my $ng = $self->{ng};
+               my $mm = $ng->mm;
                my ($beg, $end) = @$r;
                more($self, '221 Header follows');
                $self->long_response($beg, $end, sub {
@@ -606,26 +618,31 @@ sub header_obj_for {
        };
 };
 
-sub xhdr_searchmsg ($$$) {
-       my ($self, $sub, $range) = @_;
-       my $srch = $self->{ng}->search;
-       my $emit = ($sub eq 'date') ? sub {
-               my ($pfx, $m) = @_;
-               my @t = gmtime($m->header('X-PI-TS'));
-               more($self, "$pfx ". strftime('%a, %d %b %Y %T %z', @t));
-       } : sub {
-               my ($pfx, $m) = @_;
-               my $h = $m->header($sub);
-               more($self, "$pfx $h") if defined $h;
-       };
+sub hdr_searchmsg ($$$$) {
+       my ($self, $xhdr, $hdr, $range) = @_;
+       my $filter;
+       if ($hdr eq 'date') {
+               $hdr = 'X-PI-TS';
+               $filter = sub ($) {
+                       strftime('%a, %d %b %Y %T %z', gmtime($_[0]));
+               };
+       }
 
        if (defined $range && $range =~ /\A<(.+)>\z/) { # Message-ID
-               more($self, '221 Header follows');
-               my $m = header_obj_for($srch, $1);
-               $emit->($range, $m) if defined $m;
-               '.';
+               my ($ng, $n) = mid_lookup($self, $1);
+               return r430 unless $n;
+               if (my $srch = $ng->search) {
+                       my $m = header_obj_for($srch, $range);
+                       my $v = $m->header($hdr);
+                       $v = $filter->($v) if defined $v && $filter;
+                       hdr_mid_response($self, $xhdr, $ng, $n, $range, $v);
+               } else {
+                       hdr_slow($self, $xhdr, $hdr, $range);
+               }
        } else { # numeric range
                $range = $self->{article} unless defined $range;
+               my $srch = $self->{ng}->search or
+                               return hdr_slow($self, $xhdr, $hdr, $range);
                my $mm = $self->{ng}->mm;
                my $r = get_range($self, $range);
                return $r unless ref $r;
@@ -635,49 +652,89 @@ sub xhdr_searchmsg ($$$) {
                        my ($i) = @_;
                        my $mid = $mm->mid_for($$i) or return;
                        my $m = header_obj_for($srch, $mid) or return;
-                       $emit->($$i, $m);
+                       my $v = $m->header($hdr);
+                       defined $v or return;
+                       $v = $filter->($v) if $filter;
+                       more($self, "$$i $v");
                });
        }
 }
 
-sub cmd_xhdr ($$;$) {
-       my ($self, $header, $range) = @_;
-       my $ng = $self->{ng};
-       defined $ng or return '412 no news group currently selected';
+sub do_hdr ($$$;$) {
+       my ($self, $xhdr, $header, $range) = @_;
        my $sub = lc $header;
        if ($sub eq 'message-id') {
-               xhdr_message_id($self, $range);
+               hdr_message_id($self, $xhdr, $range);
        } elsif ($sub eq 'xref') {
-               xhdr_xref($self, $range);
-       } elsif ($sub =~ /\A(subject|references|date)\z/ && $ng->search) {
-               xhdr_searchmsg($self, $sub, $range);
+               hdr_xref($self, $xhdr, $range);
+       } elsif ($sub =~ /\A(subject|references|date)\z/) {
+               hdr_searchmsg($self, $xhdr, $sub, $range);
        } else {
-               xhdr_slow($self, $header, $range);
+               hdr_slow($self, $xhdr, $header, $range);
        }
 }
 
-sub xhdr_slow ($$$) {
+# RFC 3977
+sub cmd_hdr ($$;$) {
        my ($self, $header, $range) = @_;
+       do_hdr($self, 0, $header, $range);
+}
 
-       if (defined $range && $range =~ /\A<(.+)>\z/) { # Message-ID
+# RFC 2980
+sub cmd_xhdr ($$;$) {
+       my ($self, $header, $range) = @_;
+       do_hdr($self, 1, $header, $range);
+}
+
+sub hdr_mid_prefix ($$$$$) {
+       my ($self, $xhdr, $ng, $n, $mid) = @_;
+       return $mid if $xhdr;
+
+       # HDR for RFC 3977 users
+       if (my $self_ng = $self->{ng}) {
+               ($self_ng eq $ng) ? $n : '0';
+       } else {
+               '0';
+       }
+}
+
+sub hdr_mid_response ($$$$$$) {
+       my ($self, $xhdr, $ng, $n, $mid, $v) = @_; # r: art_lookup result
+       my $res = '';
+       if ($xhdr) {
+               $res .= r221 . "\r\n";
+               $res .= "$mid $v\r\n" if defined $v;
+       } else {
+               $res .= r224 . "\r\n";
+               if (defined $v) {
+                       my $pfx = hdr_mid_prefix($self, $xhdr, $ng, $n, $mid);
+                       $res .= "$pfx $v\r\n";
+               }
+       }
+       res($self, $res .= '.');
+       undef;
+}
+
+sub hdr_slow ($$$$) {
+       my ($self, $xhdr, $header, $range) = @_;
+
+       if (defined $range && $range =~ /\A<.+>\z/) { # Message-ID
                my $r = $self->art_lookup($range, 2);
                return $r unless ref $r;
-               more($self, '221 Header follows');
-               if (defined($r = xhdr($r, $header))) {
-                       more($self, "<$range> $r");
-               }
-               '.';
+               my ($n, $ng) = ($r->[0], $r->[5]);
+               my $v = hdr_val($r, $header);
+               hdr_mid_response($self, $xhdr, $ng, $n, $range, $v);
        } else { # numeric range
                $range = $self->{article} unless defined $range;
                my $r = get_range($self, $range);
                return $r unless ref $r;
                my ($beg, $end) = @$r;
-               more($self, '221 Header follows');
+               more($self, $xhdr ? r221 : r224);
                $self->long_response($beg, $end, sub {
                        my ($i) = @_;
                        $r = $self->art_lookup($$i, 2);
                        return unless ref $r;
-                       defined($r = xhdr($r, $header)) or return;
+                       defined($r = hdr_val($r, $header)) or return;
                        more($self, "$$i $r");
                });
        }
@@ -705,6 +762,32 @@ sub cmd_xrover ($;$) {
        });
 }
 
+sub over_line ($$) {
+       my ($self, $r) = @_;
+
+       more($self, join("\t", $r->[0], map {
+                               my $h = hdr_val($r, $_);
+                               defined $h ? $h : '';
+                       } @OVERVIEW ));
+}
+
+sub cmd_over ($;$) {
+       my ($self, $range) = @_;
+       if ($range && $range =~ /\A<.+>\z/) {
+               my $r = $self->art_lookup($range, 2);
+               return '430 No article with that message-id' unless ref $r;
+               more($self, '224 Overview information follows (multi-line)');
+
+               # Only set article number column if it's the current group
+               my $ng = $self->{ng};
+               $r->[0] = 0 if (!$ng || $ng ne $r->[5]);
+               over_line($self, $r);
+               '.';
+       } else {
+               cmd_xover($self, $range);
+       }
+}
+
 sub cmd_xover ($;$) {
        my ($self, $range) = @_;
        $range = $self->{article} unless defined $range;
@@ -716,11 +799,7 @@ sub cmd_xover ($;$) {
                my ($i) = @_;
                my $r = $self->art_lookup($$i, 2);
                return unless ref $r;
-               more($self, join("\t", $r->[0],
-                               map {
-                                       my $h = xhdr($r, $_);
-                                       defined $h ? $h : '';
-                               } @OVERVIEW ));
+               over_line($self, $r);
        });
 }