Skip to content

Commit f0038c1

Browse files
jimf5jimf5
committed
Tests: http resolver with ipv4/ipv6 parameters.
1 parent 7811a9a commit f0038c1

File tree

2 files changed

+221
-2
lines changed

2 files changed

+221
-2
lines changed

‎http_resolver_ipv4.t

+200
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,200 @@
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+
###############################################################################

‎proxy_protocol2.t

+21-2
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ select STDOUT; $| = 1;
2424

2525
my $t = Test::Nginx->new()->has(qw/http access realip/);
2626

27-
$t->write_file_expand('nginx.conf', <<'EOF')->plan(28);
27+
$t->write_file_expand('nginx.conf', <<'EOF')->plan(32);
2828
2929
%%TEST_GLOBALS%%
3030
@@ -41,6 +41,9 @@ http {
4141
add_header X-IP $remote_addr!$remote_port;
4242
add_header X-PP $proxy_protocol_addr!$proxy_protocol_port;
4343
add_header X-PPS $proxy_protocol_server_addr!$proxy_protocol_server_port;
44+
add_header X-TL $proxy_protocol_tlv_0x3-$proxy_protocol_tlv_0x0000ae-$proxy_protocol_tlv_0x0f;
45+
add_header X-NT $proxy_protocol_tlv_unique_id-$proxy_protocol_tlv_ssl_cn-$proxy_protocol_tlv_ssl_0x22-$proxy_protocol_tlv_ssl_verify;
46+
add_header X-NT2 $proxy_protocol_tlv_unique_id-$proxy_protocol_tlv_ssl_version-$proxy_protocol_tlv_ssl_0x21-$proxy_protocol_tlv_ssl_verify;
4447
4548
server {
4649
listen 127.0.0.1:8080 proxy_protocol;
@@ -78,8 +81,14 @@ my $p = pack("N3C", 0x0D0A0D0A, 0x000D0A51, 0x5549540A, 0x21);
7881
my $tcp4 = $p . pack("CnN2n2", 0x11, 12, 0xc0000201, 0xc0000202, 123, 5678);
7982
my $tcp6 = $p . pack("CnNx8NNx8Nn2", 0x21, 36,
8083
0x20010db8, 0x00000001, 0x20010db8, 0x00000002, 123, 5678);
81-
my $tlv = $p . pack("CnN2n2x9", 0x11, 21, 0xc0000201, 0xc0000202, 123, 5678);
8284
my $unk1 = $p . pack("Cxx", 0x01);
85+
my $tlv = $p . pack("CnN2n2N3", 0x11, 24, 0xc0000201, 0xc0000202, 123, 5678,
86+
0x03000141, 0xAE000531, 0x32333435);
87+
my $tlv2 = $p . pack("CnN2n2N7", 0x11, 40, 0xc0000201, 0xc0000202, 123, 5678,
88+
0x05000555, 0x4E495151,0x20001100, 0xdeadbeef, 0x22000966, 0x6f6f2e62,
89+
0x61727272);
90+
my $tlv3 = $p . pack("CnN2n2N5", 0x11, 32, 0xc0000201, 0xc0000202, 123, 5678,
91+
0x05000555, 0x4E495151,0x20000900, 0x00000000, 0x23000132);
8392
my $unk2 = $p . pack("CnC4", 0x41, 4, 1, 2, 3, 4);
8493
my $r;
8594

@@ -102,6 +111,16 @@ like($r, qr/SEE-THIS/, 'tlv request');
102111
like($r, qr/X-PP: 192.0.2.1!123\x0d/, 'tlv proxy');
103112
like($r, qr/X-PPS: 192.0.2.2!5678\x0d/, 'tlv proxy server');
104113
unlike($r, qr/X-IP: (192.0.2.1|[^!]+!123\x0d)/, 'tlv client');
114+
like($r, qr/X-TL: A-12345-\x0d/, 'tlv raw variables');
115+
like($r, qr/X-NT: ---\x0d/, 'tlv missing variables');
116+
117+
$r = pp_get('/t1', $tlv2);
118+
like($r, qr/X-NT: UNIQQ-foo.barrr-foo.barrr-3735928559\x0d/,
119+
'tlv named variables');
120+
121+
$r = pp_get('/t1', $tlv3);
122+
like($r, qr/X-NT2: UNIQQ-2-2-0\x0d/,
123+
'tlv named variables');
105124

106125
$r = pp_get('/t1', $unk1);
107126
like($r, qr/SEE-THIS/, 'unknown request 1');

0 commit comments

Comments
 (0)