glib/
checksum.rs

1// Copyright 2013-2016, The Gtk-rs Project Developers.
2// See the COPYRIGHT file at the top-level directory of this distribution.
3// Licensed under the MIT license, see the LICENSE file or <http://opensource.org/licenses/MIT>
4
5use glib_sys;
6use libc::size_t;
7use std::vec::Vec;
8use translate::*;
9use Checksum;
10
11impl Checksum {
12    pub fn get_digest(self) -> Vec<u8> {
13        unsafe {
14            //Don't forget update when `ChecksumType` contains type bigger that Sha512.
15            let mut digest_len: size_t = 512 / 8;
16            let mut vec = Vec::with_capacity(digest_len as usize);
17
18            glib_sys::g_checksum_get_digest(
19                mut_override(self.to_glib_none().0),
20                vec.as_mut_ptr(),
21                &mut digest_len,
22            );
23
24            vec.set_len(digest_len);
25            vec
26        }
27    }
28
29    pub fn get_string(self) -> Option<String> {
30        unsafe {
31            from_glib_none(glib_sys::g_checksum_get_string(mut_override(
32                self.to_glib_none().0,
33            )))
34        }
35    }
36}
37
38#[cfg(test)]
39mod tests {
40    use {Checksum, ChecksumType};
41
42    const CS_TYPE: ChecksumType = ChecksumType::Md5;
43    const CS_VALUE: &str = "fc3ff98e8c6a0d3087d515c0473f8677";
44    const CS_SLICE: &[u8] = &[
45        0xfc, 0x3f, 0xf9, 0x8e, 0x8c, 0x6a, 0x0d, 0x30, 0x87, 0xd5, 0x15, 0xc0, 0x47, 0x3f, 0x86,
46        0x77,
47    ];
48
49    #[test]
50    fn update() {
51        let mut cs = Checksum::new(CS_TYPE);
52        cs.update("hello world!".as_bytes());
53        assert_eq!(cs.get_string().unwrap(), CS_VALUE);
54    }
55
56    #[test]
57    fn update_multi_call() {
58        let mut cs = Checksum::new(CS_TYPE);
59        cs.update("hello ".as_bytes());
60        cs.update("world!".as_bytes());
61        assert_eq!(cs.get_string().unwrap(), CS_VALUE);
62    }
63
64    #[test]
65    fn get_digest() {
66        let mut cs = Checksum::new(CS_TYPE);
67        cs.update("hello world!".as_bytes());
68        let vec = cs.get_digest();
69        assert_eq!(vec, CS_SLICE);
70    }
71
72}