1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright 2008 - 2009 Windriver, <www.windriver.com>
4 * Author: Tom Rix <Tom.Rix@windriver.com>
5 *
6 * (C) Copyright 2014 Linaro, Ltd.
7 * Rob Herring <robh@kernel.org>
8 */
9 #include <common.h>
10 #include <command.h>
11 #include <console.h>
12 #include <g_dnl.h>
13 #include <fastboot.h>
14 #include <net.h>
15 #include <usb.h>
16 #include <watchdog.h>
17 #include <linux/stringify.h>
18
do_fastboot_udp(int argc,char * const argv[],uintptr_t buf_addr,size_t buf_size)19 static int do_fastboot_udp(int argc, char *const argv[],
20 uintptr_t buf_addr, size_t buf_size)
21 {
22 int err;
23
24 if (!IS_ENABLED(CONFIG_UDP_FUNCTION_FASTBOOT)) {
25 pr_err("Fastboot UDP not enabled\n");
26 return CMD_RET_FAILURE;
27 }
28
29 err = net_loop(FASTBOOT_UDP);
30
31 if (err < 0) {
32 printf("fastboot udp error: %d\n", err);
33 return CMD_RET_FAILURE;
34 }
35
36 return CMD_RET_SUCCESS;
37 }
38
do_fastboot_tcp(int argc,char * const argv[],uintptr_t buf_addr,size_t buf_size)39 static int do_fastboot_tcp(int argc, char *const argv[],
40 uintptr_t buf_addr, size_t buf_size)
41 {
42 int err;
43
44 if (!IS_ENABLED(CONFIG_TCP_FUNCTION_FASTBOOT)) {
45 pr_err("Fastboot TCP not enabled\n");
46 return CMD_RET_FAILURE;
47 }
48
49 err = net_loop(FASTBOOT_TCP);
50
51 if (err < 0) {
52 printf("fastboot tcp error: %d\n", err);
53 return CMD_RET_FAILURE;
54 }
55
56 return CMD_RET_SUCCESS;
57 }
58
do_fastboot_usb(int argc,char * const argv[],uintptr_t buf_addr,size_t buf_size)59 static int do_fastboot_usb(int argc, char *const argv[],
60 uintptr_t buf_addr, size_t buf_size)
61 {
62 int controller_index;
63 char *usb_controller;
64 char *endp;
65 int ret;
66
67 if (!IS_ENABLED(CONFIG_USB_FUNCTION_FASTBOOT)) {
68 pr_err("Fastboot USB not enabled\n");
69 return CMD_RET_FAILURE;
70 }
71
72 if (argc < 2)
73 return CMD_RET_USAGE;
74
75 usb_controller = argv[1];
76 controller_index = simple_strtoul(usb_controller, &endp, 0);
77 if (*endp != '\0') {
78 pr_err("Error: Wrong USB controller index format\n");
79 return CMD_RET_FAILURE;
80 }
81
82 ret = usb_gadget_initialize(controller_index);
83 if (ret) {
84 pr_err("USB init failed: %d\n", ret);
85 return CMD_RET_FAILURE;
86 }
87
88 g_dnl_clear_detach();
89 ret = g_dnl_register("usb_dnl_fastboot");
90 if (ret)
91 return ret;
92
93 if (!g_dnl_board_usb_cable_connected()) {
94 puts("\rUSB cable not detected.\n" \
95 "Command exit.\n");
96 ret = CMD_RET_FAILURE;
97 goto exit;
98 }
99
100 while (1) {
101 if (g_dnl_detach())
102 break;
103 if (ctrlc())
104 break;
105 schedule();
106 usb_gadget_handle_interrupts(controller_index);
107 }
108
109 ret = CMD_RET_SUCCESS;
110
111 exit:
112 usb_gadget_release(controller_index);
113 g_dnl_unregister();
114 g_dnl_clear_detach();
115
116 return ret;
117 }
118
do_fastboot(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])119 static int do_fastboot(struct cmd_tbl *cmdtp, int flag, int argc,
120 char *const argv[])
121 {
122 uintptr_t buf_addr = (uintptr_t)NULL;
123 size_t buf_size = 0;
124
125 if (argc < 2)
126 return CMD_RET_USAGE;
127
128 while (argc > 1 && **(argv + 1) == '-') {
129 char *arg = *++argv;
130
131 --argc;
132 while (*++arg) {
133 switch (*arg) {
134 case 'l':
135 if (--argc <= 0)
136 return CMD_RET_USAGE;
137 buf_addr = hextoul(*++argv, NULL);
138 goto NXTARG;
139
140 case 's':
141 if (--argc <= 0)
142 return CMD_RET_USAGE;
143 buf_size = hextoul(*++argv, NULL);
144 goto NXTARG;
145
146 default:
147 return CMD_RET_USAGE;
148 }
149 }
150 NXTARG:
151 ;
152 }
153
154 /* Handle case when USB controller param is just '-' */
155 if (argc == 1) {
156 pr_err("Error: Incorrect USB controller index\n");
157 return CMD_RET_USAGE;
158 }
159
160 fastboot_init((void *)buf_addr, buf_size);
161
162 if (!strcmp(argv[1], "udp"))
163 return do_fastboot_udp(argc, argv, buf_addr, buf_size);
164 if (!strcmp(argv[1], "tcp"))
165 return do_fastboot_tcp(argc, argv, buf_addr, buf_size);
166 if (!strcmp(argv[1], "usb")) {
167 argv++;
168 argc--;
169 }
170
171 return do_fastboot_usb(argc, argv, buf_addr, buf_size);
172 }
173
174 U_BOOT_CMD(
175 fastboot, CONFIG_SYS_MAXARGS, 1, do_fastboot,
176 "run as a fastboot usb or udp device",
177 "[-l addr] [-s size] usb <controller> | udp\n"
178 "\taddr - address of buffer used during data transfers ("
179 __stringify(CONFIG_FASTBOOT_BUF_ADDR) ")\n"
180 "\tsize - size of buffer used during data transfers ("
181 __stringify(CONFIG_FASTBOOT_BUF_SIZE) ")"
182 );
183