1 // SPDX-License-Identifier: GPL-2.0+ 2 /* 3 * (C) Copyright 2000-2010 4 * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 5 * 6 * (C) Copyright 2001 Sysgo Real-Time Solutions, GmbH <www.elinos.com> 7 * Andreas Heppel <aheppel@sysgo.de> 8 */ 9 10 #include <common.h> 11 #include <command.h> 12 #include <environment.h> 13 #include <linux/stddef.h> 14 #include <search.h> 15 #include <errno.h> 16 #include <malloc.h> 17 18 DECLARE_GLOBAL_DATA_PTR; 19 20 /************************************************************************ 21 * Default settings to be used when no valid environment is found 22 */ 23 #include <env_default.h> 24 25 struct hsearch_data env_htab = { 26 .change_ok = env_flags_validate, 27 }; 28 29 /* 30 * Read an environment variable as a boolean 31 * Return -1 if variable does not exist (default to true) 32 */ 33 int env_get_yesno(const char *var) 34 { 35 char *s = env_get(var); 36 37 if (s == NULL) 38 return -1; 39 return (*s == '1' || *s == 'y' || *s == 'Y' || *s == 't' || *s == 'T') ? 40 1 : 0; 41 } 42 43 /* 44 * Look up the variable from the default environment 45 */ 46 char *env_get_default(const char *name) 47 { 48 char *ret_val; 49 unsigned long really_valid = gd->env_valid; 50 unsigned long real_gd_flags = gd->flags; 51 52 /* Pretend that the image is bad. */ 53 gd->flags &= ~GD_FLG_ENV_READY; 54 gd->env_valid = ENV_INVALID; 55 ret_val = env_get(name); 56 gd->env_valid = really_valid; 57 gd->flags = real_gd_flags; 58 return ret_val; 59 } 60 61 void set_default_env(const char *s) 62 { 63 int flags = 0; 64 65 if (sizeof(default_environment) > ENV_SIZE) { 66 puts("*** Error - default environment is too large\n\n"); 67 return; 68 } 69 70 if (s) { 71 if (*s == '!') { 72 printf("*** Warning - %s, " 73 "using default environment\n\n", 74 s + 1); 75 } else { 76 flags = H_INTERACTIVE; 77 puts(s); 78 } 79 } else { 80 debug("Using default environment\n"); 81 } 82 83 if (himport_r(&env_htab, (char *)default_environment, 84 sizeof(default_environment), '\0', flags, 0, 85 0, NULL) == 0) 86 pr_err("Environment import failed: errno = %d\n", errno); 87 88 gd->flags |= GD_FLG_ENV_READY; 89 gd->flags |= GD_FLG_ENV_DEFAULT; 90 } 91 92 93 /* [re]set individual variables to their value in the default environment */ 94 int set_default_vars(int nvars, char * const vars[], int flags) 95 { 96 /* 97 * Special use-case: import from default environment 98 * (and use \0 as a separator) 99 */ 100 flags |= H_NOCLEAR; 101 return himport_r(&env_htab, (const char *)default_environment, 102 sizeof(default_environment), '\0', 103 flags, 0, nvars, vars); 104 } 105 106 /* 107 * Check if CRC is valid and (if yes) import the environment. 108 * Note that "buf" may or may not be aligned. 109 */ 110 int env_import(const char *buf, int check) 111 { 112 env_t *ep = (env_t *)buf; 113 114 if (check) { 115 uint32_t crc; 116 117 memcpy(&crc, &ep->crc, sizeof(crc)); 118 119 if (crc32(0, ep->data, ENV_SIZE) != crc) { 120 set_default_env("!bad CRC"); 121 return -EIO; 122 } 123 } 124 125 if (himport_r(&env_htab, (char *)ep->data, ENV_SIZE, '\0', 0, 0, 126 0, NULL)) { 127 gd->flags |= GD_FLG_ENV_READY; 128 return 0; 129 } 130 131 pr_err("Cannot import environment: errno = %d\n", errno); 132 133 set_default_env("!import failed"); 134 135 return -EIO; 136 } 137 138 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT 139 static unsigned char env_flags; 140 141 int env_import_redund(const char *buf1, int buf1_read_fail, 142 const char *buf2, int buf2_read_fail) 143 { 144 int crc1_ok, crc2_ok; 145 env_t *ep, *tmp_env1, *tmp_env2; 146 147 tmp_env1 = (env_t *)buf1; 148 tmp_env2 = (env_t *)buf2; 149 150 if (buf1_read_fail && buf2_read_fail) { 151 puts("*** Error - No Valid Environment Area found\n"); 152 } else if (buf1_read_fail || buf2_read_fail) { 153 puts("*** Warning - some problems detected "); 154 puts("reading environment; recovered successfully\n"); 155 } 156 157 if (buf1_read_fail && buf2_read_fail) { 158 set_default_env("!bad env area"); 159 return -EIO; 160 } else if (!buf1_read_fail && buf2_read_fail) { 161 gd->env_valid = ENV_VALID; 162 return env_import((char *)tmp_env1, 1); 163 } else if (buf1_read_fail && !buf2_read_fail) { 164 gd->env_valid = ENV_REDUND; 165 return env_import((char *)tmp_env2, 1); 166 } 167 168 crc1_ok = crc32(0, tmp_env1->data, ENV_SIZE) == 169 tmp_env1->crc; 170 crc2_ok = crc32(0, tmp_env2->data, ENV_SIZE) == 171 tmp_env2->crc; 172 173 if (!crc1_ok && !crc2_ok) { 174 set_default_env("!bad CRC"); 175 return -EIO; 176 } else if (crc1_ok && !crc2_ok) { 177 gd->env_valid = ENV_VALID; 178 } else if (!crc1_ok && crc2_ok) { 179 gd->env_valid = ENV_REDUND; 180 } else { 181 /* both ok - check serial */ 182 if (tmp_env1->flags == 255 && tmp_env2->flags == 0) 183 gd->env_valid = ENV_REDUND; 184 else if (tmp_env2->flags == 255 && tmp_env1->flags == 0) 185 gd->env_valid = ENV_VALID; 186 else if (tmp_env1->flags > tmp_env2->flags) 187 gd->env_valid = ENV_VALID; 188 else if (tmp_env2->flags > tmp_env1->flags) 189 gd->env_valid = ENV_REDUND; 190 else /* flags are equal - almost impossible */ 191 gd->env_valid = ENV_VALID; 192 } 193 194 if (gd->env_valid == ENV_VALID) 195 ep = tmp_env1; 196 else 197 ep = tmp_env2; 198 199 env_flags = ep->flags; 200 return env_import((char *)ep, 0); 201 } 202 #endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */ 203 204 /* Export the environment and generate CRC for it. */ 205 int env_export(env_t *env_out) 206 { 207 char *res; 208 ssize_t len; 209 210 res = (char *)env_out->data; 211 len = hexport_r(&env_htab, '\0', 0, &res, ENV_SIZE, 0, NULL); 212 if (len < 0) { 213 pr_err("Cannot export environment: errno = %d\n", errno); 214 return 1; 215 } 216 217 env_out->crc = crc32(0, env_out->data, ENV_SIZE); 218 219 #ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT 220 env_out->flags = ++env_flags; /* increase the serial */ 221 #endif 222 223 return 0; 224 } 225 226 void env_relocate(void) 227 { 228 #if defined(CONFIG_NEEDS_MANUAL_RELOC) 229 env_reloc(); 230 env_fix_drivers(); 231 env_htab.change_ok += gd->reloc_off; 232 #endif 233 if (gd->env_valid == ENV_INVALID) { 234 #if defined(CONFIG_ENV_IS_NOWHERE) || defined(CONFIG_SPL_BUILD) 235 /* Environment not changable */ 236 set_default_env(NULL); 237 #else 238 bootstage_error(BOOTSTAGE_ID_NET_CHECKSUM); 239 set_default_env("!bad CRC"); 240 #endif 241 } else { 242 env_load(); 243 } 244 } 245 246 #if defined(CONFIG_AUTO_COMPLETE) && !defined(CONFIG_SPL_BUILD) 247 int env_complete(char *var, int maxv, char *cmdv[], int bufsz, char *buf) 248 { 249 ENTRY *match; 250 int found, idx; 251 252 idx = 0; 253 found = 0; 254 cmdv[0] = NULL; 255 256 while ((idx = hmatch_r(var, idx, &match, &env_htab))) { 257 int vallen = strlen(match->key) + 1; 258 259 if (found >= maxv - 2 || bufsz < vallen) 260 break; 261 262 cmdv[found++] = buf; 263 memcpy(buf, match->key, vallen); 264 buf += vallen; 265 bufsz -= vallen; 266 } 267 268 qsort(cmdv, found, sizeof(cmdv[0]), strcmp_compar); 269 270 if (idx) 271 cmdv[found++] = "..."; 272 273 cmdv[found] = NULL; 274 return found; 275 } 276 #endif 277