fastboot.c 3.3 KB

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