cmd/fastboot.c

   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
  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
  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
  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
 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 );