fastboot.c 3.0 KB

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