Merge pull request websockets#94 from guille/add/headers-hook

Headers hook
bmeck · Jul 3, 2012 · 6394681 · 6394681
2 parents 41084f1 + 3ecdb99
commit 6394681
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/doc/ws.md b/doc/ws.md
@@ -37,6 +37,12 @@ When the Upgrade was successfully, the `callback` will be called with a `ws.WebS
 
 If the underlying server emits an error, it will be forwarded here.
 
+### Event: 'headers'
+
+`function (headers) { }`
+
+Emitted with the object of HTTP headers that are going to be written to the `Stream` as part of the handshake.
+
 ### Event: 'connection'
 
 `function (socket) { }`

diff --git a/lib/WebSocketServer.js b/lib/WebSocketServer.js
@@ -196,6 +196,9 @@ function handleHybiUpgrade(req, socket, upgradeHead, cb) {
       headers['Sec-WebSocket-Protocol'] = protocol;
     }
 
+    // allows external modification/inspection of handshake headers
+    self.emit('headers', headers);
+
     socket.setTimeout(0);
     socket.setNoDelay(true);
     try {