]>
Commit | Line | Data |
---|---|---|
0a580593 DM |
1 | package PMG::RuleDB::Quarantine; |
2 | ||
3 | use strict; | |
4 | use warnings; | |
0a580593 DM |
5 | use DBI; |
6 | use Digest::SHA; | |
7 | ||
8 | use PVE::SafeSyslog; | |
758c7b6b DM |
9 | |
10 | use PMG::Utils; | |
11 | use PMG::ModGroup; | |
0a580593 DM |
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 | ||
0a580593 DM |
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 | ||
7a2cf7e6 | 45 | my $self = $class->SUPER::new($class->otype(), $ogroup); |
0a580593 DM |
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 | ||
758c7b6b | 80 | $self->{id} = PMG::Utils::lastid($ruledb->{dbh}, 'object_id_seq'); |
0a580593 DM |
81 | } |
82 | ||
83 | return $self->{id}; | |
84 | } | |
85 | ||
86 | sub execute { | |
87 | my ($self, $queue, $ruledb, $mod_group, $targets, | |
88 | $msginfo, $vars, $marks, $ldap) = @_; | |
0a580593 DM |
89 | |
90 | my $subgroups = $mod_group->subgroups($targets, 1); | |
91 | ||
6c65ab40 | 92 | my $rulename = $vars->{RULE} // 'unknown'; |
365d5b95 | 93 | |
0a580593 DM |
94 | foreach my $ta (@$subgroups) { |
95 | my ($tg, $entity) = (@$ta[0], @$ta[1]); | |
96 | ||
758c7b6b | 97 | PMG::Utils::remove_marks($entity); |
0a580593 DM |
98 | |
99 | if ($queue->{vinfo}) { | |
100 | if (my $qid = $queue->quarantine_mail($ruledb, 'V', $entity, $tg, $msginfo, $vars, $ldap)) { | |
101 | ||
102 | foreach (@$tg) { | |
bb1b6f1e | 103 | syslog ('info', "$queue->{logid}: moved mail for <%s> to virus quarantine - %s (rule: %s)", $_, $qid, $rulename); |
0a580593 DM |
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) { | |
bb1b6f1e | 113 | syslog ('info', "$queue->{logid}: moved mail for <%s> to spam quarantine - %s (rule: %s)", $_, $qid, $rulename); |
0a580593 DM |
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 | ||
9578dcd7 | 127 | return 'Move to quarantine.'; |
0a580593 DM |
128 | } |
129 | ||
130 | 1; |