/// <summary>
/// Perform WebSocket client upgrade
/// </summary>
/// <param name="response">WebSocket upgrade HTTP response</param>
/// <param name="id">WebSocket client Id</param>
/// <returns>'true' if the WebSocket was successfully upgrade, 'false' if the WebSocket was not upgrade</returns>
public Boolean PerformClientUpgrade(HttpNetworkResponse response, Guid id)
{
if (response.Status != 101)
{
return(false);
}
Boolean error = false;
Boolean accept = false;
Boolean connection = false;
Boolean upgrade = false;
// Validate WebSocket handshake headers
for (Int32 i = 0; i < response.Headers; ++i)
{
Tuple <String, String> header = response.Header(i);
String key = header.Item1;
String value = header.Item2;
if (key == "Connection")
{
if (value != "Upgrade")
{
error = true;
_wsHandler.OnWsError("Invalid WebSocket handshaked response: 'Connection' header value must be 'Upgrade'");
break;
}
connection = true;
}
else if (key == "Upgrade")
{
if (value != "websocket")
{
error = true;
_wsHandler.OnWsError("Invalid WebSocket handshaked response: 'Upgrade' header value must be 'websocket'");
break;
}
upgrade = true;
}
else if (key == "Sec-WebSocket-Accept")
{
// Calculate the original WebSocket hash
String wskey = Convert.ToBase64String(Encoding.UTF8.GetBytes(id.ToString())) + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
String wshash;
using (SHA1Managed sha1 = new SHA1Managed())
{
wshash = Encoding.UTF8.GetString(sha1.ComputeHash(Encoding.UTF8.GetBytes(wskey)));
}
// Get the received WebSocket hash
wskey = Encoding.UTF8.GetString(Convert.FromBase64String(value));
// Compare original and received hashes
if (String.Compare(wskey, wshash, StringComparison.InvariantCulture) != 0)
{
error = true;
_wsHandler.OnWsError("Invalid WebSocket handshaked response: 'Sec-WebSocket-Accept' value validation failed");
break;
}
accept = true;
}
}
// Failed to perform WebSocket handshake
if (!accept || !connection || !upgrade)
{
if (!error)
{
_wsHandler.OnWsError("Invalid WebSocket response");
}
return(false);
}
// WebSocket successfully handshaked!
WsHandshaked = true;
IRandom random = RandomUtils.Create();
random.NextBytes(WsSendMask);
_wsHandler.OnWsConnected(response);
return(true);
}
