Mercurial > hg > nginx-tests

annotate stream_limit_rate.t @ 816:77359b849cd5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: stream package.
author Andrey Zelenkov <zelenkov@nginx.com>
date Mon, 20 Jul 2015 15:06:09 +0300
parents a53c310c695d
children 5048b8f0fedd
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
1 #!/usr/bin/perl
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
2
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
3 # (C) Andrey Zelenkov
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
4 # (C) Nginx, Inc.
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
5
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
6 # Tests for stream proxy module, limit rate directives.
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
7
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
8 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
9
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
10 use warnings;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
11 use strict;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
12
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
13 use Test::More;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
14
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
15 use IO::Select;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
16
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
17 BEGIN { use FindBin; chdir($FindBin::Bin); }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
18
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
19 use lib 'lib';
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
20 use Test::Nginx;
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
21 use Test::Nginx::Stream qw/ stream /;
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
22
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
23 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
24
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
25 select STDERR; $| = 1;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
26 select STDOUT; $| = 1;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
27
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
28 my $t = Test::Nginx->new()->has(qw/stream/)
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
29 ->write_file_expand('nginx.conf', <<'EOF');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
30
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
31 %%TEST_GLOBALS%%
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
32
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
33 daemon off;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
34
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
35 events {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
36 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
37
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
38 stream {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
39 # download and upload rates are set equal to the maximum
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
40 # number of bytes transmitted
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
41
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
42 # proxy_download_rate value comes from following calculations:
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
43 # test string length (1000) + whitespace (1) + time string length (10)
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
44
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
45 proxy_download_rate 1011;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
46 proxy_upload_rate 1000;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
47
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
48 server {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
49 listen 127.0.0.1:8081;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
50 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
51 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
52
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
53 server {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
54 listen 127.0.0.1:8082;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
55 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
56 proxy_download_rate 0;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
57 proxy_upload_rate 0;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
58 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
59
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
60 server {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
61 listen 127.0.0.1:8083;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
62 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
63 proxy_download_rate 1;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
64 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
65
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
66 server {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
67 listen 127.0.0.1:8084;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
68 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
69 proxy_upload_rate 1;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
70 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
71
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
72 server {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
73 listen 127.0.0.1:8085;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
74 proxy_pass 127.0.0.1:8080;
634
40675bfad7d3 Tests: fixed stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 632
diff changeset
75 proxy_download_rate 250;
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
76 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
77
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
78 server {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
79 listen 127.0.0.1:8086;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
80 proxy_pass 127.0.0.1:8090;
634
40675bfad7d3 Tests: fixed stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 632
diff changeset
81 proxy_upload_rate 250;
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
82 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
83 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
84
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
85 EOF
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
86
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
87 $t->run_daemon(\&stream_daemon, 8080);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
88 $t->run_daemon(\&stream_daemon, 8090);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
89
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
90 $t->try_run('no proxy_download_rate and/or proxy_upload_rate')->plan(8);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
91
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
92 $t->waitforsocket('127.0.0.1:8080');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
93 $t->waitforsocket('127.0.0.1:8090');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
94
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
95 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
96
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
97 my $str = '1234567890' x 100;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
98
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
99 my %r = response($str, peer => '127.0.0.1:8081');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
100 is($r{'data'}, $str, 'exact limit');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
101
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
102 %r = response($str, peer => '127.0.0.1:8082');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
103 is($r{'data'}, $str, 'unlimited');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
104
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
105 SKIP: {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
106 skip 'unsafe on VM', 2 unless $ENV{TEST_NGINX_UNSAFE};
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
107
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
108 # if interaction between backend and client is slow then proxy can add extra
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
109 # bytes to upload/download data
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
110
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
111 %r = response($str, peer => '127.0.0.1:8083', readonce => 1);
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
112 is($r{'data'}, '1', 'download - one byte');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
113
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
114 %r = response($str, peer => '127.0.0.1:8084');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
115 is($r{'data'}, '1', 'upload - one byte');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
116
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
117 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
118
787
a53c310c695d Tests: relaxed limit_rate tests timeouts.
Sergey Kandaurov <pluknet@nginx.com>
parents: 711
diff changeset
119 # Five chunks are split with four 1s delays:
710
aed139eaaf3f Tests: fixed comment after 40675bfad7d3.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 637
diff changeset
120 # the first four chunks are quarters of test string
aed139eaaf3f Tests: fixed comment after 40675bfad7d3.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 637
diff changeset
121 # and the fifth one is some extra data from backend.
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
122
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
123 %r = response($str, peer => '127.0.0.1:8085');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
124 my $diff = time() - $r{'time'};
787
a53c310c695d Tests: relaxed limit_rate tests timeouts.
Sergey Kandaurov <pluknet@nginx.com>
parents: 711
diff changeset
125 cmp_ok($diff, '>=', 4, 'download - time');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
126 is($r{'data'}, $str, 'download - data');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
127
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
128 my $time = time();
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
129 %r = response($str . 'close', peer => '127.0.0.1:8086');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
130 $diff = time() - $time;
787
a53c310c695d Tests: relaxed limit_rate tests timeouts.
Sergey Kandaurov <pluknet@nginx.com>
parents: 711
diff changeset
131 cmp_ok($diff, '>=', 4, 'upload - time');
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
132 is($r{'data'}, $str . 'close', 'upload - data');
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
133
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
134 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
135
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
136 sub response {
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
137 my ($data, %extra) = @_;
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
138
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
139 my $s = stream($extra{peer});
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
140 $s->write($data);
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
141
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
142 $data = '';
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
143 while (1) {
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
144 my $buf = $s->read();
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
145 last unless length($buf);
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
146
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
147 $data .= $buf;
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
148
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
149 last if $extra{'readonce'};
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
150 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
151 $data =~ /([\S]*)\s?(\d+)?/;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
152
816
77359b849cd5 Tests: stream package.
Andrey Zelenkov <zelenkov@nginx.com>
parents: 787
diff changeset
153 return ('data' => $1, 'time' => $2)
632
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
154 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
155
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
156 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
157
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
158 sub stream_daemon {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
159 my $port = shift;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
160 my $server = IO::Socket::INET->new(
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
161 Proto => 'tcp',
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
162 LocalAddr => '127.0.0.1',
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
163 LocalPort => $port,
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
164 Listen => 5,
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
165 Reuse => 1
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
166 )
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
167 or die "Can't create listening socket: $!\n";
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
168
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
169 my $sel = IO::Select->new($server);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
170
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
171 local $SIG{PIPE} = 'IGNORE';
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
172
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
173 while (my @ready = $sel->can_read) {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
174 foreach my $fh (@ready) {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
175 if ($server == $fh) {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
176 my $new = $fh->accept;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
177 $new->autoflush(1);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
178 $sel->add($new);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
179
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
180 } elsif (stream_handle_client($fh)) {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
181 $sel->remove($fh);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
182 $fh->close;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
183 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
184 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
185 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
186 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
187
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
188 sub stream_handle_client {
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
189 my ($client) = @_;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
190
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
191 log2c("(new connection $client)");
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
192
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
193 $client->sysread(my $buffer, 65536) or return 1;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
194
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
195 log2i("$client $buffer");
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
196
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
197 $buffer .= " " . time() if $client->sockport() eq 8080;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
198
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
199 log2o("$client $buffer");
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
200
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
201 $client->syswrite($buffer);
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
202
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
203 return $client->sockport() eq 8080 ? 1 : $buffer =~ /close/;
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
204 }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
205
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
206 sub log2i { Test::Nginx::log_core('|| <<', @_); }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
207 sub log2o { Test::Nginx::log_core('|| >>', @_); }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
208 sub log2c { Test::Nginx::log_core('||', @_); }
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
209
c8d6f816e094 Tests: stream limit rate tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
diff changeset
210 ###############################################################################