blob: 08aac47df2b6a7c5e7c04f1d9232f6db62e43bac [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Joe Hershberger2b744332013-04-08 10:32:51 +00002/*
3 * (c) Copyright 2012 by National Instruments,
4 * Joe Hershberger <joe.hershberger@ni.com>
Joe Hershberger2b744332013-04-08 10:32:51 +00005 */
6
7#include <common.h>
8
9#include <command.h>
Simon Glass0ac7d722019-08-01 09:47:00 -060010#include <env.h>
Simon Glassf3998fd2019-08-02 09:44:25 -060011#include <env_internal.h>
Joe Hershberger2b744332013-04-08 10:32:51 +000012#include <errno.h>
13#include <malloc.h>
Simon Glasscf92e052015-09-02 17:24:58 -060014#include <memalign.h>
Joe Hershberger2b744332013-04-08 10:32:51 +000015#include <search.h>
16#include <ubi_uboot.h>
17#undef crc32
18
Hamish Guthrie985186d2019-05-15 15:15:55 +020019#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
Joe Hershberger2b744332013-04-08 10:32:51 +000028DECLARE_GLOBAL_DATA_PTR;
29
Joe Hershberger2b744332013-04-08 10:32:51 +000030#ifdef CONFIG_CMD_SAVEENV
Joe Hershberger785881f2013-04-08 10:32:52 +000031#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
Simon Glasse5bce242017-08-03 12:22:01 -060032static int env_ubi_save(void)
Joe Hershberger785881f2013-04-08 10:32:52 +000033{
34 ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
Marek Vasut7ce15262014-03-05 19:59:50 +010035 int ret;
Joe Hershberger785881f2013-04-08 10:32:52 +000036
Marek Vasut7ce15262014-03-05 19:59:50 +010037 ret = env_export(env_new);
38 if (ret)
39 return ret;
Joe Hershberger785881f2013-04-08 10:32:52 +000040
Hamish Guthrie985186d2019-05-15 15:15:55 +020041 if (ubi_part(CONFIG_ENV_UBI_PART, UBI_VID_OFFSET)) {
Joe Hershberger785881f2013-04-08 10:32:52 +000042 printf("\n** Cannot find mtd partition \"%s\"\n",
43 CONFIG_ENV_UBI_PART);
44 return 1;
45 }
46
Simon Glass203e94f2017-08-03 12:21:56 -060047 if (gd->env_valid == ENV_VALID) {
Joe Hershberger785881f2013-04-08 10:32:52 +000048 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
Simon Glass203e94f2017-08-03 12:21:56 -060069 gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
Joe Hershberger785881f2013-04-08 10:32:52 +000070
71 return 0;
72}
73#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
Simon Glasse5bce242017-08-03 12:22:01 -060074static int env_ubi_save(void)
Joe Hershberger2b744332013-04-08 10:32:51 +000075{
76 ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
Marek Vasut7ce15262014-03-05 19:59:50 +010077 int ret;
Joe Hershberger2b744332013-04-08 10:32:51 +000078
Marek Vasut7ce15262014-03-05 19:59:50 +010079 ret = env_export(env_new);
80 if (ret)
81 return ret;
Joe Hershberger2b744332013-04-08 10:32:51 +000082
Hamish Guthrie985186d2019-05-15 15:15:55 +020083 if (ubi_part(CONFIG_ENV_UBI_PART, UBI_VID_OFFSET)) {
Joe Hershberger2b744332013-04-08 10:32:51 +000084 printf("\n** Cannot find mtd partition \"%s\"\n",
85 CONFIG_ENV_UBI_PART);
86 return 1;
87 }
88
Joe Hershberger2b744332013-04-08 10:32:51 +000089 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}
Joe Hershberger785881f2013-04-08 10:32:52 +000099#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
Joe Hershberger2b744332013-04-08 10:32:51 +0000100#endif /* CONFIG_CMD_SAVEENV */
101
Joe Hershberger785881f2013-04-08 10:32:52 +0000102#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
Simon Glassc5951992017-08-03 12:22:17 -0600103static int env_ubi_load(void)
Joe Hershberger785881f2013-04-08 10:32:52 +0000104{
105 ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
106 ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
Simon Goldschmidt31f044b2018-01-31 14:47:11 +0100107 int read1_fail, read2_fail;
Fiach Antaw9d364af2017-01-25 18:53:12 +1000108 env_t *tmp_env1, *tmp_env2;
Joe Hershberger785881f2013-04-08 10:32:52 +0000109
Marcin Niestrojc1f51e02016-05-24 14:59:55 +0200110 /*
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
Joe Hershberger785881f2013-04-08 10:32:52 +0000121 tmp_env1 = (env_t *)env1_buf;
122 tmp_env2 = (env_t *)env2_buf;
123
Hamish Guthrie985186d2019-05-15 15:15:55 +0200124 if (ubi_part(CONFIG_ENV_UBI_PART, UBI_VID_OFFSET)) {
Joe Hershberger785881f2013-04-08 10:32:52 +0000125 printf("\n** Cannot find mtd partition \"%s\"\n",
126 CONFIG_ENV_UBI_PART);
Simon Glass0ac7d722019-08-01 09:47:00 -0600127 env_set_default(NULL, 0);
Simon Glassc5951992017-08-03 12:22:17 -0600128 return -EIO;
Joe Hershberger785881f2013-04-08 10:32:52 +0000129 }
130
Simon Goldschmidt31f044b2018-01-31 14:47:11 +0100131 read1_fail = ubi_volume_read(CONFIG_ENV_UBI_VOLUME, (void *)tmp_env1,
132 CONFIG_ENV_SIZE);
133 if (read1_fail)
Joe Hershberger785881f2013-04-08 10:32:52 +0000134 printf("\n** Unable to read env from %s:%s **\n",
135 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
Joe Hershberger785881f2013-04-08 10:32:52 +0000136
Simon Goldschmidt31f044b2018-01-31 14:47:11 +0100137 read2_fail = ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND,
138 (void *)tmp_env2, CONFIG_ENV_SIZE);
139 if (read2_fail)
Joe Hershberger785881f2013-04-08 10:32:52 +0000140 printf("\n** Unable to read redundant env from %s:%s **\n",
141 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME_REDUND);
Joe Hershberger785881f2013-04-08 10:32:52 +0000142
Simon Goldschmidt31f044b2018-01-31 14:47:11 +0100143 return env_import_redund((char *)tmp_env1, read1_fail, (char *)tmp_env2,
144 read2_fail);
Joe Hershberger785881f2013-04-08 10:32:52 +0000145}
146#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
Simon Glassc5951992017-08-03 12:22:17 -0600147static int env_ubi_load(void)
Joe Hershberger2b744332013-04-08 10:32:51 +0000148{
149 ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
150
Marcin Niestrojc1f51e02016-05-24 14:59:55 +0200151 /*
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
Hamish Guthrie985186d2019-05-15 15:15:55 +0200161 if (ubi_part(CONFIG_ENV_UBI_PART, UBI_VID_OFFSET)) {
Joe Hershberger2b744332013-04-08 10:32:51 +0000162 printf("\n** Cannot find mtd partition \"%s\"\n",
163 CONFIG_ENV_UBI_PART);
Simon Glass0ac7d722019-08-01 09:47:00 -0600164 env_set_default(NULL, 0);
Simon Glassc5951992017-08-03 12:22:17 -0600165 return -EIO;
Joe Hershberger2b744332013-04-08 10:32:51 +0000166 }
167
Kevin Smitha7c06cd2015-10-23 17:51:47 +0000168 if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, buf, CONFIG_ENV_SIZE)) {
Joe Hershberger2b744332013-04-08 10:32:51 +0000169 printf("\n** Unable to read env from %s:%s **\n",
170 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
Simon Glass0ac7d722019-08-01 09:47:00 -0600171 env_set_default(NULL, 0);
Simon Glassc5951992017-08-03 12:22:17 -0600172 return -EIO;
Joe Hershberger2b744332013-04-08 10:32:51 +0000173 }
174
Simon Goldschmidt2166ebf2018-01-31 14:47:12 +0100175 return env_import(buf, 1);
Joe Hershberger2b744332013-04-08 10:32:51 +0000176}
Joe Hershberger785881f2013-04-08 10:32:52 +0000177#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
Simon Glass4415f1d2017-08-03 12:21:58 -0600178
179U_BOOT_ENV_LOCATION(ubi) = {
180 .location = ENVL_UBI,
Marek Vasut344ca792018-08-21 15:53:33 +0200181 ENV_NAME("UBI")
Simon Glasse5bce242017-08-03 12:22:01 -0600182 .load = env_ubi_load,
183 .save = env_save_ptr(env_ubi_save),
Simon Glass4415f1d2017-08-03 12:21:58 -0600184};