C#基于WebSocket實(shí)現(xiàn)聊天室功能

第C#基于WebSocket實(shí)現(xiàn)聊天室功能本文實(shí)例為大家分享了C#基于WebSocket實(shí)現(xiàn)聊天室功能的具體代碼，供大家參考，具體內(nèi)容如下

前面兩篇溫習(xí)了，C#Socket內(nèi)容

本章根據(jù)Socket異步聊天室修改成WebSocket聊天室

WebSocket特別的地方是握手和消息內(nèi)容的編碼、解碼（添加了ServerHelper協(xié)助處理）

ServerHelper：

usingSystem;

usingSystem.Collections;

usingSystem.Text;

usingSystem.Security.Cryptography;

namespaceSocketDemo

//Server助手負(fù)責(zé)：1握手2請(qǐng)求轉(zhuǎn)換3響應(yīng)轉(zhuǎn)換

classServerHelper

{

///summary

///輸出連接頭信息

////summary

publicstaticstringResponseHeader(stringrequestHeader)

{

Hashtabletable=newHashtable();

//拆分成鍵值對(duì)，保存到哈希表

string[]rows=requestHeader.Split(newstring[]{"\r\n"},StringSplitOptions.RemoveEmptyEntries);

foreach(stringrowinrows)

{

intsplitIndex=row.IndexOf(':');

if(splitIndex0)

{

table.Add(row.Substring(0,splitIndex).Trim(),row.Substring(splitIndex+1).Trim());

}

}

StringBuilderheader=newStringBuilder();

header.Append("HTTP/1.1101WebSocketProtocolHandshake\r\n");

header.AppendFormat("Upgrade:{0}\r\n",table.ContainsKey("Upgrade")table["Upgrade"].ToString():string.Empty);

header.AppendFormat("Connection:{0}\r\n",table.ContainsKey("Connection")table["Connection"].ToString():string.Empty);

header.AppendFormat("WebSocket-Origin:{0}\r\n",table.ContainsKey("Sec-WebSocket-Origin")table["Sec-WebSocket-Origin"].ToString():string.Empty);

header.AppendFormat("WebSocket-Location:{0}\r\n",table.ContainsKey("Host")table["Host"].ToString():string.Empty);

stringkey=table.ContainsKey("Sec-WebSocket-Key")table["Sec-WebSocket-Key"].ToString():string.Empty;

stringmagic="258EAFA5-E914-47DA-95CA-C5AB0DC85B11";

header.AppendFormat("Sec-WebSocket-Accept:{0}\r\n",Convert.ToBase64String(SHA1.Create().ComputeHash(Encoding.ASCII.GetBytes(key+magic))));

header.Append("\r\n");

returnheader.ToString();

}

///summary

///解碼請(qǐng)求內(nèi)容

////summary

publicstaticstringDecodeMsg(Byte[]buffer,intlen)

{

if(buffer[0]!=0x81

||(buffer[0]0x80)!=0x80

||(buffer[1]0x80)!=0x80)

{

returnnull;

}

Byte[]mask=newByte[4];

intbeginIndex=0;

intpayload_len=buffer[1]0x7F;

if(payload_len==0x7E)

{

Array.Copy(buffer,4,mask,0,4);

payload_len=payload_len0x00000000;

payload_len=payload_len|buffer[2];

payload_len=(payload_len8)|buffer[3];

beginIndex=8;

}

elseif(payload_len!=0x7F)

{

Array.Copy(buffer,2,mask,0,4);

beginIndex=6;

}

for(inti=0;ipayload_len;i++)

{

buffer[i+beginIndex]=(byte)(buffer[i+beginIndex]^mask[i%4]);

}

returnEncoding.UTF8.GetString(buffer,beginIndex,payload_len);

}

///summary

///編碼響應(yīng)內(nèi)容

////summary

publicstaticbyte[]EncodeMsg(stringcontent)

{

byte[]bts=null;

byte[]temp=Encoding.UTF8.GetBytes(content);

if(temp.Length126)

{

bts=newbyte[temp.Length+2];

bts[0]=0x81;

bts[1]=(byte)temp.Length;

Array.Copy(temp,0,bts,2,temp.Length);

}

elseif(temp.Length0xFFFF)

{

bts=newbyte[temp.Length+4];

bts[0]=0x81;

bts[1]=126;

bts[2]=(byte)(temp.Length0xFF);

bts[3]=(byte)(temp.Length80xFF);

Array.Copy(temp,0,bts,4,temp.Length);

}

else

{

byte[]st=System.Text.Encoding.UTF8.GetBytes(string.Format("暫不處理超長(zhǎng)內(nèi)容").ToCharArray());

}

returnbts;

}

}

}

Server：

usingSystem;

usingSystem.Collections.Generic;

usingSystem.Text;

usingSystem.Net;

usingSystem.Net.Sockets;

namespaceSocketDemo

classClientInfo

{

publicSocketSocket{get;set;}

publicboolIsOpen{get;set;}

publicstringAddress{get;set;}

}

//管理Client

classClientManager

{

staticListClientInfoclientList=newListClientInfo

publicstaticvoidAdd(ClientInfoinfo)

{

if(!IsExist(info.Address))

{

clientList.Add(info);

}

}

publicstaticboolIsExist(stringaddress)

{

returnclientList.Exists(item=string.Compare(address,item.Address,true)==0);

}

publicstaticboolIsExist(stringaddress,boolisOpen)

{

returnclientList.Exists(item=string.Compare(address,item.Address,true)==0item.IsOpen==isOpen);

}

publicstaticvoidOpen(stringaddress)

{

clientList.ForEach(item=

{

if(string.Compare(address,item.Address,true)==0)

{

item.IsOpen=true;

}

});

}

publicstaticvoidClose(stringaddress=null)

{

clientList.ForEach(item=

{

if(address==null||string.Compare(address,item.Address,true)==0)

{

item.IsOpen=false;

item.Socket.Shutdown(SocketShutdown.Both);

}

});

}

//發(fā)送消息到ClientList

publicstaticvoidSendMsgToClientList(stringmsg,stringaddress=null)

{

clientList.ForEach(item=

{

if(item.IsOpen(address==null||item.Address!=address))

{

SendMsgToClient(item.Socket,msg);

}

});

}

publicstaticvoidSendMsgToClient(Socketclient,stringmsg)

{

byte[]bt=ServerHelper.EncodeMsg(msg);

client.BeginSend(bt,0,bt.Length,SocketFlags.None,newAsyncCallback(SendTarget),client);

}

privatestaticvoidSendTarget(IAsyncResultres)

{

//Socketclient=(Socket)res.AsyncState;

//intsize=client.EndSend(res);

}

}

//接收消息

classReceiveHelper

{

publicbyte[]Bytes{get;set;}

publicvoidReceiveTarget(IAsyncResultres)

{

Socketclient=(Socket)res.AsyncState;

intsize=client.EndReceive(res);

if(size0)

{

stringaddress=client.RemoteEndPoint.ToString();//獲取Client的IP和端口

stringstringdata=null;

if(ClientManager.IsExist(address,false))//握手

{

stringdata=Encoding.UTF8.GetString(Bytes,0,size);

ClientManager.SendMsgToClient(client,ServerHelper.ResponseHeader(stringdata));

ClientManager.Open(address);

}

else

{

stringdata=ServerHelper.DecodeMsg(Bytes,size);

}

if(stringdata.IndexOf("exit")-1)

{

ClientManager.SendMsgToClientList(address+"已從服務(wù)器斷開(kāi)",address);

ClientManager.Close(address);

Console.WriteLine(address+"已從服務(wù)器斷開(kāi)");

Console.WriteLine(address+""+DateTimeOffset.Now.ToString("G"));

return;

}

else

{

Console.WriteLine(stringdata);

Console.WriteLine(address+""+DateTimeOffset.Now.ToString("G"));

ClientManager.SendMsgToClientList(stringdata,address);

}

}

//繼續(xù)等待

client.BeginReceive(Bytes,0,Bytes.Length,SocketFlags.None,newAsyncCallback(ReceiveTarget),client);

}

}

//監(jiān)聽(tīng)請(qǐng)求

classAcceptHelper

{

publicbyte[]Bytes{get;set;}

publicvoidAcceptTarget(IAsyncResultres)

{

Socketserver=(Socket)res.AsyncState;

Socketclient=server.EndAccept(res);

stringaddress=client.RemoteEndPoint.ToString();

ClientManager.Add(newClientInfo(){Socket=client,Address=address,IsOpen=false});

ReceiveHelperrs=newReceiveHelper(){Bytes=this.Bytes};

IAsyncResultrecres=client.BeginReceive(rs.Bytes,0,rs.Bytes.Length,SocketFlags.None,newAsyncCallback(rs.ReceiveTarget),client);

//繼續(xù)監(jiān)聽(tīng)

server.BeginAccept(newAsyncCallback(AcceptTarget),server);

}

}

classProgram

{

staticvoidMain(string[]args)

{

Socketserver=newSocket(AddressFamily.InterNetwork,SocketType.Stream,ProtocolType.Tcp);

server.Bind(newIPEndPoint(IPAddress.Parse(""),200));//綁定IP+端口

server.Listen(10);//開(kāi)始監(jiān)聽(tīng)

Console.WriteLine("等待連接...");

AcceptHelperca=newAcceptHelper(){Bytes=newbyte[2048]};

IAsyncResultres=server.BeginAccept(newAsyncCallback(ca.AcceptTarget),server);

stringstr=string.Empty;

while(str!="exit")

{

str=Console.ReadLine();

Console.WriteLine("ME:"+DateTimeOffset.Now.ToString("G"));

ClientManager.SendMsgToClientList(str);

}

ClientManager.Close();

server.Close();

}