Fixed null pointer dereference issue in treeblock

radareorg · Feb 12, 2024 · 8419d7d · 8419d7d
1 parent 1ee768a
commit 8419d7d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/shlr/grub/fs/sfs.c b/shlr/grub/fs/sfs.c
@@ -159,6 +159,9 @@ grub_sfs_read_extent (struct grub_sfs_data *data, unsigned int block,
     return 0;
 
   treeblock = grub_malloc (data->blocksize);
+  if(!treeblock){
+    return grub_error(GRUB_ERR_OUT_OF_MEMORY, "Failed to allocate memory for treeblock");
+  }
 
   next = grub_be_to_cpu32 (data->rblock.btree);
   tree = (struct grub_sfs_btree *) treeblock;