Compression state reset, Error message

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);
 }