diff options
Diffstat (limited to 'src/message_digest.c')
| -rw-r--r-- | src/message_digest.c | 21 |
1 files changed, 16 insertions, 5 deletions
diff --git a/src/message_digest.c b/src/message_digest.c index 24c1db5..c696780 100644 --- a/src/message_digest.c +++ b/src/message_digest.c @@ -6,7 +6,7 @@ /* By: charles <me@cacharle.xyz> +#+ +:+ +#+ */ /* +#+#+#+#+#+ +#+ */ /* Created: 2020/08/01 15:03:39 by charles #+# #+# */ -/* Updated: 2020/08/02 11:56:27 by charles ### ########.fr */ +/* Updated: 2020/08/02 13:56:26 by charles ### ########.fr */ /* */ /* ************************************************************************** */ @@ -17,10 +17,18 @@ char *message_digest(t_message_digest_param *param, uint8_t *message_origin, uin size_t i; uint64_t padding_size; uint8_t *message; + void *local_state; + char *ret; + local_state = ft_memdup(param->compression_state_init, param->compression_state_size); + if (local_state == NULL) + return (NULL); padding_size = (param->chunk_size - (size + sizeof(uint64_t)) % param->chunk_size) + sizeof(uint64_t); if ((message = malloc(size + padding_size)) == NULL) + { + free(local_state); return (NULL); + } ft_memcpy(message, message_origin, size); ft_memset(message + size, 0x0, padding_size); message[size] |= 1 << 7; @@ -29,14 +37,17 @@ char *message_digest(t_message_digest_param *param, uint8_t *message_origin, uin i = 0; while (i < size) { - param->compression_state = param->compression_func(param->compression_state, message + i); - if (param->compression_state == NULL) + local_state = param->compression_func(local_state, message + i); + if (local_state == NULL) { free(message); + free(local_state); return (NULL); } i += param->chunk_size; } - // FIXME reset state - return (bytes_to_str(param->compression_state, param->compression_state_size)); + free(message); + ret = bytes_to_str(local_state, param->compression_state_size); + free(local_state); + return (ret); } |