793c42212369a85d11047d7315e0f386e77c0e05
[oweals/u-boot.git] / 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 <command.h>
13 #include <malloc.h>
14 #include <dfu.h>
15 #include <asm/errno.h>
16 #include <g_dnl.h>
17
18 static int do_dfu(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
19 {
20         const char *str_env;
21         char *s = "dfu";
22         int ret, i = 0;
23         char *env_bkp;
24
25         if (argc < 3)
26                 return CMD_RET_USAGE;
27
28         str_env = getenv("dfu_alt_info");
29         if (str_env == NULL) {
30                 printf("%s: \"dfu_alt_info\" env variable not defined!\n",
31                        __func__);
32                 return CMD_RET_FAILURE;
33         }
34
35         env_bkp = strdup(str_env);
36         ret = dfu_config_entities(env_bkp, argv[1],
37                             (int)simple_strtoul(argv[2], NULL, 10));
38         if (ret)
39                 return CMD_RET_FAILURE;
40
41         if (argc > 3 && strcmp(argv[3], "list") == 0) {
42                 dfu_show_entities();
43                 goto done;
44         }
45
46 #ifdef CONFIG_TRATS
47         board_usb_init();
48 #endif
49
50         g_dnl_register(s);
51         while (1) {
52                 if (dfu_reset())
53                         /*
54                          * This extra number of usb_gadget_handle_interrupts()
55                          * calls is necessary to assure correct transmission
56                          * completion with dfu-util
57                          */
58                         if (++i == 10)
59                                 goto exit;
60
61                 if (ctrlc())
62                         goto exit;
63
64                 usb_gadget_handle_interrupts();
65         }
66 exit:
67         g_dnl_unregister();
68 done:
69         dfu_free_entities();
70         free(env_bkp);
71
72         if (dfu_reset())
73                 run_command("reset", 0);
74
75         return CMD_RET_SUCCESS;
76 }
77
78 U_BOOT_CMD(dfu, CONFIG_SYS_MAXARGS, 1, do_dfu,
79         "Device Firmware Upgrade",
80         "<interface> <dev> [list]\n"
81         "  - device firmware upgrade on a device <dev>\n"
82         "    attached to interface <interface>\n"
83         "    [list] - list available alt settings"
84 );