1
2
3
4
5
6
7
8
9
10
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
22static int do_dfu(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
23{
24
25 if (argc < 2)
26 return CMD_RET_USAGE;
27
28#ifdef CONFIG_DFU_OVER_USB
29 char *usb_controller = argv[1];
30#endif
31#if defined(CONFIG_DFU_OVER_USB) || defined(CONFIG_DFU_OVER_TFTP)
32 char *interface = NULL;
33 char *devstring = NULL;
34#if defined(CONFIG_DFU_TIMEOUT) || defined(CONFIG_DFU_OVER_TFTP)
35 unsigned long value = 0;
36#endif
37 if (argc >= 4) {
38 interface = argv[2];
39 devstring = argv[3];
40 }
41
42#if defined(CONFIG_DFU_TIMEOUT) || defined(CONFIG_DFU_OVER_TFTP)
43 if (argc == 5 || argc == 3)
44 value = simple_strtoul(argv[argc - 1], NULL, 0);
45#endif
46#endif
47
48 int ret = 0;
49#ifdef CONFIG_DFU_OVER_TFTP
50 if (!strcmp(argv[1], "tftp"))
51 return update_tftp(value, interface, devstring);
52#endif
53#ifdef CONFIG_DFU_OVER_USB
54 ret = dfu_init_env_entities(interface, devstring);
55 if (ret)
56 goto done;
57
58#ifdef CONFIG_DFU_TIMEOUT
59 dfu_set_timeout(value * 1000);
60#endif
61
62 ret = CMD_RET_SUCCESS;
63 if (strcmp(argv[argc - 1], "list") == 0) {
64 dfu_show_entities();
65 goto done;
66 }
67
68 int controller_index = simple_strtoul(usb_controller, NULL, 0);
69 bool retry = false;
70 do {
71 ret = run_usb_dnl_gadget(controller_index, "usb_dnl_dfu");
72
73 if (dfu_reinit_needed) {
74 dfu_free_entities();
75 ret = dfu_init_env_entities(interface, devstring);
76 if (ret)
77 goto done;
78 retry = true;
79 }
80 } while (retry);
81
82done:
83 dfu_free_entities();
84#endif
85 return ret;
86}
87
88U_BOOT_CMD(dfu, CONFIG_SYS_MAXARGS, 1, do_dfu,
89 "Device Firmware Upgrade",
90 ""
91#ifdef CONFIG_DFU_OVER_USB
92#ifdef CONFIG_DFU_TIMEOUT
93 "<USB_controller> [<interface> <dev>] [<timeout>] [list]\n"
94#else
95 "<USB_controller> [<interface> <dev>] [list]\n"
96#endif
97 " - device firmware upgrade via <USB_controller>\n"
98 " on device <dev>, attached to interface\n"
99 " <interface>\n"
100#ifdef CONFIG_DFU_TIMEOUT
101 " [<timeout>] - specify inactivity timeout in seconds\n"
102#endif
103 " [list] - list available alt settings\n"
104#endif
105#ifdef CONFIG_DFU_OVER_TFTP
106#ifdef CONFIG_DFU_OVER_USB
107 "dfu "
108#endif
109 "tftp [<interface> <dev>] [<addr>]\n"
110 " - device firmware upgrade via TFTP\n"
111 " on device <dev>, attached to interface\n"
112 " <interface>\n"
113 " [<addr>] - address where FIT image has been stored\n"
114#endif
115);
116