fastboot.c 3.6 KB

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