version 1.1, 2014/01/28 08:04:21
|
version 1.2, 2014/02/05 15:44:06
|
Line 0
|
Line 1
|
|
/************************************************************************* |
|
* (C) 2014 AITNET - Sofia/Bulgaria - <office@aitbg.com> |
|
* by Michael Pounov <misho@aitbg.com> |
|
* |
|
* $Author$ |
|
* $Id$ |
|
* |
|
************************************************************************* |
|
The ELWIX and AITNET software is distributed under the following |
|
terms: |
|
|
|
All of the documentation and software included in the ELWIX and AITNET |
|
Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org> |
|
|
|
Copyright 2004 - 2014 |
|
by Michael Pounov <misho@elwix.org>. All rights reserved. |
|
|
|
Redistribution and use in source and binary forms, with or without |
|
modification, are permitted provided that the following conditions |
|
are met: |
|
1. Redistributions of source code must retain the above copyright |
|
notice, this list of conditions and the following disclaimer. |
|
2. Redistributions in binary form must reproduce the above copyright |
|
notice, this list of conditions and the following disclaimer in the |
|
documentation and/or other materials provided with the distribution. |
|
3. All advertising materials mentioning features or use of this software |
|
must display the following acknowledgement: |
|
This product includes software developed by Michael Pounov <misho@elwix.org> |
|
ELWIX - Embedded LightWeight unIX and its contributors. |
|
4. Neither the name of AITNET nor the names of its contributors |
|
may be used to endorse or promote products derived from this software |
|
without specific prior written permission. |
|
|
|
THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND |
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
|
OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
|
HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
|
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
|
OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
|
SUCH DAMAGE. |
|
*/ |
|
#include "global.h" |
|
#include "ub_env.h" |
|
|
|
|
|
env_t *env; |
|
|
|
static int |
|
ub_flash_io(const char *csSec, int mode) |
|
{ |
|
int f, l, rlen; |
|
const char *str; |
|
ssize_t siz, esiz; |
|
ait_val_t v; |
|
u_int crc; |
|
|
|
FTRACE(4); |
|
|
|
str = cfg_getAttribute(&cfg, csSec, "drive_size"); |
|
siz = strtol(str, NULL, 0); |
|
if (!siz) |
|
return -1; |
|
str = cfg_getAttribute(&cfg, csSec, "env_size"); |
|
if (!str) |
|
esiz = siz - sizeof env->env_crc; |
|
else |
|
esiz = strtol(str, NULL, 0); |
|
str = cfg_getAttribute(&cfg, csSec, "drive_name"); |
|
if (!str) { |
|
printf("Error:: drive not found!\n"); |
|
return -1; |
|
} |
|
|
|
cfg_loadAttribute(&cfg, "ube", "lock", &v, UBE_LOCK); |
|
l = open(AIT_GET_STR(&v), O_CREAT | O_EXCL | O_WRONLY, 0644); |
|
if (l == -1) { |
|
printf("Error:: Locked u-boot-env map %s\n", AIT_GET_STR(&v)); |
|
AIT_FREE_VAL(&v); |
|
return -1; |
|
} |
|
|
|
f = open(str, mode); |
|
if (f == -1) { |
|
printf("Error:: Can't access u-boot-env device %s\n", str); |
|
close(l); |
|
unlink(AIT_GET_STR(&v)); |
|
AIT_FREE_VAL(&v); |
|
return -1; |
|
} |
|
|
|
if (mode & O_RDWR) { |
|
env->env_crc = crc_32(0, (u_char*) env->env_data, esiz); |
|
VERB(5) printf("Write CRC32 0x%x\n", env->env_crc); |
|
rlen = write(f, env, siz); |
|
if (rlen != siz) |
|
printf("Error:: written %d bytes != %d\n", rlen, siz); |
|
else |
|
VERB(3) printf("Written %d bytes\n", rlen); |
|
lseek(f, 0, SEEK_SET); |
|
} |
|
|
|
rlen = read(f, env, siz); |
|
if (rlen != siz) |
|
printf("Error:: readed %d bytes != %d\n", rlen, siz); |
|
else |
|
VERB(3) printf("Readed %d bytes\n", rlen); |
|
|
|
crc = crc_32(0, (u_char*) env->env_data, esiz); |
|
VERB(5) printf("Calculated CRC32 0x%x\n", crc); |
|
if (crc != env->env_crc) |
|
VERB(1) printf("Warning:: Bad CRC, Flash crc32 0x%x != 0x%x\n", |
|
env->env_crc, crc); |
|
|
|
close(f); |
|
close(l); |
|
unlink(AIT_GET_STR(&v)); |
|
AIT_FREE_VAL(&v); |
|
return 0; |
|
} |
|
|
|
static inline const char * |
|
ub_envmatch(const char *csName, const char *e) |
|
{ |
|
while (*csName == *e++) |
|
if (*csName++ == '=') |
|
return e; |
|
if (!*csName && *(e - 1) == '=') |
|
return e; |
|
|
|
return NULL; |
|
} |
|
|
|
int |
|
ub_load(const char *csSec) |
|
{ |
|
const char *str; |
|
size_t siz; |
|
|
|
FTRACE(4); |
|
|
|
str = cfg_getAttribute(&cfg, csSec, "drive_size"); |
|
siz = strtol(str, NULL, 0); |
|
if (!siz) |
|
return -1; |
|
|
|
env = e_malloc(siz); |
|
if (!env) { |
|
ELIBERR(elwix); |
|
return -1; |
|
} |
|
|
|
return ub_flash_io(csSec, O_RDONLY); |
|
} |
|
|
|
void |
|
ub_unload() |
|
{ |
|
FTRACE(4); |
|
|
|
if (env) |
|
e_free(env); |
|
} |
|
|
|
const char* |
|
ub_getenv(const char *csSec, const char *csName) |
|
{ |
|
char *e, *nxt; |
|
size_t dlen = 0; |
|
const char *str = NULL; |
|
|
|
FTRACE(3); |
|
|
|
str = cfg_getAttribute(&cfg, csSec, "env_size"); |
|
if (!str) { |
|
str = cfg_getAttribute(&cfg, csSec, "drive_size"); |
|
if (!str) |
|
return NULL; |
|
dlen -= sizeof env->env_crc; |
|
} |
|
dlen += strtol(str, NULL, 0); |
|
if (!dlen) |
|
return NULL; |
|
else |
|
dlen--; |
|
|
|
for (e = env->env_data; *e; e = nxt + 1) { |
|
for (nxt = e; *nxt; nxt++) |
|
if (nxt >= env->env_data + dlen) { |
|
printf("Error:: environment not terminated\n"); |
|
return NULL; |
|
} |
|
|
|
str = ub_envmatch(csName, e); |
|
if (str) |
|
break; |
|
} |
|
|
|
return str; |
|
} |
|
|
|
int |
|
ub_setenv(const char *csSec, const char *csName, const char *csValue) |
|
{ |
|
char *e, *nxt; |
|
ssize_t dlen = 0, len; |
|
const char *str, *old = NULL; |
|
|
|
FTRACE(3); |
|
|
|
str = cfg_getAttribute(&cfg, csSec, "env_size"); |
|
if (!str) { |
|
str = cfg_getAttribute(&cfg, csSec, "drive_size"); |
|
if (!str) |
|
return -1; |
|
dlen -= sizeof env->env_crc; |
|
} |
|
dlen += strtol(str, NULL, 0); |
|
if (!dlen) |
|
return -1; |
|
else |
|
dlen--; |
|
|
|
for (e = env->env_data; *e; e = nxt + 1) { |
|
for (nxt = e; *nxt; nxt++) |
|
if (nxt >= env->env_data + dlen) { |
|
printf("Error:: environment not terminated\n"); |
|
return -1; |
|
} |
|
|
|
old = ub_envmatch(csName, e); |
|
if (old) |
|
break; |
|
} |
|
|
|
/* Delete any existing definition */ |
|
if (old) { |
|
if (!*++nxt) |
|
*e = 0; |
|
else |
|
while (42) { |
|
*e = *nxt++; |
|
if (!*e && !*nxt) |
|
break; |
|
e++; |
|
} |
|
*++e = 0; |
|
} |
|
|
|
if (csValue) { |
|
/* Append new definition at the end */ |
|
for (e = env->env_data; *e || *(e + 1); e++); |
|
if (e > env->env_data) |
|
e++; |
|
/* "name" + "=" + "val" +"\0\0" check u-boot-env size */ |
|
len = strlen(csName) + 2; /* add '=' for first arg, ' ' for all others */ |
|
len += strlen(csValue) + 1; |
|
if (len > env->env_data + dlen - e) { |
|
printf("Error:: Environment overflow!\n"); |
|
return -1; |
|
} |
|
|
|
/* av pair */ |
|
while ((*e = *csName++)) |
|
e++; |
|
*e = '='; |
|
while ((*++e = *csValue++)); |
|
|
|
/* end is marked with double '\0' */ |
|
*++e = 0; |
|
} |
|
|
|
if (ub_flash_io(csSec, O_RDWR)) { |
|
printf("Error:: Can't write environment to flash!\n"); |
|
return -1; |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
int |
|
ub_env(const char *csSec) |
|
{ |
|
char *e, *nxt; |
|
ssize_t dlen = 0; |
|
const char *str; |
|
|
|
FTRACE(3); |
|
|
|
str = cfg_getAttribute(&cfg, csSec, "env_size"); |
|
if (!str) { |
|
str = cfg_getAttribute(&cfg, csSec, "drive_size"); |
|
if (!str) |
|
return -1; |
|
dlen -= sizeof env->env_crc; |
|
} |
|
dlen += strtol(str, NULL, 0); |
|
if (!dlen) |
|
return -1; |
|
else |
|
dlen--; |
|
|
|
for (e = env->env_data; *e; e = nxt + 1) { |
|
for (nxt = e; *nxt; nxt++) |
|
if (nxt >= env->env_data + dlen) { |
|
printf("Error:: environment not terminated\n"); |
|
return -1; |
|
} |
|
|
|
printf("%s\n", e); |
|
} |
|
|
|
return 0; |
|
} |