blob: fc848c582b57bd7402400b17040c636d63dd8872 [file] [log] [blame]
#!/usr/bin/perl
# (C) Sergey Kandaurov
# (C) Nginx, Inc.
# Tests for stream proxy module with datagrams, limit rate directives.
###############################################################################
use warnings;
use strict;
use Test::More;
BEGIN { use FindBin; chdir($FindBin::Bin); }
use lib 'lib';
use Test::Nginx;
use Test::Nginx::Stream qw/ dgram /;
###############################################################################
select STDERR; $| = 1;
select STDOUT; $| = 1;
my $t = Test::Nginx->new()->has(qw/stream udp/)->plan(8)
->write_file_expand('nginx.conf', <<'EOF');
%%TEST_GLOBALS%%
daemon off;
events {
}
stream {
proxy_requests 2;
proxy_responses 1;
proxy_timeout 3s;
server {
listen 127.0.0.1:%%PORT_8982_UDP%% udp;
proxy_pass 127.0.0.1:%%PORT_8980_UDP%%;
}
server {
listen 127.0.0.1:%%PORT_8983_UDP%% udp;
proxy_pass 127.0.0.1:%%PORT_8980_UDP%%;
proxy_download_rate 500;
}
server {
listen 127.0.0.1:%%PORT_8984_UDP%% udp;
proxy_pass 127.0.0.1:%%PORT_8980_UDP%%;
proxy_upload_rate 500;
}
}
EOF
$t->run_daemon(\&udp_daemon, port(8980), $t);
$t->run();
$t->waitforfile($t->testdir . '/' . port(8980));
###############################################################################
my $str = '1234567890' x 100;
# unlimited
my $s = dgram('127.0.0.1:' . port(8982));
is($s->io($str), $str, 'unlimited');
is($s->io($str), $str, 'unlimited 2');
# datagram doesn't get split
$s = dgram('127.0.0.1:' . port(8983));
is($s->io($str), $str, 'download');
my $t1 = time();
is($s->io($str), $str, 'download 2');
my $t2 = time();
cmp_ok($t1, '<', $t2, 'download 2 delayed');
# infinite event report before 1.15.9
$s = dgram('127.0.0.1:' . port(8984));
is($s->io($str), $str, 'upload');
is($s->io($str, read_timeout => 0.5), '', 'upload limited');
select undef, undef, undef, 1.6;
is($s->io($str), $str, 'upload passed');
###############################################################################
sub udp_daemon {
my ($port, $t) = @_;
my $server = IO::Socket::INET->new(
Proto => 'udp',
LocalAddr => "127.0.0.1:$port",
)
or die "Can't create listening socket: $!\n";
# signal we are ready
open my $fh, '>', $t->testdir() . "/$port";
close $fh;
while (1) {
$server->recv(my $buffer, 65536);
log2i("$server $buffer");
log2o("$server $buffer");
$server->send($buffer);
}
}
sub log2i { Test::Nginx::log_core('|| <<', @_); }
sub log2o { Test::Nginx::log_core('|| >>', @_); }
sub log2c { Test::Nginx::log_core('||', @_); }
###############################################################################