1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
# --
# Kernel/System/PostMaster/LoopProtection/DB.pm - backend module of LoopProtection
# Copyright (C) 2001-2007 OTRS GmbH, http://otrs.org/
# --
# $Id: DB.pm,v 1.7 2007/03/19 22:24:59 martin Exp $
# --
# This software comes with ABSOLUTELY NO WARRANTY. For details, see
# the enclosed file COPYING for license information (GPL). If you
# did not receive this file, see http://www.gnu.org/licenses/gpl.txt.
# --
package Kernel::System::PostMaster::LoopProtection::DB;
use strict;
use vars qw($VERSION);
$VERSION = '$Revision: 1.7 $';
$VERSION =~ s/^\$.*:\W(.*)\W.+?$/$1/;
sub new {
my $Type = shift;
my %Param = @_;
# allocate new hash for object
my $Self = {};
bless ($Self, $Type);
# get needed objects
foreach (qw(DBObject LogObject ConfigObject)) {
$Self->{$_} = $Param{$_} || die "Got no $_!";
}
# get config options
$Self->{PostmasterMaxEmails} = $Self->{ConfigObject}->Get('PostmasterMaxEmails') || 40;
# create logfile name
my ($Sec, $Min, $Hour, $Day, $Month, $Year) = localtime(time);
$Year=$Year+1900;
$Month++;
$Self->{LoopProtectionDate} .= $Year.'-'.$Month.'-'.$Day;
return $Self;
}
sub SendEmail {
my $Self = shift;
my %Param = @_;
my $To = $Param{To} || return;
# insert log
if ($Self->{DBObject}->Do(
SQL => "INSERT INTO ticket_loop_protection (sent_to, sent_date)".
" VALUES ('".$Self->{DBObject}->Quote($To)."', '$Self->{LoopProtectionDate}')",
)) {
# delete old enrties
$Self->{DBObject}->Do(
SQL => "DELETE FROM ticket_loop_protection WHERE ".
" sent_date != '$Self->{LoopProtectionDate}'",
);
return 1;
}
else {
return;
}
}
sub Check {
my $Self = shift;
my %Param = @_;
my $To = $Param{To} || return;
my $Count = 0;
# check existing logfile
my $SQL = "SELECT count(*) FROM ticket_loop_protection " .
" WHERE " .
" sent_to = '".$Self->{DBObject}->Quote($To)."' ".
" AND ".
" sent_date = '$Self->{LoopProtectionDate}'";
$Self->{DBObject}->Prepare(SQL => $SQL);
while (my @Row = $Self->{DBObject}->FetchrowArray()) {
$Count = $Row[0];
}
# check possible loop
if ($Count >= $Self->{PostmasterMaxEmails}) {
$Self->{LogObject}->Log(
Priority => 'notice',
Message => "LoopProtection!!! Send no more emails to '$To'! Max. count of $Self->{PostmasterMaxEmails} has been reached!",
);
return;
}
else {
return 1;
}
}
1;
|
