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 void 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; 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 #else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */ 136 static void env_ubi_load(void) 137 { 138 ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE); 139 140 /* 141 * In case we have restarted u-boot there is a chance that buffer 142 * contains old environment (from the previous boot). 143 * If UBI volume is zero size, ubi_volume_read() doesn't modify the 144 * buffer. 145 * We need to clear buffer manually here, so the invalid CRC will 146 * cause setting default environment as expected. 147 */ 148 memset(buf, 0x0, CONFIG_ENV_SIZE); 149 150 if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) { 151 printf("\n** Cannot find mtd partition \"%s\"\n", 152 CONFIG_ENV_UBI_PART); 153 set_default_env(NULL); 154 return; 155 } 156 157 if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, buf, CONFIG_ENV_SIZE)) { 158 printf("\n** Unable to read env from %s:%s **\n", 159 CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME); 160 set_default_env(NULL); 161 return; 162 } 163 164 env_import(buf, 1); 165 } 166 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */ 167 168 U_BOOT_ENV_LOCATION(ubi) = { 169 .location = ENVL_UBI, 170 .load = env_ubi_load, 171 .save = env_save_ptr(env_ubi_save), 172 }; 173