common/cmd_dfu.c

   1 /*
   2  * cmd_dfu.c -- dfu command
   3  *
   4  * Copyright (C) 2012 Samsung Electronics
   5  * authors: Andrzej Pietrasiewicz <andrzej.p@samsung.com>
   6  *          Lukasz Majewski <l.majewski@samsung.com>
   7  *
   8  * SPDX-License-Identifier:     GPL-2.0+
   9  */
  10
  11 #include <common.h>
  12 #include <dfu.h>
  13 #include <g_dnl.h>
  14 #include <usb.h>
  15
  16 static int do_dfu(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  17 {
  18         bool dfu_reset = false;
  19
  20         if (argc < 4)
  21                 return CMD_RET_USAGE;
  22
  23         char *usb_controller = argv[1];
  24         char *interface = argv[2];
  25         char *devstring = argv[3];
  26
  27         int ret, i = 0;
  28
  29         ret = dfu_init_env_entities(interface, devstring);
  30         if (ret)
  31                 goto done;
  32
  33         ret = CMD_RET_SUCCESS;
  34         if (argc > 4 && strcmp(argv[4], "list") == 0) {
  35                 dfu_show_entities();
  36                 goto done;
  37         }
  38
  39         int controller_index = simple_strtoul(usb_controller, NULL, 0);
  40         board_usb_init(controller_index, USB_INIT_DEVICE);
  41         g_dnl_clear_detach();
  42         g_dnl_register("usb_dnl_dfu");
  43         while (1) {
  44                 if (g_dnl_detach()) {
  45                         /*
  46                          * Check if USB bus reset is performed after detach,
  47                          * which indicates that -R switch has been passed to
  48                          * dfu-util. In this case reboot the device
  49                          */
  50                         if (dfu_usb_get_reset()) {
  51                                 dfu_reset = true;
  52                                 goto exit;
  53                         }
  54
  55                         /*
  56                          * This extra number of usb_gadget_handle_interrupts()
  57                          * calls is necessary to assure correct transmission
  58                          * completion with dfu-util
  59                          */
  60                         if (++i == 10000)
  61                                 goto exit;
  62                 }
  63
  64                 if (ctrlc())
  65                         goto exit;
  66
  67                 usb_gadget_handle_interrupts();
  68         }
  69 exit:
  70         g_dnl_unregister();
  71         board_usb_cleanup(controller_index, USB_INIT_DEVICE);
  72 done:
  73         dfu_free_entities();
  74
  75         if (dfu_reset)
  76                 run_command("reset", 0);
  77
  78         g_dnl_clear_detach();
  79
  80         return ret;
  81 }
  82
  83 U_BOOT_CMD(dfu, CONFIG_SYS_MAXARGS, 1, do_dfu,
  84         "Device Firmware Upgrade",
  85         "<USB_controller> <interface> <dev> [list]\n"
  86         "  - device firmware upgrade via <USB_controller>\n"
  87         "    on device <dev>, attached to interface\n"
  88         "    <interface>\n"
  89         "    [list] - list available alt settings\n"
  90 );