怎么把文件藏进图片里？Rust隐写工具开发实战分享

// 将每个数据字节隐藏在图片的最低有效位中
for (i, &byte) in data_with_len.iter().enumerate() {
    for bit in 0..8 {
        let bit_value = (byte >> bit) & 1;
        let pixel_index = i * 8 + bit;
        
        // 修改像素的最低有效位
        image_data[pixel_index] = (image_data[pixel_index] & 0xFE) | bit_value;
    }
}

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}
​
#[derive(Subcommand)]
enum Commands {
    /// 隐藏文件或文本到图片中
    Hide {
        /// 要隐藏的文件路径
        #[arg(short, long)]
        file: Option<String>,
​
        /// 要隐藏的文本
        #[arg(short, long)]
        text: Option<String>,
​
        /// 原始图片路径
        #[arg(short, long)]
        image: String,
​
        /// 输出图片路径
        #[arg(short, long)]
        output: String,
    },
    /// 从图片中提取隐藏的数据
    Extract {
        /// 包含隐藏数据的图片路径
        #[arg(short, long)]
        image: String,
​
        /// 输出文件路径（如果提取的是文件）
        #[arg(short, long)]
        output: Option<String>,
    },
    /// 启动Web服务
    Web {
        /// 监听地址
        #[arg(short, long, default_value = "127.0.0.1:3000")]
        bind: String,
    },
}

let app = Router::new()
    .route("/", get(root))
    .route("/api/hide-file", post(hide_file_handler))
    .route("/api/hide-text", post(hide_text_handler))
    .route("/api/extract", post(extract_handler))
    .nest_service("/static", ServeDir::new("static"));

// 确保输出路径以.png结尾以使用无损格式
let output_path = if !output_path.ends_with(".png") {
    let mut path = output_path.to_string();
    if let Some(dot_index) = path.rfind('.') {
        path.truncate(dot_index);
    }
    path.push_str(".png");
    path
} else {
    output_path.to_string()
};

let total_data_len = data.len() + 4; // 数据长度(4字节) + 实际数据
if total_data_len * 8 > image_data.len() {
    return Err("图片太小，无法容纳要隐藏的数据".into());
}

let output_file = if let Some(path) = output_path {
    path.to_string()
} else {
    format!("temp/{}", file_name)
};

$ cargo run -- hide -f secret.txt -i original.png -o output.png

$ cargo run -- extract -i output.png

$ cargo run -- hide -t "这是秘密消息" -i original.png -o output.png

$ cargo run -- extract -i output.png