stegano 0.1.1

The ultimate steganography swiss knife army CLI tool.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use clap::Parser;
use crc32_v2::byfour::crc32_little;
use std::fs::File;
use std::io::Write;
use stegano::cli::{Cli, SteganoCommands};
use stegano::jpeg::utils::read_jpeg_headers;
use stegano::models::MetaChunk;
use stegano::utils::{encrypt_payload, xor_encrypt_decrypt};

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let args = Cli::parse();

    // Run the CLI.
    match args.command {
        Some(command) => match command {
            SteganoCommands::Encrypt(encrypt_cmd) => {
                let mut file = File::open(encrypt_cmd.input.clone())?;

                let mut meta_chunk = MetaChunk::new(&mut file, encrypt_cmd.suppress)
                    .expect("Error processing the png file!");

                let mut file_writer = File::create(encrypt_cmd.output.clone())?;
                let encrypted_data: Vec<u8> = match (*encrypt_cmd.algorithm.to_lowercase()).into() {
                    "aes" => encrypt_payload(&encrypt_cmd.key, &encrypt_cmd.payload),
                    "xor" => xor_encrypt_decrypt(encrypt_cmd.payload.as_bytes(), &encrypt_cmd.key),
                    _ => {
                        return Err("Unsupported algorithm!".into());
                    }
                };
                // Calculate CRC for the encrypted data
                let mut bytes_msb = Vec::new();
                bytes_msb
                    .write_all(&meta_chunk.chk.r#type.to_be_bytes())
                    .unwrap();
                bytes_msb.write_all(&encrypted_data).unwrap();
                let crc = crc32_little(meta_chunk.chk.crc, &bytes_msb);

                // Update the MetaChunk with the encrypted data and CRC
                meta_chunk.chk.data = encrypted_data.clone();
                meta_chunk.chk.crc = crc;

                // Create a new mutable reference to file_reader
                let mut file_reader = &file;

                meta_chunk.write_encrypted_data(&mut file_reader, &encrypt_cmd, &mut file_writer);
            }
            SteganoCommands::Decrypt(decrypt_cmd) => {
                let mut file = File::open(decrypt_cmd.input.clone())?;

                let mut meta_chunk = MetaChunk::new(&mut file, decrypt_cmd.suppress)
                    .expect("Error processing the png file!");

                let mut file_writer = File::create(decrypt_cmd.output.clone()).unwrap();
                let mut file_reader = &file;
                meta_chunk.write_decrypted_data(&mut file_reader, &decrypt_cmd, &mut file_writer);
            }
            SteganoCommands::ShowMeta(show_meta_cmd) => {
                if show_meta_cmd.r#type.to_lowercase() == "jpeg" {
                    let _ = read_jpeg_headers(
                        &show_meta_cmd.input.clone(),
                        show_meta_cmd.start_chunk,
                        show_meta_cmd.end_chunk,
                        show_meta_cmd.nb_chunks,
                    );
                } else if show_meta_cmd.r#type.to_lowercase() == "png" {
                    let mut file = File::open(show_meta_cmd.input.clone())?;
                    let mut meta_chunk = MetaChunk::new(&mut file, show_meta_cmd.suppress)
                        .expect("Error processing the png file!");
                    meta_chunk.process_image(&mut file, &show_meta_cmd);
                }
                return Ok(());
            }
        },
        None => println!("\x1b[1;91mUnknown command. Use 'help' for usage instructions.\x1b[0m"),
    }
    Ok(())
}