blob: 0fa371589a5b71ad63baa4f57fd6f112f2821881 [file] [log] [blame]
#!/usr/bin/perl
# (C) Andrey Zelenkov
# (C) Dmitry Volyntsev
# (C) Nginx, Inc.
# Tests for stream njs module.
###############################################################################
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 stream /;
###############################################################################
select STDERR; $| = 1;
select STDOUT; $| = 1;
my $t = Test::Nginx->new()->has(qw/http proxy rewrite stream stream_return udp/)
->write_file_expand('nginx.conf', <<'EOF');
%%TEST_GLOBALS%%
daemon off;
events {
}
http {
%%TEST_GLOBALS_HTTP%%
js_include test.js;
server {
listen 127.0.0.1:8079;
server_name localhost;
location /njs {
js_content test_njs;
}
location /p/ {
proxy_pass http://127.0.0.1:8095/;
}
location /return {
return 200 $http_foo;
}
}
}
stream {
js_set $js_addr js_addr;
js_set $js_var js_var;
js_set $js_log js_log;
js_set $js_unk js_unk;
js_set $js_req_line js_req_line;
js_set $js_sess_unk js_sess_unk;
js_include test.js;
server {
listen 127.0.0.1:8080;
return $js_addr;
}
server {
listen 127.0.0.1:8081;
return $js_log;
}
server {
listen 127.0.0.1:8082;
return $js_var;
}
server {
listen 127.0.0.1:8083;
return $js_unk;
}
server {
listen 127.0.0.1:8084;
return $js_sess_unk;
}
server {
listen 127.0.0.1:%%PORT_8985_UDP%% udp;
return $js_addr;
}
server {
listen 127.0.0.1:8086;
js_access js_access_step;
js_preread js_preread_step;
js_filter js_filter_step;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8087;
js_access js_access_undecided;
return OK;
}
server {
listen 127.0.0.1:8088;
js_access js_access_allow;
return OK;
}
server {
listen 127.0.0.1:8089;
js_access js_access_deny;
return OK;
}
server {
listen 127.0.0.1:8091;
js_preread js_preread_async;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8092;
js_preread js_preread_data;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8093;
js_preread js_preread_req_line;
return $js_req_line;
}
server {
listen 127.0.0.1:8094;
js_filter js_filter_empty;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8095;
js_filter js_filter_header_inject;
proxy_pass 127.0.0.1:8079;
}
server {
listen 127.0.0.1:8096;
js_filter js_filter_search;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8097;
js_access js_access_except;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8098;
js_preread js_preread_except;
proxy_pass 127.0.0.1:8090;
}
server {
listen 127.0.0.1:8099;
js_filter js_filter_except;
proxy_pass 127.0.0.1:8090;
}
}
EOF
$t->write_file('test.js', <<EOF);
function test_njs(r) {
r.return(200, njs.version);
}
function js_addr(s) {
return 'addr=' + s.remoteAddress;
}
function js_var(s) {
return 'variable=' + s.variables.remote_addr;
}
function js_sess_unk(s) {
return 'sess_unk=' + s.unk;
}
function js_log(s) {
s.log("SEE-THIS");
}
var res = '';
function js_access_step(s) {
res += '1';
setTimeout(function() {
if (s.remoteAddress.match('127.0.0.1')) {
s.allow();
}
}, 1);
}
function js_preread_step(s) {
s.on('upload', function (data) {
res += '2';
if (res.length >= 3) {
s.done();
}
});
}
function js_filter_step(s) {
s.on('upload', function(data, flags) {
s.send(data);
res += '3';
});
s.on('download', function(data, flags) {
if (!flags.last) {
res += '4';
s.send(data);
} else {
res += '5';
s.send(res, {last:1});
s.off('download');
}
});
}
function js_access_undecided(s) {
s.decline();
}
function js_access_allow(s) {
if (s.remoteAddress.match('127.0.0.1')) {
s.done();
return;
}
s.abort();
}
function js_access_deny(s) {
if (s.remoteAddress.match('127.0.0.1')) {
s.abort();
return;
}
s.allow();
}
function js_preread_async(s) {
setTimeout(function() {
s.done();
}, 1);
}
function js_preread_data(s) {
s.on('upload', function (data, flags) {
if (data.indexOf('z') != -1) {
s.done();
}
});
}
var line = '';
function js_preread_req_line(s) {
s.on('upload', function (data, flags) {
var n = data.indexOf('\\n');
if (n != -1) {
line = data.substr(0, n);
s.done();
}
});
}
function js_req_line(s) {
return line;
}
function js_filter_empty(s) {
}
function js_filter_header_inject(s) {
var req = '';
s.on('upload', function(data, flags) {
req += data;
var n = req.search('\\n');
if (n != -1) {
var rest = req.substr(n + 1);
req = req.substr(0, n + 1);
s.send(req + 'Foo: foo' + '\\r\\n' + rest, flags);
s.off('upload');
}
});
}
function js_filter_search(s) {
s.on('download', function(data, flags) {
var n = data.search('y');
if (n != -1) {
s.send('z');
}
});
s.on('upload', function(data, flags) {
var n = data.search('x');
if (n != -1) {
s.send('y');
}
});
}
function js_access_except(s) {
function done() {return s.a.a};
setTimeout(done, 1);
setTimeout(done, 2);
}
function js_preread_except(s) {
var fs = require('fs');
fs.readFileSync();
}
function js_filter_except(s) {
s.on('unknown', function() {});
}
EOF
$t->run_daemon(\&stream_daemon, port(8090));
$t->try_run('no stream njs available')->plan(19);
$t->waitforsocket('127.0.0.1:' . port(8090));
###############################################################################
is(stream('127.0.0.1:' . port(8080))->read(), 'addr=127.0.0.1',
's.remoteAddress');
is(dgram('127.0.0.1:' . port(8985))->io('.'), 'addr=127.0.0.1',
's.remoteAddress udp');
is(stream('127.0.0.1:' . port(8081))->read(), 'undefined', 's.log');
is(stream('127.0.0.1:' . port(8082))->read(), 'variable=127.0.0.1',
's.variables');
is(stream('127.0.0.1:' . port(8083))->read(), '', 'stream js unknown function');
is(stream('127.0.0.1:' . port(8084))->read(), 'sess_unk=undefined', 's.unk');
is(stream('127.0.0.1:' . port(8086))->io('0'), '0122345',
'async handlers order');
is(stream('127.0.0.1:' . port(8087))->io('#'), 'OK', 'js_access_undecided');
is(stream('127.0.0.1:' . port(8088))->io('#'), 'OK', 'js_access_allow');
is(stream('127.0.0.1:' . port(8089))->io('#'), '', 'js_access_deny');
is(stream('127.0.0.1:' . port(8091))->io('#'), '#', 'js_preread_async');
is(stream('127.0.0.1:' . port(8092))->io('#z'), '#z', 'js_preread_async_data');
is(stream('127.0.0.1:' . port(8093))->io("xy\na"), 'xy', 'js_preread_req_line');
is(stream('127.0.0.1:' . port(8094))->io('x'), 'x', 'js_filter_empty');
like(get('/p/return'), qr/foo/, 'js_filter_injected_header');
is(stream('127.0.0.1:' . port(8096))->io('x'), 'z', 'js_filter_search');
stream('127.0.0.1:' . port(8097))->io('x');
stream('127.0.0.1:' . port(8098))->io('x');
stream('127.0.0.1:' . port(8099))->io('x');
$t->stop();
ok(index($t->read_file('error.log'), 'SEE-THIS') > 0, 'stream js log');
ok(index($t->read_file('error.log'), 'at fs.readFileSync') > 0,
'stream js_preread backtrace');
ok(index($t->read_file('error.log'), 'at js_filter_except') > 0,
'stream js_filter backtrace');
###############################################################################
sub stream_daemon {
my $server = IO::Socket::INET->new(
Proto => 'tcp',
LocalAddr => '127.0.0.1:' . port(8090),
Listen => 5,
Reuse => 1
)
or die "Can't create listening socket: $!\n";
local $SIG{PIPE} = 'IGNORE';
while (my $client = $server->accept()) {
$client->autoflush(1);
log2c("(new connection $client)");
$client->sysread(my $buffer, 65536) or next;
log2i("$client $buffer");
log2o("$client $buffer");
$client->syswrite($buffer);
close $client;
}
}
sub log2i { Test::Nginx::log_core('|| <<', @_); }
sub log2o { Test::Nginx::log_core('|| >>', @_); }
sub log2c { Test::Nginx::log_core('||', @_); }
sub get {
my ($url, %extra) = @_;
my $s = IO::Socket::INET->new(
Proto => 'tcp',
PeerAddr => '127.0.0.1:' . port(8079)
) or die "Can't connect to nginx: $!\n";
return http_get($url, socket => $s);
}
###############################################################################