C语言使用TCP协议向服务器发送心跳包
时间: 2023-07-20 07:37:47 浏览: 260
基于TCP协议的心跳包
TCP协议发送心跳包的方式与UDP协议略有不同,需要在保持TCP连接的同时发送心跳包。
以下是一个使用TCP协议向服务器发送心跳包的示例代码:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <errno.h>
#include <signal.h>
#define SERVER_IP "127.0.0.1"
#define SERVER_PORT 8888
#define HEARTBEAT_INTERVAL 10 // 心跳包发送间隔,单位为秒
#define MAX_RETRY_TIMES 3 // 最大重试次数
volatile sig_atomic_t can_exit = 0; // 是否可以退出程序的标志
void sigint_handler(int signum) {
can_exit = 1;
}
int heartbeat(int sockfd) {
int ret, retry_times;
char buf[1024] = "heartbeat";
while (!can_exit) {
// 向服务器发送心跳包
ret = send(sockfd, buf, strlen(buf), 0);
if (ret <= 0) {
perror("send");
return -1;
}
printf("send heartbeat\n");
sleep(HEARTBEAT_INTERVAL);
// 接收服务器响应
char recv_buf[1024];
ret = recv(sockfd, recv_buf, sizeof(recv_buf), MSG_DONTWAIT);
if (ret < 0) {
if (errno == EAGAIN || errno == EWOULDBLOCK) {
// 未收到响应,继续发送心跳包
continue;
} else {
perror("recv");
return -1;
}
} else if (ret == 0) {
// 连接已断开,需要重新连接
printf("connection closed\n");
return -1;
} else {
// 收到服务器响应,重置重试次数
retry_times = 0;
}
// 如果连续多次未收到服务器响应,则认为连接已断开
retry_times++;
if (retry_times > MAX_RETRY_TIMES) {
printf("no response from server\n");
return -1;
}
}
return 0;
}
int main() {
int sockfd;
struct sockaddr_in server_addr;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
perror("socket");
exit(-1);
}
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = inet_addr(SERVER_IP);
server_addr.sin_port = htons(SERVER_PORT);
// 建立TCP连接
int ret = connect(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr));
if (ret < 0) {
perror("connect");
exit(-1);
}
// 注册SIGINT信号处理函数,以便在收到Ctrl+C信号时可以安全退出程序
signal(SIGINT, sigint_handler);
// 开始发送心跳包
heartbeat(sockfd);
close(sockfd);
return 0;
}
```
该示例中,使用TCP协议向服务器发送心跳包。心跳包内容为字符串"heartbeat",发送间隔为10秒。在程序开始时,先建立TCP连接。然后进入心跳包发送循环,不断发送心跳包,同时接收服务器的响应。如果连续多次未收到服务器响应,则认为连接已断开,需要重新连接。如果收到服务器的响应,则重置重试次数。在程序收到Ctrl+C信号时,设置can_exit标志,以便安全退出程序。
阅读全文