src/PMG/RuleDB/Quarantine.pm

   1 package PMG::RuleDB::Quarantine;
   2
   3 use strict;
   4 use warnings;
   5 use DBI;
   6 use Digest::SHA;
   7 use Encode qw(encode);
   8
   9 use PVE::SafeSyslog;
  10
  11 use PMG::Utils;
  12 use PMG::ModGroup;
  13 use PMG::RuleDB::Object;
  14
  15 use base qw(PMG::RuleDB::Object);
  16
  17 sub otype {
  18     return 4006;
  19 }
  20
  21 sub oclass {
  22     return 'action';
  23 }
  24
  25 sub otype_text {
  26     return 'Quarantine';
  27 }
  28
  29 sub oisedit {
  30     return 0;
  31 }
  32
  33 sub final {
  34     return 1;
  35 }
  36
  37 sub priority {
  38     return 90;
  39 }
  40
  41 sub new {
  42     my ($type, $ogroup) = @_;
  43
  44     my $class = ref($type) || $type;
  45
  46     my $self = $class->SUPER::new($class->otype(), $ogroup);
  47
  48     return $self;
  49 }
  50
  51 sub load_attr {
  52     my ($type, $ruledb, $id, $ogroup, $value) = @_;
  53
  54     my $class = ref($type) || $type;
  55
  56     my $obj = $class->new($ogroup);
  57     $obj->{id} = $id;
  58
  59     $obj->{digest} = Digest::SHA::sha1_hex($id, $ogroup);
  60
  61     return $obj;
  62 }
  63
  64 sub save {
  65     my ($self, $ruledb) = @_;
  66
  67     defined($self->{ogroup}) || return undef;
  68
  69     if (defined ($self->{id})) {
  70         # update
  71
  72         # nothing to update
  73
  74     } else {
  75         # insert
  76         my $sth = $ruledb->{dbh}->prepare (
  77             "INSERT INTO Object (Objectgroup_ID, ObjectType) VALUES (?, ?);");
  78
  79         $sth->execute($self->ogroup, $self->otype);
  80
  81         $self->{id} = PMG::Utils::lastid($ruledb->{dbh}, 'object_id_seq');
  82     }
  83
  84     return $self->{id};
  85 }
  86
  87 sub execute {
  88     my ($self, $queue, $ruledb, $mod_group, $targets,
  89         $msginfo, $vars, $marks, $ldap) = @_;
  90
  91     my $subgroups = $mod_group->subgroups($targets, 1);
  92
  93     my $rulename = encode('UTF-8', $vars->{RULE} // 'unknown');
  94
  95     foreach my $ta (@$subgroups) {
  96         my ($tg, $entity) = (@$ta[0], @$ta[1]);
  97
  98         PMG::Utils::remove_marks($entity);
  99
 100         if ($queue->{vinfo}) {
 101             if (my $qid = $queue->quarantine_mail($ruledb, 'V', $entity, $tg, $msginfo, $vars, $ldap)) {
 102
 103                 foreach (@$tg) {
 104                     syslog ('info', "$queue->{logid}: moved mail for <%s> to virus quarantine - %s (rule: %s)", $_, $qid, $rulename);
 105                 }
 106
 107                 $queue->set_status ($tg, 'delivered');
 108             }
 109
 110         } else {
 111             if (my $qid = $queue->quarantine_mail($ruledb, 'S', $entity, $tg, $msginfo, $vars, $ldap)) {
 112
 113                 foreach (@$tg) {
 114                     syslog ('info', "$queue->{logid}: moved mail for <%s> to spam quarantine - %s (rule: %s)", $_, $qid, $rulename);
 115                 }
 116
 117                 $queue->set_status($tg, 'delivered');
 118             }
 119         }
 120     }
 121
 122     # warn if no subgroups
 123 }
 124
 125 sub short_desc {
 126     my $self = shift;
 127
 128     return 'Move to quarantine.';
 129 }
 130
 131 1;