🔒️ fix malloc protection

froz42 · Nov 4, 2023 · f3b52aa · f3b52aa
1 parent 3869bfa
commit f3b52aa
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/srcs/syscall/param_log/log_string.c b/srcs/syscall/param_log/log_string.c
@@ -22,6 +22,11 @@ static void buffer_add_char(buffer_t *buffer, char c)
 	{
 		buffer->size_buffer *= 2;
 		buffer->buffer = realloc(buffer->buffer, buffer->size_buffer);
+		if (buffer->buffer == NULL)
+		{
+			log_error("log_STRING", "realloc failed", true);
+			return;
+		}
 	}
 	buffer->buffer[buffer->index++] = c;
 }
@@ -43,6 +48,11 @@ int log_remote_string(pid_t pid, char *remote_str, int max_size)
 		.size_buffer = DEFAULT_BUFFER_SIZE,
 		.index = 0,
 	};
+	if (buffer.buffer == NULL)
+	{
+		log_error("log_STRING", "malloc failed", true);
+		return 0;
+	}
 	char c = 1; // dummy value that will be overwritten by the first read
 	while (c != '\0')
 	{