boot: add overflow check in GPT parser

ALIGN_TO() can overflow and return SIZE_MAX

CID#1644887

Follow-up for ccbd324a3a522362de0863e8d06cdd06a58d2fca

diff --git a/src/boot/part-discovery.c b/src/boot/part-discovery.c
index dc1aed0514b1d2b3f03898dd2a9dce6aa70c3689..25f60521acf30b4aa5881a0766ec1c826518f15f 100644 (file)
--- a/src/boot/part-discovery.c
+++ b/src/boot/part-discovery.c
@@ -117,6 +117,8 @@ static EFI_STATUS try_gpt(
 
         /* Now load the GPT entry table */
         size = ALIGN_TO((size_t) gpt->SizeOfPartitionEntry * (size_t) gpt->NumberOfPartitionEntries, 512);
+        if (size == SIZE_MAX) /* overflow check */
+                return EFI_OUT_OF_RESOURCES;
         entries_pages = xmalloc_aligned_pages(
                 AllocateMaxAddress,
                 EfiLoaderData,