Added argument parsing (file/stdin digest)

1 files changed, 11 insertions, 16 deletions

diff --git a/src/message_digest.c b/src/message_digest.c
index b08022f..24c1db5 100644
--- a/src/message_digest.c
+++ b/src/message_digest.c
@@ -6,33 +6,28 @@
 /*   By: charles <me@cacharle.xyz>                  +#+  +:+       +#+        */
 /*                                                +#+#+#+#+#+   +#+           */
 /*   Created: 2020/08/01 15:03:39 by charles           #+#    #+#             */
-/*   Updated: 2020/08/01 19:25:07 by charles          ###   ########.fr       */
+/*   Updated: 2020/08/02 11:56:27 by charles          ###   ########.fr       */
 /*                                                                            */
 /* ************************************************************************** */
 
 #include "ft_ssl.h"
 
-char	*message_digest(t_message_digest_param *param)
+char	*message_digest(t_message_digest_param *param, uint8_t *message_origin, uint64_t size)
 {
 	size_t		i;
 	uint64_t	padding_size;
 	uint8_t		*message;
 
-	padding_size = (64 - (param->size + 8) % 64) + 8;
-
-	message = malloc(param->size + padding_size);
-	if (message == NULL)
+	padding_size = (param->chunk_size - (size + sizeof(uint64_t)) % param->chunk_size) + sizeof(uint64_t);
+	if ((message = malloc(size + padding_size)) == NULL)
 		return (NULL);
-	ft_memcpy(message, param->message, param->size);
-	ft_memset(message + param->size, 0x0, padding_size);
-
-	message[param->size] |= 1 << 7;
-	*((uint64_t*)(message + param->size + padding_size - sizeof(uint64_t))) =
-		(uint64_t)param->size * 8;
-	param->size += padding_size;
-
+	ft_memcpy(message, message_origin, size);
+	ft_memset(message + size, 0x0, padding_size);
+	message[size] |= 1 << 7;
+	*((uint64_t*)(message + size + padding_size - sizeof(uint64_t))) = size * 8;
+	size += padding_size;
 	i = 0;
-	while (i < param->size)
+	while (i < size)
 	{
 		param->compression_state = param->compression_func(param->compression_state, message + i);
 		if (param->compression_state == NULL)
@@ -42,6 +37,6 @@ char	*message_digest(t_message_digest_param *param)
 		}
 		i += param->chunk_size;
 	}
-
+	// FIXME reset state
 	return (bytes_to_str(param->compression_state, param->compression_state_size));
 }