1 /*
2 * Copyright 2017 Rockchip Electronics Co., Ltd
3 * Frank Wang <frank.wang@rock-chips.com>
4 *
5 * SPDX-License-Identifier: GPL-2.0+
6 */
7
8 #include <errno.h>
9 #include <common.h>
10 #include <command.h>
11 #include <console.h>
12 #include <g_dnl.h>
13 #include <part.h>
14 #include <usb.h>
15 #include <usb_mass_storage.h>
16 #include <rockusb.h>
17
18 static struct rockusb rkusb;
19 static struct rockusb *g_rkusb;
20
rkusb_read_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,void * buf)21 static int rkusb_read_sector(struct ums *ums_dev,
22 ulong start, lbaint_t blkcnt, void *buf)
23 {
24 struct blk_desc *block_dev = &ums_dev->block_dev;
25 lbaint_t blkstart = start + ums_dev->start_sector;
26 int ret;
27
28 if ((blkstart + blkcnt) > RKUSB_READ_LIMIT_ADDR) {
29 memset(buf, 0xcc, blkcnt * SECTOR_SIZE);
30 return blkcnt;
31 } else {
32 ret = blk_dread(block_dev, blkstart, blkcnt, buf);
33 if (!ret)
34 ret = -EIO;
35 return ret;
36 }
37 }
38
rkusb_write_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt,const void * buf)39 static int rkusb_write_sector(struct ums *ums_dev,
40 ulong start, lbaint_t blkcnt, const void *buf)
41 {
42 struct blk_desc *block_dev = &ums_dev->block_dev;
43 lbaint_t blkstart = start + ums_dev->start_sector;
44 int ret;
45
46 if (block_dev->if_type == IF_TYPE_MTD)
47 block_dev->op_flag |= BLK_MTD_CONT_WRITE;
48
49 ret = blk_dwrite(block_dev, blkstart, blkcnt, buf);
50 if (!ret)
51 ret = -EIO;
52
53 if (block_dev->if_type == IF_TYPE_MTD)
54 block_dev->op_flag &= ~(BLK_MTD_CONT_WRITE);
55 return ret;
56 }
57
rkusb_erase_sector(struct ums * ums_dev,ulong start,lbaint_t blkcnt)58 static int rkusb_erase_sector(struct ums *ums_dev,
59 ulong start, lbaint_t blkcnt)
60 {
61 struct blk_desc *block_dev = &ums_dev->block_dev;
62 lbaint_t blkstart = start + ums_dev->start_sector;
63
64 return blk_derase(block_dev, blkstart, blkcnt);
65 }
66
rkusb_fini(void)67 static void rkusb_fini(void)
68 {
69 int i;
70
71 for (i = 0; i < g_rkusb->ums_cnt; i++)
72 free((void *)g_rkusb->ums[i].name);
73 free(g_rkusb->ums);
74 g_rkusb->ums = NULL;
75 g_rkusb->ums_cnt = 0;
76 g_rkusb = NULL;
77 }
78
79 #define RKUSB_NAME_LEN 16
80
rkusb_init(const char * devtype,const char * devnums_part_str)81 static int rkusb_init(const char *devtype, const char *devnums_part_str)
82 {
83 char *s, *t, *devnum_part_str, *name;
84 struct blk_desc *block_dev;
85 disk_partition_t info;
86 int partnum, cnt;
87 int ret = -1;
88 struct ums *ums_new;
89
90 s = strdup(devnums_part_str);
91 if (!s)
92 return -1;
93
94 t = s;
95 g_rkusb->ums_cnt = 0;
96
97 for (;;) {
98 devnum_part_str = strsep(&t, ",");
99 if (!devnum_part_str)
100 break;
101
102 partnum = blk_get_device_part_str(devtype, devnum_part_str,
103 &block_dev, &info, 1);
104 if (partnum < 0)
105 goto cleanup;
106
107 /* f_mass_storage.c assumes SECTOR_SIZE sectors */
108 if (block_dev->blksz != SECTOR_SIZE)
109 goto cleanup;
110
111 ums_new = realloc(g_rkusb->ums, (g_rkusb->ums_cnt + 1) *
112 sizeof(*g_rkusb->ums));
113 if (!ums_new)
114 goto cleanup;
115 g_rkusb->ums = ums_new;
116 cnt = g_rkusb->ums_cnt;
117
118 /* Expose all partitions for rockusb command */
119 g_rkusb->ums[cnt].start_sector = 0;
120 g_rkusb->ums[cnt].num_sectors = block_dev->lba;
121
122 g_rkusb->ums[cnt].read_sector = rkusb_read_sector;
123 g_rkusb->ums[cnt].write_sector = rkusb_write_sector;
124 g_rkusb->ums[cnt].erase_sector = rkusb_erase_sector;
125
126 name = malloc(RKUSB_NAME_LEN);
127 if (!name)
128 goto cleanup;
129 snprintf(name, RKUSB_NAME_LEN, "rkusb disk %d", cnt);
130 g_rkusb->ums[cnt].name = name;
131 g_rkusb->ums[cnt].block_dev = *block_dev;
132
133 printf("RKUSB: LUN %d, dev %d, hwpart %d, sector %#x, count %#x\n",
134 g_rkusb->ums_cnt,
135 g_rkusb->ums[cnt].block_dev.devnum,
136 g_rkusb->ums[cnt].block_dev.hwpart,
137 g_rkusb->ums[cnt].start_sector,
138 g_rkusb->ums[cnt].num_sectors);
139
140 g_rkusb->ums_cnt++;
141 }
142
143 if (g_rkusb->ums_cnt)
144 ret = 0;
145
146 cleanup:
147 free(s);
148 if (ret < 0)
149 rkusb_fini();
150
151 return ret;
152 }
153
do_rkusb(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])154 static int do_rkusb(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[])
155 {
156 const char *usb_controller;
157 const char *devtype;
158 const char *devnum;
159 unsigned int controller_index;
160 int rc;
161 int cable_ready_timeout __maybe_unused;
162 const char *s;
163
164 if (argc != 4)
165 return CMD_RET_USAGE;
166
167 usb_controller = argv[1];
168 devtype = argv[2];
169 devnum = argv[3];
170
171 if (!strcmp(devtype, "mmc") && !strcmp(devnum, "1")) {
172 pr_err("Forbid to flash mmc 1(sdcard)\n");
173 return CMD_RET_FAILURE;
174 } else if (!strcmp(devtype, "nvme") && !strcmp(devnum, "0")) {
175 /*
176 * Add partnum ":0" to active 'allow_whole_dev' partition
177 * search mechanism on multi storage, where there maybe not
178 * valid partition table.
179 */
180 devnum = "0:0";
181 }
182
183 g_rkusb = &rkusb;
184 rc = rkusb_init(devtype, devnum);
185 if (rc < 0)
186 return CMD_RET_FAILURE;
187
188 if (g_rkusb->ums[0].block_dev.if_type == IF_TYPE_MTD &&
189 g_rkusb->ums[0].block_dev.devnum == BLK_MTD_NAND) {
190 #ifdef CONFIG_CMD_GO
191 pr_err("Enter bootrom rockusb...\n");
192 flushc();
193 run_command("rbrom", 0);
194 #else
195 pr_err("rockusb: count not support loader upgrade!\n");
196 #endif
197 }
198
199 controller_index = (unsigned int)(simple_strtoul(
200 usb_controller, NULL, 0));
201 rc = usb_gadget_initialize(controller_index);
202 if (rc) {
203 pr_err("Couldn't init USB controller.");
204 rc = CMD_RET_FAILURE;
205 goto cleanup_rkusb;
206 }
207
208 rc = fsg_init(g_rkusb->ums, g_rkusb->ums_cnt);
209 if (rc) {
210 pr_err("fsg_init failed");
211 rc = CMD_RET_FAILURE;
212 goto cleanup_board;
213 }
214
215 s = env_get("serial#");
216 if (s) {
217 char *sn = (char *)calloc(strlen(s) + 1, sizeof(char));
218 char *sn_p = sn;
219
220 if (!sn)
221 goto cleanup_board;
222
223 memcpy(sn, s, strlen(s));
224 while (*sn_p) {
225 if (*sn_p == '\\' || *sn_p == '/')
226 *sn_p = '_';
227 sn_p++;
228 }
229
230 g_dnl_set_serialnumber(sn);
231 free(sn);
232 }
233
234 rc = g_dnl_register("rkusb_ums_dnl");
235 if (rc) {
236 pr_err("g_dnl_register failed");
237 rc = CMD_RET_FAILURE;
238 goto cleanup_board;
239 }
240
241 /* Timeout unit: seconds */
242 cable_ready_timeout = UMS_CABLE_READY_TIMEOUT;
243
244 if (!g_dnl_board_usb_cable_connected()) {
245 puts("Please connect USB cable.\n");
246
247 while (!g_dnl_board_usb_cable_connected()) {
248 if (ctrlc()) {
249 puts("\rCTRL+C - Operation aborted.\n");
250 rc = CMD_RET_SUCCESS;
251 goto cleanup_register;
252 }
253 if (!cable_ready_timeout) {
254 puts("\rUSB cable not detected.\nCommand exit.\n");
255 rc = CMD_RET_SUCCESS;
256 goto cleanup_register;
257 }
258
259 printf("\rAuto exit in: %.2d s.", cable_ready_timeout);
260 mdelay(1000);
261 cable_ready_timeout--;
262 }
263 puts("\r\n");
264 }
265
266 while (1) {
267 usb_gadget_handle_interrupts(controller_index);
268
269 rc = fsg_main_thread(NULL);
270 if (rc) {
271 /* Check I/O error */
272 if (rc == -EIO)
273 printf("\rCheck USB cable connection\n");
274
275 /* Check CTRL+C */
276 if (rc == -EPIPE)
277 printf("\rCTRL+C - Operation aborted\n");
278
279 rc = CMD_RET_SUCCESS;
280 goto cleanup_register;
281 }
282 }
283
284 cleanup_register:
285 g_dnl_unregister();
286 cleanup_board:
287 usb_gadget_release(controller_index);
288 cleanup_rkusb:
289 rkusb_fini();
290
291 return rc;
292 }
293
294 U_BOOT_CMD_ALWAYS(rockusb, 4, 1, do_rkusb,
295 "Use the rockusb Protocol",
296 "<USB_controller> <devtype> <dev[:part]> e.g. rockusb 0 mmc 0\n"
297 );
298