blake2b.rs

// Copyright 2020 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

use blake2::Blake2b;
use digest::{
    generic_array::typenum::{U20, U32},
    FixedOutput, FixedOutputReset, HashMarker, Output, OutputSizeUser, Reset, Update,
};

// blake2 has [`Blake2s256`] instance but not for 160 bits.
/// Blake2b instance with a 256-bit output.
#[derive(Clone)]
pub struct Blake2b256(Blake2b<U32>);

impl Blake2b256 {
    /// Creates a new [`Blake2b256`] instance.
    pub fn new() -> Self {
        Self(Blake2b::default())
    }
}

impl Default for Blake2b256 {
    fn default() -> Self {
        Self::new()
    }
}

impl OutputSizeUser for Blake2b256 {
    type OutputSize = U32;
}

impl Reset for Blake2b256 {
    fn reset(&mut self) {
        self.0.reset();
    }
}

impl FixedOutput for Blake2b256 {
    fn finalize_into(self, out: &mut Output<Self>) {
        self.0.finalize_into(out);
    }
}

impl FixedOutputReset for Blake2b256 {
    fn finalize_into_reset(&mut self, out: &mut Output<Self>) {
        self.0.finalize_into_reset(out);
    }
}

impl Update for Blake2b256 {
    fn update(&mut self, data: &[u8]) {
        self.0.update(data);
    }
}

impl HashMarker for Blake2b256 {}

/// Blake2b instance with a 160-bit output.
#[derive(Clone)]
pub struct Blake2b160(Blake2b<U20>);

impl Blake2b160 {
    /// Creates a new [`Blake2b160`] instance.
    pub fn new() -> Self {
        Self(Blake2b::default())
    }
}

impl Default for Blake2b160 {
    fn default() -> Self {
        Self::new()
    }
}

impl OutputSizeUser for Blake2b160 {
    type OutputSize = U20;
}

impl Reset for Blake2b160 {
    fn reset(&mut self) {
        self.0.reset();
    }
}

impl FixedOutput for Blake2b160 {
    fn finalize_into(self, out: &mut Output<Self>) {
        self.0.finalize_into(out);
    }
}

impl FixedOutputReset for Blake2b160 {
    fn finalize_into_reset(&mut self, out: &mut Output<Self>) {
        self.0.finalize_into_reset(out);
    }
}

impl Update for Blake2b160 {
    fn update(&mut self, data: &[u8]) {
        self.0.update(data);
    }
}

impl HashMarker for Blake2b160 {}