Socket服務器-Websocket支持多端連入

socket服務端：

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Security.Cryptography;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading;
using System.Threading.Tasks;
namespace ConsoleApplication1
{
    class Program
    {
        static List<Socket> Sockets = new List<Socket>();
        //static Socket listener = null;
        //static Dictionary<string, Socket> ClientConnectionItems = new Dictionary<string, Socket> { };
        static void Main(string[] args)
        {
            int port = 10;
            byte[] buffer = new byte[1024];

            IPEndPoint localEP = new IPEndPoint(IPAddress.Any, port);
            Socket listener = new Socket(localEP.Address.AddressFamily, SocketType.Stream, ProtocolType.Tcp);

            try
            {
                listener.Bind(localEP);
                listener.Listen(10);
                Console.WriteLine("等待客戶端連接....");
                while (true) //該操作用於多個客戶端連接
                {
                    Socket sc = listener.Accept();//接受一個連接
                    Sockets.Add(sc); //將連接的客戶端, 添加到內存當中
                    Thread t = new Thread(new ThreadStart(() => ReceiveData(sc))); //開啟當前Socket線程, 去執行獲取數據的動作,與客戶端通信
                    t.IsBackground = true;
                    t.Start();
                }
                //Thread threadwatch = new Thread(WatchConnecting);
                //threadwatch.IsBackground = true;
                ////啟動線程         
                //threadwatch.Start();
                //Console.WriteLine("開啟監聽......");
                //Console.ReadKey();
                //listener.Close();
            }
            catch (Exception e)
            {
                Console.WriteLine(e.ToString());
            }
            Console.ReadLine();
        }
        //public static void WatchConnecting()
        //{
        //    Socket connection = null;
        //    while (true)//持續不斷監聽客戶端發來的請求    
        //    {
        //        try
        //        {
        //            connection = listener.Accept();
        //        }
        //        catch (Exception) {
        //            Console.WriteLine();
        //            break;
        //        }
        //        //客戶端網絡結點號   
        //        string remoteEndPoint = connection.RemoteEndPoint.ToString();
        //        //添加客戶端信息             
        //        ClientConnectionItems.Add(remoteEndPoint, connection);
        //        //顯示與客戶端連接情況           
        //        Console.WriteLine("\r\n[客戶端\"" + remoteEndPoint + "\"建立連接成功！ 客戶端數量：" + ClientConnectionItems .Count+ "]");
        //        //獲取客戶端的IP和端口號             
        //        IPAddress clientIP = (connection.RemoteEndPoint as IPEndPoint).Address;
        //        int clientPort = (connection.RemoteEndPoint as IPEndPoint).Port;
        //        //讓客戶顯示"連接成功的"的信息             
        //        string sendmsg = "[" + "本地IP：" + clientIP + " 本地端口：" + clientPort.ToString() + " 連接服務端成功！]";
        //        byte[] arrSendMsg = Encoding.UTF8.GetBytes(sendmsg);
        //        connection.Send(arrSendMsg);
        //        //創建一個通信線程                 
        //        Thread thread = new Thread(recv);
        //        //設置為后台線程，隨着主線程退出而退出            
        //        thread.IsBackground = true;
        //        //啟動線程               
        //        thread.Start(connection);

        //    }
        //}
      
   
    /// <summary> 
    /// 接收客戶端發來的信息，客戶端套接字對象
    /// </summary>   
    /// <param name="socketclientpara"></param>
    //static void recv(object socketclientpara)
    //{
    //    Socket socketServer = socketclientpara as Socket;
    //    while (true)
    //    {
    //        //創建一個內存緩沖區，其大小為1024*1024字節  即1M
    //        byte[] arrServerRecMsg = new byte[1024 * 1024];
    //        //將接收到的信息存入到內存緩沖區，並返回其字節數組的長度 
    //        try
    //        {
    //            int length = socketServer.Receive(arrServerRecMsg);

    //            //將機器接受到的字節數組轉換為人可以讀懂的字符串     
    //            string strSRecMsg = Encoding.UTF8.GetString(arrServerRecMsg, 0, length);
    //            //將發送的字符串信息附加到文本框txtMsg上              
    //            Console.WriteLine("\r\n[客戶端：" + socketServer.RemoteEndPoint + " 時間：" + DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss:fff") + "]\r\n" + strSRecMsg);
    //                //

    //                //Thread.Sleep(1000);
    //                //自定義聊天內容
    //                //Console.WriteLine("輸入消息:\r\n");
    //                //string s = Console.ReadLine();
    //                //單個客戶端
    //                //socketServer.Send(Encoding.UTF8.GetBytes("[" + socketServer.RemoteEndPoint + "]：" + strSRecMsg + "\r\n"));
    //                //發送客戶端數據           
    //                if (ClientConnectionItems.Count > 0)
    //            {
    //                foreach (var socketTemp in ClientConnectionItems)
    //                {
    //                    socketTemp.Value.Send(Encoding.UTF8.GetBytes("[" + socketServer.RemoteEndPoint + "]：" + strSRecMsg));
    //                }
    //            }
    //        }
    //        catch (Exception)
    //        {
    //            ClientConnectionItems.Remove(socketServer.RemoteEndPoint.ToString());
    //            //提示套接字監聽異常     

    //            Console.WriteLine("\r\n[客戶端\"" + socketServer.RemoteEndPoint + "\"已經中斷連接！ 客戶端數量：" + ClientConnectionItems.Count + "]");
    //            //關閉之前accept出來的和客戶端進行通信的套接字          
    //            socketServer.Close();
    //            break;
    //        }
    //    }
    //}

    public static void ReceiveData(Socket sc)
        {
            byte[] buffer = new byte[1024];
            Console.WriteLine("接受到了客戶端：" + sc.RemoteEndPoint.ToString() + "連接....");
            //握手
            int length = sc.Receive(buffer);//接受客戶端握手信息
            sc.Send(PackHandShakeData(GetSecKeyAccetp(buffer, length))); while (true)
            {
                try
                {
                    //接受客戶端數據
                    Console.WriteLine("等待客戶端數據....");//運行到這步會停止，然后等客戶端消息
                    length = sc.Receive(buffer);//接受客戶端信息
                    string clientMsg = AnalyticData(buffer, length);
                    Console.WriteLine("接受到客戶端數據：" + clientMsg);
                    //發送數據
                    string sendMsg = "服務端返回信息:" + clientMsg;
                    sc.Send(PackData(sendMsg));
                }
                catch (Exception ex)
                {
                    Sockets.Remove(sc);  //如果接收的過程中,斷開, 那么內存中移除當前Socket對象, 並且退出當前線程
                    Console.WriteLine("客戶端已經斷開連接!");
                    return;
                }
            }

        }
        /// <summary>
        /// 打包握手信息
        /// </summary>
        /// <param name="secKeyAccept"></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);
            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;
        }
    }
}

websocket客戶端：

<%@ Page Language="C#" AutoEventWireup="true" CodeBehind="Test.aspx.cs" Inherits="SmartLockerWS.Test" %>

<%--<!DOCTYPE html>

<html xmlns="http://www.w3.org/1999/xhtml">
<head runat="server">
    <title></title>
</head>
<body>
    <form id="form1" runat="server">
    <div>
            <asp:Button ID="btnTest" runat="server" Text="Test1" OnClick="btnTest_Click" /><asp:Button ID="Button2" runat="server" Text="Test2" OnClick="btnTest2_Click" /><asp:Label ID="lbl1" runat="server" />
            <asp:GridView ID="gv1" runat="server" />
    </div>
    </form>
</body>
</html>--%>
<!DOCTYPE html>

<html lang="en" xmlns="http://www.w3.org/1999/xhtml">
<head>
    <meta charset="utf-8" />
    <title></title>
    <script>
var webSocket;
function connect()
{
    try
    {
        var readyState = new Array("正在連接", "已建立連接", "正在關閉連接", "已關閉連接");
        var host = "ws://localhost:10";
        webSocket = new WebSocket(host);
        var message = document.getElementById("message");
        message.innerHTML += "<p>Socket狀態：" + readyState[webSocket.readyState] + "</p>";
        webSocket.onopen = function () {
            message.innerHTML += "<p>Socket狀態：" + readyState[webSocket.readyState] + "</p>";
        }
        webSocket.onmessage = function (msg) {
            message.innerHTML += "<p>接收信息：" + msg.data + "</p>";
        }
        webSocket.onclose = function () {
            message.innerHTML += "<p>Socket狀態：" + readyState[webSocket.readyState] + "</p>";
        }

    }
    catch(exception)
    {
        message.innerHTML += "<p>有錯誤發生</p>";
    }
}
function send()
{
    var text = document.getElementById("text").value;
    var message = document.getElementById("message");
    if (text == "") {
        message.innerHTML += "<p>請輸入一些文字</p>";
        return;
    }
    try {
        webSocket.send(text);
        message.innerHTML += "<p>發送數據：" + text + "</p>";
    }
    catch (exception) {
        message.innerHTML += "<p>發送數據出錯</p>";
    }
    document.getElementById("text").value = "";

}
function disconnect()
{
    webSocket.close();
}
    </script>

</head>

<body>
    <%--<h1>WebSocket客戶端示例</h1>--%>
    <div id="message"></div>
    <p>請輸入一些文字</p>
    <input id="text" type="text">
    <button id="connect" onClick="connect();">建立連接</button>
    <button id="send" onClick="send();">發送數據</button>
    <button id="disconnect" onClick="disconnect();">斷開連接</button>

</body>
</html>

或者socket客戶端：

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace ConsoleApplication2
{
    class Program
    {
        //創建 1個客戶端套接字 和1個負責監聽服務端請求的線程 
        static Thread threadclient = null;
        static Socket socketclient = null;

        static void Main(string[] args)
        {
            //定義一個套接字監聽 
            socketclient = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);

            //獲取文本框中的IP地址 
            IPAddress address = IPAddress.Parse("127.0.0.1");

            //將獲取的IP地址和端口號綁定在網絡節點上 
            IPEndPoint point = new IPEndPoint(address, 10);

            try
            {
                //客戶端套接字連接到網絡節點上，用的是Connect 
                socketclient.Connect(point);
            }
            catch (Exception r)
            {
                return;
            }

            threadclient = new Thread(recv);
            threadclient.IsBackground = true;
            threadclient.Start();
            Thread.Sleep(1000);
            Console.WriteLine(socketclient.LocalEndPoint+":\r");
            while (true) {
                string s = Console.ReadLine();
                ClientSendMsg(s);
            }
        }

        // 接收服務端發來信息的方法   
       static void recv()
        {
            int x = 0;
            //持續監聽服務端發來的消息
            while (true)
            {
                try
                {
                    //定義一個1M的內存緩沖區，用於臨時性存儲接收到的消息 
                    byte[] arrRecvmsg = new byte[1024 * 1024];

                    //將客戶端套接字接收到的數據存入內存緩沖區，並獲取長度 
                    int length = socketclient.Receive(arrRecvmsg);

                    //將套接字獲取到的字符數組轉換為人可以看懂的字符串 
                    string strRevMsg = Encoding.UTF8.GetString(arrRecvmsg, 0, length);
                    if (x == 1)
                    {
                        Console.WriteLine(strRevMsg);
                        }
                    else
                    {
                        Console.WriteLine(strRevMsg + "\r\n");
                        x = 1;
                    }
                }
                catch (Exception ex)
                {
                    Console.WriteLine("遠程服務器已經中斷連接！" + ex.Message + "\r\n");
                    break;
                }
            }
        }

        //獲取當前系統時間 
        DateTime GetCurrentTime()
        {
            DateTime currentTime = new DateTime();
            currentTime = DateTime.Now;
            return currentTime;
        }

        //發送字符信息到服務端的方法 
       static void ClientSendMsg(string sendMsg)
        {
            //將輸入的內容字符串轉換為機器可以識別的字節數組    
            byte[] arrClientSendMsg = Encoding.UTF8.GetBytes(sendMsg);
            //調用客戶端套接字發送字節數組    
            socketclient.Send(arrClientSendMsg);
            //將發送的信息追加到聊天內容文本框中    
        } 
    }
}