快捷導(dǎo)航

c語言socket多線程編程限制客戶端連接數(shù)

更新時(shí)間：2013年12月05日 10:25:48 作者：

這篇文章主要介紹了c語言socket多線程編程,可以限制客戶端連接數(shù)，大家參考使用吧

先上一些多線程需要使用的函數(shù)定義：

DWORD WINAPI ProcessClientRequests(LPVOID lpParam) //新線程將會(huì)執(zhí)行的函數(shù)定義
{
return 0;
}

HANDLE handler=CreateThread(NULL, 0, ProcessClientRequests, &clientsocket, 0, NULL); //這里比較簡單，&clientsocket是個(gè)指針，是從主線程傳入新線程的參數(shù)

WaitForMultipleObjects(MAXCLIENTS, threads, TRUE, INFINITE); //用來阻塞主線程，直到所有創(chuàng)建的子線程都完成任務(wù)為止，才繼續(xù)執(zhí)行后面的代碼

for(int i=0;i<MAXCLIENTS; i++)
{
CloseHandle(threads[i]); //創(chuàng)建的每個(gè)子線程的HANDLE都會(huì)被保存在HANDLE數(shù)組中，這個(gè)函數(shù)用于關(guān)閉各個(gè)handle所對應(yīng)的線程空間
}

服務(wù)器端程序

主線程代碼如下：

復(fù)制代碼代碼如下:

#define MAXCLIENTS 3 //宏定義，最多3個(gè)客戶端連接

int main()
{
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2, 2), &wsaData);
    HANDLE threads[MAXCLIENTS];

SOCKET s=socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);

    sockaddr_in sockaddr;
    sockaddr.sin_family=PF_INET;
    sockaddr.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
    sockaddr.sin_port=htons(9000);
    bind(s, (SOCKADDR*)&sockaddr, sizeof(SOCKADDR));

listen(s, 1);

printf("listening on port [%d].\n", 9000);

    int existingClientCount=0;
    while(TRUE)
    {
        SOCKADDR clientAddr;
        int size=sizeof(SOCKADDR);

        SOCKET clientsocket;
        clientsocket=accept(s, &clientAddr, &size);
        printf("***SYS***    New client touched.\n");

        if(existingClientCount<MAXCLIENTS)       //判斷是否已經(jīng)超出最大連接數(shù)了
        {
            threads[existingClientCount++]=CreateThread(NULL, 0, ProcessClientRequests, &clientsocket, 0, NULL); //啟動(dòng)新線程，并且將socket傳入
        }
        else
        {
            char* msg="Exceeded Max incoming requests, will refused this connect!\r\n";
            send(clientsocket, msg, strlen(msg)+sizeof(char), NULL);       //發(fā)送拒絕連接消息給客戶端
            printf("***SYS***    REFUSED.\n");
            closesocket(clientsocket);                                     //釋放資源
            break;
        }
    }

printf("Maximize clients occurred for d%.\r\n", MAXCLIENTS);

WaitForMultipleObjects(MAXCLIENTS, threads, TRUE, INFINITE); //等待所有子線程，直到完成為止

    closesocket(s);
    for(int i=0;i<MAXCLIENTS; i++)
    {
        CloseHandle(threads[i]);                                           //清理線程資源
    }

    WSACleanup();

printf("Cleared all.\r\n");

getchar();

exit(0);
}

子線程函數(shù)定義

復(fù)制代碼代碼如下:

DWORD WINAPI ProcessClientRequests(LPVOID lpParam)
{
SOCKET* clientsocket=(SOCKET*)lpParam; //這里需要強(qiáng)制轉(zhuǎn)換，注意：指針類型的

    char* msg="Hello, my client.\r\n";
    send(*clientsocket, msg, strlen(msg)+sizeof(char), NULL);
    printf("***SYS***    HELLO.\n");

    while(TRUE)
    {
        char buffer[MAXBYTE]={0};
        recv(*clientsocket, buffer, MAXBYTE, NULL);
        if(strcmp(buffer, "exit")==0)
        {
            char* msg_bye="Bye.\r\n";
            send(*clientsocket, msg_bye, strlen(msg_bye)+sizeof(char), NULL);
            break;
        }
        printf("***Client***    %s\n", buffer);
    }

    closesocket(*clientsocket);

return 0;
}