lib/websocket/handshake/handler/server75.rb in websocket-1.2.3 vs lib/websocket/handshake/handler/server75.rb in websocket-1.2.4

@@ -14,10 +14,16 @@
           [
             %w(Upgrade WebSocket),
             %w(Connection Upgrade),
             ['WebSocket-Origin', @handshake.headers['origin']],
             ['WebSocket-Location', @handshake.uri]
-          ]
+          ] + protocol
+        end
+
+        def protocol
+          return [] unless @handshake.headers.key?('websocket-protocol')
+          proto = @handshake.headers['websocket-protocol']
+          [['WebSocket-Protocol', @handshake.protocols.include?(proto) ? proto : nil]]
         end
       end
     end
   end
 end