Marshal/Unmarshal: do no marshal unset bytes in the chunk

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

1 files changed, 2 insertions, 2 deletions

diff --git a/marshal.c b/marshal.c
index 75a8749..cae5b20 100644
--- a/marshal.c
+++ b/marshal.c
@@ -50,8 +50,8 @@ libsha2_marshal(const struct libsha2_state *restrict state, char *restrict buf)
 		*(size_t *)&buf[off] = state->chunk_size;
 	off += sizeof(size_t);
 	if (buf)
-		memcpy(&buf[off], state->chunk, state->chunk_size);
-	off += state->chunk_size;
+		memcpy(&buf[off], state->chunk, state->message_size % state->chunk_size);
+	off += state->message_size % state->chunk_size;
 
 	return off;
 }