blob: 8dc06eb6103d9ac5fa6a57d293d6d98c6eadeaa5 [file] [log] [blame]
#!/usr/bin/perl
# (C) Sergey Kandaurov
# (C) Nginx, Inc.
# Tests for mail resolver.
###############################################################################
use warnings;
use strict;
use Test::More;
BEGIN { use FindBin; chdir($FindBin::Bin); }
use lib 'lib';
use Test::Nginx;
use Test::Nginx::SMTP;
###############################################################################
select STDERR; $| = 1;
select STDOUT; $| = 1;
local $SIG{PIPE} = 'IGNORE';
my $t = Test::Nginx->new()->has(qw/mail smtp http rewrite/)->plan(8)
->write_file_expand('nginx.conf', <<'EOF');
%%TEST_GLOBALS%%
daemon off;
events {
}
mail {
auth_http http://127.0.0.1:8080/mail/auth;
smtp_auth none;
server_name locahost;
# prevent useless resend
resolver_timeout 2s;
server {
listen 127.0.0.1:8025;
protocol smtp;
resolver 127.0.0.1:%%PORT_8981_UDP%%
127.0.0.1:%%PORT_8982_UDP%%
127.0.0.1:%%PORT_8983_UDP%%;
}
server {
listen 127.0.0.1:8027;
protocol smtp;
resolver 127.0.0.1:%%PORT_8982_UDP%%;
}
server {
listen 127.0.0.1:8028;
protocol smtp;
resolver 127.0.0.1:%%PORT_8983_UDP%%;
resolver_timeout 1s;
}
server {
listen 127.0.0.1:8029;
protocol smtp;
resolver 127.0.0.1:%%PORT_8984_UDP%%;
}
server {
listen 127.0.0.1:8030;
protocol smtp;
resolver 127.0.0.1:%%PORT_8985_UDP%%;
}
server {
listen 127.0.0.1:8031;
protocol smtp;
resolver 127.0.0.1:%%PORT_8986_UDP%%;
resolver_timeout 1s;
}
server {
listen 127.0.0.1:8032;
protocol smtp;
resolver 127.0.0.1:%%PORT_8987_UDP%%;
}
}
http {
%%TEST_GLOBALS_HTTP%%
server {
listen 127.0.0.1:8080;
server_name localhost;
location = /mail/auth {
set $reply $http_client_host;
if ($http_client_host !~ UNAVAIL) {
set $reply OK;
}
add_header Auth-Status $reply;
add_header Auth-Server 127.0.0.1;
add_header Auth-Port %%PORT_8026%%;
return 204;
}
}
}
EOF
$t->run_daemon(\&Test::Nginx::SMTP::smtp_test_daemon);
$t->run_daemon(\&dns_daemon, port($_), $t) foreach (8981 .. 8987);
$t->run();
$t->waitforsocket('127.0.0.1:' . port(8026));
$t->waitforfile($t->testdir . '/' . port($_)) foreach (8981 .. 8987);
###############################################################################
# PTR
my $s = Test::Nginx::SMTP->new();
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->ok('PTR');
$s->send('QUIT');
$s->read();
# Cached PTR prevents from querying bad ns on port 8983
$s = Test::Nginx::SMTP->new();
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->ok('PTR cached');
$s->send('QUIT');
$s->read();
# SERVFAIL
$s = Test::Nginx::SMTP->new(PeerAddr => '127.0.0.1:' . port(8027));
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->check(qr/TEMPUNAVAIL/, 'PTR SERVFAIL');
$s->send('QUIT');
$s->read();
# PTR with zero length RDATA
$s = Test::Nginx::SMTP->new(PeerAddr => '127.0.0.1:' . port(8028));
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->check(qr/TEMPUNAVAIL/, 'PTR empty');
$s->send('QUIT');
$s->read();
# CNAME
TODO: {
local $TODO = 'support for CNAME RR';
$s = Test::Nginx::SMTP->new(PeerAddr => '127.0.0.1:' . port(8029));
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->ok('CNAME');
$s->send('QUIT');
$s->read();
}
# uncompressed answer
$s = Test::Nginx::SMTP->new(PeerAddr => '127.0.0.1:' . port(8030));
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->ok('uncompressed PTR');
$s->send('QUIT');
$s->read();
$s = Test::Nginx::SMTP->new(PeerAddr => '127.0.0.1:' . port(8031));
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->check(qr/TEMPUNAVAIL/, 'PTR type');
$s->send('QUIT');
$s->read();
# CNAME and PTR in one answer section
$s = Test::Nginx::SMTP->new(PeerAddr => '127.0.0.1:' . port(8032));
$s->read();
$s->send('EHLO example.com');
$s->read();
$s->send('MAIL FROM:<test@example.com> SIZE=100');
$s->read();
$s->send('RCPT TO:<test@example.com>');
$s->ok('CNAME with PTR');
$s->send('QUIT');
$s->read();
###############################################################################
sub reply_handler {
my ($recv_data, $port) = @_;
my (@name, @rdata);
use constant NOERROR => 0;
use constant SERVFAIL => 2;
use constant NXDOMAIN => 3;
use constant A => 1;
use constant CNAME => 5;
use constant PTR => 12;
use constant DNAME => 39;
use constant IN => 1;
# default values
my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600);
# decode name
my ($len, $offset) = (undef, 12);
while (1) {
$len = unpack("\@$offset C", $recv_data);
last if $len == 0;
$offset++;
push @name, unpack("\@$offset A$len", $recv_data);
$offset += $len;
}
$offset -= 1;
my ($id, $type, $class) = unpack("n x$offset n2", $recv_data);
my $name = join('.', @name);
if ($name eq 'a.example.net' && $type == A) {
push @rdata, rd_addr($ttl, '127.0.0.1');
} elsif ($name eq '1.0.0.127.in-addr.arpa' && $type == PTR) {
if ($port == port(8981)) {
push @rdata, rd_name(PTR, $ttl, 'a.example.net');
} elsif ($port == port(8982)) {
$rcode = SERVFAIL;
} elsif ($port == port(8983)) {
# zero length RDATA
push @rdata, pack("n3N n", 0xc00c, PTR, IN, $ttl, 0);
} elsif ($port == port(8984)) {
# PTR answered with CNAME
push @rdata, rd_name(CNAME, $ttl,
'1.1.0.0.127.in-addr.arpa');
} elsif ($port == port(8985)) {
# uncompressed answer
push @rdata, pack("(C/a*)6x n2N n(C/a*)3x",
('1', '0', '0', '127', 'in-addr', 'arpa'),
PTR, IN, $ttl, 15, ('a', 'example', 'net'));
} elsif ($port == port(8986)) {
push @rdata, rd_name(DNAME, $ttl, 'a.example.net');
} elsif ($port == port(8987)) {
# PTR answered with CNAME+PTR
push @rdata, rd_name(CNAME, $ttl,
'1.1.0.0.127.in-addr.arpa');
push @rdata, pack("n3N n(C/a*)3 x", 0xc034,
PTR, IN, $ttl, 15, ('a', 'example', 'net'));
}
} elsif ($name eq '1.1.0.0.127.in-addr.arpa' && $type == PTR) {
push @rdata, rd_name(PTR, $ttl, 'a.example.net');
}
$len = @name;
pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata,
0, 0, @name, $type, $class) . join('', @rdata);
}
sub rd_name {
my ($type, $ttl, $name) = @_;
my ($rdlen, @rdname);
@rdname = split /\./, $name;
$rdlen = length(join '', @rdname) + @rdname + 1;
pack("n3N n(C/a*)* x", 0xc00c, $type, IN, $ttl, $rdlen, @rdname);
}
sub rd_addr {
my ($ttl, $addr) = @_;
my $code = 'split(/\./, $addr)';
# use a special pack string to not zero pad
return pack 'n3N', 0xc00c, A, IN, $ttl if $addr eq '';
pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code);
}
sub dns_daemon {
my ($port, $t) = @_;
my ($data, $recv_data);
my $socket = IO::Socket::INET->new(
LocalAddr => '127.0.0.1',
LocalPort => $port,
Proto => 'udp',
)
or die "Can't create listening socket: $!\n";
# signal we are ready
open my $fh, '>', $t->testdir() . '/' . $port;
close $fh;
while (1) {
$socket->recv($recv_data, 65536);
$data = reply_handler($recv_data, $port);
$socket->send($data);
}
}
###############################################################################