# HG changeset patch # User Sergey Kandaurov # Date 1494938710 -10800 # Node ID eaa92d810bbbd547c724b5d9ff92fecfeef1d63b # Parent 717030fd5a94496952a8d42f206166bef4203a7d Tests: realip tests for 'unix:' and hostname in set_real_ip_from. diff --git a/realip_hostname.t b/realip_hostname.t new file mode 100644 --- /dev/null +++ b/realip_hostname.t @@ -0,0 +1,85 @@ +#!/usr/bin/perl + +# (C) Sergey Kandaurov +# (C) Nginx, Inc. + +# Tests for nginx realip module, 'unix:' and hostname in set_real_ip_from. + +############################################################################### + +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 realip proxy unix/); + +$t->write_file_expand('nginx.conf', <<'EOF'); + +%%TEST_GLOBALS%% + +daemon off; + +events { +} + +http { + %%TEST_GLOBALS_HTTP%% + + server { + listen 127.0.0.1:8080; + listen unix:%%TESTDIR%%/unix.sock; + server_name localhost; + + location /1 { + set_real_ip_from localhost; + add_header X-IP $remote_addr; + } + + location /2 { + set_real_ip_from unix:; + add_header X-IP $remote_addr; + } + + location /unix { + proxy_pass http://unix:%%TESTDIR%%/unix.sock:/; + proxy_set_header X-Real-IP 192.0.2.1; + } + + location /ip { + proxy_pass http://127.0.0.1:8080/; + proxy_set_header X-Real-IP 192.0.2.1; + } + } +} + +EOF + +$t->write_file('1', ''); +$t->write_file('2', ''); +$t->try_run('no realip hostnames support'); + +plan(skip_all => 'no 127.0.0.1 on host') + if http_get('/1') !~ /X-IP: 127.0.0.1/m; + +$t->plan(4); + +############################################################################### + +like(http_get('/unix/2'), qr/X-IP: 192.0.2.1/, 'realip unix'); +unlike(http_get('/unix/1'), qr/X-IP: 192.0.2.1/, 'realip unix - no match'); + +like(http_get('/ip/1'), qr/X-IP: 192.0.2.1/, 'realip hostname'); +unlike(http_get('/ip/2'), qr/X-IP: 192.0.2.1/, 'realip hostname - no match'); + +############################################################################### diff --git a/stream_realip_hostname.t b/stream_realip_hostname.t 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); +} + +###############################################################################