Blame view

common/dfu.c 2.12 KB
d41ce506b   Eric Lee   Initial Release, ...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
  /*
   * dfu.c -- dfu command
   *
   * Copyright (C) 2015
   * Lukasz Majewski <l.majewski@majess.pl>
   *
   * Copyright (C) 2012 Samsung Electronics
   * authors: Andrzej Pietrasiewicz <andrzej.p@samsung.com>
   *	    Lukasz Majewski <l.majewski@samsung.com>
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #include <common.h>
  #include <watchdog.h>
  #include <dfu.h>
  #include <console.h>
  #include <g_dnl.h>
  #include <usb.h>
  #include <net.h>
  
  int run_usb_dnl_gadget(int usbctrl_index, char *usb_dnl_gadget)
  {
  	bool dfu_reset = false;
  	int ret, i = 0;
  
  	ret = board_usb_init(usbctrl_index, USB_INIT_DEVICE);
  	if (ret) {
  		pr_err("board usb init failed
  ");
  		return CMD_RET_FAILURE;
  	}
  	g_dnl_clear_detach();
  	ret = g_dnl_register(usb_dnl_gadget);
  	if (ret) {
  		pr_err("g_dnl_register failed");
  		return CMD_RET_FAILURE;
  	}
  
  	while (1) {
  		if (g_dnl_detach()) {
  			/*
  			 * Check if USB bus reset is performed after detach,
  			 * which indicates that -R switch has been passed to
  			 * dfu-util. In this case reboot the device
  			 */
  			if (dfu_usb_get_reset()) {
  				dfu_reset = true;
  				goto exit;
  			}
  
  			/*
  			 * This extra number of usb_gadget_handle_interrupts()
  			 * calls is necessary to assure correct transmission
  			 * completion with dfu-util
  			 */
  			if (++i == 10000)
  				goto exit;
  		}
  
  		if (ctrlc())
  			goto exit;
  
  		if (dfu_get_defer_flush()) {
  			/*
  			 * Call to usb_gadget_handle_interrupts() is necessary
  			 * to act on ZLP OUT transaction from HOST PC after
  			 * transmitting the whole file.
  			 *
  			 * If this ZLP OUT packet is NAK'ed, the HOST libusb
  			 * function fails after timeout (by default it is set to
  			 * 5 seconds). In such situation the dfu-util program
  			 * exits with error message.
  			 */
  			usb_gadget_handle_interrupts(usbctrl_index);
  			ret = dfu_flush(dfu_get_defer_flush(), NULL, 0, 0);
  			dfu_set_defer_flush(NULL);
  			if (ret) {
  				pr_err("Deferred dfu_flush() failed!");
  				goto exit;
  			}
  		}
  
  		WATCHDOG_RESET();
  		usb_gadget_handle_interrupts(usbctrl_index);
  	}
  exit:
  	g_dnl_unregister();
  	board_usb_cleanup(usbctrl_index, USB_INIT_DEVICE);
  
  	if (dfu_reset)
  		do_reset(NULL, 0, 0, NULL);
  
  	g_dnl_clear_detach();
  
  	return ret;
  }