count_unrecognized_commands
simplified logic in a couple places consolidated duplicated message added 4 tests
This commit is contained in:
parent
74125300da
commit
5e76d66c66
@ -16,20 +16,23 @@ before we disconnect the client. Defaults to 4.
|
|||||||
|
|
||||||
=cut
|
=cut
|
||||||
|
|
||||||
|
use strict;
|
||||||
|
use warnings;
|
||||||
|
|
||||||
|
use Qpsmtpd::Constants;
|
||||||
|
|
||||||
sub register {
|
sub register {
|
||||||
my ($self, $qp, @args) = @_;
|
my ($self, $qp ) = shift, shift;
|
||||||
|
|
||||||
if (@args > 0) {
|
$self->{_unrec_cmd_max} = shift || 4;
|
||||||
$self->{_unrec_cmd_max} = $args[0];
|
|
||||||
$self->log(LOGWARN, "WARNING: Ignoring additional arguments.") if (@args > 1);
|
if ( scalar @_ ) {
|
||||||
} else {
|
$self->log(LOGWARN, "Ignoring additional arguments.");
|
||||||
$self->{_unrec_cmd_max} = 4;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
sub hook_connect {
|
sub hook_connect {
|
||||||
my ($self, $transaction) = @_;
|
my $self = shift;
|
||||||
|
|
||||||
$self->qp->connection->notes('unrec_cmd_count', 0);
|
$self->qp->connection->notes('unrec_cmd_count', 0);
|
||||||
return DECLINED;
|
return DECLINED;
|
||||||
@ -46,8 +49,9 @@ sub hook_unrecognized_command {
|
|||||||
);
|
);
|
||||||
|
|
||||||
if ($badcmdcount >= $self->{_unrec_cmd_max}) {
|
if ($badcmdcount >= $self->{_unrec_cmd_max}) {
|
||||||
$self->log(LOGINFO, "Closing connection. Too many unrecognized commands.");
|
my $msg = "Closing connection, $badcmdcount unrecognized commands.";
|
||||||
return (DENY_DISCONNECT, "Closing connection. $badcmdcount unrecognized commands. Perhaps you should read RFC 2821?");
|
$self->log(LOGINFO, "fail: $msg");
|
||||||
|
return (DENY_DISCONNECT, "$msg Perhaps you should read RFC 2821?");
|
||||||
}
|
}
|
||||||
|
|
||||||
return DECLINED;
|
return DECLINED;
|
||||||
|
31
t/plugin_tests/count_unrecognized_commands
Normal file
31
t/plugin_tests/count_unrecognized_commands
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
#!perl -w
|
||||||
|
|
||||||
|
use strict;
|
||||||
|
use warnings;
|
||||||
|
|
||||||
|
use Qpsmtpd::Constants;
|
||||||
|
|
||||||
|
sub register_tests {
|
||||||
|
my $self = shift;
|
||||||
|
|
||||||
|
$self->register_test('test_hook_unrecognized_command', 4);
|
||||||
|
};
|
||||||
|
|
||||||
|
sub test_hook_unrecognized_command {
|
||||||
|
my $self = shift;
|
||||||
|
|
||||||
|
$self->{_unrec_cmd_max} = 2;
|
||||||
|
$self->qp->connection->notes( 'unrec_cmd_count', 0 );
|
||||||
|
|
||||||
|
my ($code, $mess) = $self->hook_unrecognized_command(undef,'hiya');
|
||||||
|
cmp_ok( $code, '==', DECLINED, "good" );
|
||||||
|
|
||||||
|
$self->qp->connection->notes( 'unrec_cmd_count', 2 );
|
||||||
|
($code, $mess) = $self->hook_unrecognized_command(undef,'snookums');
|
||||||
|
cmp_ok( $code, '==', DENY_DISCONNECT, "limit" );
|
||||||
|
|
||||||
|
($code, $mess) = $self->hook_unrecognized_command(undef,'wtf');
|
||||||
|
cmp_ok( $code, '==', DENY_DISCONNECT, "over limit" );
|
||||||
|
|
||||||
|
cmp_ok( $self->qp->connection->notes( 'unrec_cmd_count'), '==', 4, "correct increment" );
|
||||||
|
};
|
Loading…
Reference in New Issue
Block a user