async-http 0.38.0 → 0.38.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 89e83fd398432897302be1ddbb07fad391a285216320e282e218c2faf07f4183
-  data.tar.gz: c9f37f4776390db5b0d050845397d7b530b145b6072f062b453957944683d7d6
+  metadata.gz: 83a5e5bd7e766404f4011a11d61d6cfcc75d1cff40c1822895f21f71ccf91925
+  data.tar.gz: ecdde5ee1128a0ee9f947a1da217c683fda1c65c281dba0694ff72749a31ca9f
 SHA512:
-  metadata.gz: 53f726aec29bd9130015446d9e05b5b11be437c85978f9cb6d22647ebd17dd033182e647dda82c31bfb42e14cf01b9c1b8bae904e781370564f33df390470916
-  data.tar.gz: c698697b2d53e17a2321d430b01cd5746ef4eaf5623cb59b8a08200f86c4d3357037d85444c4141b23198bf4eba1aedd4915c8e79646717a761c02e8dd7d79b9
+  metadata.gz: ce5c4265e43776b208ef51ebcb5b54b84f0b86e7d8d52dbccf3905869fc6094ec88d74b6afb35eba557ffcae0699dae5f130e7b72f507ac989c28e2904e55fbb
+  data.tar.gz: 74b16c8ad259f0aaded61ef8094d26043354b717d299c3de04b2a4aab0169b74ebc2808f889ff284468a5ffabd4ad1a2f0e988dd67b268d3a8660303df9b7fe3

data/lib/async/http/body/slowloris.rb

@@ -0,0 +1,70 @@
+# Copyright, 2018, by Samuel G. D. Williams. <http://www.codeotaku.com>
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+require_relative 'writable'
+
+require 'async/clock'
+
+module Async
+	module HTTP
+		module Body
+			# A dynamic body which you can write to and read from.
+			class Slowloris < Writable
+				class ThroughputError < StandardError
+					def initialize(throughput, minimum_throughput, time_since_last_write)
+						super("Slow write: #{throughput.round(1)}bytes/s less than required #{minimum_throughput.round}bytes/s.")
+					end
+				end
+				
+				# In order for this implementation to work correctly, you need to use a LimitedQueue.
+				# @param minimum_throughput [Integer] the minimum bytes per second otherwise this body will be forcefully closed.
+				def initialize(*args, minimum_throughput: 1024, **options)
+					super(*args, **options)
+					
+					@minimum_throughput = minimum_throughput
+					
+					@last_write_at = nil
+					@last_chunk_size = nil
+				end
+				
+				attr :minimum_throughput
+				
+				# If #read is called regularly to maintain throughput, that is good. If #read is not called, that is a problem. Throughput is dependent on data being available, from #write, so it doesn't seem particularly problimatic to do this check in #write.
+				def write(chunk)
+					if @last_chunk_size
+						time_since_last_write = Async::Clock.now - @last_write_at
+						throughput = @last_chunk_size / time_since_last_write
+						
+						if throughput < @minimum_throughput
+							error = ThroughputError.new(throughput, @minimum_throughput, time_since_last_write)
+							
+							self.close(error)
+						end
+					end
+					
+					super.tap do
+						@last_write_at = Async::Clock.now
+						@last_chunk_size = chunk&.bytesize
+					end
+				end
+			end
+		end
+	end
+end

data/lib/async/http/protocol/http2/request.rb

@@ -51,6 +51,9 @@ module Async
 						return request.stream
 					end
 					
+					def close!(state)
+					end
+					
 					# @return [Stream] the promised stream, on which to send data.
 					def push(path, headers = nil)
 						push_headers = [

data/lib/async/http/protocol/http2/response.rb

@@ -52,6 +52,10 @@ module Async
 						return promise.stream
 					end
 					
+					def close!(state)
+						self.promises.enqueue(nil)
+					end
+					
 					# Notify anyone waiting on the response headers to be received (or failure).
 					protected def notify!
 						if @notification

data/lib/async/http/protocol/http2/stream.rb

@@ -43,6 +43,12 @@ module Async
 						@delegate.create_promise_stream(headers, stream_id)
 					end
 					
+					def close!(state = :closed)
+						super
+						
+						@delegate.close!(state)
+					end
+					
 					def send_body(body, task: Async::Task.current)
 						# TODO Might need to stop this task when body is cancelled.
 						@task = task.async do |subtask|

data/lib/async/http/version.rb

@@ -20,6 +20,6 @@
 
 module Async
 	module HTTP
-		VERSION = "0.38.0"
+		VERSION = "0.38.1"
 	end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: async-http
 version: !ruby/object:Gem::Version
-  version: 0.38.0
+  version: 0.38.1
 platform: ruby
 authors:
 - Samuel Williams
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-02-09 00:00:00.000000000 Z
+date: 2019-02-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async
@@ -161,6 +161,7 @@ files:
 - lib/async/http/body/readable.rb
 - lib/async/http/body/reader.rb
 - lib/async/http/body/rewindable.rb
+- lib/async/http/body/slowloris.rb
 - lib/async/http/body/streamable.rb
 - lib/async/http/body/wrapper.rb
 - lib/async/http/body/writable.rb