使用 Rust 实现连接远程 Linux 服务器、发送文件、执行命令
一、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、服务器文件
