写了个优选IP程序

niconiconi

写了个优选IP程序，实测50W个IP ping 4次取平均值，获取延迟最低前十只需要几秒钟，会动手的mjj自己编译，刚开始学有点糙

下载地址 https://drive.kamibook.com/raw/ipopt.7z

./ipopt.exe ipv4 ip.txt

ip.txt内容 示例 1.1.1.1/24  一行一个

1. use std::{
   
2.     result,
   
3.     env,
   
4.     net::IpAddr,
   
5.     time::Duration,
   
6.     io::BufRead,
   
7.     collections::HashMap,
   
8. };
   
9. use futures::future::join_all;
   
10. use rand::random;
    
11. use surge_ping::{Client, Config, IcmpPacket, PingIdentifier, PingSequence, ICMP};
    
12. use tokio::time;
    
13. use ipnet::{Ipv4Net, Ipv6Net};
    
14. 
    
15. #[tokio::main]
    
16. async fn main() -> Result<(), Box<dyn std::error::Error>> {
    
17.     let args: Vec<String> = env::args().collect();
    
18.     let ips = get_ip_from_file(&args[2]).await?;
    
19.     let ipaddrs = get_ip_range(&args[1], ips).await?;
    
20.     let client_v4 = Client::new(&Config::default())?;
    
21.     let client_v6 = Client::new(&Config::builder().kind(ICMP::V6).build())?;
    
22.     let mut tasks = Vec::new();
    
23. 
    
24.     for ip in &ipaddrs {
    
25.         match ip.parse() {
    
26.             Ok(IpAddr::V4(addr)) => {
    
27.                 tasks.push(tokio::spawn(ping(client_v4.clone(), IpAddr::V4(addr))))
    
28. 
    
29.             }
    
30.             Ok(IpAddr::V6(addr)) => {
    
31.                 tasks.push(tokio::spawn(ping(client_v6.clone(), IpAddr::V6(addr))))
    
32.             }
    
33.             Err(e) => println!("{} parse to ipaddr error: {}", ip, e),
    
34.         }
    
35.     }
    
36.     let results: Vec<_> = join_all(tasks).await.into_iter().filter_map(Result::ok).collect();
    
37.     let mut ip_map = HashMap::new();
    
38. 
    
39.     for result in results {
    
40.         match result {
    
41.             Ok((ip, delay)) => {
    
42.                 ip_map.insert(ip, delay);
    
43.             }
    
44.             Err(e) => {
    
45.                 eprintln!("Error occurred: {:?}", e);
    
46.             }
    
47.         }
    
48.     }
    
49.     let mut top_10: Vec<_> = ip_map.into_iter()
    
50.     .filter(|(_, delay)| delay.as_millis() > 0)
    
51.     .collect::<Vec<_>>();
    
52.     top_10.sort_by(|(_, delay1), (_, delay2)| delay1.cmp(delay2));
    
53.     let top_10 = top_10.into_iter().take(10).collect::<Vec<_>>();
    
54. 
    
55.     for (ip, delay) in top_10 {
    
56.         println!("{} {:?}", ip, delay.as_millis());
    
57.     }
    
58.     Ok(())
    
59. }
    
60. async fn ping(client: Client, addr: IpAddr) -> Result<(IpAddr, Duration), Box<dyn std::error::Error + Send + 'static>> {    let payload = [0; 56];
    
61.     let mut pinger = client.pinger(addr, PingIdentifier(random())).await;
    
62.     let mut interval = time::interval(Duration::from_secs(1));
    
63.     let mut delays = Vec::new();
    
64. 
    
65.     for idx in 0..4 {
    
66.         interval.tick().await;
    
67.         match pinger.ping(PingSequence(idx), &payload).await {
    
68.             Ok((IcmpPacket::V4(_packet), dur)) => {
    
69.                 delays.push(dur);
    
70.             }
    
71.             Ok((IcmpPacket::V6(_packet), dur)) => {
    
72.                 delays.push(dur);
    
73.             }
    
74.             Err(_e) => {
    
75.             }
    
76.         };
    
77.     }
    
78.     let average_delay = if !delays.is_empty() {
    
79.         let total: Duration = delays.iter().sum();
    
80.         total / delays.len() as u32
    
81.     } else {
    
82.         Duration::new(0, 0)
    
83.     };
    
84.     Ok((addr, average_delay))
    
85. }
    
86. 
    
87. async fn get_ip_from_file(file_path: &str) -> Result<Vec<String>, Box<dyn std::error::Error>> {
    
88.     let mut ips = Vec::new();
    
89.     let file = std::fs::File::open(file_path)?;
    
90.     let mut buf_reader = std::io::BufReader::new(file);
    
91.     let mut line = String::new();
    
92.    
    
93.     while buf_reader.read_line(&mut line).unwrap() > 0 {
    
94.         let ip = line.trim();
    
95.         ips.push(ip.to_string());
    
96.         line.clear();
    
97.     }
    
98.     Ok(ips)
    
99. }
    
100. 
     
101. 
     
102. async fn get_ip_range(ip_type: &str, ips: Vec<String>) -> result::Result<Vec<String>, Box<dyn std::error::Error>> {
     
103.     let mut ip_subnets = Vec::new();
     
104. 
     
105.     match ip_type {
     
106.         "ipv4" => {
     
107.             for ip in ips {
     
108.                 let net: Ipv4Net = ip.parse().unwrap();
     
109.                 let subnets = net.subnets(32).expect("PrefixLenError: new prefix length cannot be shorter than existing");
     
110.                 for (_i, n) in subnets.enumerate() {
     
111.                     ip_subnets.push(n.to_string());
     
112.                 }
     
113.             }
     
114.         }
     
115.         "ipv6" => {
     
116.             for ip in ips {
     
117.                 let net: Ipv6Net = ip.parse().unwrap();
     
118.                 let subnets = net.subnets(128).expect("PrefixLenError: new prefix length cannot be shorter than existing");
     
119.                 for (_i, n) in subnets.enumerate() {
     
120.                     ip_subnets.push(n.to_string());
     
121.                 }
     
122.             }
     
123.         }
     
124.         _ => {
     
125.             return Err(Box::new(std::io::Error::new(std::io::ErrorKind::InvalidInput, "Invalid IP type")));
     
126.         }
     
127.     }
     
128.     let ip_range: Vec<String> =  ip_subnets.into_iter().map(|subnet: String| subnet.split('/').next().unwrap().to_string()).collect();
     
129.     Ok(ip_range)
     
130. }

复制代码

网盘

大佬怎么用呢？  

easonxnuw

不是有7Z吗 ，下载下来用啊

niconiconi

由于多线程并发，会创建大量任务，这也是为什么在几秒钟内完成，所以cpu 内存会占用很大，50W占用内存800M

skysf

loc还有会rust的，罕见

niconiconi

skysf 发表于 2024-11-29 00:00
loc还有会rust的，罕见

loc人才济济

HOH

skysf 发表于 2024-11-29 00:00
loc还有会rust的，罕见

没什么罕见的，rust跟其他一样都是用库而已，你说看到.cpp .c那才叫罕见

弟弟不爱吃饭

编译不通过

niconiconi

弟弟不爱吃饭 发表于 2024-11-29 00:30
编译不通过

错误截图

虎谷

loc果然高手如云啊