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