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 };