利用Rust实现一个简单的Ping应用

目录
  • 目标
    • 命令行解析
    • 实现Ping
    • 周期性发送
    • 其他
  • 验证
  • 总结

这两年Rust火的一塌糊涂,甚至都烧到了前端,再不学习怕是要落伍了。最近翻了翻文档,写了个简单的Ping应用练练手,被所有权折腾的够呛,相比起Golang上手难度大很多,现将开发中的一些问题总结如下,所有源码见ring

目标

实现一个Ping,功能包含:

命令行解析

实现ICMP协议,pnet包中已经包含了ICMP包定义,可以使用socket2库发送

周期性发送Ping,通过多线程发送,再汇总结果

监听退出信号

命令行解析

系统库std::env::args可以解析命令行参数,但对于一些复杂的参数使用起来比较繁琐,更推荐clap。利用clap的注解,通过结构体定义命令行参数

/// ping but with rust, rust + ping -> ring
#[derive(Parser, Debug, Clone)] // Parser生成clap命令行解析方法
#[command(author, version, about, long_about = None)]
pub struct Args {
    /// Count of ping times
    #[arg(short, default_value_t = 4)] // short表示开启短命名,默认为第一个字母,可以指定;default_value_t设置默认值
    count: u16,

    /// Ping packet size
    #[arg(short = 's', default_value_t = 64)]
    packet_size: usize,

    /// Ping ttl
    #[arg(short = 't', default_value_t = 64)]
    ttl: u32,

    /// Ping timeout seconds
    #[arg(short = 'w', default_value_t = 1)]
    timeout: u64,

    /// Ping interval duration milliseconds
    #[arg(short = 'i', default_value_t = 1000)]
    interval: u64,

    /// Ping destination, ip or domain
    #[arg(value_parser=Address::parse)] // 自定义解析
    destination: Address,
}

clap可以方便的指定参数命名、默认值、解析方法等,运行结果如下

➜  ring git:(main) cargo run -- -h
   Compiling ring v0.1.0 (/home/i551329/work/ring)
    Finished dev [unoptimized + debuginfo] target(s) in 1.72s
     Running `target/debug/ring -h`
ping but with rust, rust + ping -> ring

Usage: ring [OPTIONS] <DESTINATION>

Arguments:
  <DESTINATION>  Ping destination, ip or domain

Options:
  -c <COUNT>            Count of ping times [default: 4]
  -s <PACKET_SIZE>      Ping packet size [default: 64]
  -t <TTL>              Ping ttl [default: 64]
  -w <TIMEOUT>          Ping timeout seconds [default: 1]
  -i <INTERVAL>         Ping interval duration milliseconds [default: 1000]
  -h, --help            Print help information
  -V, --version         Print version information

实现Ping

pnet中提供了ICMP包的定义,socket2可以将定义好的ICMP包发送给目标IP,另一种实现是通过pnet_transport::transport_channel发送原始数据包,但需要过滤结果而且权限要求较高。

首先定义ICMP包

let mut buf = vec![0; self.config.packet_size];
let mut icmp = MutableEchoRequestPacket::new(&mut buf[..]).ok_or(RingError::InvalidBufferSize)?;
icmp.set_icmp_type(IcmpTypes::EchoRequest); // 设置为EchoRequest类型
icmp.set_icmp_code(IcmpCodes::NoCode);
icmp.set_sequence_number(self.config.sequence + seq_offset); // 序列号
icmp.set_identifier(self.config.id);
icmp.set_checksum(util::checksum(icmp.packet(), 1)); // 校验函数

通过socket2发送请求

let socket = Socket::new(Domain::IPV4, Type::DGRAM, Some(Protocol::ICMPV4))?;
let src = SocketAddr::new(net::IpAddr::V4(Ipv4Addr::UNSPECIFIED), 0);
socket.bind(&src.into())?; // 绑定源地址
socket.set_ttl(config.ttl)?;
socket.set_read_timeout(Some(Duration::from_secs(config.timeout)))?; // 超时配置
socket.set_write_timeout(Some(Duration::from_secs(config.timeout)))?;

​​​​​​​// 发送
socket.send_to(icmp.packet_mut(), &self.dest.into())?;

最后处理相应,转换成pnet中的EchoReplyPacket

let mut mem_buf = unsafe { &mut *(buf.as_mut_slice() as *mut [u8] as *mut [std::mem::MaybeUninit<u8>]) };
let (size, _) = self.socket.recv_from(&mut mem_buf)?;

​​​​​​​// 转换成EchoReply
let reply = EchoReplyPacket::new(&buf).ok_or(RingError::InvalidPacket)?;

至此,一次Ping请求完成。

周期性发送

Ping需要周期性的发送请求,比如秒秒请求一次,如果直接通过循环实现,一次请求卡住将影响主流程,必须通过多线程来保证固定周期的发送。

发送请求

let send = Arc::new(AtomicU64::new(0)); // 统计发送次数
let _send = send.clone();
let this = Arc::new(self.clone());
let (sx, rx) = bounded(this.config.count as usize); // channel接受线程handler
thread::spawn(move || {
    for i in 0..this.config.count {
        let _this = this.clone();
        sx.send(thread::spawn(move || _this.ping(i))).unwrap(); // 线程中运行ping,并将handler发送到channel中

        _send.fetch_add(1, Ordering::SeqCst); // 发送一次,send加1

        if i < this.config.count - 1 {
            thread::sleep(Duration::from_millis(this.config.interval));
        }
    }
    drop(sx); // 发送完成关闭channel
});
  • thread::spawn可以快速创建线程,但需要注意所有权的转移,如果在线程内部调用方法获取变量,需要通过Arc原子引用计数
  • send变量用来统计发送数,原子类型,并且用Arc包裹;this是当前类的Arc克隆,会转移到线程中
  • 第一个线程内周期性调用ping(),并且其在单独线程中运行
  • 通过bounded来定义channel(类似Golang中的chan),用来处理结果,发送完成关闭

处理结果

let success = Arc::new(AtomicU64::new(0)); // 定义请求成功的请求
let _success = success.clone();
let (summary_s, summary_r) = bounded(1); // channel来判断是否处理完成
thread::spawn(move || {
    for handle in rx.iter() {
        if let Some(res) = handle.join().ok() {
            if res.is_ok() {
                _success.fetch_add(1, Ordering::SeqCst); // 如果handler结果正常,success加1
            }
        }
    }
    summary_s.send(()).unwrap(); // 处理完成
});

第二个线程用来统计结果,channel通道取出ping线程的handler,如果返回正常则加1

处理信号

let stop = signal_notify()?; // 监听退出信号
select!(
    recv(stop) -> sig => {
        if let Some(s) = sig.ok() { // 收到退出信号
            println!("Receive signal {:?}", s);
        }
    },
    recv(summary_r) -> summary => { // 任务完成
        if let Some(e) = summary.err() {
            println!("Error on summary: {}", e);
        }
    },
);

通过select来处理信号(类似Golang中的select),到收到退出信号或者任务完成时继续往下执行。

信号处理

Golang中可以很方便的处理信号,但在Rust中官方库没有提供类似功能,可以通过signal_hook与crossbeam_channel实现监听退出信号

fn signal_notify() -> std::io::Result<Receiver<i32>> {
    let (s, r) = bounded(1); // 定义channel,用来异步接受退出信号

    let mut signals = signal_hook::iterator::Signals::new(&[SIGINT, SIGTERM])?; // 创建信号

    thread::spawn(move || {
        for signal in signals.forever() { // 如果结果到信号发送到channel中
            s.send(signal).unwrap();
            break;
        }
    });

    Ok(r) // 返回接受channel
}

其他

很多吐槽人Golang的错误处理,Rust也不遑多让,不过提供了?语法糖,也可以配合anyhow与thiserror来简化错误处理。

验证

Ping域名/IP

ring git:(main)  cargo run -- www.baidu.com

PING www.baidu.com(103.235.46.40)
64 bytes from 103.235.46.40: icmp_seq=1 ttl=64 time=255.85ms
64 bytes from 103.235.46.40: icmp_seq=2 ttl=64 time=254.17ms
64 bytes from 103.235.46.40: icmp_seq=3 ttl=64 time=255.41ms
64 bytes from 103.235.46.40: icmp_seq=4 ttl=64 time=256.50ms

--- www.baidu.com ping statistics ---
4 packets transmitted, 4 received, 0% packet loss, time 3257.921ms

测试退出信息,运行中通过Ctrl+C中止

cargo run 8.8.8.8 -c 10

PING 8.8.8.8(8.8.8.8)
64 bytes from 8.8.8.8: icmp_seq=1 ttl=64 time=4.32ms
64 bytes from 8.8.8.8: icmp_seq=2 ttl=64 time=3.02ms
64 bytes from 8.8.8.8: icmp_seq=3 ttl=64 time=3.24ms
^CReceive signal 2

​​​​​​​--- 8.8.8.8 ping statistics ---
3 packets transmitted, 3 received, 0% packet loss, time 2365.104ms

总结

Rust为了安全高效,通过引入所有权来解决GC问题,也带来了许多不便,编程时必须要考虑到变量的声明周期、借用等问题,所有语言都是在方便、性能、安全之间做权衡,要么程序员不方便,要么编译器多做点功。换一个角度来说Bug总是不可避免的,在编译阶段出现总好过运行阶段。

到此这篇关于利用Rust实现一个简单的Ping应用的文章就介绍到这了,更多相关Rust实现Ping内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们!

(0)

相关推荐

  • vbs ping实现的两种方式

    对于vbs中ping的实现可以通过两种方式 : 1.调用系统ping命令: 2.使用wmi查询pingstate类处理. 1.调用系统ping命令 Set wshell = CreateObject("WScript.Shell") wshell.run("ping 182.183.101.1",0.true) 对于以上调用,如果想对其进行过滤,可以考虑将运行结果重定向到文件,在读到一个string中,查找其中是否有timeout或超时字符,判断是否超时.本打算直接

  • VBS 批量Ping的项目实现

    本文用vb编写的 ping程序实现,具体如下: '判断当前VBS脚本是否由CScript执行 If InStr(LCase(WScript.FullName), "cscript.exe") = 0 Then     '若不是由CScript执行,则使用CScript重新执行当前脚本     Set objShell = CreateObject("Shell.Application")      objShell.ShellExecute "cscrip

  • C++代码实现网络Ping功能

    目录 (一)main.cpp文件 (二)ping.h文件 (1)IP头结构体: (2)ICMP头结构体: (3)ICMP响应报文结构体: (4)Ping类及相关变量的定义: (三)ping.cpp文件 (1)char *m_szICMPData; BOOL m_bIsInitSucc; (2)BOOL Ping(char *szDestIP, PingReply *pPingReply = NULL, DWORD dwTimeout = 2000); (3)BOOL PingCore(DWORD

  • Python利用pythonping处理ping的示例详解

    目录 简介 历史攻略 Ping命令可以进行以下操作 安装 案例源码1 案例源码2 简介 ping (Packet Internet Groper)是一种因特网包探索器,用于测试网络连接量的程序 .Ping是工作在 TCP/IP网络体系结构中应用层的一个服务命令, 主要是向特定的目的主机发送 ICMP(Internet Control Message Protocol 因特网报文控制协议)Echo 请求报文,测试目的站是否可达及了解其有关状态.ping用于确定本地主机是否能与另一台主机成功交换(发

  • 利用Rust实现一个简单的Ping应用

    目录 目标 命令行解析 实现Ping 周期性发送 其他 验证 总结 这两年Rust火的一塌糊涂,甚至都烧到了前端,再不学习怕是要落伍了.最近翻了翻文档,写了个简单的Ping应用练练手,被所有权折腾的够呛,相比起Golang上手难度大很多,现将开发中的一些问题总结如下,所有源码见ring. 目标 实现一个Ping,功能包含: 命令行解析 实现ICMP协议,pnet包中已经包含了ICMP包定义,可以使用socket2库发送 周期性发送Ping,通过多线程发送,再汇总结果 监听退出信号 命令行解析 系

  • 利用Rust编写一个简单的字符串时钟

    目录 1.简介 2.用到的知识点 2.1 取utc时间 2.2 图片变换为像素图案 2.3 字符方式显示当前时间 2.4 时间刷新 1.简介 用rust写的一个简单的练手的demo,一个字符串时钟,在终端用字符串方式显示当前时间.本质是对图片取灰度,然后每个像素按灰度门限用星号代替灰度值,就把图片变为由星号组成的字符型图案.把时间字符串的每个字符按照字母和数字图片的样式转换为字符,然后拼接字符图案就实现了字符时钟的效果. 主要用到的知识有:rust操作时间.字符串.vector,字符串和vect

  • C# 利用VS编写一个简单的网游客户端

    目录 一.测试连接服务器 二.设计客户端 三.运行效果 四.总结 一.测试连接服务器 1.打开cmd,输入ping 10.1.230.74 2.输入telnet,进入telnet界面 3.输入set localecho,打开本地回显: 4.连接服务器,输入命令open 10.1.230.74 3900 二.设计客户端 1.新建项目 打开VS2022选择新建Windows窗体应用 如果没找到,说明没有安装相应的配置,可以添加工具 选择 安装成功就能找到了. 2.设计界面 新建完成后,会直接来到Fo

  • 利用jQuery实现一个简单的表格上下翻页效果

    前言 本文主要介绍的是利用jQuery实现一个简单的表格上下翻页效果,注:实现原理与轮播图相似.下面话不多说,来看看详细的 实现方法吧. html: <div class="popup day02-popup04"> <div class="group-caption"> <span>日期</span><span>参与团购场次</span><span class="result&

  • python实现一个简单的ping工具方法

    继上一篇计算checksum校验和,本章通过socket套接字,struct字节打包成二进制,select返回套接字的文件描述符的结合,实现一个简单的ping工具. #!/usr/bin/python3.6.4 #!coding:utf-8 __author__ = 'Rosefinch' __date__ = '2018/5/31 22:27' import time import struct import socket import select import sys def chesks

  • 利用c++写一个简单的推箱子小游戏

    效果图 相信各位都肯定完整这种推箱子的小游戏.游戏玩法很简单,那就是一个人把所有的箱子推动到对应的位置那就可以赢了. 那么我们接下来看看这个推箱子的游戏改怎么写 char map[10][10]= { {'#','#','#','#','#','#','#','#','#','#'}, {'#','#','#','#',' ',' ','!',' ',' ','#'}, {'#',' ',' ',' ',' ','o',' ',' ',' ','#'}, {'#',' ',' ',' ',' '

  • 利用JavaScript写一个简单计算器

    效果如下: 参考程序: <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <meta http-equiv="X-UA-Compatible" content="IE=edge"> <meta name="viewport" content="width=d

  • 利用Python制作一个简单的天气播报系统

    目录 前言 工具 天气数据来源 代码实现 总结 前言 大家好,我是辣条 相信大家都能感觉到最近天气的多变,好几次出门半路天气转变.辣条也深受其扰,直接给我整感冒,就差被隔离起来了,既然天气我没法做主,那不如用python整个天气爬虫来获取天气情况.这样也好可以进行一个提前预防 工具 python3.7 pycharm pyttsx3:语音播报库 天气数据来源 找寻一个天气网站 比如说我们要查询某地的天气,在输入地名后就能看到结果. 我们可以看到网站的url会有变化: 每个城市的天气信息url就是

  • 如何利用PyQt5制作一个简单的登录界面

    目录 环境配置 额外工具配置 生成UI界面 总结 环境配置 新建python虚拟环境并激活 conda create -n pyqt python=3.8 conda activate py36 安装pyqt5 pip install pyqt5 安装pyqt5-tools pip install pyqt5-tools 在PyCharm中新建一个qtdemo工程,并使用这个新建的python虚拟环境作为工程环境 额外工具配置 依次点击File---Settings---Tools---Exte

  • 利用Python实现一个简单的Web汇率计算器

    目录 Dash是什么 网页搭建步骤 安装相关依赖(库) 导入相关包 构建app 构建结果输出函数 网页结构Layout搭建 callback回调参数设定 界面效果 前段时间刚接触到前端网页开发,但是对于刚入门的小白而言,像flask.Django等这类稍大型的框架确实不太适合,今天这个Dash是集众家之长于一体的轻量化Web开发库. Dash是什么 Dash 是一个用于构建基于 Web 的应用程序的 Python 库,无需 JavaScript . Dash 同时也是用于创建分析 Web 应用程

随机推荐