mla 2.0.0

Multi Layer Archive - A pure rust encrypted and compressed archive file format
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

use sha2::{Digest, Sha256, Sha512};
use std::io::Read;
use std::io::{self, Write};

pub(crate) type Sha256Hash = [u8; 32];

pub(crate) struct HashWrapperReader<'a, R: Read, H = Sha256> {
    /// Wrapper over a `impl Read` updating `hash` on each call to `read`
    inner: R,
    hash: &'a mut H,
}

impl<'a, R: Read, H> HashWrapperReader<'a, R, H> {
    pub(crate) fn new(inner: R, hash: &'a mut H) -> Self {
        Self { inner, hash }
    }

    pub(crate) fn into_inner(self) -> R {
        self.inner
    }
}

impl<R: Read, H: Digest> Read for HashWrapperReader<'_, R, H> {
    /// Wrapper on inner with hash update
    fn read(&mut self, into: &mut [u8]) -> io::Result<usize> {
        let read = self.inner.read(into)?;
        self.hash.update(&into[..read]);
        Ok(read)
    }
}

pub(crate) struct HashWrapperWriter<'a, W, H = Sha512> {
    /// Wrapper over a `impl Write` updating `hash` on each call to `write`
    inner: W,
    hash: &'a mut H,
}

impl<'a, W, H> HashWrapperWriter<'a, W, H> {
    pub(crate) fn new(inner: W, hash: &'a mut H) -> Self {
        Self { inner, hash }
    }

    pub(crate) fn into_inner(self) -> W {
        self.inner
    }
}

impl<W: Write, H: Digest> Write for HashWrapperWriter<'_, W, H> {
    /// Wrapper on inner with hash update
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        self.hash.update(buf);
        self.inner.write(buf)
    }

    fn flush(&mut self) -> io::Result<()> {
        self.inner.flush()
    }
}