183d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
22e192b24SSimon Glass /*
32e192b24SSimon Glass * Copyright (C) 2011 Samsung Electronics
42e192b24SSimon Glass * Lukasz Majewski <l.majewski@samsung.com>
52e192b24SSimon Glass *
62e192b24SSimon Glass * Copyright (c) 2015, NVIDIA CORPORATION. All rights reserved.
72e192b24SSimon Glass */
82e192b24SSimon Glass
92e192b24SSimon Glass #include <errno.h>
102e192b24SSimon Glass #include <common.h>
112e192b24SSimon Glass #include <command.h>
122e192b24SSimon Glass #include <console.h>
132e192b24SSimon Glass #include <g_dnl.h>
142e192b24SSimon Glass #include <part.h>
152e192b24SSimon Glass #include <usb.h>
162e192b24SSimon Glass #include <usb_mass_storage.h>
172e192b24SSimon Glass
ums_read_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,void * buf)182e192b24SSimon Glass static int ums_read_sector(struct ums *ums_dev,
192e192b24SSimon Glass ulong start, lbaint_t blkcnt, void *buf)
202e192b24SSimon Glass {
214101f687SSimon Glass struct blk_desc *block_dev = &ums_dev->block_dev;
222e192b24SSimon Glass lbaint_t blkstart = start + ums_dev->start_sector;
232e192b24SSimon Glass
24c9f3c5f9SSimon Glass return blk_dread(block_dev, blkstart, blkcnt, buf);
252e192b24SSimon Glass }
262e192b24SSimon Glass
ums_write_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,const void * buf)272e192b24SSimon Glass static int ums_write_sector(struct ums *ums_dev,
282e192b24SSimon Glass ulong start, lbaint_t blkcnt, const void *buf)
292e192b24SSimon Glass {
304101f687SSimon Glass struct blk_desc *block_dev = &ums_dev->block_dev;
312e192b24SSimon Glass lbaint_t blkstart = start + ums_dev->start_sector;
322e192b24SSimon Glass
33c9f3c5f9SSimon Glass return blk_dwrite(block_dev, blkstart, blkcnt, buf);
342e192b24SSimon Glass }
352e192b24SSimon Glass
362e192b24SSimon Glass static struct ums *ums;
372e192b24SSimon Glass static int ums_count;
382e192b24SSimon Glass
ums_fini(void)392e192b24SSimon Glass static void ums_fini(void)
402e192b24SSimon Glass {
412e192b24SSimon Glass int i;
422e192b24SSimon Glass
432e192b24SSimon Glass for (i = 0; i < ums_count; i++)
442e192b24SSimon Glass free((void *)ums[i].name);
452e192b24SSimon Glass free(ums);
46d419026aSFabio Estevam ums = NULL;
472e192b24SSimon Glass ums_count = 0;
482e192b24SSimon Glass }
492e192b24SSimon Glass
502e192b24SSimon Glass #define UMS_NAME_LEN 16
512e192b24SSimon Glass
ums_init(const char * devtype,const char * devnums_part_str)52a2e3a1d8SJohn Tobias static int ums_init(const char *devtype, const char *devnums_part_str)
532e192b24SSimon Glass {
54a2e3a1d8SJohn Tobias char *s, *t, *devnum_part_str, *name;
554101f687SSimon Glass struct blk_desc *block_dev;
56a2e3a1d8SJohn Tobias disk_partition_t info;
57a2e3a1d8SJohn Tobias int partnum;
58a238b0daSTom Rini int ret = -1;
592e192b24SSimon Glass struct ums *ums_new;
602e192b24SSimon Glass
61a2e3a1d8SJohn Tobias s = strdup(devnums_part_str);
622e192b24SSimon Glass if (!s)
632e192b24SSimon Glass return -1;
642e192b24SSimon Glass
652e192b24SSimon Glass t = s;
662e192b24SSimon Glass ums_count = 0;
672e192b24SSimon Glass
682e192b24SSimon Glass for (;;) {
69a2e3a1d8SJohn Tobias devnum_part_str = strsep(&t, ",");
70a2e3a1d8SJohn Tobias if (!devnum_part_str)
712e192b24SSimon Glass break;
722e192b24SSimon Glass
73a2e3a1d8SJohn Tobias partnum = blk_get_device_part_str(devtype, devnum_part_str,
74a2e3a1d8SJohn Tobias &block_dev, &info, 1);
75a2e3a1d8SJohn Tobias
76a2e3a1d8SJohn Tobias if (partnum < 0)
772e192b24SSimon Glass goto cleanup;
782e192b24SSimon Glass
79a2e3a1d8SJohn Tobias /* Check if the argument is in legacy format. If yes,
80a2e3a1d8SJohn Tobias * expose all partitions by setting the partnum = 0
81a2e3a1d8SJohn Tobias * e.g. ums 0 mmc 0
82a2e3a1d8SJohn Tobias */
83a2e3a1d8SJohn Tobias if (!strchr(devnum_part_str, ':'))
84a2e3a1d8SJohn Tobias partnum = 0;
85a2e3a1d8SJohn Tobias
862e192b24SSimon Glass /* f_mass_storage.c assumes SECTOR_SIZE sectors */
87a238b0daSTom Rini if (block_dev->blksz != SECTOR_SIZE)
882e192b24SSimon Glass goto cleanup;
892e192b24SSimon Glass
902e192b24SSimon Glass ums_new = realloc(ums, (ums_count + 1) * sizeof(*ums));
91a238b0daSTom Rini if (!ums_new)
922e192b24SSimon Glass goto cleanup;
932e192b24SSimon Glass ums = ums_new;
942e192b24SSimon Glass
95a2e3a1d8SJohn Tobias /* if partnum = 0, expose all partitions */
96a2e3a1d8SJohn Tobias if (partnum == 0) {
972e192b24SSimon Glass ums[ums_count].start_sector = 0;
982e192b24SSimon Glass ums[ums_count].num_sectors = block_dev->lba;
99a2e3a1d8SJohn Tobias } else {
100a2e3a1d8SJohn Tobias ums[ums_count].start_sector = info.start;
101a2e3a1d8SJohn Tobias ums[ums_count].num_sectors = info.size;
102a2e3a1d8SJohn Tobias }
103a2e3a1d8SJohn Tobias
104a2e3a1d8SJohn Tobias ums[ums_count].read_sector = ums_read_sector;
105a2e3a1d8SJohn Tobias ums[ums_count].write_sector = ums_write_sector;
106a2e3a1d8SJohn Tobias
1072e192b24SSimon Glass name = malloc(UMS_NAME_LEN);
108a238b0daSTom Rini if (!name)
1092e192b24SSimon Glass goto cleanup;
1102e192b24SSimon Glass snprintf(name, UMS_NAME_LEN, "UMS disk %d", ums_count);
1112e192b24SSimon Glass ums[ums_count].name = name;
1122e192b24SSimon Glass ums[ums_count].block_dev = *block_dev;
1132e192b24SSimon Glass
1142e192b24SSimon Glass printf("UMS: LUN %d, dev %d, hwpart %d, sector %#x, count %#x\n",
115bcce53d0SSimon Glass ums_count, ums[ums_count].block_dev.devnum,
1162e192b24SSimon Glass ums[ums_count].block_dev.hwpart,
1172e192b24SSimon Glass ums[ums_count].start_sector,
1182e192b24SSimon Glass ums[ums_count].num_sectors);
1192e192b24SSimon Glass
1202e192b24SSimon Glass ums_count++;
1212e192b24SSimon Glass }
1222e192b24SSimon Glass
123a238b0daSTom Rini if (ums_count)
1242e192b24SSimon Glass ret = 0;
1252e192b24SSimon Glass
1262e192b24SSimon Glass cleanup:
1272e192b24SSimon Glass free(s);
1282e192b24SSimon Glass
1292e192b24SSimon Glass if (ret < 0)
1302e192b24SSimon Glass ums_fini();
1312e192b24SSimon Glass
1322e192b24SSimon Glass return ret;
1332e192b24SSimon Glass }
1342e192b24SSimon Glass
do_usb_mass_storage(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])135b9203429SFabio Estevam static int do_usb_mass_storage(cmd_tbl_t *cmdtp, int flag,
1362e192b24SSimon Glass int argc, char * const argv[])
1372e192b24SSimon Glass {
1382e192b24SSimon Glass const char *usb_controller;
1392e192b24SSimon Glass const char *devtype;
1402e192b24SSimon Glass const char *devnum;
1412e192b24SSimon Glass unsigned int controller_index;
1422e192b24SSimon Glass int rc;
1432e192b24SSimon Glass int cable_ready_timeout __maybe_unused;
1442e192b24SSimon Glass
1452e192b24SSimon Glass if (argc < 3)
1462e192b24SSimon Glass return CMD_RET_USAGE;
1472e192b24SSimon Glass
1482e192b24SSimon Glass usb_controller = argv[1];
1492e192b24SSimon Glass if (argc >= 4) {
1502e192b24SSimon Glass devtype = argv[2];
1512e192b24SSimon Glass devnum = argv[3];
1522e192b24SSimon Glass } else {
1532e192b24SSimon Glass devtype = "mmc";
1542e192b24SSimon Glass devnum = argv[2];
1552e192b24SSimon Glass }
1562e192b24SSimon Glass
1572e192b24SSimon Glass rc = ums_init(devtype, devnum);
1582e192b24SSimon Glass if (rc < 0)
1592e192b24SSimon Glass return CMD_RET_FAILURE;
1602e192b24SSimon Glass
1612e192b24SSimon Glass controller_index = (unsigned int)(simple_strtoul(
1622e192b24SSimon Glass usb_controller, NULL, 0));
163*a06955aeSJean-Jacques Hiblot if (usb_gadget_initialize(controller_index)) {
16471002b50SSeung-Woo Kim pr_err("Couldn't init USB controller.\n");
1652e192b24SSimon Glass rc = CMD_RET_FAILURE;
1662e192b24SSimon Glass goto cleanup_ums_init;
1672e192b24SSimon Glass }
1682e192b24SSimon Glass
1692e192b24SSimon Glass rc = fsg_init(ums, ums_count);
1702e192b24SSimon Glass if (rc) {
17171002b50SSeung-Woo Kim pr_err("fsg_init failed\n");
1722e192b24SSimon Glass rc = CMD_RET_FAILURE;
1732e192b24SSimon Glass goto cleanup_board;
1742e192b24SSimon Glass }
1752e192b24SSimon Glass
1762e192b24SSimon Glass rc = g_dnl_register("usb_dnl_ums");
1772e192b24SSimon Glass if (rc) {
17871002b50SSeung-Woo Kim pr_err("g_dnl_register failed\n");
1792e192b24SSimon Glass rc = CMD_RET_FAILURE;
1802e192b24SSimon Glass goto cleanup_board;
1812e192b24SSimon Glass }
1822e192b24SSimon Glass
1832e192b24SSimon Glass /* Timeout unit: seconds */
1842e192b24SSimon Glass cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
1852e192b24SSimon Glass
1862e192b24SSimon Glass if (!g_dnl_board_usb_cable_connected()) {
1872e192b24SSimon Glass /*
1882e192b24SSimon Glass * Won't execute if we don't know whether the cable is
1892e192b24SSimon Glass * connected.
1902e192b24SSimon Glass */
1912e192b24SSimon Glass puts("Please connect USB cable.\n");
1922e192b24SSimon Glass
1932e192b24SSimon Glass while (!g_dnl_board_usb_cable_connected()) {
1942e192b24SSimon Glass if (ctrlc()) {
1952e192b24SSimon Glass puts("\rCTRL+C - Operation aborted.\n");
1962e192b24SSimon Glass rc = CMD_RET_SUCCESS;
1972e192b24SSimon Glass goto cleanup_register;
1982e192b24SSimon Glass }
1992e192b24SSimon Glass if (!cable_ready_timeout) {
2002e192b24SSimon Glass puts("\rUSB cable not detected.\n" \
2012e192b24SSimon Glass "Command exit.\n");
2022e192b24SSimon Glass rc = CMD_RET_SUCCESS;
2032e192b24SSimon Glass goto cleanup_register;
2042e192b24SSimon Glass }
2052e192b24SSimon Glass
2062e192b24SSimon Glass printf("\rAuto exit in: %.2d s.", cable_ready_timeout);
2072e192b24SSimon Glass mdelay(1000);
2082e192b24SSimon Glass cable_ready_timeout--;
2092e192b24SSimon Glass }
2102e192b24SSimon Glass puts("\r\n");
2112e192b24SSimon Glass }
2122e192b24SSimon Glass
2132e192b24SSimon Glass while (1) {
2142e192b24SSimon Glass usb_gadget_handle_interrupts(controller_index);
2152e192b24SSimon Glass
2162e192b24SSimon Glass rc = fsg_main_thread(NULL);
2172e192b24SSimon Glass if (rc) {
2182e192b24SSimon Glass /* Check I/O error */
2192e192b24SSimon Glass if (rc == -EIO)
2202e192b24SSimon Glass printf("\rCheck USB cable connection\n");
2212e192b24SSimon Glass
2222e192b24SSimon Glass /* Check CTRL+C */
2232e192b24SSimon Glass if (rc == -EPIPE)
2242e192b24SSimon Glass printf("\rCTRL+C - Operation aborted\n");
2252e192b24SSimon Glass
2262e192b24SSimon Glass rc = CMD_RET_SUCCESS;
2272e192b24SSimon Glass goto cleanup_register;
2282e192b24SSimon Glass }
2292e192b24SSimon Glass }
2302e192b24SSimon Glass
2312e192b24SSimon Glass cleanup_register:
2322e192b24SSimon Glass g_dnl_unregister();
2332e192b24SSimon Glass cleanup_board:
234*a06955aeSJean-Jacques Hiblot usb_gadget_release(controller_index);
2352e192b24SSimon Glass cleanup_ums_init:
2362e192b24SSimon Glass ums_fini();
2372e192b24SSimon Glass
2382e192b24SSimon Glass return rc;
2392e192b24SSimon Glass }
2402e192b24SSimon Glass
2412e192b24SSimon Glass U_BOOT_CMD(ums, 4, 1, do_usb_mass_storage,
2422e192b24SSimon Glass "Use the UMS [USB Mass Storage]",
243a2e3a1d8SJohn Tobias "<USB_controller> [<devtype>] <dev[:part]> e.g. ums 0 mmc 0\n"
2442e192b24SSimon Glass " devtype defaults to mmc"
2452e192b24SSimon Glass );
246