socket下server端支持多客戶端並發訪問簡單實現

/*

*Author: 　wainiwann

*Source：　博客園 http://www.cnblogs.com/wainiwann

*Remarks:  轉載請說明出處！！！

*/

server端開啟之后始終有兩個線程在處理連接請求，一個是只負責客戶端的請求連接的（這里是只針對TCP協議），當客戶端connect的時候記錄當前客戶端連接存放到數據組中當中，而這個數組聲明為全局成員，其實在線程內處理外部成員的話，也沒必要非要用靜態或者全局成員，今天聽經理說也可以在創建該線程時，把某類的this指針傳遞過去，同樣好像也可以訪問public成員的，具體行不行，還沒試不過真的是不錯的方法。要知道很多在項目很避諱使用全局的東西，甚至有的公司直接不讓使用全局的東西。這里扯的有點遠了。

另外一個同步允許的線程就是對accept記錄的數組進行操作，依次處理各個客戶端請求通信和狀態監控等，當數組內socket成員不為空時記錄當前索引然后在create發送或者獲取線程處理函數並發響應多個客戶端的連接請求通信。

加載套接字庫：

BOOL TcpServer::InitSocket()
{
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
    wVersionRequested = MAKEWORD( 1, 1 );
    err = WSAStartup( wVersionRequested, &wsaData );
    if ( err != 0 ) 
    {
        return FALSE;
    }

    if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) 
    {
        WSACleanup( );
        return FALSE;
    }

    //創建套接字
    //SOCKET m_socket=socket(AF_INET,SOCK_STREAM,0);

    
    return TRUE;
}

開啟執行Accept線程處理函數：

BOOL TcpServer::SatartServer()
{
    //創建線程
    HANDLE hThread = CreateThread(NULL,0,ThreadProc_Accept,NULL,0,NULL);
    //關閉該接收線程句柄，釋放引用計數
    CloseHandle(hThread);

    return TRUE;
}

線程處理函數：

DWORD WINAPI TcpServer::ThreadProc_Accept(LPVOID lpParameter)
{
    int len = sizeof(SOCKADDR);
    int err;
    m_socket=socket(AF_INET,SOCK_STREAM,0);
    if (m_socket == INVALID_SOCKET)
    {
        AfxMessageBox(_T("套接字創建失敗！"));
        return FALSE;
    }

    SOCKADDR_IN addrSrv;
    addrSrv.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
    addrSrv.sin_family=AF_INET;
    addrSrv.sin_port=htons(8099);

    err = bind(m_socket,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));    //綁定本地端口
    if (err==SOCKET_ERROR)
    {
        closesocket(m_socket);
        AfxMessageBox(_T("綁定失敗！"));
        return FALSE;
    }
    listen(m_socket,10);//開啟監聽

    //創建線程
    HANDLE hThread = CreateThread(NULL,0,ThreadProc_Select,NULL,0,NULL);
    //關閉該接收線程句柄，釋放引用計數
    CloseHandle(hThread);

    while (TRUE)
    {
        m_CliSocketArr[m_ToolConn++] = accept(m_socket,(SOCKADDR*)&addrSrv,&len);
    }
    return 0;
}

同時在該線程函數內創建處理客戶端數組的線程處理函數：

DWORD WINAPI TcpServer::ThreadProc_Select(LPVOID lpParameter)
{
    int recvflag=0;

    fd_set    fdread;                //讀集fdread
    int        ret;                //查看某個套接字的狀態
    struct timeval tv = {1,0};    //實例化timeval變量

    while (TRUE)
    {
        //判斷當前連接數是否為 0
        if (m_ToolConn == 0)
        {
            Sleep(50);
            continue;
        }

        FD_ZERO(&fdread);
        for (int i = 0;i < m_ToolConn;i++)
        {
            FD_SET(m_CliSocketArr[i],&fdread);
        }
        ret = select(0,&fdread,NULL,NULL,&tv);
        if (ret == 0)
        {
            continue;
        }
        for (int i =0;i<m_ToolConn;i++)
        {
            if (FD_ISSET(m_CliSocketArr[i],&fdread))
            {
                ret = recv(m_CliSocketArr[i],(char*)&recvflag,sizeof(int)+1,0);
                if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
                {
                    closesocket(m_CliSocketArr[i]);
                    if (i < m_ToolConn-1)
                    {
                        m_CliSocketArr[i] = m_CliSocketArr[--m_ToolConn];
                    }else
                    {
                        --m_ToolConn;
                    }

                }else
                {
                    INDEX * inx = new INDEX;
                    inx->flag  = recvflag;
                    inx->index = i;

                    //創建線程
                    HANDLE hThread = CreateThread(NULL,0,ThreadProc_Response,(LPVOID)inx,0,NULL);
                    //關閉該接收線程句柄，釋放引用計數
                    CloseHandle(hThread);

                }
            }//if
        }//for
    }//while

    return 0;
}

下面就是一次創建線程並發處理客戶端請求線程處理函數：

DWORD WINAPI TcpServer::ThreadProc_Response(LPVOID lpParameter)
{
    int ix = ((INDEX*)lpParameter)->index;
    int flag = ((INDEX*)lpParameter)->flag;

    delete lpParameter;

    if (flag == 1)
    {
        //.............................
        unsigned char sendBuffer[1024] = {'a'};
        send(m_CliSocketArr[ix],(char*)sendBuffer,sizeof(sendBuffer)+1,0);
    }

    return 0;
}

這里線程的創建以及處理也可以使用_beginthread（）去實現。

另外里面沒有做太多的資源釋放等等，如果真正去用的話，是應該注意的。