1.創建 TCP 套接字
int server_sockfd = socket(AF_INET,SOCK_STREAM, 0);
函數原型:
#include <sys/socket.h>
int socket(int domain, int type, int protocol);
domain | 協議族(地址族) | AF_INET (IPv4) |
type | 套接字類型 | SOCK_STREAM (TCP) |
protocol | 協議類型 | 0 (自動選擇) |
(1)?domain
(協議族)
AF_INET
:IPv4 地址族(最常用)。AF_INET6
:IPv6 地址族。AF_UNIX
(或?AF_LOCAL
):本地進程間通信(UNIX 域套接字)。
(2)?type
(套接字類型)
SOCK_STREAM
:- 面向連接的?TCP 套接字。
- 提供可靠、雙向、基于字節流的通信。
SOCK_DGRAM
:- 無連接的?UDP 套接字。
- 提供不可靠、無邊界的數據報服務。
SOCK_RAW
:- 原始套接字,用于自定義協議(如 ICMP)。
(3)?protocol
(協議類型)
0
:- 讓系統自動選擇與?
domain
?和?type
?匹配的協議。 - 對于?
AF_INET + SOCK_STREAM
,默認選擇?TCP。
- 讓系統自動選擇與?
- 其他常見值:
IPPROTO_TCP
(顯式指定 TCP,但通常用?0
?即可)。IPPROTO_UDP
(用于?SOCK_DGRAM
)。
?
返回值
- 成功:返回一個?非負整數,即?套接字描述符(
server_sockfd
)。 - 失敗:返回?
-1
,并設置?errno
(如?EMFILE
、ENFILE
、EACCES
?等)。
2.設置套接字選項?SO_REUSEADDR
?的標準用法,用于控制套接字的行為
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))?
- 作用:
- 允許套接字(
sockfd
)綁定到?處于?TIME_WAIT
?狀態的地址(例如服務器重啟時)。 - 避免?
bind()
?失敗(EADDRINUSE
?錯誤)。
- 允許套接字(
- 參數說明:
sockfd
:目標套接字描述符。SOL_SOCKET
:表示操作套接字層選項(通用選項)。SO_REUSEADDR
:選項名稱,允許地址重用。&reuse
:指向選項值的指針(int
?類型,1
?啟用,0
?禁用)。sizeof(reuse)
:選項值的大小。
3.TCP 服務器綁定(bind()
)操作?
struct sockaddr_in server_sockaddr;server_sockaddr.sin_family = AF_INET; // IPv4 協議族server_sockaddr.sin_addr.s_addr = htonl(INADDR_ANY); // 綁定到所有本地接口(0.0.0.0) server_sockaddr.sin_port = htons(voice_SysParameter.port); // 綁定到指定端口(網絡字節序)// 綁定套接字
if (bind(server_sockfd, (struct sockaddr *)&server_sockaddr, sizeof(server_sockaddr)) == -1)
{perror("bind"); // 輸出錯誤信息 goto voice_tcp_Thread_TaskProcError; // 錯誤處理}
- 關鍵點:
AF_INET
:使用 IPv4 協議。INADDR_ANY
:綁定到所有本地網絡接口(即?0.0.0.0
),允許通過任何本地 IP 訪問服務。htons()
?和?htonl()
:將主機字節序(小端或大端)轉換為網絡字節序(大端)。bind()
?返回值:成功返回?0
,失敗返回?-1
?并設置?errno
。
4.TCP 連接超時檢測與清理??
for (int i = 0; i < voice_MaxConnectNum; i++) {// 檢查條件:// 1. ConnfdCurTime[i] > 0(記錄過心跳時間)// 2. 當前時間 - 最后一次心跳時間 > 允許的心跳超時時間(voice_SysParameter.heartBeat)// 3. Connfd[i] > 0(連接有效)if ((ConnfdCurTime[i] > 0) && (ConnfdCurTime[i] + voice_SysParameter.heartBeat < HeartBeatCnt) && (Connfd[i] > 0)) {struct sockaddr_in sa = {0};int len = sizeof(sa);// 獲取客戶端的 IP 地址和端口信息getpeername(Connfd[i], (struct sockaddr *)&sa, &len);// 關閉超時連接,并打印日志(連接描述符 + 客戶端 IP)close(Connfd[i]);printf("Connfd Tcp__closed=%d %s\n", Connfd[i], inet_ntoa(sa.sin_addr));// 重置連接狀態Connfd[i] = 0;ConnfdCurTime[i] = 0;}
}
5.檢查所有已記錄的子設備(SubDev)狀態
for (int j = 0; j < voice_SysParameter.connectNo; j++) {// 如果子設備的 IP 地址為空,跳過處理if (strcmp(voice_SysParameter.SubDevState.SubDevState[j].ip, "") == 0) continue;int i = 0;// 遍歷所有 TCP 連接,檢查是否有連接匹配當前子設備的 IPfor (i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] == 0) continue; // 跳過無效連接(Connfd[i]=0 表示空閑)struct sockaddr_in sa = {0};int len = sizeof(sa);// 獲取當前連接的客戶端 IP 地址getpeername(Connfd[i], (struct sockaddr *)&sa, &len);// 如果當前連接的 IP 匹配子設備的 IP,跳出循環if (strcmp(voice_SysParameter.SubDevState.SubDevState[j].ip, inet_ntoa(sa.sin_addr)) == 0) break;}// 如果遍歷完所有連接都沒找到匹配的 IP(i == voice_MaxConnectNum)if (i == voice_MaxConnectNum) {// 清除子設備狀態(標記為未連接)voice_SysParameter.SubDevState.SubDevState[j].slaveNo = (char)-1;strcpy(voice_SysParameter.SubDevState.SubDevState[j].ip, "");voice_SysParameter.SubDevState.SubDevState[j].status = 0;}
}
?6.使用?select()
?監聽多個文件描述符(file descriptors)的可讀事件
// 清空文件描述符集合
FD_ZERO(&rset);
FD_SET(server_sockfd, &rset); // 將服務器套接字加入監聽集合
maxfd = server_sockfd; // 初始化最大文件描述符為服務器套接字// 遍歷所有客戶端連接,更新監聽集合和最大文件描述符
for (int i = 0; i < voice_MaxConnectNum; i++) {maxfd = maxfd > Connfd[i] ? maxfd : Connfd[i]; // 更新最大文件描述符if (Connfd[i] > 0) // 只監聽有效的連接FD_SET(Connfd[i], &rset); // 加入監聽集合
}// 調用 select() 監聽 I/O 事件
nready = select(maxfd + 1, &rset, NULL, NULL, &timeout);// 超時處理
if (nready == 0) {continue; // select 超時,繼續循環
}
- 核心功能:使用?
select()
?監聽服務器套接字和客戶端連接的可讀事件,實現 I/O 多路復用。 - 關鍵點:
FD_ZERO()
?+?FD_SET()
?初始化監聽集合。select(maxfd + 1, &rset, ...)
?阻塞等待 I/O 事件。FD_ISSET()
?檢查哪個文件描述符就緒。
- 改進方向:
- 改用?
epoll()
/kqueue()
?提高性能。 - 增加錯誤處理和連接管理。
- 動態調整超時時間。
- 改用?
7.處理?select()
?返回的就緒文件描述符?
for (int i = 0; i < voice_MaxConnectNum; i++) {if (FD_ISSET(Connfd[i], &rset)) { // 檢查 Connfd[i] 是否就緒(可讀)ConnfdCurTime[i] = HeartBeatCnt; // 更新心跳時間戳if (voice_ConTask(Connfd[i]) == -1) { // 處理客戶端數據,返回 -1 表示錯誤struct sockaddr_in sa = {0};int len = sizeof(sa);// 1. 獲取客戶端 IP 地址(用于日志或調試)getpeername(Connfd[i], (struct sockaddr *)&sa, &len);// 2. 關閉連接并清理資源close(Connfd[i]);printf("Connfd Tcp__closed=%d %s\n", Connfd[i], inet_ntoa(sa.sin_addr));Connfd[i] = 0; // 標記連接為無效ConnfdCurTime[i] = 0; // 清空心跳時間戳}}
}
8.處理單個客戶端連接的數據接收、解析和響應
int voice_ConTask(int sockfd) {char buffer[2048]; // 接收數據的緩沖區memset(buffer, 0, sizeof(buffer)); // 清空緩沖區// 1. 接收客戶端數據int len = recv(sockfd, buffer, sizeof(buffer), 0);// 2. 檢查是否收到 "exit" 命令(客戶端主動關閉)if (strcmp(buffer, "exit") == 0) {printf("sockfd: %d exited.1\n", sockfd);return -1; // 返回 -1 表示連接需要關閉}// 3. 檢查連接是否已關閉(len=0 表示客戶端斷開)else if (len == 0) {printf("sockfd: %d exited.2\n", sockfd);return -1; // 返回 -1 表示連接需要關閉}// 4. 處理有效數據else {// 調用 voice_TaskCommand 處理數據,并返回響應長度len = voice_TaskCommand(buffer, len);if (len > 0) {// 發送響應數據給客戶端send(sockfd, buffer, len, 0);}}return 1; // 返回 1 表示處理成功,繼續保持連接
}
9.TCP 服務器?處理?新客戶端連接?的邏輯
// 1. 檢查服務器 socket 是否可讀(即是否有新連接到達)
if (FD_ISSET(server_sockfd, &rset)) {struct sockaddr_in client_addr;socklen_t length = sizeof(client_addr);// 2. 接受新連接int conn = accept(server_sockfd, (struct sockaddr*)&client_addr, &length);if (conn < 0) {perror("New client connect Error");continue; // 接受失敗,跳過本次循環} else {printf("new client accepted.\n");}// 3. 將新連接存入連接池(Connfd 數組)for (int i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] == 0) { // 找到空閑位置Connfd[i] = conn; // 存儲新連接的文件描述符printf("Connfd[%d]=%d\n\r", i, conn);printf("IP:%s Connected...\n\r", inet_ntoa(client_addr.sin_addr));break; // 存入后跳出循環}// 4. 處理連接數超限if (i == voice_MaxConnectNum - 1) { // 遍歷完所有位置printf("New Conn Num limit to %d \n\r", voice_MaxConnectNum);close(conn); // 關閉新連接(因為無法存儲)}}
}
完整代碼
/*** @brief TCP 服務器線程處理函數* @param p 線程參數(結構體指針,包含線程啟動標志)* @return void* 線程返回值(未使用)*/
void* voice_tcp_Thread_TaskProc(void* p) {// 1. 解析線程參數struct voice_Thread_PARA_S *pstPara = (struct voice_Thread_PARA_S*)p;// 2. 創建服務器 socket(IPv4 + TCP)int server_sockfd = socket(AF_INET, SOCK_STREAM, 0);if (server_sockfd < 0) {perror("socket creation failed");goto voice_tcp_Thread_TaskProcError;}// 3. 初始化連接池和心跳時間記錄數組int Connfd[voice_MaxConnectNum] = {0}; // 存儲客戶端連接的文件描述符int ConnfdCurTime[voice_MaxConnectNum] = {0}; // 記錄最后一次心跳時間(秒)// 4. 設置 SO_REUSEADDR 選項(避免端口占用)int reuse = 1;if (setsockopt(server_sockfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0) {perror("Setting SO_REUSEADDR failed");goto voice_tcp_Thread_TaskProcError;}// 5. 綁定服務器地址和端口struct sockaddr_in server_sockaddr;memset(&server_sockaddr, 0, sizeof(server_sockaddr));server_sockaddr.sin_family = AF_INET;server_sockaddr.sin_addr.s_addr = htonl(INADDR_ANY); // 監聽所有網卡server_sockaddr.sin_port = htons(voice_SysParameter.port); // 轉換端口為網絡字節序if (bind(server_sockfd, (struct sockaddr*)&server_sockaddr, sizeof(server_sockaddr)) == -1) {perror("bind failed");goto voice_tcp_Thread_TaskProcError;}printf("bind success.\n");// 6. 開始監聽(待處理連接隊列長度為 20)if (listen(server_sockfd, 20) == -1) {perror("listen failed");goto voice_tcp_Thread_TaskProcError;}printf("listen success.\n");printf("Tcp Server: %s:%d\n", voice_SysParameter.ip, voice_SysParameter.port);// 7. 主循環(處理客戶端連接和數據)while (pstPara->bThreadStart == 1) {// 7.1 獲取當前時間(用于心跳檢測)struct timeval tv;gettimeofday(&tv, NULL);int current_time = tv.tv_sec;// 7.2 心跳檢測:關閉超時未活動的連接for (int i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] > 0 && ConnfdCurTime[i] > 0 && (current_time - ConnfdCurTime[i] > voice_SysParameter.heartBeat)) {struct sockaddr_in client_addr;socklen_t len = sizeof(client_addr);getpeername(Connfd[i], (struct sockaddr*)&client_addr, &len); // 獲取客戶端 IPprintf("Heartbeat timeout, close connection: fd=%d, IP=%s\n", Connfd[i], inet_ntoa(client_addr.sin_addr));close(Connfd[i]); // 關閉連接Connfd[i] = 0; // 清空連接池位置ConnfdCurTime[i] = 0;}}// 7.3 檢查設備狀態(如果客戶端斷開,更新設備狀態)for (int j = 0; j < voice_SysParameter.connectNo; j++) {if (strcmp(voice_SysParameter.SubDevState.SubDevState[j].ip, "") == 0) continue;int i;for (i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] <= 0) continue; // 跳過無效連接struct sockaddr_in client_addr;socklen_t len = sizeof(client_addr);getpeername(Connfd[i], (struct sockaddr*)&client_addr, &len);if (strcmp(voice_SysParameter.SubDevState.SubDevState[j].ip, inet_ntoa(client_addr.sin_addr)) == 0) {break; // 找到匹配的連接}}// 如果未找到匹配連接,清空設備狀態if (i == voice_MaxConnectNum) {voice_SysParameter.SubDevState.SubDevState[j].slaveNo = -1;strcpy(voice_SysParameter.SubDevState.SubDevState[j].ip, "");voice_SysParameter.SubDevState.SubDevState[j].status = 0;}}// 7.4 使用 select 監聽 I/O 事件fd_set rset;FD_ZERO(&rset);FD_SET(server_sockfd, &rset); // 監聽服務器 socket(接受新連接)int maxfd = server_sockfd;for (int i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] > 0) {FD_SET(Connfd[i], &rset); // 監聽所有客戶端連接maxfd = (Connfd[i] > maxfd) ? Connfd[i] : maxfd; // 更新最大文件描述符}}// 設置 select 超時時間(1 秒)struct timeval timeout;timeout.tv_sec = 1;timeout.tv_usec = 0;int nready = select(maxfd + 1, &rset, NULL, NULL, &timeout);if (nready <= 0) continue; // 超時或錯誤,繼續循環// 7.5 處理客戶端數據for (int i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] > 0 && FD_ISSET(Connfd[i], &rset)) {ConnfdCurTime[i] = current_time; // 更新心跳時間// 調用業務邏輯處理函數(如解析協議、處理請求)if (voice_ConTask(Connfd[i]) == -1) {struct sockaddr_in client_addr;socklen_t len = sizeof(client_addr);getpeername(Connfd[i], (struct sockaddr*)&client_addr, &len);printf("Client disconnected or error, close connection: fd=%d, IP=%s\n", Connfd[i], inet_ntoa(client_addr.sin_addr));close(Connfd[i]);Connfd[i] = 0;ConnfdCurTime[i] = 0;}}}// 7.6 處理新連接if (FD_ISSET(server_sockfd, &rset)) {struct sockaddr_in client_addr;socklen_t length = sizeof(client_addr);int conn = accept(server_sockfd, (struct sockaddr*)&client_addr, &length);if (conn < 0) {perror("accept failed");continue;}printf("New client accepted: fd=%d, IP=%s\n", conn, inet_ntoa(client_addr.sin_addr));// 將新連接存入連接池for (int i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] == 0) {Connfd[i] = conn;ConnfdCurTime[i] = current_time; // 初始化心跳時間printf("Connfd[%d]=%d, IP=%s Connected\n", i, conn, inet_ntoa(client_addr.sin_addr));break;}// 連接數超限if (i == voice_MaxConnectNum - 1) {printf("Connection limit reached (%d), reject new client\n", voice_MaxConnectNum);close(conn); // 關閉新連接}}}}// 8. 清理資源(線程退出時)for (int i = 0; i < voice_MaxConnectNum; i++) {if (Connfd[i] > 0) {close(Connfd[i]);Connfd[i] = 0;}}close(server_sockfd);return NULL;voice_tcp_Thread_TaskProcError:if (server_sockfd >= 0) close(server_sockfd);return NULL;
}