common/cmd_dfu.c

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