使用 Rust 实现连接远程 Linux 服务器、发送文件、执行命令

以你之姓@ 2023-10-13 13:53 15阅读 0赞

# 使用 Rust 实现连接远程 Linux 服务器、发送文件、执行命令 #

### 文章目录 ###

*  使用 Rust 实现连接远程 Linux 服务器、发送文件、执行命令
 *  一、Rust 概述
 *   *   *  使用场景
         *  优点
         *  缺点
 *  二、功能实现
 *   *  1、代码
     *  2、运行日志
     *  3、服务器文件

# 一、Rust 概述 #

> Rust 已经听了无数遍，我很清楚它很强！但我也刚了解到它比较难学！那么加上 ChatGPT 呢？下面的代码就是在 ChatGPT 加持下写出来的，我只起到了辅助作用！只能说它们都很厉害！

Rust（又称为 Rust programming language）是一种系统编程语言，主要用于开发底层系统和应用程序，如操作系统、网络协议、编译器等。

### 使用场景 ###

*  系统编程：Rust 适用于编写操作系统、网络协议、驱动程序等底层系统程序。
 *  并行编程：Rust 的并发模型和内存管理机制使其成为并行编程的优秀选择。
 *  嵌入式系统：Rust 的性能和可靠性使其在嵌入式系统中得到广泛应用。
 *  安全性：Rust 提供了强大的安全功能，如内存安全、防止数据竞争等，使其成为编写安全关键应用程序的选择。优缺点：

### 优点 ###

*  内存安全：Rust 提供了内存安全保证，避免了常见的内存泄漏、越界访问等问题。
 *  高性能：Rust 的性能接近于 C++，使得其适用于对性能要求较高的场景。
 *  并行性：Rust 的并发编程能力强大，适合于开发多线程应用程序。
 *  可靠性：Rust 的错误处理机制和内存管理机制使得应用程序更加可靠。
 *  简洁性：Rust 的语法简洁明了，易于学习。

### 缺点 ###

*  学习曲线：**Rust 的语法和概念较为新颖**，对于熟悉其他编程语言的人来说，可能需要一些时间来适应。
 *  库生态系统：尽管 Rust 的标准库生态系统较为完善，但在某些特定领域的库可能还不够成熟。
 *  静态类型系统：虽然静态类型有助于增强代码的可维护性和可读性，但也可能会限制代码的灵活性。

# 二、功能实现 #

## 1、代码 ##

> 依赖 ssh2  
> ssh2 = “0.9.4”

#![allow(unused_assignments)] // 禁用未使用赋值的警告
    
    use std::io::{
        Read, Write};
    use std::path::Path;
    use ssh2::Session;
    
    const PROGRESS_UPDATE_INTERVAL: usize = 1024 * 1024; // 每传输 1MB 打印一次进度信息
    
    fn main() {
        
        // 连接远程服务器
        let tcp = std::net::TcpStream::connect("222.222.222.222:22").unwrap(); // 连接到远程服务器
        let mut sess = Session::new().unwrap(); // 创建一个新的会话
        sess.set_tcp_stream(tcp); // 设置会话的 TCP 流
        sess.handshake().unwrap(); // 进行握手
        sess.userauth_password("root", "222222222").unwrap(); // 使用用户名和密码进行身份验证
    
        // 执行远程命令
        let mut channel = sess.channel_session().unwrap(); // 创建一个新的会话通道
        channel.exec("pwd").unwrap(); // 执行命令
        let mut output = Vec::new(); // 创建一个空的字节向量
        channel.read_to_end(&mut output).unwrap(); // 读取命令输出
        println!("{}", String::from_utf8_lossy(&output)); // 打印命令输出
    
        // 传输文件
        let local_file_path = "C:\\Users\\Administrator\\Desktop\\app.jar"; // 本地文件路径
        let remote_file_path = "/home/zibo/app.jar"; // 远程文件路径
        let file_size = get_file_size(local_file_path); // 获取文件大小
        let mut channel = sess.scp_send(Path::new(remote_file_path), 0o644, file_size, None).unwrap(); // 创建一个新的 SCP 通道
        let mut file = std::fs::File::open(local_file_path).unwrap(); // 打开本地文件
        let mut buffer = Vec::new(); // 创建一个空的字节向量
        file.read_to_end(&mut buffer).unwrap(); // 读取文件内容
    
        let mut total_bytes_sent = 0; // 已发送的总字节数
        let mut total_mb = (file_size as f64) / (1024.0 * 1024.0); // 文件总大小（MB）
        total_mb = (total_mb * 100.0).round() / 100.0; // 保留2位小数
        let mut transferred_mb = 0.0; // 已传输的文件大小（MB）
    
        // 记录已发送文件的大小
        let mut bytes_sent = 0; // 已发送的字节数
        for (i, chunk) in buffer.chunks(PROGRESS_UPDATE_INTERVAL).enumerate() {
         // 循环发送文件内容
            // 循环发送，发完为止！
            while bytes_sent < chunk.len() {
        
                let result = channel.write(chunk).unwrap(); // 发送文件内容
                bytes_sent += result; // 更新已发送的字节数
            }
            total_bytes_sent += bytes_sent; // 更新已发送的总字节数
            bytes_sent = 0; // 重置已发送的字节数
            transferred_mb = (total_bytes_sent as f64) / (1024.0 * 1024.0); // 更新已传输的文件大小（MB）
            transferred_mb = (transferred_mb * 100.0).round() / 100.0; // 保留2位小数
    
            if (i + 1) * PROGRESS_UPDATE_INTERVAL <= buffer.len() {
         // 如果还有剩余的文件内容需要发送
                let progress = (total_bytes_sent as f64) / (file_size as f64) * 100.0; // 计算传输进度
                println!("进度: {:.2}% ({:.2} MB / {:.2} MB)", progress, transferred_mb, total_mb); // 打印传输进度信息
            } else {
         // 文件传输完毕
                println!("进度: 100% 文件传输完毕！");
            }
        }
        channel.send_eof().unwrap(); // 发送 EOF 标志
    }
    
    // 获取文件大小
    fn get_file_size(file_path: &str) -> u64 {
        
        std::fs::metadata(file_path) // 获取文件元数据
            .map(|metadata| metadata.len()) // 获取文件大小
            .unwrap_or(0) // 如果获取失败，则返回 0
    }

## 2、运行日志 ##

C:/Users/Administrator/.cargo/bin/cargo.exe run --color=always --package hello-rust --bin hello-rust
        Finished dev [unoptimized + debuginfo] target(s) in 0.03s
         Running `target\debug\hello-rust.exe`
    /root
    
    进度: 23.10% (1.03 MB / 4.46 MB)
    进度: 46.20% (2.06 MB / 4.46 MB)
    进度: 69.30% (3.09 MB / 4.46 MB)
    进度: 92.40% (4.12 MB / 4.46 MB)
    进度: 100% 文件传输完毕！
    
    Process finished with exit code 0

## 3、服务器文件 ##

![image-20230708204644799][]

[image-20230708204644799]: https://img-blog.csdnimg.cn/img_convert/9a0b500efbcb23a1944902c84480b929.png