[C++] 从零实现一个ping服务

news2024/10/5 17:28:11
在这里插入图片描述

💻文章目录

  • 前言
  • ICMP
    • 概念
    • 报文格式
  • Ping服务实现
    • 系统调用函数
    • 具体实现
    • 运行测试
  • 总结


前言

ping命令,因为其简单、易用等特点,几乎所有的操作系统都内置了一个ping命令。如果你是一名C++初学者,对网络编程、系统编程有所了解,但又没有多少实操经验的话,不妨来尝试动手实现一个属于自己的ping命令。这样一来,也能提高你对系统编程、网络编程的能力。

ICMP

概念

ICMP是工作在网络层的一种不可靠的传输协议,意在辅助IP协议获取报文传输与网络连接的情况,被广泛运用于网络诊断工具(如:ping 和 traceroute)。

ICMP协议可以控制路由将报文错误原因返回给源主机,从而实现对网络状况的诊断。

在这里插入图片描述

报文格式

ICMP协议被封装在IP协议之中,以下为ICMP的报文固定格式:

在这里插入图片描述

  • 类型:用于标识报文的类型,ICMP报文类型分为两类:信息类报文、差错类报文。

  • 代码:用于标识差错类报文的具体错误信息。

  • 校验和:用于计算报文是否出现损坏(发送方填写,接收方校验)。

「ICMP常见消息类型」

ICMP 类型描述
0回显应答(Echo Reply):对回显请求的响应,通常用于ping操作。
3目的不可达(Destination Unreachable):目标地址无法到达时发送,包括网络不可达、主机不可达等子类型。
4源抑制(Source Quench):请求发送方降低发送速率,以防止网络拥塞(现已弃用)。
5重定向(Redirect):建议主机将数据包发送到不同的路由器,提供更优路径。
8回显请求(Echo Request):请求目标主机返回应答消息,通常用于ping操作。
11超时(Time Exceeded):数据包在网络中传输时间超过TTL值,或在分片重组过程中超时。
12参数问题(Parameter Problem):数据包的IP头部存在错误,导致无法处理。

「Linux中的实现」

Linux中ICMP报文格式有不少成员,但只是实现ping服务只需要以下成员:

  • icmp_type:icmp报文的类型。

  • icmp_cksum:校验和,用于计算数据是否损坏。

  • icmp_id:用于标识报文的唯一性。

  • icmp_seq:序列号字段,多用于echo、echoreply功能。

  • icmp_data:报文的内容,只有8bit大小

「Linux中ICMP报文的描述」

/*Linux中icmp的有较多成员变量,嫌麻烦可以看#define部分来认识主要成员变量*/
struct icmp
{
  uint8_t  icmp_type;	/* icmp类型; type of message, see below */
  uint8_t  icmp_code;	/* type sub code */
  uint16_t icmp_cksum;	/*校验和,用于确定报文是否完整无损*/
  union
  {
    unsigned char ih_pptr;	/* ICMP_PARAMPROB */
    struct in_addr ih_gwaddr;	/* gateway address */
    struct ih_idseq		/* echo datagram */
    {
      uint16_t icd_id;
      uint16_t icd_seq;
    } ih_idseq;
    uint32_t ih_void;

    /* ICMP_UNREACH_NEEDFRAG -- Path MTU Discovery (RFC1191) */
    struct ih_pmtu
    {
      uint16_t ipm_void;
      uint16_t ipm_nextmtu;
    } ih_pmtu;

    struct ih_rtradv
    {
      uint8_t irt_num_addrs;
      uint8_t irt_wpa;
      uint16_t irt_lifetime;
    } ih_rtradv;
  } icmp_hun;
#define	icmp_pptr	icmp_hun.ih_pptr
#define	icmp_gwaddr	icmp_hun.ih_gwaddr
#define	icmp_id		icmp_hun.ih_idseq.icd_id
#define	icmp_seq	icmp_hun.ih_idseq.icd_seq
#define	icmp_void	icmp_hun.ih_void
#define	icmp_pmvoid	icmp_hun.ih_pmtu.ipm_void
#define	icmp_nextmtu	icmp_hun.ih_pmtu.ipm_nextmtu
#define	icmp_num_addrs	icmp_hun.ih_rtradv.irt_num_addrs
#define	icmp_wpa	icmp_hun.ih_rtradv.irt_wpa
#define	icmp_lifetime	icmp_hun.ih_rtradv.irt_lifetime
  union
  {
    struct    //存储时间戳
    {
      uint32_t its_otime;        // 原始时间戳,发送时的时间
      uint32_t its_rtime;        // 接受时间戳,接受时的时间
      uint32_t its_ttime;        // 传输时间戳,传输所用时间
    } id_ts;
    struct
    {
      struct ip idi_ip;
      /* options and then 64 bits of data */
    } id_ip;
    struct icmp_ra_addr id_radv;
    uint32_t   id_mask;
    uint8_t    id_data[1];
  } icmp_dun;
#define	icmp_otime	icmp_dun.id_ts.its_otime
#define	icmp_rtime	icmp_dun.id_ts.its_rtime
#define	icmp_ttime	icmp_dun.id_ts.its_ttime
#define	icmp_ip		icmp_dun.id_ip.idi_ip
#define	icmp_radv	icmp_dun.id_radv
#define	icmp_mask	icmp_dun.id_mask
#define	icmp_data	icmp_dun.id_data
};

Ping服务实现

系统调用函数

原始套接字

要使用ICMP协议就必须绕过传输层(TCP/UDP),直接操作网络层,所以必须使用原始套接字,在Mac、Linux中使用原始套接字可能会需要root权限

//函数原型
int socket(int domain, int type, int protocol);

int _sockfd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);  //使用原始套接字

信号转换

在Linux中的ping服务一般通过ctl+c来实现终止,所以得要将信号执行函数替换成自己的函数。

//函数原型
void (*signal(int sig, void (*func)(int)))(int);

//使用方式
signal(SIGINT, [](int sig)
{
    printf("sig:%d", sig);
} );

「域名转换为IP地址」

在Linux中将域名转成ip地址的函数有gethostbyname,但其在新版本的linux中已经被废弃,所以这里使用较新的getaddrinfo。

/*通过getaddrinfo获取的数据将存进该结构体*/
struct addrinfo {
   int              ai_flags;
   int              ai_family;    //协议族
   int              ai_socktype;
   int              ai_protocol;
   socklen_t        ai_addrlen;  // sockaddr 的长度
   struct sockaddr *ai_addr;     // 根据需求转换成sockaddr_in
   char            *ai_canonname;
   struct addrinfo *ai_next;     //下一个addrinfo,使用链表来连接匹配的IP。
};

int getaddrinfo(const char *restrict node,                  //需要转换的域名
                   const char *restrict service,            //DNS服务器地址,可为空
                   const struct addrinfo *restrict hints,   //用于限定获取的数据
                   struct addrinfo **restrict res);         //结果存放的指针

具体实现

ping服务的实现使用了类来进行封装,从而使得其更简洁易懂。

头文件声明

#include <netdb.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/ip_icmp.h>
#include <string>
#include <iostream>
#include <format>
#include <thread>


class PingServer
{
public:
    PingServer(const char* ip);   

    void Start();  

    static void TimeEnd();   // ping计算总结,ctrl+c调用。

private:
    void Init();     // 初始化类

    void SendData();  //发送数据

    void RecvData();  //接受数据

    unsigned short CheckSum(void* data, int len);   //计算校验和

private:
    static std::chrono::system_clock::time_point _oldTime;   //计算ping服务运行时间
    static int _sendSeq;  //发送数据次数
    static int _recvSeq;  //接受数据次数

    struct sockaddr_in _destAddr;  //远端地址信息
    const char* _ip;    //需要ping的ip/hostname;
    char _recvData[1024];   //接受数据缓冲区
    
    int _sockfd;   //套接字

    unsigned short _id;   //用于标识ip报文唯一性。
};

//初始化静态成员
std::chrono::system_clock::time_point PingServer::_oldTime = std::chrono::system_clock::now();
int PingServer::_sendSeq = 0;
int PingServer::_recvSeq = 0;

介绍完类的成员,也该到其实现了⬇️。

#include <netdb.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>
#include <stdio.h>
#include <string.h>
#include <netinet/ip_icmp.h>
#include <string>
#include <iostream>
#include <format>
#include <future>
#include <thread>


//TODO chrono时钟实现超时

class PingServer
{
public:
    PingServer(const char* ip)
        :_ip(ip), _id(htons(getpid()))
    {
        Init();
    }

    void Start()
    {
        std::thread(&PingServer::SendData, this).detach();
        RecvData();
    }

    static void TimeEnd()
    {
        auto now = std::chrono::system_clock::now();
        auto sum = std::chrono::duration_cast<std::chrono::milliseconds>(now-_oldTime).count();
        int loss = ((double)(_sendSeq - _recvSeq) / _sendSeq) * 100;

        std::cout << std::format("\n{} packets transimitted, {} received, {}% packet loss, time {}ms", _sendSeq, _recvSeq, loss, sum) << std::endl;
    }

private:
    void Init()
    {
        _sockfd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);  //使用原始套接字
        if(_sockfd < 0) 
        {
            std::cerr << "socket error" << std::endl;  
            exit(-1);
        }

        struct addrinfo hints{}, *res{};   
        hints.ai_family = AF_INET;  //限定获取IP为IPV4

        if(getaddrinfo(_ip, nullptr, &hints, &res) != 0) //正确返回0
        {
            std::cerr << "hostname error" << std::endl;
            exit(EXIT_FAILURE);
        }

        sockaddr_in* ipv4 = (sockaddr_in*)res->ai_addr;  //转换成sockaddr_in结构 sockaddr->sockaddr_in
        memcpy(&_destAddr, ipv4, sizeof(sockaddr_in));
    }

    void SendData()
    {
        while (1)
        {
            //装包
            struct icmp icmphdr{};  //需要发送的ICMP报文
            icmphdr.icmp_seq = ++_sendSeq;  
            icmphdr.icmp_type = ICMP_ECHO;  //ICMP报文的类型
            // icmphdr.icmp_type = ICMP_TIMESTAMP;      
            icmphdr.icmp_id = _id;      

            auto now = std::chrono::system_clock::now();     // 获取时间戳, 8bit
            memcpy(icmphdr.icmp_data, &now, sizeof(now));    

            icmphdr.icmp_cksum = CheckSum(&icmphdr, sizeof(icmphdr));   // 计算校验和

            if(sendto(_sockfd, &icmphdr, sizeof(icmphdr), 0, (struct sockaddr*)&_destAddr, sizeof(_destAddr)) <= 0)
            {   //发送数据
                std::cout << "send data fail " << _ip << std::endl;
                exit(EXIT_FAILURE);
            }

            std::this_thread::sleep_for(std::chrono::seconds(1));   //每个一秒发送一次
        }
    }

    void RecvData()
    {
        while (1)
        {
            sockaddr_in addr{};
            socklen_t fromLen = sizeof(_destAddr);
            ssize_t n = recvfrom(_sockfd, _recvData, sizeof(_recvData), 0, (sockaddr*)&addr, &fromLen);
            if(n > 0)
            {   
                struct ip* ip_hdr = (struct ip*)_recvData;  
                // 获取ICMP报文位置,IP头部计算为首部字段长度*4;
                struct icmp* icmp_hdr = (struct icmp*)(_recvData + (ip_hdr->ip_hl << 2));   

                if (icmp_hdr->icmp_type == ICMP_ECHOREPLY && icmp_hdr->icmp_id == _id)  //筛选
                {
                    ++_recvSeq;
                    //计算耗时
                    auto now = std::chrono::system_clock::now();
                    auto data = (std::chrono::system_clock::time_point*)icmp_hdr->icmp_data;
                    auto sum = std::chrono::duration_cast<std::chrono::milliseconds>(now - *data).count();

                    std::cout << std::format("{} bytes from {}: icmp_seq={} ttl={} time={}ms",
                        n, inet_ntoa(_destAddr.sin_addr), icmp_hdr->icmp_seq, ip_hdr->ip_ttl, sum) << std::endl;
                }
                // else 
                // {
                //     std::cout << std::format("icmp_type: {}, icmp_ip: {}, icmp_code: {}", icmp_hdr->icmp_type, icmp_hdr->icmp_id, icmp_hdr->icmp_code) << std::endl;
                // }
            }
            else if(n <= 0)
            {
                std::cerr << "Recv fail" << std::endl;
                exit(EXIT_FAILURE);
            }
        }
        
    }

    unsigned short CheckSum(void* data, int len)
    {   
        unsigned short* buf = (unsigned short*)data;
        unsigned sum = 0;

        // 计算数据的和
        while(len > 1)
        {
            sum += *buf++;
            len -= 2;
        }
        if(len == 1)
        {
            sum += *(unsigned char*)buf;
        }

        // 把高16位和低16位相加
        sum = (sum >> 16) + (sum & 0xffff);
        sum += (sum >> 16);
        // 取反
        return (unsigned short)(~sum);
    }



private:
    static std::chrono::system_clock::time_point _oldTime;  
    static int _sendSeq;
    static int _recvSeq;

    unsigned short _id;
    int _sockfd;

    struct sockaddr_in _destAddr;
    const char* _ip;    //需要ping的ip;
    char _recvData[1024];
};

std::chrono::system_clock::time_point PingServer::_oldTime = std::chrono::system_clock::now();
int PingServer::_sendSeq = 0;
int PingServer::_recvSeq = 0;

main函数

#include "Ping.hpp"

//TOOD 初始化

void Usage()
{
    std::cout << "ping <ip/hostname>" << std::endl;
}

int main(int argc, char* argv[])
{
    if(argc != 2)
    {
        Usage();
        return 1;
    }

    signal(SIGINT, [](int sig)  //当使用 ctl+c 时中断程序。
    {
        PingServer::TimeEnd();
        exit(0);
    });

    PingServer ping(argv[1]);

    ping.Start();

    return 0;
}

运行测试

CMakeList

cmake_minimum_required(VERSION 3.29)
project(PingServer)

set(CMAKE_CXX_STANDARD 20)
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

add_executable(test test.cpp
        Ping.hpp
)

运行结果:

在这里插入图片描述

总结

本篇文章实现了一个简易的ping指令,其对系统编程、网络编程都有所涉及,但真实的ping指令可远不止这么简单,感兴趣的读者可以通过访问Linux开源项目来了解真正的实现。

📜博客主页:主页
📫我的专栏:C++
📱我的github:github

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1829692.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

使用百度的长文本转语音API时无法下载.MP3文件

今天是学生们交作业的时候&#xff0c;结果是我最忙碌的一天&#xff0c;各种改bug。 有个学生来问&#xff1a; 我在百度提供的API代码(长文本转语音)的基础上添加了下载生成的.MP3文件的代码&#xff0c;运行之后成功建成了.MP3文件&#xff0c;但是文件的内容确实以下的报错…

监控室,屏幕显示不支持码流

1号屏&#xff0c;出现不支持码流 如下原因 老是录像机 无法关闭自动添加摄像头功能&#xff0c; 其他杂牌摄像头 会自动还ip 最终导致 ip冲突 更换ip 可以解决

openstack删除实例卡死在正在删除中

删除实例 问题描述解决办法 实验环境&#xff1b;服务器&#xff0c;openstackY版 问题描述 openstack在删除实例时一直显示正在删除中 解决办法 进入数据库修改实例状态&#xff0c;修改为错误&#xff0c;然后重新删除 首先查看对应实例id 进入数据库修改 rootcompute:~…

数据库原理(关系型数据库基本理论)——(

一、关系的概念 1.关系的定义 &#xff08;1&#xff09;域 域是一组具有相同数据类型的值的集合&#xff0c;可以理解为int[]&#xff08;int类型的数组&#xff09;是一个域。 &#xff08;2&#xff09;笛卡儿积 简单来说&#xff0c;若干个域的笛卡儿积就是将这几个域的…

算法与数据结构--决策树算法

欢迎来到 Papicatch的博客 文章目录 &#x1f349;决策树算法介绍 &#x1f348;原理 &#x1f348;核心思想包括 &#x1f34d;递归分割 &#x1f34d;选择标准 &#x1f34d;剪枝 &#x1f348;解题过程 &#x1f34d;数据准备 &#x1f34d;选择最佳分割特征 &…

pdf转图片,pdf转图片在线转

pdf转图片的方法&#xff0c;对于许多人来说可能是一个稍显陌生的操作。然而&#xff0c;在日常生活和工作中&#xff0c;我们有时确实需要将pdf文件转换为图片格式&#xff0c;以便于在特定的场合或平台上进行分享、展示或编辑。以下&#xff0c;我们将详细介绍一个pdf转成图片…

【每天学会一个渗透测试工具】dirsearch安装及使用指南

&#x1f31d;博客主页&#xff1a;泥菩萨 &#x1f496;专栏&#xff1a;Linux探索之旅 | 网络安全的神秘世界 | 专接本 | 每天学会一个渗透测试工具 ✨dirsearch介绍 dirsearch安装包百度网盘 disearch是基于Python开发的&#xff0c;因此需要确保你的系统中已经安装了pyth…

linux中acl策略+文件权限

文档归属的局限性 - 任何人只属于三种角色&#xff1a;属主 属组 其他人- 无法实现更精细的控制 acl访问策略 - 能够对个别用户个别组设置独立的权限- 大多数挂载ext3/4,xfs文件系统默认已支持 Usage: setfacl [-bkndRLP] { -m|-M|-x|-X ... } file ...setfacl [选项] u:用户名…

消息队列-分布式消息队列技术选型

Kafka Kafka 是 LinkedIn 开源的一个分布式流式处理平台&#xff0c;已经成为 Apache 顶级项目&#xff0c;早期被用来用于处理海量的日志&#xff0c;后面才慢慢发展成了一款功能全面的高性能消息队列。 流式处理平台具有三个关键功能&#xff1a; 消息队列&#xff1a;发布和…

【C++入门(1)】命名空间

一、C出世 我们先简单认识下C的来历&#xff0c;C是在C语言的基础上发展来的。 当年C的设计者Bjarne Stroustrup&#xff0c;本贾尼斯特劳斯特卢普先生设计C语言之初&#xff0c;是为了对C语言做出一些更改&#xff0c;弥补C语言在一些方面的不足&#xff0c;或者做出其他的设…

JAVA开发 选择多个文件,系统运行后自动生成ZIP压缩包

选择多个文件&#xff0c;系统运行后自动生成ZIP压缩包 实现方法1.1 代码块1.2 运行结果截取 相关知识 实现方法 案例简述&#xff1a;通过启动java代码来打开文件选择器对话框&#xff0c;用户选择确认需要进行压缩的文件&#xff0c;可一次性选择多个文件&#xff0c;选择完…

whisper 模型源码解读

whisper官方源码 whisper 模型官方代码&#xff1a;https://github.com/openai/whisper/blob/main/whisper/model.py &#xff1b;注释如下 import base64 import gzip from dataclasses import dataclass from typing import Dict, Iterable, Optionalimport numpy as np impo…

ffmpeg解封装rtsp并录制视频-(2)使用VLC模拟一个rtsp服务器并用ffmpeg解封装该rtsp流

VCL模拟服务器并打开播放该视频文件&#xff1a; - 准备好一个mp4文件&#xff0c;打开vlc软件 - 选择“媒体”》“流” - 添加一个mp4文件 - 点击下方按钮选择“串流” - 下一步目标选择rtsp 点击“添加” - 端口默认8554 - 路径设置 /test - 用…

XML Encoding = ‘GBK‘ after STRANS,中文乱码

最近帮同事处理了一个中信银行银企直连接口的一个问题,同事反馈,使用STRANS转换XML后,encoding始终是’utf-16’,就算指定了GBK也不行。尝试了很多办法始终不行,发到银行的数据中,中文始终是乱码。 Debug使用HTML视图看报文时也可以看到中文是乱码。 解决方案: 使用cl…

飞腾银河麒麟V10安装Todesk

下载安装包 下载地址 https://www.todesk.com/linux.html 安装 yum makecache yum install libappindicator-gtk3-devel.aarch64 rpm -ivh 下载的安装包文件后台启动 service todeskd start修改配置 编辑 /opt/todesk/config/config.ini 移除自动更新临时密码 passupda…

STM32CubeMX配置-RTC周期唤醒

一、简介 MCU为STM32G070&#xff0c;采用内部时钟32KHZ&#xff0c;配置为周期6s唤醒&#xff0c;调用回调函数&#xff0c;进行喂狗操作。 二、配置 初始时间、日期、周期唤醒时间配置。 开启周期唤醒中断 三、生成代码 调用回调函数&#xff0c;进行喂狗操作。 //RTC唤醒回…

STM32理论 —— μCOS-Ⅲ(2/2):时间管理、消息队列、信号量、任务内嵌信号量/队列、事件标志、软件定时器、内存管理

文章目录 9. 时间管理9.1 OSTimeDly()9.2 OSTimeDlyHMSM()9.3 OSTimeDlyResume()9.4 延时函数实验 10. 消息队列10.1 创建消息队列函数OSQCreate()10.2 发送消息到消息队列函数(写入队列)OSQPost()10.3 获取消息队列中的消息函数(读出队列)OSQPend()10.4 消息队列操作实验 11. …

绿色版DirectoryOpus功能强大且高度可定制的Windows文件管理器

Directory Opus&#xff08;通常简称为DOpus&#xff09;是一款功能强大且高度可定制的Windows文件管理器。它提供了许多超越Windows默认文件资源管理器&#xff08;Explorer&#xff09;的功能&#xff0c;使得文件和文件夹的管理变得更加高效和直观。以下是对Directory Opus的…

Hex-Rays IDA Pro V7安装教程 (交互式反汇编工具)

前言 DA Pro就像提供了一张二进制的地图&#xff0c;标注了系统函数以及分析人员注解的函数调用&#xff0c;同时展现出各级函数和代码块之间的调用关系。此外&#xff0c;IDA Pro的扩展性能很好&#xff0c;可以利用IDA Pro提供的API接口和IDC脚本来扩展应用&#xff0c;而且…