From 58daab21cd043e1dc37024a7f99b396788372918 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sat, 9 Mar 2024 14:19:48 +0100 Subject: Merging upstream version 1.44.3. Signed-off-by: Daniel Baumann --- web/server/h2o/libh2o/t/50mruby.t | 541 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 541 insertions(+) create mode 100644 web/server/h2o/libh2o/t/50mruby.t (limited to 'web/server/h2o/libh2o/t/50mruby.t') diff --git a/web/server/h2o/libh2o/t/50mruby.t b/web/server/h2o/libh2o/t/50mruby.t new file mode 100644 index 000000000..d93f17e04 --- /dev/null +++ b/web/server/h2o/libh2o/t/50mruby.t @@ -0,0 +1,541 @@ +use strict; +use warnings; +use Digest::MD5 qw(md5_hex); +use File::Temp qw(tempdir); +use Test::More; +use t::Util; + +plan skip_all => 'mruby support is off' + unless server_features()->{mruby}; + +plan skip_all => 'curl not found' + unless prog_exists('curl'); + +subtest "handler-file" => sub { + my $server = spawn_h2o(<< 'EOT'); +hosts: + default: + paths: + /: + mruby.handler-file: t/50mruby/hello.rb +EOT + my ($headers, $body) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + is $body, "hello from h2o_mruby\n"; + like $headers, qr{^HTTP/1\.1 200 OK\r\n}s; + like $headers, qr{^content-type: text/plain; charset=utf-8\r$}im; +}; + +subtest "basic" => sub { + my $server = spawn_h2o(<< 'EOT'); +hosts: + default: + paths: + /inline: + mruby.handler: | + Proc.new do |env| + [200, {}, ["hello from h2o_mruby\n"]] + end + /return-404: + mruby.handler: | + Proc.new do |env| + [404, {}, ["not found"]] + end + file.dir: examples/doc_root + /fallthru: + mruby.handler: | + Proc.new do |env| + [399, {}, []] + end + file.dir: t/50mruby/ + /echo: + mruby.handler: | + Proc.new do |env| + [200, {}, [JSON.generate(env)]] + end + /headers: + mruby.handler: | + Proc.new do |env| + [200, {"foo" => "123\n456", "bar" => "baz"}, []] + end + /headers-each: + mruby.handler: | + Proc.new do |env| + [200, [["content-type", "text/plain"], ["hello", "world"]], []] + end +EOT + my $fetch = sub { + my $path = shift; + run_prog("curl --silent -A h2o_mruby_test --dump-header /dev/stderr http://127.0.0.1:$server->{port}$path"); + }; + my ($headers, $body) = $fetch->("/inline/"); + is $body, "hello from h2o_mruby\n", "inline"; + subtest "return-404" => sub { + ($headers, $body) = $fetch->("/return-404/"); + like $headers, qr{^HTTP/1\.1 404 }is; + is $body, "not found"; + }; + subtest "fallthru" => sub { + ($headers, $body) = $fetch->("/fallthru/"); + like $headers, qr{^HTTP/1\.1 200 OK\r\n}is; + is md5_hex($body), md5_file("t/50mruby/index.html"); + }; + subtest "echo" => sub { + ($headers, $body) = $fetch->("/echo/abc?def"); + like $body, qr{"REQUEST_METHOD":"GET"}, "REQUEST_METHOD"; + like $body, qr{"SCRIPT_NAME":"/echo"}, "SCRIPT_NAME"; + like $body, qr{"PATH_INFO":"/abc"}, "PATH_INFO"; + like $body, qr{"QUERY_STRING":"def"}, "QUERY_STRING"; + like $body, qr{"SERVER_NAME":"default"}, "SERVER_NAME"; + like $body, qr{"SERVER_ADDR":"127.0.0.1"}, "SERVER_ADDR"; + like $body, qr{"SERVER_PORT":"$server->{port}"}, "SERVER_PORT"; + like $body, qr{"HTTP_HOST":"127.0.0.1:$server->{port}"}, "HTTP_HOST"; + like $body, qr{"SERVER_ADDR":"127.0.0.1"}, "REMOTE_ADDR"; + like $body, qr{"SERVER_PORT":"[0-9]+"}, "REMOTE_PORT"; + like $body, qr{"HTTP_USER_AGENT":"h2o_mruby_test"}, "HTTP_USER_AGENT"; + like $body, qr{"rack.url_scheme":"http"}, "url_scheme"; + like $body, qr{"SERVER_SOFTWARE":"h2o/[0-9]+\.[0-9]+\.[0-9]+}, "SERVER_SOFTWARE"; + }; + subtest "protocol" => sub { + run_with_curl($server, sub { + my ($proto, $port, $curl) = @_; + my $content = `$curl --silent --show-error $proto://127.0.0.1:$port/echo`; + if ($curl =~ /http2/) { + like $content, qr{"SERVER_PROTOCOL":"HTTP/2"}, "SERVER_PROTOCOL"; + } else { + like $content, qr{"SERVER_PROTOCOL":"HTTP/1\.1"}, "SERVER_PROTOCOL"; + } + }); + }; + subtest "headers" => sub { + ($headers, $body) = $fetch->("/headers/"); + like $headers, qr{^foo: 123\r$}mi; + like $headers, qr{^foo: 456\r$}mi; + like $headers, qr{^bar: baz\r$}mi; + }; + subtest "headers-each" => sub { + ($headers, $body) = $fetch->("/headers-each/"); + like $headers, qr{^content-type: text/plain\r$}mi; + like $headers, qr{^hello: world\r$}mi; + }; +}; + +subtest "reprocess_request" => sub { + my $server = spawn_h2o(<< "EOT"); +hosts: + default: + reproxy: ON + paths: + /: + mruby.handler: | + Proc.new do |env| + [200, {"x-reproxy-url" => "http://default/dest#{env["PATH_INFO"]}"}, ["should never see this"]] + end + /307: + mruby.handler: | + Proc.new do |env| + [307, {"x-reproxy-url" => "http://default/dest#{env["PATH_INFO"]}"}, ["should never see this"]] + end + /dest: + mruby.handler: | + Proc.new do |env| + [200, {}, ["#{env["SCRIPT_NAME"]}#{env["PATH_INFO"]};#{env["CONTENT_LENGTH"]}"]] + end +EOT + my ($stderr, $stdout) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + is $stdout, "/dest/;"; + ($stderr, $stdout) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/hoge"); + is $stdout, "/dest/hoge;"; + subtest "preserve-method" => sub { + ($stderr, $stdout) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/307/"); + is $stdout, "/dest/;"; + ($stderr, $stdout) = run_prog("curl --data hello --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/307/"); + is $stdout, "/dest/;5"; + ($stderr, $stdout) = run_prog("curl --data hello --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + is $stdout, "/dest/;"; + }; +}; + +subtest "server-push" => sub { + plan skip_all => 'nghttp not found' + unless prog_exists('nghttp'); + my $server = spawn_h2o(<< "EOT"); +hosts: + default: + paths: + /: + mruby.handler: | + Proc.new do |env| + push_paths = [] + if env["PATH_INFO"] == "/index.txt" + push_paths << "/index.js" + end + [399, push_paths.empty? ? {} : {"link" => push_paths.map{|p| "<#{p}>; rel=preload"}.join()}, []] + end + file.dir: t/assets/doc_root +EOT + my $resp = `nghttp -n --stat https://127.0.0.1:$server->{tls_port}/index.txt`; + like $resp, qr{\nid\s*responseEnd\s.*\s/index\.js\n.*\s/index\.txt}is, "receives index.js then /index.txt"; +}; + +subtest "server-push / nopush" => sub { + plan skip_all => 'nghttp not found' + unless prog_exists('nghttp'); + my $server = spawn_h2o(<< "EOT"); +hosts: + default: + paths: + /: + mruby.handler: | + Proc.new do |env| + push_paths = [] + if env["PATH_INFO"] == "/index.txt" + push_paths << "/index.js" + end + [399, push_paths.empty? ? {} : {"link" => push_paths.map{|p| "<#{p}>; rel=preload; nopush"}.join()}, []] + end + file.dir: t/assets/doc_root +EOT + my $resp = `nghttp -n --stat https://127.0.0.1:$server->{tls_port}/index.txt`; + unlike $resp, qr{/index\.js}is, "receives only /index.txt"; + like $resp, qr{/index\.txt}is, "receives only /index.txt"; +}; + +subtest "infinite-reprocess" => sub { + my $server = spawn_h2o(sub { + my ($port, $tls_port) = @_; + return << "EOT"; +hosts: + "127.0.0.1:$port": + paths: + /: + reproxy: ON + mruby.handler: | + Proc.new do |env| + [200,{"x-reproxy-url" => "http://127.0.0.1:$port/"},[]] + end +EOT + }); + my ($stderr, $stdout) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $stderr, qr{^HTTP\/1.1 502 }s, "502 response"; + like $stdout, qr{too many internal delegations}, "reason"; +}; + +subtest "send-file" => sub { + my $server = spawn_h2o(<< "EOT"); +hosts: + default: + paths: + /: + mruby.handler: | + Proc.new do |env| + [200,{}, File::open("t/50mruby/index.html")] + end +EOT + my ($headers, $body) = run_prog("curl --silent -A h2o_mruby_test --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 OK\r\n}is; + is md5_hex($body), md5_file("t/50mruby/index.html"); +}; + +subtest "exception" => sub { + my $server = spawn_h2o(<< 'EOT'); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + cnt = 0 + Proc.new do |env| + cnt += 1 + if cnt % 2 != 0 + [200, {}, ["hello\n"]] + else + raise "error from rack" + end + end +EOT + my $fetch = sub { + run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}"); + }; + for (1..3) { + my ($headers, $body) = $fetch->(); + like $headers, qr{^HTTP/1\.1 200 }is; + is $body, "hello\n"; + ($headers, $body) = $fetch->(); + like $headers, qr{^HTTP/1\.1 500 }is; + } +}; + +subtest "post" => sub { + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + Proc.new do |env| + body = [] + 3.times do + env["rack.input"].rewind + body << env["rack.input"].read + body << "\\n" + end + [200, {}, body] + end +EOT + my ($headers, $body) = run_prog("curl --silent --data 'hello' --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 OK\r\n}is; + is $body, "hello\n" x 3; +}; + +subtest "InputStream#read-after-close" => sub { + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + prev_input = nil + Proc.new do |env| + if !prev_input + prev_input = env["rack.input"] + resp = "not cached" + else + begin + prev_input.read + resp = "must not seed this" + rescue IOError => e + resp = "got IOError" + end + end + [200, {}, [resp]] + end +EOT + my ($headers, $body) = run_prog("curl --silent --data 'hello' --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 OK\r\n}is; + is $body, "not cached"; + ($headers, $body) = run_prog("curl --silent --data 'hello' --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 OK\r\n}is; + is $body, "got IOError"; +}; + +subtest "header-concat" => sub { + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + Proc.new do |env| + [200, {}, [env["HTTP_COOKIE"]]] + end +EOT + run_with_curl($server, sub { + my ($proto, $port, $curl) = @_; + my ($headers, $body) = run_prog("$curl --silent -H 'cookie: a=b' -H 'cookie: c=d' --dump-header /dev/stderr $proto://127.0.0.1:$port/"); + like $headers, qr{^HTTP/\S+ 200}is; + like $body, qr{^a=b;\s*c=d$}is; + }); +}; + +subtest "close-called" => sub { + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + is_open = false + lambda do |env| + if is_open + return [500, {}, ["close not called"]] + end + is_open = true + return [ + 200, + {}, + Class.new do + def each + yield "hello" + end + define_method(:close) do + is_open = false + end + end.new, + ] + end +EOT + my ($headers, $body) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 }is; + is $body, "hello"; + ($headers, $body) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 }is; + is $body, "hello"; +}; + +subtest "close-called-on-exception" => sub { + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + is_open = false + lambda do |env| + if is_open + return [500, {}, ["close not called"]] + end + is_open = true + return [ + 200, + {}, + Class.new do + def each + yield "hello" + raise "yeah!" + end + define_method(:close) do + is_open = false + end + end.new, + ] + end +EOT + my ($headers, $body) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 }is; + is $body, "hello"; + ($headers, $body) = run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 200 }is; + is $body, "hello"; +}; + +subtest "log lineno" => sub { + my $tester = sub { + my ($name, $conf, $expected) = @_; + + subtest $name => sub { + my $tempdir = tempdir(CLEANUP => 1); + unlink "$tempdir/error_log"; + my $server = spawn_h2o(<< "EOT"); +$conf +error-log: $tempdir/error_log +EOT + run_prog("curl --silent --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + my @log = do { + open my $fh, "<", "$tempdir/error_log" + or die "failed to open error_log:$!"; + map { my $l = $_; chomp $l; $l } <$fh>; + }; + @log = grep { $_ =~ /^\[h2o_mruby\]/ } @log; + like $log[$#log], qr{\[h2o_mruby\] in request:/:mruby raised: @{[$server->{conf_file}]}:$expected:\s*hoge \(RuntimeError\)}; + }; + }; + $tester->("flow style", <<"EOT", 5); +hosts: + default: + paths: + /: + mruby.handler: Proc.new do |env| raise "hoge" end +EOT + $tester->("block style", <<"EOT", 7); +hosts: + default: + paths: + /: + mruby.handler: | + Proc.new do |env| + raise "hoge" + end +EOT +}; + +subtest 'response with specific statuses should not contain content-length header' => sub { + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + proc {|env| + [204, {}, []] + } +EOT + my ($headers, $body) = run_prog("curl --silent --data 'hello' --dump-header /dev/stderr http://127.0.0.1:$server->{port}/"); + like $headers, qr{^HTTP/1\.1 204 OK\r\n}is; + unlike $headers, qr{^content-length:}im; +}; + +subtest 'PATH_INFO and SCRIPT_NAME' => sub { + plan skip_all => "nc not found" + unless prog_exists("nc"); + + my $server = spawn_h2o(<< "EOT"); +num-threads: 1 +hosts: + default: + paths: + /: + mruby.handler: | + proc {|env| + [200, {}, ['handler1, ' + env['SCRIPT_NAME'] + ', ' + env['PATH_INFO']]] + } + /abc: + mruby.handler: | + proc {|env| + [200, {}, ['handler2, ' + env['SCRIPT_NAME'] + ', ' + env['PATH_INFO']]] + } + "/foo bar": + mruby.handler: | + proc {|env| + [200, {}, ['handler3, ' + env['SCRIPT_NAME'] + ', ' + env['PATH_INFO']]] + } +EOT + my $nc = sub { + my $path = shift; + my $cmd = "echo 'GET $path HTTP/1.1\\r\\nHost: 127.0.0.1\\r\\n\\r' | nc 127.0.0.1 $server->{port}"; + (undef, my $r) = run_prog($cmd); + split(/\r\n\r\n/, $r, 2); + }; + + my $body; + (undef, $body) = $nc->('/abc/def%20ghi'); + is $body, 'handler2, /abc, /def%20ghi', 'should be kept undecoded'; + + (undef, $body) = $nc->('/abc/def/../ghi/../jhk'); + is $body, 'handler2, /abc, /def/../ghi/../jhk', 'https://github.com/h2o/h2o/pull/1480#issuecomment-339614160'; + + (undef, $body) = $nc->('/123/../abc/def/../ghi'); + is $body, 'handler2, /abc, /def/../ghi', 'https://github.com/h2o/h2o/pull/1480#issuecomment-339658134'; + + (undef, $body) = $nc->('/foo%20bar/baz'); + is $body, 'handler3, /foo bar, /baz', 'paths should be decoded'; + + (undef, $body) = $nc->('/xxx/../hoge'); + is $body, 'handler1, , /xxx/../hoge', 'string size is too big issue 1'; + + (undef, $body) = $nc->('/../abc'); + is $body, 'handler2, /abc, ', 'string size is too big issue 2'; + + (undef, $body) = $nc->('abc'); + is $body, 'handler2, /abc, ', 'no leading slash 1'; + + (undef, $body) = $nc->('abc/def'); + is $body, 'handler2, /abc, /def', 'no leading slash 2'; + + (undef, $body) = $nc->('123/../abc/def/../ghi'); + is $body, 'handler2, /abc, /def/../ghi', 'no leading slash 3'; + + (undef, $body) = $nc->('xyz'); + is $body, 'handler1, , xyz', 'no leading slash 4'; + + (undef, $body) = $nc->(''); + is $body, 'handler1, , ', 'empty path'; +}; + +done_testing(); -- cgit v1.2.3