Merge git://git.denx.de/u-boot-spi
[platform/kernel/u-boot.git] / env / ubi.c
1 /*
2  * (c) Copyright 2012 by National Instruments,
3  *        Joe Hershberger <joe.hershberger@ni.com>
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
7
8 #include <common.h>
9
10 #include <command.h>
11 #include <environment.h>
12 #include <errno.h>
13 #include <malloc.h>
14 #include <memalign.h>
15 #include <search.h>
16 #include <ubi_uboot.h>
17 #undef crc32
18
19 DECLARE_GLOBAL_DATA_PTR;
20
21 #ifdef CONFIG_CMD_SAVEENV
22 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
23 static int env_ubi_save(void)
24 {
25         ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
26         int ret;
27
28         ret = env_export(env_new);
29         if (ret)
30                 return ret;
31
32         if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
33                 printf("\n** Cannot find mtd partition \"%s\"\n",
34                        CONFIG_ENV_UBI_PART);
35                 return 1;
36         }
37
38         if (gd->env_valid == ENV_VALID) {
39                 puts("Writing to redundant UBI... ");
40                 if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME_REDUND,
41                                      (void *)env_new, CONFIG_ENV_SIZE)) {
42                         printf("\n** Unable to write env to %s:%s **\n",
43                                CONFIG_ENV_UBI_PART,
44                                CONFIG_ENV_UBI_VOLUME_REDUND);
45                         return 1;
46                 }
47         } else {
48                 puts("Writing to UBI... ");
49                 if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME,
50                                      (void *)env_new, CONFIG_ENV_SIZE)) {
51                         printf("\n** Unable to write env to %s:%s **\n",
52                                CONFIG_ENV_UBI_PART,
53                                CONFIG_ENV_UBI_VOLUME);
54                         return 1;
55                 }
56         }
57
58         puts("done\n");
59
60         gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
61
62         return 0;
63 }
64 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
65 static int env_ubi_save(void)
66 {
67         ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
68         int ret;
69
70         ret = env_export(env_new);
71         if (ret)
72                 return ret;
73
74         if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
75                 printf("\n** Cannot find mtd partition \"%s\"\n",
76                        CONFIG_ENV_UBI_PART);
77                 return 1;
78         }
79
80         if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME, (void *)env_new,
81                              CONFIG_ENV_SIZE)) {
82                 printf("\n** Unable to write env to %s:%s **\n",
83                        CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
84                 return 1;
85         }
86
87         puts("done\n");
88         return 0;
89 }
90 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
91 #endif /* CONFIG_CMD_SAVEENV */
92
93 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
94 static int env_ubi_load(void)
95 {
96         ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
97         ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
98         env_t *tmp_env1, *tmp_env2;
99
100         /*
101          * In case we have restarted u-boot there is a chance that buffer
102          * contains old environment (from the previous boot).
103          * If UBI volume is zero size, ubi_volume_read() doesn't modify the
104          * buffer.
105          * We need to clear buffer manually here, so the invalid CRC will
106          * cause setting default environment as expected.
107          */
108         memset(env1_buf, 0x0, CONFIG_ENV_SIZE);
109         memset(env2_buf, 0x0, CONFIG_ENV_SIZE);
110
111         tmp_env1 = (env_t *)env1_buf;
112         tmp_env2 = (env_t *)env2_buf;
113
114         if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
115                 printf("\n** Cannot find mtd partition \"%s\"\n",
116                        CONFIG_ENV_UBI_PART);
117                 set_default_env(NULL);
118                 return -EIO;
119         }
120
121         if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, (void *)tmp_env1,
122                             CONFIG_ENV_SIZE)) {
123                 printf("\n** Unable to read env from %s:%s **\n",
124                        CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
125         }
126
127         if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND, (void *)tmp_env2,
128                             CONFIG_ENV_SIZE)) {
129                 printf("\n** Unable to read redundant env from %s:%s **\n",
130                        CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME_REDUND);
131         }
132
133         env_import_redund((char *)tmp_env1, (char *)tmp_env2);
134
135         return 0;
136 }
137 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
138 static int env_ubi_load(void)
139 {
140         ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
141
142         /*
143          * In case we have restarted u-boot there is a chance that buffer
144          * contains old environment (from the previous boot).
145          * If UBI volume is zero size, ubi_volume_read() doesn't modify the
146          * buffer.
147          * We need to clear buffer manually here, so the invalid CRC will
148          * cause setting default environment as expected.
149          */
150         memset(buf, 0x0, CONFIG_ENV_SIZE);
151
152         if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
153                 printf("\n** Cannot find mtd partition \"%s\"\n",
154                        CONFIG_ENV_UBI_PART);
155                 set_default_env(NULL);
156                 return -EIO;
157         }
158
159         if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, buf, CONFIG_ENV_SIZE)) {
160                 printf("\n** Unable to read env from %s:%s **\n",
161                        CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
162                 set_default_env(NULL);
163                 return -EIO;
164         }
165
166         env_import(buf, 1);
167
168         return 0;
169 }
170 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
171
172 U_BOOT_ENV_LOCATION(ubi) = {
173         .location       = ENVL_UBI,
174         .load           = env_ubi_load,
175         .save           = env_save_ptr(env_ubi_save),
176 };