openethereum/ethash/src/keccak.rs

// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of OpenEthereum.

// OpenEthereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// OpenEthereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with OpenEthereum.  If not, see <http://www.gnu.org/licenses/>.

extern crate keccak_hash as hash;

pub type H256 = [u8; 32];

pub mod keccak_512 {
    use super::hash;

    pub use self::hash::keccak_512_unchecked as unchecked;

    pub fn write(input: &[u8], output: &mut [u8]) {
        hash::keccak_512(input, output);
    }

    pub fn inplace(input: &mut [u8]) {
        // This is safe since `keccak_*` uses an internal buffer and copies the result to the output. This
        // means that we can reuse the input buffer for both input and output.
        unsafe {
            hash::keccak_512_unchecked(
                input.as_mut_ptr(),
                input.len(),
                input.as_ptr(),
                input.len(),
            );
        }
    }
}

pub mod keccak_256 {
    use super::hash;

    pub use self::hash::keccak_256_unchecked as unchecked;

    #[allow(dead_code)]
    pub fn write(input: &[u8], output: &mut [u8]) {
        hash::keccak_256(input, output);
    }

    pub fn inplace(input: &mut [u8]) {
        // This is safe since `keccak_*` uses an internal buffer and copies the result to the output. This
        // means that we can reuse the input buffer for both input and output.
        unsafe {
            hash::keccak_256_unchecked(
                input.as_mut_ptr(),
                input.len(),
                input.as_ptr(),
                input.len(),
            );
        }
    }
}