diff options
author | Eric Wong <normalperson@yhbt.net> | 2010-07-19 10:09:56 +0000 |
---|---|---|
committer | Eric Wong <normalperson@yhbt.net> | 2010-07-19 17:04:26 -0700 |
commit | 1e6d3d19da2b62bfe7f8fd7827dcad3ee3fe9923 (patch) | |
tree | 5de875617e5cf3befb7c0ae9e1add9b2ee2f451e /lib/rainbows/ev_core.rb | |
parent | cc18035c5105751a3e67a8e449ee0021fd313ea9 (diff) | |
download | rainbows-1e6d3d19da2b62bfe7f8fd7827dcad3ee3fe9923.tar.gz |
If a response proxying a pipe (or socket) includes a Content-Length, do not attempt to outsmart the application and just use the given Content-Length. This helps avoid exposing applications to weird internals such as env["rainbows.autochunk"] and X-Rainbows-* response headers.
Diffstat (limited to 'lib/rainbows/ev_core.rb')
-rw-r--r-- | lib/rainbows/ev_core.rb | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/lib/rainbows/ev_core.rb b/lib/rainbows/ev_core.rb index 5ca693b..dbcdeba 100644 --- a/lib/rainbows/ev_core.rb +++ b/lib/rainbows/ev_core.rb @@ -6,6 +6,7 @@ module Rainbows module EvCore include Unicorn include Rainbows::Const + include Rainbows::Response G = Rainbows::G NULL_IO = Unicorn::HttpRequest::NULL_IO @@ -33,6 +34,19 @@ module Rainbows quit end + # returns whether to enable response chunking for autochunk models + def stream_response_headers(status, headers) + if headers['Content-Length'] + rv = false + else + rv = !!(headers['Transfer-Encoding'] =~ %r{\Achunked\z}i) + rv = false if headers.delete('X-Rainbows-Autochunk') == 'no' + end + headers[CONNECTION] = CLOSE # TODO: allow keep-alive + write(response_header(status, headers)) + rv + end + # TeeInput doesn't map too well to this right now... def on_read(data) case @state |