stdio: Correct a build error with driver model
[platform/kernel/u-boot.git] / 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 );