From 1468604be898d17c2cf2da519dccd493c58f4282 Mon Sep 17 00:00:00 2001
From: Eric Wong <normalperson@yhbt.net>
Date: Fri, 10 Jun 2011 22:08:04 +0000
Subject: ev_core: do not autochunk HTTP/1.0 (and 0.9) responses

Do not assume middlewares/applications are stupid and blindly
add chunking to responses (we have precedence set by
Rack::Chunked).
---
 lib/rainbows/dev_fd_response.rb |  9 ++++++---
 lib/rainbows/ev_core.rb         | 17 ++++++++++++++---
 2 files changed, 20 insertions(+), 6 deletions(-)

(limited to 'lib')

diff --git a/lib/rainbows/dev_fd_response.rb b/lib/rainbows/dev_fd_response.rb
index eaa4af4..edc39af 100644
--- a/lib/rainbows/dev_fd_response.rb
+++ b/lib/rainbows/dev_fd_response.rb
@@ -19,7 +19,6 @@ class Rainbows::DevFdResponse < Struct.new(:app)
   Transfer_Encoding = "Transfer-Encoding".freeze
   Rainbows_autochunk = "rainbows.autochunk".freeze
   Rainbows_model = "rainbows.model"
-  HTTP_1_0 = "HTTP/1.0"
   HTTP_VERSION = "HTTP_VERSION"
   Chunked = "chunked"
 
@@ -55,8 +54,12 @@ class Rainbows::DevFdResponse < Struct.new(:app)
       headers.delete(Transfer_Encoding)
     elsif st.pipe? || st.socket? # epoll-able things
       unless headers.include?(Content_Length)
-        if env[Rainbows_autochunk] && HTTP_1_0 != env[HTTP_VERSION]
-          headers[Transfer_Encoding] = Chunked
+        if env[Rainbows_autochunk]
+          case env[HTTP_VERSION]
+          when "HTTP/1.0", nil
+          else
+            headers[Transfer_Encoding] = Chunked
+          end
         else
           env[Rainbows_autochunk] = false
         end
diff --git a/lib/rainbows/ev_core.rb b/lib/rainbows/ev_core.rb
index 29e0e81..46feaff 100644
--- a/lib/rainbows/ev_core.rb
+++ b/lib/rainbows/ev_core.rb
@@ -10,6 +10,7 @@ module Rainbows::EvCore
   RBUF = ""
   Z = "".freeze
   Rainbows.config!(self, :client_header_buffer_size)
+  HTTP_VERSION = "HTTP_VERSION"
 
   # Apps may return this Rack response: AsyncResponse = [ -1, {}, [] ]
   ASYNC_CALLBACK = "async.callback".freeze
@@ -54,13 +55,23 @@ module Rainbows::EvCore
   def stream_response_headers(status, headers, alive)
     headers = Rack::Utils::HeaderHash.new(headers) unless Hash === headers
     if headers.include?(Content_Length)
-      rv = false
+      write_headers(status, headers, alive)
+      return false
+    end
+
+    case @env[HTTP_VERSION]
+    when "HTTP/1.0" # disable HTTP/1.0 keepalive to stream
+      write_headers(status, headers, false)
+      @hp.clear
+      false
+    when nil # "HTTP/0.9"
+      false
     else
       rv = !!(headers[Transfer_Encoding] =~ %r{\Achunked\z}i)
       rv = false unless @env["rainbows.autochunk"]
+      write_headers(status, headers, alive)
+      rv
     end
-    write_headers(status, headers, alive)
-    rv
   end
 
   def prepare_request_body
-- 
cgit v1.2.3-24-ge0c7