X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Fcompat%2Fcompat-poll.c;h=bc79eed0aaa5321c48fcd055858698bcdc5fa375;hp=cff9f44b8e17c369a4b015866a825044bb43ef48;hb=ac018a8b4890dcf6f0eb9d4b5dee600269e79b29;hpb=ab30f567fb55cc0336b2e7018295bb23ba90abb5

diff --git a/src/common/compat/compat-poll.c b/src/common/compat/compat-poll.c
index cff9f44b8..bc79eed0a 100644
--- a/src/common/compat/compat-poll.c
+++ b/src/common/compat/compat-poll.c
@@ -40,6 +40,11 @@ static int resize_poll_event(struct compat_poll_event_array *array,
 
 	assert(array);
 
+	/* Refuse to resize the array more than the max size. */
+	if (new_size > poll_max_size) {
+		goto error;
+	}
+
 	ptr = realloc(array->events, new_size * sizeof(*ptr));
 	if (ptr == NULL) {
 		PERROR("realloc epoll add");
@@ -200,11 +205,6 @@ int compat_poll_del(struct lttng_poll_event *events, int fd)
 	/* Ease our life a bit. */
 	current = &events->current;
 
-	/* Safety check on size */
-	if (new_size > poll_max_size) {
-		new_size = poll_max_size;
-	}
-
 	/* Check if we need to shrink it down. */
 	if ((current->nb_fd << 1UL) <= current->alloc_size &&
 			current->nb_fd >= current->init_size) {