rackup 2.0.0 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c58b0cded0e6edaba53fc8bd56f6b86b362feb6c96933ad81ee31f618015adcb
-  data.tar.gz: e5bf647ba5b1108790c63ed32dd9684f948fa7d6347359bf70ad39497ac5e144
+  metadata.gz: a0b31e72267a0dc7a7cc3bbe41d06b49ea16ec3612edc7474fe5c99c97e568cf
+  data.tar.gz: 6b9b9f197c6929bed47b7bb3c81f13f956427e1d078ab2a4b12afc0c4400f881
 SHA512:
-  metadata.gz: b9b36c45895e8e373d4308b4893e4cb93d2db3cf0e9fa24043f773eac0c4e735eecceba460591caa78bf99ff7271b98e14e161aa590658a3801e7007f6cbb807
-  data.tar.gz: 32b1b2c2768b0f5cf1b45b1f5b8e75742318c6a172c4af7d8e64990739d0754823534091520b677026630fd96a7395ee275f414fdfef52c8b5ab6413725f1fb7
+  metadata.gz: 21aba248befe16cd582ac7c867bc28cecd41155e2f8979e8cecba248e2f1623894c3f943484a85a189d3e2bdec45c836ad8f9f2995a2362809332b1b99dcd3d5
+  data.tar.gz: 258c30b02fc3d454df9dcd1d6285f0f3b6d592728f856c53e761c37db2a56925bc0a90bf0b3330441c754fddec280501abc13aa2032a120f3ce5917bd38c817f

data/lib/rackup/handler/webrick.rb

@@ -11,28 +11,11 @@ require 'rack/constants'
 require_relative '../handler'
 require_relative '../version'
 
-# This monkey patch allows for applications to perform their own chunking
-# through WEBrick::HTTPResponse if rack is set to true.
-class WEBrick::HTTPResponse
-  attr_accessor :rack
-
-  alias _rack_setup_header setup_header
-  def setup_header
-    app_chunking = rack && @header['transfer-encoding'] == 'chunked'
-
-    @chunked = app_chunking if app_chunking
-
-    _rack_setup_header
-
-    @chunked = false if app_chunking
-  end
-end
+require_relative '../stream'
 
 module Rackup
   module Handler
     class WEBrick < ::WEBrick::HTTPServlet::AbstractServlet
-      include Rack
-      
       def self.run(app, **options)
         environment  = ENV['RACK_ENV'] || 'development'
         default_host = environment == 'development' ? 'localhost' : nil
@@ -73,43 +56,75 @@ module Rackup
         @app = app
       end
 
+      # This handles mapping the WEBrick request to a Rack input stream.
+      class Input
+        include Stream::Reader
+
+        def initialize(request)
+          @request = request
+
+          @reader = Fiber.new do
+            @request.body do |chunk|
+              Fiber.yield(chunk)
+            end
+
+            Fiber.yield(nil)
+
+            # End of stream:
+            @reader = nil
+          end
+        end
+
+        def close
+          @request = nil
+          @reader = nil
+        end
+
+        private
+
+        # Read one chunk from the request body.
+        def read_next
+          @reader&.resume
+        end
+      end
+
       def service(req, res)
-        res.rack = true
         env = req.meta_vars
         env.delete_if { |k, v| v.nil? }
 
-        rack_input = StringIO.new(req.body.to_s)
-        rack_input.set_encoding(Encoding::BINARY)
+        input = Input.new(req)
 
         env.update(
-          RACK_INPUT        => rack_input,
-          RACK_ERRORS       => $stderr,
-          RACK_URL_SCHEME   => ["yes", "on", "1"].include?(env[HTTPS]) ? "https" : "http",
-          RACK_IS_HIJACK    => true,
+          ::Rack::RACK_INPUT => input,
+          ::Rack::RACK_ERRORS => $stderr,
+          ::Rack::RACK_URL_SCHEME => ["yes", "on", "1"].include?(env[::Rack::HTTPS]) ? "https" : "http",
+          ::Rack::RACK_IS_HIJACK => true,
         )
 
-        env[QUERY_STRING] ||= ""
-        unless env[PATH_INFO] == ""
-          path, n = req.request_uri.path, env[SCRIPT_NAME].length
-          env[PATH_INFO] = path[n, path.length - n]
+        env[::Rack::QUERY_STRING] ||= ""
+        unless env[::Rack::PATH_INFO] == ""
+          path, n = req.request_uri.path, env[::Rack::SCRIPT_NAME].length
+          env[::Rack::PATH_INFO] = path[n, path.length - n]
         end
-        env[REQUEST_PATH] ||= [env[SCRIPT_NAME], env[PATH_INFO]].join
+        env[::Rack::REQUEST_PATH] ||= [env[::Rack::SCRIPT_NAME], env[::Rack::PATH_INFO]].join
 
         status, headers, body = @app.call(env)
         begin
           res.status = status
 
-          if value = headers[RACK_HIJACK]
+          if value = headers[::Rack::RACK_HIJACK]
             io_lambda = value
+            body = nil
           elsif !body.respond_to?(:to_path) && !body.respond_to?(:each)
             io_lambda = body
+            body = nil
           end
 
           if value = headers.delete('set-cookie')
             res.cookies.concat(Array(value))
           end
 
-          headers.each { |key, value|
+          headers.each do |key, value|
             # Skip keys starting with rack., per Rack SPEC
             next if key.start_with?('rack.')
 
@@ -117,22 +132,27 @@ module Rackup
             # merge the values per RFC 1945 section 4.2.
             value = value.join(", ") if Array === value
             res[key] = value
-          }
+          end
 
           if io_lambda
-            rd, wr = IO.pipe
-            res.body = rd
-            res.chunked = true
-            io_lambda.call wr
+            protocol = headers['rack.protocol'] || headers['upgrade']
+
+            if protocol
+              # Set all the headers correctly for an upgrade response:
+              res.upgrade!(protocol)
+            end
+            res.body = io_lambda
           elsif body.respond_to?(:to_path)
             res.body = ::File.open(body.to_path, 'rb')
           else
-            body.each { |part|
-              res.body << part
-            }
+            buffer = String.new
+            body.each do |part|
+              buffer << part
+            end
+            res.body = buffer
           end
         ensure
-          body.close if body.respond_to? :close
+          body.close if body.respond_to?(:close)
         end
       end
     end

data/lib/rackup/stream.rb

@@ -0,0 +1,199 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2019-2022, by Samuel Williams.
+
+module Rackup
+  # The input stream is an IO-like object which contains the raw HTTP POST data. When applicable, its external encoding must be “ASCII-8BIT” and it must be opened in binary mode, for Ruby 1.9 compatibility. The input stream must respond to gets, each, read and rewind.
+  class Stream
+    def initialize(input = nil, output = Buffered.new)
+      @input = input
+      @output = output
+
+      raise ArgumentError, "Non-writable output!" unless output.respond_to?(:write)
+
+      # Will hold remaining data in `#read`.
+      @buffer = nil
+      @closed = false
+    end
+
+    attr :input
+    attr :output
+
+    # This provides a read-only interface for data, which is surprisingly tricky to implement correctly.
+    module Reader
+      # rack.hijack_io must respond to:
+      # read, write, read_nonblock, write_nonblock, flush, close, close_read, close_write, closed?
+
+      # read behaves like IO#read. Its signature is read([length, [buffer]]). If given, length must be a non-negative Integer (>= 0) or nil, and buffer must be a String and may not be nil. If length is given and not nil, then this method reads at most length bytes from the input stream. If length is not given or nil, then this method reads all data until EOF. When EOF is reached, this method returns nil if length is given and not nil, or “” if length is not given or is nil. If buffer is given, then the read data will be placed into buffer instead of a newly created String object.
+      # @param length [Integer] the amount of data to read
+      # @param buffer [String] the buffer which will receive the data
+      # @return a buffer containing the data
+      def read(length = nil, buffer = nil)
+        return '' if length == 0
+
+        buffer ||= String.new.force_encoding(Encoding::BINARY)
+
+        # Take any previously buffered data and replace it into the given buffer.
+        if @buffer
+          buffer.replace(@buffer)
+          @buffer = nil
+        else
+          buffer.clear
+        end
+
+        if length
+          while buffer.bytesize < length and chunk = read_next
+            buffer << chunk
+          end
+
+          # This ensures the subsequent `slice!` works correctly.
+          buffer.force_encoding(Encoding::BINARY)
+
+          # This will be at least one copy:
+          @buffer = buffer.byteslice(length, buffer.bytesize)
+
+          # This should be zero-copy:
+          buffer.slice!(length, buffer.bytesize)
+
+          if buffer.empty?
+            return nil
+          else
+            return buffer
+          end
+        else
+          while chunk = read_next
+            buffer << chunk
+          end
+
+          return buffer
+        end
+      end
+
+      # Read at most `length` bytes from the stream. Will avoid reading from the underlying stream if possible.
+      def read_partial(length = nil)
+        if @buffer
+          buffer = @buffer
+          @buffer = nil
+        else
+          buffer = read_next
+        end
+
+        if buffer and length
+          if buffer.bytesize > length
+            # This ensures the subsequent `slice!` works correctly.
+            buffer.force_encoding(Encoding::BINARY)
+
+            @buffer = buffer.byteslice(length, buffer.bytesize)
+            buffer.slice!(length, buffer.bytesize)
+          end
+        end
+
+        return buffer
+      end
+
+      def gets
+        read_partial
+      end
+
+      def each
+        while chunk = read_partial
+          yield chunk
+        end
+      end
+
+      def read_nonblock(length, buffer = nil)
+        @buffer ||= read_next
+        chunk = nil
+
+        unless @buffer
+          buffer&.clear
+          return
+        end
+
+        if @buffer.bytesize > length
+          chunk = @buffer.byteslice(0, length)
+          @buffer = @buffer.byteslice(length, @buffer.bytesize)
+        else
+          chunk = @buffer
+          @buffer = nil
+        end
+
+        if buffer
+          buffer.replace(chunk)
+        else
+          buffer = chunk
+        end
+
+        return buffer
+      end
+    end
+
+    include Reader
+
+    def write(buffer)
+      if @output
+        @output.write(buffer)
+        return buffer.bytesize
+      else
+        raise IOError, "Stream is not writable, output has been closed!"
+      end
+    end
+
+    def write_nonblock(buffer)
+      write(buffer)
+    end
+
+    def <<(buffer)
+      write(buffer)
+    end
+
+    def flush
+    end
+
+    def close_read
+      @input&.close
+      @input = nil
+    end
+
+    # close must never be called on the input stream. huh?
+    def close_write
+      if @output.respond_to?(:close)
+        @output&.close
+      end
+
+      @output = nil
+    end
+
+    # Close the input and output bodies.
+    def close(error = nil)
+      self.close_read
+      self.close_write
+
+      return nil
+    ensure
+      @closed = true
+    end
+
+    # Whether the stream has been closed.
+    def closed?
+      @closed
+    end
+
+    # Whether there are any output chunks remaining?
+    def empty?
+      @output.empty?
+    end
+
+    private
+
+    def read_next
+      if @input
+        return @input.read
+      else
+        @input = nil
+        raise IOError, "Stream is not readable, input has been closed!"
+      end
+    end
+  end
+end

data/lib/rackup/version.rb

@@ -4,5 +4,5 @@
 # Copyright, 2022-2023, by Samuel Williams.
 
 module Rackup
-  VERSION = "2.0.0"
+  VERSION = "2.1.0"
 end

data/license.md

@@ -3,7 +3,7 @@
 Copyright, 2007-2009, by Leah Neukirchen.  
 Copyright, 2008, by Marc-André Cournoyer.  
 Copyright, 2009, by Aaron Pfeifer.  
-Copyright, 2009, by Graham Batty.  
+Copyright, 2009-2010, by Megan Batty.  
 Copyright, 2009-2010, by Michael Fellinger.  
 Copyright, 2009, by Genki Takiuchi.  
 Copyright, 2009, by Joshua Peek.  
@@ -16,7 +16,6 @@ Copyright, 2010, by Loren Segal.
 Copyright, 2010, by Andrew Bortz.  
 Copyright, 2010, by John Barnette.  
 Copyright, 2010, by John Sumsion.  
-Copyright, 2010, by Graham.  
 Copyright, 2011-2018, by Aaron Patterson.  
 Copyright, 2011, by Konstantin Haase.  
 Copyright, 2011, by Blake Mizerany.  

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rackup
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.1.0
 platform: ruby
 authors:
 - Samuel Williams
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-18 00:00:00.000000000 Z
+date: 2023-01-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack
@@ -29,16 +29,16 @@ dependencies:
   name: webrick
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.8'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -125,6 +125,7 @@ files:
 - lib/rackup/handler/webrick.rb
 - lib/rackup/lobster.rb
 - lib/rackup/server.rb
+- lib/rackup/stream.rb
 - lib/rackup/version.rb
 - license.md
 - readme.md