mail_resolver.t - nginx-tests - Git at Google

 #!/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/)
 	->run_daemon(\&Test::Nginx::SMTP::smtp_test_daemon);

 $t->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 1s;

     server {
         listen    127.0.0.1:8025;
         protocol  smtp;
         resolver  127.0.0.1:8081 127.0.0.1:8082 127.0.0.1:8083;
     }

     server {
         listen    127.0.0.1:8027;
         protocol  smtp;
         resolver  127.0.0.1:8082;
     }

     server {
         listen    127.0.0.1:8028;
         protocol  smtp;
         resolver  127.0.0.1:8083;

     }

     server {
         listen    127.0.0.1:8029;
         protocol  smtp;
         resolver  127.0.0.1:8084;
     }

     server {
         listen    127.0.0.1:8030;
         protocol  smtp;
         resolver  127.0.0.1:8085;
     }

     server {
         listen    127.0.0.1:8031;
         protocol  smtp;
         resolver  127.0.0.1:8086;
     }
 }

 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 8026;
             return 204;
         }
     }
 }

 EOF

 for (8081 .. 8086) {
 	$t->run_daemon(\&dns_daemon, $_, $t);
 }
 $t->run();

 for (8081 .. 8086) {
 	$t->waitforfile($t->testdir . "/$_");
 }

 $t->plan(7);

 ###############################################################################

 # 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 8083

 $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: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: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: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('PTR with CNAME');

 $s->send('QUIT');
 $s->read();

 }

 # uncompressed answer

 TODO: {
 local $TODO = 'support for uncompressed name in PTR';

 $s = Test::Nginx::SMTP->new(PeerAddr => "127.0.0.1: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();

 }

 TODO: {
 local $TODO = 'PTR type checking';

 $s = Test::Nginx::SMTP->new(PeerAddr => "127.0.0.1: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();

 }

 ###############################################################################

 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 == 8081) {
 			push @rdata, rd_name(PTR, $ttl, 'a.example.net');

 		} elsif ($port == 8082) {
 			$rcode = SERVFAIL;

 		} elsif ($port == 8083) {
 			# zero length RDATA

 			push @rdata, pack("n3N n", 0xc00c, PTR, IN, $ttl, 0);

 		} elsif ($port == 8084) {
 			# PTR answered with CNAME

 			push @rdata, rd_name(CNAME, $ttl,
 				'1.1.0.0.127.in-addr.arpa');

 		} elsif ($port == 8085) {
 			# 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 == 8086) {
 			push @rdata, rd_name(DNAME, $ttl, '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);
 	}
 }

 ###############################################################################
	#!/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/)
	->run_daemon(\&Test::Nginx::SMTP::smtp_test_daemon);

	$t->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 1s;

	server {
	listen 127.0.0.1:8025;
	protocol smtp;
	resolver 127.0.0.1:8081 127.0.0.1:8082 127.0.0.1:8083;
	}

	server {
	listen 127.0.0.1:8027;
	protocol smtp;
	resolver 127.0.0.1:8082;
	}

	server {
	listen 127.0.0.1:8028;
	protocol smtp;
	resolver 127.0.0.1:8083;

	}

	server {
	listen 127.0.0.1:8029;
	protocol smtp;
	resolver 127.0.0.1:8084;
	}

	server {
	listen 127.0.0.1:8030;
	protocol smtp;
	resolver 127.0.0.1:8085;
	}

	server {
	listen 127.0.0.1:8031;
	protocol smtp;
	resolver 127.0.0.1:8086;
	}
	}

	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 8026;
	return 204;
	}
	}
	}

	EOF

	for (8081 .. 8086) {
	$t->run_daemon(\&dns_daemon, $_, $t);
	}
	$t->run();

	for (8081 .. 8086) {
	$t->waitforfile($t->testdir . "/$_");
	}

	$t->plan(7);

	###############################################################################

	# 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 8083

	$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: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: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: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('PTR with CNAME');

	$s->send('QUIT');
	$s->read();

	}

	# uncompressed answer

	TODO: {
	local $TODO = 'support for uncompressed name in PTR';

	$s = Test::Nginx::SMTP->new(PeerAddr => "127.0.0.1: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();

	}

	TODO: {
	local $TODO = 'PTR type checking';

	$s = Test::Nginx::SMTP->new(PeerAddr => "127.0.0.1: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();

	}

	###############################################################################

	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 == 8081) {
	push @rdata, rd_name(PTR, $ttl, 'a.example.net');

	} elsif ($port == 8082) {
	$rcode = SERVFAIL;

	} elsif ($port == 8083) {
	# zero length RDATA

	push @rdata, pack("n3N n", 0xc00c, PTR, IN, $ttl, 0);

	} elsif ($port == 8084) {
	# PTR answered with CNAME

	push @rdata, rd_name(CNAME, $ttl,
	'1.1.0.0.127.in-addr.arpa');

	} elsif ($port == 8085) {
	# 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 == 8086) {
	push @rdata, rd_name(DNAME, $ttl, '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);
	}
	}

	###############################################################################