1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * (C) Copyright 2008 - 2009
4 * Windriver, <www.windriver.com>
5 * Tom Rix <Tom.Rix@windriver.com>
6 *
7 * Copyright 2011 Sebastian Andrzej Siewior <bigeasy@linutronix.de>
8 *
9 * Copyright 2014 Linaro, Ltd.
10 * Rob Herring <robh@kernel.org>
11 */
12
13 #include <common.h>
14 #include <command.h>
15 #include <env.h>
16 #include <fastboot.h>
17 #include <net/fastboot.h>
18
19 /**
20 * fastboot_buf_addr - base address of the fastboot download buffer
21 */
22 void *fastboot_buf_addr;
23
24 /**
25 * fastboot_buf_size - size of the fastboot download buffer
26 */
27 u32 fastboot_buf_size;
28
29 /**
30 * fastboot_progress_callback - callback executed during long operations
31 */
32 void (*fastboot_progress_callback)(const char *msg);
33
34 /**
35 * fastboot_response() - Writes a response of the form "$tag$reason".
36 *
37 * @tag: The first part of the response
38 * @response: Pointer to fastboot response buffer
39 * @format: printf style format string
40 */
fastboot_response(const char * tag,char * response,const char * format,...)41 void fastboot_response(const char *tag, char *response,
42 const char *format, ...)
43 {
44 va_list args;
45
46 strlcpy(response, tag, FASTBOOT_RESPONSE_LEN);
47 if (format) {
48 va_start(args, format);
49 vsnprintf(response + strlen(response),
50 FASTBOOT_RESPONSE_LEN - strlen(response) - 1,
51 format, args);
52 va_end(args);
53 }
54 }
55
56 /**
57 * fastboot_fail() - Write a FAIL response of the form "FAIL$reason".
58 *
59 * @reason: Pointer to returned reason string
60 * @response: Pointer to fastboot response buffer
61 */
fastboot_fail(const char * reason,char * response)62 void fastboot_fail(const char *reason, char *response)
63 {
64 fastboot_response("FAIL", response, "%s", reason);
65 }
66
67 /**
68 * fastboot_okay() - Write an OKAY response of the form "OKAY$reason".
69 *
70 * @reason: Pointer to returned reason string, or NULL to send a bare "OKAY"
71 * @response: Pointer to fastboot response buffer
72 */
fastboot_okay(const char * reason,char * response)73 void fastboot_okay(const char *reason, char *response)
74 {
75 if (reason)
76 fastboot_response("OKAY", response, "%s", reason);
77 else
78 fastboot_response("OKAY", response, NULL);
79 }
80
81 /**
82 * fastboot_set_reboot_flag() - Set flag to indicate reboot-bootloader
83 *
84 * Set flag which indicates that we should reboot into the bootloader
85 * following the reboot that fastboot executes after this function.
86 *
87 * This function should be overridden in your board file with one
88 * which sets whatever flag your board specific Android bootloader flow
89 * requires in order to re-enter the bootloader.
90 */
fastboot_set_reboot_flag(void)91 int __weak fastboot_set_reboot_flag(void)
92 {
93 return -ENOSYS;
94 }
95
96 /**
97 * fastboot_get_progress_callback() - Return progress callback
98 *
99 * Return: Pointer to function called during long operations
100 */
fastboot_get_progress_callback(void)101 void (*fastboot_get_progress_callback(void))(const char *)
102 {
103 return fastboot_progress_callback;
104 }
105
106 /**
107 * fastboot_boot() - Execute fastboot boot command
108 *
109 * If ${fastboot_bootcmd} is set, run that command to execute the boot
110 * process, if that returns, then exit the fastboot server and return
111 * control to the caller.
112 *
113 * Otherwise execute "bootm <fastboot_buf_addr>", if that fails, reset
114 * the board.
115 */
fastboot_boot(void)116 void fastboot_boot(void)
117 {
118 char *s;
119
120 s = env_get("fastboot_bootcmd");
121 if (s) {
122 run_command(s, CMD_FLAG_ENV);
123 } else {
124 static char boot_addr_start[20];
125 static char *const bootm_args[] = {
126 "bootm", boot_addr_start, NULL
127 };
128
129 snprintf(boot_addr_start, sizeof(boot_addr_start) - 1,
130 "0x%p", fastboot_buf_addr);
131 printf("Booting kernel at %s...\n\n\n", boot_addr_start);
132
133 do_bootm(NULL, 0, 2, bootm_args);
134
135 /*
136 * This only happens if image is somehow faulty so we start
137 * over. We deliberately leave this policy to the invocation
138 * of fastbootcmd if that's what's being run
139 */
140 do_reset(NULL, 0, 0, NULL);
141 }
142 }
143
144 /**
145 * fastboot_set_progress_callback() - set progress callback
146 *
147 * @progress: Pointer to progress callback
148 *
149 * Set a callback which is invoked periodically during long running operations
150 * (flash and erase). This can be used (for example) by the UDP transport to
151 * send INFO responses to keep the client alive whilst those commands are
152 * executing.
153 */
fastboot_set_progress_callback(void (* progress)(const char * msg))154 void fastboot_set_progress_callback(void (*progress)(const char *msg))
155 {
156 fastboot_progress_callback = progress;
157 }
158
159 /*
160 * fastboot_init() - initialise new fastboot protocol session
161 *
162 * @buf_addr: Pointer to download buffer, or NULL for default
163 * @buf_size: Size of download buffer, or zero for default
164 */
fastboot_init(void * buf_addr,u32 buf_size)165 void fastboot_init(void *buf_addr, u32 buf_size)
166 {
167 fastboot_buf_addr = buf_addr ? buf_addr :
168 (void *)CONFIG_FASTBOOT_BUF_ADDR;
169 fastboot_buf_size = buf_size ? buf_size : CONFIG_FASTBOOT_BUF_SIZE;
170 fastboot_set_progress_callback(NULL);
171 }
172