blob: 996682ebc7e0797e5981adafdc7548d1df984262 [file] [log] [blame]
wdenk6aff3112002-12-17 01:51:00 +00001/*
Detlev Zundel0aef7bc2010-07-30 11:22:15 +02002 * (C) Copyright 2000-2010
wdenk6aff3112002-12-17 01:51:00 +00003 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4 *
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02005 * (C) Copyright 2008
6 * Guennadi Liakhovetski, DENX Software Engineering, lg@denx.de.
7 *
wdenk6aff3112002-12-17 01:51:00 +00008 * 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
wdenk4a6fd342003-04-12 23:38:12 +000018 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
wdenk6aff3112002-12-17 01:51:00 +000019 * 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 <fcntl.h>
29#include <stdio.h>
30#include <stdlib.h>
31#include <stddef.h>
32#include <string.h>
33#include <sys/types.h>
34#include <sys/ioctl.h>
35#include <sys/stat.h>
36#include <unistd.h>
wdenk6aff3112002-12-17 01:51:00 +000037
Markus Klotzbücher6de66b32007-11-27 10:23:20 +010038#ifdef MTD_OLD
Wolfgang Denk1711f3b2008-09-02 21:17:36 +020039# include <stdint.h>
Markus Klotzbücher6de66b32007-11-27 10:23:20 +010040# include <linux/mtd/mtd.h>
41#else
Wolfgang Denkc83d7ca2008-01-08 22:58:27 +010042# define __user /* nothing */
Markus Klotzbücher6de66b32007-11-27 10:23:20 +010043# include <mtd/mtd-user.h>
44#endif
45
46#include "fw_env.h"
wdenk6aff3112002-12-17 01:51:00 +000047
Stefano Babicbd7b26f2010-05-24 12:08:16 +020048#define WHITESPACE(c) ((c == '\t') || (c == ' '))
wdenk6aff3112002-12-17 01:51:00 +000049
Guennadi Liakhovetski56086922008-09-04 13:01:49 +020050#define min(x, y) ({ \
51 typeof(x) _min1 = (x); \
52 typeof(y) _min2 = (y); \
53 (void) (&_min1 == &_min2); \
54 _min1 < _min2 ? _min1 : _min2; })
55
56struct envdev_s {
Markus Klotzbücher6de66b32007-11-27 10:23:20 +010057 char devname[16]; /* Device name */
wdenk4a6fd342003-04-12 23:38:12 +000058 ulong devoff; /* Device offset */
59 ulong env_size; /* environment size */
60 ulong erase_size; /* device erase size */
Guennadi Liakhovetski56086922008-09-04 13:01:49 +020061 ulong env_sectors; /* number of environment sectors */
62 uint8_t mtd_type; /* type of the MTD device */
63};
wdenk6aff3112002-12-17 01:51:00 +000064
Guennadi Liakhovetski56086922008-09-04 13:01:49 +020065static struct envdev_s envdevices[2] =
66{
67 {
68 .mtd_type = MTD_ABSENT,
69 }, {
70 .mtd_type = MTD_ABSENT,
71 },
72};
73static int dev_current;
wdenk6aff3112002-12-17 01:51:00 +000074
75#define DEVNAME(i) envdevices[(i)].devname
wdenkd0fb80c2003-01-11 09:48:40 +000076#define DEVOFFSET(i) envdevices[(i)].devoff
wdenk6aff3112002-12-17 01:51:00 +000077#define ENVSIZE(i) envdevices[(i)].env_size
78#define DEVESIZE(i) envdevices[(i)].erase_size
Guennadi Liakhovetski56086922008-09-04 13:01:49 +020079#define ENVSECTORS(i) envdevices[(i)].env_sectors
80#define DEVTYPE(i) envdevices[(i)].mtd_type
wdenk6aff3112002-12-17 01:51:00 +000081
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +020082#define CONFIG_ENV_SIZE ENVSIZE(dev_current)
wdenk6aff3112002-12-17 01:51:00 +000083
wdenkd0fb80c2003-01-11 09:48:40 +000084#define ENV_SIZE getenvsize()
wdenk6aff3112002-12-17 01:51:00 +000085
Guennadi Liakhovetski56086922008-09-04 13:01:49 +020086struct env_image_single {
87 uint32_t crc; /* CRC32 over data bytes */
88 char data[];
89};
wdenk6aff3112002-12-17 01:51:00 +000090
Guennadi Liakhovetski56086922008-09-04 13:01:49 +020091struct env_image_redundant {
92 uint32_t crc; /* CRC32 over data bytes */
93 unsigned char flags; /* active or obsolete */
94 char data[];
95};
96
97enum flag_scheme {
98 FLAG_NONE,
99 FLAG_BOOLEAN,
100 FLAG_INCREMENTAL,
101};
102
103struct environment {
104 void *image;
105 uint32_t *crc;
106 unsigned char *flags;
107 char *data;
108 enum flag_scheme flag_scheme;
109};
110
111static struct environment environment = {
112 .flag_scheme = FLAG_NONE,
113};
wdenk6aff3112002-12-17 01:51:00 +0000114
wdenkd0fb80c2003-01-11 09:48:40 +0000115static int HaveRedundEnv = 0;
116
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100117static unsigned char active_flag = 1;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200118/* obsolete_flag must be 0 to efficiently set it on NOR flash without erasing */
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100119static unsigned char obsolete_flag = 0;
wdenkd0fb80c2003-01-11 09:48:40 +0000120
wdenk6aff3112002-12-17 01:51:00 +0000121
122#define XMK_STR(x) #x
123#define MK_STR(x) XMK_STR(x)
124
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100125static char default_environment[] = {
wdenkd0fb80c2003-01-11 09:48:40 +0000126#if defined(CONFIG_BOOTARGS)
wdenk4a6fd342003-04-12 23:38:12 +0000127 "bootargs=" CONFIG_BOOTARGS "\0"
wdenk6aff3112002-12-17 01:51:00 +0000128#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000129#if defined(CONFIG_BOOTCOMMAND)
wdenk4a6fd342003-04-12 23:38:12 +0000130 "bootcmd=" CONFIG_BOOTCOMMAND "\0"
wdenk6aff3112002-12-17 01:51:00 +0000131#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000132#if defined(CONFIG_RAMBOOTCOMMAND)
wdenk4a6fd342003-04-12 23:38:12 +0000133 "ramboot=" CONFIG_RAMBOOTCOMMAND "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000134#endif
135#if defined(CONFIG_NFSBOOTCOMMAND)
wdenk4a6fd342003-04-12 23:38:12 +0000136 "nfsboot=" CONFIG_NFSBOOTCOMMAND "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000137#endif
138#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
wdenk4a6fd342003-04-12 23:38:12 +0000139 "bootdelay=" MK_STR (CONFIG_BOOTDELAY) "\0"
wdenk6aff3112002-12-17 01:51:00 +0000140#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000141#if defined(CONFIG_BAUDRATE) && (CONFIG_BAUDRATE >= 0)
wdenk4a6fd342003-04-12 23:38:12 +0000142 "baudrate=" MK_STR (CONFIG_BAUDRATE) "\0"
wdenk6aff3112002-12-17 01:51:00 +0000143#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000144#ifdef CONFIG_LOADS_ECHO
wdenk4a6fd342003-04-12 23:38:12 +0000145 "loads_echo=" MK_STR (CONFIG_LOADS_ECHO) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000146#endif
wdenk6aff3112002-12-17 01:51:00 +0000147#ifdef CONFIG_ETHADDR
wdenk4a6fd342003-04-12 23:38:12 +0000148 "ethaddr=" MK_STR (CONFIG_ETHADDR) "\0"
wdenk6aff3112002-12-17 01:51:00 +0000149#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000150#ifdef CONFIG_ETH1ADDR
wdenk4a6fd342003-04-12 23:38:12 +0000151 "eth1addr=" MK_STR (CONFIG_ETH1ADDR) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000152#endif
153#ifdef CONFIG_ETH2ADDR
wdenk4a6fd342003-04-12 23:38:12 +0000154 "eth2addr=" MK_STR (CONFIG_ETH2ADDR) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000155#endif
wdenke2ffd592004-12-31 09:32:47 +0000156#ifdef CONFIG_ETH3ADDR
157 "eth3addr=" MK_STR (CONFIG_ETH3ADDR) "\0"
158#endif
richardretanubunc68a05f2008-09-29 18:28:23 -0400159#ifdef CONFIG_ETH4ADDR
160 "eth4addr=" MK_STR (CONFIG_ETH4ADDR) "\0"
161#endif
162#ifdef CONFIG_ETH5ADDR
163 "eth5addr=" MK_STR (CONFIG_ETH5ADDR) "\0"
164#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000165#ifdef CONFIG_ETHPRIME
wdenk4a6fd342003-04-12 23:38:12 +0000166 "ethprime=" CONFIG_ETHPRIME "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000167#endif
wdenk6aff3112002-12-17 01:51:00 +0000168#ifdef CONFIG_IPADDR
wdenk4a6fd342003-04-12 23:38:12 +0000169 "ipaddr=" MK_STR (CONFIG_IPADDR) "\0"
wdenk6aff3112002-12-17 01:51:00 +0000170#endif
171#ifdef CONFIG_SERVERIP
wdenk4a6fd342003-04-12 23:38:12 +0000172 "serverip=" MK_STR (CONFIG_SERVERIP) "\0"
wdenk6aff3112002-12-17 01:51:00 +0000173#endif
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200174#ifdef CONFIG_SYS_AUTOLOAD
175 "autoload=" CONFIG_SYS_AUTOLOAD "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000176#endif
177#ifdef CONFIG_ROOTPATH
Joe Hershberger8b3637c2011-10-13 13:03:47 +0000178 "rootpath=" CONFIG_ROOTPATH "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000179#endif
180#ifdef CONFIG_GATEWAYIP
wdenk4a6fd342003-04-12 23:38:12 +0000181 "gatewayip=" MK_STR (CONFIG_GATEWAYIP) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000182#endif
183#ifdef CONFIG_NETMASK
wdenk4a6fd342003-04-12 23:38:12 +0000184 "netmask=" MK_STR (CONFIG_NETMASK) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000185#endif
186#ifdef CONFIG_HOSTNAME
wdenk4a6fd342003-04-12 23:38:12 +0000187 "hostname=" MK_STR (CONFIG_HOSTNAME) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000188#endif
189#ifdef CONFIG_BOOTFILE
Joe Hershbergerb3f44c22011-10-13 13:03:48 +0000190 "bootfile=" CONFIG_BOOTFILE "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000191#endif
192#ifdef CONFIG_LOADADDR
wdenk4a6fd342003-04-12 23:38:12 +0000193 "loadaddr=" MK_STR (CONFIG_LOADADDR) "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000194#endif
195#ifdef CONFIG_PREBOOT
wdenk4a6fd342003-04-12 23:38:12 +0000196 "preboot=" CONFIG_PREBOOT "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000197#endif
198#ifdef CONFIG_CLOCKS_IN_MHZ
wdenk4a6fd342003-04-12 23:38:12 +0000199 "clocks_in_mhz=" "1" "\0"
wdenkd0fb80c2003-01-11 09:48:40 +0000200#endif
stroesead10dd92003-02-14 11:21:23 +0000201#if defined(CONFIG_PCI_BOOTDELAY) && (CONFIG_PCI_BOOTDELAY > 0)
wdenk4a6fd342003-04-12 23:38:12 +0000202 "pcidelay=" MK_STR (CONFIG_PCI_BOOTDELAY) "\0"
stroesead10dd92003-02-14 11:21:23 +0000203#endif
wdenkd0fb80c2003-01-11 09:48:40 +0000204#ifdef CONFIG_EXTRA_ENV_SETTINGS
205 CONFIG_EXTRA_ENV_SETTINGS
206#endif
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200207 "\0" /* Termimate struct environment data with 2 NULs */
wdenk6aff3112002-12-17 01:51:00 +0000208};
209
wdenk4a6fd342003-04-12 23:38:12 +0000210static int flash_io (int mode);
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100211static char *envmatch (char * s1, char * s2);
wdenk4a6fd342003-04-12 23:38:12 +0000212static int parse_config (void);
213
wdenkd0fb80c2003-01-11 09:48:40 +0000214#if defined(CONFIG_FILE)
wdenk4a6fd342003-04-12 23:38:12 +0000215static int get_config (char *);
wdenkd0fb80c2003-01-11 09:48:40 +0000216#endif
wdenk4a6fd342003-04-12 23:38:12 +0000217static inline ulong getenvsize (void)
wdenkd0fb80c2003-01-11 09:48:40 +0000218{
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +0200219 ulong rc = CONFIG_ENV_SIZE - sizeof (long);
wdenk4a6fd342003-04-12 23:38:12 +0000220
wdenkd0fb80c2003-01-11 09:48:40 +0000221 if (HaveRedundEnv)
wdenk4a6fd342003-04-12 23:38:12 +0000222 rc -= sizeof (char);
wdenkd0fb80c2003-01-11 09:48:40 +0000223 return rc;
224}
wdenk6aff3112002-12-17 01:51:00 +0000225
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200226static char *fw_string_blank(char *s, int noblank)
227{
228 int i;
229 int len = strlen(s);
230
231 for (i = 0; i < len; i++, s++) {
232 if ((noblank && !WHITESPACE(*s)) ||
233 (!noblank && WHITESPACE(*s)))
234 break;
235 }
236 if (i == len)
237 return NULL;
238
239 return s;
240}
241
wdenk6aff3112002-12-17 01:51:00 +0000242/*
243 * Search the environment for a variable.
244 * Return the value, if found, or NULL, if not found.
245 */
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100246char *fw_getenv (char *name)
wdenk6aff3112002-12-17 01:51:00 +0000247{
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100248 char *env, *nxt;
wdenk6aff3112002-12-17 01:51:00 +0000249
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200250 if (fw_env_open())
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200251 return NULL;
wdenk6aff3112002-12-17 01:51:00 +0000252
wdenk4a6fd342003-04-12 23:38:12 +0000253 for (env = environment.data; *env; env = nxt + 1) {
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100254 char *val;
wdenk6aff3112002-12-17 01:51:00 +0000255
wdenk4a6fd342003-04-12 23:38:12 +0000256 for (nxt = env; *nxt; ++nxt) {
wdenk6aff3112002-12-17 01:51:00 +0000257 if (nxt >= &environment.data[ENV_SIZE]) {
258 fprintf (stderr, "## Error: "
259 "environment not terminated\n");
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200260 return NULL;
wdenk6aff3112002-12-17 01:51:00 +0000261 }
262 }
wdenk4a6fd342003-04-12 23:38:12 +0000263 val = envmatch (name, env);
wdenk6aff3112002-12-17 01:51:00 +0000264 if (!val)
265 continue;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200266 return val;
wdenk6aff3112002-12-17 01:51:00 +0000267 }
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200268 return NULL;
wdenk6aff3112002-12-17 01:51:00 +0000269}
270
271/*
272 * Print the current definition of one, or more, or all
273 * environment variables
274 */
Grant Ericksonbc117562008-05-06 20:16:15 -0700275int fw_printenv (int argc, char *argv[])
wdenk6aff3112002-12-17 01:51:00 +0000276{
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100277 char *env, *nxt;
wdenk6aff3112002-12-17 01:51:00 +0000278 int i, n_flag;
Grant Ericksonbc117562008-05-06 20:16:15 -0700279 int rc = 0;
wdenk6aff3112002-12-17 01:51:00 +0000280
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200281 if (fw_env_open())
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200282 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000283
wdenk4a6fd342003-04-12 23:38:12 +0000284 if (argc == 1) { /* Print all env variables */
285 for (env = environment.data; *env; env = nxt + 1) {
286 for (nxt = env; *nxt; ++nxt) {
wdenk6aff3112002-12-17 01:51:00 +0000287 if (nxt >= &environment.data[ENV_SIZE]) {
288 fprintf (stderr, "## Error: "
289 "environment not terminated\n");
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200290 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000291 }
292 }
293
wdenk4a6fd342003-04-12 23:38:12 +0000294 printf ("%s\n", env);
wdenk6aff3112002-12-17 01:51:00 +0000295 }
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200296 return 0;
wdenk6aff3112002-12-17 01:51:00 +0000297 }
298
wdenk4a6fd342003-04-12 23:38:12 +0000299 if (strcmp (argv[1], "-n") == 0) {
wdenk6aff3112002-12-17 01:51:00 +0000300 n_flag = 1;
301 ++argv;
302 --argc;
303 if (argc != 2) {
304 fprintf (stderr, "## Error: "
305 "`-n' option requires exactly one argument\n");
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200306 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000307 }
308 } else {
309 n_flag = 0;
310 }
311
wdenk4a6fd342003-04-12 23:38:12 +0000312 for (i = 1; i < argc; ++i) { /* print single env variables */
Markus Klotzbücher6de66b32007-11-27 10:23:20 +0100313 char *name = argv[i];
314 char *val = NULL;
wdenk6aff3112002-12-17 01:51:00 +0000315
wdenk4a6fd342003-04-12 23:38:12 +0000316 for (env = environment.data; *env; env = nxt + 1) {
wdenk6aff3112002-12-17 01:51:00 +0000317
wdenk4a6fd342003-04-12 23:38:12 +0000318 for (nxt = env; *nxt; ++nxt) {
wdenk6aff3112002-12-17 01:51:00 +0000319 if (nxt >= &environment.data[ENV_SIZE]) {
320 fprintf (stderr, "## Error: "
321 "environment not terminated\n");
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200322 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000323 }
324 }
wdenk4a6fd342003-04-12 23:38:12 +0000325 val = envmatch (name, env);
wdenk6aff3112002-12-17 01:51:00 +0000326 if (val) {
327 if (!n_flag) {
328 fputs (name, stdout);
wdenk4a6fd342003-04-12 23:38:12 +0000329 putc ('=', stdout);
wdenk6aff3112002-12-17 01:51:00 +0000330 }
wdenk4a6fd342003-04-12 23:38:12 +0000331 puts (val);
wdenk6aff3112002-12-17 01:51:00 +0000332 break;
333 }
334 }
Grant Ericksonbc117562008-05-06 20:16:15 -0700335 if (!val) {
wdenk4a6fd342003-04-12 23:38:12 +0000336 fprintf (stderr, "## Error: \"%s\" not defined\n", name);
Grant Ericksonbc117562008-05-06 20:16:15 -0700337 rc = -1;
338 }
wdenk6aff3112002-12-17 01:51:00 +0000339 }
Grant Ericksonbc117562008-05-06 20:16:15 -0700340
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200341 return rc;
wdenk6aff3112002-12-17 01:51:00 +0000342}
343
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200344int fw_env_close(void)
wdenk6aff3112002-12-17 01:51:00 +0000345{
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200346 /*
347 * Update CRC
348 */
349 *environment.crc = crc32(0, (uint8_t *) environment.data, ENV_SIZE);
wdenk6aff3112002-12-17 01:51:00 +0000350
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200351 /* write environment back to flash */
352 if (flash_io(O_RDWR)) {
353 fprintf(stderr,
354 "Error: can't write fw_env to flash\n");
355 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000356 }
357
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200358 return 0;
359}
wdenk6aff3112002-12-17 01:51:00 +0000360
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200361
362/*
363 * Set/Clear a single variable in the environment.
364 * This is called in sequence to update the environment
365 * in RAM without updating the copy in flash after each set
366 */
367int fw_env_write(char *name, char *value)
368{
369 int len;
370 char *env, *nxt;
371 char *oldval = NULL;
wdenk6aff3112002-12-17 01:51:00 +0000372
373 /*
374 * search if variable with this name already exists
375 */
wdenk4a6fd342003-04-12 23:38:12 +0000376 for (nxt = env = environment.data; *env; env = nxt + 1) {
377 for (nxt = env; *nxt; ++nxt) {
wdenk6aff3112002-12-17 01:51:00 +0000378 if (nxt >= &environment.data[ENV_SIZE]) {
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200379 fprintf(stderr, "## Error: "
wdenk6aff3112002-12-17 01:51:00 +0000380 "environment not terminated\n");
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200381 errno = EINVAL;
382 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000383 }
384 }
wdenk4a6fd342003-04-12 23:38:12 +0000385 if ((oldval = envmatch (name, env)) != NULL)
wdenk6aff3112002-12-17 01:51:00 +0000386 break;
387 }
388
389 /*
390 * Delete any existing definition
391 */
392 if (oldval) {
393 /*
394 * Ethernet Address and serial# can be set only once
395 */
396 if ((strcmp (name, "ethaddr") == 0) ||
wdenk4a6fd342003-04-12 23:38:12 +0000397 (strcmp (name, "serial#") == 0)) {
wdenk6aff3112002-12-17 01:51:00 +0000398 fprintf (stderr, "Can't overwrite \"%s\"\n", name);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200399 errno = EROFS;
400 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000401 }
402
403 if (*++nxt == '\0') {
404 *env = '\0';
405 } else {
406 for (;;) {
407 *env = *nxt++;
408 if ((*env == '\0') && (*nxt == '\0'))
409 break;
410 ++env;
411 }
412 }
413 *++env = '\0';
414 }
415
416 /* Delete only ? */
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200417 if (!value || !strlen(value))
418 return 0;
wdenk6aff3112002-12-17 01:51:00 +0000419
420 /*
421 * Append new definition at the end
422 */
wdenk4a6fd342003-04-12 23:38:12 +0000423 for (env = environment.data; *env || *(env + 1); ++env);
wdenk6aff3112002-12-17 01:51:00 +0000424 if (env > environment.data)
425 ++env;
426 /*
427 * Overflow when:
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +0200428 * "name" + "=" + "val" +"\0\0" > CONFIG_ENV_SIZE - (env-environment)
wdenk6aff3112002-12-17 01:51:00 +0000429 */
wdenk4a6fd342003-04-12 23:38:12 +0000430 len = strlen (name) + 2;
wdenk6aff3112002-12-17 01:51:00 +0000431 /* add '=' for first arg, ' ' for all others */
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200432 len += strlen(value) + 1;
433
wdenk4a6fd342003-04-12 23:38:12 +0000434 if (len > (&environment.data[ENV_SIZE] - env)) {
wdenk6aff3112002-12-17 01:51:00 +0000435 fprintf (stderr,
436 "Error: environment overflow, \"%s\" deleted\n",
437 name);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200438 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000439 }
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200440
wdenk6aff3112002-12-17 01:51:00 +0000441 while ((*env = *name++) != '\0')
442 env++;
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200443 *env = '=';
444 while ((*++env = *value++) != '\0')
445 ;
wdenk6aff3112002-12-17 01:51:00 +0000446
447 /* end is marked with double '\0' */
448 *++env = '\0';
449
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200450 return 0;
451}
wdenk6aff3112002-12-17 01:51:00 +0000452
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200453/*
454 * Deletes or sets environment variables. Returns -1 and sets errno error codes:
455 * 0 - OK
456 * EINVAL - need at least 1 argument
457 * EROFS - certain variables ("ethaddr", "serial#") cannot be
458 * modified or deleted
459 *
460 */
461int fw_setenv(int argc, char *argv[])
462{
463 int i, len;
464 char *name;
465 char *value = NULL;
466 char *tmpval = NULL;
wdenk6aff3112002-12-17 01:51:00 +0000467
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200468 if (argc < 2) {
469 errno = EINVAL;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200470 return -1;
wdenk6aff3112002-12-17 01:51:00 +0000471 }
472
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200473 if (fw_env_open()) {
474 fprintf(stderr, "Error: environment not initialized\n");
475 return -1;
476 }
477
478 name = argv[1];
479
480 len = strlen(name) + 2;
481 for (i = 2; i < argc; ++i)
482 len += strlen(argv[i]) + 1;
483
484 /* Allocate enough place to the data string */
485 for (i = 2; i < argc; ++i) {
486 char *val = argv[i];
487 if (!value) {
488 value = (char *)malloc(len - strlen(name));
489 if (!value) {
490 fprintf(stderr,
Luka Perkov8603b692011-09-05 23:40:13 +0200491 "Cannot malloc %zu bytes: %s\n",
Stefano Babicbd7b26f2010-05-24 12:08:16 +0200492 len - strlen(name), strerror(errno));
493 return -1;
494 }
495 memset(value, 0, len - strlen(name));
496 tmpval = value;
497 }
498 if (i != 2)
499 *tmpval++ = ' ';
500 while (*val != '\0')
501 *tmpval++ = *val++;
502 }
503
504 fw_env_write(name, value);
505
506 if (value)
507 free(value);
508
509 return fw_env_close();
510}
511
512/*
513 * Parse a file and configure the u-boot variables.
514 * The script file has a very simple format, as follows:
515 *
516 * Each line has a couple with name, value:
517 * <white spaces>variable_name<white spaces>variable_value
518 *
519 * Both variable_name and variable_value are interpreted as strings.
520 * Any character after <white spaces> and before ending \r\n is interpreted
521 * as variable's value (no comment allowed on these lines !)
522 *
523 * Comments are allowed if the first character in the line is #
524 *
525 * Returns -1 and sets errno error codes:
526 * 0 - OK
527 * -1 - Error
528 */
529int fw_parse_script(char *fname)
530{
531 FILE *fp;
532 char dump[1024]; /* Maximum line length in the file */
533 char *name;
534 char *val;
535 int lineno = 0;
536 int len;
537 int ret = 0;
538
539 if (fw_env_open()) {
540 fprintf(stderr, "Error: environment not initialized\n");
541 return -1;
542 }
543
544 if (strcmp(fname, "-") == 0)
545 fp = stdin;
546 else {
547 fp = fopen(fname, "r");
548 if (fp == NULL) {
549 fprintf(stderr, "I cannot open %s for reading\n",
550 fname);
551 return -1;
552 }
553 }
554
555 while (fgets(dump, sizeof(dump), fp)) {
556 lineno++;
557 len = strlen(dump);
558
559 /*
560 * Read a whole line from the file. If the line is too long
561 * or is not terminated, reports an error and exit.
562 */
563 if (dump[len - 1] != '\n') {
564 fprintf(stderr,
565 "Line %d not corrected terminated or too long\n",
566 lineno);
567 ret = -1;
568 break;
569 }
570
571 /* Drop ending line feed / carriage return */
572 while (len > 0 && (dump[len - 1] == '\n' ||
573 dump[len - 1] == '\r')) {
574 dump[len - 1] = '\0';
575 len--;
576 }
577
578 /* Skip comment or empty lines */
579 if ((len == 0) || dump[0] == '#')
580 continue;
581
582 /*
583 * Search for variable's name,
584 * remove leading whitespaces
585 */
586 name = fw_string_blank(dump, 1);
587 if (!name)
588 continue;
589
590 /* The first white space is the end of variable name */
591 val = fw_string_blank(name, 0);
592 len = strlen(name);
593 if (val) {
594 *val++ = '\0';
595 if ((val - name) < len)
596 val = fw_string_blank(val, 1);
597 else
598 val = NULL;
599 }
600
601#ifdef DEBUG
602 fprintf(stderr, "Setting %s : %s\n",
603 name, val ? val : " removed");
604#endif
605
606 /*
607 * If there is an error setting a variable,
608 * try to save the environment and returns an error
609 */
610 if (fw_env_write(name, val)) {
611 fprintf(stderr,
612 "fw_env_write returns with error : %s\n",
613 strerror(errno));
614 ret = -1;
615 break;
616 }
617
618 }
619
620 /* Close file if not stdin */
621 if (strcmp(fname, "-") != 0)
622 fclose(fp);
623
624 ret |= fw_env_close();
625
626 return ret;
627
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200628}
629
630/*
631 * Test for bad block on NAND, just returns 0 on NOR, on NAND:
632 * 0 - block is good
633 * > 0 - block is bad
634 * < 0 - failed to test
635 */
636static int flash_bad_block (int fd, uint8_t mtd_type, loff_t *blockstart)
637{
638 if (mtd_type == MTD_NANDFLASH) {
639 int badblock = ioctl (fd, MEMGETBADBLOCK, blockstart);
640
641 if (badblock < 0) {
642 perror ("Cannot read bad block mark");
643 return badblock;
644 }
645
646 if (badblock) {
647#ifdef DEBUG
648 fprintf (stderr, "Bad block at 0x%llx, "
649 "skipping\n", *blockstart);
650#endif
651 return badblock;
652 }
653 }
654
655 return 0;
656}
657
658/*
659 * Read data from flash at an offset into a provided buffer. On NAND it skips
660 * bad blocks but makes sure it stays within ENVSECTORS (dev) starting from
661 * the DEVOFFSET (dev) block. On NOR the loop is only run once.
662 */
663static int flash_read_buf (int dev, int fd, void *buf, size_t count,
664 off_t offset, uint8_t mtd_type)
665{
666 size_t blocklen; /* erase / write length - one block on NAND,
667 0 on NOR */
668 size_t processed = 0; /* progress counter */
669 size_t readlen = count; /* current read length */
670 off_t top_of_range; /* end of the last block we may use */
671 off_t block_seek; /* offset inside the current block to the start
672 of the data */
673 loff_t blockstart; /* running start of the current block -
674 MEMGETBADBLOCK needs 64 bits */
675 int rc;
676
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100677 blockstart = (offset / DEVESIZE (dev)) * DEVESIZE (dev);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200678
679 /* Offset inside a block */
680 block_seek = offset - blockstart;
681
682 if (mtd_type == MTD_NANDFLASH) {
683 /*
684 * NAND: calculate which blocks we are reading. We have
685 * to read one block at a time to skip bad blocks.
686 */
687 blocklen = DEVESIZE (dev);
688
689 /*
690 * To calculate the top of the range, we have to use the
691 * global DEVOFFSET (dev), which can be different from offset
692 */
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100693 top_of_range = ((DEVOFFSET(dev) / blocklen) +
694 ENVSECTORS (dev)) * blocklen;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200695
696 /* Limit to one block for the first read */
697 if (readlen > blocklen - block_seek)
698 readlen = blocklen - block_seek;
699 } else {
700 blocklen = 0;
701 top_of_range = offset + count;
702 }
703
704 /* This only runs once on NOR flash */
705 while (processed < count) {
706 rc = flash_bad_block (fd, mtd_type, &blockstart);
707 if (rc < 0) /* block test failed */
708 return -1;
709
710 if (blockstart + block_seek + readlen > top_of_range) {
711 /* End of range is reached */
712 fprintf (stderr,
713 "Too few good blocks within range\n");
714 return -1;
715 }
716
717 if (rc) { /* block is bad */
718 blockstart += blocklen;
719 continue;
720 }
721
722 /*
723 * If a block is bad, we retry in the next block at the same
724 * offset - see common/env_nand.c::writeenv()
725 */
726 lseek (fd, blockstart + block_seek, SEEK_SET);
727
728 rc = read (fd, buf + processed, readlen);
729 if (rc != readlen) {
730 fprintf (stderr, "Read error on %s: %s\n",
731 DEVNAME (dev), strerror (errno));
732 return -1;
733 }
734#ifdef DEBUG
735 fprintf (stderr, "Read 0x%x bytes at 0x%llx\n",
736 rc, blockstart + block_seek);
737#endif
738 processed += readlen;
739 readlen = min (blocklen, count - processed);
740 block_seek = 0;
741 blockstart += blocklen;
742 }
743
744 return processed;
745}
746
747/*
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100748 * Write count bytes at offset, but stay within ENVSECTORS (dev) sectors of
749 * DEVOFFSET (dev). Similar to the read case above, on NOR and dataflash we
750 * erase and write the whole data at once.
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200751 */
752static int flash_write_buf (int dev, int fd, void *buf, size_t count,
753 off_t offset, uint8_t mtd_type)
754{
755 void *data;
756 struct erase_info_user erase;
757 size_t blocklen; /* length of NAND block / NOR erase sector */
758 size_t erase_len; /* whole area that can be erased - may include
759 bad blocks */
760 size_t erasesize; /* erase / write length - one block on NAND,
761 whole area on NOR */
762 size_t processed = 0; /* progress counter */
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100763 size_t write_total; /* total size to actually write - excluding
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200764 bad blocks */
765 off_t erase_offset; /* offset to the first erase block (aligned)
766 below offset */
767 off_t block_seek; /* offset inside the erase block to the start
768 of the data */
769 off_t top_of_range; /* end of the last block we may use */
770 loff_t blockstart; /* running start of the current block -
771 MEMGETBADBLOCK needs 64 bits */
772 int rc;
773
774 blocklen = DEVESIZE (dev);
775
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100776 top_of_range = ((DEVOFFSET(dev) / blocklen) +
777 ENVSECTORS (dev)) * blocklen;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200778
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100779 erase_offset = (offset / blocklen) * blocklen;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200780
781 /* Maximum area we may use */
782 erase_len = top_of_range - erase_offset;
783
784 blockstart = erase_offset;
785 /* Offset inside a block */
786 block_seek = offset - erase_offset;
787
788 /*
789 * Data size we actually have to write: from the start of the block
790 * to the start of the data, then count bytes of data, and to the
791 * end of the block
792 */
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100793 write_total = ((block_seek + count + blocklen - 1) /
794 blocklen) * blocklen;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200795
796 /*
797 * Support data anywhere within erase sectors: read out the complete
798 * area to be erased, replace the environment image, write the whole
799 * block back again.
800 */
801 if (write_total > count) {
802 data = malloc (erase_len);
803 if (!data) {
804 fprintf (stderr,
Luka Perkov8603b692011-09-05 23:40:13 +0200805 "Cannot malloc %zu bytes: %s\n",
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200806 erase_len, strerror (errno));
807 return -1;
808 }
809
810 rc = flash_read_buf (dev, fd, data, write_total, erase_offset,
811 mtd_type);
812 if (write_total != rc)
813 return -1;
814
815 /* Overwrite the old environment */
816 memcpy (data + block_seek, buf, count);
817 } else {
818 /*
819 * We get here, iff offset is block-aligned and count is a
820 * multiple of blocklen - see write_total calculation above
821 */
822 data = buf;
823 }
824
825 if (mtd_type == MTD_NANDFLASH) {
826 /*
827 * NAND: calculate which blocks we are writing. We have
828 * to write one block at a time to skip bad blocks.
829 */
830 erasesize = blocklen;
831 } else {
832 erasesize = erase_len;
833 }
834
835 erase.length = erasesize;
836
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100837 /* This only runs once on NOR flash and SPI-dataflash */
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200838 while (processed < write_total) {
839 rc = flash_bad_block (fd, mtd_type, &blockstart);
840 if (rc < 0) /* block test failed */
841 return rc;
842
843 if (blockstart + erasesize > top_of_range) {
844 fprintf (stderr, "End of range reached, aborting\n");
845 return -1;
846 }
847
848 if (rc) { /* block is bad */
849 blockstart += blocklen;
850 continue;
851 }
852
853 erase.start = blockstart;
854 ioctl (fd, MEMUNLOCK, &erase);
855
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100856 /* Dataflash does not need an explicit erase cycle */
857 if (mtd_type != MTD_DATAFLASH)
858 if (ioctl (fd, MEMERASE, &erase) != 0) {
859 fprintf (stderr, "MTD erase error on %s: %s\n",
860 DEVNAME (dev),
861 strerror (errno));
862 return -1;
863 }
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200864
865 if (lseek (fd, blockstart, SEEK_SET) == -1) {
866 fprintf (stderr,
867 "Seek error on %s: %s\n",
868 DEVNAME (dev), strerror (errno));
869 return -1;
870 }
871
872#ifdef DEBUG
873 printf ("Write 0x%x bytes at 0x%llx\n", erasesize, blockstart);
874#endif
875 if (write (fd, data + processed, erasesize) != erasesize) {
876 fprintf (stderr, "Write error on %s: %s\n",
877 DEVNAME (dev), strerror (errno));
878 return -1;
879 }
880
881 ioctl (fd, MEMLOCK, &erase);
882
883 processed += blocklen;
884 block_seek = 0;
885 blockstart += blocklen;
886 }
887
888 if (write_total > count)
889 free (data);
890
891 return processed;
892}
893
894/*
895 * Set obsolete flag at offset - NOR flash only
896 */
897static int flash_flag_obsolete (int dev, int fd, off_t offset)
898{
899 int rc;
Detlev Zundel0aef7bc2010-07-30 11:22:15 +0200900 struct erase_info_user erase;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200901
Detlev Zundel0aef7bc2010-07-30 11:22:15 +0200902 erase.start = DEVOFFSET (dev);
903 erase.length = DEVESIZE (dev);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200904 /* This relies on the fact, that obsolete_flag == 0 */
905 rc = lseek (fd, offset, SEEK_SET);
906 if (rc < 0) {
907 fprintf (stderr, "Cannot seek to set the flag on %s \n",
908 DEVNAME (dev));
909 return rc;
910 }
Detlev Zundel0aef7bc2010-07-30 11:22:15 +0200911 ioctl (fd, MEMUNLOCK, &erase);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200912 rc = write (fd, &obsolete_flag, sizeof (obsolete_flag));
Detlev Zundel0aef7bc2010-07-30 11:22:15 +0200913 ioctl (fd, MEMLOCK, &erase);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200914 if (rc < 0)
915 perror ("Could not set obsolete flag");
916
917 return rc;
918}
919
920static int flash_write (int fd_current, int fd_target, int dev_target)
921{
922 int rc;
923
924 switch (environment.flag_scheme) {
925 case FLAG_NONE:
926 break;
927 case FLAG_INCREMENTAL:
928 (*environment.flags)++;
929 break;
930 case FLAG_BOOLEAN:
931 *environment.flags = active_flag;
932 break;
933 default:
934 fprintf (stderr, "Unimplemented flash scheme %u \n",
935 environment.flag_scheme);
936 return -1;
937 }
938
939#ifdef DEBUG
940 printf ("Writing new environment at 0x%lx on %s\n",
941 DEVOFFSET (dev_target), DEVNAME (dev_target));
942#endif
943 rc = flash_write_buf (dev_target, fd_target, environment.image,
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +0200944 CONFIG_ENV_SIZE, DEVOFFSET (dev_target),
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200945 DEVTYPE(dev_target));
946 if (rc < 0)
947 return rc;
948
949 if (environment.flag_scheme == FLAG_BOOLEAN) {
950 /* Have to set obsolete flag */
951 off_t offset = DEVOFFSET (dev_current) +
952 offsetof (struct env_image_redundant, flags);
953#ifdef DEBUG
954 printf ("Setting obsolete flag in environment at 0x%lx on %s\n",
955 DEVOFFSET (dev_current), DEVNAME (dev_current));
956#endif
957 flash_flag_obsolete (dev_current, fd_current, offset);
958 }
959
960 return 0;
961}
962
963static int flash_read (int fd)
964{
965 struct mtd_info_user mtdinfo;
966 int rc;
967
968 rc = ioctl (fd, MEMGETINFO, &mtdinfo);
969 if (rc < 0) {
970 perror ("Cannot get MTD information");
971 return -1;
972 }
973
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +0100974 if (mtdinfo.type != MTD_NORFLASH &&
975 mtdinfo.type != MTD_NANDFLASH &&
976 mtdinfo.type != MTD_DATAFLASH) {
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200977 fprintf (stderr, "Unsupported flash type %u\n", mtdinfo.type);
978 return -1;
979 }
980
981 DEVTYPE(dev_current) = mtdinfo.type;
982
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +0200983 rc = flash_read_buf (dev_current, fd, environment.image, CONFIG_ENV_SIZE,
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200984 DEVOFFSET (dev_current), mtdinfo.type);
985
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +0200986 return (rc != CONFIG_ENV_SIZE) ? -1 : 0;
wdenk6aff3112002-12-17 01:51:00 +0000987}
988
989static int flash_io (int mode)
990{
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200991 int fd_current, fd_target, rc, dev_target;
wdenk6aff3112002-12-17 01:51:00 +0000992
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200993 /* dev_current: fd_current, erase_current */
994 fd_current = open (DEVNAME (dev_current), mode);
995 if (fd_current < 0) {
wdenk4a6fd342003-04-12 23:38:12 +0000996 fprintf (stderr,
Guennadi Liakhovetski56086922008-09-04 13:01:49 +0200997 "Can't open %s: %s\n",
998 DEVNAME (dev_current), strerror (errno));
999 return -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001000 }
wdenk6aff3112002-12-17 01:51:00 +00001001
1002 if (mode == O_RDWR) {
wdenkd0fb80c2003-01-11 09:48:40 +00001003 if (HaveRedundEnv) {
1004 /* switch to next partition for writing */
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001005 dev_target = !dev_current;
1006 /* dev_target: fd_target, erase_target */
1007 fd_target = open (DEVNAME (dev_target), mode);
1008 if (fd_target < 0) {
wdenk4a6fd342003-04-12 23:38:12 +00001009 fprintf (stderr,
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001010 "Can't open %s: %s\n",
1011 DEVNAME (dev_target),
1012 strerror (errno));
1013 rc = -1;
1014 goto exit;
wdenkd0fb80c2003-01-11 09:48:40 +00001015 }
1016 } else {
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001017 dev_target = dev_current;
1018 fd_target = fd_current;
wdenk6aff3112002-12-17 01:51:00 +00001019 }
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001020
1021 rc = flash_write (fd_current, fd_target, dev_target);
wdenk6aff3112002-12-17 01:51:00 +00001022
wdenkd0fb80c2003-01-11 09:48:40 +00001023 if (HaveRedundEnv) {
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001024 if (close (fd_target)) {
wdenkd0fb80c2003-01-11 09:48:40 +00001025 fprintf (stderr,
wdenk4a6fd342003-04-12 23:38:12 +00001026 "I/O error on %s: %s\n",
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001027 DEVNAME (dev_target),
wdenk4a6fd342003-04-12 23:38:12 +00001028 strerror (errno));
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001029 rc = -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001030 }
wdenk6aff3112002-12-17 01:51:00 +00001031 }
wdenk6aff3112002-12-17 01:51:00 +00001032 } else {
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001033 rc = flash_read (fd_current);
wdenk6aff3112002-12-17 01:51:00 +00001034 }
1035
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001036exit:
1037 if (close (fd_current)) {
wdenk6aff3112002-12-17 01:51:00 +00001038 fprintf (stderr,
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001039 "I/O error on %s: %s\n",
1040 DEVNAME (dev_current), strerror (errno));
1041 return -1;
wdenk6aff3112002-12-17 01:51:00 +00001042 }
1043
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001044 return rc;
wdenk6aff3112002-12-17 01:51:00 +00001045}
1046
1047/*
1048 * s1 is either a simple 'name', or a 'name=value' pair.
1049 * s2 is a 'name=value' pair.
1050 * If the names match, return the value of s2, else NULL.
1051 */
1052
Markus Klotzbücher6de66b32007-11-27 10:23:20 +01001053static char *envmatch (char * s1, char * s2)
wdenk6aff3112002-12-17 01:51:00 +00001054{
1055
1056 while (*s1 == *s2++)
1057 if (*s1++ == '=')
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001058 return s2;
wdenk4a6fd342003-04-12 23:38:12 +00001059 if (*s1 == '\0' && *(s2 - 1) == '=')
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001060 return s2;
1061 return NULL;
wdenk6aff3112002-12-17 01:51:00 +00001062}
1063
1064/*
1065 * Prevent confusion if running from erased flash memory
1066 */
Stefano Babicbd7b26f2010-05-24 12:08:16 +02001067int fw_env_open(void)
wdenk6aff3112002-12-17 01:51:00 +00001068{
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001069 int crc0, crc0_ok;
Jon Povey735eb0f2011-03-11 14:10:56 +09001070 unsigned char flag0;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001071 void *addr0;
wdenkd0fb80c2003-01-11 09:48:40 +00001072
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001073 int crc1, crc1_ok;
Jon Povey735eb0f2011-03-11 14:10:56 +09001074 unsigned char flag1;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001075 void *addr1;
1076
1077 struct env_image_single *single;
1078 struct env_image_redundant *redundant;
wdenk6aff3112002-12-17 01:51:00 +00001079
wdenk4a6fd342003-04-12 23:38:12 +00001080 if (parse_config ()) /* should fill envdevices */
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001081 return -1;
wdenk4a6fd342003-04-12 23:38:12 +00001082
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +02001083 addr0 = calloc (1, CONFIG_ENV_SIZE);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001084 if (addr0 == NULL) {
wdenk4a6fd342003-04-12 23:38:12 +00001085 fprintf (stderr,
1086 "Not enough memory for environment (%ld bytes)\n",
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +02001087 CONFIG_ENV_SIZE);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001088 return -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001089 }
wdenk4a6fd342003-04-12 23:38:12 +00001090
wdenkd0fb80c2003-01-11 09:48:40 +00001091 /* read environment from FLASH to local buffer */
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001092 environment.image = addr0;
1093
1094 if (HaveRedundEnv) {
1095 redundant = addr0;
1096 environment.crc = &redundant->crc;
1097 environment.flags = &redundant->flags;
1098 environment.data = redundant->data;
1099 } else {
1100 single = addr0;
1101 environment.crc = &single->crc;
1102 environment.flags = NULL;
1103 environment.data = single->data;
wdenkd0fb80c2003-01-11 09:48:40 +00001104 }
wdenk4a6fd342003-04-12 23:38:12 +00001105
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001106 dev_current = 0;
1107 if (flash_io (O_RDONLY))
1108 return -1;
1109
1110 crc0 = crc32 (0, (uint8_t *) environment.data, ENV_SIZE);
1111 crc0_ok = (crc0 == *environment.crc);
wdenkd0fb80c2003-01-11 09:48:40 +00001112 if (!HaveRedundEnv) {
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001113 if (!crc0_ok) {
wdenk4a6fd342003-04-12 23:38:12 +00001114 fprintf (stderr,
1115 "Warning: Bad CRC, using default environment\n");
Wolfgang Denkf07217c2006-03-12 23:27:46 +01001116 memcpy(environment.data, default_environment, sizeof default_environment);
wdenk6aff3112002-12-17 01:51:00 +00001117 }
wdenkd0fb80c2003-01-11 09:48:40 +00001118 } else {
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001119 flag0 = *environment.flags;
wdenk4a6fd342003-04-12 23:38:12 +00001120
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001121 dev_current = 1;
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +02001122 addr1 = calloc (1, CONFIG_ENV_SIZE);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001123 if (addr1 == NULL) {
wdenk4a6fd342003-04-12 23:38:12 +00001124 fprintf (stderr,
1125 "Not enough memory for environment (%ld bytes)\n",
Jean-Christophe PLAGNIOL-VILLARD0e8d1582008-09-10 22:48:06 +02001126 CONFIG_ENV_SIZE);
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001127 return -1;
wdenk4a6fd342003-04-12 23:38:12 +00001128 }
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001129 redundant = addr1;
wdenk4a6fd342003-04-12 23:38:12 +00001130
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001131 /*
1132 * have to set environment.image for flash_read(), careful -
1133 * other pointers in environment still point inside addr0
1134 */
1135 environment.image = addr1;
1136 if (flash_io (O_RDONLY))
1137 return -1;
1138
1139 /* Check flag scheme compatibility */
1140 if (DEVTYPE(dev_current) == MTD_NORFLASH &&
1141 DEVTYPE(!dev_current) == MTD_NORFLASH) {
1142 environment.flag_scheme = FLAG_BOOLEAN;
1143 } else if (DEVTYPE(dev_current) == MTD_NANDFLASH &&
1144 DEVTYPE(!dev_current) == MTD_NANDFLASH) {
1145 environment.flag_scheme = FLAG_INCREMENTAL;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001146 } else if (DEVTYPE(dev_current) == MTD_DATAFLASH &&
1147 DEVTYPE(!dev_current) == MTD_DATAFLASH) {
1148 environment.flag_scheme = FLAG_BOOLEAN;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001149 } else {
1150 fprintf (stderr, "Incompatible flash types!\n");
1151 return -1;
wdenk6aff3112002-12-17 01:51:00 +00001152 }
wdenk4a6fd342003-04-12 23:38:12 +00001153
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001154 crc1 = crc32 (0, (uint8_t *) redundant->data, ENV_SIZE);
1155 crc1_ok = (crc1 == redundant->crc);
1156 flag1 = redundant->flags;
wdenk4a6fd342003-04-12 23:38:12 +00001157
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001158 if (crc0_ok && !crc1_ok) {
1159 dev_current = 0;
1160 } else if (!crc0_ok && crc1_ok) {
1161 dev_current = 1;
1162 } else if (!crc0_ok && !crc1_ok) {
wdenk4a6fd342003-04-12 23:38:12 +00001163 fprintf (stderr,
1164 "Warning: Bad CRC, using default environment\n");
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001165 memcpy (environment.data, default_environment,
1166 sizeof default_environment);
1167 dev_current = 0;
1168 } else {
1169 switch (environment.flag_scheme) {
1170 case FLAG_BOOLEAN:
1171 if (flag0 == active_flag &&
1172 flag1 == obsolete_flag) {
1173 dev_current = 0;
1174 } else if (flag0 == obsolete_flag &&
1175 flag1 == active_flag) {
1176 dev_current = 1;
1177 } else if (flag0 == flag1) {
1178 dev_current = 0;
1179 } else if (flag0 == 0xFF) {
1180 dev_current = 0;
1181 } else if (flag1 == 0xFF) {
1182 dev_current = 1;
1183 } else {
1184 dev_current = 0;
1185 }
1186 break;
1187 case FLAG_INCREMENTAL:
Jon Povey735eb0f2011-03-11 14:10:56 +09001188 if (flag0 == 255 && flag1 == 0)
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001189 dev_current = 1;
1190 else if ((flag1 == 255 && flag0 == 0) ||
Jon Povey735eb0f2011-03-11 14:10:56 +09001191 flag0 >= flag1)
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001192 dev_current = 0;
Jon Povey735eb0f2011-03-11 14:10:56 +09001193 else /* flag1 > flag0 */
1194 dev_current = 1;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001195 break;
1196 default:
1197 fprintf (stderr, "Unknown flag scheme %u \n",
1198 environment.flag_scheme);
1199 return -1;
1200 }
1201 }
1202
1203 /*
1204 * If we are reading, we don't need the flag and the CRC any
1205 * more, if we are writing, we will re-calculate CRC and update
1206 * flags before writing out
1207 */
1208 if (dev_current) {
1209 environment.image = addr1;
1210 environment.crc = &redundant->crc;
1211 environment.flags = &redundant->flags;
1212 environment.data = redundant->data;
1213 free (addr0);
1214 } else {
1215 environment.image = addr0;
1216 /* Other pointers are already set */
wdenk4a6fd342003-04-12 23:38:12 +00001217 free (addr1);
wdenk6aff3112002-12-17 01:51:00 +00001218 }
wdenk6aff3112002-12-17 01:51:00 +00001219 }
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001220 return 0;
wdenk6aff3112002-12-17 01:51:00 +00001221}
1222
1223
wdenk4a6fd342003-04-12 23:38:12 +00001224static int parse_config ()
wdenk6aff3112002-12-17 01:51:00 +00001225{
1226 struct stat st;
1227
wdenkd0fb80c2003-01-11 09:48:40 +00001228#if defined(CONFIG_FILE)
1229 /* Fills in DEVNAME(), ENVSIZE(), DEVESIZE(). Or don't. */
wdenk4a6fd342003-04-12 23:38:12 +00001230 if (get_config (CONFIG_FILE)) {
wdenkd0fb80c2003-01-11 09:48:40 +00001231 fprintf (stderr,
wdenk4a6fd342003-04-12 23:38:12 +00001232 "Cannot parse config file: %s\n", strerror (errno));
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001233 return -1;
wdenk6aff3112002-12-17 01:51:00 +00001234 }
wdenkd0fb80c2003-01-11 09:48:40 +00001235#else
wdenk4a6fd342003-04-12 23:38:12 +00001236 strcpy (DEVNAME (0), DEVICE1_NAME);
1237 DEVOFFSET (0) = DEVICE1_OFFSET;
1238 ENVSIZE (0) = ENV1_SIZE;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001239 /* Default values are: erase-size=env-size, #sectors=1 */
1240 DEVESIZE (0) = ENVSIZE (0);
1241 ENVSECTORS (0) = 1;
1242#ifdef DEVICE1_ESIZE
wdenk4a6fd342003-04-12 23:38:12 +00001243 DEVESIZE (0) = DEVICE1_ESIZE;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001244#endif
1245#ifdef DEVICE1_ENVSECTORS
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001246 ENVSECTORS (0) = DEVICE1_ENVSECTORS;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001247#endif
1248
wdenk6aff3112002-12-17 01:51:00 +00001249#ifdef HAVE_REDUND
wdenk4a6fd342003-04-12 23:38:12 +00001250 strcpy (DEVNAME (1), DEVICE2_NAME);
1251 DEVOFFSET (1) = DEVICE2_OFFSET;
1252 ENVSIZE (1) = ENV2_SIZE;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001253 /* Default values are: erase-size=env-size, #sectors=1 */
1254 DEVESIZE (1) = ENVSIZE (1);
1255 ENVSECTORS (1) = 1;
1256#ifdef DEVICE2_ESIZE
wdenk4a6fd342003-04-12 23:38:12 +00001257 DEVESIZE (1) = DEVICE2_ESIZE;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001258#endif
1259#ifdef DEVICE2_ENVSECTORS
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001260 ENVSECTORS (1) = DEVICE2_ENVSECTORS;
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001261#endif
wdenkd0fb80c2003-01-11 09:48:40 +00001262 HaveRedundEnv = 1;
wdenk6aff3112002-12-17 01:51:00 +00001263#endif
wdenkd0fb80c2003-01-11 09:48:40 +00001264#endif
wdenk4a6fd342003-04-12 23:38:12 +00001265 if (stat (DEVNAME (0), &st)) {
1266 fprintf (stderr,
1267 "Cannot access MTD device %s: %s\n",
1268 DEVNAME (0), strerror (errno));
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001269 return -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001270 }
wdenk4a6fd342003-04-12 23:38:12 +00001271
1272 if (HaveRedundEnv && stat (DEVNAME (1), &st)) {
1273 fprintf (stderr,
1274 "Cannot access MTD device %s: %s\n",
Wolfgang Denke2146b62005-09-25 01:20:38 +02001275 DEVNAME (1), strerror (errno));
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001276 return -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001277 }
wdenk6aff3112002-12-17 01:51:00 +00001278 return 0;
1279}
wdenkd0fb80c2003-01-11 09:48:40 +00001280
1281#if defined(CONFIG_FILE)
1282static int get_config (char *fname)
1283{
1284 FILE *fp;
1285 int i = 0;
1286 int rc;
1287 char dump[128];
1288
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001289 fp = fopen (fname, "r");
1290 if (fp == NULL)
1291 return -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001292
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001293 while (i < 2 && fgets (dump, sizeof (dump), fp)) {
wdenkd0fb80c2003-01-11 09:48:40 +00001294 /* Skip incomplete conversions and comment strings */
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001295 if (dump[0] == '#')
wdenkd0fb80c2003-01-11 09:48:40 +00001296 continue;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001297
1298 rc = sscanf (dump, "%s %lx %lx %lx %lx",
1299 DEVNAME (i),
1300 &DEVOFFSET (i),
1301 &ENVSIZE (i),
1302 &DEVESIZE (i),
1303 &ENVSECTORS (i));
1304
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001305 if (rc < 3)
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001306 continue;
1307
Remy Bohmer9eeaa8e2011-02-12 19:06:26 +01001308 if (rc < 4)
1309 /* Assume the erase size is the same as the env-size */
1310 DEVESIZE(i) = ENVSIZE(i);
1311
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001312 if (rc < 5)
1313 /* Default - 1 sector */
1314 ENVSECTORS (i) = 1;
wdenkd0fb80c2003-01-11 09:48:40 +00001315
1316 i++;
1317 }
wdenk4a6fd342003-04-12 23:38:12 +00001318 fclose (fp);
1319
wdenkd0fb80c2003-01-11 09:48:40 +00001320 HaveRedundEnv = i - 1;
wdenk4a6fd342003-04-12 23:38:12 +00001321 if (!i) { /* No valid entries found */
wdenkd0fb80c2003-01-11 09:48:40 +00001322 errno = EINVAL;
Guennadi Liakhovetski56086922008-09-04 13:01:49 +02001323 return -1;
wdenkd0fb80c2003-01-11 09:48:40 +00001324 } else
1325 return 0;
1326}
1327#endif