qpsmtpd/plugins/noop_counter

62 lines
1.6 KiB
Perl

#!perl -w
=head1 NAME
noop_counter - disconnect after too many consecutive NOOPs, example plugin for the hook_noop()
=head1 DESCRIPTION
The B<noop_counter> counts the number of consecutive C<NOOP> commands given
by a client and disconnects after a given number.
Any other command than a C<NOOP> resets the counter.
One argument may be given: the number of C<NOOP>s after which the client will
be disconnected.
=head1 NOTE
This plugin should be loaded early to be able to reset the counter on any other
command.
=cut
sub register {
my ($self, $qp, @args) = @_;
$self->{_noop_count} = 0;
$self->{_max_noop} = 3;
if ($args[0] && $args[0] =~ /^\d+$/) {
$self->{_max_noop} = shift @args;
}
}
sub hook_noop {
my ($self, $transaction, @args) = @_;
++$self->{_noop_count};
### the following block is not used, RFC 2821 says we SHOULD ignore
### any arguments... so we MAY return an error if we want to :-)
# return DENY, "Syntax error, NOOP does not take any arguments"
# if $args[0];
if ($self->{_noop_count} >= $self->{_max_noop}) {
return DENY_DISCONNECT, "Stop wasting my time, too many consecutive NOOPs";
}
return DECLINED;
}
sub reset_noop_counter {
$_[0]->{_noop_count} = 0;
return DECLINED;
}
# and bind the counter reset to the hooks, QUIT not useful here:
*hook_helo = *hook_ehlo = # HELO / EHLO
*hook_mail = # MAIL FROM:
*hook_rcpt = # RCPT TO:
*hook_data = # DATA
*hook_reset_transaction = # RSET
*hook_vrfy = # VRFY
*hook_help = # HELP
\&reset_noop_counter;