common/cmd_usb_mass_storage.c

   1 /*
   2  * Copyright (C) 2011 Samsung Electronics
   3  * Lukasz Majewski <[email protected]>
   4  *
   5  * SPDX-License-Identifier:     GPL-2.0+
   6  */
   7
   8 #include <errno.h>
   9 #include <common.h>
  10 #include <command.h>
  11 #include <g_dnl.h>
  12 #include <usb.h>
  13 #include <usb_mass_storage.h>
  14
  15 int do_usb_mass_storage(cmd_tbl_t *cmdtp, int flag,
  16                                int argc, char * const argv[])
  17 {
  18         if (argc < 3)
  19                 return CMD_RET_USAGE;
  20
  21         const char *usb_controller = argv[1];
  22         const char *mmc_devstring  = argv[2];
  23
  24         unsigned int dev_num = simple_strtoul(mmc_devstring, NULL, 0);
  25
  26         struct ums *ums = ums_init(dev_num);
  27         if (!ums)
  28                 return CMD_RET_FAILURE;
  29
  30         unsigned int controller_index = (unsigned int)(simple_strtoul(
  31                                         usb_controller, NULL, 0));
  32         if (board_usb_init(controller_index, USB_INIT_DEVICE)) {
  33                 error("Couldn't init USB controller.");
  34                 return CMD_RET_FAILURE;
  35         }
  36
  37         int rc = fsg_init(ums);
  38         if (rc) {
  39                 error("fsg_init failed");
  40                 return CMD_RET_FAILURE;
  41         }
  42
  43         g_dnl_register("ums");
  44
  45         /* Timeout unit: seconds */
  46         int cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
  47
  48         if (!usb_cable_connected()) {
  49                 puts("Please connect USB cable.\n");
  50
  51                 while (!usb_cable_connected()) {
  52                         if (ctrlc()) {
  53                                 puts("\rCTRL+C - Operation aborted.\n");
  54                                 goto exit;
  55                         }
  56                         if (!cable_ready_timeout) {
  57                                 puts("\rUSB cable not detected.\n" \
  58                                      "Command exit.\n");
  59                                 goto exit;
  60                         }
  61
  62                         printf("\rAuto exit in: %.2d s.", cable_ready_timeout);
  63                         mdelay(1000);
  64                         cable_ready_timeout--;
  65                 }
  66                 puts("\r\n");
  67         }
  68
  69         while (1) {
  70                 usb_gadget_handle_interrupts();
  71
  72                 rc = fsg_main_thread(NULL);
  73                 if (rc) {
  74                         /* Check I/O error */
  75                         if (rc == -EIO)
  76                                 printf("\rCheck USB cable connection\n");
  77
  78                         /* Check CTRL+C */
  79                         if (rc == -EPIPE)
  80                                 printf("\rCTRL+C - Operation aborted\n");
  81
  82                         goto exit;
  83                 }
  84         }
  85 exit:
  86         g_dnl_unregister();
  87         return CMD_RET_SUCCESS;
  88 }
  89
  90 U_BOOT_CMD(ums, CONFIG_SYS_MAXARGS, 1, do_usb_mass_storage,
  91         "Use the UMS [User Mass Storage]",
  92         "ums <USB_controller> <mmc_dev>  e.g. ums 0 0"
  93 );