blob: 753ae4f42a7a0bd3671032f14ddfb61a82205a71 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Lukasz Majewskib528f712013-03-05 12:10:17 +01002/*
3 * Copyright (C) 2011 Samsung Electronics
4 * Lukasz Majewski <l.majewski@samsung.com>
5 *
Stephen Warren02585eb2015-12-07 11:38:50 -07006 * Copyright (c) 2015, NVIDIA CORPORATION. All rights reserved.
Lukasz Majewskib528f712013-03-05 12:10:17 +01007 */
8
Przemyslaw Marczak351e9b22013-10-23 14:30:46 +02009#include <errno.h>
Lukasz Majewskib528f712013-03-05 12:10:17 +010010#include <common.h>
11#include <command.h>
Simon Glass24b852a2015-11-08 23:47:45 -070012#include <console.h>
Lukasz Majewskib528f712013-03-05 12:10:17 +010013#include <g_dnl.h>
Stephen Warrenabfe8af2014-05-05 10:40:15 -060014#include <part.h>
Mateusz Zalega16297cf2013-10-04 19:22:26 +020015#include <usb.h>
Lukasz Majewskib528f712013-03-05 12:10:17 +010016#include <usb_mass_storage.h>
17
Stephen Warrenabfe8af2014-05-05 10:40:15 -060018static int ums_read_sector(struct ums *ums_dev,
19 ulong start, lbaint_t blkcnt, void *buf)
20{
Simon Glass4101f682016-02-29 15:25:34 -070021 struct blk_desc *block_dev = &ums_dev->block_dev;
Stephen Warrenabfe8af2014-05-05 10:40:15 -060022 lbaint_t blkstart = start + ums_dev->start_sector;
Stephen Warrenabfe8af2014-05-05 10:40:15 -060023
Simon Glassc9f3c5f2016-07-05 17:10:17 -060024 return blk_dread(block_dev, blkstart, blkcnt, buf);
Stephen Warrenabfe8af2014-05-05 10:40:15 -060025}
26
27static int ums_write_sector(struct ums *ums_dev,
28 ulong start, lbaint_t blkcnt, const void *buf)
29{
Simon Glass4101f682016-02-29 15:25:34 -070030 struct blk_desc *block_dev = &ums_dev->block_dev;
Stephen Warrenabfe8af2014-05-05 10:40:15 -060031 lbaint_t blkstart = start + ums_dev->start_sector;
Stephen Warrenabfe8af2014-05-05 10:40:15 -060032
Simon Glassc9f3c5f2016-07-05 17:10:17 -060033 return blk_dwrite(block_dev, blkstart, blkcnt, buf);
Stephen Warrenabfe8af2014-05-05 10:40:15 -060034}
35
Stephen Warren02585eb2015-12-07 11:38:50 -070036static struct ums *ums;
37static int ums_count;
Stephen Warrenabfe8af2014-05-05 10:40:15 -060038
Stephen Warren02585eb2015-12-07 11:38:50 -070039static void ums_fini(void)
Stephen Warrenabfe8af2014-05-05 10:40:15 -060040{
Stephen Warren02585eb2015-12-07 11:38:50 -070041 int i;
42
43 for (i = 0; i < ums_count; i++)
44 free((void *)ums[i].name);
45 free(ums);
Fabio Estevamd4190262016-05-27 22:37:10 -030046 ums = NULL;
Stephen Warren02585eb2015-12-07 11:38:50 -070047 ums_count = 0;
48}
49
50#define UMS_NAME_LEN 16
51
John Tobiasa2e3a1d2016-04-13 12:06:51 -070052static int ums_init(const char *devtype, const char *devnums_part_str)
Stephen Warren02585eb2015-12-07 11:38:50 -070053{
John Tobiasa2e3a1d2016-04-13 12:06:51 -070054 char *s, *t, *devnum_part_str, *name;
Simon Glass4101f682016-02-29 15:25:34 -070055 struct blk_desc *block_dev;
John Tobiasa2e3a1d2016-04-13 12:06:51 -070056 disk_partition_t info;
57 int partnum;
Tom Rinia238b0d2016-04-20 10:59:32 -040058 int ret = -1;
Stephen Warren02585eb2015-12-07 11:38:50 -070059 struct ums *ums_new;
Stephen Warrenabfe8af2014-05-05 10:40:15 -060060
John Tobiasa2e3a1d2016-04-13 12:06:51 -070061 s = strdup(devnums_part_str);
Stephen Warren02585eb2015-12-07 11:38:50 -070062 if (!s)
63 return -1;
64
65 t = s;
66 ums_count = 0;
67
68 for (;;) {
John Tobiasa2e3a1d2016-04-13 12:06:51 -070069 devnum_part_str = strsep(&t, ",");
70 if (!devnum_part_str)
Stephen Warren02585eb2015-12-07 11:38:50 -070071 break;
72
John Tobiasa2e3a1d2016-04-13 12:06:51 -070073 partnum = blk_get_device_part_str(devtype, devnum_part_str,
74 &block_dev, &info, 1);
75
76 if (partnum < 0)
Stephen Warren02585eb2015-12-07 11:38:50 -070077 goto cleanup;
78
John Tobiasa2e3a1d2016-04-13 12:06:51 -070079 /* Check if the argument is in legacy format. If yes,
80 * expose all partitions by setting the partnum = 0
81 * e.g. ums 0 mmc 0
82 */
83 if (!strchr(devnum_part_str, ':'))
84 partnum = 0;
85
Stephen Warren02585eb2015-12-07 11:38:50 -070086 /* f_mass_storage.c assumes SECTOR_SIZE sectors */
Tom Rinia238b0d2016-04-20 10:59:32 -040087 if (block_dev->blksz != SECTOR_SIZE)
Stephen Warren02585eb2015-12-07 11:38:50 -070088 goto cleanup;
Stephen Warren02585eb2015-12-07 11:38:50 -070089
90 ums_new = realloc(ums, (ums_count + 1) * sizeof(*ums));
Tom Rinia238b0d2016-04-20 10:59:32 -040091 if (!ums_new)
Stephen Warren02585eb2015-12-07 11:38:50 -070092 goto cleanup;
Stephen Warren02585eb2015-12-07 11:38:50 -070093 ums = ums_new;
94
John Tobiasa2e3a1d2016-04-13 12:06:51 -070095 /* if partnum = 0, expose all partitions */
96 if (partnum == 0) {
97 ums[ums_count].start_sector = 0;
98 ums[ums_count].num_sectors = block_dev->lba;
99 } else {
100 ums[ums_count].start_sector = info.start;
101 ums[ums_count].num_sectors = info.size;
102 }
103
Stephen Warren02585eb2015-12-07 11:38:50 -0700104 ums[ums_count].read_sector = ums_read_sector;
105 ums[ums_count].write_sector = ums_write_sector;
John Tobiasa2e3a1d2016-04-13 12:06:51 -0700106
Stephen Warren02585eb2015-12-07 11:38:50 -0700107 name = malloc(UMS_NAME_LEN);
Tom Rinia238b0d2016-04-20 10:59:32 -0400108 if (!name)
Stephen Warren02585eb2015-12-07 11:38:50 -0700109 goto cleanup;
Stephen Warren02585eb2015-12-07 11:38:50 -0700110 snprintf(name, UMS_NAME_LEN, "UMS disk %d", ums_count);
111 ums[ums_count].name = name;
112 ums[ums_count].block_dev = *block_dev;
113
114 printf("UMS: LUN %d, dev %d, hwpart %d, sector %#x, count %#x\n",
Simon Glassbcce53d2016-02-29 15:25:51 -0700115 ums_count, ums[ums_count].block_dev.devnum,
Stephen Warren02585eb2015-12-07 11:38:50 -0700116 ums[ums_count].block_dev.hwpart,
117 ums[ums_count].start_sector,
118 ums[ums_count].num_sectors);
119
120 ums_count++;
121 }
122
Tom Rinia238b0d2016-04-20 10:59:32 -0400123 if (ums_count)
Stephen Warren02585eb2015-12-07 11:38:50 -0700124 ret = 0;
125
126cleanup:
127 free(s);
128
Stephen Warrend0cc4562014-05-05 10:40:16 -0600129 if (ret < 0)
Stephen Warren02585eb2015-12-07 11:38:50 -0700130 ums_fini();
Stephen Warrenabfe8af2014-05-05 10:40:15 -0600131
Stephen Warren02585eb2015-12-07 11:38:50 -0700132 return ret;
Stephen Warrenabfe8af2014-05-05 10:40:15 -0600133}
134
Fabio Estevamb9203422016-05-27 22:37:11 -0300135static int do_usb_mass_storage(cmd_tbl_t *cmdtp, int flag,
Lukasz Majewskib528f712013-03-05 12:10:17 +0100136 int argc, char * const argv[])
137{
Stephen Warren1725f122014-05-05 10:40:17 -0600138 const char *usb_controller;
139 const char *devtype;
140 const char *devnum;
Stephen Warren1725f122014-05-05 10:40:17 -0600141 unsigned int controller_index;
142 int rc;
143 int cable_ready_timeout __maybe_unused;
144
Mateusz Zalega16297cf2013-10-04 19:22:26 +0200145 if (argc < 3)
146 return CMD_RET_USAGE;
Lukasz Majewskib528f712013-03-05 12:10:17 +0100147
Stephen Warren1725f122014-05-05 10:40:17 -0600148 usb_controller = argv[1];
Stephen Warren8c600452014-05-05 10:40:18 -0600149 if (argc >= 4) {
150 devtype = argv[2];
151 devnum = argv[3];
152 } else {
153 devtype = "mmc";
154 devnum = argv[2];
155 }
Lukasz Majewskib528f712013-03-05 12:10:17 +0100156
Stephen Warren02585eb2015-12-07 11:38:50 -0700157 rc = ums_init(devtype, devnum);
158 if (rc < 0)
Przemyslaw Marczakf4dacf72013-10-23 14:30:43 +0200159 return CMD_RET_FAILURE;
Lukasz Majewskib528f712013-03-05 12:10:17 +0100160
Stephen Warren1725f122014-05-05 10:40:17 -0600161 controller_index = (unsigned int)(simple_strtoul(
162 usb_controller, NULL, 0));
Jean-Jacques Hiblota06955a2018-11-29 10:52:41 +0100163 if (usb_gadget_initialize(controller_index)) {
Seung-Woo Kim71002b52018-06-04 16:04:51 +0900164 pr_err("Couldn't init USB controller.\n");
Stephen Warren02585eb2015-12-07 11:38:50 -0700165 rc = CMD_RET_FAILURE;
166 goto cleanup_ums_init;
Mateusz Zalega16297cf2013-10-04 19:22:26 +0200167 }
Lukasz Majewskib528f712013-03-05 12:10:17 +0100168
Stephen Warren02585eb2015-12-07 11:38:50 -0700169 rc = fsg_init(ums, ums_count);
Mateusz Zalega16297cf2013-10-04 19:22:26 +0200170 if (rc) {
Seung-Woo Kim71002b52018-06-04 16:04:51 +0900171 pr_err("fsg_init failed\n");
Stephen Warren02585eb2015-12-07 11:38:50 -0700172 rc = CMD_RET_FAILURE;
173 goto cleanup_board;
Mateusz Zalega16297cf2013-10-04 19:22:26 +0200174 }
175
Stephen Warren66b88b02014-05-01 15:42:10 -0600176 rc = g_dnl_register("usb_dnl_ums");
177 if (rc) {
Seung-Woo Kim71002b52018-06-04 16:04:51 +0900178 pr_err("g_dnl_register failed\n");
Stephen Warren02585eb2015-12-07 11:38:50 -0700179 rc = CMD_RET_FAILURE;
180 goto cleanup_board;
Stephen Warren66b88b02014-05-01 15:42:10 -0600181 }
Lukasz Majewskib528f712013-03-05 12:10:17 +0100182
Przemyslaw Marczak3603e312014-01-07 15:08:37 +0100183 /* Timeout unit: seconds */
Stephen Warren1725f122014-05-05 10:40:17 -0600184 cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
Przemyslaw Marczak3603e312014-01-07 15:08:37 +0100185
Mateusz Zalega75504e92014-04-30 13:07:48 +0200186 if (!g_dnl_board_usb_cable_connected()) {
187 /*
188 * Won't execute if we don't know whether the cable is
189 * connected.
190 */
Przemyslaw Marczak3603e312014-01-07 15:08:37 +0100191 puts("Please connect USB cable.\n");
192
Mateusz Zalega75504e92014-04-30 13:07:48 +0200193 while (!g_dnl_board_usb_cable_connected()) {
Przemyslaw Marczak3603e312014-01-07 15:08:37 +0100194 if (ctrlc()) {
195 puts("\rCTRL+C - Operation aborted.\n");
Stephen Warren02585eb2015-12-07 11:38:50 -0700196 rc = CMD_RET_SUCCESS;
197 goto cleanup_register;
Przemyslaw Marczak3603e312014-01-07 15:08:37 +0100198 }
199 if (!cable_ready_timeout) {
200 puts("\rUSB cable not detected.\n" \
201 "Command exit.\n");
Stephen Warren02585eb2015-12-07 11:38:50 -0700202 rc = CMD_RET_SUCCESS;
203 goto cleanup_register;
Przemyslaw Marczak3603e312014-01-07 15:08:37 +0100204 }
205
206 printf("\rAuto exit in: %.2d s.", cable_ready_timeout);
207 mdelay(1000);
208 cable_ready_timeout--;
209 }
210 puts("\r\n");
211 }
212
Lukasz Majewskib528f712013-03-05 12:10:17 +0100213 while (1) {
Kishon Vijay Abraham I2d48aa62015-02-23 18:40:23 +0530214 usb_gadget_handle_interrupts(controller_index);
Przemyslaw Marczak351e9b22013-10-23 14:30:46 +0200215
216 rc = fsg_main_thread(NULL);
217 if (rc) {
218 /* Check I/O error */
219 if (rc == -EIO)
220 printf("\rCheck USB cable connection\n");
221
222 /* Check CTRL+C */
223 if (rc == -EPIPE)
224 printf("\rCTRL+C - Operation aborted\n");
225
Stephen Warren02585eb2015-12-07 11:38:50 -0700226 rc = CMD_RET_SUCCESS;
227 goto cleanup_register;
Lukasz Majewskib528f712013-03-05 12:10:17 +0100228 }
Lukasz Majewskib528f712013-03-05 12:10:17 +0100229 }
Stephen Warren02585eb2015-12-07 11:38:50 -0700230
231cleanup_register:
Lukasz Majewskib528f712013-03-05 12:10:17 +0100232 g_dnl_unregister();
Stephen Warren02585eb2015-12-07 11:38:50 -0700233cleanup_board:
Jean-Jacques Hiblota06955a2018-11-29 10:52:41 +0100234 usb_gadget_release(controller_index);
Stephen Warren02585eb2015-12-07 11:38:50 -0700235cleanup_ums_init:
236 ums_fini();
237
238 return rc;
Lukasz Majewskib528f712013-03-05 12:10:17 +0100239}
240
Stephen Warren8c600452014-05-05 10:40:18 -0600241U_BOOT_CMD(ums, 4, 1, do_usb_mass_storage,
Fabio Estevamee02a652015-02-28 15:11:46 -0300242 "Use the UMS [USB Mass Storage]",
John Tobiasa2e3a1d2016-04-13 12:06:51 -0700243 "<USB_controller> [<devtype>] <dev[:part]> e.g. ums 0 mmc 0\n"
Stephen Warren8c600452014-05-05 10:40:18 -0600244 " devtype defaults to mmc"
Lukasz Majewskib528f712013-03-05 12:10:17 +0100245);