/*
   * (C) Copyright 2001
   * Denis Peter, MPL AG Switzerland
   *

   * Adapted for U-Boot driver model
   * (C) Copyright 2015 Google, Inc
   *

   * Most of this source has been derived from the Linux USB
   * project.
   *

   * SPDX-License-Identifier:	GPL-2.0+

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

  #include <console.h>

  #include <dm.h>

  #include <dm/uclass-internal.h>

  #include <memalign.h>

  #include <asm/byteorder.h>

  #include <asm/unaligned.h>

  #include <part.h>

  #include <usb.h>

  #ifdef CONFIG_USB_STORAGE

  static int usb_stor_curr_dev = -1; /* current device */

  #endif

  #if defined(CONFIG_USB_HOST_ETHER) && !defined(CONFIG_DM_ETH)

  static int __maybe_unused usb_ether_curr_dev = -1; /* current ethernet device */

  #endif

  /* some display routines (info command) */

  static char *usb_get_class_desc(unsigned char dclass)

  {

  	switch (dclass) {
  	case USB_CLASS_PER_INTERFACE:
  		return "See Interface";
  	case USB_CLASS_AUDIO:
  		return "Audio";
  	case USB_CLASS_COMM:
  		return "Communication";
  	case USB_CLASS_HID:
  		return "Human Interface";
  	case USB_CLASS_PRINTER:
  		return "Printer";
  	case USB_CLASS_MASS_STORAGE:
  		return "Mass Storage";
  	case USB_CLASS_HUB:
  		return "Hub";
  	case USB_CLASS_DATA:
  		return "CDC Data";
  	case USB_CLASS_VENDOR_SPEC:
  		return "Vendor specific";
  	default:
  		return "";

  	}
  }

  static void usb_display_class_sub(unsigned char dclass, unsigned char subclass,
  				  unsigned char proto)

  {

  	switch (dclass) {
  	case USB_CLASS_PER_INTERFACE:
  		printf("See Interface");
  		break;
  	case USB_CLASS_HID:
  		printf("Human Interface, Subclass: ");
  		switch (subclass) {
  		case USB_SUB_HID_NONE:
  			printf("None");

  			break;

  		case USB_SUB_HID_BOOT:
  			printf("Boot ");
  			switch (proto) {
  			case USB_PROT_HID_NONE:
  				printf("None");
  				break;
  			case USB_PROT_HID_KEYBOARD:
  				printf("Keyboard");
  				break;
  			case USB_PROT_HID_MOUSE:
  				printf("Mouse");
  				break;
  			default:
  				printf("reserved");
  				break;

  			}
  			break;

  		default:
  			printf("reserved");
  			break;
  		}
  		break;
  	case USB_CLASS_MASS_STORAGE:
  		printf("Mass Storage, ");
  		switch (subclass) {
  		case US_SC_RBC:
  			printf("RBC ");
  			break;
  		case US_SC_8020:
  			printf("SFF-8020i (ATAPI)");
  			break;
  		case US_SC_QIC:
  			printf("QIC-157 (Tape)");
  			break;
  		case US_SC_UFI:
  			printf("UFI");
  			break;
  		case US_SC_8070:
  			printf("SFF-8070");
  			break;
  		case US_SC_SCSI:
  			printf("Transp. SCSI");
  			break;
  		default:
  			printf("reserved");
  			break;
  		}
  		printf(", ");
  		switch (proto) {
  		case US_PR_CB:
  			printf("Command/Bulk");
  			break;
  		case US_PR_CBI:
  			printf("Command/Bulk/Int");
  			break;
  		case US_PR_BULK:
  			printf("Bulk only");

  			break;
  		default:

  			printf("reserved");
  			break;
  		}
  		break;
  	default:
  		printf("%s", usb_get_class_desc(dclass));
  		break;

  	}
  }

  static void usb_display_string(struct usb_device *dev, int index)

  {

  	ALLOC_CACHE_ALIGN_BUFFER(char, buffer, 256);

  	if (index != 0) {
  		if (usb_string(dev, index, &buffer[0], 256) > 0)
  			printf("String: \"%s\"", buffer);

  	}
  }

  static void usb_display_desc(struct usb_device *dev)

  {

  	uint packet_size = dev->descriptor.bMaxPacketSize0;

  	if (dev->descriptor.bDescriptorType == USB_DT_DEVICE) {
  		printf("%d: %s,  USB Revision %x.%x
  ", dev->devnum,

  		usb_get_class_desc(dev->config.if_desc[0].desc.bInterfaceClass),

  				   (dev->descriptor.bcdUSB>>8) & 0xff,
  				   dev->descriptor.bcdUSB & 0xff);
  
  		if (strlen(dev->mf) || strlen(dev->prod) ||
  		    strlen(dev->serial))
  			printf(" - %s %s %s
  ", dev->mf, dev->prod,
  				dev->serial);

  		if (dev->descriptor.bDeviceClass) {
  			printf(" - Class: ");

  			usb_display_class_sub(dev->descriptor.bDeviceClass,
  					      dev->descriptor.bDeviceSubClass,
  					      dev->descriptor.bDeviceProtocol);

  			printf("
  ");

  		} else {
  			printf(" - Class: (from Interface) %s
  ",
  			       usb_get_class_desc(

  				dev->config.if_desc[0].desc.bInterfaceClass));

  		}

  		if (dev->descriptor.bcdUSB >= cpu_to_le16(0x0300))
  			packet_size = 1 << packet_size;

  		printf(" - PacketSize: %d  Configurations: %d
  ",

  			packet_size, dev->descriptor.bNumConfigurations);

  		printf(" - Vendor: 0x%04x  Product 0x%04x Version %d.%d
  ",
  			dev->descriptor.idVendor, dev->descriptor.idProduct,
  			(dev->descriptor.bcdDevice>>8) & 0xff,
  			dev->descriptor.bcdDevice & 0xff);

  	}
  
  }

  static void usb_display_conf_desc(struct usb_config_descriptor *config,

  				  struct usb_device *dev)

  {

  	printf("   Configuration: %d
  ", config->bConfigurationValue);
  	printf("   - Interfaces: %d %s%s%dmA
  ", config->bNumInterfaces,
  	       (config->bmAttributes & 0x40) ? "Self Powered " : "Bus Powered ",
  	       (config->bmAttributes & 0x20) ? "Remote Wakeup " : "",

  		config->bMaxPower*2);

  	if (config->iConfiguration) {
  		printf("   - ");

  		usb_display_string(dev, config->iConfiguration);

  		printf("
  ");
  	}
  }

  static void usb_display_if_desc(struct usb_interface_descriptor *ifdesc,
  				struct usb_device *dev)

  {

  	printf("     Interface: %d
  ", ifdesc->bInterfaceNumber);
  	printf("     - Alternate Setting %d, Endpoints: %d
  ",
  		ifdesc->bAlternateSetting, ifdesc->bNumEndpoints);

  	printf("     - Class ");

  	usb_display_class_sub(ifdesc->bInterfaceClass,
  		ifdesc->bInterfaceSubClass, ifdesc->bInterfaceProtocol);

  	printf("
  ");
  	if (ifdesc->iInterface) {
  		printf("     - ");

  		usb_display_string(dev, ifdesc->iInterface);

  		printf("
  ");
  	}
  }

  static void usb_display_ep_desc(struct usb_endpoint_descriptor *epdesc)

  {

  	printf("     - Endpoint %d %s ", epdesc->bEndpointAddress & 0xf,
  		(epdesc->bEndpointAddress & 0x80) ? "In" : "Out");
  	switch ((epdesc->bmAttributes & 0x03)) {
  	case 0:
  		printf("Control");
  		break;
  	case 1:
  		printf("Isochronous");
  		break;
  	case 2:
  		printf("Bulk");
  		break;
  	case 3:
  		printf("Interrupt");
  		break;

  	}

  	printf(" MaxPacket %d", get_unaligned(&epdesc->wMaxPacketSize));

  	if ((epdesc->bmAttributes & 0x03) == 0x3)
  		printf(" Interval %dms", epdesc->bInterval);

  	printf("
  ");
  }
  
  /* main routine to diasplay the configs, interfaces and endpoints */

  static void usb_display_config(struct usb_device *dev)

  {

  	struct usb_config *config;
  	struct usb_interface *ifdesc;

  	struct usb_endpoint_descriptor *epdesc;

  	int i, ii;
  
  	config = &dev->config;

  	usb_display_conf_desc(&config->desc, dev);

  	for (i = 0; i < config->no_of_if; i++) {
  		ifdesc = &config->if_desc[i];

  		usb_display_if_desc(&ifdesc->desc, dev);

  		for (ii = 0; ii < ifdesc->no_of_ep; ii++) {
  			epdesc = &ifdesc->ep_desc[ii];

  			usb_display_ep_desc(epdesc);
  		}
  	}
  	printf("
  ");
  }

  /*
   * With driver model this isn't right since we can have multiple controllers
   * and the device numbering starts at 1 on each bus.
   * TODO(sjg@chromium.org): Add a way to specify the controller/bus.
   */

  static struct usb_device *usb_find_device(int devnum)
  {

  #ifdef CONFIG_DM_USB
  	struct usb_device *udev;
  	struct udevice *hub;
  	struct uclass *uc;
  	int ret;
  
  	/* Device addresses start at 1 */
  	devnum++;
  	ret = uclass_get(UCLASS_USB_HUB, &uc);
  	if (ret)
  		return NULL;
  
  	uclass_foreach_dev(hub, uc) {
  		struct udevice *dev;
  
  		if (!device_active(hub))
  			continue;

  		udev = dev_get_parent_priv(hub);

  		if (udev->devnum == devnum)
  			return udev;
  
  		for (device_find_first_child(hub, &dev);
  		     dev;
  		     device_find_next_child(&dev)) {
  			if (!device_active(hub))
  				continue;

  			udev = dev_get_parent_priv(dev);

  			if (udev->devnum == devnum)
  				return udev;
  		}
  	}
  #else

  	struct usb_device *udev;

  	int d;
  
  	for (d = 0; d < USB_MAX_DEVICE; d++) {

  		udev = usb_get_dev_index(d);
  		if (udev == NULL)

  			return NULL;

  		if (udev->devnum == devnum)
  			return udev;

  	}

  #endif

  
  	return NULL;
  }

  static inline char *portspeed(int speed)
  {

  	char *speed_str;
  
  	switch (speed) {
  	case USB_SPEED_SUPER:
  		speed_str = "5 Gb/s";
  		break;
  	case USB_SPEED_HIGH:
  		speed_str = "480 Mb/s";
  		break;
  	case USB_SPEED_LOW:
  		speed_str = "1.5 Mb/s";
  		break;
  	default:
  		speed_str = "12 Mb/s";
  		break;
  	}
  
  	return speed_str;

  }

  /* shows the device tree recursively */

  static void usb_show_tree_graph(struct usb_device *dev, char *pre)

  {

  	int index;

  	int has_child, last_child;

  	index = strlen(pre);
  	printf(" %s", pre);

  #ifdef CONFIG_DM_USB
  	has_child = device_has_active_children(dev->dev);

  	if (device_get_uclass_id(dev->dev) == UCLASS_MASS_STORAGE) {
  		struct udevice *child;
  
  		for (device_find_first_child(dev->dev, &child);
  		     child;
  		     device_find_next_child(&child)) {
  			if (device_get_uclass_id(child) == UCLASS_BLK)
  				has_child = 0;
  		}
  	}

  #else

  	/* check if the device has connected children */

  	int i;

  	has_child = 0;
  	for (i = 0; i < dev->maxchild; i++) {
  		if (dev->children[i] != NULL)
  			has_child = 1;

  	}

  #endif

  	/* check if we are the last one */

  #ifdef CONFIG_DM_USB

  	/* Not the root of the usb tree? */
  	if (device_get_uclass_id(dev->dev->parent) != UCLASS_USB) {
  		last_child = device_is_last_sibling(dev->dev);

  #else

  	if (dev->parent != NULL) { /* not root? */
  		last_child = 1;

  		for (i = 0; i < dev->parent->maxchild; i++) {

  			/* search for children */

  			if (dev->parent->children[i] == dev) {
  				/* found our pointer, see if we have a
  				 * little sister
  				 */

  				while (i++ < dev->parent->maxchild) {
  					if (dev->parent->children[i] != NULL) {

  						/* found a sister */

  						last_child = 0;

  						break;
  					} /* if */
  				} /* while */
  			} /* device found */
  		} /* for all children of the parent */

  #endif

  		printf("\b+-");
  		/* correct last child */

  		if (last_child && index)

  			pre[index-1] = ' ';

  	} /* if not root hub */
  	else
  		printf(" ");

  	printf("%d ", dev->devnum);
  	pre[index++] = ' ';
  	pre[index++] = has_child ? '|' : ' ';
  	pre[index] = 0;
  	printf(" %s (%s, %dmA)
  ", usb_get_class_desc(

  					dev->config.if_desc[0].desc.bInterfaceClass),

  					portspeed(dev->speed),

  					dev->config.desc.bMaxPower * 2);

  	if (strlen(dev->mf) || strlen(dev->prod) || strlen(dev->serial))
  		printf(" %s  %s %s %s
  ", pre, dev->mf, dev->prod, dev->serial);
  	printf(" %s
  ", pre);

  #ifdef CONFIG_DM_USB
  	struct udevice *child;
  
  	for (device_find_first_child(dev->dev, &child);
  	     child;
  	     device_find_next_child(&child)) {
  		struct usb_device *udev;
  
  		if (!device_active(child))
  			continue;

  		udev = dev_get_parent_priv(child);

  		/*
  		 * Ignore emulators and block child devices, we only want
  		 * real devices
  		 */
  		if ((device_get_uclass_id(child) != UCLASS_USB_EMUL) &&
  		    (device_get_uclass_id(child) != UCLASS_BLK)) {

  			usb_show_tree_graph(udev, pre);
  			pre[index] = 0;
  		}
  	}
  #else

  	if (dev->maxchild > 0) {
  		for (i = 0; i < dev->maxchild; i++) {
  			if (dev->children[i] != NULL) {
  				usb_show_tree_graph(dev->children[i], pre);
  				pre[index] = 0;

  			}
  		}
  	}

  #endif

  }
  
  /* main routine for the tree command */

  static void usb_show_subtree(struct usb_device *dev)

  {
  	char preamble[32];

  	memset(preamble, '\0', sizeof(preamble));

  	usb_show_tree_graph(dev, &preamble[0]);

  }

  #ifdef CONFIG_DM_USB

  typedef void (*usb_dev_func_t)(struct usb_device *udev);
  
  static void usb_for_each_root_dev(usb_dev_func_t func)
  {

  	struct udevice *bus;

  	for (uclass_find_first_device(UCLASS_USB, &bus);

  		bus;

  		uclass_find_next_device(&bus)) {

  		struct usb_device *udev;
  		struct udevice *dev;

  		if (!device_active(bus))
  			continue;

  		device_find_first_child(bus, &dev);
  		if (dev && device_active(dev)) {
  			udev = dev_get_parent_priv(dev);

  			func(udev);

  		}
  	}

  }
  #endif
  
  void usb_show_tree(void)
  {
  #ifdef CONFIG_DM_USB
  	usb_for_each_root_dev(usb_show_subtree);

  #else
  	struct usb_device *udev;
  	int i;
  
  	for (i = 0; i < USB_MAX_DEVICE; i++) {
  		udev = usb_get_dev_index(i);
  		if (udev == NULL)
  			break;
  		if (udev->parent == NULL)
  			usb_show_subtree(udev);
  	}
  #endif
  }

  static int usb_test(struct usb_device *dev, int port, char* arg)
  {
  	int mode;
  
  	if (port > dev->maxchild) {
  		printf("Device is no hub or does not have %d ports.
  ", port);
  		return 1;
  	}
  
  	switch (arg[0]) {
  	case 'J':
  	case 'j':
  		printf("Setting Test_J mode");
  		mode = USB_TEST_MODE_J;
  		break;
  	case 'K':
  	case 'k':
  		printf("Setting Test_K mode");
  		mode = USB_TEST_MODE_K;
  		break;
  	case 'S':
  	case 's':
  		printf("Setting Test_SE0_NAK mode");
  		mode = USB_TEST_MODE_SE0_NAK;
  		break;
  	case 'P':
  	case 'p':
  		printf("Setting Test_Packet mode");
  		mode = USB_TEST_MODE_PACKET;
  		break;
  	case 'F':
  	case 'f':
  		printf("Setting Test_Force_Enable mode");
  		mode = USB_TEST_MODE_FORCE_ENABLE;
  		break;
  	default:
  		printf("Unrecognized test mode: %s
  Available modes: "
  		       "J, K, S[E0_NAK], P[acket], F[orce_Enable]
  ", arg);
  		return 1;
  	}
  
  	if (port)
  		printf(" on downstream facing port %d...
  ", port);
  	else
  		printf(" on upstream facing port...
  ");
  
  	if (usb_control_msg(dev, usb_sndctrlpipe(dev, 0), USB_REQ_SET_FEATURE,
  			    port ? USB_RT_PORT : USB_RECIP_DEVICE,
  			    port ? USB_PORT_FEAT_TEST : USB_FEAT_TEST,
  			    (mode << 8) | port,
  			    NULL, 0, USB_CNTL_TIMEOUT) == -1) {
  		printf("Error during SET_FEATURE.
  ");
  		return 1;
  	} else {
  		printf("Test mode successfully set. Use 'usb start' "
  		       "to return to normal operation.
  ");
  		return 0;
  	}
  }

  /******************************************************************************
   * usb boot command intepreter. Derived from diskboot
   */
  #ifdef CONFIG_USB_STORAGE

  static int do_usbboot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])

  {

  	return common_diskboot(cmdtp, "usb", argc, argv);

  }
  #endif /* CONFIG_USB_STORAGE */

  static int do_usb_stop_keyboard(int force)

  {

  #if !defined CONFIG_DM_USB && defined CONFIG_USB_KEYBOARD

  	if (usb_kbd_deregister(force) != 0) {

  		printf("USB not stopped: usbkbd still using USB
  ");
  		return 1;
  	}
  #endif
  	return 0;
  }

  static void do_usb_start(void)
  {
  	bootstage_mark_name(BOOTSTAGE_ID_USB_START, "usb_start");
  
  	if (usb_init() < 0)
  		return;

  	/* Driver model will probe the devices as they are found */

  # ifdef CONFIG_USB_STORAGE

  	/* try to recognize storage devices immediately */
  	usb_stor_curr_dev = usb_stor_scan(1);

  # endif

  #ifndef CONFIG_DM_USB

  # ifdef CONFIG_USB_KEYBOARD
  	drv_usb_kbd_init();
  # endif
  #endif /* !CONFIG_DM_USB */

  #ifdef CONFIG_USB_HOST_ETHER

  # ifdef CONFIG_DM_ETH

  #  ifndef CONFIG_DM_USB
  #   error "You must use CONFIG_DM_USB if you want to use CONFIG_USB_HOST_ETHER with CONFIG_DM_ETH"
  #  endif
  # else

  	/* try to recognize ethernet devices immediately */
  	usb_ether_curr_dev = usb_host_eth_scan(1);

  # endif

  #endif

  }

  #ifdef CONFIG_DM_USB

  static void usb_show_info(struct usb_device *udev)

  {
  	struct udevice *child;

  	usb_display_desc(udev);
  	usb_display_config(udev);

  	for (device_find_first_child(udev->dev, &child);

  	     child;
  	     device_find_next_child(&child)) {

  		if (device_active(child) &&
  		    (device_get_uclass_id(child) != UCLASS_USB_EMUL) &&
  		    (device_get_uclass_id(child) != UCLASS_BLK)) {

  			udev = dev_get_parent_priv(child);
  			usb_show_info(udev);

  		}
  	}

  }
  #endif

  /******************************************************************************

   * usb command intepreter
   */

  static int do_usb(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])

  {

  	struct usb_device *udev = NULL;

  	int i;

  	extern char usb_started;

  	if (argc < 2)

  		return CMD_RET_USAGE;

  	if (strncmp(argv[1], "start", 5) == 0) {
  		if (usb_started)
  			return 0; /* Already started */
  		printf("starting USB...
  ");
  		do_usb_start();
  		return 0;
  	}
  
  	if (strncmp(argv[1], "reset", 5) == 0) {
  		printf("resetting USB...
  ");

  		if (do_usb_stop_keyboard(1) != 0)

  			return 1;

  		usb_stop();

  		do_usb_start();

  		return 0;
  	}

  	if (strncmp(argv[1], "stop", 4) == 0) {

  		if (argc != 2)

  			console_assign(stdin, "serial");

  		if (do_usb_stop_keyboard(0) != 0)

  			return 1;

  		printf("stopping USB..
  ");
  		usb_stop();
  		return 0;
  	}

  	if (!usb_started) {
  		printf("USB is stopped. Please issue 'usb start' first.
  ");
  		return 1;
  	}

  	if (strncmp(argv[1], "tree", 4) == 0) {

  		puts("USB device tree:
  ");

  		usb_show_tree();

  		return 0;
  	}

  	if (strncmp(argv[1], "inf", 3) == 0) {

  		if (argc == 2) {

  #ifdef CONFIG_DM_USB

  			usb_for_each_root_dev(usb_show_info);

  #else
  			int d;

  			for (d = 0; d < USB_MAX_DEVICE; d++) {

  				udev = usb_get_dev_index(d);
  				if (udev == NULL)

  					break;

  				usb_display_desc(udev);
  				usb_display_config(udev);

  			}

  #endif

  			return 0;

  		} else {

  			/*
  			 * With driver model this isn't right since we can
  			 * have multiple controllers and the device numbering
  			 * starts at 1 on each bus.
  			 */

  			i = simple_strtoul(argv[2], NULL, 10);

  			printf("config for device %d
  ", i);

  			udev = usb_find_device(i);
  			if (udev == NULL) {

  				printf("*** No device available ***
  ");

  				return 0;

  			} else {

  				usb_display_desc(udev);
  				usb_display_config(udev);

  			}
  		}
  		return 0;
  	}

  	if (strncmp(argv[1], "test", 4) == 0) {
  		if (argc < 5)
  			return CMD_RET_USAGE;
  		i = simple_strtoul(argv[2], NULL, 10);

  		udev = usb_find_device(i);
  		if (udev == NULL) {

  			printf("Device %d does not exist.
  ", i);
  			return 1;
  		}
  		i = simple_strtoul(argv[3], NULL, 10);

  		return usb_test(udev, i, argv[4]);

  	}

  #ifdef CONFIG_USB_STORAGE

  	if (strncmp(argv[1], "stor", 4) == 0)

  		return usb_stor_info();

  	return blk_common_cmd(argc, argv, IF_TYPE_USB, &usb_stor_curr_dev);
  #else

  	return CMD_RET_USAGE;

  #endif /* CONFIG_USB_STORAGE */

  }

  U_BOOT_CMD(
  	usb,	5,	1,	do_usb,

  	"USB sub-system",

  	"start - start (scan) USB controller
  "
  	"usb reset - reset (rescan) USB controller
  "

  	"usb stop [f] - stop USB [f]=force stop
  "
  	"usb tree - show USB device tree
  "

  	"usb info [dev] - show available USB devices
  "

  	"usb test [dev] [port] [mode] - set USB 2.0 test mode
  "
  	"    (specify port 0 to indicate the device's upstream port)
  "
  	"    Available modes: J, K, S[E0_NAK], P[acket], F[orce_Enable]
  "
  #ifdef CONFIG_USB_STORAGE

  	"usb storage - show details of USB storage devices
  "

  	"usb dev [dev] - show or set current USB storage device
  "

  	"usb part [dev] - print partition table of one or all USB storage"

  	"    devices
  "

  	"usb read addr blk# cnt - read `cnt' blocks starting at block `blk#'
  "

  	"    to memory address `addr'
  "

  	"usb write addr blk# cnt - write `cnt' blocks starting at block `blk#'
  "
  	"    from memory address `addr'"

  #endif /* CONFIG_USB_STORAGE */

  );

  #ifdef CONFIG_USB_STORAGE

  U_BOOT_CMD(
  	usbboot,	3,	1,	do_usbboot,

  	"boot from USB device",

  	"loadAddr dev:part"

  );

  #endif /* CONFIG_USB_STORAGE */