mirror of
				https://git.proxmox.com/git/grub2
				synced 2025-11-04 12:55:16 +00:00 
			
		
		
		
	This fixes a performance issue when pc & gpt partmap iterators
        didn't abort iteration even after our hook found what it was
        looking for (often causing expensive probes of non-existant drives).
        Some callers relied on previous buggy behaviour, since they would
        rise an error when their own hooks caused early abortion of its
        iteration.
        * kern/device.c (grub_device_open): Improve error message.
        * disk/lvm.c (grub_lvm_open): Likewise.
        * disk/raid.c (grub_raid_open): Likewise.
        * partmap/pc.c (pc_partition_map_iterate): Abort parent iteration
        when hook requests it, independently of grub_errno.
        (pc_partition_map_probe): Do not fail when find_func() caused
        early abortion of pc_partition_map_iterate().
        * partmap/gpt.c (gpt_partition_map_iterate): Abort parent iteration
        when hook requests it, independently of grub_errno.
        (gpt_partition_map_probe): Do not fail when find_func() caused
        early abortion of gpt_partition_map_iterate().
        * kern/partition.c (grub_partition_iterate): Abort parent iteration
        when hook requests it, independently of grub_errno.  Do not fail when
        part_map_iterate_hook() caused early abortion of p->iterate().
        * util/biosdisk.c (grub_util_biosdisk_get_grub_dev): Do not fail
        when grub_partition_iterate() returned with non-zero.
		
	
			
		
			
				
	
	
		
			140 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			140 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* device.c - device manager */
 | 
						|
/*
 | 
						|
 *  GRUB  --  GRand Unified Bootloader
 | 
						|
 *  Copyright (C) 2002,2005,2007,2008  Free Software Foundation, Inc.
 | 
						|
 *
 | 
						|
 *  GRUB is free software: you can redistribute it and/or modify
 | 
						|
 *  it under the terms of the GNU General Public License as published by
 | 
						|
 *  the Free Software Foundation, either version 3 of the License, or
 | 
						|
 *  (at your option) any later version.
 | 
						|
 *
 | 
						|
 *  GRUB is distributed in the hope that it will be useful,
 | 
						|
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
 *  GNU General Public License for more details.
 | 
						|
 *
 | 
						|
 *  You should have received a copy of the GNU General Public License
 | 
						|
 *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
#include <grub/device.h>
 | 
						|
#include <grub/disk.h>
 | 
						|
#include <grub/net.h>
 | 
						|
#include <grub/fs.h>
 | 
						|
#include <grub/mm.h>
 | 
						|
#include <grub/misc.h>
 | 
						|
#include <grub/env.h>
 | 
						|
#include <grub/partition.h>
 | 
						|
 | 
						|
grub_device_t
 | 
						|
grub_device_open (const char *name)
 | 
						|
{
 | 
						|
  grub_disk_t disk = 0;
 | 
						|
  grub_device_t dev = 0;
 | 
						|
 | 
						|
  if (! name)
 | 
						|
    {
 | 
						|
      name = grub_env_get ("root");
 | 
						|
      if (*name == '\0')
 | 
						|
	{
 | 
						|
	  grub_error (GRUB_ERR_BAD_DEVICE, "no device is set");
 | 
						|
	  goto fail;
 | 
						|
	}
 | 
						|
    }
 | 
						|
    
 | 
						|
  dev = grub_malloc (sizeof (*dev));
 | 
						|
  if (! dev)
 | 
						|
    goto fail;
 | 
						|
  
 | 
						|
  /* Try to open a disk.  */
 | 
						|
  disk = grub_disk_open (name);
 | 
						|
  if (! disk)
 | 
						|
    {
 | 
						|
      grub_error (GRUB_ERR_BAD_DEVICE, "unknown device %s", name);
 | 
						|
      goto fail;
 | 
						|
    }
 | 
						|
 | 
						|
  dev->disk = disk;
 | 
						|
  dev->net = 0;	/* FIXME */
 | 
						|
 | 
						|
  return dev;
 | 
						|
 | 
						|
 fail:
 | 
						|
  if (disk)
 | 
						|
    grub_disk_close (disk);
 | 
						|
  
 | 
						|
  grub_free (dev);
 | 
						|
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
 | 
						|
grub_err_t
 | 
						|
grub_device_close (grub_device_t device)
 | 
						|
{
 | 
						|
  if (device->disk)
 | 
						|
    grub_disk_close (device->disk);
 | 
						|
 | 
						|
  grub_free (device);
 | 
						|
 | 
						|
  return grub_errno;
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
grub_device_iterate (int (*hook) (const char *name))
 | 
						|
{
 | 
						|
  auto int iterate_disk (const char *disk_name);
 | 
						|
  auto int iterate_partition (grub_disk_t disk,
 | 
						|
			      const grub_partition_t partition);
 | 
						|
  
 | 
						|
  int iterate_disk (const char *disk_name)
 | 
						|
    {
 | 
						|
      grub_device_t dev;
 | 
						|
 | 
						|
      if (hook (disk_name))
 | 
						|
	return 1;
 | 
						|
      
 | 
						|
      dev = grub_device_open (disk_name);
 | 
						|
      if (! dev)
 | 
						|
	return 0;
 | 
						|
      
 | 
						|
      if (dev->disk && dev->disk->has_partitions)
 | 
						|
	if (grub_partition_iterate (dev->disk, iterate_partition))
 | 
						|
	  {
 | 
						|
	    grub_device_close (dev);
 | 
						|
	    return 1;
 | 
						|
	  }
 | 
						|
 | 
						|
      grub_device_close (dev);
 | 
						|
      return 0;
 | 
						|
    }
 | 
						|
  
 | 
						|
  int iterate_partition (grub_disk_t disk, const grub_partition_t partition)
 | 
						|
    {
 | 
						|
      char *partition_name;
 | 
						|
      char *device_name;
 | 
						|
      int ret;
 | 
						|
      
 | 
						|
      partition_name = grub_partition_get_name (partition);
 | 
						|
      if (! partition_name)
 | 
						|
	return 1;
 | 
						|
      
 | 
						|
      device_name = grub_malloc (grub_strlen (disk->name) + 1
 | 
						|
				 + grub_strlen (partition_name) + 1);
 | 
						|
      if (! device_name)
 | 
						|
	{
 | 
						|
	  grub_free (partition_name);
 | 
						|
	  return 1;
 | 
						|
	}
 | 
						|
 | 
						|
      grub_sprintf (device_name, "%s,%s", disk->name, partition_name);
 | 
						|
      grub_free (partition_name);
 | 
						|
 | 
						|
      ret = hook (device_name);
 | 
						|
      grub_free (device_name);
 | 
						|
      return ret;
 | 
						|
    }
 | 
						|
 | 
						|
  /* Only disk devices are supported at the moment.  */
 | 
						|
  return grub_disk_dev_iterate (iterate_disk);
 | 
						|
}
 |