diff options
author | Eric Wong <e@80x24.org> | 2016-12-14 06:59:52 +0000 |
---|---|---|
committer | Eric Wong <e@80x24.org> | 2016-12-14 08:02:59 +0000 |
commit | 1a1499d67d7b1b8525b2a73478cb3588e132b731 (patch) | |
tree | 3ec86e26333806216199636e07b166998bac74d4 | |
parent | d34fed5788c4ca8175ca4ef210041ae0348f56f4 (diff) | |
download | yahns-1a1499d67d7b1b8525b2a73478cb3588e132b731.tar.gz |
As with the previous commit ("response: do not set chunked header on bodyless responses"), blindly setting "Transfer-Encoding: chunked" is wrong and confuses "curl -T" on 204 responses, at least.
-rw-r--r-- | lib/yahns/proxy_http_response.rb | 2 | ||||
-rw-r--r-- | test/test_proxy_pass.rb | 33 |
2 files changed, 34 insertions, 1 deletions
diff --git a/lib/yahns/proxy_http_response.rb b/lib/yahns/proxy_http_response.rb index 74f5ce5..1776291 100644 --- a/lib/yahns/proxy_http_response.rb +++ b/lib/yahns/proxy_http_response.rb @@ -120,7 +120,7 @@ module Yahns::HttpResponse # :nodoc: # chunk the response ourselves if the client supports it, # but the backend does not terminate properly - if alive && ! term + if alive && ! term && have_body if env['HTTP_VERSION'] == 'HTTP/1.1'.freeze res << "Transfer-Encoding: chunked\r\n".freeze else # we can't persist HTTP/1.0 and HTTP/0.9 w/o Content-Length diff --git a/test/test_proxy_pass.rb b/test/test_proxy_pass.rb index 4c4b53a..22f1802 100644 --- a/test/test_proxy_pass.rb +++ b/test/test_proxy_pass.rb @@ -154,6 +154,9 @@ class TestProxyPass < Testcase env['rack.hijack'].call.close # should not be seen: [ 123, [ %w(Content-Type text/html), %w(Content-Length 0) ], [] ] + when '/204' + buf = env['rack.input'].read # drain + [ 204, {}, [] ] else buf = env['rack.input'].read [ 201, { @@ -271,6 +274,7 @@ class TestProxyPass < Testcase end end + check_204_on_put(host, port) check_forbidden_put(host, port) check_eof_body(host, port) check_pipelining(host, port) @@ -639,4 +643,33 @@ class TestProxyPass < Testcase ensure to_close.each(&:close) end + + def check_204_on_put(host, port) + s = TCPSocket.new(host, port) + s.write("PUT /204 HTTP/1.1\r\nHost: example.com\r\n" \ + "Content-Length: 11\r\n" \ + "Content-Type: application/octet-stream\r\n" \ + "\r\nhello worldPUT") + buf = s.readpartial(1024) + assert_match %r{\AHTTP/1\.1 204}, buf + assert_match %r{\r\n\r\n\z}, buf + refute_match %r{^Transfer-Encoding}i, buf + refute_match %r{^Content-Length}i, buf + assert_match %r{^Connection: keep-alive\r\n}, buf + assert_nil IO.select([s], nil, nil, 1), 'connection persists..' + + # make sure another on the same connection works + s.write(" / HTTP/1.1\r\nHost: example.com\r\n" \ + "Content-Length: 11\r\n" \ + "Content-Type: application/octet-stream\r\n" \ + "\r\nhello world") + buf = s.readpartial(1024) + assert_match %r{\r\n\r\nhello world\z}, buf + assert_match %r{\AHTTP/1\.1 201}, buf + assert_match(%r{^Content-Length: 11\r\n}, buf) + assert_match %r{^Connection: keep-alive\r\n}, buf + assert_nil IO.select([s], nil, nil, 1), 'connection persists..' + ensure + s.close if s + end end |