Mercurial > hg > nginx-tests

comparison proxy_protocol2_server.t @ 1527:7676944968c1

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: proxy protocol server variables.
author Sergey Kandaurov <pluknet@nginx.com>
date Thu, 24 Oct 2019 15:00:30 +0300
parents
children 5ac6efbe5552
comparison
equal deleted inserted replaced
1526:b3bbb59dc324 1527:7676944968c1
1 #!/usr/bin/perl
2
3 # (C) Sergey Kandaurov
4 # (C) Nginx, Inc.
5
6 # Tests for haproxy protocol.
7
8 ###############################################################################
9
10 use warnings;
11 use strict;
12
13 use Test::More;
14
15 BEGIN { use FindBin; chdir($FindBin::Bin); }
16
17 use lib 'lib';
18 use Test::Nginx;
19
20 ###############################################################################
21
22 select STDERR; $| = 1;
23 select STDOUT; $| = 1;
24
25 my $t = Test::Nginx->new()->has(qw/http access realip/);
26
27 $t->write_file_expand('nginx.conf', <<'EOF');
28
29 %%TEST_GLOBALS%%
30
31 daemon off;
32
33 events {
34 }
35
36 http {
37 %%TEST_GLOBALS_HTTP%%
38
39 log_format pp $remote_addr:$remote_port;
40
41 add_header X-IP $remote_addr!$remote_port;
42 add_header X-PP $proxy_protocol_addr!$proxy_protocol_port;
43 add_header X-PPS $proxy_protocol_server_addr!$proxy_protocol_server_port;
44
45 server {
46 listen 127.0.0.1:8080 proxy_protocol;
47 server_name localhost;
48
49 set_real_ip_from 127.0.0.1/32;
50
51 location /pp {
52 real_ip_header proxy_protocol;
53 error_page 404 =200 /t1;
54
55 location /pp_4 {
56 deny 192.0.2.1/32;
57 access_log %%TESTDIR%%/pp4.log pp;
58 }
59
60 location /pp_6 {
61 deny 2001:DB8::1/128;
62 access_log %%TESTDIR%%/pp6.log pp;
63 }
64 }
65
66 location / { }
67 }
68 }
69
70 EOF
71
72 $t->write_file('t1', 'SEE-THIS');
73 $t->try_run('no proxy_protocol_server_addr')->plan(28);
74
75 ###############################################################################
76
77 my $p = pack("N3C", 0x0D0A0D0A, 0x000D0A51, 0x5549540A, 0x21);
78 my $tcp4 = $p . pack("CnN2n2", 0x11, 12, 0xc0000201, 0xc0000202, 123, 567);
79 my $tcp6 = $p . pack("CnNx8NNx8Nn2", 0x21, 36,
80 0x20010db8, 0x00000001, 0x20010db8, 0x00000002, 123, 567);
81 my $tlv = $p . pack("CnN2n2x9", 0x11, 21, 0xc0000201, 0xc0000202, 123, 567);
82 my $unk1 = $p . pack("Cxx", 0x01);
83 my $unk2 = $p . pack("CnC4", 0x41, 4, 1, 2, 3, 4);
84 my $r;
85
86 # no realip, just PROXY header parsing
87
88 $r = pp_get('/t1', $tcp4);
89 like($r, qr/SEE-THIS/, 'tcp4 request');
90 like($r, qr/X-PP: 192.0.2.1!123\x0d/, 'tcp4 proxy');
91 like($r, qr/X-PPS: 192.0.2.2!567\x0d/, 'tcp4 proxy server');
92 unlike($r, qr/X-IP: (192.0.2.1|[^!]+!123\x0d)/, 'tcp4 client');
93
94 $r = pp_get('/t1', $tcp6);
95 like($r, qr/SEE-THIS/, 'tcp6 request');
96 like($r, qr/X-PP: 2001:DB8::1!123\x0d/i, 'tcp6 proxy');
97 like($r, qr/X-PPS: 2001:DB8::2!567\x0d/i, 'tcp6 proxy server');
98 unlike($r, qr/X-IP: (2001:DB8::1|[^!]+!123\x0d)/i, 'tcp6 client');
99
100 $r = pp_get('/t1', $tlv);
101 like($r, qr/SEE-THIS/, 'tlv request');
102 like($r, qr/X-PP: 192.0.2.1!123\x0d/, 'tlv proxy');
103 like($r, qr/X-PPS: 192.0.2.2!567\x0d/, 'tlv proxy server');
104 unlike($r, qr/X-IP: (192.0.2.1|[^!]+!123\x0d)/, 'tlv client');
105
106 $r = pp_get('/t1', $unk1);
107 like($r, qr/SEE-THIS/, 'unknown request 1');
108 like($r, qr/X-PP: !\x0d/, 'unknown proxy 1');
109 like($r, qr/X-PPS: !\x0d/, 'unknown proxy server 1');
110
111 $r = pp_get('/t1', $unk2);
112 like($r, qr/SEE-THIS/, 'unknown request 2');
113 like($r, qr/X-PP: !\x0d/, 'unknown proxy 2');
114 like($r, qr/X-PPS: !\x0d/, 'unknown proxy server 2');
115
116 # realip
117
118 $r = pp_get('/pp', $tcp4);
119 like($r, qr/SEE-THIS/, 'tcp4 request realip');
120 like($r, qr/X-PP: 192.0.2.1!123\x0d/, 'tcp4 proxy realip');
121 like($r, qr/X-IP: 192.0.2.1!123\x0d/, 'tcp4 client realip');
122
123 $r = pp_get('/pp', $tcp6);
124 like($r, qr/SEE-THIS/, 'tcp6 request realip');
125 like($r, qr/X-PP: 2001:DB8::1!123\x0d/i, 'tcp6 proxy realip');
126 like($r, qr/X-IP: 2001:DB8::1!123\x0d/i, 'tcp6 client realip');
127
128 # access
129
130 $r = pp_get('/pp_4', $tcp4);
131 like($r, qr/403 Forbidden/, 'tcp4 access');
132
133 $r = pp_get('/pp_6', $tcp6);
134 like($r, qr/403 Forbidden/, 'tcp6 access');
135
136 # client address in access.log
137
138 $t->stop();
139
140 is($t->read_file('pp4.log'), "192.0.2.1:123\n", 'tcp4 log');
141 is($t->read_file('pp6.log'), "2001:db8::1:123\n", 'tcp6 log');
142
143 ###############################################################################
144
145 sub pp_get {
146 my ($url, $proxy) = @_;
147 return http($proxy . <<EOF);
148 GET $url HTTP/1.0
149 Host: localhost
150
151 EOF
152 }
153
154 ###############################################################################