C# 实现WebSocket服务端
using System;
using ;
using ;
using ;
using ;
using ;
using ;
using ;
using ;
namespace WebSocketServer
{
public class WebSocket
{
private Dictionary<string, Session> SessionPool = new Dictionary<string, Session>();
private Dictionary<string, string> MsgPool = new Dictionary<string, string>();
#region 启动WebSocket服务
/// <summary>
/// 启动WebSocket服务
/// </summary>
public void start(int port)
{
Socket SockeServer = new Socket(, , );
(new IPEndPoint(, port));
(20);
(new AsyncCallback(Accept), SockeServer);
("服务已启动");
("按任意键关闭服务");
();
}
#endregion
#region 处理客户端连接请求
/// <summary>
/// 处理客户端连接请求
/// </summary>
/// <param name="result"></param>
private void Accept(IAsyncResult socket)
{
// 还原传入的原始套接字
Socket SockeServer = (Socket);
// 在原始套接字上调用EndAccept方法，返回新的套接字
Socket SockeClient = (socket);
byte[] buffer = new byte[4096];
try
{
//接收客户端的数据
(buffer, 0, , , new AsyncCallback(Recieve), SockeClient);
//保存登录的客户端
Session session = new Session();
= SockeClient;
= ();
= buffer;
lock (SessionPool)
{
if (())
{
this.();
}
this.(, session);
}
//准备接受下一个客户端
(new AsyncCallback(Accept), SockeServer);
(string.Format("Client {0} connected", ));
}
catch (Exception ex)
{
("Error : " + ());
}
}
#endregion
#region 处理接收的数据
/// <summary>
/// 处理接受的数据
/// </summary>
/// <param name="socket"></param>
private void Recieve(IAsyncResult socket)
{
Socket SockeClient = (Socket);
string IP = ();
if (SockeClient == null || !(IP))
{
return;
}
try
{
int length = (socket);
byte[] buffer = SessionPool[IP].buffer;
(buffer, 0, , , new AsyncCallback(Recieve), SockeClient);
string msg = Encoding.(buffer, 0, length);
// websocket建立连接的时候，除了TCP连接的三次握手，websocket协议中客户端与服务器想建立连接需要一次额外的握手动作
if (("Sec-WebSocket-Key"))
{
(PackageHandShakeData(buffer, length));
SessionPool[IP].isWeb = true;
return;
}
if (SessionPool[IP].isWeb)
{
msg = AnalyzeClientData(buffer, length);
}
byte[] msgBuffer = PackageServerData(msg);
foreach (Session se in )
{
(msgBuffer, , );
}
}
catch
{
(true);
("客户端 {0} 断开连接", IP);
(IP);
}
}
#endregion
#region 客户端和服务端的响应
/*
* 客户端向服务器发送请求
*
* GET / HTTP/1.1
* Origin: http://localhost:1416
* Sec-WebSocket-Key: vDyPp55hT1PphRU5OAe2Wg==
* Connection: Upgrade
* Upgrade: Websocket
*Sec-WebSocket-Version: 13
* User-Agent: Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko
* Host: localhost:8064
* DNT: 1
* Cache-Control: no-cache
* Cookie: DTRememberName=admin
*
* 服务器给出响应
*
* HTTP/1.1 101 Switching Protocols
* Upgrade: websocket
* Connection: Upgrade
* Sec-WebSocket-Accept: xsOSgr30aKL2GNZKNHKmeT1qYjA=
*
* 在请求中的“Sec-WebSocket-Key”是随机的，服务器端会用这些数据来构造出一个SHA-1的信息摘要。把“Sec-WebSocket-Key”加上一个魔幻字符串
* “258EAFA5-E914-47DA-95CA-C5AB0DC85B11”。使用 SHA-1 加密，之后进行 BASE-64编码，将结果做为 “Sec-WebSocket-Accept” 头的值，返回给客户端
*/
#endregion
#region 打包请求连接数据
/// <summary>
/// 打包请求连接数据
/// </summary>
/// <param name="handShakeBytes"></param>
/// <param name="length"></param>
/// <returns></returns>
private byte[] PackageHandShakeData(byte[] handShakeBytes, int length)
{
string handShakeText = Encoding.(handShakeBytes, 0, length);
string key = string.Empty;
Regex reg = new Regex(@"Sec\-WebSocket\-Key:(.*?)\r\n");
Match m = (handShakeText);
if ( != "")
{
key = (, @"Sec\-WebSocket\-Key:(.*?)\r\n", "$1").Trim();
}
byte[] secKeyBytes = ().ComputeHash((key + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"));
string secKey = Convert.ToBase64String(secKeyBytes);
var responseBuilder = new StringBuilder();
("HTTP/1.1 101 Switching Protocols" + "\r\n");
("Upgrade: websocket" + "\r\n");
("Connection: Upgrade" + "\r\n");
("Sec-WebSocket-Accept: " + secKey + "\r\n\r\n");
return Encoding.(());
}
#endregion
#region 处理接收的数据
/// <summary>
/// 处理接收的数据
/// 参考 /smark/archive/2012/11/26/
/// </summary>
/// <param name="recBytes"></param>
/// <param name="length"></param>
/// <returns></returns>
private string AnalyzeClientData(byte[] recBytes, int length)
{
int start = 0;
// 如果有数据则至少包括3位
if (length < 2) return "";
// 判断是否为结束针
bool IsEof = (recBytes[start] >> 7) > 0;
// 暂不处理超过一帧的数据
if (!IsEof) return "";
start++;
// 是否包含掩码
bool hasMask = (recBytes[start] >> 7) > 0;
// 不包含掩码的暂不处理
if (!hasMask) return "";
// 获取数据长度
UInt64 mPackageLength = (UInt64)recBytes[start] & 0x7F;
start++;
// 存储4位掩码值
byte[] Masking_key = new byte[4];
// 存储数据
byte[] mDataPackage;
if (mPackageLength == 126)
{
// 等于126 随后的两个字节16位表示数据长度
mPackageLength = (UInt64)(recBytes[start] << 8 | recBytes[start + 1]);
start += 2;
}
if (mPackageLength == 127)
{
// 等于127 随后的八个字节64位表示数据长度
mPackageLength = (UInt64)(recBytes[start] << (8 * 7) | recBytes[start] << (8 * 6) | recBytes[start] << (8 * 5) | recBytes[start] << (8 * 4) | recBytes[start] << (8 * 3) | recBytes[start] << (8 * 2) | recBytes[start] << 8 | recBytes[start + 1]);
start += 8;
}
mDataPackage = new byte[mPackageLength];
for (UInt64 i = 0; i < mPackageLength; i++)
{
mDataPackage[i] = recBytes[i + (UInt64)start + 4];
}
(recBytes, start, Masking_key, 0, 4);
for (UInt64 i = 0; i < mPackageLength; i++)
{
mDataPackage[i] = (byte)(mDataPackage[i] ^ Masking_key[i % 4]);
}
return Encoding.(mDataPackage);
}
#endregion
#region 发送数据
/// <summary>
/// 把发送给客户端消息打包处理（拼接上谁什么时候发的什么消息）
/// </summary>
/// <returns>The data.</returns>
/// <param name="message">Message.</param>
private byte[] PackageServerData(string msg)
{
byte[] content = null;
byte[] temp = Encoding.(msg);
if ( < 126)
{
content = new byte[ + 2];
content[0] = 0x81;
content[1] = (byte);
(temp, 0, content, 2, );
}
else if ( < 0xFFFF)
{
content = new byte[ + 4];
content[0] = 0x81;
content[1] = 126;
content[2] = (byte)( & 0xFF);
content[3] = (byte)( >> 8 & 0xFF);
(temp, 0, content, 4, );
}
return content;
}
#endregion
}
}