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>
19 DECLARE_GLOBAL_DATA_PTR
;
21 #ifdef CONFIG_CMD_SAVEENV
22 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
23 static int env_ubi_save(void)
25 ALLOC_CACHE_ALIGN_BUFFER(env_t
, env_new
, 1);
28 ret
= env_export(env_new
);
32 if (ubi_part(CONFIG_ENV_UBI_PART
, NULL
)) {
33 printf("\n** Cannot find mtd partition \"%s\"\n",
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",
44 CONFIG_ENV_UBI_VOLUME_REDUND
);
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",
53 CONFIG_ENV_UBI_VOLUME
);
60 gd
->env_valid
= gd
->env_valid
== ENV_REDUND
? ENV_VALID
: ENV_REDUND
;
64 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
65 static int env_ubi_save(void)
67 ALLOC_CACHE_ALIGN_BUFFER(env_t
, env_new
, 1);
70 ret
= env_export(env_new
);
74 if (ubi_part(CONFIG_ENV_UBI_PART
, NULL
)) {
75 printf("\n** Cannot find mtd partition \"%s\"\n",
80 if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME
, (void *)env_new
,
82 printf("\n** Unable to write env to %s:%s **\n",
83 CONFIG_ENV_UBI_PART
, CONFIG_ENV_UBI_VOLUME
);
90 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
91 #endif /* CONFIG_CMD_SAVEENV */
93 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
94 static int env_ubi_load(void)
96 ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf
, CONFIG_ENV_SIZE
);
97 ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf
, CONFIG_ENV_SIZE
);
98 int read1_fail
, read2_fail
;
99 env_t
*tmp_env1
, *tmp_env2
;
102 * In case we have restarted u-boot there is a chance that buffer
103 * contains old environment (from the previous boot).
104 * If UBI volume is zero size, ubi_volume_read() doesn't modify the
106 * We need to clear buffer manually here, so the invalid CRC will
107 * cause setting default environment as expected.
109 memset(env1_buf
, 0x0, CONFIG_ENV_SIZE
);
110 memset(env2_buf
, 0x0, CONFIG_ENV_SIZE
);
112 tmp_env1
= (env_t
*)env1_buf
;
113 tmp_env2
= (env_t
*)env2_buf
;
115 if (ubi_part(CONFIG_ENV_UBI_PART
, NULL
)) {
116 printf("\n** Cannot find mtd partition \"%s\"\n",
117 CONFIG_ENV_UBI_PART
);
118 set_default_env(NULL
);
122 read1_fail
= 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
);
128 read2_fail
= ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND
,
129 (void *)tmp_env2
, CONFIG_ENV_SIZE
);
131 printf("\n** Unable to read redundant env from %s:%s **\n",
132 CONFIG_ENV_UBI_PART
, CONFIG_ENV_UBI_VOLUME_REDUND
);
134 return env_import_redund((char *)tmp_env1
, read1_fail
, (char *)tmp_env2
,
137 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
138 static int 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
);
166 return env_import(buf
, 1);
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
),