xref: /openbmc/u-boot/common/update.c (revision 1e52fea3)
1 /*
2  * (C) Copyright 2008 Semihalf
3  *
4  * Written by: Rafal Czubak <rcz@semihalf.com>
5  *             Bartlomiej Sieka <tur@semihalf.com>
6  *
7  * See file CREDITS for list of people who contributed to this
8  * project.
9  *
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License as
12  * published by the Free Software Foundation; either version 2 of
13  * the License, or (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
23  * MA 02111-1307 USA
24  *
25  */
26 
27 #include <common.h>
28 
29 #if !(defined(CONFIG_FIT) && defined(CONFIG_OF_LIBFDT))
30 #error "CONFIG_FIT and CONFIG_OF_LIBFDT are required for auto-update feature"
31 #endif
32 
33 #if defined(CONFIG_SYS_NO_FLASH)
34 #error "CONFIG_SYS_NO_FLASH defined, but FLASH is required for auto-update feature"
35 #endif
36 
37 #include <command.h>
38 #include <flash.h>
39 #include <net.h>
40 #include <malloc.h>
41 
42 /* env variable holding the location of the update file */
43 #define UPDATE_FILE_ENV		"updatefile"
44 
45 /* set configuration defaults if needed */
46 #ifndef CONFIG_UPDATE_LOAD_ADDR
47 #define CONFIG_UPDATE_LOAD_ADDR	0x100000
48 #endif
49 
50 #ifndef CONFIG_UPDATE_TFTP_MSEC_MAX
51 #define CONFIG_UPDATE_TFTP_MSEC_MAX	100
52 #endif
53 
54 #ifndef CONFIG_UPDATE_TFTP_CNT_MAX
55 #define CONFIG_UPDATE_TFTP_CNT_MAX	0
56 #endif
57 
58 extern ulong TftpRRQTimeoutMSecs;
59 extern int TftpRRQTimeoutCountMax;
60 extern flash_info_t flash_info[];
61 extern ulong load_addr;
62 
63 static uchar *saved_prot_info;
64 
65 static int update_load(char *filename, ulong msec_max, int cnt_max, ulong addr)
66 {
67 	int size, rv;
68 	ulong saved_timeout_msecs;
69 	int saved_timeout_count;
70 	char *saved_netretry, *saved_bootfile;
71 
72 	rv = 0;
73 	/* save used globals and env variable */
74 	saved_timeout_msecs = TftpRRQTimeoutMSecs;
75 	saved_timeout_count = TftpRRQTimeoutCountMax;
76 	saved_netretry = strdup(getenv("netretry"));
77 	saved_bootfile = strdup(BootFile);
78 
79 	/* set timeouts for auto-update */
80 	TftpRRQTimeoutMSecs = msec_max;
81 	TftpRRQTimeoutCountMax = cnt_max;
82 
83 	/* we don't want to retry the connection if errors occur */
84 	setenv("netretry", "no");
85 
86 	/* download the update file */
87 	load_addr = addr;
88 	copy_filename(BootFile, filename, sizeof(BootFile));
89 	size = NetLoop(TFTPGET);
90 
91 	if (size < 0)
92 		rv = 1;
93 	else if (size > 0)
94 		flush_cache(addr, size);
95 
96 	/* restore changed globals and env variable */
97 	TftpRRQTimeoutMSecs = saved_timeout_msecs;
98 	TftpRRQTimeoutCountMax = saved_timeout_count;
99 
100 	setenv("netretry", saved_netretry);
101 	if (saved_netretry != NULL)
102 		free(saved_netretry);
103 
104 	if (saved_bootfile != NULL) {
105 		copy_filename(BootFile, saved_bootfile, sizeof(BootFile));
106 		free(saved_bootfile);
107 	}
108 
109 	return rv;
110 }
111 
112 static int update_flash_protect(int prot, ulong addr_first, ulong addr_last)
113 {
114 	uchar *sp_info_ptr;
115 	ulong s;
116 	int i, bank, cnt;
117 	flash_info_t *info;
118 
119 	sp_info_ptr = NULL;
120 
121 	if (prot == 0) {
122 		saved_prot_info =
123 			calloc(CONFIG_SYS_MAX_FLASH_BANKS * CONFIG_SYS_MAX_FLASH_SECT, 1);
124 		if (!saved_prot_info)
125 			return 1;
126 	}
127 
128 	for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
129 		cnt = 0;
130 		info = &flash_info[bank];
131 
132 		/* Nothing to do if the bank doesn't exist */
133 		if (info->sector_count == 0)
134 			return 0;
135 
136 		/* Point to current bank protection information */
137 		sp_info_ptr = saved_prot_info + (bank * CONFIG_SYS_MAX_FLASH_SECT);
138 
139 		/*
140 		 * Adjust addr_first or addr_last if we are on bank boundary.
141 		 * Address space between banks must be continuous for other
142 		 * flash functions (like flash_sect_erase or flash_write) to
143 		 * succeed. Banks must also be numbered in correct order,
144 		 * according to increasing addresses.
145 		 */
146 		if (addr_last > info->start[0] + info->size - 1)
147 			addr_last = info->start[0] + info->size - 1;
148 		if (addr_first < info->start[0])
149 			addr_first = info->start[0];
150 
151 		for (i = 0; i < info->sector_count; i++) {
152 			/* Save current information about protected sectors */
153 			if (prot == 0) {
154 				s = info->start[i];
155 				if ((s >= addr_first) && (s <= addr_last))
156 					sp_info_ptr[i] = info->protect[i];
157 
158 			}
159 
160 			/* Protect/unprotect sectors */
161 			if (sp_info_ptr[i] == 1) {
162 #if defined(CONFIG_SYS_FLASH_PROTECTION)
163 				if (flash_real_protect(info, i, prot))
164 					return 1;
165 #else
166 				info->protect[i] = prot;
167 #endif
168 				cnt++;
169 			}
170 		}
171 
172 		if (cnt) {
173 			printf("%sProtected %d sectors\n",
174 						prot ? "": "Un-", cnt);
175 		}
176 	}
177 
178 	if((prot == 1) && saved_prot_info)
179 		free(saved_prot_info);
180 
181 	return 0;
182 }
183 
184 static int update_flash(ulong addr_source, ulong addr_first, ulong size)
185 {
186 	ulong addr_last = addr_first + size - 1;
187 
188 	/* round last address to the sector boundary */
189 	if (flash_sect_roundb(&addr_last) > 0)
190 		return 1;
191 
192 	if (addr_first >= addr_last) {
193 		printf("Error: end address exceeds addressing space\n");
194 		return 1;
195 	}
196 
197 	/* remove protection on processed sectors */
198 	if (update_flash_protect(0, addr_first, addr_last) > 0) {
199 		printf("Error: could not unprotect flash sectors\n");
200 		return 1;
201 	}
202 
203 	printf("Erasing 0x%08lx - 0x%08lx", addr_first, addr_last);
204 	if (flash_sect_erase(addr_first, addr_last) > 0) {
205 		printf("Error: could not erase flash\n");
206 		return 1;
207 	}
208 
209 	printf("Copying to flash...");
210 	if (flash_write((char *)addr_source, addr_first, size) > 0) {
211 		printf("Error: could not copy to flash\n");
212 		return 1;
213 	}
214 	printf("done\n");
215 
216 	/* enable protection on processed sectors */
217 	if (update_flash_protect(1, addr_first, addr_last) > 0) {
218 		printf("Error: could not protect flash sectors\n");
219 		return 1;
220 	}
221 
222 	return 0;
223 }
224 
225 static int update_fit_getparams(const void *fit, int noffset, ulong *addr,
226 						ulong *fladdr, ulong *size)
227 {
228 	const void *data;
229 
230 	if (fit_image_get_data(fit, noffset, &data, (size_t *)size))
231 		return 1;
232 
233 	if (fit_image_get_load(fit, noffset, (ulong *)fladdr))
234 		return 1;
235 
236 	*addr = (ulong)data;
237 
238 	return 0;
239 }
240 
241 int update_tftp(ulong addr)
242 {
243 	char *filename, *env_addr;
244 	int images_noffset, ndepth, noffset;
245 	ulong update_addr, update_fladdr, update_size;
246 	void *fit;
247 	int ret = 0;
248 
249 	/* use already present image */
250 	if (addr)
251 		goto got_update_file;
252 
253 	printf("Auto-update from TFTP: ");
254 
255 	/* get the file name of the update file */
256 	filename = getenv(UPDATE_FILE_ENV);
257 	if (filename == NULL) {
258 		printf("failed, env. variable '%s' not found\n",
259 							UPDATE_FILE_ENV);
260 		return 1;
261 	}
262 
263 	printf("trying update file '%s'\n", filename);
264 
265 	/* get load address of downloaded update file */
266 	if ((env_addr = getenv("loadaddr")) != NULL)
267 		addr = simple_strtoul(env_addr, NULL, 16);
268 	else
269 		addr = CONFIG_UPDATE_LOAD_ADDR;
270 
271 
272 	if (update_load(filename, CONFIG_UPDATE_TFTP_MSEC_MAX,
273 					CONFIG_UPDATE_TFTP_CNT_MAX, addr)) {
274 		printf("Can't load update file, aborting auto-update\n");
275 		return 1;
276 	}
277 
278 got_update_file:
279 	fit = (void *)addr;
280 
281 	if (!fit_check_format((void *)fit)) {
282 		printf("Bad FIT format of the update file, aborting "
283 							"auto-update\n");
284 		return 1;
285 	}
286 
287 	/* process updates */
288 	images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
289 
290 	ndepth = 0;
291 	noffset = fdt_next_node(fit, images_noffset, &ndepth);
292 	while (noffset >= 0 && ndepth > 0) {
293 		if (ndepth != 1)
294 			goto next_node;
295 
296 		printf("Processing update '%s' :",
297 			fit_get_name(fit, noffset, NULL));
298 
299 		if (!fit_image_check_hashes(fit, noffset)) {
300 			printf("Error: invalid update hash, aborting\n");
301 			ret = 1;
302 			goto next_node;
303 		}
304 
305 		printf("\n");
306 		if (update_fit_getparams(fit, noffset, &update_addr,
307 					&update_fladdr, &update_size)) {
308 			printf("Error: can't get update parameteres, "
309 								"aborting\n");
310 			ret = 1;
311 			goto next_node;
312 		}
313 		if (update_flash(update_addr, update_fladdr, update_size)) {
314 			printf("Error: can't flash update, aborting\n");
315 			ret = 1;
316 			goto next_node;
317 		}
318 next_node:
319 		noffset = fdt_next_node(fit, noffset, &ndepth);
320 	}
321 
322 	return ret;
323 }
324