[SETUPLIB] FindSupportedSystemPartition(): Add missing NULL check

Addendum to commit 59acff7.
oleg-dubinskiy · Jun 10, 2024 · c752e8c · c752e8c
1 parent ae27ffc
commit c752e8c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/base/setup/lib/utils/partlist.c b/base/setup/lib/utils/partlist.c
@@ -3346,6 +3346,11 @@ FindSupportedSystemPartition(
  * should be our system partition.
  */
  DiskEntry = GetSystemDisk(List);
+ if (!DiskEntry)
+ {
+ /* No system disk found, directly go check the alternative disk */
+ goto UseAlternativeDisk;
+ }
 
  if (DiskEntry->DiskStyle == PARTITION_STYLE_GPT)
  {