chug-cli 0.1.3

The fastest way to consume Homebrew bottles
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

use std::io;

use ring::digest::{self, SHA256};

pub struct Validate<R> {
    inner: R,
    digest_context: digest::Context,
    sha256: Vec<u8>,
}

impl<R: io::Read> Validate<R> {
    pub(crate) fn new(inner: R, sha256: Vec<u8>) -> Self {
        Validate {
            inner,
            digest_context: digest::Context::new(&SHA256),
            sha256,
        }
    }

    pub fn validate(self) -> anyhow::Result<()> {
        let checksum = self.digest_context.finish();
        anyhow::ensure!(
            checksum.as_ref() == self.sha256.as_slice(),
            "Checksum mismatch",
        );
        Ok(())
    }
}

impl<R: io::Read> io::Read for Validate<R> {
    fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
        let len = self.inner.read(buf)?;
        self.digest_context.update(&buf[..len]);
        Ok(len)
    }
}