/* raid.c - RAID support for GRUB utils.  */
/*
 *  GRUB  --  GRand Unified Bootloader
 *  Copyright (C) 2006,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/>.
 */

/* We only support RAID on Linux.  */
#ifdef __linux__
#include <grub/util/misc.h>
#include <grub/util/raid.h>

#include <string.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <errno.h>

#include <linux/types.h>
#include <linux/major.h>
#include <linux/raid/md_p.h>
#include <linux/raid/md_u.h>

static char *
grub_util_getdiskname (int major, int minor)
{
  char *name = xmalloc (15);

  if (major == LOOP_MAJOR)
    sprintf (name, "/dev/loop%d", minor);
  else if (major == IDE0_MAJOR)
    sprintf (name, "/dev/hd%c", 'a' + minor / 64);
  else if (major == IDE1_MAJOR)
    sprintf (name, "/dev/hd%c", 'c' + minor / 64);
  else if (major == IDE2_MAJOR)
    sprintf (name, "/dev/hd%c", 'e' + minor / 64);
  else if (major == IDE3_MAJOR)
    sprintf (name, "/dev/hd%c", 'g' + minor / 64);
  else if (major == SCSI_DISK0_MAJOR)
    sprintf (name, "/dev/sd%c", 'a' + minor / 16);
  else
    grub_util_error ("Unknown device number: %d, %d", major, minor);

  return name;
}

char **
grub_util_raid_getmembers (char *name)
{
  int fd, ret, i, j;
  char *devname;
  char **devicelist;
  mdu_version_t version;
  mdu_array_info_t info;
  mdu_disk_info_t disk;

  devname = xmalloc (strlen (name) + 6);
  strcpy (devname, "/dev/");
  strcpy (devname+5, name);

  fd = open (devname, O_RDONLY);

  if (fd == -1)
    grub_util_error ("Can't open %s: %s", devname, strerror (errno));

  free (devname);

  ret = ioctl (fd, RAID_VERSION, &version);
  if (ret != 0)
    grub_util_error ("ioctl RAID_VERSION error: %s", strerror (errno));

  if (version.major != 0 || version.minor != 90)
    grub_util_error ("Unsupported RAID version: %d.%d",
		     version.major, version.minor);

  ret = ioctl (fd, GET_ARRAY_INFO, &info);
  if (ret != 0)
    grub_util_error ("ioctl GET_ARRAY_INFO error: %s", strerror (errno));

  devicelist = xmalloc ((info.nr_disks + 1) * sizeof (char *));

  for (i = 0, j = 0; i <info.nr_disks; i++)
    {
      disk.number = i;
      ret = ioctl (fd, GET_DISK_INFO, &disk);
      if (ret != 0)
	grub_util_error ("ioctl GET_DISK_INFO error: %s", strerror (errno));

      if (disk.state & (1 << MD_DISK_ACTIVE))
	{
	  devicelist[j] = grub_util_getdiskname (disk.major, disk.minor);
	  j++;
	}
    }

  devicelist[j] = NULL;

  return devicelist;
}

#endif /* ! __linux__ */