C# 实现WEBSOCKET聊天应用示例

ProtocolDraft10.cs
public class ProtocolDraft10 : IProtocol
{
private const String WebSocketKeyPattern = @"Sec\-WebSocket\-Key:\s+(?<key>.*)\r\n";
private const String MagicKey = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
private const Char charOne = ‘1‘;
private const Char charZero = ‘0‘;

#region Handshake

// 发送回复信息完成握手
public Byte[] ProduceResponse(string request)
{
String webSocketKey = Common.GetRegexValue(request, WebSocketKeyPattern)[0].Groups["key"].Value;
String acceptKey = produceAcceptKey(webSocketKey);
StringBuilder stringBuilder = new StringBuilder();
stringBuilder.Append(String.Concat("HTTP/1.1 101 Web Socket Protocol Handshake", Environment.NewLine));
stringBuilder.Append(String.Concat("Upgrade: WebSocket", Environment.NewLine));
stringBuilder.Append(String.Concat("Connection: Upgrade", Environment.NewLine));
stringBuilder.Append(String.Concat("Sec-WebSocket-Accept: ", acceptKey, Environment.NewLine, Environment.NewLine));
String asd = stringBuilder.ToString();
return Encoding.UTF8.GetBytes(stringBuilder.ToString());
}
// 根据Sec-WebSocket-Key和MagicKey生成AcceptKey
private String produceAcceptKey(String webSocketKey)
{
Byte[] acceptKey = SHA1.Create().ComputeHash(Encoding.ASCII.GetBytes(webSocketKey + MagicKey));
return Convert.ToBase64String(acceptKey);
}

#endregion

#region Decode
// 对客户端发来的数据进行解析
public Message Decode(Byte[] data)
{
Byte[] buffer = new Byte[14];
if (data.Length >= 14)
Buffer.BlockCopy(data, 0, buffer, 0, 14);
else
Buffer.BlockCopy(data, 0, buffer, 0, data.Length);
MessageHeader header = analyseHead(buffer);
Message msg = new Message();
msg.header = header;

Byte[] payload;
if (header != null)
{
payload = new Byte[data.Length - header.PayloadDataStartIndex];
Buffer.BlockCopy(data, header.PayloadDataStartIndex, payload, 0, payload.Length);
if (header.MASK == charOne)
{
for (int i = 0; i < payload.Length; i++)
{
payload[i] = (Byte)(payload[i] ^ header.Maskey[i % 4]);
}
}
}
else
{
msg.Data = Encoding.UTF8.GetString(data);
return msg;
}

if(header.Opcode==OperType.Text)
msg.Data=Encoding.UTF8.GetString(payload);

return msg;
}
privateMessageHeader analyseHead(Byte[] buffer)
{
MessageHeader header =newMessageHeader();
header.FIN =(buffer[0]&0x80)==0x80? charOne : charZero;
header.RSV1 =(buffer[0]&0x40)==0x40? charOne : charZero;
header.RSV2 =(buffer[0]&0x20)==0x20? charOne : charZero;
header.RSV3 =(buffer[0]&0x10)==0x10? charOne : charZero;

if((buffer[0]&0xA)==0xA)
header.Opcode=OperType.Pong;
elseif((buffer[0]&0x9)==0x9)
header.Opcode=OperType.Ping;
elseif((buffer[0]&0x8)==0x8)
header.Opcode=OperType.Close;
elseif((buffer[0]&0x2)==0x2)
header.Opcode=OperType.Binary;
elseif((buffer[0]&0x1)==0x1)
header.Opcode=OperType.Text;
elseif((buffer[0]&0x0)==0x0)
header.Opcode=OperType.Row;

header.MASK =(buffer[1]&0x80)==0x80? charOne : charZero;
Int32 len = buffer[1]&0x7F;
if(len ==126)
{
header.Payloadlen=(UInt16)(buffer[2]<<8| buffer[3]);
if(header.MASK == charOne)
{
header.Maskey=newByte[4];
Buffer.BlockCopy(buffer,4, header.Maskey,0,4);
header.PayloadDataStartIndex=8;
}
else
header.PayloadDataStartIndex=4;
}
elseif(len ==127)
{
Byte[] byteLen =newByte[8];
Buffer.BlockCopy(buffer,4, byteLen,0,8);
header.Payloadlen=BitConverter.ToUInt64(byteLen,0);
if(header.MASK == charOne)
{
header.Maskey=newByte[4];
Buffer.BlockCopy(buffer,10, header.Maskey,0,4);
header.PayloadDataStartIndex=14;
}
else
header.PayloadDataStartIndex=10;
}
else
{
if(header.MASK == charOne)
{
header.Maskey=newByte[4];
Buffer.BlockCopy(buffer,2, header.Maskey,0,4);
header.PayloadDataStartIndex=6;
}
else
header.PayloadDataStartIndex=2;
}
return header;
}

#endregion

#region Encode
// 对要发送的数据进行编码一符合草案10的规则
publicByte[]Encode(String msg)
{
Byte[] data =Encoding.UTF8.GetBytes(msg);
Int32 dataLength = data.Length;

Byte[] head = packetHeader(OperType.Text, dataLength);
for(int i =0; i < data.Length; i++)
{
data[i]=(Byte)(data[i]^ maskKey[i %4]);
}

Byte[] result =newByte[head.Length+ dataLength];
Buffer.BlockCopy(head,0, result,0, head.Length);
Buffer.BlockCopy(data,0, result, head.Length, dataLength);
return result;
}
privateconstByte byte80 =0x80;
privateByte[] maskKey =newByte[]{113,105,97,110};
privateByte[] packetHeader(OperType operType,Int32 length)
{
Byte byteHead =(Byte)(byte80 |(Byte)operType);
Byte[] byteLen;
if(length <126)
{
byteLen =newByte[1];
byteLen[0]=(Byte)(byte80 |(Byte)length);
}
elseif(length <65535)
{
byteLen =newByte[3];
byteLen[0]=(Byte)(byte80 |(Byte)126);
for(int i =1; i <3; i++)
byteLen[i]=(Byte)(length >>(8*(2- i)));
}
else
{
byteLen =newByte[9];
byteLen[0]=(Byte)(byte80 |(Byte)127);
for(int i =1; i <9; i++)
byteLen[i]=(Byte)(length >>(8*(8- i)));
}

Byte[] packet =newByte[1+ byteLen.Length+ maskKey.Length];
packet[0]= byteHead;
Buffer.BlockCopy(byteLen,0, packet,1, byteLen.Length);
Buffer.BlockCopy(maskKey,0, packet,1+ byteLen.Length, maskKey.Length);
return packet;
}

#endregion
}

<script type="text/javascript">
var noSupportMessage = "您的浏览器不支持WebSocket!";
var ws;

function connectSocketServer() {
var messageBoard = $("#messageBoard");

var support = "MozWebSocket" in window ? ‘MozWebSocket‘ : (‘WebSocket‘ in window ? ‘WebSocket‘ : null);

if (support == null) {
alert(noSupportMessage);
messageBoard.append(‘*‘ + noSupportMessage + "<br />");
return;
}

messageBoard.append("* Connecting to server..<br />");
try {
ws = new WebSocket(‘ws://localhost:3000‘);
} catch (e) {
alert(e.Message);
}

ws.onmessage = function (event) {
messageBoard.append(event.data + "<br />");
}

ws.onopen = function () {
messageBoard.append(‘* Connection open<br />‘);
}

ws.onclose = function () {
messageBoard.append(‘* Connection closed<br />‘);
}
}

function sendMessage() {
if (ws) {
var mssageBox = document.getElementById("messageBox");
var user = document.getElementById("users");
var msg = user.value + "<separator>" + mssageBox.value;
ws.send(msg);
mssageBox.value = "";
} else {
alert(noSupportMessage);
}
}

window.onload = function () {
connectSocketServer();
}
</script>