add Async::HTTP::Protocol::HTTP to auto-detect h1,h2 for inbound http:// connections

Author: zarqman

lib/async/http/protocol/http.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2023, by Thomas Morgan.
+
+require_relative 'http1'
+require_relative 'http2'
+
+module Async
+	module HTTP
+		module Protocol
+			# HTTP is an http:// server that auto-selects HTTP/1.1 or HTTP/2 by detecting the HTTP/2
+			# connection preface.
+			module HTTP
+				HTTP2_PREFACE = "PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n"
+				HTTP2_PREFACE_SIZE = HTTP2_PREFACE.bytesize
+				
+				def self.protocol_for(stream)
+					# Detect HTTP/2 connection preface
+					# https://www.rfc-editor.org/rfc/rfc9113.html#section-3.4
+					preface = stream.peek do |read_buffer|
+						if read_buffer.bytesize >= HTTP2_PREFACE_SIZE
+							break read_buffer[0, HTTP2_PREFACE_SIZE]
+						elsif read_buffer.bytesize > 0
+							# If partial read_buffer already doesn't match, no need to wait for more bytes.
+							break read_buffer unless HTTP2_PREFACE[read_buffer]
+						end
+					end
+					if preface == HTTP2_PREFACE
+						HTTP2
+					else
+						HTTP1
+					end
+				end
+				
+				# Only inbound connections can detect HTTP1 vs HTTP2 for http://.
+				# Outbound connections default to HTTP1.
+				def self.client(peer, **kwargs)
+					HTTP1.client(peer, **kwargs)
+				end
+				
+				def self.server(peer, **kwargs)
+					stream = IO::Stream.new(peer, sync: true)
+					protocol_for(stream).server(stream, **kwargs)
+				end
+				
+				def self.names
+					["h2", "http/1.1", "http/1.0"]
+				end
+			end
+		end
+	end
+end

lib/async/http/protocol/http1.rb

@@ -2,6 +2,7 @@
 
 # Released under the MIT License.
 # Copyright, 2017-2023, by Samuel Williams.
+# Copyright, 2023, by Thomas Morgan.
 
 require_relative 'http1/client'
 require_relative 'http1/server'
@@ -27,7 +28,7 @@ def self.client(peer)
 				end
 
 				def self.server(peer)
-					stream = IO::Stream.new(peer, sync: true)
+					stream = peer.is_a?(IO::Stream) ? peer : IO::Stream.new(peer, sync: true)
 
 					return HTTP1::Server.new(stream, VERSION)
 				end

lib/async/http/protocol/http10.rb

@@ -2,6 +2,7 @@
 
 # Released under the MIT License.
 # Copyright, 2017-2023, by Samuel Williams.
+# Copyright, 2023, by Thomas Morgan.
 
 require_relative 'http1'
 
@@ -26,7 +27,7 @@ def self.client(peer)
 				end
 
 				def self.server(peer)
-					stream = IO::Stream.new(peer, sync: true)
+					stream = peer.is_a?(IO::Stream) ? peer : IO::Stream.new(peer, sync: true)
 
 					return HTTP1::Server.new(stream, VERSION)
 				end

lib/async/http/protocol/http11.rb

@@ -3,6 +3,7 @@
 # Released under the MIT License.
 # Copyright, 2017-2023, by Samuel Williams.
 # Copyright, 2018, by Janko Marohnić.
+# Copyright, 2023, by Thomas Morgan.
 
 require_relative 'http1'
 
@@ -27,7 +28,7 @@ def self.client(peer)
 				end
 
 				def self.server(peer)
-					stream = IO::Stream.new(peer, sync: true)
+					stream = peer.is_a?(IO::Stream) ? peer : IO::Stream.new(peer, sync: true)
 
 					return HTTP1::Server.new(stream, VERSION)
 				end

lib/async/http/protocol/http2.rb

@@ -2,6 +2,7 @@
 
 # Released under the MIT License.
 # Copyright, 2018-2023, by Samuel Williams.
+# Copyright, 2023, by Thomas Morgan.
 
 require_relative 'http2/client'
 require_relative 'http2/server'
@@ -46,7 +47,7 @@ def self.client(peer, settings = CLIENT_SETTINGS)
 				end
 
 				def self.server(peer, settings = SERVER_SETTINGS)
-					stream = IO::Stream.new(peer, sync: true)
+					stream = peer.is_a?(IO::Stream) ? peer : IO::Stream.new(peer, sync: true)
 
 					server = Server.new(stream)
 

test/async/http/protocol/http.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2023, by Thomas Morgan.
+
+require 'async/http/protocol/http'
+require 'async/http/a_protocol'
+
+describe Async::HTTP::Protocol::HTTP do
+	with 'server' do
+		include Sus::Fixtures::Async::HTTP::ServerContext
+		let(:protocol) {subject}
+		
+		with 'http11 client' do
+			it 'should make a successful request' do
+				response = client.get('/')
+				expect(response).to be(:success?)
+				expect(response.version).to be == 'HTTP/1.1'
+				response.read
+			end
+		end
+		
+		with 'http2 client' do
+			def make_client(endpoint, **options)
+				options[:protocol] = Async::HTTP::Protocol::HTTP2
+				super
+			end
+			
+			it 'should make a successful request' do
+				response = client.get('/')
+				expect(response).to be(:success?)
+				expect(response.version).to be == 'HTTP/2'
+				response.read
+			end
+		end
+	end
+end