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 <watchdog.h>
15 #include <dfu.h>
16 #include <console.h>
17 #include <g_dnl.h>
18 #include <usb.h>
19 #include <net.h>
20
run_usb_dnl_gadget(int usbctrl_index,char * usb_dnl_gadget)21 int run_usb_dnl_gadget(int usbctrl_index, char *usb_dnl_gadget)
22 {
23 bool dfu_reset = false;
24 int ret, i = 0;
25
26 ret = usb_gadget_initialize(usbctrl_index);
27 if (ret) {
28 pr_err("usb_gadget_initialize failed\n");
29 return CMD_RET_FAILURE;
30 }
31 g_dnl_clear_detach();
32 ret = g_dnl_register(usb_dnl_gadget);
33 if (ret) {
34 pr_err("g_dnl_register failed");
35 return CMD_RET_FAILURE;
36 }
37
38 while (1) {
39 if (g_dnl_detach()) {
40 /*
41 * Check if USB bus reset is performed after detach,
42 * which indicates that -R switch has been passed to
43 * dfu-util. In this case reboot the device
44 */
45 if (dfu_usb_get_reset()) {
46 dfu_reset = true;
47 goto exit;
48 }
49
50 /*
51 * This extra number of usb_gadget_handle_interrupts()
52 * calls is necessary to assure correct transmission
53 * completion with dfu-util
54 */
55 if (++i == 10000)
56 goto exit;
57 }
58
59 if (ctrlc())
60 goto exit;
61
62 if (dfu_get_defer_flush()) {
63 /*
64 * Call to usb_gadget_handle_interrupts() is necessary
65 * to act on ZLP OUT transaction from HOST PC after
66 * transmitting the whole file.
67 *
68 * If this ZLP OUT packet is NAK'ed, the HOST libusb
69 * function fails after timeout (by default it is set to
70 * 5 seconds). In such situation the dfu-util program
71 * exits with error message.
72 */
73 usb_gadget_handle_interrupts(usbctrl_index);
74 ret = dfu_flush(dfu_get_defer_flush(), NULL, 0, 0);
75 dfu_set_defer_flush(NULL);
76 if (ret) {
77 pr_err("Deferred dfu_flush() failed!");
78 goto exit;
79 }
80 }
81
82 WATCHDOG_RESET();
83 usb_gadget_handle_interrupts(usbctrl_index);
84 }
85 exit:
86 g_dnl_unregister();
87 usb_gadget_release(usbctrl_index);
88
89 if (dfu_reset)
90 do_reset(NULL, 0, 0, NULL);
91
92 g_dnl_clear_detach();
93
94 return ret;
95 }
96