Add missing Sec-WebSocket-Protocol header to hybi protocols

LearnBoost · Mar 10, 2012 · 5093e49 · 5093e49
1 parent 8dc184f
commit 5093e49
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/protocols/hybi-07-12.js b/lib/protocols/hybi-07-12.js
@@ -123,6 +123,11 @@ WebSocket.prototype.onOpen = function () {
     , 'Sec-WebSocket-Accept: ' + key
   ];
 
+  if (this.req.headers['sec-websocket-protocol']){
+    headers.push('Sec-WebSocket-Protocol: '
+        + this.req.headers['sec-websocket-protocol']);
+  }
+
   try {
     this.socket.write(headers.concat('', '').join('\r\n'));
     this.socket.setTimeout(0);

diff --git a/lib/protocols/hybi-16.js b/lib/protocols/hybi-16.js
@@ -125,6 +125,11 @@ WebSocket.prototype.onOpen = function () {
     , 'Sec-WebSocket-Accept: ' + key
   ];
 
+  if (this.req.headers['sec-websocket-protocol']){
+    headers.push('Sec-WebSocket-Protocol: '
+        + this.req.headers['sec-websocket-protocol']);
+  }
+
   try {
     this.socket.write(headers.concat('', '').join('\r\n'));
     this.socket.setTimeout(0);