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