Additional fix to readline_into_buffer() with CONFIG_CMDLINE_EDITING before relocating
[platform/kernel/u-boot.git] / common / env_flash.c
1 /*
2  * (C) Copyright 2000-2002
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * (C) Copyright 2001 Sysgo Real-Time Solutions, GmbH <www.elinos.com>
6  * Andreas Heppel <aheppel@sysgo.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 /* #define DEBUG */
28
29 #include <common.h>
30
31 #if defined(CFG_ENV_IS_IN_FLASH) /* Environment is in Flash */
32
33 #include <command.h>
34 #include <environment.h>
35 #include <linux/stddef.h>
36 #include <malloc.h>
37
38 DECLARE_GLOBAL_DATA_PTR;
39
40 #if defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_FLASH)
41 #define CMD_SAVEENV
42 #elif defined(CFG_ENV_ADDR_REDUND)
43 #error Cannot use CFG_ENV_ADDR_REDUND without CONFIG_CMD_ENV & CONFIG_CMD_FLASH
44 #endif
45
46 #if defined(CFG_ENV_SIZE_REDUND) && (CFG_ENV_SIZE_REDUND < CFG_ENV_SIZE)
47 #error CFG_ENV_SIZE_REDUND should not be less then CFG_ENV_SIZE
48 #endif
49
50 #ifdef CONFIG_INFERNO
51 # ifdef CFG_ENV_ADDR_REDUND
52 #error CFG_ENV_ADDR_REDUND is not implemented for CONFIG_INFERNO
53 # endif
54 #endif
55
56 char * env_name_spec = "Flash";
57
58 #ifdef ENV_IS_EMBEDDED
59
60 extern uchar environment[];
61 env_t *env_ptr = (env_t *)(&environment[0]);
62
63 #ifdef CMD_SAVEENV
64 /* static env_t *flash_addr = (env_t *)(&environment[0]);-broken on ARM-wd-*/
65 static env_t *flash_addr = (env_t *)CFG_ENV_ADDR;
66 #endif
67
68 #else /* ! ENV_IS_EMBEDDED */
69
70 env_t *env_ptr = (env_t *)CFG_ENV_ADDR;
71 #ifdef CMD_SAVEENV
72 static env_t *flash_addr = (env_t *)CFG_ENV_ADDR;
73 #endif
74
75 #endif /* ENV_IS_EMBEDDED */
76
77 #ifdef CFG_ENV_ADDR_REDUND
78 static env_t *flash_addr_new = (env_t *)CFG_ENV_ADDR_REDUND;
79
80 /* CFG_ENV_ADDR is supposed to be on sector boundary */
81 static ulong end_addr = CFG_ENV_ADDR + CFG_ENV_SECT_SIZE - 1;
82 static ulong end_addr_new = CFG_ENV_ADDR_REDUND + CFG_ENV_SECT_SIZE - 1;
83
84 #define ACTIVE_FLAG   1
85 #define OBSOLETE_FLAG 0
86 #endif /* CFG_ENV_ADDR_REDUND */
87
88 extern uchar default_environment[];
89 extern int default_environment_size;
90
91
92 uchar env_get_char_spec (int index)
93 {
94         return ( *((uchar *)(gd->env_addr + index)) );
95 }
96
97 #ifdef CFG_ENV_ADDR_REDUND
98
99 int  env_init(void)
100 {
101         int crc1_ok = 0, crc2_ok = 0;
102
103         uchar flag1 = flash_addr->flags;
104         uchar flag2 = flash_addr_new->flags;
105
106         ulong addr_default = (ulong)&default_environment[0];
107         ulong addr1 = (ulong)&(flash_addr->data);
108         ulong addr2 = (ulong)&(flash_addr_new->data);
109
110         crc1_ok = (crc32(0, flash_addr->data, ENV_SIZE) == flash_addr->crc);
111         crc2_ok = (crc32(0, flash_addr_new->data, ENV_SIZE) == flash_addr_new->crc);
112
113         if (crc1_ok && ! crc2_ok) {
114                 gd->env_addr  = addr1;
115                 gd->env_valid = 1;
116         } else if (! crc1_ok && crc2_ok) {
117                 gd->env_addr  = addr2;
118                 gd->env_valid = 1;
119         } else if (! crc1_ok && ! crc2_ok) {
120                 gd->env_addr  = addr_default;
121                 gd->env_valid = 0;
122         } else if (flag1 == ACTIVE_FLAG && flag2 == OBSOLETE_FLAG) {
123                 gd->env_addr  = addr1;
124                 gd->env_valid = 1;
125         } else if (flag1 == OBSOLETE_FLAG && flag2 == ACTIVE_FLAG) {
126                 gd->env_addr  = addr2;
127                 gd->env_valid = 1;
128         } else if (flag1 == flag2) {
129                 gd->env_addr  = addr1;
130                 gd->env_valid = 2;
131         } else if (flag1 == 0xFF) {
132                 gd->env_addr  = addr1;
133                 gd->env_valid = 2;
134         } else if (flag2 == 0xFF) {
135                 gd->env_addr  = addr2;
136                 gd->env_valid = 2;
137         }
138
139         return (0);
140 }
141
142 #ifdef CMD_SAVEENV
143 int saveenv(void)
144 {
145         char *saved_data = NULL;
146         int rc = 1;
147         char flag = OBSOLETE_FLAG, new_flag = ACTIVE_FLAG;
148 #if CFG_ENV_SECT_SIZE > CFG_ENV_SIZE
149         ulong up_data = 0;
150 #endif
151
152         debug ("Protect off %08lX ... %08lX\n",
153                 (ulong)flash_addr, end_addr);
154
155         if (flash_sect_protect (0, (ulong)flash_addr, end_addr)) {
156                 goto Done;
157         }
158
159         debug ("Protect off %08lX ... %08lX\n",
160                 (ulong)flash_addr_new, end_addr_new);
161
162         if (flash_sect_protect (0, (ulong)flash_addr_new, end_addr_new)) {
163                 goto Done;
164         }
165
166 #if CFG_ENV_SECT_SIZE > CFG_ENV_SIZE
167         up_data = (end_addr_new + 1 - ((long)flash_addr_new + CFG_ENV_SIZE));
168         debug ("Data to save 0x%x\n", up_data);
169         if (up_data) {
170                 if ((saved_data = malloc(up_data)) == NULL) {
171                         printf("Unable to save the rest of sector (%ld)\n",
172                                 up_data);
173                         goto Done;
174                 }
175                 memcpy(saved_data,
176                         (void *)((long)flash_addr_new + CFG_ENV_SIZE), up_data);
177                 debug ("Data (start 0x%x, len 0x%x) saved at 0x%x\n",
178                            (long)flash_addr_new + CFG_ENV_SIZE,
179                                 up_data, saved_data);
180         }
181 #endif
182         puts ("Erasing Flash...");
183         debug (" %08lX ... %08lX ...",
184                 (ulong)flash_addr_new, end_addr_new);
185
186         if (flash_sect_erase ((ulong)flash_addr_new, end_addr_new)) {
187                 goto Done;
188         }
189
190         puts ("Writing to Flash... ");
191         debug (" %08lX ... %08lX ...",
192                 (ulong)&(flash_addr_new->data),
193                 sizeof(env_ptr->data)+(ulong)&(flash_addr_new->data));
194         if ((rc = flash_write((char *)env_ptr->data,
195                         (ulong)&(flash_addr_new->data),
196                         sizeof(env_ptr->data))) ||
197             (rc = flash_write((char *)&(env_ptr->crc),
198                         (ulong)&(flash_addr_new->crc),
199                         sizeof(env_ptr->crc))) ||
200             (rc = flash_write(&flag,
201                         (ulong)&(flash_addr->flags),
202                         sizeof(flash_addr->flags))) ||
203             (rc = flash_write(&new_flag,
204                         (ulong)&(flash_addr_new->flags),
205                         sizeof(flash_addr_new->flags))))
206         {
207                 flash_perror (rc);
208                 goto Done;
209         }
210         puts ("done\n");
211
212 #if CFG_ENV_SECT_SIZE > CFG_ENV_SIZE
213         if (up_data) { /* restore the rest of sector */
214                 debug ("Restoring the rest of data to 0x%x len 0x%x\n",
215                            (long)flash_addr_new + CFG_ENV_SIZE, up_data);
216                 if (flash_write(saved_data,
217                                 (long)flash_addr_new + CFG_ENV_SIZE,
218                                 up_data)) {
219                         flash_perror(rc);
220                         goto Done;
221                 }
222         }
223 #endif
224         {
225                 env_t * etmp = flash_addr;
226                 ulong ltmp = end_addr;
227
228                 flash_addr = flash_addr_new;
229                 flash_addr_new = etmp;
230
231                 end_addr = end_addr_new;
232                 end_addr_new = ltmp;
233         }
234
235         rc = 0;
236 Done:
237
238         if (saved_data)
239                 free (saved_data);
240         /* try to re-protect */
241         (void) flash_sect_protect (1, (ulong)flash_addr, end_addr);
242         (void) flash_sect_protect (1, (ulong)flash_addr_new, end_addr_new);
243
244         return rc;
245 }
246 #endif /* CMD_SAVEENV */
247
248 #else /* ! CFG_ENV_ADDR_REDUND */
249
250 int  env_init(void)
251 {
252         if (crc32(0, env_ptr->data, ENV_SIZE) == env_ptr->crc) {
253                 gd->env_addr  = (ulong)&(env_ptr->data);
254                 gd->env_valid = 1;
255                 return(0);
256         }
257
258         gd->env_addr  = (ulong)&default_environment[0];
259         gd->env_valid = 0;
260         return (0);
261 }
262
263 #ifdef CMD_SAVEENV
264
265 int saveenv(void)
266 {
267         int     len, rc;
268         ulong   end_addr;
269         ulong   flash_sect_addr;
270 #if defined(CFG_ENV_SECT_SIZE) && (CFG_ENV_SECT_SIZE > CFG_ENV_SIZE)
271         ulong   flash_offset;
272         uchar   env_buffer[CFG_ENV_SECT_SIZE];
273 #else
274         uchar *env_buffer = (uchar *)env_ptr;
275 #endif  /* CFG_ENV_SECT_SIZE */
276         int rcode = 0;
277
278 #if defined(CFG_ENV_SECT_SIZE) && (CFG_ENV_SECT_SIZE > CFG_ENV_SIZE)
279
280         flash_offset    = ((ulong)flash_addr) & (CFG_ENV_SECT_SIZE-1);
281         flash_sect_addr = ((ulong)flash_addr) & ~(CFG_ENV_SECT_SIZE-1);
282
283         debug ( "copy old content: "
284                 "sect_addr: %08lX  env_addr: %08lX  offset: %08lX\n",
285                 flash_sect_addr, (ulong)flash_addr, flash_offset);
286
287         /* copy old contents to temporary buffer */
288         memcpy (env_buffer, (void *)flash_sect_addr, CFG_ENV_SECT_SIZE);
289
290         /* copy current environment to temporary buffer */
291         memcpy ((uchar *)((unsigned long)env_buffer + flash_offset),
292                 env_ptr,
293                 CFG_ENV_SIZE);
294
295         len      = CFG_ENV_SECT_SIZE;
296 #else
297         flash_sect_addr = (ulong)flash_addr;
298         len      = CFG_ENV_SIZE;
299 #endif  /* CFG_ENV_SECT_SIZE */
300
301 #ifndef CONFIG_INFERNO
302         end_addr = flash_sect_addr + len - 1;
303 #else
304         /* this is the last sector, and the size is hardcoded here */
305         /* otherwise we will get stack problems on loading 128 KB environment */
306         end_addr = flash_sect_addr + 0x20000 - 1;
307 #endif
308
309         debug ("Protect off %08lX ... %08lX\n",
310                 (ulong)flash_sect_addr, end_addr);
311
312         if (flash_sect_protect (0, flash_sect_addr, end_addr))
313                 return 1;
314
315         puts ("Erasing Flash...");
316         if (flash_sect_erase (flash_sect_addr, end_addr))
317                 return 1;
318
319         puts ("Writing to Flash... ");
320         rc = flash_write((char *)env_buffer, flash_sect_addr, len);
321         if (rc != 0) {
322                 flash_perror (rc);
323                 rcode = 1;
324         } else {
325                 puts ("done\n");
326         }
327
328         /* try to re-protect */
329         (void) flash_sect_protect (1, flash_sect_addr, end_addr);
330         return rcode;
331 }
332
333 #endif /* CMD_SAVEENV */
334
335 #endif /* CFG_ENV_ADDR_REDUND */
336
337 void env_relocate_spec (void)
338 {
339 #if !defined(ENV_IS_EMBEDDED) || defined(CFG_ENV_ADDR_REDUND)
340 #ifdef CFG_ENV_ADDR_REDUND
341         if (gd->env_addr != (ulong)&(flash_addr->data)) {
342                 env_t * etmp = flash_addr;
343                 ulong ltmp = end_addr;
344
345                 flash_addr = flash_addr_new;
346                 flash_addr_new = etmp;
347
348                 end_addr = end_addr_new;
349                 end_addr_new = ltmp;
350         }
351
352         if (flash_addr_new->flags != OBSOLETE_FLAG &&
353             crc32(0, flash_addr_new->data, ENV_SIZE) ==
354             flash_addr_new->crc) {
355                 char flag = OBSOLETE_FLAG;
356
357                 gd->env_valid = 2;
358                 flash_sect_protect (0, (ulong)flash_addr_new, end_addr_new);
359                 flash_write(&flag,
360                             (ulong)&(flash_addr_new->flags),
361                             sizeof(flash_addr_new->flags));
362                 flash_sect_protect (1, (ulong)flash_addr_new, end_addr_new);
363         }
364
365         if (flash_addr->flags != ACTIVE_FLAG &&
366             (flash_addr->flags & ACTIVE_FLAG) == ACTIVE_FLAG) {
367                 char flag = ACTIVE_FLAG;
368
369                 gd->env_valid = 2;
370                 flash_sect_protect (0, (ulong)flash_addr, end_addr);
371                 flash_write(&flag,
372                             (ulong)&(flash_addr->flags),
373                             sizeof(flash_addr->flags));
374                 flash_sect_protect (1, (ulong)flash_addr, end_addr);
375         }
376
377         if (gd->env_valid == 2)
378                 puts ("*** Warning - some problems detected "
379                       "reading environment; recovered successfully\n\n");
380 #endif /* CFG_ENV_ADDR_REDUND */
381 #ifdef CMD_SAVEENV
382         memcpy (env_ptr, (void*)flash_addr, CFG_ENV_SIZE);
383 #endif
384 #endif /* ! ENV_IS_EMBEDDED || CFG_ENV_ADDR_REDUND */
385 }
386
387 #endif /* CFG_ENV_IS_IN_FLASH */