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
// Take a look at the license at the top of the repository in the LICENSE file.

use crate::translate::*;
use crate::Checksum;
use libc::size_t;
use std::vec::Vec;

impl Checksum {
    /// Gets the digest from `self` as a raw binary vector and places it
    /// into `buffer`. The size of the digest depends on the type of checksum.
    ///
    /// Once this function has been called, the [`Checksum`][crate::Checksum] is closed and can
    /// no longer be updated with [`update()`][Self::update()].
    /// ## `buffer`
    /// output buffer
    #[doc(alias = "g_checksum_get_digest")]
    #[doc(alias = "get_digest")]
    pub fn digest(self) -> Vec<u8> {
        unsafe {
            //Don't forget update when `ChecksumType` contains type bigger that Sha512.
            let mut digest_len: size_t = 512 / 8;
            let mut vec = Vec::with_capacity(digest_len as usize);

            ffi::g_checksum_get_digest(
                mut_override(self.to_glib_none().0),
                vec.as_mut_ptr(),
                &mut digest_len,
            );

            vec.set_len(digest_len);
            vec
        }
    }

    /// Gets the digest as a hexadecimal string.
    ///
    /// Once this function has been called the [`Checksum`][crate::Checksum] can no longer be
    /// updated with [`update()`][Self::update()].
    ///
    /// The hexadecimal characters will be lower case.
    ///
    /// # Returns
    ///
    /// the hexadecimal representation of the checksum. The
    ///  returned string is owned by the checksum and should not be modified
    ///  or freed.
    #[doc(alias = "g_checksum_get_string")]
    #[doc(alias = "get_string")]
    pub fn string(self) -> Option<String> {
        unsafe {
            from_glib_none(ffi::g_checksum_get_string(mut_override(
                self.to_glib_none().0,
            )))
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::{Checksum, ChecksumType};

    const CS_TYPE: ChecksumType = ChecksumType::Md5;
    const CS_VALUE: &str = "fc3ff98e8c6a0d3087d515c0473f8677";
    const CS_SLICE: &[u8] = &[
        0xfc, 0x3f, 0xf9, 0x8e, 0x8c, 0x6a, 0x0d, 0x30, 0x87, 0xd5, 0x15, 0xc0, 0x47, 0x3f, 0x86,
        0x77,
    ];

    #[test]
    fn update() {
        let mut cs = Checksum::new(CS_TYPE).unwrap();
        cs.update(b"hello world!");
        assert_eq!(cs.string().unwrap(), CS_VALUE);
    }

    #[test]
    fn update_multi_call() {
        let mut cs = Checksum::new(CS_TYPE).unwrap();
        cs.update(b"hello ");
        cs.update(b"world!");
        assert_eq!(cs.string().unwrap(), CS_VALUE);
    }

    #[test]
    #[doc(alias = "get_digest")]
    fn digest() {
        let mut cs = Checksum::new(CS_TYPE).unwrap();
        cs.update(b"hello world!");
        let vec = cs.digest();
        assert_eq!(vec, CS_SLICE);
    }
}