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
  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         int ret;
  42
  43         if (argc < 2)
  44                 return CMD_RET_USAGE;
  45
  46         usb_controller = argv[1];
  47         controller_index = simple_strtoul(usb_controller, NULL, 0);
  48
  49         ret = board_usb_init(controller_index, USB_INIT_DEVICE);
  50         if (ret) {
  51                 pr_err("USB init failed: %d\n", ret);
  52                 return CMD_RET_FAILURE;
  53         }
  54
  55         g_dnl_clear_detach();
  56         ret = g_dnl_register("usb_dnl_fastboot");
  57         if (ret)
  58                 return ret;
  59
  60         if (!g_dnl_board_usb_cable_connected()) {
  61                 puts("\rUSB cable not detected.\n" \
  62                      "Command exit.\n");
  63                 ret = CMD_RET_FAILURE;
  64                 goto exit;
  65         }
  66
  67         while (1) {
  68                 if (g_dnl_detach())
  69                         break;
  70                 if (ctrlc())
  71                         break;
  72                 usb_gadget_handle_interrupts(controller_index);
  73         }
  74
  75         ret = CMD_RET_SUCCESS;
  76
  77 exit:
  78         g_dnl_unregister();
  79         g_dnl_clear_detach();
  80         board_usb_cleanup(controller_index, USB_INIT_DEVICE);
  81
  82         return ret;
  83 #else
  84         pr_err("Fastboot USB not enabled\n");
  85         return CMD_RET_FAILURE;
  86 #endif
  87 }
  88
  89 static int do_fastboot(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[])
  90 {
  91         uintptr_t buf_addr = (uintptr_t)NULL;
  92         size_t buf_size = 0;
  93
  94         if (argc < 2)
  95                 return CMD_RET_USAGE;
  96
  97         while (argc > 1 && **(argv + 1) == '-') {
  98                 char *arg = *++argv;
  99
 100                 --argc;
 101                 while (*++arg) {
 102                         switch (*arg) {
 103                         case 'l':
 104                                 if (--argc <= 0)
 105                                         return CMD_RET_USAGE;
 106                                 buf_addr = simple_strtoul(*++argv, NULL, 16);
 107                                 goto NXTARG;
 108
 109                         case 's':
 110                                 if (--argc <= 0)
 111                                         return CMD_RET_USAGE;
 112                                 buf_size = simple_strtoul(*++argv, NULL, 16);
 113                                 goto NXTARG;
 114
 115                         default:
 116                                 return CMD_RET_USAGE;
 117                         }
 118                 }
 119 NXTARG:
 120                 ;
 121         }
 122
 123         fastboot_init((void *)buf_addr, buf_size);
 124
 125         if (!strcmp(argv[1], "udp"))
 126                 return do_fastboot_udp(argc, argv, buf_addr, buf_size);
 127
 128         if (!strcmp(argv[1], "usb")) {
 129                 argv++;
 130                 argc--;
 131         }
 132
 133         return do_fastboot_usb(argc, argv, buf_addr, buf_size);
 134 }
 135
 136 #ifdef CONFIG_SYS_LONGHELP
 137 static char fastboot_help_text[] =
 138         "[-l addr] [-s size] usb <controller> | udp\n"
 139         "\taddr - address of buffer used during data transfers ("
 140         __stringify(CONFIG_FASTBOOT_BUF_ADDR) ")\n"
 141         "\tsize - size of buffer used during data transfers ("
 142         __stringify(CONFIG_FASTBOOT_BUF_SIZE) ")"
 143         ;
 144 #endif
 145
 146 U_BOOT_CMD(
 147         fastboot, CONFIG_SYS_MAXARGS, 1, do_fastboot,
 148         "run as a fastboot usb or udp device", fastboot_help_text
 149 );