364 lines
10 KiB
Rust
364 lines
10 KiB
Rust
// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
|
|
// An implementation of the Blake2b cryptographic hash function.
|
|
// The implementation closely follows: https://tools.ietf.org/html/rfc7693
|
|
//
|
|
// "BLAKE2 is a cryptographic hash function faster than MD5, SHA-1, SHA-2, and
|
|
// SHA-3, yet is at least as secure as the latest standard SHA-3."
|
|
// according to their own website :)
|
|
//
|
|
// Indeed this implementation is two to three times as fast as our SHA-256
|
|
// implementation. If you have the luxury of being able to use crates from
|
|
// crates.io, you can go there and find still faster implementations.
|
|
|
|
use std::mem;
|
|
use std::slice;
|
|
|
|
#[repr(C)]
|
|
struct Blake2bCtx {
|
|
b: [u8; 128],
|
|
h: [u64; 8],
|
|
t: [u64; 2],
|
|
c: usize,
|
|
outlen: u16,
|
|
finalized: bool,
|
|
|
|
#[cfg(debug_assertions)]
|
|
fnv_hash: u64,
|
|
}
|
|
|
|
#[cfg(debug_assertions)]
|
|
impl ::std::fmt::Debug for Blake2bCtx {
|
|
fn fmt(&self, fmt: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
|
|
write!(fmt, "{:x}", self.fnv_hash)
|
|
}
|
|
}
|
|
|
|
#[cfg(not(debug_assertions))]
|
|
impl ::std::fmt::Debug for Blake2bCtx {
|
|
fn fmt(&self, fmt: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
|
|
write!(fmt, "Enable debug_assertions() for more info.")
|
|
}
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn b2b_g(v: &mut [u64; 16],
|
|
a: usize,
|
|
b: usize,
|
|
c: usize,
|
|
d: usize,
|
|
x: u64,
|
|
y: u64)
|
|
{
|
|
v[a] = v[a].wrapping_add(v[b]).wrapping_add(x);
|
|
v[d] = (v[d] ^ v[a]).rotate_right(32);
|
|
v[c] = v[c].wrapping_add(v[d]);
|
|
v[b] = (v[b] ^ v[c]).rotate_right(24);
|
|
v[a] = v[a].wrapping_add(v[b]).wrapping_add(y);
|
|
v[d] = (v[d] ^ v[a]).rotate_right(16);
|
|
v[c] = v[c].wrapping_add(v[d]);
|
|
v[b] = (v[b] ^ v[c]).rotate_right(63);
|
|
}
|
|
|
|
// Initialization vector
|
|
const BLAKE2B_IV: [u64; 8] = [
|
|
0x6A09E667F3BCC908, 0xBB67AE8584CAA73B,
|
|
0x3C6EF372FE94F82B, 0xA54FF53A5F1D36F1,
|
|
0x510E527FADE682D1, 0x9B05688C2B3E6C1F,
|
|
0x1F83D9ABFB41BD6B, 0x5BE0CD19137E2179
|
|
];
|
|
|
|
fn blake2b_compress(ctx: &mut Blake2bCtx, last: bool) {
|
|
|
|
const SIGMA: [[usize; 16]; 12] = [
|
|
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 ],
|
|
[14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3 ],
|
|
[11, 8, 12, 0, 5, 2, 15, 13, 10, 14, 3, 6, 7, 1, 9, 4 ],
|
|
[7, 9, 3, 1, 13, 12, 11, 14, 2, 6, 5, 10, 4, 0, 15, 8 ],
|
|
[9, 0, 5, 7, 2, 4, 10, 15, 14, 1, 11, 12, 6, 8, 3, 13 ],
|
|
[2, 12, 6, 10, 0, 11, 8, 3, 4, 13, 7, 5, 15, 14, 1, 9 ],
|
|
[12, 5, 1, 15, 14, 13, 4, 10, 0, 7, 6, 3, 9, 2, 8, 11 ],
|
|
[13, 11, 7, 14, 12, 1, 3, 9, 5, 0, 15, 4, 8, 6, 2, 10 ],
|
|
[6, 15, 14, 9, 11, 3, 0, 8, 12, 2, 13, 7, 1, 4, 10, 5 ],
|
|
[10, 2, 8, 4, 7, 6, 1, 5, 15, 11, 9, 14, 3, 12, 13, 0 ],
|
|
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 ],
|
|
[14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3 ]
|
|
];
|
|
|
|
let mut v: [u64; 16] = [
|
|
ctx.h[0],
|
|
ctx.h[1],
|
|
ctx.h[2],
|
|
ctx.h[3],
|
|
ctx.h[4],
|
|
ctx.h[5],
|
|
ctx.h[6],
|
|
ctx.h[7],
|
|
|
|
BLAKE2B_IV[0],
|
|
BLAKE2B_IV[1],
|
|
BLAKE2B_IV[2],
|
|
BLAKE2B_IV[3],
|
|
BLAKE2B_IV[4],
|
|
BLAKE2B_IV[5],
|
|
BLAKE2B_IV[6],
|
|
BLAKE2B_IV[7],
|
|
];
|
|
|
|
v[12] ^= ctx.t[0]; // low 64 bits of offset
|
|
v[13] ^= ctx.t[1]; // high 64 bits
|
|
if last {
|
|
v[14] = !v[14];
|
|
}
|
|
|
|
{
|
|
// Re-interpret the input buffer in the state as an array
|
|
// of little-endian u64s, converting them to machine
|
|
// endianness. It's OK to modify the buffer in place
|
|
// since this is the last time this data will be accessed
|
|
// before it's overwritten.
|
|
|
|
let m: &mut [u64; 16] = unsafe {
|
|
let b: &mut [u8; 128] = &mut ctx.b;
|
|
::std::mem::transmute(b)
|
|
};
|
|
|
|
if cfg!(target_endian = "big") {
|
|
for word in &mut m[..] {
|
|
*word = u64::from_le(*word);
|
|
}
|
|
}
|
|
|
|
for i in 0 .. 12 {
|
|
b2b_g(&mut v, 0, 4, 8, 12, m[SIGMA[i][ 0]], m[SIGMA[i][ 1]]);
|
|
b2b_g(&mut v, 1, 5, 9, 13, m[SIGMA[i][ 2]], m[SIGMA[i][ 3]]);
|
|
b2b_g(&mut v, 2, 6, 10, 14, m[SIGMA[i][ 4]], m[SIGMA[i][ 5]]);
|
|
b2b_g(&mut v, 3, 7, 11, 15, m[SIGMA[i][ 6]], m[SIGMA[i][ 7]]);
|
|
b2b_g(&mut v, 0, 5, 10, 15, m[SIGMA[i][ 8]], m[SIGMA[i][ 9]]);
|
|
b2b_g(&mut v, 1, 6, 11, 12, m[SIGMA[i][10]], m[SIGMA[i][11]]);
|
|
b2b_g(&mut v, 2, 7, 8, 13, m[SIGMA[i][12]], m[SIGMA[i][13]]);
|
|
b2b_g(&mut v, 3, 4, 9, 14, m[SIGMA[i][14]], m[SIGMA[i][15]]);
|
|
}
|
|
}
|
|
|
|
for i in 0 .. 8 {
|
|
ctx.h[i] ^= v[i] ^ v[i + 8];
|
|
}
|
|
}
|
|
|
|
fn blake2b_new(outlen: usize, key: &[u8]) -> Blake2bCtx {
|
|
assert!(outlen > 0 && outlen <= 64 && key.len() <= 64);
|
|
|
|
let mut ctx = Blake2bCtx {
|
|
b: [0; 128],
|
|
h: BLAKE2B_IV,
|
|
t: [0; 2],
|
|
c: 0,
|
|
outlen: outlen as u16,
|
|
finalized: false,
|
|
|
|
#[cfg(debug_assertions)]
|
|
fnv_hash: 0xcbf29ce484222325,
|
|
};
|
|
|
|
ctx.h[0] ^= 0x01010000 ^ ((key.len() << 8) as u64) ^ (outlen as u64);
|
|
|
|
if key.len() > 0 {
|
|
blake2b_update(&mut ctx, key);
|
|
ctx.c = ctx.b.len();
|
|
}
|
|
|
|
ctx
|
|
}
|
|
|
|
fn blake2b_update(ctx: &mut Blake2bCtx, mut data: &[u8]) {
|
|
assert!(!ctx.finalized, "Blake2bCtx already finalized");
|
|
|
|
let mut bytes_to_copy = data.len();
|
|
let mut space_in_buffer = ctx.b.len() - ctx.c;
|
|
|
|
while bytes_to_copy > space_in_buffer {
|
|
checked_mem_copy(data, &mut ctx.b[ctx.c .. ], space_in_buffer);
|
|
|
|
ctx.t[0] = ctx.t[0].wrapping_add(ctx.b.len() as u64);
|
|
if ctx.t[0] < (ctx.b.len() as u64) {
|
|
ctx.t[1] += 1;
|
|
}
|
|
blake2b_compress(ctx, false);
|
|
ctx.c = 0;
|
|
|
|
data = &data[space_in_buffer .. ];
|
|
bytes_to_copy -= space_in_buffer;
|
|
space_in_buffer = ctx.b.len();
|
|
}
|
|
|
|
if bytes_to_copy > 0 {
|
|
checked_mem_copy(data, &mut ctx.b[ctx.c .. ], bytes_to_copy);
|
|
ctx.c += bytes_to_copy;
|
|
}
|
|
|
|
#[cfg(debug_assertions)]
|
|
{
|
|
// compute additional FNV hash for simpler to read debug output
|
|
const MAGIC_PRIME: u64 = 0x00000100000001b3;
|
|
|
|
for &byte in data {
|
|
ctx.fnv_hash = (ctx.fnv_hash ^ byte as u64).wrapping_mul(MAGIC_PRIME);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn blake2b_final(ctx: &mut Blake2bCtx)
|
|
{
|
|
assert!(!ctx.finalized, "Blake2bCtx already finalized");
|
|
|
|
ctx.t[0] = ctx.t[0].wrapping_add(ctx.c as u64);
|
|
if ctx.t[0] < ctx.c as u64 {
|
|
ctx.t[1] += 1;
|
|
}
|
|
|
|
while ctx.c < 128 {
|
|
ctx.b[ctx.c] = 0;
|
|
ctx.c += 1;
|
|
}
|
|
|
|
blake2b_compress(ctx, true);
|
|
|
|
// Modify our buffer to little-endian format as it will be read
|
|
// as a byte array. It's OK to modify the buffer in place since
|
|
// this is the last time this data will be accessed.
|
|
if cfg!(target_endian = "big") {
|
|
for word in &mut ctx.h {
|
|
*word = word.to_le();
|
|
}
|
|
}
|
|
|
|
ctx.finalized = true;
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn checked_mem_copy<T1, T2>(from: &[T1], to: &mut [T2], byte_count: usize) {
|
|
let from_size = from.len() * mem::size_of::<T1>();
|
|
let to_size = to.len() * mem::size_of::<T2>();
|
|
assert!(from_size >= byte_count);
|
|
assert!(to_size >= byte_count);
|
|
let from_byte_ptr = from.as_ptr() as * const u8;
|
|
let to_byte_ptr = to.as_mut_ptr() as * mut u8;
|
|
unsafe {
|
|
::std::ptr::copy_nonoverlapping(from_byte_ptr, to_byte_ptr, byte_count);
|
|
}
|
|
}
|
|
|
|
pub fn blake2b(out: &mut [u8], key: &[u8], data: &[u8])
|
|
{
|
|
let mut ctx = blake2b_new(out.len(), key);
|
|
blake2b_update(&mut ctx, data);
|
|
blake2b_final(&mut ctx);
|
|
checked_mem_copy(&ctx.h, out, ctx.outlen as usize);
|
|
}
|
|
|
|
pub struct Blake2bHasher(Blake2bCtx);
|
|
|
|
impl ::std::hash::Hasher for Blake2bHasher {
|
|
fn write(&mut self, bytes: &[u8]) {
|
|
blake2b_update(&mut self.0, bytes);
|
|
}
|
|
|
|
fn finish(&self) -> u64 {
|
|
assert!(self.0.outlen == 8,
|
|
"Hasher initialized with incompatible output length");
|
|
u64::from_le(self.0.h[0])
|
|
}
|
|
}
|
|
|
|
impl Blake2bHasher {
|
|
pub fn new(outlen: usize, key: &[u8]) -> Blake2bHasher {
|
|
Blake2bHasher(blake2b_new(outlen, key))
|
|
}
|
|
|
|
pub fn finalize(&mut self) -> &[u8] {
|
|
if !self.0.finalized {
|
|
blake2b_final(&mut self.0);
|
|
}
|
|
debug_assert!(mem::size_of_val(&self.0.h) >= self.0.outlen as usize);
|
|
let raw_ptr = (&self.0.h[..]).as_ptr() as * const u8;
|
|
unsafe {
|
|
slice::from_raw_parts(raw_ptr, self.0.outlen as usize)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ::std::fmt::Debug for Blake2bHasher {
|
|
fn fmt(&self, fmt: &mut ::std::fmt::Formatter) -> Result<(), ::std::fmt::Error> {
|
|
write!(fmt, "{:?}", self.0)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
fn selftest_seq(out: &mut [u8], seed: u32)
|
|
{
|
|
let mut a: u32 = 0xDEAD4BADu32.wrapping_mul(seed);
|
|
let mut b: u32 = 1;
|
|
|
|
for i in 0 .. out.len() {
|
|
let t: u32 = a.wrapping_add(b);
|
|
a = b;
|
|
b = t;
|
|
out[i] = ((t >> 24) & 0xFF) as u8;
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn blake2b_selftest()
|
|
{
|
|
use std::hash::Hasher;
|
|
|
|
// grand hash of hash results
|
|
const BLAKE2B_RES: [u8; 32] = [
|
|
0xC2, 0x3A, 0x78, 0x00, 0xD9, 0x81, 0x23, 0xBD,
|
|
0x10, 0xF5, 0x06, 0xC6, 0x1E, 0x29, 0xDA, 0x56,
|
|
0x03, 0xD7, 0x63, 0xB8, 0xBB, 0xAD, 0x2E, 0x73,
|
|
0x7F, 0x5E, 0x76, 0x5A, 0x7B, 0xCC, 0xD4, 0x75
|
|
];
|
|
|
|
// parameter sets
|
|
const B2B_MD_LEN: [usize; 4] = [20, 32, 48, 64];
|
|
const B2B_IN_LEN: [usize; 6] = [0, 3, 128, 129, 255, 1024];
|
|
|
|
let mut data = [0u8; 1024];
|
|
let mut md = [0u8; 64];
|
|
let mut key = [0u8; 64];
|
|
|
|
let mut hasher = Blake2bHasher::new(32, &[]);
|
|
|
|
for i in 0 .. 4 {
|
|
let outlen = B2B_MD_LEN[i];
|
|
for j in 0 .. 6 {
|
|
let inlen = B2B_IN_LEN[j];
|
|
|
|
selftest_seq(&mut data[.. inlen], inlen as u32); // unkeyed hash
|
|
blake2b(&mut md[.. outlen], &[], &data[.. inlen]);
|
|
hasher.write(&md[.. outlen]); // hash the hash
|
|
|
|
selftest_seq(&mut key[0 .. outlen], outlen as u32); // keyed hash
|
|
blake2b(&mut md[.. outlen], &key[.. outlen], &data[.. inlen]);
|
|
hasher.write(&md[.. outlen]); // hash the hash
|
|
}
|
|
}
|
|
|
|
// compute and compare the hash of hashes
|
|
let md = hasher.finalize();
|
|
for i in 0 .. 32 {
|
|
assert_eq!(md[i], BLAKE2B_RES[i]);
|
|
}
|
|
}
|