2 * (c) Copyright 2012 by National Instruments,
3 * Joe Hershberger <joe.hershberger@ni.com>
5 * SPDX-License-Identifier: GPL-2.0+
11 #include <environment.h>
16 #include <ubi_uboot.h>
21 DECLARE_GLOBAL_DATA_PTR;
23 #ifdef CONFIG_CMD_SAVEENV
24 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
25 static int env_ubi_save(void)
27 ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
30 ret = env_export(env_new);
34 if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
35 printf("\n** Cannot find mtd partition \"%s\"\n",
40 if (gd->env_valid == ENV_VALID) {
41 puts("Writing to redundant UBI... ");
42 if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME_REDUND,
43 (void *)env_new, CONFIG_ENV_SIZE)) {
44 printf("\n** Unable to write env to %s:%s **\n",
46 CONFIG_ENV_UBI_VOLUME_REDUND);
50 puts("Writing to UBI... ");
51 if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME,
52 (void *)env_new, CONFIG_ENV_SIZE)) {
53 printf("\n** Unable to write env to %s:%s **\n",
55 CONFIG_ENV_UBI_VOLUME);
62 gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
66 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
67 static int env_ubi_save(void)
69 ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
72 ret = env_export(env_new);
76 if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
77 printf("\n** Cannot find mtd partition \"%s\"\n",
82 if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME, (void *)env_new,
84 printf("\n** Unable to write env to %s:%s **\n",
85 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
92 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
93 #endif /* CONFIG_CMD_SAVEENV */
95 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
96 static void env_ubi_load(void)
98 ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
99 ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
100 env_t *tmp_env1, *tmp_env2;
103 * In case we have restarted u-boot there is a chance that buffer
104 * contains old environment (from the previous boot).
105 * If UBI volume is zero size, ubi_volume_read() doesn't modify the
107 * We need to clear buffer manually here, so the invalid CRC will
108 * cause setting default environment as expected.
110 memset(env1_buf, 0x0, CONFIG_ENV_SIZE);
111 memset(env2_buf, 0x0, CONFIG_ENV_SIZE);
113 tmp_env1 = (env_t *)env1_buf;
114 tmp_env2 = (env_t *)env2_buf;
116 if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
117 printf("\n** Cannot find mtd partition \"%s\"\n",
118 CONFIG_ENV_UBI_PART);
119 set_default_env(NULL);
123 if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, (void *)tmp_env1,
125 printf("\n** Unable to read env from %s:%s **\n",
126 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
129 if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND, (void *)tmp_env2,
131 printf("\n** Unable to read redundant env from %s:%s **\n",
132 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME_REDUND);
135 env_import_redund((char *)tmp_env1, (char *)tmp_env2);
137 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
138 static void env_ubi_load(void)
140 ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
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
147 * We need to clear buffer manually here, so the invalid CRC will
148 * cause setting default environment as expected.
150 memset(buf, 0x0, CONFIG_ENV_SIZE);
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);
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);
168 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
170 U_BOOT_ENV_LOCATION(ubi) = {
171 .location = ENVL_UBI,
172 .load = env_ubi_load,
173 .save = env_save_ptr(env_ubi_save),