]>
Commit | Line | Data |
---|---|---|
6aff3115 | 1 | /* |
0aef7bc7 | 2 | * (C) Copyright 2000-2010 |
6aff3115 WD |
3 | * Wolfgang Denk, DENX Software Engineering, wd@denx.de. |
4 | * | |
56086921 GL |
5 | * (C) Copyright 2008 |
6 | * Guennadi Liakhovetski, DENX Software Engineering, lg@denx.de. | |
7 | * | |
1a459660 | 8 | * SPDX-License-Identifier: GPL-2.0+ |
6aff3115 WD |
9 | */ |
10 | ||
26e355d1 JK |
11 | #define _GNU_SOURCE |
12 | ||
69bf2d2f | 13 | #include <compiler.h> |
6aff3115 | 14 | #include <errno.h> |
30fd4fad | 15 | #include <env_flags.h> |
6aff3115 | 16 | #include <fcntl.h> |
f4742ca0 | 17 | #include <linux/fs.h> |
06134211 | 18 | #include <linux/stringify.h> |
10667e15 | 19 | #include <ctype.h> |
6aff3115 WD |
20 | #include <stdio.h> |
21 | #include <stdlib.h> | |
22 | #include <stddef.h> | |
23 | #include <string.h> | |
24 | #include <sys/types.h> | |
25 | #include <sys/ioctl.h> | |
26 | #include <sys/stat.h> | |
27 | #include <unistd.h> | |
6aff3115 | 28 | |
6de66b35 | 29 | #ifdef MTD_OLD |
1711f3bd | 30 | # include <stdint.h> |
6de66b35 MK |
31 | # include <linux/mtd/mtd.h> |
32 | #else | |
c83d7ca4 | 33 | # define __user /* nothing */ |
6de66b35 MK |
34 | # include <mtd/mtd-user.h> |
35 | #endif | |
36 | ||
9d80b49a | 37 | #include "fw_env_private.h" |
6de66b35 | 38 | #include "fw_env.h" |
6aff3115 | 39 | |
14070e69 AF |
40 | struct env_opts default_opts = { |
41 | #ifdef CONFIG_FILE | |
42 | .config_file = CONFIG_FILE | |
43 | #endif | |
44 | }; | |
45 | ||
a8a752c0 MV |
46 | #define DIV_ROUND_UP(n, d) (((n) + (d) - 1) / (d)) |
47 | ||
56086921 GL |
48 | #define min(x, y) ({ \ |
49 | typeof(x) _min1 = (x); \ | |
50 | typeof(y) _min2 = (y); \ | |
51 | (void) (&_min1 == &_min2); \ | |
52 | _min1 < _min2 ? _min1 : _min2; }) | |
53 | ||
54 | struct envdev_s { | |
229695fe | 55 | const char *devname; /* Device name */ |
f4742ca0 | 56 | long long devoff; /* Device offset */ |
4a6fd34b WD |
57 | ulong env_size; /* environment size */ |
58 | ulong erase_size; /* device erase size */ | |
56086921 GL |
59 | ulong env_sectors; /* number of environment sectors */ |
60 | uint8_t mtd_type; /* type of the MTD device */ | |
61 | }; | |
6aff3115 | 62 | |
56086921 GL |
63 | static struct envdev_s envdevices[2] = |
64 | { | |
65 | { | |
66 | .mtd_type = MTD_ABSENT, | |
67 | }, { | |
68 | .mtd_type = MTD_ABSENT, | |
69 | }, | |
70 | }; | |
71 | static int dev_current; | |
6aff3115 WD |
72 | |
73 | #define DEVNAME(i) envdevices[(i)].devname | |
d0fb80c3 | 74 | #define DEVOFFSET(i) envdevices[(i)].devoff |
6aff3115 WD |
75 | #define ENVSIZE(i) envdevices[(i)].env_size |
76 | #define DEVESIZE(i) envdevices[(i)].erase_size | |
56086921 GL |
77 | #define ENVSECTORS(i) envdevices[(i)].env_sectors |
78 | #define DEVTYPE(i) envdevices[(i)].mtd_type | |
6aff3115 | 79 | |
497f2053 | 80 | #define CUR_ENVSIZE ENVSIZE(dev_current) |
6aff3115 | 81 | |
f71cee4b AF |
82 | static unsigned long usable_envsize; |
83 | #define ENV_SIZE usable_envsize | |
6aff3115 | 84 | |
56086921 GL |
85 | struct env_image_single { |
86 | uint32_t crc; /* CRC32 over data bytes */ | |
87 | char data[]; | |
88 | }; | |
6aff3115 | 89 | |
56086921 GL |
90 | struct env_image_redundant { |
91 | uint32_t crc; /* CRC32 over data bytes */ | |
92 | unsigned char flags; /* active or obsolete */ | |
93 | char data[]; | |
94 | }; | |
95 | ||
96 | enum flag_scheme { | |
97 | FLAG_NONE, | |
98 | FLAG_BOOLEAN, | |
99 | FLAG_INCREMENTAL, | |
100 | }; | |
101 | ||
102 | struct environment { | |
103 | void *image; | |
104 | uint32_t *crc; | |
105 | unsigned char *flags; | |
106 | char *data; | |
107 | enum flag_scheme flag_scheme; | |
108 | }; | |
109 | ||
110 | static struct environment environment = { | |
111 | .flag_scheme = FLAG_NONE, | |
112 | }; | |
6aff3115 | 113 | |
dcdc1f6a | 114 | static int env_aes_cbc_crypt(char *data, const int enc, uint8_t *key); |
a8a752c0 | 115 | |
d0fb80c3 WD |
116 | static int HaveRedundEnv = 0; |
117 | ||
6de66b35 | 118 | static unsigned char active_flag = 1; |
56086921 | 119 | /* obsolete_flag must be 0 to efficiently set it on NOR flash without erasing */ |
6de66b35 | 120 | static unsigned char obsolete_flag = 0; |
d0fb80c3 | 121 | |
ddd8418f JH |
122 | #define DEFAULT_ENV_INSTANCE_STATIC |
123 | #include <env_default.h> | |
6aff3115 | 124 | |
4a6fd34b | 125 | static int flash_io (int mode); |
81974f44 | 126 | static int parse_config(struct env_opts *opts); |
4a6fd34b | 127 | |
d0fb80c3 | 128 | #if defined(CONFIG_FILE) |
4a6fd34b | 129 | static int get_config (char *); |
d0fb80c3 | 130 | #endif |
6aff3115 | 131 | |
938c29ff | 132 | static char *skip_chars(char *s) |
bd7b26f8 | 133 | { |
938c29ff AF |
134 | for (; *s != '\0'; s++) { |
135 | if (isblank(*s)) | |
136 | return s; | |
137 | } | |
138 | return NULL; | |
139 | } | |
bd7b26f8 | 140 | |
938c29ff AF |
141 | static char *skip_blanks(char *s) |
142 | { | |
143 | for (; *s != '\0'; s++) { | |
144 | if (!isblank(*s)) | |
9583efcc | 145 | return s; |
bd7b26f8 | 146 | } |
9583efcc | 147 | return NULL; |
bd7b26f8 SB |
148 | } |
149 | ||
6aff3115 | 150 | /* |
1b7427cd AF |
151 | * s1 is either a simple 'name', or a 'name=value' pair. |
152 | * s2 is a 'name=value' pair. | |
153 | * If the names match, return the value of s2, else NULL. | |
154 | */ | |
155 | static char *envmatch(char *s1, char *s2) | |
156 | { | |
157 | if (s1 == NULL || s2 == NULL) | |
158 | return NULL; | |
159 | ||
160 | while (*s1 == *s2++) | |
161 | if (*s1++ == '=') | |
162 | return s2; | |
163 | if (*s1 == '\0' && *(s2 - 1) == '=') | |
164 | return s2; | |
165 | return NULL; | |
166 | } | |
167 | ||
168 | /** | |
6aff3115 WD |
169 | * Search the environment for a variable. |
170 | * Return the value, if found, or NULL, if not found. | |
171 | */ | |
6de66b35 | 172 | char *fw_getenv (char *name) |
6aff3115 | 173 | { |
6de66b35 | 174 | char *env, *nxt; |
6aff3115 | 175 | |
4a6fd34b | 176 | for (env = environment.data; *env; env = nxt + 1) { |
6de66b35 | 177 | char *val; |
6aff3115 | 178 | |
4a6fd34b | 179 | for (nxt = env; *nxt; ++nxt) { |
6aff3115 WD |
180 | if (nxt >= &environment.data[ENV_SIZE]) { |
181 | fprintf (stderr, "## Error: " | |
182 | "environment not terminated\n"); | |
56086921 | 183 | return NULL; |
6aff3115 WD |
184 | } |
185 | } | |
4a6fd34b | 186 | val = envmatch (name, env); |
6aff3115 WD |
187 | if (!val) |
188 | continue; | |
56086921 | 189 | return val; |
6aff3115 | 190 | } |
56086921 | 191 | return NULL; |
6aff3115 WD |
192 | } |
193 | ||
267541f7 JH |
194 | /* |
195 | * Search the default environment for a variable. | |
196 | * Return the value, if found, or NULL, if not found. | |
197 | */ | |
198 | char *fw_getdefenv(char *name) | |
199 | { | |
200 | char *env, *nxt; | |
201 | ||
202 | for (env = default_environment; *env; env = nxt + 1) { | |
203 | char *val; | |
204 | ||
205 | for (nxt = env; *nxt; ++nxt) { | |
206 | if (nxt >= &default_environment[ENV_SIZE]) { | |
207 | fprintf(stderr, "## Error: " | |
208 | "default environment not terminated\n"); | |
209 | return NULL; | |
210 | } | |
211 | } | |
212 | val = envmatch(name, env); | |
213 | if (!val) | |
214 | continue; | |
215 | return val; | |
216 | } | |
217 | return NULL; | |
218 | } | |
219 | ||
371ee137 | 220 | int parse_aes_key(char *key, uint8_t *bin_key) |
a8a752c0 MV |
221 | { |
222 | char tmp[5] = { '0', 'x', 0, 0, 0 }; | |
223 | unsigned long ul; | |
224 | int i; | |
225 | ||
226 | if (strnlen(key, 64) != 32) { | |
227 | fprintf(stderr, | |
228 | "## Error: '-a' option requires 16-byte AES key\n"); | |
229 | return -1; | |
230 | } | |
231 | ||
232 | for (i = 0; i < 16; i++) { | |
233 | tmp[2] = key[0]; | |
234 | tmp[3] = key[1]; | |
235 | errno = 0; | |
236 | ul = strtoul(tmp, NULL, 16); | |
237 | if (errno) { | |
238 | fprintf(stderr, | |
239 | "## Error: '-a' option requires valid AES key\n"); | |
240 | return -1; | |
241 | } | |
8b27abd8 | 242 | bin_key[i] = ul & 0xff; |
a8a752c0 MV |
243 | key += 2; |
244 | } | |
a8a752c0 MV |
245 | return 0; |
246 | } | |
247 | ||
6aff3115 WD |
248 | /* |
249 | * Print the current definition of one, or more, or all | |
250 | * environment variables | |
251 | */ | |
81974f44 | 252 | int fw_printenv(int argc, char *argv[], int value_only, struct env_opts *opts) |
6aff3115 | 253 | { |
371ee137 | 254 | int i, rc = 0; |
6aff3115 | 255 | |
c5c41c45 AF |
256 | if (value_only && argc != 1) { |
257 | fprintf(stderr, | |
258 | "## Error: `-n' option requires exactly one argument\n"); | |
259 | return -1; | |
260 | } | |
261 | ||
14070e69 AF |
262 | if (!opts) |
263 | opts = &default_opts; | |
264 | ||
81974f44 | 265 | if (fw_env_open(opts)) |
56086921 | 266 | return -1; |
6aff3115 | 267 | |
1ce68697 | 268 | if (argc == 0) { /* Print all env variables */ |
c5c41c45 | 269 | char *env, *nxt; |
4a6fd34b WD |
270 | for (env = environment.data; *env; env = nxt + 1) { |
271 | for (nxt = env; *nxt; ++nxt) { | |
6aff3115 WD |
272 | if (nxt >= &environment.data[ENV_SIZE]) { |
273 | fprintf (stderr, "## Error: " | |
274 | "environment not terminated\n"); | |
56086921 | 275 | return -1; |
6aff3115 WD |
276 | } |
277 | } | |
278 | ||
4a6fd34b | 279 | printf ("%s\n", env); |
6aff3115 | 280 | } |
33f0086c | 281 | fw_env_close(opts); |
56086921 | 282 | return 0; |
6aff3115 WD |
283 | } |
284 | ||
c5c41c45 | 285 | for (i = 0; i < argc; ++i) { /* print a subset of env variables */ |
6de66b35 MK |
286 | char *name = argv[i]; |
287 | char *val = NULL; | |
6aff3115 | 288 | |
c5c41c45 | 289 | val = fw_getenv(name); |
bc11756d | 290 | if (!val) { |
4a6fd34b | 291 | fprintf (stderr, "## Error: \"%s\" not defined\n", name); |
bc11756d | 292 | rc = -1; |
c5c41c45 | 293 | continue; |
bc11756d | 294 | } |
c5c41c45 AF |
295 | |
296 | if (value_only) { | |
297 | puts(val); | |
298 | break; | |
299 | } | |
300 | ||
301 | printf("%s=%s\n", name, val); | |
6aff3115 | 302 | } |
bc11756d | 303 | |
33f0086c SB |
304 | fw_env_close(opts); |
305 | ||
56086921 | 306 | return rc; |
6aff3115 WD |
307 | } |
308 | ||
33f0086c | 309 | int fw_env_flush(struct env_opts *opts) |
6aff3115 | 310 | { |
a8a752c0 | 311 | int ret; |
81974f44 | 312 | |
14070e69 AF |
313 | if (!opts) |
314 | opts = &default_opts; | |
315 | ||
81974f44 | 316 | if (opts->aes_flag) { |
dcdc1f6a | 317 | ret = env_aes_cbc_crypt(environment.data, 1, |
81974f44 | 318 | opts->aes_key); |
a8a752c0 MV |
319 | if (ret) { |
320 | fprintf(stderr, | |
321 | "Error: can't encrypt env for flash\n"); | |
322 | return ret; | |
323 | } | |
324 | } | |
325 | ||
bd7b26f8 SB |
326 | /* |
327 | * Update CRC | |
328 | */ | |
329 | *environment.crc = crc32(0, (uint8_t *) environment.data, ENV_SIZE); | |
6aff3115 | 330 | |
bd7b26f8 SB |
331 | /* write environment back to flash */ |
332 | if (flash_io(O_RDWR)) { | |
333 | fprintf(stderr, | |
334 | "Error: can't write fw_env to flash\n"); | |
335 | return -1; | |
6aff3115 WD |
336 | } |
337 | ||
bd7b26f8 SB |
338 | return 0; |
339 | } | |
6aff3115 | 340 | |
bd7b26f8 SB |
341 | |
342 | /* | |
343 | * Set/Clear a single variable in the environment. | |
344 | * This is called in sequence to update the environment | |
345 | * in RAM without updating the copy in flash after each set | |
346 | */ | |
347 | int fw_env_write(char *name, char *value) | |
348 | { | |
349 | int len; | |
350 | char *env, *nxt; | |
351 | char *oldval = NULL; | |
267541f7 | 352 | int deleting, creating, overwriting; |
6aff3115 WD |
353 | |
354 | /* | |
355 | * search if variable with this name already exists | |
356 | */ | |
4a6fd34b WD |
357 | for (nxt = env = environment.data; *env; env = nxt + 1) { |
358 | for (nxt = env; *nxt; ++nxt) { | |
6aff3115 | 359 | if (nxt >= &environment.data[ENV_SIZE]) { |
bd7b26f8 | 360 | fprintf(stderr, "## Error: " |
6aff3115 | 361 | "environment not terminated\n"); |
56086921 GL |
362 | errno = EINVAL; |
363 | return -1; | |
6aff3115 WD |
364 | } |
365 | } | |
4a6fd34b | 366 | if ((oldval = envmatch (name, env)) != NULL) |
6aff3115 WD |
367 | break; |
368 | } | |
369 | ||
267541f7 JH |
370 | deleting = (oldval && !(value && strlen(value))); |
371 | creating = (!oldval && (value && strlen(value))); | |
372 | overwriting = (oldval && (value && strlen(value))); | |
373 | ||
374 | /* check for permission */ | |
375 | if (deleting) { | |
376 | if (env_flags_validate_varaccess(name, | |
377 | ENV_FLAGS_VARACCESS_PREVENT_DELETE)) { | |
378 | printf("Can't delete \"%s\"\n", name); | |
379 | errno = EROFS; | |
380 | return -1; | |
381 | } | |
382 | } else if (overwriting) { | |
383 | if (env_flags_validate_varaccess(name, | |
384 | ENV_FLAGS_VARACCESS_PREVENT_OVERWR)) { | |
385 | printf("Can't overwrite \"%s\"\n", name); | |
386 | errno = EROFS; | |
387 | return -1; | |
388 | } else if (env_flags_validate_varaccess(name, | |
389 | ENV_FLAGS_VARACCESS_PREVENT_NONDEF_OVERWR)) { | |
390 | const char *defval = fw_getdefenv(name); | |
391 | ||
392 | if (defval == NULL) | |
393 | defval = ""; | |
394 | if (strcmp(oldval, defval) | |
395 | != 0) { | |
396 | printf("Can't overwrite \"%s\"\n", name); | |
397 | errno = EROFS; | |
398 | return -1; | |
399 | } | |
400 | } | |
401 | } else if (creating) { | |
402 | if (env_flags_validate_varaccess(name, | |
403 | ENV_FLAGS_VARACCESS_PREVENT_CREATE)) { | |
404 | printf("Can't create \"%s\"\n", name); | |
405 | errno = EROFS; | |
406 | return -1; | |
407 | } | |
408 | } else | |
409 | /* Nothing to do */ | |
410 | return 0; | |
411 | ||
412 | if (deleting || overwriting) { | |
6aff3115 WD |
413 | if (*++nxt == '\0') { |
414 | *env = '\0'; | |
415 | } else { | |
416 | for (;;) { | |
417 | *env = *nxt++; | |
418 | if ((*env == '\0') && (*nxt == '\0')) | |
419 | break; | |
420 | ++env; | |
421 | } | |
422 | } | |
423 | *++env = '\0'; | |
424 | } | |
425 | ||
426 | /* Delete only ? */ | |
bd7b26f8 SB |
427 | if (!value || !strlen(value)) |
428 | return 0; | |
6aff3115 WD |
429 | |
430 | /* | |
431 | * Append new definition at the end | |
432 | */ | |
4a6fd34b | 433 | for (env = environment.data; *env || *(env + 1); ++env); |
6aff3115 WD |
434 | if (env > environment.data) |
435 | ++env; | |
436 | /* | |
437 | * Overflow when: | |
497f2053 | 438 | * "name" + "=" + "val" +"\0\0" > CUR_ENVSIZE - (env-environment) |
6aff3115 | 439 | */ |
4a6fd34b | 440 | len = strlen (name) + 2; |
6aff3115 | 441 | /* add '=' for first arg, ' ' for all others */ |
bd7b26f8 SB |
442 | len += strlen(value) + 1; |
443 | ||
4a6fd34b | 444 | if (len > (&environment.data[ENV_SIZE] - env)) { |
6aff3115 WD |
445 | fprintf (stderr, |
446 | "Error: environment overflow, \"%s\" deleted\n", | |
447 | name); | |
56086921 | 448 | return -1; |
6aff3115 | 449 | } |
bd7b26f8 | 450 | |
6aff3115 WD |
451 | while ((*env = *name++) != '\0') |
452 | env++; | |
bd7b26f8 SB |
453 | *env = '='; |
454 | while ((*++env = *value++) != '\0') | |
455 | ; | |
456 | ||
457 | /* end is marked with double '\0' */ | |
458 | *++env = '\0'; | |
459 | ||
460 | return 0; | |
461 | } | |
462 | ||
463 | /* | |
464 | * Deletes or sets environment variables. Returns -1 and sets errno error codes: | |
465 | * 0 - OK | |
466 | * EINVAL - need at least 1 argument | |
467 | * EROFS - certain variables ("ethaddr", "serial#") cannot be | |
468 | * modified or deleted | |
469 | * | |
470 | */ | |
81974f44 | 471 | int fw_setenv(int argc, char *argv[], struct env_opts *opts) |
bd7b26f8 | 472 | { |
371ee137 | 473 | int i; |
3779c8e3 | 474 | size_t len; |
167f5258 | 475 | char *name, **valv; |
ddc6a9de | 476 | char *oldval; |
bd7b26f8 | 477 | char *value = NULL; |
167f5258 | 478 | int valc; |
33f0086c | 479 | int ret; |
bd7b26f8 | 480 | |
14070e69 AF |
481 | if (!opts) |
482 | opts = &default_opts; | |
483 | ||
1ce68697 AF |
484 | if (argc < 1) { |
485 | fprintf(stderr, "## Error: variable name missing\n"); | |
a8a752c0 MV |
486 | errno = EINVAL; |
487 | return -1; | |
488 | } | |
489 | ||
81974f44 | 490 | if (fw_env_open(opts)) { |
bd7b26f8 SB |
491 | fprintf(stderr, "Error: environment not initialized\n"); |
492 | return -1; | |
493 | } | |
494 | ||
1ce68697 AF |
495 | name = argv[0]; |
496 | valv = argv + 1; | |
497 | valc = argc - 1; | |
bd7b26f8 | 498 | |
33f0086c SB |
499 | if (env_flags_validate_env_set_params(name, valv, valc) < 0) { |
500 | fw_env_close(opts); | |
fd4e3280 | 501 | return -1; |
33f0086c | 502 | } |
30fd4fad | 503 | |
62a34a04 | 504 | len = 0; |
167f5258 AF |
505 | for (i = 0; i < valc; ++i) { |
506 | char *val = valv[i]; | |
62a34a04 JH |
507 | size_t val_len = strlen(val); |
508 | ||
ce2f5800 JH |
509 | if (value) |
510 | value[len - 1] = ' '; | |
ddc6a9de | 511 | oldval = value; |
62a34a04 | 512 | value = realloc(value, len + val_len + 1); |
bd7b26f8 | 513 | if (!value) { |
62a34a04 | 514 | fprintf(stderr, |
8603b69b | 515 | "Cannot malloc %zu bytes: %s\n", |
62a34a04 | 516 | len, strerror(errno)); |
ddc6a9de | 517 | free(oldval); |
62a34a04 | 518 | return -1; |
bd7b26f8 | 519 | } |
62a34a04 JH |
520 | |
521 | memcpy(value + len, val, val_len); | |
522 | len += val_len; | |
ce2f5800 | 523 | value[len++] = '\0'; |
6aff3115 WD |
524 | } |
525 | ||
bd7b26f8 | 526 | fw_env_write(name, value); |
6aff3115 | 527 | |
62a34a04 | 528 | free(value); |
6aff3115 | 529 | |
33f0086c SB |
530 | ret = fw_env_flush(opts); |
531 | fw_env_close(opts); | |
532 | ||
533 | return ret; | |
bd7b26f8 | 534 | } |
6aff3115 | 535 | |
bd7b26f8 SB |
536 | /* |
537 | * Parse a file and configure the u-boot variables. | |
538 | * The script file has a very simple format, as follows: | |
539 | * | |
540 | * Each line has a couple with name, value: | |
541 | * <white spaces>variable_name<white spaces>variable_value | |
542 | * | |
543 | * Both variable_name and variable_value are interpreted as strings. | |
544 | * Any character after <white spaces> and before ending \r\n is interpreted | |
545 | * as variable's value (no comment allowed on these lines !) | |
546 | * | |
547 | * Comments are allowed if the first character in the line is # | |
548 | * | |
549 | * Returns -1 and sets errno error codes: | |
550 | * 0 - OK | |
551 | * -1 - Error | |
552 | */ | |
81974f44 | 553 | int fw_parse_script(char *fname, struct env_opts *opts) |
bd7b26f8 SB |
554 | { |
555 | FILE *fp; | |
556 | char dump[1024]; /* Maximum line length in the file */ | |
557 | char *name; | |
558 | char *val; | |
559 | int lineno = 0; | |
560 | int len; | |
561 | int ret = 0; | |
562 | ||
14070e69 AF |
563 | if (!opts) |
564 | opts = &default_opts; | |
565 | ||
81974f44 | 566 | if (fw_env_open(opts)) { |
bd7b26f8 | 567 | fprintf(stderr, "Error: environment not initialized\n"); |
56086921 | 568 | return -1; |
6aff3115 WD |
569 | } |
570 | ||
bd7b26f8 SB |
571 | if (strcmp(fname, "-") == 0) |
572 | fp = stdin; | |
573 | else { | |
574 | fp = fopen(fname, "r"); | |
575 | if (fp == NULL) { | |
576 | fprintf(stderr, "I cannot open %s for reading\n", | |
577 | fname); | |
578 | return -1; | |
579 | } | |
580 | } | |
581 | ||
582 | while (fgets(dump, sizeof(dump), fp)) { | |
583 | lineno++; | |
584 | len = strlen(dump); | |
585 | ||
586 | /* | |
587 | * Read a whole line from the file. If the line is too long | |
588 | * or is not terminated, reports an error and exit. | |
589 | */ | |
590 | if (dump[len - 1] != '\n') { | |
591 | fprintf(stderr, | |
592 | "Line %d not corrected terminated or too long\n", | |
593 | lineno); | |
594 | ret = -1; | |
595 | break; | |
596 | } | |
597 | ||
598 | /* Drop ending line feed / carriage return */ | |
69067a34 AF |
599 | dump[--len] = '\0'; |
600 | if (len && dump[len - 1] == '\r') | |
601 | dump[--len] = '\0'; | |
bd7b26f8 SB |
602 | |
603 | /* Skip comment or empty lines */ | |
69067a34 | 604 | if (len == 0 || dump[0] == '#') |
bd7b26f8 SB |
605 | continue; |
606 | ||
607 | /* | |
608 | * Search for variable's name, | |
609 | * remove leading whitespaces | |
610 | */ | |
938c29ff | 611 | name = skip_blanks(dump); |
bd7b26f8 SB |
612 | if (!name) |
613 | continue; | |
614 | ||
615 | /* The first white space is the end of variable name */ | |
938c29ff | 616 | val = skip_chars(name); |
bd7b26f8 SB |
617 | len = strlen(name); |
618 | if (val) { | |
619 | *val++ = '\0'; | |
620 | if ((val - name) < len) | |
938c29ff | 621 | val = skip_blanks(val); |
bd7b26f8 SB |
622 | else |
623 | val = NULL; | |
624 | } | |
625 | ||
626 | #ifdef DEBUG | |
627 | fprintf(stderr, "Setting %s : %s\n", | |
628 | name, val ? val : " removed"); | |
629 | #endif | |
630 | ||
30fd4fad JH |
631 | if (env_flags_validate_type(name, val) < 0) { |
632 | ret = -1; | |
633 | break; | |
634 | } | |
635 | ||
bd7b26f8 SB |
636 | /* |
637 | * If there is an error setting a variable, | |
638 | * try to save the environment and returns an error | |
639 | */ | |
640 | if (fw_env_write(name, val)) { | |
641 | fprintf(stderr, | |
642 | "fw_env_write returns with error : %s\n", | |
643 | strerror(errno)); | |
644 | ret = -1; | |
645 | break; | |
646 | } | |
647 | ||
648 | } | |
649 | ||
650 | /* Close file if not stdin */ | |
651 | if (strcmp(fname, "-") != 0) | |
652 | fclose(fp); | |
653 | ||
33f0086c SB |
654 | ret |= fw_env_flush(opts); |
655 | ||
656 | fw_env_close(opts); | |
bd7b26f8 SB |
657 | |
658 | return ret; | |
6aff3115 WD |
659 | } |
660 | ||
e2c9351d AF |
661 | /** |
662 | * environment_end() - compute offset of first byte right after environemnt | |
663 | * @dev - index of enviroment buffer | |
664 | * Return: | |
665 | * device offset of first byte right after environemnt | |
666 | */ | |
667 | off_t environment_end(int dev) | |
668 | { | |
669 | /* environment is block aligned */ | |
670 | return DEVOFFSET(dev) + ENVSECTORS(dev) * DEVESIZE(dev); | |
671 | } | |
672 | ||
56086921 GL |
673 | /* |
674 | * Test for bad block on NAND, just returns 0 on NOR, on NAND: | |
675 | * 0 - block is good | |
676 | * > 0 - block is bad | |
677 | * < 0 - failed to test | |
678 | */ | |
c6012bbc | 679 | static int flash_bad_block(int fd, uint8_t mtd_type, loff_t blockstart) |
6aff3115 | 680 | { |
56086921 | 681 | if (mtd_type == MTD_NANDFLASH) { |
c6012bbc | 682 | int badblock = ioctl(fd, MEMGETBADBLOCK, &blockstart); |
6aff3115 | 683 | |
56086921 GL |
684 | if (badblock < 0) { |
685 | perror ("Cannot read bad block mark"); | |
686 | return badblock; | |
687 | } | |
688 | ||
689 | if (badblock) { | |
690 | #ifdef DEBUG | |
81c878dd | 691 | fprintf (stderr, "Bad block at 0x%llx, skipping\n", |
c6012bbc | 692 | (unsigned long long)blockstart); |
56086921 GL |
693 | #endif |
694 | return badblock; | |
695 | } | |
6aff3115 WD |
696 | } |
697 | ||
56086921 GL |
698 | return 0; |
699 | } | |
700 | ||
701 | /* | |
702 | * Read data from flash at an offset into a provided buffer. On NAND it skips | |
703 | * bad blocks but makes sure it stays within ENVSECTORS (dev) starting from | |
704 | * the DEVOFFSET (dev) block. On NOR the loop is only run once. | |
705 | */ | |
706 | static int flash_read_buf (int dev, int fd, void *buf, size_t count, | |
ff95e579 | 707 | off_t offset) |
56086921 GL |
708 | { |
709 | size_t blocklen; /* erase / write length - one block on NAND, | |
710 | 0 on NOR */ | |
711 | size_t processed = 0; /* progress counter */ | |
712 | size_t readlen = count; /* current read length */ | |
56086921 GL |
713 | off_t block_seek; /* offset inside the current block to the start |
714 | of the data */ | |
715 | loff_t blockstart; /* running start of the current block - | |
716 | MEMGETBADBLOCK needs 64 bits */ | |
717 | int rc; | |
718 | ||
9eeaa8e6 | 719 | blockstart = (offset / DEVESIZE (dev)) * DEVESIZE (dev); |
56086921 GL |
720 | |
721 | /* Offset inside a block */ | |
722 | block_seek = offset - blockstart; | |
723 | ||
ff95e579 | 724 | if (DEVTYPE(dev) == MTD_NANDFLASH) { |
56086921 GL |
725 | /* |
726 | * NAND: calculate which blocks we are reading. We have | |
727 | * to read one block at a time to skip bad blocks. | |
728 | */ | |
729 | blocklen = DEVESIZE (dev); | |
730 | ||
56086921 GL |
731 | /* Limit to one block for the first read */ |
732 | if (readlen > blocklen - block_seek) | |
733 | readlen = blocklen - block_seek; | |
734 | } else { | |
735 | blocklen = 0; | |
d0fb80c3 | 736 | } |
6aff3115 | 737 | |
56086921 GL |
738 | /* This only runs once on NOR flash */ |
739 | while (processed < count) { | |
ff95e579 | 740 | rc = flash_bad_block(fd, DEVTYPE(dev), blockstart); |
56086921 GL |
741 | if (rc < 0) /* block test failed */ |
742 | return -1; | |
6aff3115 | 743 | |
e2c9351d | 744 | if (blockstart + block_seek + readlen > environment_end(dev)) { |
56086921 GL |
745 | /* End of range is reached */ |
746 | fprintf (stderr, | |
747 | "Too few good blocks within range\n"); | |
748 | return -1; | |
d0fb80c3 WD |
749 | } |
750 | ||
56086921 GL |
751 | if (rc) { /* block is bad */ |
752 | blockstart += blocklen; | |
753 | continue; | |
6aff3115 WD |
754 | } |
755 | ||
56086921 GL |
756 | /* |
757 | * If a block is bad, we retry in the next block at the same | |
758 | * offset - see common/env_nand.c::writeenv() | |
759 | */ | |
760 | lseek (fd, blockstart + block_seek, SEEK_SET); | |
6aff3115 | 761 | |
56086921 GL |
762 | rc = read (fd, buf + processed, readlen); |
763 | if (rc != readlen) { | |
764 | fprintf (stderr, "Read error on %s: %s\n", | |
765 | DEVNAME (dev), strerror (errno)); | |
766 | return -1; | |
6aff3115 | 767 | } |
56086921 | 768 | #ifdef DEBUG |
74620e1c | 769 | fprintf(stderr, "Read 0x%x bytes at 0x%llx on %s\n", |
81c878dd MN |
770 | rc, (unsigned long long) blockstart + block_seek, |
771 | DEVNAME(dev)); | |
56086921 GL |
772 | #endif |
773 | processed += readlen; | |
774 | readlen = min (blocklen, count - processed); | |
775 | block_seek = 0; | |
776 | blockstart += blocklen; | |
777 | } | |
778 | ||
779 | return processed; | |
780 | } | |
781 | ||
782 | /* | |
24307d63 AF |
783 | * Write count bytes from begin of environment, but stay within |
784 | * ENVSECTORS(dev) sectors of | |
9eeaa8e6 RB |
785 | * DEVOFFSET (dev). Similar to the read case above, on NOR and dataflash we |
786 | * erase and write the whole data at once. | |
56086921 | 787 | */ |
24307d63 | 788 | static int flash_write_buf(int dev, int fd, void *buf, size_t count) |
56086921 GL |
789 | { |
790 | void *data; | |
791 | struct erase_info_user erase; | |
792 | size_t blocklen; /* length of NAND block / NOR erase sector */ | |
793 | size_t erase_len; /* whole area that can be erased - may include | |
794 | bad blocks */ | |
795 | size_t erasesize; /* erase / write length - one block on NAND, | |
796 | whole area on NOR */ | |
797 | size_t processed = 0; /* progress counter */ | |
9eeaa8e6 | 798 | size_t write_total; /* total size to actually write - excluding |
56086921 GL |
799 | bad blocks */ |
800 | off_t erase_offset; /* offset to the first erase block (aligned) | |
801 | below offset */ | |
802 | off_t block_seek; /* offset inside the erase block to the start | |
803 | of the data */ | |
56086921 GL |
804 | loff_t blockstart; /* running start of the current block - |
805 | MEMGETBADBLOCK needs 64 bits */ | |
806 | int rc; | |
807 | ||
e387efbd OM |
808 | /* |
809 | * For mtd devices only offset and size of the environment do matter | |
810 | */ | |
ff95e579 | 811 | if (DEVTYPE(dev) == MTD_ABSENT) { |
e387efbd | 812 | blocklen = count; |
e387efbd | 813 | erase_len = blocklen; |
24307d63 | 814 | blockstart = DEVOFFSET(dev); |
e387efbd OM |
815 | block_seek = 0; |
816 | write_total = blocklen; | |
817 | } else { | |
818 | blocklen = DEVESIZE(dev); | |
56086921 | 819 | |
24307d63 | 820 | erase_offset = DEVOFFSET(dev); |
6aff3115 | 821 | |
e387efbd | 822 | /* Maximum area we may use */ |
e2c9351d | 823 | erase_len = environment_end(dev) - erase_offset; |
56086921 | 824 | |
e387efbd | 825 | blockstart = erase_offset; |
24307d63 | 826 | |
e387efbd | 827 | /* Offset inside a block */ |
24307d63 | 828 | block_seek = DEVOFFSET(dev) - erase_offset; |
56086921 | 829 | |
e387efbd OM |
830 | /* |
831 | * Data size we actually write: from the start of the block | |
832 | * to the start of the data, then count bytes of data, and | |
833 | * to the end of the block | |
834 | */ | |
835 | write_total = ((block_seek + count + blocklen - 1) / | |
836 | blocklen) * blocklen; | |
837 | } | |
56086921 GL |
838 | |
839 | /* | |
840 | * Support data anywhere within erase sectors: read out the complete | |
841 | * area to be erased, replace the environment image, write the whole | |
842 | * block back again. | |
843 | */ | |
844 | if (write_total > count) { | |
845 | data = malloc (erase_len); | |
846 | if (!data) { | |
d0fb80c3 | 847 | fprintf (stderr, |
8603b69b | 848 | "Cannot malloc %zu bytes: %s\n", |
56086921 GL |
849 | erase_len, strerror (errno)); |
850 | return -1; | |
d0fb80c3 | 851 | } |
56086921 | 852 | |
ff95e579 | 853 | rc = flash_read_buf(dev, fd, data, write_total, erase_offset); |
56086921 GL |
854 | if (write_total != rc) |
855 | return -1; | |
856 | ||
74620e1c JH |
857 | #ifdef DEBUG |
858 | fprintf(stderr, "Preserving data "); | |
859 | if (block_seek != 0) | |
860 | fprintf(stderr, "0x%x - 0x%lx", 0, block_seek - 1); | |
861 | if (block_seek + count != write_total) { | |
862 | if (block_seek != 0) | |
863 | fprintf(stderr, " and "); | |
81c878dd MN |
864 | fprintf(stderr, "0x%lx - 0x%lx", |
865 | (unsigned long) block_seek + count, | |
866 | (unsigned long) write_total - 1); | |
74620e1c JH |
867 | } |
868 | fprintf(stderr, "\n"); | |
869 | #endif | |
56086921 GL |
870 | /* Overwrite the old environment */ |
871 | memcpy (data + block_seek, buf, count); | |
872 | } else { | |
873 | /* | |
874 | * We get here, iff offset is block-aligned and count is a | |
875 | * multiple of blocklen - see write_total calculation above | |
876 | */ | |
877 | data = buf; | |
878 | } | |
879 | ||
ff95e579 | 880 | if (DEVTYPE(dev) == MTD_NANDFLASH) { |
56086921 GL |
881 | /* |
882 | * NAND: calculate which blocks we are writing. We have | |
883 | * to write one block at a time to skip bad blocks. | |
884 | */ | |
885 | erasesize = blocklen; | |
886 | } else { | |
887 | erasesize = erase_len; | |
888 | } | |
889 | ||
890 | erase.length = erasesize; | |
891 | ||
9eeaa8e6 | 892 | /* This only runs once on NOR flash and SPI-dataflash */ |
56086921 | 893 | while (processed < write_total) { |
ff95e579 | 894 | rc = flash_bad_block(fd, DEVTYPE(dev), blockstart); |
56086921 GL |
895 | if (rc < 0) /* block test failed */ |
896 | return rc; | |
897 | ||
e2c9351d | 898 | if (blockstart + erasesize > environment_end(dev)) { |
56086921 GL |
899 | fprintf (stderr, "End of range reached, aborting\n"); |
900 | return -1; | |
6aff3115 | 901 | } |
56086921 GL |
902 | |
903 | if (rc) { /* block is bad */ | |
904 | blockstart += blocklen; | |
905 | continue; | |
906 | } | |
907 | ||
ff95e579 | 908 | if (DEVTYPE(dev) != MTD_ABSENT) { |
e387efbd OM |
909 | erase.start = blockstart; |
910 | ioctl(fd, MEMUNLOCK, &erase); | |
911 | /* These do not need an explicit erase cycle */ | |
ff95e579 | 912 | if (DEVTYPE(dev) != MTD_DATAFLASH) |
e387efbd OM |
913 | if (ioctl(fd, MEMERASE, &erase) != 0) { |
914 | fprintf(stderr, | |
915 | "MTD erase error on %s: %s\n", | |
916 | DEVNAME(dev), strerror(errno)); | |
917 | return -1; | |
918 | } | |
919 | } | |
56086921 GL |
920 | |
921 | if (lseek (fd, blockstart, SEEK_SET) == -1) { | |
6aff3115 | 922 | fprintf (stderr, |
56086921 GL |
923 | "Seek error on %s: %s\n", |
924 | DEVNAME (dev), strerror (errno)); | |
925 | return -1; | |
6aff3115 | 926 | } |
56086921 GL |
927 | |
928 | #ifdef DEBUG | |
81c878dd MN |
929 | fprintf(stderr, "Write 0x%llx bytes at 0x%llx\n", |
930 | (unsigned long long) erasesize, | |
931 | (unsigned long long) blockstart); | |
56086921 GL |
932 | #endif |
933 | if (write (fd, data + processed, erasesize) != erasesize) { | |
934 | fprintf (stderr, "Write error on %s: %s\n", | |
935 | DEVNAME (dev), strerror (errno)); | |
936 | return -1; | |
6aff3115 | 937 | } |
56086921 | 938 | |
ff95e579 | 939 | if (DEVTYPE(dev) != MTD_ABSENT) |
e387efbd | 940 | ioctl(fd, MEMLOCK, &erase); |
56086921 | 941 | |
4b774ff1 | 942 | processed += erasesize; |
56086921 | 943 | block_seek = 0; |
4b774ff1 | 944 | blockstart += erasesize; |
56086921 GL |
945 | } |
946 | ||
947 | if (write_total > count) | |
948 | free (data); | |
949 | ||
950 | return processed; | |
951 | } | |
952 | ||
953 | /* | |
954 | * Set obsolete flag at offset - NOR flash only | |
955 | */ | |
956 | static int flash_flag_obsolete (int dev, int fd, off_t offset) | |
957 | { | |
958 | int rc; | |
0aef7bc7 | 959 | struct erase_info_user erase; |
56086921 | 960 | |
0aef7bc7 DZ |
961 | erase.start = DEVOFFSET (dev); |
962 | erase.length = DEVESIZE (dev); | |
56086921 GL |
963 | /* This relies on the fact, that obsolete_flag == 0 */ |
964 | rc = lseek (fd, offset, SEEK_SET); | |
965 | if (rc < 0) { | |
966 | fprintf (stderr, "Cannot seek to set the flag on %s \n", | |
967 | DEVNAME (dev)); | |
968 | return rc; | |
969 | } | |
0aef7bc7 | 970 | ioctl (fd, MEMUNLOCK, &erase); |
56086921 | 971 | rc = write (fd, &obsolete_flag, sizeof (obsolete_flag)); |
0aef7bc7 | 972 | ioctl (fd, MEMLOCK, &erase); |
56086921 GL |
973 | if (rc < 0) |
974 | perror ("Could not set obsolete flag"); | |
975 | ||
976 | return rc; | |
977 | } | |
978 | ||
a8a752c0 | 979 | /* Encrypt or decrypt the environment before writing or reading it. */ |
dcdc1f6a | 980 | static int env_aes_cbc_crypt(char *payload, const int enc, uint8_t *key) |
a8a752c0 MV |
981 | { |
982 | uint8_t *data = (uint8_t *)payload; | |
f71cee4b | 983 | const int len = usable_envsize; |
a8a752c0 MV |
984 | uint8_t key_exp[AES_EXPAND_KEY_LENGTH]; |
985 | uint32_t aes_blocks; | |
986 | ||
987 | /* First we expand the key. */ | |
dcdc1f6a | 988 | aes_expand_key(key, key_exp); |
a8a752c0 MV |
989 | |
990 | /* Calculate the number of AES blocks to encrypt. */ | |
991 | aes_blocks = DIV_ROUND_UP(len, AES_KEY_LENGTH); | |
992 | ||
993 | if (enc) | |
994 | aes_cbc_encrypt_blocks(key_exp, data, data, aes_blocks); | |
995 | else | |
996 | aes_cbc_decrypt_blocks(key_exp, data, data, aes_blocks); | |
997 | ||
998 | return 0; | |
999 | } | |
1000 | ||
56086921 GL |
1001 | static int flash_write (int fd_current, int fd_target, int dev_target) |
1002 | { | |
1003 | int rc; | |
1004 | ||
1005 | switch (environment.flag_scheme) { | |
1006 | case FLAG_NONE: | |
1007 | break; | |
1008 | case FLAG_INCREMENTAL: | |
1009 | (*environment.flags)++; | |
1010 | break; | |
1011 | case FLAG_BOOLEAN: | |
1012 | *environment.flags = active_flag; | |
1013 | break; | |
1014 | default: | |
1015 | fprintf (stderr, "Unimplemented flash scheme %u \n", | |
1016 | environment.flag_scheme); | |
1017 | return -1; | |
1018 | } | |
1019 | ||
1020 | #ifdef DEBUG | |
f4742ca0 | 1021 | fprintf(stderr, "Writing new environment at 0x%llx on %s\n", |
56086921 GL |
1022 | DEVOFFSET (dev_target), DEVNAME (dev_target)); |
1023 | #endif | |
a8a752c0 | 1024 | |
497f2053 | 1025 | rc = flash_write_buf(dev_target, fd_target, environment.image, |
24307d63 | 1026 | CUR_ENVSIZE); |
56086921 GL |
1027 | if (rc < 0) |
1028 | return rc; | |
1029 | ||
1030 | if (environment.flag_scheme == FLAG_BOOLEAN) { | |
1031 | /* Have to set obsolete flag */ | |
1032 | off_t offset = DEVOFFSET (dev_current) + | |
1033 | offsetof (struct env_image_redundant, flags); | |
1034 | #ifdef DEBUG | |
74620e1c | 1035 | fprintf(stderr, |
f4742ca0 | 1036 | "Setting obsolete flag in environment at 0x%llx on %s\n", |
56086921 GL |
1037 | DEVOFFSET (dev_current), DEVNAME (dev_current)); |
1038 | #endif | |
1039 | flash_flag_obsolete (dev_current, fd_current, offset); | |
1040 | } | |
1041 | ||
1042 | return 0; | |
1043 | } | |
1044 | ||
1045 | static int flash_read (int fd) | |
1046 | { | |
56086921 GL |
1047 | int rc; |
1048 | ||
497f2053 | 1049 | rc = flash_read_buf(dev_current, fd, environment.image, CUR_ENVSIZE, |
ff95e579 | 1050 | DEVOFFSET(dev_current)); |
a8a752c0 MV |
1051 | if (rc != CUR_ENVSIZE) |
1052 | return -1; | |
56086921 | 1053 | |
a8a752c0 | 1054 | return 0; |
56086921 GL |
1055 | } |
1056 | ||
1057 | static int flash_io (int mode) | |
1058 | { | |
1059 | int fd_current, fd_target, rc, dev_target; | |
1060 | ||
1061 | /* dev_current: fd_current, erase_current */ | |
1062 | fd_current = open (DEVNAME (dev_current), mode); | |
1063 | if (fd_current < 0) { | |
1064 | fprintf (stderr, | |
1065 | "Can't open %s: %s\n", | |
1066 | DEVNAME (dev_current), strerror (errno)); | |
1067 | return -1; | |
1068 | } | |
1069 | ||
1070 | if (mode == O_RDWR) { | |
d0fb80c3 | 1071 | if (HaveRedundEnv) { |
56086921 GL |
1072 | /* switch to next partition for writing */ |
1073 | dev_target = !dev_current; | |
1074 | /* dev_target: fd_target, erase_target */ | |
1075 | fd_target = open (DEVNAME (dev_target), mode); | |
1076 | if (fd_target < 0) { | |
d0fb80c3 | 1077 | fprintf (stderr, |
56086921 GL |
1078 | "Can't open %s: %s\n", |
1079 | DEVNAME (dev_target), | |
1080 | strerror (errno)); | |
1081 | rc = -1; | |
1082 | goto exit; | |
d0fb80c3 | 1083 | } |
56086921 GL |
1084 | } else { |
1085 | dev_target = dev_current; | |
1086 | fd_target = fd_current; | |
6aff3115 | 1087 | } |
56086921 GL |
1088 | |
1089 | rc = flash_write (fd_current, fd_target, dev_target); | |
1090 | ||
7e99e14d MH |
1091 | if (fsync (fd_current)) { |
1092 | fprintf (stderr, | |
1093 | "fsync failed on %s: %s\n", | |
1094 | DEVNAME (dev_current), strerror (errno)); | |
1095 | } | |
1096 | ||
d0fb80c3 | 1097 | if (HaveRedundEnv) { |
7e99e14d MH |
1098 | if (fsync (fd_target)) { |
1099 | fprintf (stderr, | |
1100 | "fsync failed on %s: %s\n", | |
1101 | DEVNAME (dev_current), strerror (errno)); | |
1102 | } | |
1103 | ||
56086921 | 1104 | if (close (fd_target)) { |
d0fb80c3 | 1105 | fprintf (stderr, |
4a6fd34b | 1106 | "I/O error on %s: %s\n", |
56086921 | 1107 | DEVNAME (dev_target), |
4a6fd34b | 1108 | strerror (errno)); |
56086921 | 1109 | rc = -1; |
d0fb80c3 | 1110 | } |
6aff3115 | 1111 | } |
6aff3115 | 1112 | } else { |
56086921 | 1113 | rc = flash_read (fd_current); |
6aff3115 WD |
1114 | } |
1115 | ||
56086921 GL |
1116 | exit: |
1117 | if (close (fd_current)) { | |
6aff3115 | 1118 | fprintf (stderr, |
56086921 GL |
1119 | "I/O error on %s: %s\n", |
1120 | DEVNAME (dev_current), strerror (errno)); | |
1121 | return -1; | |
6aff3115 WD |
1122 | } |
1123 | ||
56086921 | 1124 | return rc; |
6aff3115 WD |
1125 | } |
1126 | ||
6aff3115 WD |
1127 | /* |
1128 | * Prevent confusion if running from erased flash memory | |
1129 | */ | |
81974f44 | 1130 | int fw_env_open(struct env_opts *opts) |
6aff3115 | 1131 | { |
56086921 | 1132 | int crc0, crc0_ok; |
735eb0f0 | 1133 | unsigned char flag0; |
33f0086c | 1134 | void *addr0 = NULL; |
56086921 | 1135 | |
6aff3115 | 1136 | int crc1, crc1_ok; |
735eb0f0 | 1137 | unsigned char flag1; |
33f0086c | 1138 | void *addr1 = NULL; |
d0fb80c3 | 1139 | |
a8a752c0 MV |
1140 | int ret; |
1141 | ||
56086921 GL |
1142 | struct env_image_single *single; |
1143 | struct env_image_redundant *redundant; | |
6aff3115 | 1144 | |
14070e69 AF |
1145 | if (!opts) |
1146 | opts = &default_opts; | |
1147 | ||
81974f44 | 1148 | if (parse_config(opts)) /* should fill envdevices */ |
33f0086c | 1149 | return -EINVAL; |
4a6fd34b | 1150 | |
497f2053 | 1151 | addr0 = calloc(1, CUR_ENVSIZE); |
56086921 | 1152 | if (addr0 == NULL) { |
497f2053 | 1153 | fprintf(stderr, |
4a6fd34b | 1154 | "Not enough memory for environment (%ld bytes)\n", |
497f2053 | 1155 | CUR_ENVSIZE); |
33f0086c SB |
1156 | ret = -ENOMEM; |
1157 | goto open_cleanup; | |
d0fb80c3 | 1158 | } |
4a6fd34b | 1159 | |
d0fb80c3 | 1160 | /* read environment from FLASH to local buffer */ |
56086921 GL |
1161 | environment.image = addr0; |
1162 | ||
1163 | if (HaveRedundEnv) { | |
1164 | redundant = addr0; | |
1165 | environment.crc = &redundant->crc; | |
1166 | environment.flags = &redundant->flags; | |
1167 | environment.data = redundant->data; | |
1168 | } else { | |
1169 | single = addr0; | |
1170 | environment.crc = &single->crc; | |
1171 | environment.flags = NULL; | |
1172 | environment.data = single->data; | |
d0fb80c3 | 1173 | } |
4a6fd34b | 1174 | |
56086921 | 1175 | dev_current = 0; |
33f0086c SB |
1176 | if (flash_io(O_RDONLY)) { |
1177 | ret = -EIO; | |
1178 | goto open_cleanup; | |
1179 | } | |
56086921 GL |
1180 | |
1181 | crc0 = crc32 (0, (uint8_t *) environment.data, ENV_SIZE); | |
a8a752c0 | 1182 | |
81974f44 | 1183 | if (opts->aes_flag) { |
dcdc1f6a | 1184 | ret = env_aes_cbc_crypt(environment.data, 0, |
81974f44 | 1185 | opts->aes_key); |
a8a752c0 | 1186 | if (ret) |
33f0086c | 1187 | goto open_cleanup; |
a8a752c0 MV |
1188 | } |
1189 | ||
56086921 | 1190 | crc0_ok = (crc0 == *environment.crc); |
d0fb80c3 | 1191 | if (!HaveRedundEnv) { |
56086921 | 1192 | if (!crc0_ok) { |
4a6fd34b WD |
1193 | fprintf (stderr, |
1194 | "Warning: Bad CRC, using default environment\n"); | |
f07217c9 | 1195 | memcpy(environment.data, default_environment, sizeof default_environment); |
6aff3115 | 1196 | } |
d0fb80c3 | 1197 | } else { |
56086921 | 1198 | flag0 = *environment.flags; |
4a6fd34b | 1199 | |
56086921 | 1200 | dev_current = 1; |
497f2053 | 1201 | addr1 = calloc(1, CUR_ENVSIZE); |
56086921 | 1202 | if (addr1 == NULL) { |
497f2053 | 1203 | fprintf(stderr, |
4a6fd34b | 1204 | "Not enough memory for environment (%ld bytes)\n", |
497f2053 | 1205 | CUR_ENVSIZE); |
33f0086c SB |
1206 | ret = -ENOMEM; |
1207 | goto open_cleanup; | |
4a6fd34b | 1208 | } |
56086921 | 1209 | redundant = addr1; |
4a6fd34b | 1210 | |
56086921 GL |
1211 | /* |
1212 | * have to set environment.image for flash_read(), careful - | |
1213 | * other pointers in environment still point inside addr0 | |
1214 | */ | |
1215 | environment.image = addr1; | |
33f0086c SB |
1216 | if (flash_io(O_RDONLY)) { |
1217 | ret = -EIO; | |
1218 | goto open_cleanup; | |
1219 | } | |
56086921 GL |
1220 | |
1221 | /* Check flag scheme compatibility */ | |
1222 | if (DEVTYPE(dev_current) == MTD_NORFLASH && | |
1223 | DEVTYPE(!dev_current) == MTD_NORFLASH) { | |
1224 | environment.flag_scheme = FLAG_BOOLEAN; | |
1225 | } else if (DEVTYPE(dev_current) == MTD_NANDFLASH && | |
1226 | DEVTYPE(!dev_current) == MTD_NANDFLASH) { | |
1227 | environment.flag_scheme = FLAG_INCREMENTAL; | |
9eeaa8e6 RB |
1228 | } else if (DEVTYPE(dev_current) == MTD_DATAFLASH && |
1229 | DEVTYPE(!dev_current) == MTD_DATAFLASH) { | |
1230 | environment.flag_scheme = FLAG_BOOLEAN; | |
785881f7 JH |
1231 | } else if (DEVTYPE(dev_current) == MTD_UBIVOLUME && |
1232 | DEVTYPE(!dev_current) == MTD_UBIVOLUME) { | |
1233 | environment.flag_scheme = FLAG_INCREMENTAL; | |
23ef62d7 OM |
1234 | } else if (DEVTYPE(dev_current) == MTD_ABSENT && |
1235 | DEVTYPE(!dev_current) == MTD_ABSENT) { | |
1236 | environment.flag_scheme = FLAG_INCREMENTAL; | |
56086921 GL |
1237 | } else { |
1238 | fprintf (stderr, "Incompatible flash types!\n"); | |
33f0086c SB |
1239 | ret = -EINVAL; |
1240 | goto open_cleanup; | |
6aff3115 | 1241 | } |
4a6fd34b | 1242 | |
56086921 | 1243 | crc1 = crc32 (0, (uint8_t *) redundant->data, ENV_SIZE); |
a8a752c0 | 1244 | |
81974f44 | 1245 | if (opts->aes_flag) { |
dcdc1f6a | 1246 | ret = env_aes_cbc_crypt(redundant->data, 0, |
81974f44 | 1247 | opts->aes_key); |
a8a752c0 | 1248 | if (ret) |
33f0086c | 1249 | goto open_cleanup; |
a8a752c0 MV |
1250 | } |
1251 | ||
56086921 GL |
1252 | crc1_ok = (crc1 == redundant->crc); |
1253 | flag1 = redundant->flags; | |
1254 | ||
1255 | if (crc0_ok && !crc1_ok) { | |
1256 | dev_current = 0; | |
1257 | } else if (!crc0_ok && crc1_ok) { | |
1258 | dev_current = 1; | |
1259 | } else if (!crc0_ok && !crc1_ok) { | |
4a6fd34b WD |
1260 | fprintf (stderr, |
1261 | "Warning: Bad CRC, using default environment\n"); | |
56086921 GL |
1262 | memcpy (environment.data, default_environment, |
1263 | sizeof default_environment); | |
1264 | dev_current = 0; | |
1265 | } else { | |
1266 | switch (environment.flag_scheme) { | |
1267 | case FLAG_BOOLEAN: | |
1268 | if (flag0 == active_flag && | |
1269 | flag1 == obsolete_flag) { | |
1270 | dev_current = 0; | |
1271 | } else if (flag0 == obsolete_flag && | |
1272 | flag1 == active_flag) { | |
1273 | dev_current = 1; | |
1274 | } else if (flag0 == flag1) { | |
1275 | dev_current = 0; | |
1276 | } else if (flag0 == 0xFF) { | |
1277 | dev_current = 0; | |
1278 | } else if (flag1 == 0xFF) { | |
1279 | dev_current = 1; | |
1280 | } else { | |
1281 | dev_current = 0; | |
1282 | } | |
1283 | break; | |
1284 | case FLAG_INCREMENTAL: | |
735eb0f0 | 1285 | if (flag0 == 255 && flag1 == 0) |
56086921 GL |
1286 | dev_current = 1; |
1287 | else if ((flag1 == 255 && flag0 == 0) || | |
735eb0f0 | 1288 | flag0 >= flag1) |
56086921 | 1289 | dev_current = 0; |
735eb0f0 JP |
1290 | else /* flag1 > flag0 */ |
1291 | dev_current = 1; | |
56086921 GL |
1292 | break; |
1293 | default: | |
1294 | fprintf (stderr, "Unknown flag scheme %u \n", | |
1295 | environment.flag_scheme); | |
1296 | return -1; | |
1297 | } | |
1298 | } | |
1299 | ||
1300 | /* | |
1301 | * If we are reading, we don't need the flag and the CRC any | |
1302 | * more, if we are writing, we will re-calculate CRC and update | |
1303 | * flags before writing out | |
1304 | */ | |
1305 | if (dev_current) { | |
1306 | environment.image = addr1; | |
1307 | environment.crc = &redundant->crc; | |
1308 | environment.flags = &redundant->flags; | |
1309 | environment.data = redundant->data; | |
1310 | free (addr0); | |
1311 | } else { | |
1312 | environment.image = addr0; | |
1313 | /* Other pointers are already set */ | |
4a6fd34b | 1314 | free (addr1); |
6aff3115 | 1315 | } |
74620e1c JH |
1316 | #ifdef DEBUG |
1317 | fprintf(stderr, "Selected env in %s\n", DEVNAME(dev_current)); | |
1318 | #endif | |
6aff3115 | 1319 | } |
56086921 | 1320 | return 0; |
33f0086c SB |
1321 | |
1322 | open_cleanup: | |
1323 | if (addr0) | |
1324 | free(addr0); | |
1325 | ||
1326 | if (addr1) | |
1327 | free(addr0); | |
1328 | ||
1329 | return ret; | |
1330 | } | |
1331 | ||
1332 | /* | |
1333 | * Simply free allocated buffer with environment | |
1334 | */ | |
1335 | int fw_env_close(struct env_opts *opts) | |
1336 | { | |
1337 | if (environment.image) | |
1338 | free(environment.image); | |
1339 | ||
1340 | environment.image = NULL; | |
1341 | ||
1342 | return 0; | |
6aff3115 WD |
1343 | } |
1344 | ||
14fb5b25 SA |
1345 | static int check_device_config(int dev) |
1346 | { | |
1347 | struct stat st; | |
1348 | int fd, rc = 0; | |
1349 | ||
1350 | fd = open(DEVNAME(dev), O_RDONLY); | |
1351 | if (fd < 0) { | |
1352 | fprintf(stderr, | |
1353 | "Cannot open %s: %s\n", | |
1354 | DEVNAME(dev), strerror(errno)); | |
1355 | return -1; | |
1356 | } | |
1357 | ||
1358 | rc = fstat(fd, &st); | |
1359 | if (rc < 0) { | |
1360 | fprintf(stderr, "Cannot stat the file %s\n", | |
1361 | DEVNAME(dev)); | |
1362 | goto err; | |
1363 | } | |
1364 | ||
1365 | if (S_ISCHR(st.st_mode)) { | |
1366 | struct mtd_info_user mtdinfo; | |
1367 | rc = ioctl(fd, MEMGETINFO, &mtdinfo); | |
1368 | if (rc < 0) { | |
1369 | fprintf(stderr, "Cannot get MTD information for %s\n", | |
1370 | DEVNAME(dev)); | |
1371 | goto err; | |
1372 | } | |
1373 | if (mtdinfo.type != MTD_NORFLASH && | |
1374 | mtdinfo.type != MTD_NANDFLASH && | |
1375 | mtdinfo.type != MTD_DATAFLASH && | |
1376 | mtdinfo.type != MTD_UBIVOLUME) { | |
1377 | fprintf(stderr, "Unsupported flash type %u on %s\n", | |
1378 | mtdinfo.type, DEVNAME(dev)); | |
1379 | goto err; | |
1380 | } | |
1381 | DEVTYPE(dev) = mtdinfo.type; | |
333ee16d MK |
1382 | if (DEVESIZE(dev) == 0) |
1383 | /* Assume the erase size is the same as the env-size */ | |
1384 | DEVESIZE(dev) = ENVSIZE(dev); | |
14fb5b25 | 1385 | } else { |
f4742ca0 | 1386 | uint64_t size; |
14fb5b25 | 1387 | DEVTYPE(dev) = MTD_ABSENT; |
333ee16d MK |
1388 | if (DEVESIZE(dev) == 0) |
1389 | /* Assume the erase size to be 512 bytes */ | |
1390 | DEVESIZE(dev) = 0x200; | |
f4742ca0 SA |
1391 | |
1392 | /* | |
1393 | * Check for negative offsets, treat it as backwards offset | |
1394 | * from the end of the block device | |
1395 | */ | |
1396 | if (DEVOFFSET(dev) < 0) { | |
1397 | rc = ioctl(fd, BLKGETSIZE64, &size); | |
1398 | if (rc < 0) { | |
1399 | fprintf(stderr, "Could not get block device size on %s\n", | |
1400 | DEVNAME(dev)); | |
1401 | goto err; | |
1402 | } | |
1403 | ||
1404 | DEVOFFSET(dev) = DEVOFFSET(dev) + size; | |
1405 | #ifdef DEBUG | |
1406 | fprintf(stderr, "Calculated device offset 0x%llx on %s\n", | |
1407 | DEVOFFSET(dev), DEVNAME(dev)); | |
1408 | #endif | |
1409 | } | |
14fb5b25 SA |
1410 | } |
1411 | ||
333ee16d MK |
1412 | if (ENVSECTORS(dev) == 0) |
1413 | /* Assume enough sectors to cover the environment */ | |
1414 | ENVSECTORS(dev) = DIV_ROUND_UP(ENVSIZE(dev), DEVESIZE(dev)); | |
1415 | ||
1416 | if (DEVOFFSET(dev) % DEVESIZE(dev) != 0) { | |
1417 | fprintf(stderr, "Environment does not start on (erase) block boundary\n"); | |
1418 | errno = EINVAL; | |
1419 | return -1; | |
1420 | } | |
1421 | ||
1422 | if (ENVSIZE(dev) > ENVSECTORS(dev) * DEVESIZE(dev)) { | |
1423 | fprintf(stderr, "Environment does not fit into available sectors\n"); | |
1424 | errno = EINVAL; | |
1425 | return -1; | |
1426 | } | |
1427 | ||
14fb5b25 SA |
1428 | err: |
1429 | close(fd); | |
1430 | return rc; | |
1431 | } | |
6aff3115 | 1432 | |
81974f44 | 1433 | static int parse_config(struct env_opts *opts) |
6aff3115 | 1434 | { |
14fb5b25 | 1435 | int rc; |
6aff3115 | 1436 | |
14070e69 AF |
1437 | if (!opts) |
1438 | opts = &default_opts; | |
925c97c2 | 1439 | |
14070e69 | 1440 | #if defined(CONFIG_FILE) |
d0fb80c3 | 1441 | /* Fills in DEVNAME(), ENVSIZE(), DEVESIZE(). Or don't. */ |
81974f44 | 1442 | if (get_config(opts->config_file)) { |
371ee137 | 1443 | fprintf(stderr, "Cannot parse config file '%s': %m\n", |
81974f44 | 1444 | opts->config_file); |
56086921 | 1445 | return -1; |
6aff3115 | 1446 | } |
d0fb80c3 | 1447 | #else |
229695fe | 1448 | DEVNAME (0) = DEVICE1_NAME; |
4a6fd34b WD |
1449 | DEVOFFSET (0) = DEVICE1_OFFSET; |
1450 | ENVSIZE (0) = ENV1_SIZE; | |
333ee16d MK |
1451 | |
1452 | /* Set defaults for DEVESIZE, ENVSECTORS later once we | |
1453 | * know DEVTYPE | |
1454 | */ | |
9eeaa8e6 | 1455 | #ifdef DEVICE1_ESIZE |
4a6fd34b | 1456 | DEVESIZE (0) = DEVICE1_ESIZE; |
9eeaa8e6 RB |
1457 | #endif |
1458 | #ifdef DEVICE1_ENVSECTORS | |
56086921 | 1459 | ENVSECTORS (0) = DEVICE1_ENVSECTORS; |
9eeaa8e6 RB |
1460 | #endif |
1461 | ||
6aff3115 | 1462 | #ifdef HAVE_REDUND |
229695fe | 1463 | DEVNAME (1) = DEVICE2_NAME; |
4a6fd34b WD |
1464 | DEVOFFSET (1) = DEVICE2_OFFSET; |
1465 | ENVSIZE (1) = ENV2_SIZE; | |
333ee16d MK |
1466 | |
1467 | /* Set defaults for DEVESIZE, ENVSECTORS later once we | |
1468 | * know DEVTYPE | |
1469 | */ | |
9eeaa8e6 | 1470 | #ifdef DEVICE2_ESIZE |
4a6fd34b | 1471 | DEVESIZE (1) = DEVICE2_ESIZE; |
9eeaa8e6 RB |
1472 | #endif |
1473 | #ifdef DEVICE2_ENVSECTORS | |
56086921 | 1474 | ENVSECTORS (1) = DEVICE2_ENVSECTORS; |
9eeaa8e6 | 1475 | #endif |
d0fb80c3 | 1476 | HaveRedundEnv = 1; |
6aff3115 | 1477 | #endif |
d0fb80c3 | 1478 | #endif |
14fb5b25 SA |
1479 | rc = check_device_config(0); |
1480 | if (rc < 0) | |
1481 | return rc; | |
4a6fd34b | 1482 | |
14fb5b25 SA |
1483 | if (HaveRedundEnv) { |
1484 | rc = check_device_config(1); | |
1485 | if (rc < 0) | |
1486 | return rc; | |
f71cee4b | 1487 | |
14fb5b25 | 1488 | if (ENVSIZE(0) != ENVSIZE(1)) { |
14fb5b25 | 1489 | fprintf(stderr, |
490365c3 AF |
1490 | "Redundant environments have unequal size"); |
1491 | return -1; | |
14fb5b25 | 1492 | } |
f71cee4b AF |
1493 | } |
1494 | ||
1495 | usable_envsize = CUR_ENVSIZE - sizeof(uint32_t); | |
1496 | if (HaveRedundEnv) | |
1497 | usable_envsize -= sizeof(char); | |
1498 | ||
81974f44 | 1499 | if (opts->aes_flag) |
f71cee4b AF |
1500 | usable_envsize &= ~(AES_KEY_LENGTH - 1); |
1501 | ||
6aff3115 WD |
1502 | return 0; |
1503 | } | |
d0fb80c3 WD |
1504 | |
1505 | #if defined(CONFIG_FILE) | |
1506 | static int get_config (char *fname) | |
1507 | { | |
1508 | FILE *fp; | |
1509 | int i = 0; | |
1510 | int rc; | |
1511 | char dump[128]; | |
229695fe | 1512 | char *devname; |
d0fb80c3 | 1513 | |
56086921 GL |
1514 | fp = fopen (fname, "r"); |
1515 | if (fp == NULL) | |
1516 | return -1; | |
d0fb80c3 | 1517 | |
56086921 | 1518 | while (i < 2 && fgets (dump, sizeof (dump), fp)) { |
d0fb80c3 | 1519 | /* Skip incomplete conversions and comment strings */ |
56086921 | 1520 | if (dump[0] == '#') |
d0fb80c3 | 1521 | continue; |
56086921 | 1522 | |
f4742ca0 SA |
1523 | rc = sscanf(dump, "%ms %lli %lx %lx %lx", |
1524 | &devname, | |
1525 | &DEVOFFSET(i), | |
1526 | &ENVSIZE(i), | |
1527 | &DEVESIZE(i), | |
1528 | &ENVSECTORS(i)); | |
56086921 | 1529 | |
9eeaa8e6 | 1530 | if (rc < 3) |
56086921 GL |
1531 | continue; |
1532 | ||
229695fe TR |
1533 | DEVNAME(i) = devname; |
1534 | ||
333ee16d MK |
1535 | /* Set defaults for DEVESIZE, ENVSECTORS later once we |
1536 | * know DEVTYPE | |
1537 | */ | |
d0fb80c3 WD |
1538 | |
1539 | i++; | |
1540 | } | |
4a6fd34b WD |
1541 | fclose (fp); |
1542 | ||
d0fb80c3 | 1543 | HaveRedundEnv = i - 1; |
4a6fd34b | 1544 | if (!i) { /* No valid entries found */ |
d0fb80c3 | 1545 | errno = EINVAL; |
56086921 | 1546 | return -1; |
d0fb80c3 WD |
1547 | } else |
1548 | return 0; | |
1549 | } | |
1550 | #endif |