對於網路通信中的服務端來說,顯然不可能是一對一的,我們所希望的是服務端啟用一份則可以選擇性的與特定一個客戶端通信,而當不需要與客戶端通信時,則只需要將該套接字掛到鏈表中存儲並等待後續操作,套接字服務端通過多線程實現存儲套接字和選擇通信,可以提高服務端的併發性能,使其能夠同時處理多個客戶端的請求。在實... ...
對於網路通信中的服務端來說,顯然不可能是一對一的,我們所希望的是服務端啟用一份則可以選擇性的與特定一個客戶端通信,而當不需要與客戶端通信時,則只需要將該套接字掛到鏈表中存儲並等待後續操作,套接字服務端通過多線程實現存儲套接字和選擇通信,可以提高服務端的併發性能,使其能夠同時處理多個客戶端的請求。在實際應用場景中,這種技術被廣泛應用於網路編程、互聯網應用等領域。
該功能的具體實現思路可以總結為如下流程;
在服務端啟動時,創建套接字併進行綁定,然後開啟一個線程(稱為主線程)用於監聽客戶端的連接請求。主線程在接收到新的連接請求後,會將對應的套接字加入一個數據結構(例如鏈表、隊列、哈希表等)中進行存儲。同時,主線程會將存儲套接字的數據結構傳遞給每個子線程,並開啟多個子線程進行服務,每個子線程從存儲套接字的數據結構中取出套接字,然後通過套接字與客戶端進行通信。
在選擇通信方面,用戶可以指定要與哪個客戶端進行通信。服務端會在存儲套接字的數據結構中尋找符合條件的套接字,然後將通信數據發送給對應的客戶端。
首先為了能實現套接字的存儲功能,此處我們需要定義一個ClientInfo
該結構被定義的作用只有一個那就是存儲套接字的FD
句柄,以及該套接字的IP
地址與埠信息,這個結構體應該定義為如下樣子;
typedef struct
{
SOCKET client;
sockaddr_in saddr;
char address[128];
unsigned short port;
}ClientInfo;
接著我們來看主函數中的實現,首先主函數中listen
正常偵聽套接字連接情況,當有新的套接字接入後則直接通過CreateThread
函數開闢一個子線程,該子線程通過EstablishConnect
函數掛在後臺,在掛入後臺之前通過std::vector<ClientInfo *> info
全局變數用來保存套接字。
當讀者需要發送數據時,只需要調用SendMessageConnect
函數,函數接收一個套接字鏈表,並接收需要操作的IP
地址信息,以及需要發送的數據包,當有了這些信息後,函數內部會首先依次根據IP
地址判斷是否是我們所需要通信的IP
,如果是則從全局鏈表內取出套接字併發送數據包給特定的客戶端。
彈出一個套接字調用PopConnect
該函數接收一個全局鏈表,以及一個字元串IP
地址,其內部通過枚舉鏈表的方式尋找IP
地址,如果找到了則直接使用ptr.erase(it)
方法將找到的套接字彈出鏈表,並以此實現關閉通信的目的。
輸出套接字元素時,通過調用ShowList
函數實現,該函數內部首先通過迴圈枚舉所有的套接字並依次Ping
測試,如果發現存在掉線的套接字則直接剔除鏈表,如果沒有掉線則客戶端會反饋一個pong
以表示自己還在,此時即可直接輸出該套接字信息。
14.10.1 服務端實現
服務端的實現方式在上述概述中已經簡單介紹過了,服務端實現的原理概括起來就是,通過多線程技術等待客戶端上線,當有客戶端上線後就直接將其加入到全局鏈表內等待操作,主函數執行死迴圈,等待用戶輸入數據,用於選擇與某個套接字通信。
#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <string>
#include <vector>
#pragma comment(lib,"ws2_32.lib")
using namespace std;
typedef struct
{
SOCKET client;
sockaddr_in saddr;
char address[128];
unsigned short port;
}ClientInfo;
std::vector<ClientInfo *> info; // 全局主機列表
SOCKET server; // 本地套接字
sockaddr_in sai_server; // 存放伺服器IP、埠
// 彈出下線的主機
void PopConnect(std::vector<ClientInfo *> &ptr, char *address)
{
// 迴圈迭代器,查找需要彈出的元素
for (std::vector<ClientInfo *>::iterator it = ptr.begin(); it != ptr.end(); it++)
{
ClientInfo *client = *it;
// 如果找到了,則將其從鏈表中移除
if (strcmp(client->address, address) == 0)
{
ptr.erase(it);
// std::cout << "地址: " << client->address << " 已下線" << std::endl;
return;
}
}
}
// 輸出當前主機列表
void ShowList(std::vector<ClientInfo *> &ptr)
{
for (int x = 0; x < ptr.size(); x++)
{
// 發送Ping信號,探測
bool ref = send(ptr[x]->client, "Ping", 4, 0);
if (ref != true)
{
PopConnect(info, ptr[x]->address);
continue;
}
// 接收探測信號,看是否存活
char ref_buf[32] = { 0 };
recv(ptr[x]->client, ref_buf, 32, 0);
if (strcmp(ref_buf, "Pong") != 0)
{
PopConnect(info, ptr[x]->address);
continue;
}
std::cout << "主機: " << ptr[x]->address << " 埠: " << ptr[x]->port << std::endl;
}
}
// 發送消息
void SendMessageConnect(std::vector<ClientInfo *> &ptr, char *address, char *send_data)
{
for (int x = 0; x < ptr.size(); x++)
{
// std::cout << ptr[x]->address << std::endl;
// 判斷是否為需要發送的IP
if (strcmp(ptr[x]->address, address) == 0)
{
// 對選中主機發送數據
send(ptr[x]->client, send_data, strlen(send_data), 0);
int error_send = GetLastError();
if (error_send != 0)
{
// std::cout << ptr[x]->address << " 已離線" << endl;
// 彈出元素
PopConnect(info, address);
return;
}
// 獲取執行結果
char recv_message[4096] = { 0 };
recv(ptr[x]->client, recv_message, 4096, 0);
std::cout << recv_message << std::endl;
}
}
}
// 建立套接字
void EstablishConnect()
{
while (1)
{
ClientInfo* cInfo = new ClientInfo();
int len_client = sizeof(sockaddr);
cInfo->client = accept(server, (sockaddr*)&cInfo->saddr, &len_client);
// 填充主機地址和埠
char array_ip[20] = { 0 };
inet_ntop(AF_INET, &cInfo->saddr.sin_addr, array_ip, 16);
strcpy(cInfo->address, array_ip);
cInfo->port = ntohs(cInfo->saddr.sin_port);
info.push_back(cInfo);
}
}
int main(int argc, char* argv[])
{
// 初始化 WSA ,激活 socket
WSADATA wsaData;
WSAStartup(MAKEWORD(2, 2), &wsaData);
// 初始化 socket、伺服器信息
server = socket(AF_INET, SOCK_STREAM, 0);
sai_server.sin_addr.S_un.S_addr = 0; // IP地址
sai_server.sin_family = AF_INET; // IPV4
sai_server.sin_port = htons(8090); // 傳輸協議埠
// 本地地址關聯套接字
if (bind(server, (sockaddr*)&sai_server, sizeof(sai_server)))
{
WSACleanup();
}
// 套接字進入監聽狀態
listen(server, SOMAXCONN);
// 建立子線程實現偵聽連接
CreateThread(0, 0, (LPTHREAD_START_ROUTINE)EstablishConnect, 0, 0, 0);
while (1)
{
char command[4096] = { 0 };
input:
memset(command, 0, 4096);
std::cout << "[ LyShell ] # ";
// 發送命令
int inputLine = 0;
while ((command[inputLine++] = getchar()) != '\n');
if (strlen(command) == 1)
goto input;
// 輸出主機列表
if (strcmp(command, "list\n") == 0)
{
ShowList(info);
}
// 發送消息
else if (strcmp(command, "send\n") == 0)
{
SendMessageConnect(info, "127.0.0.1", "Send");
}
// 發送CPU數據
else if (strcmp(command, "GetCPU\n") == 0)
{
SendMessageConnect(info, "127.0.0.1", "GetCPU");
}
// 發送退出消息
else if (strcmp(command, "Exit\n") == 0)
{
SendMessageConnect(info, "127.0.0.1", "Exit");
}
}
return 0;
}
14.10.2 客戶端實現
客戶端的實現與之前文章中的實現方式是一樣的,由於客戶端無需使用多線程技術所以在如下代碼中我們只需要通過一個死迴圈每隔5000
毫秒調用connect
對服務端進行連接,如果沒有連接成功則繼續等待,如果連接成功了則直接進入內部死迴圈,在迴圈體內根據不同的命令執行不同的返回信息,如下是客戶端實現完整代碼片段。
#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <string>
#pragma comment(lib,"ws2_32.lib")
using namespace std;
int main(int argc, char* argv[])
{
while (1)
{
WSADATA WSAData;
SOCKET sock;
struct sockaddr_in ClientAddr;
if (WSAStartup(MAKEWORD(2, 0), &WSAData) != SOCKET_ERROR)
{
ClientAddr.sin_family = AF_INET;
ClientAddr.sin_port = htons(8090);
ClientAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
sock = socket(AF_INET, SOCK_STREAM, 0);
int Ret = connect(sock, (LPSOCKADDR)&ClientAddr, sizeof(ClientAddr));
if (Ret == 0)
{
while (1)
{
char buf[4096] = { 0 };
memset(buf, 0, sizeof(buf));
recv(sock, buf, 4096, 0);
// 獲取CPU數據
if (strcmp(buf, "GetCPU") == 0)
{
char* cpu_idea = "10%";
int ServerRet = send(sock, cpu_idea, sizeof("10%"), 0);
if (ServerRet != 0)
{
std::cout << "發送CPU數據包" << std::endl;
}
}
// 發送消息
else if (strcmp(buf, "Send") == 0)
{
char* message = "hello lyshark";
int ServerRet = send(sock, message, sizeof("hello lyshark"), 0);
if (ServerRet != 0)
{
std::cout << "發送消息數據包" << std::endl;
}
}
// 終止客戶端
else if (strcmp(buf, "Exit") == 0)
{
closesocket(sock);
WSACleanup();
exit(0);
}
// 存活探測信號
else if (strcmp(buf, "Ping") == 0)
{
int ServerRet = send(sock, "Pong", 4, 0);
if (ServerRet != 0)
{
std::cout << "Ping 存活探測..." << std::endl;
}
}
}
}
}
closesocket(sock);
WSACleanup();
Sleep(5000);
}
return 0;
}
讀者可自行編譯並運行上述代碼,當服務端啟動後客戶端上線,此時讀者可根據輸入不同的命令來操作不同的套接字,如下圖所示;
本文作者: 王瑞
本文鏈接: https://www.lyshark.com/post/4acbc25e.html
版權聲明: 本博客所有文章除特別聲明外,均採用 BY-NC-SA 許可協議。轉載請註明出處!
文章出處:https://www.cnblogs.com/LyShark/p/17771226.html
本博客所有文章除特別聲明外,均採用 BY-NC-SA 許可協議。轉載請註明出處!