check that socket is writable in handshake

websockets · Jun 19, 2012 · 40321a7 · 40321a7
1 parent d3629c9
commit 40321a7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/WebSocketServer.js b/lib/WebSocketServer.js
@@ -198,7 +198,8 @@ function handleHixieUpgrade(req, socket, upgradeHead, cb) {
 		if (typeof origin != 'undefined') {
 			headers.push('Sec-WebSocket-Origin: ' + origin);
 		}
-
+
+		if (!socket.writable) return;
 		socket.setTimeout(60 * 60 * 1000);
 		socket.setNoDelay(true);
 		try {
@@ -290,7 +291,8 @@ function handleHybiUpgrade(req, socket, upgradeHead, cb) {
 		} catch(_) {}
 		return false;
 	}
-
+
+	if (!socket.writable) return;
 	socket.setTimeout(60 * 60 * 1000);
 	socket.setNoDelay(true);
 	var client = new WebSocket([req, socket, upgradeHead], {