qpsmtpd/plugins/count_unrecognized_commands
Matt Simerson 5e76d66c66 count_unrecognized_commands
simplified logic in a couple places
consolidated duplicated message
added 4 tests
2012-05-20 23:34:58 -07:00

60 lines
1.3 KiB
Perl

#!perl -w
=head1 NAME
count_unrecognized_commands - Count unrecognized commands and disconnect when we have too many
=head1 DESCRIPTION
Disconnect the client if it sends too many unrecognized commands.
Good for rejecting spam sent through open HTTP proxies.
=head1 CONFIGURATION
Takes one parameter, the number of allowed unrecognized commands
before we disconnect the client. Defaults to 4.
=cut
use strict;
use warnings;
use Qpsmtpd::Constants;
sub register {
my ($self, $qp ) = shift, shift;
$self->{_unrec_cmd_max} = shift || 4;
if ( scalar @_ ) {
$self->log(LOGWARN, "Ignoring additional arguments.");
}
}
sub hook_connect {
my $self = shift;
$self->qp->connection->notes('unrec_cmd_count', 0);
return DECLINED;
}
sub hook_unrecognized_command {
my ($self, $cmd) = @_[0,2];
$self->log(LOGINFO, "Unrecognized command '$cmd'");
my $badcmdcount =
$self->qp->connection->notes( 'unrec_cmd_count',
($self->qp->connection->notes('unrec_cmd_count') || 0) + 1
);
if ($badcmdcount >= $self->{_unrec_cmd_max}) {
my $msg = "Closing connection, $badcmdcount unrecognized commands.";
$self->log(LOGINFO, "fail: $msg");
return (DENY_DISCONNECT, "$msg Perhaps you should read RFC 2821?");
}
return DECLINED;
}