1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (C) 2011 Samsung Electronics
4 * Lukasz Majewski <l.majewski@samsung.com>
5 *
6 * Copyright (c) 2015, NVIDIA CORPORATION. All rights reserved.
7 */
8
9 #include <errno.h>
10 #include <common.h>
11 #include <command.h>
12 #include <console.h>
13 #include <g_dnl.h>
14 #include <part.h>
15 #include <usb.h>
16 #include <usb_mass_storage.h>
17 #include <watchdog.h>
18
ums_read_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,void * buf)19 static int ums_read_sector(struct ums *ums_dev,
20 ulong start, lbaint_t blkcnt, void *buf)
21 {
22 struct blk_desc *block_dev = &ums_dev->block_dev;
23 lbaint_t blkstart = start + ums_dev->start_sector;
24
25 return blk_dread(block_dev, blkstart, blkcnt, buf);
26 }
27
ums_write_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,const void * buf)28 static int ums_write_sector(struct ums *ums_dev,
29 ulong start, lbaint_t blkcnt, const void *buf)
30 {
31 struct blk_desc *block_dev = &ums_dev->block_dev;
32 lbaint_t blkstart = start + ums_dev->start_sector;
33
34 return blk_dwrite(block_dev, blkstart, blkcnt, buf);
35 }
36
37 static struct ums *ums;
38 static int ums_count;
39
ums_fini(void)40 static void ums_fini(void)
41 {
42 int i;
43
44 for (i = 0; i < ums_count; i++)
45 free((void *)ums[i].name);
46 free(ums);
47 ums = NULL;
48 ums_count = 0;
49 }
50
51 #define UMS_NAME_LEN 16
52
ums_init(const char * devtype,const char * devnums_part_str)53 static int ums_init(const char *devtype, const char *devnums_part_str)
54 {
55 char *s, *t, *devnum_part_str, *name;
56 struct blk_desc *block_dev;
57 disk_partition_t info;
58 int partnum;
59 int ret = -1;
60 struct ums *ums_new;
61
62 s = strdup(devnums_part_str);
63 if (!s)
64 return -1;
65
66 t = s;
67 ums_count = 0;
68
69 for (;;) {
70 devnum_part_str = strsep(&t, ",");
71 if (!devnum_part_str)
72 break;
73
74 partnum = blk_get_device_part_str(devtype, devnum_part_str,
75 &block_dev, &info, 1);
76
77 if (partnum < 0)
78 goto cleanup;
79
80 /* Check if the argument is in legacy format. If yes,
81 * expose all partitions by setting the partnum = 0
82 * e.g. ums 0 mmc 0
83 */
84 if (!strchr(devnum_part_str, ':'))
85 partnum = 0;
86
87 /* f_mass_storage.c assumes SECTOR_SIZE sectors */
88 if (block_dev->blksz != SECTOR_SIZE)
89 goto cleanup;
90
91 ums_new = realloc(ums, (ums_count + 1) * sizeof(*ums));
92 if (!ums_new)
93 goto cleanup;
94 ums = ums_new;
95
96 /* if partnum = 0, expose all partitions */
97 if (partnum == 0) {
98 ums[ums_count].start_sector = 0;
99 ums[ums_count].num_sectors = block_dev->lba;
100 } else {
101 ums[ums_count].start_sector = info.start;
102 ums[ums_count].num_sectors = info.size;
103 }
104
105 ums[ums_count].read_sector = ums_read_sector;
106 ums[ums_count].write_sector = ums_write_sector;
107
108 name = malloc(UMS_NAME_LEN);
109 if (!name)
110 goto cleanup;
111 snprintf(name, UMS_NAME_LEN, "UMS disk %d", ums_count);
112 ums[ums_count].name = name;
113 ums[ums_count].block_dev = *block_dev;
114
115 printf("UMS: LUN %d, dev %d, hwpart %d, sector %#x, count %#x\n",
116 ums_count, ums[ums_count].block_dev.devnum,
117 ums[ums_count].block_dev.hwpart,
118 ums[ums_count].start_sector,
119 ums[ums_count].num_sectors);
120
121 ums_count++;
122 }
123
124 if (ums_count)
125 ret = 0;
126
127 cleanup:
128 free(s);
129
130 if (ret < 0)
131 ums_fini();
132
133 return ret;
134 }
135
do_usb_mass_storage(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])136 static int do_usb_mass_storage(cmd_tbl_t *cmdtp, int flag,
137 int argc, char * const argv[])
138 {
139 const char *usb_controller;
140 const char *devtype;
141 const char *devnum;
142 unsigned int controller_index;
143 int rc;
144 int cable_ready_timeout __maybe_unused;
145
146 if (argc < 3)
147 return CMD_RET_USAGE;
148
149 usb_controller = argv[1];
150 if (argc >= 4) {
151 devtype = argv[2];
152 devnum = argv[3];
153 } else {
154 devtype = "mmc";
155 devnum = argv[2];
156 }
157
158 rc = ums_init(devtype, devnum);
159 if (rc < 0)
160 return CMD_RET_FAILURE;
161
162 controller_index = (unsigned int)(simple_strtoul(
163 usb_controller, NULL, 0));
164 if (usb_gadget_initialize(controller_index)) {
165 pr_err("Couldn't init USB controller.\n");
166 rc = CMD_RET_FAILURE;
167 goto cleanup_ums_init;
168 }
169
170 rc = fsg_init(ums, ums_count);
171 if (rc) {
172 pr_err("fsg_init failed\n");
173 rc = CMD_RET_FAILURE;
174 goto cleanup_board;
175 }
176
177 rc = g_dnl_register("usb_dnl_ums");
178 if (rc) {
179 pr_err("g_dnl_register failed\n");
180 rc = CMD_RET_FAILURE;
181 goto cleanup_board;
182 }
183
184 /* Timeout unit: seconds */
185 cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
186
187 if (!g_dnl_board_usb_cable_connected()) {
188 /*
189 * Won't execute if we don't know whether the cable is
190 * connected.
191 */
192 puts("Please connect USB cable.\n");
193
194 while (!g_dnl_board_usb_cable_connected()) {
195 if (ctrlc()) {
196 puts("\rCTRL+C - Operation aborted.\n");
197 rc = CMD_RET_SUCCESS;
198 goto cleanup_register;
199 }
200 if (!cable_ready_timeout) {
201 puts("\rUSB cable not detected.\n" \
202 "Command exit.\n");
203 rc = CMD_RET_SUCCESS;
204 goto cleanup_register;
205 }
206
207 printf("\rAuto exit in: %.2d s.", cable_ready_timeout);
208 mdelay(1000);
209 cable_ready_timeout--;
210 }
211 puts("\r\n");
212 }
213
214 while (1) {
215 usb_gadget_handle_interrupts(controller_index);
216
217 rc = fsg_main_thread(NULL);
218 if (rc) {
219 /* Check I/O error */
220 if (rc == -EIO)
221 printf("\rCheck USB cable connection\n");
222
223 /* Check CTRL+C */
224 if (rc == -EPIPE)
225 printf("\rCTRL+C - Operation aborted\n");
226
227 rc = CMD_RET_SUCCESS;
228 goto cleanup_register;
229 }
230
231 WATCHDOG_RESET();
232 }
233
234 cleanup_register:
235 g_dnl_unregister();
236 cleanup_board:
237 usb_gadget_release(controller_index);
238 cleanup_ums_init:
239 ums_fini();
240
241 return rc;
242 }
243
244 U_BOOT_CMD(ums, 4, 1, do_usb_mass_storage,
245 "Use the UMS [USB Mass Storage]",
246 "<USB_controller> [<devtype>] <dev[:part]> e.g. ums 0 mmc 0\n"
247 " devtype defaults to mmc"
248 );
249