Mercurial > hg > nginx-tests
comparison http_resolver_ipv4.t @ 1797:0b06942f0b8b
Tests: http resolver with ipv4/ipv6 parameters.
author | Eugene Grebenschikov <e.grebenshchikov@f5.com> |
---|---|
date | Wed, 12 Oct 2022 11:55:32 -0700 |
parents | |
children | 1786f49cca17 |
comparison
equal
deleted
inserted
replaced
1796:e1fd234baac0 | 1797:0b06942f0b8b |
---|---|
1 #!/usr/bin/perl | |
2 | |
3 # (C) Sergey Kandaurov | |
4 # (C) Eugene Grebenschikov | |
5 # (C) Nginx, Inc. | |
6 | |
7 # Tests for http resolver with ipv4/ipv6 parameters. | |
8 | |
9 ############################################################################### | |
10 | |
11 use warnings; | |
12 use strict; | |
13 | |
14 use Test::More; | |
15 | |
16 BEGIN { use FindBin; chdir($FindBin::Bin); } | |
17 | |
18 use lib 'lib'; | |
19 use Test::Nginx; | |
20 | |
21 ############################################################################### | |
22 | |
23 select STDERR; $| = 1; | |
24 select STDOUT; $| = 1; | |
25 | |
26 my $t = Test::Nginx->new()->has(qw/http proxy rewrite/); | |
27 | |
28 $t->write_file_expand('nginx.conf', <<'EOF'); | |
29 | |
30 %%TEST_GLOBALS%% | |
31 | |
32 daemon off; | |
33 | |
34 events { | |
35 } | |
36 | |
37 http { | |
38 %%TEST_GLOBALS_HTTP%% | |
39 | |
40 server { | |
41 listen 127.0.0.1:8080; | |
42 server_name localhost; | |
43 | |
44 location / { | |
45 proxy_pass http://$arg_h:%%PORT_8081%%/; | |
46 resolver 127.0.0.1:%%PORT_8980_UDP%% ipv4=on ipv6=on; | |
47 } | |
48 | |
49 location /ipv4 { | |
50 proxy_pass http://$arg_h:%%PORT_8081%%/; | |
51 resolver 127.0.0.1:%%PORT_8980_UDP%% ipv4=on ipv6=off; | |
52 } | |
53 | |
54 location /ipv6 { | |
55 proxy_pass http://$arg_h:%%PORT_8081%%/; | |
56 resolver 127.0.0.1:%%PORT_8980_UDP%% ipv4=off ipv6=on; | |
57 } | |
58 } | |
59 | |
60 server { | |
61 listen 127.0.0.1:8081; | |
62 server_name localhost; | |
63 | |
64 location / { | |
65 return 200 "ipv4"; | |
66 } | |
67 } | |
68 | |
69 server { | |
70 listen [::1]:%%PORT_8081%%; | |
71 server_name localhost; | |
72 | |
73 location / { | |
74 return 200 "ipv6"; | |
75 } | |
76 } | |
77 } | |
78 | |
79 EOF | |
80 | |
81 $t->try_run('no resolver ipv4')->plan(3); | |
82 | |
83 $t->run_daemon(\&dns_daemon, port(8980), $t); | |
84 $t->waitforfile($t->testdir . '/' . port(8980)); | |
85 | |
86 ############################################################################### | |
87 | |
88 like(many('/', 10), qr/ipv4: \d+, ipv6: \d+/, 'ipv4 ipv6'); | |
89 is(many('/ipv4', 10), 'ipv4: 10', 'ipv4 only'); | |
90 is(many('/ipv6', 10), 'ipv6: 10', 'ipv6 only'); | |
91 | |
92 ############################################################################### | |
93 | |
94 sub many { | |
95 my ($uri, $count) = @_; | |
96 my %hits; | |
97 | |
98 for (1 .. $count) { | |
99 if (http_get("$uri?h=example.com") =~ /(ipv(4|6))/) {; | |
100 $hits{$1} = 0 unless defined $hits{$1}; | |
101 $hits{$1}++; | |
102 } | |
103 } | |
104 | |
105 return join ', ', map { $_ . ": " . $hits{$_} } sort keys %hits; | |
106 } | |
107 | |
108 ############################################################################### | |
109 | |
110 sub reply_handler { | |
111 my ($recv_data, $port) = @_; | |
112 | |
113 my (@name, @rdata); | |
114 | |
115 use constant NOERROR => 0; | |
116 | |
117 use constant A => 1; | |
118 use constant AAAA => 28; | |
119 | |
120 use constant IN => 1; | |
121 | |
122 # default values | |
123 | |
124 my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600); | |
125 | |
126 # decode name | |
127 | |
128 my ($len, $offset) = (undef, 12); | |
129 while (1) { | |
130 $len = unpack("\@$offset C", $recv_data); | |
131 last if $len == 0; | |
132 $offset++; | |
133 push @name, unpack("\@$offset A$len", $recv_data); | |
134 $offset += $len; | |
135 } | |
136 | |
137 $offset -= 1; | |
138 my ($id, $type, $class) = unpack("n x$offset n2", $recv_data); | |
139 | |
140 my $name = join('.', @name); | |
141 if ($name eq 'example.com') { | |
142 if ($type == A) { | |
143 push @rdata, rd_addr($ttl, '127.0.0.1'); | |
144 } | |
145 if ($type == AAAA) { | |
146 push @rdata, rd_addr6($ttl, "::1"); | |
147 } | |
148 } | |
149 | |
150 $len = @name; | |
151 pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata, | |
152 0, 0, @name, $type, $class) . join('', @rdata); | |
153 } | |
154 | |
155 sub rd_addr { | |
156 my ($ttl, $addr) = @_; | |
157 | |
158 my $code = 'split(/\./, $addr)'; | |
159 | |
160 pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code); | |
161 } | |
162 | |
163 sub expand_ip6 { | |
164 my ($addr) = @_; | |
165 | |
166 substr ($addr, index($addr, "::"), 2) = | |
167 join "0", map { ":" } (0 .. 8 - (split /:/, $addr) + 1); | |
168 map { hex "0" x (4 - length $_) . "$_" } split /:/, $addr; | |
169 } | |
170 | |
171 sub rd_addr6 { | |
172 my ($ttl, $addr) = @_; | |
173 | |
174 pack 'n3N nn8', 0xc00c, AAAA, IN, $ttl, 16, expand_ip6($addr); | |
175 } | |
176 | |
177 sub dns_daemon { | |
178 my ($port, $t) = @_; | |
179 | |
180 my ($data, $recv_data); | |
181 my $socket = IO::Socket::INET->new( | |
182 LocalAddr => '127.0.0.1', | |
183 LocalPort => $port, | |
184 Proto => 'udp', | |
185 ) | |
186 or die "Can't create listening socket: $!\n"; | |
187 | |
188 # signal we are ready | |
189 | |
190 open my $fh, '>', $t->testdir() . '/' . $port; | |
191 close $fh; | |
192 | |
193 while (1) { | |
194 $socket->recv($recv_data, 65536); | |
195 $data = reply_handler($recv_data, $port); | |
196 $socket->send($data); | |
197 } | |
198 } | |
199 | |
200 ############################################################################### |