Copy protocol if successful.

socketry · Sep 3, 2024 · 5c338e3 · 5c338e3
1 parent 9dffa99
commit 5c338e3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lib/async/http/protocol/http2/response.rb b/lib/async/http/protocol/http2/response.rb
@@ -54,6 +54,11 @@ def receive_initial_headers(headers, end_stream)
 							@response.status = status
 							@headers = ::Protocol::HTTP::Headers.new
 
+							# If the protocol request was successful, ensure the response protocol matches:
+							if status == 200 and protocol = @response.request.protocol
+								@response.protocol = Array(protocol).first
+							end
+
 							headers.each do |key, value|
 								# It's guaranteed that this should be the first header:
 								if key == CONTENT_LENGTH