网络编程之排除udp判断客户端是否断开

思路：每几秒发送一条不显示的信息，客户端断开则不再发送信息，超时则表现客户端断开毗连。（心跳包）

服务器

1. #include <head.h>
3. #define MAX_CLIENTS 100  // 最大支持100个客户端
4. #define TIMEOUT 5        // 5秒超时
6. struct Client {
7.     struct sockaddr_in addr;
8.     time_t last_seen;  // 记录最后一次收到该客户端数据的时间
9. };
11. struct Client client_list[MAX_CLIENTS];
12. int client_count = 0;
14. // **更新客户端心跳时间**
15. void update_client(struct sockaddr_in *client_addr) {
16.     time_t now = time(NULL);
18.     for (int i = 0; i < client_count; i++) {
19.         if (memcmp(&client_list[i].addr, client_addr, sizeof(struct sockaddr_in)) == 0) {
20.             client_list[i].last_seen = now;  // 更新时间
21.             return;
22.         }
23.     }
25.     // **如果客户端不在列表中，则添加**
26.     if (client_count < MAX_CLIENTS) {
27.         client_list[client_count].addr = *client_addr;
28.         client_list[client_count].last_seen = now;
29.         client_count++;
30.     }
31. }
33. // **检查超时客户端**
34. void check_clients() {
35.     time_t now = time(NULL);
36.     for (int i = 0; i < client_count; i++) {
37.         if (now - client_list[i].last_seen > TIMEOUT) {
38.             printf("客户端 %s:%d 断开\n",
39.                    inet_ntoa(client_list[i].addr.sin_addr),
40.                    ntohs(client_list[i].addr.sin_port));
42.             // **移除客户端**
43.             for (int j = i; j < client_count - 1; j++) {
44.                 client_list[j] = client_list[j + 1];
45.             }
46.             client_count--;
47.             i--;  // **继续检查下一个**
48.         }
49.     }
50. }
52. int main(int argc, const char *argv[]) {
53.     if (argc < 2) {
54.         printf("请输入端口号\n");
55.         return 1;
56.     }
57.     short port = atoi(argv[1]);
59.     // **创建 UDP 套接字**
60.     int receiver = socket(AF_INET, SOCK_DGRAM, 0);
61.     struct sockaddr_in addr = {0};
62.     addr.sin_family = AF_INET;
63.     addr.sin_port = htons(port);
64.     addr.sin_addr.s_addr = inet_addr("0.0.0.0");
66.     if (bind(receiver, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
67.         perror("bind");
68.         return 1;
69.     }
71.     printf("服务器启动，监听端口 %d\n", port);
73.     struct sockaddr_in client_addr;
74.     socklen_t addr_len = sizeof(client_addr);
75.     char buf[64];
77.     fd_set readfds;
78.     struct timeval timeout;
80.     while (1) {
81.         // **使用 select 进行超时检测**
82.         FD_ZERO(&readfds);
83.         FD_SET(receiver, &readfds);
84.         timeout.tv_sec = 1;  // 每秒检查一次
85.         timeout.tv_usec = 0;
87.         int activity = select(receiver + 1, &readfds, NULL, NULL, &timeout);
89.         if (activity > 0) {
90.             // **接收数据**
91.             memset(buf, 0, sizeof(buf));
92.             int len = recvfrom(receiver, buf, sizeof(buf) - 1, 0, (struct sockaddr*)&client_addr, &addr_len);
93.             if (len > 0) {
94.                 buf[len] = '\0';
95.                 update_client(&client_addr);  // **更新心跳时间**
97.                 // **如果是心跳包 "PING"，不打印、不转发**
98.                 if (strcmp(buf, "PING") == 0) {
99.                     continue;
100.                 }
102.                 printf("收到消息: %s\n", buf);
104.                 // **转发消息给所有在线客户端**
105.                 for (int i = 0; i < client_count; i++) {
106.                     sendto(receiver, buf, strlen(buf), 0,
107.                            (struct sockaddr*)&client_list[i].addr, sizeof(client_list[i].addr));
108.                 }
109.             }
110.         }
112.         // **检查超时客户端**
113.         check_clients();
114.     }
116.     return 0;
117. }

复制代码

客户端

1. #include <head.h>
2. #include <pthread.h>
4. #define BUF_SIZE 64
6. int sender;  // 套接字
8. struct sockaddr_in addr;
10. void *heartbeat(void* arg)
11. {
12.     while(1)
13.     {
14.         // 发送空的心跳包
15.         sendto(sender, "", 1, 0, (struct sockaddr*)&addr, sizeof(addr));
16.         sleep(2);  // 每2秒发送一次心跳包
17.     }
18. }
20. int main(int argc, const char *argv[])
21. {
22.     if (argc < 2) {
23.         printf("请输入端口号\n");
24.         return 1;
25.     }
26.     short port = atoi(argv[1]);
28.     // 创建套接字
29.     sender = socket(AF_INET, SOCK_DGRAM, 0);
30.     if (sender == -1) {
31.         perror("创建套接字失败");
32.         return 1;
33.     }
35.     addr.sin_family = AF_INET;
36.     addr.sin_port = htons(port);
37.     addr.sin_addr.s_addr = inet_addr("192.168.128.20");  // 服务器的IP地址
39.     // 启动心跳包线程
40.     pthread_t heart;
41.     pthread_create(&heart, NULL, heartbeat, NULL);
43.     while (1) {
44.         char buf[BUF_SIZE] = "";
45.         printf("输入：");
46.         scanf("%s", buf);
47.         getchar();  // 读取输入并去掉换行符
49.         // 发送普通消息
50.         sendto(sender, buf, strlen(buf), 0, (struct sockaddr*)&addr, sizeof(addr));
52.         // 接收服务器的回复
53.         int len = recvfrom(sender, buf, BUF_SIZE - 1, 0, NULL, NULL);
54.         if (len > 0) {
55.             buf[len] = '\0';  // 确保字符串以 '\0' 结尾
56.             printf("接收到回复的消息: %s\n", buf);
57.         } else {
58.             printf("接收服务器消息失败\n");
59.         }
60.     }
62.     // 关闭套接字
63.     close(sender);
64.     return 0;
65. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。