TCP socket心跳包示例程序

最新推荐文章于 2023-11-25 14:54:07 发布
最新推荐文章于 2023-11-25 14:54:07 发布
阅读量3.6w 收藏 90
点赞数 7
分类专栏： System - Linux 文章标签： tcp socket 心跳包 服务器 select
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/lisonglisonglisong/article/details/51327695
版权

在做游戏开发时，经常需要在应用层实现自己的心跳机制，即定时发送一个自定义的结构体（心跳包），让对方知道自己还活着，以确保连接的有效性。

在TCP socket心跳机制中，心跳包可以由服务器发送给客户端，也可以由客户端发送给服务器，不过比较起来，前者开销可能更大。—— 这里实现的是由客户端给服务器发送心跳包，基本思路是：

1） 服务器为每个客户端保存了IP和计数器count，即map<fd, pair<ip, count>>。服务端主线程采用 select 实现多路IO复用，监听新连接以及接受数据包（心跳包），子线程用于检测心跳：

  • 如果主线程接收到的是心跳包，将该客户端对应的计数器 count 清零；
  • 在子线程中，每隔3秒遍历一次所有客户端的计数器 count：
    • 若 count 小于 5，将 count 计数器加 1；
    • 若 count 等于 5，说明已经15秒未收到该用户心跳包，判定该用户已经掉线；

2） 客户端则只是开辟子线程，定时给服务器发送心跳包（本示例中定时时间为3秒）。


下面是Linux下一个socket心跳包的简单实现：

/*************************************************************************
    > File Name: Server.cpp
    > Author: SongLee
    > E-mail: lisong.shine@qq.com
    > Created Time: 2016年05月05日 星期四 22时50分23秒
    > Personal Blog: http://songlee24.github.io/
 ************************************************************************/
#include<netinet/in.h>   // sockaddr_in
#include<sys/types.h>    // socket
#include<sys/socket.h>   // socket
#include<arpa/inet.h>
#include<unistd.h>
#include<sys/select.h>   // select
#include<sys/ioctl.h>
#include<sys/time.h>
#include<iostream>
#include<vector>
#include<map>
#include<string>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
#define BUFFER_SIZE 1024

enum Type {HEART, OTHER};

struct PACKET_HEAD
{
    Type type;
    int length;
};

void* heart_handler(void* arg);

class Server
{
private:
    struct sockaddr_in server_addr;
    socklen_t server_addr_len;
    int listen_fd;    // 监听的fd
    int max_fd;       // 最大的fd
    fd_set master_set;   // 所有fd集合，包括监听fd和客户端fd   
    fd_set working_set;  // 工作集合
    struct timeval timeout;
    map<int, pair<string, int> > mmap;   // 记录连接的客户端fd--><ip, count>
public:
    Server(int port);
    ~Server();
    void Bind();
    void Listen(int queue_len = 20);
    void Accept();
    void Run();
    void Recv(int nums);
    friend void* heart_handler(void* arg);
};

Server::Server(int port)
{
    bzero(&server_addr, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htons(INADDR_ANY);
    server_addr.sin_port = htons(port);
    // create socket to listen
    listen_fd = socket(PF_INET, SOCK_STREAM, 0);
    if(listen_fd < 0)
    {
        cout << "Create Socket Failed!";
        exit(1);
    }
    int opt = 1;
    setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
}

Server::~Server()
{
    for(int fd=0; fd<=max_fd; ++fd)
    {
        if(FD_ISSET(fd, &master_set))
        {
            close(fd);
        }
    }
}

void Server::Bind()
{
    if(-1 == (bind(listen_fd, (struct sockaddr*)&server_addr, sizeof(server_addr))))
    {
        cout << "Server Bind Failed!";
        exit(1);
    }
    cout << "Bind Successfully.\n"; 
}

void Server::Listen(int queue_len)
{
    if(-1 == listen(listen_fd, queue_len))
    {
        cout << "Server Listen Failed!";
        exit(1);
    }
    cout << "Listen Successfully.\n";
}

void Server::Accept()
{
    struct sockaddr_in client_addr;
    socklen_t client_addr_len = sizeof(client_addr);

    int new_fd = accept(listen_fd, (struct sockaddr*)&client_addr, &client_addr_len);
    if(new_fd < 0)
    {
        cout << "Server Accept Failed!";
        exit(1);
    }

    string ip(inet_ntoa(client_addr.sin_addr));    // 获取客户端IP

    cout << ip << " new connection was accepted.\n";

    mmap.insert(make_pair(new_fd, make_pair(ip, 0)));

    // 将新建立的连接的fd加入master_set
    FD_SET(new_fd, &master_set);
    if(new_fd > max_fd)
    {
        max_fd = new_fd;
    }
}   

void Server::Recv(int nums)
{
    for(int fd=0; fd<=max_fd; ++fd)
    {
        if(FD_ISSET(fd, &working_set))
        {
            bool close_conn = false;  // 标记当前连接是否断开了

            PACKET_HEAD head;
            recv(fd, &head, sizeof(head), 0);   // 先接受包头

            if(head.type == HEART)
            {
                mmap[fd].second = 0;        // 每次收到心跳包，count置0
                cout << "Received heart-beat from client.\n";
            }
            else
            {
                // 数据包，通过head.length确认数据包长度 
            }   

            if(close_conn)  // 当前这个连接有问题，关闭它
            {
                close(fd);
                FD_CLR(fd, &master_set);
                if(fd == max_fd)  // 需要更新max_fd;
                {
                    while(FD_ISSET(max_fd, &master_set) == false)
                        --max_fd;
                }
            }
        }
    }   
}

void Server::Run()
{
    pthread_t id;     // 创建心跳检测线程
    int ret = pthread_create(&id, NULL, heart_handler, (void*)this);
    if(ret != 0)
    {
        cout << "Can not create heart-beat checking thread.\n";
    }

    max_fd = listen_fd;   // 初始化max_fd
    FD_ZERO(&master_set);
    FD_SET(listen_fd, &master_set);  // 添加监听fd

    while(1)
    {
        FD_ZERO(&working_set);
        memcpy(&working_set, &master_set, sizeof(master_set));

        timeout.tv_sec = 30;
        timeout.tv_usec = 0;

        int nums = select(max_fd+1, &working_set, NULL, NULL, &timeout);
        if(nums < 0)
        {
            cout << "select() error!";
            exit(1);
        }

        if(nums == 0)
        {
            //cout << "select() is timeout!";
            continue;
        }

        if(FD_ISSET(listen_fd, &working_set))
            Accept();   // 有新的客户端请求
        else
            Recv(nums); // 接收客户端的消息
    }
}


// thread function
void* heart_handler(void* arg)
{
    cout << "The heartbeat checking thread started.\n";
    Server* s = (Server*)arg;
    while(1)
    {
        map<int, pair<string, int> >::iterator it = s->mmap.begin();
        for( ; it!=s->mmap.end(); )
        {   
            if(it->second.second == 5)   // 3s*5没有收到心跳包，判定客户端掉线
            {
                cout << "The client " << it->second.first << " has be offline.\n";

                int fd = it->first;
                close(fd);            // 关闭该连接
                FD_CLR(fd, &s->master_set);
                if(fd == s->max_fd)      // 需要更新max_fd;
                {
                    while(FD_ISSET(s->max_fd, &s->master_set) == false)
                        s->max_fd--;
                }

                s->mmap.erase(it++);  // 从map中移除该记录
            }
            else if(it->second.second < 5 && it->second.second >= 0)
            {
                it->second.second += 1;
                ++it;
            }
            else
            {
                ++it;
            }
        }
        sleep(3);   // 定时三秒
    }
}

int main()
{
    Server server(15000);
    server.Bind();
    server.Listen();
    server.Run();
    return 0;
}


/*************************************************************************
    > File Name: Client.cpp
    > Author: SongLee
    > E-mail: lisong.shine@qq.com
    > Created Time: 2016年05月05日 星期四 23时41分56秒
    > Personal Blog: http://songlee24.github.io/
 ************************************************************************/
#include<netinet/in.h>   // sockaddr_in
#include<sys/types.h>    // socket
#include<sys/socket.h>   // socket
#include<arpa/inet.h>
#include<sys/ioctl.h>
#include<unistd.h>
#include<iostream>
#include<string>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
#define BUFFER_SIZE 1024

enum Type {HEART, OTHER};

struct PACKET_HEAD
{
    Type type;
    int length;
};

void* send_heart(void* arg); 

class Client 
{
private:
    struct sockaddr_in server_addr;
    socklen_t server_addr_len;
    int fd;
public:
    Client(string ip, int port);
    ~Client();
    void Connect();
    void Run();
    friend void* send_heart(void* arg); 
};

Client::Client(string ip, int port)
{
    bzero(&server_addr, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    if(inet_pton(AF_INET, ip.c_str(), &server_addr.sin_addr) == 0)
    {
        cout << "Server IP Address Error!";
        exit(1);
    }
    server_addr.sin_port = htons(port);
    server_addr_len = sizeof(server_addr);
    // create socket
    fd = socket(AF_INET, SOCK_STREAM, 0);
    if(fd < 0)
    {
        cout << "Create Socket Failed!";
        exit(1);
    }
}

Client::~Client()
{
    close(fd);
}

void Client::Connect()
{
    cout << "Connecting......" << endl;
    if(connect(fd, (struct sockaddr*)&server_addr, server_addr_len) < 0)
    {
        cout << "Can not Connect to Server IP!";
        exit(1);
    }
    cout << "Connect to Server successfully." << endl;
}

void Client::Run()
{
    pthread_t id;
    int ret = pthread_create(&id, NULL, send_heart, (void*)this);
    if(ret != 0)
    {
        cout << "Can not create thread!";
        exit(1);
    }
}

// thread function
void* send_heart(void* arg)
{
    cout << "The heartbeat sending thread started.\n";
    Client* c = (Client*)arg;
    int count = 0;  // 测试
    while(1) 
    {
        PACKET_HEAD head;
        head.type = HEART;
        head.length = 0;    
        send(c->fd, &head, sizeof(head), 0);
        sleep(3);     // 定时3秒

        ++count;      // 测试：发送15次心跳包就停止发送
        if(count > 15)
            break;
    }
}

int main()
{
    Client client("127.0.0.1", 15000);
    client.Connect();
    client.Run();
    while(1)
    {
        string msg;
        getline(cin, msg);
        if(msg == "exit")
            break;
        cout << "msg\n";
    }
    return 0;
}



可以看出，客户端启动以后发送了15次心跳包，然后停止发送心跳包。在经过一段时间后（3s*5），服务器就判断该客户端掉线，并断开了连接。




神奕
关注
专栏目录
socketTCP通信心跳包实例
12-24
winform的socket通信中连接是否正常，最好通过心跳包来确定，本程序就是一个简单的心跳包实例，简单、易懂~
【服务程序的运行策略】——心跳进程
最新发布
weixin_56520780的博客
10-02 890
本文先介绍进程的心跳，从原理上实现基于共享内存的心跳进程。
TCP socket通信实例（心跳）
10-18
一个简单的SOCKET 通信实例，加入了心跳信息。实用 。
C# Socket心跳包跟图片传输，TCP/IP协议响应Demo供学习之用
07-14
很多人在写Socket的时候会遇到心跳包响应机制跟主线程消息传输机制如何规避而互不影响，本Demo就解决了此问题，而且简单易懂，都有备注，欢迎下载学习之用。
TCP/IP协议栈的心跳、丢包重传、连接超时机制实例详解
关注微信公众号【开源Linux】，后台回复『10T』，领取10T学习资源大放送，涵盖Linux、虚拟化、容器、云计算、网络、Python、Go等书籍和视频
07-10 424
最近有个项目，客户的网络环境不太稳定，会时不时出现丢包和网络抖动的情况，导致我们软件客户端会时不时和服务器断链，导致正在进行中的视频会议会被中断。本文借此机会，结合具体的问题实例，详细讲解一下TCP/IP协议栈的心跳机制、丢包重传机制等内容，给大家提供一个借鉴和参考。1、问题概述虽然软件底层模块在网络恢复后能自动重连上服务器，但会议因为网络问题已经退出，需要重新加入会议。因为客户特殊的网络运行环境...
c++ tcp socket心跳包示例程序
01-09
TCP的心跳包是指在TCP连接...以上是一个简单的TCP socket心跳包示例程序，通过该程序可以实现定时向服务器发送心跳包，从而保持TCP连接的稳定性。在实际应用中，可以根据具体需求对心跳包的内容、发送频率等进行调整。
c++ socket tcp demo
02-08
本示例中的"TCP Demo"提供了C++实现TCP客户端和服务器的简单应用，通过socketserver.cpp和socketclient.cpp两个文件来展示。下面我们将深入探讨TCP协议、C++ Socket API以及这两个文件可能包含的关键知识点。 首先...
C#使用Socket实现心跳的方法示例
08-25
第三，需要在Timer的Elapsed事件中，使用Socket对象发送心跳包。 第四，需要在心跳包发送成功后，更新连接状态，以确保连接保持活跃。 在示例代码中，使用了SocketListener类来监听客户端连接，并在客户端连接成功...
socket中的短连接与长连接,心跳包示例详解
热门推荐
北雨南萍
06-06 1万+
TCP连接简介 当网络通信时采用TCP协议时，在真正的读写操作之前，server与client之间必须建立一个连接， 当读写操作完成后，双方不再需要这个连接时它们可以释放这个连接， 连接的建立是需要三次握手的，而释放则需要4次握手， 所以说每个连接的建立都是需要资源消耗和时间消耗的 ​经典的三次握手示意图：​ ​ ​ ​ 经典的四次握手关闭图：   ​
TCP之心跳包理解
07-05
让初学者更容易理解心跳包是怎么回事，客户端和服务端是如何处理心跳包的
TCP通信心跳包源码C语言实现
04-29
TCP通信心跳包源码C语言实现，基于Linux平台写的，可以整合到自己的代码中去。
TCP接口心跳包学习笔记
ssz__的博客
06-21 1011
即时通讯：虚拟机模拟服务器，可以暂停运行的代码，Xshell当客户端访问服务器（telnet 192.168.56.103 9001），进入聊天室。此时暂停服务器，模拟服务器掉线，客户端ping不同，在没有心跳机制的情况下，服务端掉线，对方不可知，客户端不能发消息，网络不可知。例子：一个接口网线、CPU、电线、网卡都正常，但是此时这个接口已经死循环，无法处理请求，此种情况检查不出来。设想：如果客户端掉线，服务端不知道，此时还维持链接，就是额外无效系统开销。2.TCP 》》》长链接 （后面转成TCP）
c语言tcp心跳程序,c语言多进程tcp服务器示例
weixin_39775872的博客
05-18 188
#include "server.h"/** Create a TCP Server* Return socketfd*/int create_tcpsvr(char *ip,int port){int sockfd;struct sockaddr_in addr;if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == EOF){perror("crea...
心跳包的实现
m0_46376834的博客
11-25 800
然后，程序会休眠一段时间（这里是20秒）后调用。这个示例展示了一个基本的心跳包实现，你可以根据实际需求，在。的类，用于发送心跳包。函数中添加具体的心跳包发送逻辑。上述代码定义了一个名为。
【项目】基于TCP/IP的socket编程之心跳机制
lixungogogo的博客
08-11 9927
什么是心跳机制？ 　　　　想一下， 当tcp连接被破坏后， 如果是死连接了， 服务端和客户端怎样才能知道信息能不能到达对方呢？ 很自然的想法是， 不断地给对方发探测信号， 看有没有回应， 这就是心跳机制的直白原理。 　　所谓的心跳即是数据包， 发心跳就是一方向另一方发送的数据包， 不断地发送， 如果收不到回应， 那么就有理由认为是tcp连接出了问题。 那为什么要叫心跳呢？ 你摸一下你的心， 你看
心跳包：告诉别人，我还活着
weixin_34082695的博客
07-14 229
心跳包是什么？有什么用？ 心跳一般是指客户端（也可以是服务器端）向对方每隔一段时间发送一个应用层的自定义指令，以确保连接的有效性。因为是固定间隔，同时是检测存活，就像人的心跳一样，顾名思义，称为心跳包。一般是用于长连接，对消息实时性要求比较高的服务中，比如IM服务，推送服务。 长连接有什么用？ 在即时通讯领域和推送服务中，对消息的实时性和可用性要求非常高，建立长连接，可以有效节省DNS解释时间，T...
学习笔记 c++（ socket网络编程，心跳监测，断线重连，结构体发送接收）
qq_42145185的博客
11-07 2661
server.h /* * @Author: Li Chao * @Date: 2020-11-03 08:38:56 * @Last Modified by: Li Chao * @Last Modified time: 2020-11-03 17:03:45 */ #include <arpa/inet.h> #include <errno.h> #include <netinet/in.h> #include <poll.h> #in
评论 8
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值