Mercurial > hg > nginx-tests
diff stream_realip_hostname.t @ 1176:eaa92d810bbb
Tests: realip tests for 'unix:' and hostname in set_real_ip_from.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Tue, 16 May 2017 15:45:10 +0300 |
parents | |
children | 97c8280de681 |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/stream_realip_hostname.t @@ -0,0 +1,100 @@ +#!/usr/bin/perl + +# (C) Sergey Kandaurov +# (C) Nginx, Inc. + +# Tests for stream realip module, 'unix:' and hostname in set_real_ip_from. + +############################################################################### + +use warnings; +use strict; + +use Test::More; + +use Socket qw/ $CRLF /; + +BEGIN { use FindBin; chdir($FindBin::Bin); } + +use lib 'lib'; +use Test::Nginx; +use Test::Nginx::Stream qw/ stream /; + +############################################################################### + +select STDERR; $| = 1; +select STDOUT; $| = 1; + +my $t = Test::Nginx->new()->has(qw/stream stream_return stream_realip unix/) + ->write_file_expand('nginx.conf', <<'EOF'); + +%%TEST_GLOBALS%% + +daemon off; + +events { +} + +stream { + server { + listen unix:%%TESTDIR%%/unix.sock proxy_protocol; + listen 127.0.0.1:8080; + listen 127.0.0.1:8082 proxy_protocol; + return $remote_addr; + + set_real_ip_from unix:; + } + + server { + listen 127.0.0.1:8081; + proxy_pass unix:%%TESTDIR%%/unix.sock; + } + + server { + listen 127.0.0.1:8085 proxy_protocol; + listen unix:%%TESTDIR%%/unix2.sock proxy_protocol; + return $remote_addr; + + set_real_ip_from localhost; + } + + server { + listen 127.0.0.1:8083; + proxy_pass 127.0.0.1:8085; + } + + server { + listen 127.0.0.1:8084; + proxy_pass unix:%%TESTDIR%%/unix2.sock; + } +} + +EOF + +$t->try_run('no stream realip hostnames support'); + +plan(skip_all => 'no 127.0.0.1 on host') + if http_get('/') ne '127.0.0.1'; + +$t->plan(4); + +############################################################################### + +is(pp_get(8081, "PROXY TCP4 192.0.2.1 192.0.2.2 1234 5678${CRLF}"), + '192.0.2.1', 'realip unix'); +isnt(pp_get(8082, "PROXY TCP4 192.0.2.1 192.0.2.2 1234 5678${CRLF}"), + '192.0.2.1', 'realip unix - no match'); + +is(pp_get(8083, "PROXY TCP4 192.0.2.1 192.0.2.2 1234 5678${CRLF}"), + '192.0.2.1', 'realip hostname'); +isnt(pp_get(8084, "PROXY TCP4 192.0.2.1 192.0.2.2 1234 5678${CRLF}"), + '192.0.2.1', 'realip hostname - no match'); + +############################################################################### + +sub pp_get { + my ($port, $proxy) = @_; + stream(PeerPort => port($port))->io($proxy); +} + +###############################################################################