868
|
1 #!/usr/bin/perl
|
|
2
|
|
3 # (C) Andrey Zelenkov
|
|
4 # (C) Nginx, Inc.
|
|
5
|
|
6 # Tests for stream limit_conn module with datagrams.
|
|
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 use Test::Nginx::Stream qw/ dgram /;
|
|
20
|
|
21 ###############################################################################
|
|
22
|
|
23 select STDERR; $| = 1;
|
|
24 select STDOUT; $| = 1;
|
|
25
|
|
26 my $t = Test::Nginx->new()->has(qw/stream stream_limit_conn shmem/)
|
|
27 ->write_file_expand('nginx.conf', <<'EOF');
|
|
28
|
|
29 %%TEST_GLOBALS%%
|
|
30
|
|
31 daemon off;
|
|
32
|
|
33 events {
|
|
34 }
|
|
35
|
|
36 stream {
|
|
37 limit_conn_zone $binary_remote_addr zone=zone:1m;
|
|
38 limit_conn_zone $binary_remote_addr zone=zone2:1m;
|
|
39
|
|
40 proxy_responses 1;
|
|
41 proxy_timeout 1s;
|
|
42
|
|
43 server {
|
|
44 listen 127.0.0.1:8081 udp;
|
|
45 proxy_pass 127.0.0.1:8080;
|
|
46
|
|
47 limit_conn zone 1;
|
|
48 proxy_responses 2;
|
|
49 }
|
|
50
|
|
51 server {
|
|
52 listen 127.0.0.1:8082 udp;
|
|
53 proxy_pass 127.0.0.1:8080;
|
|
54 limit_conn zone2 1;
|
|
55 }
|
|
56
|
|
57 server {
|
|
58 listen 127.0.0.1:8083 udp;
|
|
59 proxy_pass 127.0.0.1:8080;
|
|
60 limit_conn zone 5;
|
|
61 }
|
|
62
|
|
63 server {
|
|
64 listen 127.0.0.1:8084 udp;
|
|
65 proxy_pass 127.0.0.1:8081;
|
|
66 limit_conn zone2 1;
|
|
67 }
|
|
68
|
|
69 server {
|
|
70 listen 127.0.0.1:8085 udp;
|
|
71 proxy_pass 127.0.0.1:8081;
|
|
72 limit_conn zone 1;
|
|
73 }
|
|
74 }
|
|
75
|
|
76 EOF
|
|
77
|
|
78 $t->try_run('no stream udp')->plan(8);
|
|
79 $t->run_daemon(\&udp_daemon, $t);
|
|
80 $t->waitforfile($t->testdir . '/8080');
|
|
81
|
|
82 ###############################################################################
|
|
83
|
|
84 my $str = '.';
|
|
85
|
|
86 # same and other zones
|
|
87
|
|
88 my $s = dgram('127.0.0.1:8081');
|
|
89
|
|
90 is($s->io('1'), '1', 'passed');
|
|
91
|
|
92 is(dgram('127.0.0.1:8081')->io('1'), '', 'rejected same zone');
|
|
93 is(dgram('127.0.0.1:8082')->io('1'), '1', 'passed different zone');
|
|
94 is(dgram('127.0.0.1:8083')->io('1'), '1', 'passed same zone unlimited');
|
|
95
|
|
96 is($s->io('2', read => 2), '12', 'long connection');
|
|
97
|
|
98 is(dgram('127.0.0.1:8081')->io('2', read => 2), '12', 'passed 2');
|
|
99
|
|
100 # zones proxy chain
|
|
101
|
|
102 is(dgram('127.0.0.1:8084')->io('1'), '1', 'passed proxy');
|
|
103 is(dgram('127.0.0.1:8085')->io('1'), '', 'rejected proxy');
|
|
104
|
|
105 ###############################################################################
|
|
106
|
|
107 sub udp_daemon {
|
|
108 my $t = shift;
|
|
109
|
|
110 my $server = IO::Socket::INET->new(
|
|
111 Proto => 'udp',
|
|
112 LocalAddr => '127.0.0.1:8080',
|
|
113 Reuse => 1,
|
|
114 )
|
|
115 or die "Can't create listening socket: $!\n";
|
|
116
|
|
117 # signal we are ready
|
|
118
|
|
119 open my $fh, '>', $t->testdir() . '/8080';
|
|
120 close $fh;
|
|
121
|
|
122 while (1) {
|
|
123 $server->recv(my $buffer, 65536);
|
|
124 $server->send($_) for (1 .. $buffer);
|
|
125 }
|
|
126 }
|
|
127
|
|
128 ###############################################################################
|