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         if (argc < 4)
  19                 return CMD_RET_USAGE;
  20
  21         char *usb_controller = argv[1];
  22         char *interface = argv[2];
  23         char *devstring = argv[3];
  24
  25         char *s = "dfu";
  26         int ret, i = 0;
  27
  28         ret = dfu_init_env_entities(interface, simple_strtoul(devstring,
  29                                                               NULL, 10));
  30         if (ret)
  31                 return ret;
  32
  33         if (argc > 4 && strcmp(argv[4], "list") == 0) {
  34                 dfu_show_entities();
  35                 goto done;
  36         }
  37
  38         int controller_index = simple_strtoul(usb_controller, NULL, 0);
  39         board_usb_init(controller_index, USB_INIT_DEVICE);
  40
  41         g_dnl_register(s);
  42         while (1) {
  43                 if (dfu_reset())
  44                         /*
  45                          * This extra number of usb_gadget_handle_interrupts()
  46                          * calls is necessary to assure correct transmission
  47                          * completion with dfu-util
  48                          */
  49                         if (++i == 10)
  50                                 goto exit;
  51
  52                 if (ctrlc())
  53                         goto exit;
  54
  55                 usb_gadget_handle_interrupts();
  56         }
  57 exit:
  58         g_dnl_unregister();
  59 done:
  60         dfu_free_entities();
  61
  62         if (dfu_reset())
  63                 run_command("reset", 0);
  64
  65         return CMD_RET_SUCCESS;
  66 }
  67
  68 U_BOOT_CMD(dfu, CONFIG_SYS_MAXARGS, 1, do_dfu,
  69         "Device Firmware Upgrade",
  70         "<USB_controller> <interface> <dev> [list]\n"
  71         "  - device firmware upgrade via <USB_controller>\n"
  72         "    on device <dev>, attached to interface\n"
  73         "    <interface>\n"
  74         "    [list] - list available alt settings\n"
  75 );