Add support for extended hash + add support for output hash to custom buffer when supported

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 9 insertions, 4 deletions

diff --git a/libhashsum_init_blake512_hasher.c b/libhashsum_init_blake512_hasher.c
index a42b429..6407a7b 100644
--- a/libhashsum_init_blake512_hasher.c
+++ b/libhashsum_init_blake512_hasher.c
@@ -27,13 +27,15 @@ finalise_const(struct libhashsum_hasher *this, const void *data, size_t bytes, u
 	m = &m[r];
 	bytes -= r;
 
+	if (!this->hash_output)
+		this->hash_output = this->state.blake512.buf;
+
 	memcpy(this->state.blake512.buf, m, bytes + (size_t)(extra_bits > 0U));
 	if (extra_bits)
 		this->state.blake512.buf[bytes] = libhashsum_reverse_byte__(this->state.blake512.buf[bytes]);
 	libblake_blake512_digest(&this->state.blake512.s, this->state.blake512.buf, bytes,
-	                         extra_bits, NULL, this->state.blake512.buf);
+	                         extra_bits, NULL, this->hash_output);
 	memset(&this->state.blake512.s, 0, sizeof(this->state.blake512.s));
-	this->hash_output = this->state.blake512.buf;
 	return 0;
 }
 
@@ -55,15 +57,17 @@ finalise(struct libhashsum_hasher *this, void *data, size_t bytes, unsigned extr
 	bytes -= r;
 	size -= r;
 
+	if (!this->hash_output)
+		this->hash_output = this->state.blake512.buf;
+
 	if (size < libblake_blake512_digest_get_required_input_size(bytes, extra_bits, NULL)) {
 		memcpy(this->state.blake512.buf, m, bytes + (size_t)(extra_bits > 0U));
 		m = this->state.blake512.buf;
 	}
 	if (extra_bits)
 		m[bytes] = libhashsum_reverse_byte__(m[bytes]);
-	libblake_blake512_digest(&this->state.blake512.s, m, bytes, extra_bits, NULL, this->state.blake512.buf);
+	libblake_blake512_digest(&this->state.blake512.s, m, bytes, extra_bits, NULL, this->hash_output);
 	memset(&this->state.blake512.s, 0, sizeof(this->state.blake512.s));
-	this->hash_output = this->state.blake512.buf;
 	return 0;
 }
 
@@ -104,6 +108,7 @@ libhashsum_init_blake512_hasher(struct libhashsum_hasher *this, const void *salt
 	this->process = &process;
 	this->finalise_const = &finalise_const;
 	this->finalise = &finalise;
+	this->stretch = NULL;
 	this->destroy = NULL;
 	libblake_blake512_init2(&this->state.blake512.s, salt);
 	return 0;