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
ums_read_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,void * buf)18 static int ums_read_sector(struct ums *ums_dev,
19 ulong start, lbaint_t blkcnt, void *buf)
20 {
21 struct blk_desc *block_dev = &ums_dev->block_dev;
22 lbaint_t blkstart = start + ums_dev->start_sector;
23
24 return blk_dread(block_dev, blkstart, blkcnt, buf);
25 }
26
ums_write_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,const void * buf)27 static int ums_write_sector(struct ums *ums_dev,
28 ulong start, lbaint_t blkcnt, const void *buf)
29 {
30 struct blk_desc *block_dev = &ums_dev->block_dev;
31 lbaint_t blkstart = start + ums_dev->start_sector;
32
33 return blk_dwrite(block_dev, blkstart, blkcnt, buf);
34 }
35
36 static struct ums *ums;
37 static int ums_count;
38
ums_fini(void)39 static void ums_fini(void)
40 {
41 int i;
42
43 for (i = 0; i < ums_count; i++)
44 free((void *)ums[i].name);
45 free(ums);
46 ums = NULL;
47 ums_count = 0;
48 }
49
50 #define UMS_NAME_LEN 16
51
ums_init(const char * devtype,const char * devnums_part_str)52 static int ums_init(const char *devtype, const char *devnums_part_str)
53 {
54 char *s, *t, *devnum_part_str, *name;
55 struct blk_desc *block_dev;
56 disk_partition_t info;
57 int partnum;
58 int ret = -1;
59 struct ums *ums_new;
60
61 s = strdup(devnums_part_str);
62 if (!s)
63 return -1;
64
65 t = s;
66 ums_count = 0;
67
68 for (;;) {
69 devnum_part_str = strsep(&t, ",");
70 if (!devnum_part_str)
71 break;
72
73 partnum = blk_get_device_part_str(devtype, devnum_part_str,
74 &block_dev, &info, 1);
75
76 if (partnum < 0)
77 goto cleanup;
78
79 /* 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
86 /* f_mass_storage.c assumes SECTOR_SIZE sectors */
87 if (block_dev->blksz != SECTOR_SIZE)
88 goto cleanup;
89
90 ums_new = realloc(ums, (ums_count + 1) * sizeof(*ums));
91 if (!ums_new)
92 goto cleanup;
93 ums = ums_new;
94
95 /* 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
104 ums[ums_count].read_sector = ums_read_sector;
105 ums[ums_count].write_sector = ums_write_sector;
106
107 name = malloc(UMS_NAME_LEN);
108 if (!name)
109 goto cleanup;
110 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",
115 ums_count, ums[ums_count].block_dev.devnum,
116 ums[ums_count].block_dev.hwpart,
117 ums[ums_count].start_sector,
118 ums[ums_count].num_sectors);
119
120 ums_count++;
121 }
122
123 if (ums_count)
124 ret = 0;
125
126 cleanup:
127 free(s);
128
129 if (ret < 0)
130 ums_fini();
131
132 return ret;
133 }
134
do_usb_mass_storage(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])135 static int do_usb_mass_storage(cmd_tbl_t *cmdtp, int flag,
136 int argc, char * const argv[])
137 {
138 const char *usb_controller;
139 const char *devtype;
140 const char *devnum;
141 unsigned int controller_index;
142 int rc;
143 int cable_ready_timeout __maybe_unused;
144
145 if (argc < 3)
146 return CMD_RET_USAGE;
147
148 usb_controller = argv[1];
149 if (argc >= 4) {
150 devtype = argv[2];
151 devnum = argv[3];
152 } else {
153 devtype = "mmc";
154 devnum = argv[2];
155 }
156
157 rc = ums_init(devtype, devnum);
158 if (rc < 0)
159 return CMD_RET_FAILURE;
160
161 controller_index = (unsigned int)(simple_strtoul(
162 usb_controller, NULL, 0));
163 if (usb_gadget_initialize(controller_index)) {
164 pr_err("Couldn't init USB controller.\n");
165 rc = CMD_RET_FAILURE;
166 goto cleanup_ums_init;
167 }
168
169 rc = fsg_init(ums, ums_count);
170 if (rc) {
171 pr_err("fsg_init failed\n");
172 rc = CMD_RET_FAILURE;
173 goto cleanup_board;
174 }
175
176 rc = g_dnl_register("usb_dnl_ums");
177 if (rc) {
178 pr_err("g_dnl_register failed\n");
179 rc = CMD_RET_FAILURE;
180 goto cleanup_board;
181 }
182
183 /* Timeout unit: seconds */
184 cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
185
186 if (!g_dnl_board_usb_cable_connected()) {
187 /*
188 * Won't execute if we don't know whether the cable is
189 * connected.
190 */
191 puts("Please connect USB cable.\n");
192
193 while (!g_dnl_board_usb_cable_connected()) {
194 if (ctrlc()) {
195 puts("\rCTRL+C - Operation aborted.\n");
196 rc = CMD_RET_SUCCESS;
197 goto cleanup_register;
198 }
199 if (!cable_ready_timeout) {
200 puts("\rUSB cable not detected.\n" \
201 "Command exit.\n");
202 rc = CMD_RET_SUCCESS;
203 goto cleanup_register;
204 }
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
213 while (1) {
214 usb_gadget_handle_interrupts(controller_index);
215
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
226 rc = CMD_RET_SUCCESS;
227 goto cleanup_register;
228 }
229 }
230
231 cleanup_register:
232 g_dnl_unregister();
233 cleanup_board:
234 usb_gadget_release(controller_index);
235 cleanup_ums_init:
236 ums_fini();
237
238 return rc;
239 }
240
241 U_BOOT_CMD(ums, 4, 1, do_usb_mass_storage,
242 "Use the UMS [USB Mass Storage]",
243 "<USB_controller> [<devtype>] <dev[:part]> e.g. ums 0 mmc 0\n"
244 " devtype defaults to mmc"
245 );
246