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 17 static int do_fastboot_udp(int argc, char *const argv[], 18 uintptr_t buf_addr, size_t buf_size) 19 { 20 #if CONFIG_IS_ENABLED(UDP_FUNCTION_FASTBOOT) 21 int err = net_loop(FASTBOOT); 22 23 if (err < 0) { 24 printf("fastboot udp error: %d\n", err); 25 return CMD_RET_FAILURE; 26 } 27 28 return CMD_RET_SUCCESS; 29 #else 30 pr_err("Fastboot UDP not enabled\n"); 31 return CMD_RET_FAILURE; 32 #endif 33 } 34 35 static int do_fastboot_usb(int argc, char *const argv[], 36 uintptr_t buf_addr, size_t buf_size) 37 { 38 #if CONFIG_IS_ENABLED(USB_FUNCTION_FASTBOOT) 39 int controller_index; 40 char *usb_controller; 41 char *endp; 42 int ret; 43 44 if (argc < 2) 45 return CMD_RET_USAGE; 46 47 usb_controller = argv[1]; 48 controller_index = simple_strtoul(usb_controller, &endp, 0); 49 if (*endp != '\0') { 50 pr_err("Error: Wrong USB controller index format\n"); 51 return CMD_RET_FAILURE; 52 } 53 54 ret = usb_gadget_initialize(controller_index); 55 if (ret) { 56 pr_err("USB init failed: %d\n", ret); 57 return CMD_RET_FAILURE; 58 } 59 60 g_dnl_clear_detach(); 61 ret = g_dnl_register("usb_dnl_fastboot"); 62 if (ret) 63 return ret; 64 65 if (!g_dnl_board_usb_cable_connected()) { 66 puts("\rUSB cable not detected.\n" \ 67 "Command exit.\n"); 68 ret = CMD_RET_FAILURE; 69 goto exit; 70 } 71 72 while (1) { 73 if (g_dnl_detach()) 74 break; 75 if (ctrlc()) 76 break; 77 usb_gadget_handle_interrupts(controller_index); 78 } 79 80 ret = CMD_RET_SUCCESS; 81 82 exit: 83 g_dnl_unregister(); 84 g_dnl_clear_detach(); 85 usb_gadget_release(controller_index); 86 87 return ret; 88 #else 89 pr_err("Fastboot USB not enabled\n"); 90 return CMD_RET_FAILURE; 91 #endif 92 } 93 94 static int do_fastboot(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) 95 { 96 uintptr_t buf_addr = (uintptr_t)NULL; 97 size_t buf_size = 0; 98 99 if (argc < 2) 100 return CMD_RET_USAGE; 101 102 while (argc > 1 && **(argv + 1) == '-') { 103 char *arg = *++argv; 104 105 --argc; 106 while (*++arg) { 107 switch (*arg) { 108 case 'l': 109 if (--argc <= 0) 110 return CMD_RET_USAGE; 111 buf_addr = simple_strtoul(*++argv, NULL, 16); 112 goto NXTARG; 113 114 case 's': 115 if (--argc <= 0) 116 return CMD_RET_USAGE; 117 buf_size = simple_strtoul(*++argv, NULL, 16); 118 goto NXTARG; 119 120 default: 121 return CMD_RET_USAGE; 122 } 123 } 124 NXTARG: 125 ; 126 } 127 128 /* Handle case when USB controller param is just '-' */ 129 if (argc == 1) { 130 pr_err("Error: Incorrect USB controller index\n"); 131 return CMD_RET_USAGE; 132 } 133 134 fastboot_init((void *)buf_addr, buf_size); 135 136 if (!strcmp(argv[1], "udp")) 137 return do_fastboot_udp(argc, argv, buf_addr, buf_size); 138 139 if (!strcmp(argv[1], "usb")) { 140 argv++; 141 argc--; 142 } 143 144 return do_fastboot_usb(argc, argv, buf_addr, buf_size); 145 } 146 147 #ifdef CONFIG_SYS_LONGHELP 148 static char fastboot_help_text[] = 149 "[-l addr] [-s size] usb <controller> | udp\n" 150 "\taddr - address of buffer used during data transfers (" 151 __stringify(CONFIG_FASTBOOT_BUF_ADDR) ")\n" 152 "\tsize - size of buffer used during data transfers (" 153 __stringify(CONFIG_FASTBOOT_BUF_SIZE) ")" 154 ; 155 #endif 156 157 U_BOOT_CMD( 158 fastboot, CONFIG_SYS_MAXARGS, 1, do_fastboot, 159 "run as a fastboot usb or udp device", fastboot_help_text 160 ); 161