diff --git a/ChangeLog b/ChangeLog index 6fc912903a..66101c1d1b 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,9 @@ +Tue Apr 10 18:19:32 2012 NARUSE, Yui <naruse@ruby-lang.org> + + * lib/net/http.rb (Net::HTTP#send_request_with_body_stream): + use IO.copy_stream for requests using body_stream. + patched by Eric Wong. [ruby-core:40898] [Feature #5605] + Tue Apr 10 16:53:21 2012 Nobuyoshi Nakada <nobu@ruby-lang.org> * thread_pthread.c: add prototype declarations for older Mac OS X. diff --git a/lib/net/http.rb b/lib/net/http.rb index 65816f52e8..f6da628878 100644 --- a/lib/net/http.rb +++ b/lib/net/http.rb @@ -1987,6 +1987,25 @@ module Net #:nodoc: private + class Chunker #:nodoc: + def initialize(sock) + @sock = sock + @prev = nil + end + + def write(buf) + # avoid memcpy() of buf, buf can huge and eat memory bandwidth + @sock.write("#{buf.bytesize.to_s(16)}\r\n") + rv = @sock.write(buf) + @sock.write("\r\n") + rv + end + + def finish + @sock.write("0\r\n\r\n") + end + end + def send_request_with_body(sock, ver, path, body) self.content_length = body.bytesize delete 'Transfer-Encoding' @@ -2005,14 +2024,13 @@ module Net #:nodoc: write_header sock, ver, path wait_for_continue sock, ver if sock.continue_timeout if chunked? - while s = f.read(1024) - sock.write(sprintf("%x\r\n", s.length) << s << "\r\n") - end - sock.write "0\r\n\r\n" + chunker = Chunker.new(sock) + IO.copy_stream(f, chunker) + chunker.finish else - while s = f.read(1024) - sock.write s - end + # copy_stream can sendfile() to sock.io unless we use SSL. + # If sock.io is an SSLSocket, copy_stream will hit SSL_write() + IO.copy_stream(f, sock.io) end end