前面的程式,不管伺服器端還是用戶端,都有一個問題,就是處理完一個請求立即退出了,沒有太大的實際意義。能不能像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,下次發起請求時需要重新建立。後續我們會進行詳細講解。