PMG/RuleDB/Quarantine.pm

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