如何讓伺服器端持續監聽用戶端的請求?

2020-07-16 10:04:43
前面的程式,不管伺服器端還是用戶端,都有一個問題,就是處理完一個請求立即退出了,沒有太大的實際意義。能不能像Web伺服器那樣一直接受用戶端的請求呢?能,使用 while 迴圈即可。

修改前面的回聲程式,使伺服器端可以不斷響應用戶端的請求。

伺服器端 server.cpp:
#include <stdio.h>
#include <winsock2.h>
#pragma comment (lib, "ws2_32.lib")  //載入 ws2_32.dll

#define BUF_SIZE 100

int main(){
    WSADATA wsaData;
    WSAStartup( MAKEWORD(2, 2), &wsaData);

    //建立通訊端
    SOCKET servSock = socket(AF_INET, SOCK_STREAM, 0);

    //係結通訊端
    sockaddr_in sockAddr;
    memset(&sockAddr, 0, sizeof(sockAddr));  //每個位元組都用0填充
    sockAddr.sin_family = PF_INET;  //使用IPv4地址
    sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");  //具體的IP地址
    sockAddr.sin_port = htons(1234);  //埠
    bind(servSock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));

    //進入監聽狀態
    listen(servSock, 20);

    //接收用戶端請求
    SOCKADDR clntAddr;
    int nSize = sizeof(SOCKADDR);
    char buffer[BUF_SIZE] = {0};  //緩衝區
    while(1){
        SOCKET clntSock = accept(servSock, (SOCKADDR*)&clntAddr, &nSize);
        int strLen = recv(clntSock, buffer, BUF_SIZE, 0);  //接收用戶端發來的資料
        send(clntSock, buffer, strLen, 0);  //將資料原樣返回

        closesocket(clntSock);  //關閉通訊端
        memset(buffer, 0, BUF_SIZE);  //重置緩衝區
    }

    //關閉通訊端
    closesocket(servSock);

    //終止 DLL 的使用
    WSACleanup();

    return 0;
}

用戶端 client.cpp:
#include <stdio.h>
#include <WinSock2.h>
#include <windows.h>
#pragma comment(lib, "ws2_32.lib")  //載入 ws2_32.dll

#define BUF_SIZE 100

int main(){
    //初始化DLL
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2, 2), &wsaData);

    //向伺服器發起請求
    sockaddr_in sockAddr;
    memset(&sockAddr, 0, sizeof(sockAddr));  //每個位元組都用0填充
    sockAddr.sin_family = PF_INET;
    sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    sockAddr.sin_port = htons(1234);
   
    char bufSend[BUF_SIZE] = {0};
    char bufRecv[BUF_SIZE] = {0};

    while(1){
        //建立通訊端
        SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
        connect(sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
        //獲取使用者輸入的字串並行送給伺服器
        printf("Input a string: ");
        gets(bufSend);
        send(sock, bufSend, strlen(bufSend), 0);
        //接收伺服器傳回的資料
        recv(sock, bufRecv, BUF_SIZE, 0);
        //輸出接收到的資料
        printf("Message form server: %sn", bufRecv);
       
        memset(bufSend, 0, BUF_SIZE);  //重置緩衝區
        memset(bufRecv, 0, BUF_SIZE);  //重置緩衝區
        closesocket(sock);  //關閉通訊端
    }

    WSACleanup();  //終止使用 DLL
    return 0;
}
先執行伺服器端,再執行用戶端,結果如下:
Input a string: c language
Message form server: c language
Input a string: C語言中文網
Message form server: C語言中文網
Input a string: 學習C/C++程式設計的好網站
Message form server: 學習C/C++程式設計的好網站

while(1) 讓程式碼進入死迴圈,除非使用者關閉程式,否則伺服器端會一直監聽用戶端的請求。用戶端也是一樣,會不斷向伺服器發起連線。

需要注意的是:server.cpp 中呼叫 closesocket() 不僅會關閉伺服器端的 socket,還會通知用戶端連線已斷開,用戶端也會清理 socket 相關資源,所以 client.cpp 中需要將 socket() 放在 while 迴圈內部,因為每次請求完畢都會清理 socket,下次發起請求時需要重新建立。後續我們會進行詳細講解。