changeset 825:ff8c68cd2be4

Tests: http resolver tests with CNAME, complicated cases.
author Sergey Kandaurov <pluknet@nginx.com>
date Wed, 27 Jan 2016 14:08:02 +0300
parents 9caacdb56b53
children 64abcc611242
files http_resolver_cname.t
diffstat 1 files changed, 281 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/http_resolver_cname.t
@@ -0,0 +1,281 @@
+#!/usr/bin/perl
+
+# (C) Sergey Kandaurov
+# (C) Nginx, Inc.
+
+# Tests for http resolver with CNAME.
+
+###############################################################################
+
+use warnings;
+use strict;
+
+use Test::More;
+
+BEGIN { use FindBin; chdir($FindBin::Bin); }
+
+use lib 'lib';
+use Test::Nginx qw/ :DEFAULT http_end /;
+
+###############################################################################
+
+select STDERR; $| = 1;
+select STDOUT; $| = 1;
+
+my $t = Test::Nginx->new()->has(qw/http proxy/)->plan(11);
+
+$t->write_file_expand('nginx.conf', <<'EOF');
+
+%%TEST_GLOBALS%%
+
+daemon off;
+
+events {
+}
+
+http {
+    %%TEST_GLOBALS_HTTP%%
+
+    server {
+        listen       127.0.0.1:8080;
+        server_name  localhost;
+
+        location /short {
+            resolver    127.0.0.1:8081;
+            resolver_timeout 2s;
+
+            proxy_pass  http://$host:8080/t;
+        }
+
+        location /long {
+            resolver    127.0.0.1:8081;
+            resolver_timeout 5s;
+
+            proxy_pass  http://$host:8080/t;
+        }
+
+        location / { }
+    }
+}
+
+EOF
+
+$t->run_daemon(\&dns_daemon, 8081, $t);
+
+$t->write_file('t', '');
+$t->run();
+
+$t->waitforfile($t->testdir . '/8081');
+
+###############################################################################
+
+# CNAME pointing to name which times out
+
+like(http_host('cn01.example.net', '/short'), qr/502 Bad/, 'CNAME timeout');
+
+# several requests on CNAME pointing to invalid name
+
+my @s;
+
+push @s, http_host('cn03.example.net', '/long', start => 1);
+push @s, http_host('cn03.example.net', '/long', start => 1);
+
+like(http_end(pop @s), qr/502 Bad/, 'invalid CNAME - first');
+like(http_end(pop @s), qr/502 Bad/, 'invalid CNAME - last');
+
+# several requests on CNAME pointing to cached name
+
+@s = ();
+
+http_host('a.example.net', '/long');
+
+push @s, http_host('cn04.example.net', '/long', start => 1);
+push @s, http_host('cn04.example.net', '/long', start => 1);
+
+like(http_end(pop @s), qr/200 OK/, 'cached CNAME - first');
+like(http_end(pop @s), qr/200 OK/, 'cached CNAME - last');
+
+# several requests on CNAME pointing to name being resolved
+
+@s = ();
+
+my $s = http_host('cn06.example.net', '/long', start => 1);
+
+sleep 1;
+
+push @s, http_host('cn05.example.net', '/long', start => 1);
+push @s, http_host('cn05.example.net', '/long', start => 1);
+
+like(http_end(pop @s), qr/502 Bad/, 'CNAME in progress - first');
+like(http_end(pop @s), qr/502 Bad/, 'CNAME in progress - last');
+
+# several requests on CNAME pointing to name which times out
+# 1st request receives CNAME with short ttl
+# 2nd request replaces expired CNAME
+
+@s = ();
+
+push @s, http_host('cn07.example.net', '/long', start => 1);
+
+sleep 2;
+
+push @s, http_host('cn07.example.net', '/long', start => 1);
+
+like(http_end(pop @s), qr/502 Bad/, 'CNAME ttl - first');
+like(http_end(pop @s), qr/502 Bad/, 'CNAME ttl - last');
+
+# several requests on CNAME pointing to name
+# 1st request aborts before name is resolved
+# 2nd request finishes with name resolved
+
+@s = ();
+
+push @s, http_host('cn09.example.net', '/long', start => 1);
+push @s, http_host('cn09.example.net', '/long', start => 1);
+
+sleep 2;
+
+close(shift @s);
+
+like(http_end(pop @s), qr/200 OK/, 'abort on CNAME');
+
+like(http_host('cn001.example.net', '/short'), qr/502 Bad/, 'recurse uncached');
+
+###############################################################################
+
+sub http_host {
+	my ($host, $uri, %extra) = @_;
+	return http(<<EOF, %extra);
+GET $uri HTTP/1.0
+Host: $host
+
+EOF
+}
+
+###############################################################################
+
+sub reply_handler {
+	my ($recv_data, $port) = @_;
+
+	my (@name, @rdata);
+
+	use constant NOERROR	=> 0;
+
+	use constant A		=> 1;
+	use constant CNAME	=> 5;
+
+	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 'b.example.net' && $type == A) {
+		sleep 2;
+		push @rdata, rd_addr($ttl, '127.0.0.1');
+
+	} elsif ($name eq 'cn01.example.net') {
+		$ttl = 1;
+		push @rdata, pack("n3N nCa4n", 0xc00c, CNAME, IN, $ttl,
+			7, 4, "cn02", 0xc011);
+
+	} elsif ($name =~ /cn0[268].example.net/) {
+		# resolver timeout
+
+		return;
+
+	} elsif ($name eq 'cn03.example.net') {
+		select undef, undef, undef, 1.1;
+		push @rdata, pack("n3N nC", 0xc00c, CNAME, IN, $ttl, 0);
+
+	} elsif ($name eq 'cn04.example.net') {
+		select undef, undef, undef, 1.1;
+		push @rdata, pack("n3N nCa1n", 0xc00c, CNAME, IN, $ttl,
+			4, 1, "a", 0xc011);
+
+	} elsif ($name eq 'cn05.example.net') {
+		select undef, undef, undef, 1.1;
+		push @rdata, pack("n3N nCa4n", 0xc00c, CNAME, IN, $ttl,
+			7, 4, "cn06", 0xc011);
+
+	} elsif ($name eq 'cn07.example.net') {
+		$ttl = 1;
+		push @rdata, pack("n3N nCa4n", 0xc00c, CNAME, IN, $ttl,
+			7, 4, "cn08", 0xc011);
+
+	} elsif ($name eq 'cn09.example.net') {
+		if ($type == A) {
+			select undef, undef, undef, 1.1;
+		}
+		push @rdata, pack("n3N nCa1n", 0xc00c, CNAME, IN, $ttl,
+			4, 1, "b", 0xc011);
+
+	} elsif ($name eq 'cn052.example.net') {
+		if ($type == A) {
+			push @rdata, rd_addr($ttl, '127.0.0.1');
+		}
+
+	} elsif ($name =~ /cn0\d+.example.net/) {
+		my ($id) = $name =~ /cn(\d+)/;
+		$id++;
+		push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl,
+			8, 5, "cn$id", 0xc012);
+	}
+
+	$len = @name;
+	pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata,
+		0, 0, @name, $type, $class) . join('', @rdata);
+}
+
+sub rd_addr {
+	my ($ttl, $addr) = @_;
+
+	my $code = 'split(/\./, $addr)';
+
+	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);
+	}
+}
+
+###############################################################################