seagul_core 0.1.0

A library to encode (and decode) arbitrary information into images
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use core::panic;
use std::fs::File;

use seagul_core::{decoder::ImageDecoder, prelude::*};
use seagul_core::encoder::ImageEncoder;

fn ensure_out_dir() -> std::io::Result<()> {
    std::fs::create_dir_all("tests/out")
}

#[test]
fn encode_bytes() {
    ensure_out_dir().expect("Could not create output directory");

    let verses = b"Midway upon the journey of our life
I found myself within a forest dark,
For the straightforward pathway had been lost.
Ah me! how hard a thing it is to say
What was this forest savage, rough, and stern,
Which in the very thought renews the fear.
So bitter is it, death is little more;
But of the good to treat, which there I found,
Speak will I of the other things I saw there.
I cannot well repeat how there I entered,
So full was I of slumber at the moment
In which I had abandoned the true way.--";

    let encode_result = ImageEncoder::from("tests/images/red_panda.jpg")
        .set_offset(0)
        .set_use_n_lsb(2)
        .encode_bytes(verses);

    if let Err(e) = encode_result {
        panic!("{}", e.as_str());
    }

    encode_result
        .unwrap()
        .save("tests/out/red_panda_steg.png", ImageFormat::Png)
        .expect("Could not create output file");

    let mut created_image =
        File::open("tests/out/red_panda_steg.png").expect("Failed to open created image");

    let decoded = ImageDecoder::from(&mut created_image)
        .set_offset(0)
        .set_use_n_lsb(2)
        .until_marker(Some(b"--"))
        .decode();

    assert!(decoded.is_ok());

    let decoded = decoded.unwrap();
    let decoded_string = decoded.as_raw();

    println!("Raw decoded:\n{}", decoded_string);

    assert_eq!(decoded.hit_marker(), true);
}

#[test]
fn encode_bytes_spread() {
    ensure_out_dir().expect("Could not create output directory");

    let verses = b"Midway upon the journey of our life
I found myself within a forest dark,
For the straightforward pathway had been lost.
Ah me! how hard a thing it is to say
What was this forest savage, rough, and stern,
Which in the very thought renews the fear.
So bitter is it, death is little more;
But of the good to treat, which there I found,
Speak will I of the other things I saw there.
I cannot well repeat how there I entered,
So full was I of slumber at the moment
In which I had abandoned the true way.--";

    let encode_result = ImageEncoder::from("tests/images/red_panda.jpg")
        .set_offset(0)
        .set_spread(true)
        .set_use_n_lsb(2)
        .encode_bytes(verses);

    if let Err(e) = encode_result {
        panic!("{}", e.as_str());
    }

    encode_result
        .unwrap()
        .save("tests/out/red_panda_spread.png", ImageFormat::Png)
        .expect("Could not create output file");

    let mut created_image =
        File::open("tests/out/red_panda_spread.png").expect("Failed to open created image");

    let decoded = ImageDecoder::from(&mut created_image)
        .set_offset(0)
        .set_use_n_lsb(2)
        .decode();

    assert!(decoded.is_ok());

    let decoded = decoded.unwrap();
    let decoded_string = decoded.as_raw();

    println!("Raw decoded:\n{}", decoded_string);

    assert_eq!(decoded.hit_marker(), false);
}