![Joe Hershberger](/assets/img/avatar_default.png)
Move the read of the old value to inside the check function. In some cases it can be avoided all together and at the least the code is only called from one place. Also name the function and the callback to more clearly describe what it does. Pass the ENTRY instead of just the name for direct access to the whole data structure. Pass an enum to the callback that specifies the operation being approved. Signed-off-by: Joe Hershberger <joe.hershberger@ni.com>
211 lines
4.7 KiB
C
211 lines
4.7 KiB
C
/*
|
|
* (C) Copyright 2000-2010
|
|
* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
|
|
*
|
|
* (C) Copyright 2001 Sysgo Real-Time Solutions, GmbH <www.elinos.com>
|
|
* Andreas Heppel <aheppel@sysgo.de>
|
|
*
|
|
* See file CREDITS for list of people who contributed to this
|
|
* project.
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation; either version 2 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
|
|
* MA 02111-1307 USA
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <command.h>
|
|
#include <environment.h>
|
|
#include <linux/stddef.h>
|
|
#include <search.h>
|
|
#include <errno.h>
|
|
#include <malloc.h>
|
|
|
|
DECLARE_GLOBAL_DATA_PTR;
|
|
|
|
/************************************************************************
|
|
* Default settings to be used when no valid environment is found
|
|
*/
|
|
#include <env_default.h>
|
|
|
|
struct hsearch_data env_htab = {
|
|
.change_ok = env_change_ok,
|
|
};
|
|
|
|
static uchar __env_get_char_spec(int index)
|
|
{
|
|
return *((uchar *)(gd->env_addr + index));
|
|
}
|
|
uchar env_get_char_spec(int)
|
|
__attribute__((weak, alias("__env_get_char_spec")));
|
|
|
|
static uchar env_get_char_init(int index)
|
|
{
|
|
/* if crc was bad, use the default environment */
|
|
if (gd->env_valid)
|
|
return env_get_char_spec(index);
|
|
else
|
|
return default_environment[index];
|
|
}
|
|
|
|
uchar env_get_char_memory(int index)
|
|
{
|
|
return *env_get_addr(index);
|
|
}
|
|
|
|
uchar env_get_char(int index)
|
|
{
|
|
/* if relocated to RAM */
|
|
if (gd->flags & GD_FLG_RELOC)
|
|
return env_get_char_memory(index);
|
|
else
|
|
return env_get_char_init(index);
|
|
}
|
|
|
|
const uchar *env_get_addr(int index)
|
|
{
|
|
if (gd->env_valid)
|
|
return (uchar *)(gd->env_addr + index);
|
|
else
|
|
return &default_environment[index];
|
|
}
|
|
|
|
void set_default_env(const char *s)
|
|
{
|
|
int flags = 0;
|
|
|
|
if (sizeof(default_environment) > ENV_SIZE) {
|
|
puts("*** Error - default environment is too large\n\n");
|
|
return;
|
|
}
|
|
|
|
if (s) {
|
|
if (*s == '!') {
|
|
printf("*** Warning - %s, "
|
|
"using default environment\n\n",
|
|
s + 1);
|
|
} else {
|
|
flags = H_INTERACTIVE;
|
|
puts(s);
|
|
}
|
|
} else {
|
|
puts("Using default environment\n\n");
|
|
}
|
|
|
|
if (himport_r(&env_htab, (char *)default_environment,
|
|
sizeof(default_environment), '\0', flags,
|
|
0, NULL) == 0)
|
|
error("Environment import failed: errno = %d\n", errno);
|
|
|
|
gd->flags |= GD_FLG_ENV_READY;
|
|
}
|
|
|
|
|
|
/* [re]set individual variables to their value in the default environment */
|
|
int set_default_vars(int nvars, char * const vars[])
|
|
{
|
|
/*
|
|
* Special use-case: import from default environment
|
|
* (and use \0 as a separator)
|
|
*/
|
|
return himport_r(&env_htab, (const char *)default_environment,
|
|
sizeof(default_environment), '\0',
|
|
H_NOCLEAR | H_INTERACTIVE, nvars, vars);
|
|
}
|
|
|
|
#ifndef CONFIG_SPL_BUILD
|
|
/*
|
|
* Check if CRC is valid and (if yes) import the environment.
|
|
* Note that "buf" may or may not be aligned.
|
|
*/
|
|
int env_import(const char *buf, int check)
|
|
{
|
|
env_t *ep = (env_t *)buf;
|
|
|
|
if (check) {
|
|
uint32_t crc;
|
|
|
|
memcpy(&crc, &ep->crc, sizeof(crc));
|
|
|
|
if (crc32(0, ep->data, ENV_SIZE) != crc) {
|
|
set_default_env("!bad CRC");
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
if (himport_r(&env_htab, (char *)ep->data, ENV_SIZE, '\0', 0,
|
|
0, NULL)) {
|
|
gd->flags |= GD_FLG_ENV_READY;
|
|
return 1;
|
|
}
|
|
|
|
error("Cannot import environment: errno = %d\n", errno);
|
|
|
|
set_default_env("!import failed");
|
|
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
void env_relocate(void)
|
|
{
|
|
#if defined(CONFIG_NEEDS_MANUAL_RELOC)
|
|
env_reloc();
|
|
env_htab.change_ok += gd->reloc_off;
|
|
#endif
|
|
if (gd->env_valid == 0) {
|
|
#if defined(CONFIG_ENV_IS_NOWHERE) || defined(CONFIG_SPL_BUILD)
|
|
/* Environment not changable */
|
|
set_default_env(NULL);
|
|
#else
|
|
bootstage_error(BOOTSTAGE_ID_NET_CHECKSUM);
|
|
set_default_env("!bad CRC");
|
|
#endif
|
|
} else {
|
|
env_relocate_spec();
|
|
}
|
|
}
|
|
|
|
#if defined(CONFIG_AUTO_COMPLETE) && !defined(CONFIG_SPL_BUILD)
|
|
int env_complete(char *var, int maxv, char *cmdv[], int bufsz, char *buf)
|
|
{
|
|
ENTRY *match;
|
|
int found, idx;
|
|
|
|
idx = 0;
|
|
found = 0;
|
|
cmdv[0] = NULL;
|
|
|
|
while ((idx = hmatch_r(var, idx, &match, &env_htab))) {
|
|
int vallen = strlen(match->key) + 1;
|
|
|
|
if (found >= maxv - 2 || bufsz < vallen)
|
|
break;
|
|
|
|
cmdv[found++] = buf;
|
|
memcpy(buf, match->key, vallen);
|
|
buf += vallen;
|
|
bufsz -= vallen;
|
|
}
|
|
|
|
qsort(cmdv, found, sizeof(cmdv[0]), strcmp_compar);
|
|
|
|
if (idx)
|
|
cmdv[found++] = "...";
|
|
|
|
cmdv[found] = NULL;
|
|
return found;
|
|
}
|
|
#endif
|