--- embedtools/src/ub_env.c 2014/01/29 01:40:35 1.1.2.10 +++ embedtools/src/ub_env.c 2014/01/30 07:36:53 1.1.2.18 @@ -1,25 +1,75 @@ +/************************************************************************* + * (C) 2014 AITNET - Sofia/Bulgaria - + * by Michael Pounov + * + * $Author: misho $ + * $Id: ub_env.c,v 1.1.2.18 2014/01/30 07:36:53 misho Exp $ + * + ************************************************************************* +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 + +Copyright 2004 - 2014 + by Michael Pounov . 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 +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; - size_t siz; + ssize_t siz, esiz; ait_val_t v; + u_int crc; FTRACE(4); - str = cfg_getAttribute(&cfg, csSec, "size"); + str = cfg_getAttribute(&cfg, csSec, "drive_size"); siz = strtol(str, NULL, 0); if (!siz) return -1; - str = cfg_getAttribute(&cfg, csSec, "drive"); + 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; @@ -43,6 +93,8 @@ ub_flash_io(const char *csSec, int mode) } 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); @@ -57,6 +109,12 @@ ub_flash_io(const char *csSec, int mode) 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)); @@ -67,8 +125,6 @@ ub_flash_io(const char *csSec, int mode) static inline const char * ub_envmatch(const char *csName, const char *e) { - const char *str = NULL; - while (*csName == *e++) if (*csName++ == '=') return e; @@ -86,7 +142,7 @@ ub_load(const char *csSec) FTRACE(4); - str = cfg_getAttribute(&cfg, csSec, "size"); + str = cfg_getAttribute(&cfg, csSec, "drive_size"); siz = strtol(str, NULL, 0); if (!siz) return -1; @@ -113,14 +169,19 @@ const char* ub_getenv(const char *csSec, const char *csName) { char *e, *nxt; - size_t dlen; + size_t dlen = 0; const char *str = NULL; - FTRACE(3); - str = cfg_getAttribute(&cfg, csSec, "size"); - dlen = strtol(str, NULL, 0); + 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 @@ -144,8 +205,79 @@ ub_getenv(const char *csSec, const char *csName) 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; } @@ -153,13 +285,19 @@ int ub_env(const char *csSec) { char *e, *nxt; - size_t dlen; + ssize_t dlen = 0; const char *str; FTRACE(3); - str = cfg_getAttribute(&cfg, csSec, "size"); - dlen = strtol(str, NULL, 0); + 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