/*
   * (C) Copyright 2001
   * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
   *

   * SPDX-License-Identifier:	GPL-2.0+

   */
  
  #include <common.h>
  #include <command.h>

  #include <errno.h>

  #include <ide.h>

  #include <malloc.h>

  #include <part.h>

  #include <ubifs_uboot.h>

  
  #undef	PART_DEBUG
  
  #ifdef	PART_DEBUG
  #define	PRINTF(fmt,args...)	printf (fmt ,##args)
  #else
  #define PRINTF(fmt,args...)
  #endif

  /* Check all partition types */
  #define PART_TYPE_ALL		-1

  DECLARE_GLOBAL_DATA_PTR;

  #ifdef CONFIG_HAVE_BLOCK_DEVICE

  static struct part_driver *part_driver_lookup_type(int part_type)
  {
  	struct part_driver *drv =
  		ll_entry_start(struct part_driver, part_driver);
  	const int n_ents = ll_entry_count(struct part_driver, part_driver);
  	struct part_driver *entry;
  
  	for (entry = drv; entry != drv + n_ents; entry++) {
  		if (part_type == entry->part_type)
  			return entry;
  	}
  
  	/* Not found */
  	return NULL;
  }

  static struct blk_desc *get_dev_hwpart(const char *ifname, int dev, int hwpart)

  {

  	struct blk_desc *dev_desc;
  	int ret;

  	dev_desc = blk_get_devnum_by_typename(ifname, dev);
  	if (!dev_desc) {
  		debug("%s: No device for iface '%s', dev %d
  ", __func__,
  		      ifname, dev);

  		return NULL;

  	}

  	ret = blk_dselect_hwpart(dev_desc, hwpart);
  	if (ret) {
  		debug("%s: Failed to select h/w partition: err-%d
  ", __func__,
  		      ret);
  		return NULL;
  	}
  
  	return dev_desc;

  }

  struct blk_desc *blk_get_dev(const char *ifname, int dev)

  {

  	return get_dev_hwpart(ifname, dev, 0);

  }

  #else

  struct blk_desc *get_dev_hwpart(const char *ifname, int dev, int hwpart)

  {
  	return NULL;
  }

  struct blk_desc *blk_get_dev(const char *ifname, int dev)

  {
  	return NULL;
  }
  #endif

  #ifdef CONFIG_HAVE_BLOCK_DEVICE

  /* ------------------------------------------------------------------------- */
  /*
   * reports device info to the user
   */

  #ifdef CONFIG_LBA48

  typedef uint64_t lba512_t;

  #else

  typedef lbaint_t lba512_t;

  #endif

  /*
   * Overflowless variant of (block_count * mul_by / div_by)
   * when div_by > mul_by
   */

  static lba512_t lba512_muldiv(lba512_t block_count, lba512_t mul_by, lba512_t div_by)

  {
  	lba512_t bc_quot, bc_rem;
  
  	/* x * m / d == x / d * m + (x % d) * m / d */
  	bc_quot = block_count / div_by;
  	bc_rem  = block_count - div_by * bc_quot;
  	return bc_quot * mul_by + (bc_rem * mul_by) / div_by;
  }

  void dev_print (struct blk_desc *dev_desc)

  {
  	lba512_t lba512; /* number of blocks if 512bytes block size */

  	if (dev_desc->type == DEV_TYPE_UNKNOWN) {
  		puts ("not available
  ");
  		return;
  	}

  	switch (dev_desc->if_type) {

  	case IF_TYPE_SCSI:
  		printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s
  ",
  			dev_desc->target,dev_desc->lun,

  			dev_desc->vendor,

  			dev_desc->product,
  			dev_desc->revision);
  		break;

  	case IF_TYPE_ATAPI:

  	case IF_TYPE_IDE:
  	case IF_TYPE_SATA:

  		printf ("Model: %s Firm: %s Ser#: %s
  ",
  			dev_desc->vendor,
  			dev_desc->revision,
  			dev_desc->product);

  		break;

  	case IF_TYPE_SD:
  	case IF_TYPE_MMC:

  	case IF_TYPE_USB:

  	case IF_TYPE_NVME:

  		printf ("Vendor: %s Rev: %s Prod: %s
  ",
  			dev_desc->vendor,
  			dev_desc->revision,
  			dev_desc->product);
  		break;

  	case IF_TYPE_DOC:
  		puts("device type DOC
  ");
  		return;

  	case IF_TYPE_UNKNOWN:

  		puts("device type unknown
  ");
  		return;

  	default:

  		printf("Unhandled device type: %i
  ", dev_desc->if_type);

  		return;

  	}
  	puts ("            Type: ");
  	if (dev_desc->removable)
  		puts ("Removable ");
  	switch (dev_desc->type & 0x1F) {

  	case DEV_TYPE_HARDDISK:
  		puts ("Hard Disk");
  		break;
  	case DEV_TYPE_CDROM:
  		puts ("CD ROM");
  		break;
  	case DEV_TYPE_OPDISK:
  		puts ("Optical Device");
  		break;
  	case DEV_TYPE_TAPE:
  		puts ("Tape");
  		break;
  	default:
  		printf ("# %02X #", dev_desc->type & 0x1F);
  		break;

  	}
  	puts ("
  ");

  	if (dev_desc->lba > 0L && dev_desc->blksz > 0L) {

  		ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem;

  		lbaint_t lba;

  
  		lba = dev_desc->lba;

  		lba512 = (lba * (dev_desc->blksz/512));

  		/* round to 1 digit */

  		/* 2048 = (1024 * 1024) / 512 MB */
  		mb = lba512_muldiv(lba512, 10, 2048);

  		mb_quot	= mb / 10;
  		mb_rem	= mb - (10 * mb_quot);
  
  		gb = mb / 1024;
  		gb_quot	= gb / 10;
  		gb_rem	= gb - (10 * gb_quot);

  #ifdef CONFIG_LBA48

  		if (dev_desc->lba48)

  			printf ("            Supports 48-bit addressing
  ");
  #endif

  #if defined(CONFIG_SYS_64BIT_LBA)

  		printf ("            Capacity: %lu.%lu MB = %lu.%lu GB (%llu x %lu)
  ",

  			mb_quot, mb_rem,
  			gb_quot, gb_rem,
  			lba,
  			dev_desc->blksz);
  #else

  		printf ("            Capacity: %lu.%lu MB = %lu.%lu GB (%lu x %lu)
  ",

  			mb_quot, mb_rem,
  			gb_quot, gb_rem,

  			(ulong)lba,

  			dev_desc->blksz);

  #endif

  	} else {
  		puts ("            Capacity: not available
  ");
  	}
  }

  #endif

  #ifdef CONFIG_HAVE_BLOCK_DEVICE

  void part_init(struct blk_desc *dev_desc)

  {

  	struct part_driver *drv =
  		ll_entry_start(struct part_driver, part_driver);
  	const int n_ents = ll_entry_count(struct part_driver, part_driver);
  	struct part_driver *entry;

  	blkcache_invalidate(dev_desc->if_type, dev_desc->devnum);

  	dev_desc->part_type = PART_TYPE_UNKNOWN;

  	for (entry = drv; entry != drv + n_ents; entry++) {
  		int ret;
  
  		ret = entry->test(dev_desc);
  		debug("%s: try '%s': ret=%d
  ", __func__, entry->name, ret);
  		if (!ret) {
  			dev_desc->part_type = entry->part_type;
  			break;
  		}
  	}

  }

  static void print_part_header(const char *type, struct blk_desc *dev_desc)
  {

  #if CONFIG_IS_ENABLED(MAC_PARTITION) || \

  	CONFIG_IS_ENABLED(DOS_PARTITION) || \

  	CONFIG_IS_ENABLED(ISO_PARTITION) || \

  	CONFIG_IS_ENABLED(AMIGA_PARTITION) || \

  	CONFIG_IS_ENABLED(EFI_PARTITION)

  	puts ("
  Partition Map for ");
  	switch (dev_desc->if_type) {

  	case IF_TYPE_IDE:
  		puts ("IDE");
  		break;
  	case IF_TYPE_SATA:
  		puts ("SATA");
  		break;
  	case IF_TYPE_SCSI:
  		puts ("SCSI");
  		break;
  	case IF_TYPE_ATAPI:
  		puts ("ATAPI");
  		break;
  	case IF_TYPE_USB:
  		puts ("USB");
  		break;
  	case IF_TYPE_DOC:
  		puts ("DOC");
  		break;

  	case IF_TYPE_MMC:
  		puts ("MMC");
  		break;

  	case IF_TYPE_HOST:

  		puts ("HOST");
  		break;
  	case IF_TYPE_NVME:
  		puts ("NVMe");

  		break;

  	default:
  		puts ("UNKNOWN");
  		break;

  	}
  	printf (" device %d  --   Partition Type: %s
  
  ",

  			dev_desc->devnum, type);

  #endif /* any CONFIG_..._PARTITION */

  }

  void part_print(struct blk_desc *dev_desc)

  {

  	struct part_driver *drv;

  	drv = part_driver_lookup_type(dev_desc->part_type);
  	if (!drv) {
  		printf("## Unknown partition table type %x
  ",
  		       dev_desc->part_type);

  		return;

  	}

  
  	PRINTF("## Testing for valid %s partition ##
  ", drv->name);
  	print_part_header(drv->name, dev_desc);
  	if (drv->print)
  		drv->print(dev_desc);

  }

  #endif /* CONFIG_HAVE_BLOCK_DEVICE */

  int part_get_info(struct blk_desc *dev_desc, int part,

  		       disk_partition_t *info)

  {

  #ifdef CONFIG_HAVE_BLOCK_DEVICE

  	struct part_driver *drv;

  #if CONFIG_IS_ENABLED(PARTITION_UUIDS)

  	/* The common case is no UUID support */
  	info->uuid[0] = 0;
  #endif

  #ifdef CONFIG_PARTITION_TYPE_GUID
  	info->type_guid[0] = 0;
  #endif

  	drv = part_driver_lookup_type(dev_desc->part_type);
  	if (!drv) {
  		debug("## Unknown partition table type %x
  ",
  		      dev_desc->part_type);
  		return -EPROTONOSUPPORT;
  	}
  	if (!drv->get_info) {

  		PRINTF("## Driver %s does not have the get_info() method
  ",
  		       drv->name);

  		return -ENOSYS;
  	}
  	if (drv->get_info(dev_desc, part, info) == 0) {
  		PRINTF("## Valid %s partition found ##
  ", drv->name);
  		return 0;

  	}

  #endif /* CONFIG_HAVE_BLOCK_DEVICE */

  
  	return -1;
  }

  int part_get_info_whole_disk(struct blk_desc *dev_desc, disk_partition_t *info)
  {
  	info->start = 0;
  	info->size = dev_desc->lba;
  	info->blksz = dev_desc->blksz;
  	info->bootable = 0;
  	strcpy((char *)info->type, BOOT_PART_TYPE);
  	strcpy((char *)info->name, "Whole Disk");
  #if CONFIG_IS_ENABLED(PARTITION_UUIDS)
  	info->uuid[0] = 0;
  #endif
  #ifdef CONFIG_PARTITION_TYPE_GUID
  	info->type_guid[0] = 0;
  #endif
  
  	return 0;
  }

  int blk_get_device_by_str(const char *ifname, const char *dev_hwpart_str,
  			  struct blk_desc **dev_desc)

  {
  	char *ep;

  	char *dup_str = NULL;
  	const char *dev_str, *hwpart_str;
  	int dev, hwpart;
  
  	hwpart_str = strchr(dev_hwpart_str, '.');
  	if (hwpart_str) {
  		dup_str = strdup(dev_hwpart_str);
  		dup_str[hwpart_str - dev_hwpart_str] = 0;
  		dev_str = dup_str;
  		hwpart_str++;
  	} else {
  		dev_str = dev_hwpart_str;

  		hwpart = 0;

  	}

  
  	dev = simple_strtoul(dev_str, &ep, 16);
  	if (*ep) {
  		printf("** Bad device specification %s %s **
  ",
  		       ifname, dev_str);

  		dev = -EINVAL;

  		goto cleanup;
  	}
  
  	if (hwpart_str) {
  		hwpart = simple_strtoul(hwpart_str, &ep, 16);
  		if (*ep) {
  			printf("** Bad HW partition specification %s %s **
  ",
  			    ifname, hwpart_str);

  			dev = -EINVAL;

  			goto cleanup;
  		}

  	}

  	*dev_desc = get_dev_hwpart(ifname, dev, hwpart);

  	if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) {

  		printf("** Bad device %s %s **
  ", ifname, dev_hwpart_str);

  		dev = -ENOENT;

  		goto cleanup;

  	}

  #ifdef CONFIG_HAVE_BLOCK_DEVICE

  	/*
  	 * Updates the partition table for the specified hw partition.
  	 * Does not need to be done for hwpart 0 since it is default and
  	 * already loaded.
  	 */
  	if(hwpart != 0)

  		part_init(*dev_desc);

  #endif

  cleanup:
  	free(dup_str);

  	return dev;
  }

  #define PART_UNSPECIFIED -2
  #define PART_AUTO -1

  int blk_get_device_part_str(const char *ifname, const char *dev_part_str,

  			     struct blk_desc **dev_desc,

  			     disk_partition_t *info, int allow_whole_dev)

  {

  	int ret = -1;
  	const char *part_str;
  	char *dup_str = NULL;
  	const char *dev_str;

  	int dev;

  	char *ep;
  	int p;
  	int part;
  	disk_partition_t tmpinfo;

  #ifdef CONFIG_SANDBOX

  	/*

  	 * Special-case a pseudo block device "hostfs", to allow access to the

  	 * host's own filesystem.
  	 */
  	if (0 == strcmp(ifname, "hostfs")) {
  		*dev_desc = NULL;
  		info->start = 0;
  		info->size = 0;
  		info->blksz = 0;
  		info->bootable = 0;
  		strcpy((char *)info->type, BOOT_PART_TYPE);
  		strcpy((char *)info->name, "Sandbox host");

  #if CONFIG_IS_ENABLED(PARTITION_UUIDS)

  		info->uuid[0] = 0;
  #endif

  #ifdef CONFIG_PARTITION_TYPE_GUID
  		info->type_guid[0] = 0;
  #endif

  
  		return 0;
  	}

  #endif

  #ifdef CONFIG_CMD_UBIFS
  	/*
  	 * Special-case ubi, ubi goes through a mtd, rathen then through
  	 * a regular block device.
  	 */
  	if (0 == strcmp(ifname, "ubi")) {
  		if (!ubifs_is_mounted()) {
  			printf("UBIFS not mounted, use ubifsmount to mount volume first!
  ");
  			return -1;
  		}
  
  		*dev_desc = NULL;
  		memset(info, 0, sizeof(*info));
  		strcpy((char *)info->type, BOOT_PART_TYPE);
  		strcpy((char *)info->name, "UBI");

  #if CONFIG_IS_ENABLED(PARTITION_UUIDS)

  		info->uuid[0] = 0;
  #endif
  		return 0;
  	}
  #endif

  	/* If no dev_part_str, use bootdevice environment variable */

  	if (!dev_part_str || !strlen(dev_part_str) ||
  	    !strcmp(dev_part_str, "-"))

  		dev_part_str = env_get("bootdevice");

  
  	/* If still no dev_part_str, it's an error */
  	if (!dev_part_str) {
  		printf("** No device specified **
  ");
  		goto cleanup;
  	}
  
  	/* Separate device and partition ID specification */
  	part_str = strchr(dev_part_str, ':');
  	if (part_str) {
  		dup_str = strdup(dev_part_str);
  		dup_str[part_str - dev_part_str] = 0;
  		dev_str = dup_str;
  		part_str++;
  	} else {
  		dev_str = dev_part_str;

  	}

  	/* Look up the device */

  	dev = blk_get_device_by_str(ifname, dev_str, dev_desc);

  	if (dev < 0)
  		goto cleanup;
  
  	/* Convert partition ID string to number */
  	if (!part_str || !*part_str) {
  		part = PART_UNSPECIFIED;
  	} else if (!strcmp(part_str, "auto")) {
  		part = PART_AUTO;
  	} else {
  		/* Something specified -> use exactly that */
  		part = (int)simple_strtoul(part_str, &ep, 16);
  		/*
  		 * Less than whole string converted,
  		 * or request for whole device, but caller requires partition.
  		 */
  		if (*ep || (part == 0 && !allow_whole_dev)) {
  			printf("** Bad partition specification %s %s **
  ",
  			    ifname, dev_part_str);
  			goto cleanup;
  		}
  	}
  
  	/*
  	 * No partition table on device,
  	 * or user requested partition 0 (entire device).
  	 */
  	if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) ||
  	    (part == 0)) {
  		if (!(*dev_desc)->lba) {
  			printf("** Bad device size - %s %s **
  ", ifname,
  			       dev_str);
  			goto cleanup;
  		}

  		/*
  		 * If user specified a partition ID other than 0,
  		 * or the calling command only accepts partitions,
  		 * it's an error.
  		 */
  		if ((part > 0) || (!allow_whole_dev)) {
  			printf("** No partition table - %s %s **
  ", ifname,
  			       dev_str);
  			goto cleanup;

  		}

  		(*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);

  		part_get_info_whole_disk(*dev_desc, info);

  		ret = 0;
  		goto cleanup;

  	}

  	/*
  	 * Now there's known to be a partition table,
  	 * not specifying a partition means to pick partition 1.
  	 */
  	if (part == PART_UNSPECIFIED)
  		part = 1;
  
  	/*
  	 * If user didn't specify a partition number, or did specify something
  	 * other than "auto", use that partition number directly.
  	 */
  	if (part != PART_AUTO) {

  		ret = part_get_info(*dev_desc, part, info);

  		if (ret) {
  			printf("** Invalid partition %d **
  ", part);
  			goto cleanup;
  		}
  	} else {
  		/*
  		 * Find the first bootable partition.
  		 * If none are bootable, fall back to the first valid partition.
  		 */
  		part = 0;
  		for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {

  			ret = part_get_info(*dev_desc, p, info);

  			if (ret)
  				continue;
  
  			/*
  			 * First valid partition, or new better partition?
  			 * If so, save partition ID.
  			 */
  			if (!part || info->bootable)
  				part = p;
  
  			/* Best possible partition? Stop searching. */
  			if (info->bootable)
  				break;
  
  			/*
  			 * We now need to search further for best possible.
  			 * If we what we just queried was the best so far,
  			 * save the info since we over-write it next loop.
  			 */
  			if (part == p)
  				tmpinfo = *info;
  		}
  		/* If we found any acceptable partition */
  		if (part) {
  			/*
  			 * If we searched all possible partition IDs,
  			 * return the first valid partition we found.
  			 */
  			if (p == MAX_SEARCH_PARTITIONS + 1)
  				*info = tmpinfo;

  		} else {
  			printf("** No valid partitions found **
  ");

  			ret = -1;

  			goto cleanup;
  		}

  	}
  	if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) {
  		printf("** Invalid partition type \"%.32s\""
  			" (expect \"" BOOT_PART_TYPE "\")
  ",
  			info->type);

  		ret  = -1;
  		goto cleanup;

  	}

  	(*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);

  	ret = part;
  	goto cleanup;

  cleanup:
  	free(dup_str);
  	return ret;

  }

  int part_get_info_by_name_type(struct blk_desc *dev_desc, const char *name,
  			       disk_partition_t *info, int part_type)

  {
  	struct part_driver *first_drv =
  		ll_entry_start(struct part_driver, part_driver);
  	const int n_drvs = ll_entry_count(struct part_driver, part_driver);
  	struct part_driver *part_drv;
  
  	for (part_drv = first_drv; part_drv != first_drv + n_drvs; part_drv++) {
  		int ret;
  		int i;
  		for (i = 1; i < part_drv->max_entries; i++) {

  			if (part_type >= 0 && part_type != part_drv->part_type)
  				break;

  			ret = part_drv->get_info(dev_desc, i, info);
  			if (ret != 0) {
  				/* no more entries in table */
  				break;
  			}
  			if (strcmp(name, (const char *)info->name) == 0) {
  				/* matched */

  				return i;

  			}
  		}
  	}
  	return -1;
  }

  int part_get_info_by_name(struct blk_desc *dev_desc, const char *name,
  			  disk_partition_t *info)
  {
  	return part_get_info_by_name_type(dev_desc, name, info, PART_TYPE_ALL);
  }

  void part_set_generic_name(const struct blk_desc *dev_desc,
  	int part_num, char *name)
  {
  	char *devtype;
  
  	switch (dev_desc->if_type) {
  	case IF_TYPE_IDE:
  	case IF_TYPE_SATA:
  	case IF_TYPE_ATAPI:
  		devtype = "hd";
  		break;
  	case IF_TYPE_SCSI:
  		devtype = "sd";
  		break;
  	case IF_TYPE_USB:
  		devtype = "usbd";
  		break;
  	case IF_TYPE_DOC:
  		devtype = "docd";
  		break;
  	case IF_TYPE_MMC:
  	case IF_TYPE_SD:
  		devtype = "mmcsd";
  		break;
  	default:
  		devtype = "xx";
  		break;
  	}
  
  	sprintf(name, "%s%c%d", devtype, 'a' + dev_desc->devnum, part_num);
  }