public-inbox-mda

   1 #!/usr/bin/perl -w
   2 # Copyright (C) 2013, Eric Wong <normalperson@yhbt.net> and all contributors
   3 # License: AGPLv3 or later (https://www.gnu.org/licenses/agpl-3.0.txt)
   4 use strict;
   5 use warnings;
   6 use Email::Filter;
   7 use Email::Address;
   8 use PublicInbox::Filter;
   9 use IPC::Run qw(run);
  10 my $usage = "public-inbox-mda main_repo fail_repo < rfc2822_message";
  11 my $filter = Email::Filter->new(emergency => "~/emergency.mbox");
  12 my $main_repo = shift @ARGV or die "Usage: $usage\n";
  13 my $fail_repo = shift @ARGV or die "Usage: $usage\n";
  14 my $max = 1024 * 500; # same as spamc
  15
  16 my $filtered;
  17 if (length($filter->simple->as_string) <= $max
  18     && PublicInbox->recipient_specified($filter)
  19     && do_spamc($filter->simple, \$filtered)) {
  20         # update our message with SA headers (in case our filter rejects it)
  21         my $simple = Email::Simple->new($filtered);
  22         $filtered = undef;
  23         $filter->simple($simple);
  24
  25         if (PublicInbox::Filter->run($simple)) {
  26                 # run spamc again on the HTML-free message
  27                 if (do_spamc($simple, \$filtered)) {
  28                         $filter->simple(Email::Simple->new($filtered));
  29                         $filter->pipe("ssoma-mda", $main_repo);
  30                 } else {
  31                         $filter->pipe("ssoma-mda", $fail_repo);
  32                 }
  33         } else {
  34                 # PublicInbox::Filter nuked everything, oops :x
  35                 $filter->pipe("ssoma-mda", $fail_repo);
  36         }
  37 } else {
  38         # if SA thinks it's spam or there's an error:
  39         # don't bother with our own filtering
  40         $filter->pipe("ssoma-mda", $fail_repo);
  41 }
  42 die "Email::Filter failed to exit\n";
  43
  44 # we depend on "report_safe 0" in /etc/spamassassin/*.cf with --headers
  45 # not using Email::Filter->pipe here since we want the stdout of
  46 # the command even on failure (spamc will set $? on error).
  47 sub do_spamc {
  48         my ($simple, $out) = @_;
  49         eval {
  50                 my $orig = $simple->as_string;
  51                 run([qw/spamc -E --headers/], \$orig, $out);
  52         };
  53
  54         return ($@ || $? || !defined($$out) || length($$out) == 0) ? 0 : 1;
  55 }