2 * Copyright (C) 2011 Samsung Electronics
3 * Lukasz Majewski <l.majewski@samsung.com>
5 * SPDX-License-Identifier: GPL-2.0+
15 #include <usb_mass_storage.h>
17 static int ums_read_sector(struct ums *ums_dev,
18 ulong start, lbaint_t blkcnt, void *buf)
20 block_dev_desc_t *block_dev = ums_dev->block_dev;
21 lbaint_t blkstart = start + ums_dev->start_sector;
22 int dev_num = block_dev->dev;
24 return block_dev->block_read(dev_num, blkstart, blkcnt, buf);
27 static int ums_write_sector(struct ums *ums_dev,
28 ulong start, lbaint_t blkcnt, const void *buf)
30 block_dev_desc_t *block_dev = ums_dev->block_dev;
31 lbaint_t blkstart = start + ums_dev->start_sector;
32 int dev_num = block_dev->dev;
34 return block_dev->block_write(dev_num, blkstart, blkcnt, buf);
37 static struct ums ums_dev = {
38 .read_sector = ums_read_sector,
39 .write_sector = ums_write_sector,
43 struct ums *ums_init(unsigned int dev_num)
45 struct mmc *mmc = NULL;
47 mmc = find_mmc_device(dev_num);
48 if (!mmc || mmc_init(mmc))
51 ums_dev.block_dev = &mmc->block_dev;
52 ums_dev.start_sector = 0;
53 ums_dev.num_sectors = mmc->capacity / SECTOR_SIZE;
55 printf("UMS: disk start sector: %#x, count: %#x\n",
56 ums_dev.start_sector, ums_dev.num_sectors);
61 int do_usb_mass_storage(cmd_tbl_t *cmdtp, int flag,
62 int argc, char * const argv[])
67 const char *usb_controller = argv[1];
68 const char *mmc_devstring = argv[2];
70 unsigned int dev_num = simple_strtoul(mmc_devstring, NULL, 0);
72 struct ums *ums = ums_init(dev_num);
74 return CMD_RET_FAILURE;
76 unsigned int controller_index = (unsigned int)(simple_strtoul(
77 usb_controller, NULL, 0));
78 if (board_usb_init(controller_index, USB_INIT_DEVICE)) {
79 error("Couldn't init USB controller.");
80 return CMD_RET_FAILURE;
83 int rc = fsg_init(ums);
85 error("fsg_init failed");
86 return CMD_RET_FAILURE;
89 rc = g_dnl_register("usb_dnl_ums");
91 error("g_dnl_register failed");
92 return CMD_RET_FAILURE;
95 /* Timeout unit: seconds */
96 int cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
98 if (!g_dnl_board_usb_cable_connected()) {
100 * Won't execute if we don't know whether the cable is
103 puts("Please connect USB cable.\n");
105 while (!g_dnl_board_usb_cable_connected()) {
107 puts("\rCTRL+C - Operation aborted.\n");
110 if (!cable_ready_timeout) {
111 puts("\rUSB cable not detected.\n" \
116 printf("\rAuto exit in: %.2d s.", cable_ready_timeout);
118 cable_ready_timeout--;
124 usb_gadget_handle_interrupts();
126 rc = fsg_main_thread(NULL);
128 /* Check I/O error */
130 printf("\rCheck USB cable connection\n");
134 printf("\rCTRL+C - Operation aborted\n");
141 return CMD_RET_SUCCESS;
144 U_BOOT_CMD(ums, CONFIG_SYS_MAXARGS, 1, do_usb_mass_storage,
145 "Use the UMS [User Mass Storage]",
146 "ums <USB_controller> <mmc_dev> e.g. ums 0 0"