diff options
-rw-r--r-- | ChangeLog | 8 | ||||
-rw-r--r-- | lib/net/http.rb | 13 | ||||
-rw-r--r-- | test/net/http/test_httpresponse.rb | 40 |
3 files changed, 58 insertions, 3 deletions
@@ -1,3 +1,11 @@ +Fri Oct 30 22:09:47 2009 NAKAMURA, Hiroshi <nahi@ruby-lang.org> + + * lib/net/http.rb (Net::HTTPResponse#each_response_header): + accept multiline message header of HTTP response. see #1796. + cf. RFC 2616 '4.2 Message Header'. + + * test/net/http/test_httpresponse.rb: added. + Fri Oct 30 18:54:04 2009 Yukihiro Matsumoto <matz@ruby-lang.org> * string.c (trnext): detect empty range and raise exception. diff --git a/lib/net/http.rb b/lib/net/http.rb index 64dbef4cbb..57a2b2eea5 100644 --- a/lib/net/http.rb +++ b/lib/net/http.rb @@ -2158,13 +2158,20 @@ module Net #:nodoc: end def each_response_header(sock) + key = value = nil while true line = sock.readuntil("\n", true).sub(/\s+\z/, '') break if line.empty? - m = /\A([^:]+):\s*/.match(line) or - raise HTTPBadResponse, 'wrong header line format' - yield m[1], m.post_match + if line[0] == ?\ or line[0] == ?\t and value + value << ' ' unless value.empty? + value << line.strip + else + yield key, value if key + key, value = line.strip.split(/\s*:\s*/, 2) + raise HTTPBadResponse, 'wrong header line format' if value.nil? + end end + yield key, value if key end end diff --git a/test/net/http/test_httpresponse.rb b/test/net/http/test_httpresponse.rb new file mode 100644 index 0000000000..ab6fdd0ea9 --- /dev/null +++ b/test/net/http/test_httpresponse.rb @@ -0,0 +1,40 @@ +require 'net/http' +require 'test/unit' +require 'stringio' + +class HTTPResponseTest < Test::Unit::TestCase + def test_singleline_header + io = dummy_io(<<EOS.gsub(/\n/, "\r\n")) +HTTP/1.1 200 OK +Content-Length: 5 +Connection: close + +hello +EOS + res = Net::HTTPResponse.read_new(io) + assert_equal('5', res.header['content-length']) + assert_equal('close', res.header['connection']) + end + + def test_multiline_header + io = dummy_io(<<EOS.gsub(/\n/, "\r\n")) +HTTP/1.1 200 OK +X-Foo: XXX + YYY +X-Bar: + XXX +\tYYY + +hello +EOS + res = Net::HTTPResponse.read_new(io) + assert_equal('XXX YYY', res.header['x-foo']) + assert_equal('XXX YYY', res.header['x-bar']) + end + +private + + def dummy_io(str) + Net::BufferedIO.new(StringIO.new(str)) + end +end |