blob: 2342eb6308af476a1463ee35fb2836b59fbe1e56 [file] [log] [blame]
#!/usr/bin/perl
# (C) Sergey Kandaurov
# (C) Nginx, Inc.
# Tests for haproxy protocol.
###############################################################################
use warnings;
use strict;
use Test::More;
BEGIN { use FindBin; chdir($FindBin::Bin); }
use lib 'lib';
use Test::Nginx;
###############################################################################
select STDERR; $| = 1;
select STDOUT; $| = 1;
my $t = Test::Nginx->new()->has(qw/http access realip/);
$t->write_file_expand('nginx.conf', <<'EOF')->plan(21);
%%TEST_GLOBALS%%
daemon off;
events {
}
http {
%%TEST_GLOBALS_HTTP%%
log_format pp '$remote_addr $request';
server {
listen 127.0.0.1:8080 proxy_protocol;
server_name localhost;
set_real_ip_from 127.0.0.1/32;
add_header X-IP $remote_addr;
add_header X-PP $proxy_protocol_addr;
location /pp {
real_ip_header proxy_protocol;
error_page 404 =200 /t1;
access_log %%TESTDIR%%/pp.log pp;
location /pp_4 {
deny 192.0.2.1/32;
}
location /pp_6 {
deny 2001:DB8::1/128;
}
}
}
}
EOF
$t->write_file('t1', 'SEE-THIS');
$t->run();
###############################################################################
my $p = pack("N3C", 0x0D0A0D0A, 0x000D0A51, 0x5549540A, 0x21);
my $tcp4 = $p . pack("CnN2n2", 0x11, 12, 0xc0000201, 0xc0000202, 1234, 5678);
my $tcp6 = $p . pack("CnNx8NNx8Nn2", 0x21, 36,
0x20010db8, 0x00000001, 0x20010db8, 0x00000002, 1234, 5678);
my $tlv = $p . pack("CnN2n2x9", 0x11, 21, 0xc0000201, 0xc0000202, 1234, 5678);
my $unk1 = $p . pack("Cxx", 0x01);
my $unk2 = $p . pack("CnC4", 0x41, 4, 1, 2, 3, 4);
my $r;
# no realip, just PROXY header parsing
$r = pp_get('/t1', $tcp4);
like($r, qr/SEE-THIS/, 'tcp4 request');
like($r, qr/X-PP: 192.0.2.1/, 'tcp4 proxy');
unlike($r, qr/X-IP: 192.0.2.1/, 'tcp4 client');
$r = pp_get('/t1', $tcp6);
like($r, qr/SEE-THIS/, 'tcp6 request');
like($r, qr/X-PP: 2001:DB8::1/i, 'tcp6 proxy');
unlike($r, qr/X-IP: 2001:DB8::1/i, 'tcp6 client');
$r = pp_get('/t1', $tlv);
like($r, qr/SEE-THIS/, 'tlv request');
like($r, qr/X-PP: 192.0.2.1/, 'tlv proxy');
unlike($r, qr/X-IP: 192.0.2.1/, 'tlv client');
like(pp_get('/t1', $unk1), qr/SEE-THIS/, 'unknown request 1');
like(pp_get('/t1', $unk2), qr/SEE-THIS/, 'unknown request 2');
# realip
$r = pp_get('/pp', $tcp4);
like($r, qr/SEE-THIS/, 'tcp4 request realip');
like($r, qr/X-PP: 192.0.2.1/, 'tcp4 proxy realip');
like($r, qr/X-IP: 192.0.2.1/, 'tcp4 client realip');
$r = pp_get('/pp', $tcp6);
like($r, qr/SEE-THIS/, 'tcp6 request realip');
like($r, qr/X-PP: 2001:DB8::1/i, 'tcp6 proxy realip');
like($r, qr/X-IP: 2001:DB8::1/i, 'tcp6 client realip');
# access
$r = pp_get('/pp_4', $tcp4);
like($r, qr/403 Forbidden/, 'tcp4 access');
$r = pp_get('/pp_6', $tcp6);
like($r, qr/403 Forbidden/, 'tcp6 access');
# client address in access.log
$t->stop();
my $log = $t->read_file('pp.log');
like($log, qr!^192\.0\.2\.1 GET /pp_4!m, 'tcp4 access log');
like($log, qr!^2001:DB8::1 GET /pp_6!mi, 'tcp6 access log');
###############################################################################
sub pp_get {
my ($url, $proxy) = @_;
return http($proxy . <<EOF);
GET $url HTTP/1.0
Host: localhost
EOF
}
###############################################################################