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