boot: Restrict block IO buf size

2022-04-07 18:45:21 +02:00
parent 7faded269d
commit 5ec9285611
1 changed files with 2 additions and 2 deletions
@@ -2042,10 +2042,10 @@ static EFI_STATUS boot_windows_bitlocker(void) {
        for (UINTN i = 0; i < n_handles; i++) {
                EFI_BLOCK_IO *block_io;
                err = BS->HandleProtocol(handles[i], &BlockIoProtocol, (void **) &block_io);
-                if (EFI_ERROR(err) || block_io->Media->BlockSize < 512)
+                if (EFI_ERROR(err) || block_io->Media->BlockSize < 512 || block_io->Media->BlockSize > 4096)
                        continue;

-                CHAR8 buf[block_io->Media->BlockSize];
+                CHAR8 buf[4096];
                err = block_io->ReadBlocks(block_io, block_io->Media->MediaId, 0, sizeof(buf), buf);
                if (EFI_ERROR(err))
                        continue;