最近使用的web项目中，需要服务器直接触发前端显示效果。
所以研究了一下websocket:
名词解释：
WebSocket
WebSocket协议是一种双向通信协议，它建立在TCP之上，同http一样通过TCP来传输数据，但是它和http最大的不同有两点：
1.WebSocket是一种双向通信协议，在建立连接后，WebSocket服务器和Browser/UA都能主动的向对方发送或接收数据，就像Socket一样，不同的是WebSocket是一种建立在Web基础上的一种简单模拟Socket的协议；
2.WebSocket需要通过握手连接，类似于TCP它也需要客户端和服务器端进行握手连接，连接成功后才能相互通信。
当Web应用程序调用new WebSocket(url)接口时，Browser就开始了与地址为url的WebServer建立握手连接的过程。
客户端向服务器发送请求：
GET /chat HTTP/1.1
Host: server.example.com
Upgrade: websocket
Connection: Upgrade
Sec-WebSocket-Key:dGhlIHNhbXBsZSBub25jZQ==
Origin: http://example.com
Sec-WebSocket-Protocol: chat,superchat
Sec-WebSocket-Version: 13
服务器端返回内容:
HTTP/1.1 101 Switching Protocols
Upgrade: websocket
Connection: Upgrade
Sec-WebSocket-Accept: xsOSgr30aKL2GNZKNHKmeT1qYjA=
请求中的【Sec-WebSocket-Key】是随机发送的。而服务器返回的【Sec-WebSocket-Accept】是将【Sec-WebSocket-Key】加上一个固定字符串【258EAFA5-E914-47DA-95CA-C5AB0DC85B11】，并使用SHA-1加密后，再进行BASE-64编码生成的。
一个简单的DEMO:
前端页面:
<!DOCTYPE html>
<html>
<head>
<title>Web Socket Client</title>
</head>
<body style="padding:10px;">
<h1>聊天室</h1>
<div style="margin:5px 0px;">
Address:
<div><input id="address" type="text" value="ws://127.0.0.1:1818" style="width:400px;" /></div>
</div>
<div style="margin:5px 0px;">
Name:
<div><input id="name" type="text" value="Byron" style="width:400px;" /></div>
</div>
<div>
<button id="connect" onclick="connect();">connect server</button>
<button id="disconnect" onclick="quit();">disconnect</button>
<button id="clear" onclick="clearMsg();">clear</button>
</div>
<h5 style="margin:4px 0px;">Message:</h5>
<div id="message" style="border:solid 1px #333; padding:4px; width:400px; overflow:auto;
background-color:#404040; height:300px; margin-bottom:8px; font-size:14px;">
</div>
<input id="text" type="text" onkeypress="enter(event);" style="width:340px" />
<button id="send" onclick="send();">send</button> <script type="text/javascript">
var name=document.getElementById('name').value;
var msgContainer=document.getElementById('message');
var text=document.getElementById('text'); function connect () {
var address=document.getElementById('address').value; ws=new WebSocket(address);
ws.onopen=function(e){
var msg=document.createElement('div');
msg.style.color='#0f0';
msg.innerHTML="Server > connection open.";
msgContainer.appendChild(msg);
ws.send('{'+document.getElementById('name').value+'}');
};
ws.onmessage=function(e){
var msg=document.createElement('div');
msg.style.color='#fff';
msg.innerHTML=e.data;
msgContainer.appendChild(msg);
};
ws.onerror=function(e){
var msg=document.createElement('div');
msg.style.color='#0f0';
msg.innerHTML='Server > '+e.data;
msgContainer.appendChild(msg);
};
ws.onclose=function(e){
var msg=document.createElement('div');
msg.style.color='#0f0';
msg.innerHTML="Server > connection closed by server.";
msgContainer.appendChild(msg);
};
text.focus();
} function quit(){
if(ws){
ws.close();
var msg=document.createElement('div');
msg.style.color='#0f0';
msg.innerHTML='Server > connection closed.';
msgContainer.appendChild(msg);
ws=null;
}
} function send(){
ws.send(text.value);
setTimeout(function(){
msgContainer.scrollTop=msgContainer.getBoundingClientRect().height;
},100);
text.value='';
text.focus();
} function clearMsg(){
msgContainer.innerHTML="";
} function enter(event){
if(event.keyCode==13){
send();
}
}
</script>
</body>
</html>
后台代码（C#):
class Program
{ private static Socket listener;
private static Hashtable ht;
static void Main(string[] args)
{
int port = 1818;//监听端口为1818端口
ht = new Hashtable();//用于存放客户端的连接socket
byte[] buffer = new byte[1024]; var localEP = new IPEndPoint(IPAddress.Any, port);
listener = new Socket(localEP.Address.AddressFamily, SocketType.Stream, ProtocolType.Tcp); try
{
listener.Bind(localEP);
listener.Listen(10); Console.WriteLine("等待客户端连接....");
while (true)
{
Socket clientSocket = listener.Accept(); //接收到客户端的连接
var th = new Thread(new ParameterizedThreadStart(Receive));
th.Start(clientSocket);
}
}
catch (Exception e)
{
Console.WriteLine(e.ToString());
}
}
/// <summary>
/// 线程调用
/// </summary>
private static void Receive(object o)//Socket clientSocket)
{
Socket clientSocket = (Socket)o;
clientSocket.Blocking = true;
IPEndPoint clientipe = (IPEndPoint)clientSocket.RemoteEndPoint;
// Console.WriteLine("[" + clientipe.Address.ToString() + "] Connected");
var key = string.Format("{0}-X-Q-X-{1}", clientipe.Address.ToString(), clientipe.Port);
if (!ht.ContainsKey(key))
{
//将ip地址设置为hashTable的key值 若hashTable中存在该ip地址则不再ht中添加socket以免发送重复数据
ht.Add(key, clientSocket);
}
Console.WriteLine("接收到了客户端：ip" + clientSocket.RemoteEndPoint.ToString() + "的连接");
byte[] buffer = new byte[1024];
int length = clientSocket.Receive(buffer);
clientSocket.Send(PackHandShakeData(GetSecKeyAccetp(buffer, length)));
Console.WriteLine("已经发送握手协议了...."); //接收用户姓名信息
length = clientSocket.Receive(buffer);
string xm = AnalyticData(buffer, length); clientSocket.Send(PackData("连接时间：" + DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"))); try
{ while (true)
{
var errLs = new List<object>();
//接受客户端数据
Console.WriteLine("等待客户端数据....");
try
{
length = clientSocket.Receive(buffer);//接受客户端信息
}
catch (SocketException e)
{
//去除字典
Console.WriteLine(e.Message);
try
{
foreach (DictionaryEntry de in ht)
{
if (de.Value == clientSocket)
{
ht.Remove(de.Key);
}
}
}
catch (Exception ex)
{
Console.WriteLine(ex.Message);
} }
string clientMsg = AnalyticData(buffer, length);
Console.WriteLine("接受到客户端数据：" + clientMsg); //发送数据
string sendMsg = "" + clientMsg;
Console.WriteLine("发送数据：“" + sendMsg + "” 至客户端....");
//遍历hashTable中的数据获取Socket发送数据
foreach (DictionaryEntry de in ht)
{
try
{
var sc = (Socket)de.Value;
sc.Send(PackData(clientSocket.RemoteEndPoint.ToString() + xm + "说：" + sendMsg));
}
catch (Exception e)
{
Console.WriteLine("Num:{0} err:{1}", ht.Count, e);
errLs.Add(de.Key);
}
} if (errLs != null && errLs.Any())
{
foreach (var item in errLs)
{
ht.Remove(item);
}
} Thread.Sleep(1000);
}
}
catch (SocketException e)
{
//去除字典
Console.WriteLine(e.Message);
}
catch (System.ObjectDisposedException e)
{
//去除字典
Console.WriteLine(e.Message);
}
catch (Exception e)
{
Console.WriteLine(e.Message);
} }
/// <summary>
/// 打包握手信息
/// </summary>
/// <param name="secKeyAccept">Sec-WebSocket-Accept</param>
/// <returns>数据包</returns>
private static byte[] PackHandShakeData(string secKeyAccept)
{
var responseBuilder = new StringBuilder();
responseBuilder.Append("HTTP/1.1 101 Switching Protocols" + Environment.NewLine);
responseBuilder.Append("Upgrade: websocket" + Environment.NewLine);
responseBuilder.Append("Connection: Upgrade" + Environment.NewLine);
responseBuilder.Append("Sec-WebSocket-Accept: " + secKeyAccept + Environment.NewLine + Environment.NewLine);
//如果把上一行换成下面两行，才是thewebsocketprotocol-17协议，但居然握手不成功，目前仍没弄明白！
//responseBuilder.Append("Sec-WebSocket-Accept: " + secKeyAccept + Environment.NewLine);
//responseBuilder.Append("Sec-WebSocket-Protocol: chat" + Environment.NewLine); return Encoding.UTF8.GetBytes(responseBuilder.ToString());
} /// <summary>
/// 生成Sec-WebSocket-Accept
/// </summary>
/// <param name="handShakeText">客户端握手信息</param>
/// <returns>Sec-WebSocket-Accept</returns>
private static string GetSecKeyAccetp(byte[] handShakeBytes, int bytesLength)
{
string handShakeText = Encoding.UTF8.GetString(handShakeBytes, 0, bytesLength);
string key = string.Empty;
Regex r = new Regex(@"Sec\-WebSocket\-Key:(.*?)\r\n");
Match m = r.Match(handShakeText);
if (m.Groups.Count != 0)
{
key = Regex.Replace(m.Value, @"Sec\-WebSocket\-Key:(.*?)\r\n", "$1").Trim();
}
byte[] encryptionString = SHA1.Create().ComputeHash(Encoding.ASCII.GetBytes(key + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"));
return Convert.ToBase64String(encryptionString);
} /// <summary>
/// 解析客户端数据包
/// </summary>
/// <param name="recBytes">服务器接收的数据包</param>
/// <param name="recByteLength">有效数据长度</param>
/// <returns></returns>
private static string AnalyticData(byte[] recBytes, int recByteLength)
{
if (recByteLength < 2) { return string.Empty; } bool fin = (recBytes[0] & 0x80) == 0x80; // 1bit，1表示最后一帧
if (!fin)
{
return string.Empty;// 超过一帧暂不处理
} bool mask_flag = (recBytes[1] & 0x80) == 0x80; // 是否包含掩码
if (!mask_flag)
{
return string.Empty;// 不包含掩码的暂不处理
} int payload_len = recBytes[1] & 0x7F; // 数据长度 byte[] masks = new byte[4];
byte[] payload_data; if (payload_len == 126)
{
Array.Copy(recBytes, 4, masks, 0, 4);
payload_len = (UInt16)(recBytes[2] << 8 | recBytes[3]);
payload_data = new byte[payload_len];
Array.Copy(recBytes, 8, payload_data, 0, payload_len); }
else if (payload_len == 127)
{
Array.Copy(recBytes, 10, masks, 0, 4);
byte[] uInt64Bytes = new byte[8];
for (int i = 0; i < 8; i++)
{
uInt64Bytes[i] = recBytes[9 - i];
}
UInt64 len = BitConverter.ToUInt64(uInt64Bytes, 0); payload_data = new byte[len];
for (UInt64 i = 0; i < len; i++)
{
payload_data[i] = recBytes[i + 14];
}
}
else
{
Array.Copy(recBytes, 2, masks, 0, 4);
payload_data = new byte[payload_len];
Array.Copy(recBytes, 6, payload_data, 0, payload_len); } for (var i = 0; i < payload_len; i++)
{
payload_data[i] = (byte)(payload_data[i] ^ masks[i % 4]);
} return Encoding.UTF8.GetString(payload_data);
} /// <summary>
/// 打包服务器数据
/// </summary>
/// <param name="message">数据</param>
/// <returns>数据包</returns>
private static byte[] PackData(string message)
{
byte[] contentBytes = null;
byte[] temp = Encoding.UTF8.GetBytes(message); if (temp.Length < 126)
{
contentBytes = new byte[temp.Length + 2];
contentBytes[0] = 0x81;
contentBytes[1] = (byte)temp.Length;
Array.Copy(temp, 0, contentBytes, 2, temp.Length);
}
else if (temp.Length < 0xFFFF)
{
contentBytes = new byte[temp.Length + 4];
contentBytes[0] = 0x81;
contentBytes[1] = 126;
contentBytes[2] = (byte)(temp.Length & 0xFF);
contentBytes[3] = (byte)(temp.Length >> 8 & 0xFF);
Array.Copy(temp, 0, contentBytes, 4, temp.Length);
}
else
{
// 暂不处理超长内容
} return contentBytes;
}
}
运行效果：
