command: Remove the cmd_tbl_t typedef
[platform/kernel/u-boot.git] / common / dfu.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * dfu.c -- dfu command
4  *
5  * Copyright (C) 2015
6  * Lukasz Majewski <l.majewski@majess.pl>
7  *
8  * Copyright (C) 2012 Samsung Electronics
9  * authors: Andrzej Pietrasiewicz <andrzej.p@samsung.com>
10  *          Lukasz Majewski <l.majewski@samsung.com>
11  */
12
13 #include <common.h>
14 #include <command.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 int run_usb_dnl_gadget(int usbctrl_index, char *usb_dnl_gadget)
23 {
24         bool dfu_reset = false;
25         int ret, i = 0;
26
27         ret = usb_gadget_initialize(usbctrl_index);
28         if (ret) {
29                 pr_err("usb_gadget_initialize failed\n");
30                 return CMD_RET_FAILURE;
31         }
32         g_dnl_clear_detach();
33         ret = g_dnl_register(usb_dnl_gadget);
34         if (ret) {
35                 pr_err("g_dnl_register failed");
36                 return CMD_RET_FAILURE;
37         }
38
39 #ifdef CONFIG_DFU_TIMEOUT
40         unsigned long start_time = get_timer(0);
41 #endif
42
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                 if (dfu_get_defer_flush()) {
68                         /*
69                          * Call to usb_gadget_handle_interrupts() is necessary
70                          * to act on ZLP OUT transaction from HOST PC after
71                          * transmitting the whole file.
72                          *
73                          * If this ZLP OUT packet is NAK'ed, the HOST libusb
74                          * function fails after timeout (by default it is set to
75                          * 5 seconds). In such situation the dfu-util program
76                          * exits with error message.
77                          */
78                         usb_gadget_handle_interrupts(usbctrl_index);
79                         ret = dfu_flush(dfu_get_defer_flush(), NULL, 0, 0);
80                         dfu_set_defer_flush(NULL);
81                         if (ret) {
82                                 pr_err("Deferred dfu_flush() failed!");
83                                 goto exit;
84                         }
85                 }
86
87 #ifdef CONFIG_DFU_TIMEOUT
88                 unsigned long wait_time = dfu_get_timeout();
89
90                 if (wait_time) {
91                         unsigned long current_time = get_timer(start_time);
92
93                         if (current_time > wait_time) {
94                                 debug("Inactivity timeout, abort DFU\n");
95                                 goto exit;
96                         }
97                 }
98 #endif
99
100                 WATCHDOG_RESET();
101                 usb_gadget_handle_interrupts(usbctrl_index);
102         }
103 exit:
104         g_dnl_unregister();
105         usb_gadget_release(usbctrl_index);
106
107         if (dfu_reset)
108                 do_reset(NULL, 0, 0, NULL);
109
110         g_dnl_clear_detach();
111
112         return ret;
113 }