1a6826fbcSWolfgang Denk /* 2a6826fbcSWolfgang Denk * This implementation is based on code from uClibc-0.9.30.3 but was 3a6826fbcSWolfgang Denk * modified and extended for use within U-Boot. 4a6826fbcSWolfgang Denk * 5a6826fbcSWolfgang Denk * Copyright (C) 2010 Wolfgang Denk <wd@denx.de> 6a6826fbcSWolfgang Denk * 7a6826fbcSWolfgang Denk * Original license header: 8a6826fbcSWolfgang Denk * 9a6826fbcSWolfgang Denk * Copyright (C) 1993, 1995, 1996, 1997, 2002 Free Software Foundation, Inc. 10a6826fbcSWolfgang Denk * This file is part of the GNU C Library. 11a6826fbcSWolfgang Denk * Contributed by Ulrich Drepper <drepper@gnu.ai.mit.edu>, 1993. 12a6826fbcSWolfgang Denk * 13a6826fbcSWolfgang Denk * The GNU C Library is free software; you can redistribute it and/or 14a6826fbcSWolfgang Denk * modify it under the terms of the GNU Lesser General Public 15a6826fbcSWolfgang Denk * License as published by the Free Software Foundation; either 16a6826fbcSWolfgang Denk * version 2.1 of the License, or (at your option) any later version. 17a6826fbcSWolfgang Denk * 18a6826fbcSWolfgang Denk * The GNU C Library is distributed in the hope that it will be useful, 19a6826fbcSWolfgang Denk * but WITHOUT ANY WARRANTY; without even the implied warranty of 20a6826fbcSWolfgang Denk * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 21a6826fbcSWolfgang Denk * Lesser General Public License for more details. 22a6826fbcSWolfgang Denk * 23a6826fbcSWolfgang Denk * You should have received a copy of the GNU Lesser General Public 24a6826fbcSWolfgang Denk * License along with the GNU C Library; if not, write to the Free 25a6826fbcSWolfgang Denk * Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 26a6826fbcSWolfgang Denk * 02111-1307 USA. 27a6826fbcSWolfgang Denk */ 28a6826fbcSWolfgang Denk 29a6826fbcSWolfgang Denk #include <errno.h> 30a6826fbcSWolfgang Denk #include <malloc.h> 31a6826fbcSWolfgang Denk 32a6826fbcSWolfgang Denk #ifdef USE_HOSTCC /* HOST build */ 33a6826fbcSWolfgang Denk # include <string.h> 34a6826fbcSWolfgang Denk # include <assert.h> 354d91a6ecSJason Hobbs # include <ctype.h> 36a6826fbcSWolfgang Denk 37a6826fbcSWolfgang Denk # ifndef debug 38a6826fbcSWolfgang Denk # ifdef DEBUG 39a6826fbcSWolfgang Denk # define debug(fmt,args...) printf(fmt ,##args) 40a6826fbcSWolfgang Denk # else 41a6826fbcSWolfgang Denk # define debug(fmt,args...) 42a6826fbcSWolfgang Denk # endif 43a6826fbcSWolfgang Denk # endif 44a6826fbcSWolfgang Denk #else /* U-Boot build */ 45a6826fbcSWolfgang Denk # include <common.h> 46a6826fbcSWolfgang Denk # include <linux/string.h> 474d91a6ecSJason Hobbs # include <linux/ctype.h> 48a6826fbcSWolfgang Denk #endif 49a6826fbcSWolfgang Denk 50fc5fc76bSAndreas Bießmann #ifndef CONFIG_ENV_MIN_ENTRIES /* minimum number of entries */ 51fc5fc76bSAndreas Bießmann #define CONFIG_ENV_MIN_ENTRIES 64 52fc5fc76bSAndreas Bießmann #endif 53ea882bafSWolfgang Denk #ifndef CONFIG_ENV_MAX_ENTRIES /* maximum number of entries */ 54ea882bafSWolfgang Denk #define CONFIG_ENV_MAX_ENTRIES 512 55ea882bafSWolfgang Denk #endif 56ea882bafSWolfgang Denk 57*170ab110SJoe Hershberger #include <env_callback.h> 58*170ab110SJoe Hershberger #include <search.h> 59a6826fbcSWolfgang Denk 60a6826fbcSWolfgang Denk /* 61a6826fbcSWolfgang Denk * [Aho,Sethi,Ullman] Compilers: Principles, Techniques and Tools, 1986 62a6826fbcSWolfgang Denk * [Knuth] The Art of Computer Programming, part 3 (6.4) 63a6826fbcSWolfgang Denk */ 64a6826fbcSWolfgang Denk 65a6826fbcSWolfgang Denk /* 66a6826fbcSWolfgang Denk * The reentrant version has no static variables to maintain the state. 67a6826fbcSWolfgang Denk * Instead the interface of all functions is extended to take an argument 68a6826fbcSWolfgang Denk * which describes the current status. 69a6826fbcSWolfgang Denk */ 707afcf3a5SJoe Hershberger 71a6826fbcSWolfgang Denk typedef struct _ENTRY { 72c81c1222SPeter Barada int used; 73a6826fbcSWolfgang Denk ENTRY entry; 74a6826fbcSWolfgang Denk } _ENTRY; 75a6826fbcSWolfgang Denk 76a6826fbcSWolfgang Denk 777afcf3a5SJoe Hershberger static void _hdelete(const char *key, struct hsearch_data *htab, ENTRY *ep, 787afcf3a5SJoe Hershberger int idx); 797afcf3a5SJoe Hershberger 80a6826fbcSWolfgang Denk /* 81a6826fbcSWolfgang Denk * hcreate() 82a6826fbcSWolfgang Denk */ 83a6826fbcSWolfgang Denk 84a6826fbcSWolfgang Denk /* 85a6826fbcSWolfgang Denk * For the used double hash method the table size has to be a prime. To 86a6826fbcSWolfgang Denk * correct the user given table size we need a prime test. This trivial 87a6826fbcSWolfgang Denk * algorithm is adequate because 88a6826fbcSWolfgang Denk * a) the code is (most probably) called a few times per program run and 89a6826fbcSWolfgang Denk * b) the number is small because the table must fit in the core 90a6826fbcSWolfgang Denk * */ 91a6826fbcSWolfgang Denk static int isprime(unsigned int number) 92a6826fbcSWolfgang Denk { 93a6826fbcSWolfgang Denk /* no even number will be passed */ 94a6826fbcSWolfgang Denk unsigned int div = 3; 95a6826fbcSWolfgang Denk 96a6826fbcSWolfgang Denk while (div * div < number && number % div != 0) 97a6826fbcSWolfgang Denk div += 2; 98a6826fbcSWolfgang Denk 99a6826fbcSWolfgang Denk return number % div != 0; 100a6826fbcSWolfgang Denk } 101a6826fbcSWolfgang Denk 102a6826fbcSWolfgang Denk /* 103a6826fbcSWolfgang Denk * Before using the hash table we must allocate memory for it. 104a6826fbcSWolfgang Denk * Test for an existing table are done. We allocate one element 105a6826fbcSWolfgang Denk * more as the found prime number says. This is done for more effective 106a6826fbcSWolfgang Denk * indexing as explained in the comment for the hsearch function. 107a6826fbcSWolfgang Denk * The contents of the table is zeroed, especially the field used 108a6826fbcSWolfgang Denk * becomes zero. 109a6826fbcSWolfgang Denk */ 1102eb1573fSMike Frysinger 111a6826fbcSWolfgang Denk int hcreate_r(size_t nel, struct hsearch_data *htab) 112a6826fbcSWolfgang Denk { 113a6826fbcSWolfgang Denk /* Test for correct arguments. */ 114a6826fbcSWolfgang Denk if (htab == NULL) { 115a6826fbcSWolfgang Denk __set_errno(EINVAL); 116a6826fbcSWolfgang Denk return 0; 117a6826fbcSWolfgang Denk } 118a6826fbcSWolfgang Denk 119a6826fbcSWolfgang Denk /* There is still another table active. Return with error. */ 120a6826fbcSWolfgang Denk if (htab->table != NULL) 121a6826fbcSWolfgang Denk return 0; 122a6826fbcSWolfgang Denk 123a6826fbcSWolfgang Denk /* Change nel to the first prime number not smaller as nel. */ 124a6826fbcSWolfgang Denk nel |= 1; /* make odd */ 125a6826fbcSWolfgang Denk while (!isprime(nel)) 126a6826fbcSWolfgang Denk nel += 2; 127a6826fbcSWolfgang Denk 128a6826fbcSWolfgang Denk htab->size = nel; 129a6826fbcSWolfgang Denk htab->filled = 0; 130a6826fbcSWolfgang Denk 131a6826fbcSWolfgang Denk /* allocate memory and zero out */ 132a6826fbcSWolfgang Denk htab->table = (_ENTRY *) calloc(htab->size + 1, sizeof(_ENTRY)); 133a6826fbcSWolfgang Denk if (htab->table == NULL) 134a6826fbcSWolfgang Denk return 0; 135a6826fbcSWolfgang Denk 136a6826fbcSWolfgang Denk /* everything went alright */ 137a6826fbcSWolfgang Denk return 1; 138a6826fbcSWolfgang Denk } 139a6826fbcSWolfgang Denk 140a6826fbcSWolfgang Denk 141a6826fbcSWolfgang Denk /* 142a6826fbcSWolfgang Denk * hdestroy() 143a6826fbcSWolfgang Denk */ 144a6826fbcSWolfgang Denk 145a6826fbcSWolfgang Denk /* 146a6826fbcSWolfgang Denk * After using the hash table it has to be destroyed. The used memory can 147a6826fbcSWolfgang Denk * be freed and the local static variable can be marked as not used. 148a6826fbcSWolfgang Denk */ 1492eb1573fSMike Frysinger 150c4e0057fSJoe Hershberger void hdestroy_r(struct hsearch_data *htab) 151a6826fbcSWolfgang Denk { 152a6826fbcSWolfgang Denk int i; 153a6826fbcSWolfgang Denk 154a6826fbcSWolfgang Denk /* Test for correct arguments. */ 155a6826fbcSWolfgang Denk if (htab == NULL) { 156a6826fbcSWolfgang Denk __set_errno(EINVAL); 157a6826fbcSWolfgang Denk return; 158a6826fbcSWolfgang Denk } 159a6826fbcSWolfgang Denk 160a6826fbcSWolfgang Denk /* free used memory */ 161a6826fbcSWolfgang Denk for (i = 1; i <= htab->size; ++i) { 162c81c1222SPeter Barada if (htab->table[i].used > 0) { 163a6826fbcSWolfgang Denk ENTRY *ep = &htab->table[i].entry; 164c4e0057fSJoe Hershberger 16584b5e802SWolfgang Denk free((void *)ep->key); 166a6826fbcSWolfgang Denk free(ep->data); 167a6826fbcSWolfgang Denk } 168a6826fbcSWolfgang Denk } 169a6826fbcSWolfgang Denk free(htab->table); 170a6826fbcSWolfgang Denk 171a6826fbcSWolfgang Denk /* the sign for an existing table is an value != NULL in htable */ 172a6826fbcSWolfgang Denk htab->table = NULL; 173a6826fbcSWolfgang Denk } 174a6826fbcSWolfgang Denk 175a6826fbcSWolfgang Denk /* 176a6826fbcSWolfgang Denk * hsearch() 177a6826fbcSWolfgang Denk */ 178a6826fbcSWolfgang Denk 179a6826fbcSWolfgang Denk /* 180a6826fbcSWolfgang Denk * This is the search function. It uses double hashing with open addressing. 181a6826fbcSWolfgang Denk * The argument item.key has to be a pointer to an zero terminated, most 182a6826fbcSWolfgang Denk * probably strings of chars. The function for generating a number of the 183a6826fbcSWolfgang Denk * strings is simple but fast. It can be replaced by a more complex function 184a6826fbcSWolfgang Denk * like ajw (see [Aho,Sethi,Ullman]) if the needs are shown. 185a6826fbcSWolfgang Denk * 186a6826fbcSWolfgang Denk * We use an trick to speed up the lookup. The table is created by hcreate 187a6826fbcSWolfgang Denk * with one more element available. This enables us to use the index zero 188a6826fbcSWolfgang Denk * special. This index will never be used because we store the first hash 189a6826fbcSWolfgang Denk * index in the field used where zero means not used. Every other value 190a6826fbcSWolfgang Denk * means used. The used field can be used as a first fast comparison for 191a6826fbcSWolfgang Denk * equality of the stored and the parameter value. This helps to prevent 192a6826fbcSWolfgang Denk * unnecessary expensive calls of strcmp. 193a6826fbcSWolfgang Denk * 194a6826fbcSWolfgang Denk * This implementation differs from the standard library version of 195a6826fbcSWolfgang Denk * this function in a number of ways: 196a6826fbcSWolfgang Denk * 197a6826fbcSWolfgang Denk * - While the standard version does not make any assumptions about 198a6826fbcSWolfgang Denk * the type of the stored data objects at all, this implementation 199a6826fbcSWolfgang Denk * works with NUL terminated strings only. 200a6826fbcSWolfgang Denk * - Instead of storing just pointers to the original objects, we 201a6826fbcSWolfgang Denk * create local copies so the caller does not need to care about the 202a6826fbcSWolfgang Denk * data any more. 203a6826fbcSWolfgang Denk * - The standard implementation does not provide a way to update an 204a6826fbcSWolfgang Denk * existing entry. This version will create a new entry or update an 205a6826fbcSWolfgang Denk * existing one when both "action == ENTER" and "item.data != NULL". 206a6826fbcSWolfgang Denk * - Instead of returning 1 on success, we return the index into the 207a6826fbcSWolfgang Denk * internal hash table, which is also guaranteed to be positive. 208a6826fbcSWolfgang Denk * This allows us direct access to the found hash table slot for 209a6826fbcSWolfgang Denk * example for functions like hdelete(). 210a6826fbcSWolfgang Denk */ 211a6826fbcSWolfgang Denk 212a000b795SKim Phillips /* 213a000b795SKim Phillips * hstrstr_r - return index to entry whose key and/or data contains match 214a000b795SKim Phillips */ 215a000b795SKim Phillips int hstrstr_r(const char *match, int last_idx, ENTRY ** retval, 216a000b795SKim Phillips struct hsearch_data *htab) 217a000b795SKim Phillips { 218a000b795SKim Phillips unsigned int idx; 219a000b795SKim Phillips 220a000b795SKim Phillips for (idx = last_idx + 1; idx < htab->size; ++idx) { 221a000b795SKim Phillips if (htab->table[idx].used <= 0) 222a000b795SKim Phillips continue; 223a000b795SKim Phillips if (strstr(htab->table[idx].entry.key, match) || 224a000b795SKim Phillips strstr(htab->table[idx].entry.data, match)) { 225a000b795SKim Phillips *retval = &htab->table[idx].entry; 226a000b795SKim Phillips return idx; 227a000b795SKim Phillips } 228a000b795SKim Phillips } 229a000b795SKim Phillips 230a000b795SKim Phillips __set_errno(ESRCH); 231a000b795SKim Phillips *retval = NULL; 232a000b795SKim Phillips return 0; 233a000b795SKim Phillips } 234a000b795SKim Phillips 235560d424bSMike Frysinger int hmatch_r(const char *match, int last_idx, ENTRY ** retval, 236560d424bSMike Frysinger struct hsearch_data *htab) 237560d424bSMike Frysinger { 238560d424bSMike Frysinger unsigned int idx; 239560d424bSMike Frysinger size_t key_len = strlen(match); 240560d424bSMike Frysinger 241560d424bSMike Frysinger for (idx = last_idx + 1; idx < htab->size; ++idx) { 242af4d9074SKim Phillips if (htab->table[idx].used <= 0) 243560d424bSMike Frysinger continue; 244560d424bSMike Frysinger if (!strncmp(match, htab->table[idx].entry.key, key_len)) { 245560d424bSMike Frysinger *retval = &htab->table[idx].entry; 246560d424bSMike Frysinger return idx; 247560d424bSMike Frysinger } 248560d424bSMike Frysinger } 249560d424bSMike Frysinger 250560d424bSMike Frysinger __set_errno(ESRCH); 251560d424bSMike Frysinger *retval = NULL; 252560d424bSMike Frysinger return 0; 253560d424bSMike Frysinger } 254560d424bSMike Frysinger 2553d3b52f2SJoe Hershberger /* 2563d3b52f2SJoe Hershberger * Compare an existing entry with the desired key, and overwrite if the action 2573d3b52f2SJoe Hershberger * is ENTER. This is simply a helper function for hsearch_r(). 2583d3b52f2SJoe Hershberger */ 2593d3b52f2SJoe Hershberger static inline int _compare_and_overwrite_entry(ENTRY item, ACTION action, 2603d3b52f2SJoe Hershberger ENTRY **retval, struct hsearch_data *htab, int flag, 2613d3b52f2SJoe Hershberger unsigned int hval, unsigned int idx) 2623d3b52f2SJoe Hershberger { 2633d3b52f2SJoe Hershberger if (htab->table[idx].used == hval 2643d3b52f2SJoe Hershberger && strcmp(item.key, htab->table[idx].entry.key) == 0) { 2653d3b52f2SJoe Hershberger /* Overwrite existing value? */ 2663d3b52f2SJoe Hershberger if ((action == ENTER) && (item.data != NULL)) { 2677afcf3a5SJoe Hershberger /* check for permission */ 2687afcf3a5SJoe Hershberger if (htab->change_ok != NULL && htab->change_ok( 2697afcf3a5SJoe Hershberger &htab->table[idx].entry, item.data, 2707afcf3a5SJoe Hershberger env_op_overwrite, flag)) { 2717afcf3a5SJoe Hershberger debug("change_ok() rejected setting variable " 2727afcf3a5SJoe Hershberger "%s, skipping it!\n", item.key); 2737afcf3a5SJoe Hershberger __set_errno(EPERM); 2747afcf3a5SJoe Hershberger *retval = NULL; 2757afcf3a5SJoe Hershberger return 0; 2767afcf3a5SJoe Hershberger } 2777afcf3a5SJoe Hershberger 278*170ab110SJoe Hershberger /* If there is a callback, call it */ 279*170ab110SJoe Hershberger if (htab->table[idx].entry.callback && 280*170ab110SJoe Hershberger htab->table[idx].entry.callback(item.key, 281*170ab110SJoe Hershberger item.data, env_op_overwrite, flag)) { 282*170ab110SJoe Hershberger debug("callback() rejected setting variable " 283*170ab110SJoe Hershberger "%s, skipping it!\n", item.key); 284*170ab110SJoe Hershberger __set_errno(EINVAL); 285*170ab110SJoe Hershberger *retval = NULL; 286*170ab110SJoe Hershberger return 0; 287*170ab110SJoe Hershberger } 288*170ab110SJoe Hershberger 2893d3b52f2SJoe Hershberger free(htab->table[idx].entry.data); 2903d3b52f2SJoe Hershberger htab->table[idx].entry.data = strdup(item.data); 2913d3b52f2SJoe Hershberger if (!htab->table[idx].entry.data) { 2923d3b52f2SJoe Hershberger __set_errno(ENOMEM); 2933d3b52f2SJoe Hershberger *retval = NULL; 2943d3b52f2SJoe Hershberger return 0; 2953d3b52f2SJoe Hershberger } 2963d3b52f2SJoe Hershberger } 2973d3b52f2SJoe Hershberger /* return found entry */ 2983d3b52f2SJoe Hershberger *retval = &htab->table[idx].entry; 2993d3b52f2SJoe Hershberger return idx; 3003d3b52f2SJoe Hershberger } 3013d3b52f2SJoe Hershberger /* keep searching */ 3023d3b52f2SJoe Hershberger return -1; 3033d3b52f2SJoe Hershberger } 3043d3b52f2SJoe Hershberger 305a6826fbcSWolfgang Denk int hsearch_r(ENTRY item, ACTION action, ENTRY ** retval, 306c4e0057fSJoe Hershberger struct hsearch_data *htab, int flag) 307a6826fbcSWolfgang Denk { 308a6826fbcSWolfgang Denk unsigned int hval; 309a6826fbcSWolfgang Denk unsigned int count; 310a6826fbcSWolfgang Denk unsigned int len = strlen(item.key); 311a6826fbcSWolfgang Denk unsigned int idx; 312c81c1222SPeter Barada unsigned int first_deleted = 0; 3133d3b52f2SJoe Hershberger int ret; 314a6826fbcSWolfgang Denk 315a6826fbcSWolfgang Denk /* Compute an value for the given string. Perhaps use a better method. */ 316a6826fbcSWolfgang Denk hval = len; 317a6826fbcSWolfgang Denk count = len; 318a6826fbcSWolfgang Denk while (count-- > 0) { 319a6826fbcSWolfgang Denk hval <<= 4; 320a6826fbcSWolfgang Denk hval += item.key[count]; 321a6826fbcSWolfgang Denk } 322a6826fbcSWolfgang Denk 323a6826fbcSWolfgang Denk /* 324a6826fbcSWolfgang Denk * First hash function: 325a6826fbcSWolfgang Denk * simply take the modul but prevent zero. 326a6826fbcSWolfgang Denk */ 327a6826fbcSWolfgang Denk hval %= htab->size; 328a6826fbcSWolfgang Denk if (hval == 0) 329a6826fbcSWolfgang Denk ++hval; 330a6826fbcSWolfgang Denk 331a6826fbcSWolfgang Denk /* The first index tried. */ 332a6826fbcSWolfgang Denk idx = hval; 333a6826fbcSWolfgang Denk 334a6826fbcSWolfgang Denk if (htab->table[idx].used) { 335a6826fbcSWolfgang Denk /* 336a6826fbcSWolfgang Denk * Further action might be required according to the 337a6826fbcSWolfgang Denk * action value. 338a6826fbcSWolfgang Denk */ 339a6826fbcSWolfgang Denk unsigned hval2; 340a6826fbcSWolfgang Denk 341c81c1222SPeter Barada if (htab->table[idx].used == -1 342c81c1222SPeter Barada && !first_deleted) 343c81c1222SPeter Barada first_deleted = idx; 344c81c1222SPeter Barada 3453d3b52f2SJoe Hershberger ret = _compare_and_overwrite_entry(item, action, retval, htab, 3463d3b52f2SJoe Hershberger flag, hval, idx); 3473d3b52f2SJoe Hershberger if (ret != -1) 3483d3b52f2SJoe Hershberger return ret; 349a6826fbcSWolfgang Denk 350a6826fbcSWolfgang Denk /* 351a6826fbcSWolfgang Denk * Second hash function: 352a6826fbcSWolfgang Denk * as suggested in [Knuth] 353a6826fbcSWolfgang Denk */ 354a6826fbcSWolfgang Denk hval2 = 1 + hval % (htab->size - 2); 355a6826fbcSWolfgang Denk 356a6826fbcSWolfgang Denk do { 357a6826fbcSWolfgang Denk /* 358a6826fbcSWolfgang Denk * Because SIZE is prime this guarantees to 359a6826fbcSWolfgang Denk * step through all available indices. 360a6826fbcSWolfgang Denk */ 361a6826fbcSWolfgang Denk if (idx <= hval2) 362a6826fbcSWolfgang Denk idx = htab->size + idx - hval2; 363a6826fbcSWolfgang Denk else 364a6826fbcSWolfgang Denk idx -= hval2; 365a6826fbcSWolfgang Denk 366a6826fbcSWolfgang Denk /* 367a6826fbcSWolfgang Denk * If we visited all entries leave the loop 368a6826fbcSWolfgang Denk * unsuccessfully. 369a6826fbcSWolfgang Denk */ 370a6826fbcSWolfgang Denk if (idx == hval) 371a6826fbcSWolfgang Denk break; 372a6826fbcSWolfgang Denk 373a6826fbcSWolfgang Denk /* If entry is found use it. */ 3743d3b52f2SJoe Hershberger ret = _compare_and_overwrite_entry(item, action, retval, 3753d3b52f2SJoe Hershberger htab, flag, hval, idx); 3763d3b52f2SJoe Hershberger if (ret != -1) 3773d3b52f2SJoe Hershberger return ret; 378a6826fbcSWolfgang Denk } 379a6826fbcSWolfgang Denk while (htab->table[idx].used); 380a6826fbcSWolfgang Denk } 381a6826fbcSWolfgang Denk 382a6826fbcSWolfgang Denk /* An empty bucket has been found. */ 383a6826fbcSWolfgang Denk if (action == ENTER) { 384a6826fbcSWolfgang Denk /* 385a6826fbcSWolfgang Denk * If table is full and another entry should be 386a6826fbcSWolfgang Denk * entered return with error. 387a6826fbcSWolfgang Denk */ 388a6826fbcSWolfgang Denk if (htab->filled == htab->size) { 389a6826fbcSWolfgang Denk __set_errno(ENOMEM); 390a6826fbcSWolfgang Denk *retval = NULL; 391a6826fbcSWolfgang Denk return 0; 392a6826fbcSWolfgang Denk } 393a6826fbcSWolfgang Denk 394a6826fbcSWolfgang Denk /* 395a6826fbcSWolfgang Denk * Create new entry; 396a6826fbcSWolfgang Denk * create copies of item.key and item.data 397a6826fbcSWolfgang Denk */ 398c81c1222SPeter Barada if (first_deleted) 399c81c1222SPeter Barada idx = first_deleted; 400c81c1222SPeter Barada 401a6826fbcSWolfgang Denk htab->table[idx].used = hval; 402a6826fbcSWolfgang Denk htab->table[idx].entry.key = strdup(item.key); 403a6826fbcSWolfgang Denk htab->table[idx].entry.data = strdup(item.data); 404a6826fbcSWolfgang Denk if (!htab->table[idx].entry.key || 405a6826fbcSWolfgang Denk !htab->table[idx].entry.data) { 406a6826fbcSWolfgang Denk __set_errno(ENOMEM); 407a6826fbcSWolfgang Denk *retval = NULL; 408a6826fbcSWolfgang Denk return 0; 409a6826fbcSWolfgang Denk } 410a6826fbcSWolfgang Denk 411a6826fbcSWolfgang Denk ++htab->filled; 412a6826fbcSWolfgang Denk 413*170ab110SJoe Hershberger /* This is a new entry, so look up a possible callback */ 414*170ab110SJoe Hershberger env_callback_init(&htab->table[idx].entry); 415*170ab110SJoe Hershberger 4167afcf3a5SJoe Hershberger /* check for permission */ 4177afcf3a5SJoe Hershberger if (htab->change_ok != NULL && htab->change_ok( 4187afcf3a5SJoe Hershberger &htab->table[idx].entry, item.data, env_op_create, flag)) { 4197afcf3a5SJoe Hershberger debug("change_ok() rejected setting variable " 4207afcf3a5SJoe Hershberger "%s, skipping it!\n", item.key); 4217afcf3a5SJoe Hershberger _hdelete(item.key, htab, &htab->table[idx].entry, idx); 4227afcf3a5SJoe Hershberger __set_errno(EPERM); 4237afcf3a5SJoe Hershberger *retval = NULL; 4247afcf3a5SJoe Hershberger return 0; 4257afcf3a5SJoe Hershberger } 4267afcf3a5SJoe Hershberger 427*170ab110SJoe Hershberger /* If there is a callback, call it */ 428*170ab110SJoe Hershberger if (htab->table[idx].entry.callback && 429*170ab110SJoe Hershberger htab->table[idx].entry.callback(item.key, item.data, 430*170ab110SJoe Hershberger env_op_create, flag)) { 431*170ab110SJoe Hershberger debug("callback() rejected setting variable " 432*170ab110SJoe Hershberger "%s, skipping it!\n", item.key); 433*170ab110SJoe Hershberger _hdelete(item.key, htab, &htab->table[idx].entry, idx); 434*170ab110SJoe Hershberger __set_errno(EINVAL); 435*170ab110SJoe Hershberger *retval = NULL; 436*170ab110SJoe Hershberger return 0; 437*170ab110SJoe Hershberger } 438*170ab110SJoe Hershberger 439a6826fbcSWolfgang Denk /* return new entry */ 440a6826fbcSWolfgang Denk *retval = &htab->table[idx].entry; 441a6826fbcSWolfgang Denk return 1; 442a6826fbcSWolfgang Denk } 443a6826fbcSWolfgang Denk 444a6826fbcSWolfgang Denk __set_errno(ESRCH); 445a6826fbcSWolfgang Denk *retval = NULL; 446a6826fbcSWolfgang Denk return 0; 447a6826fbcSWolfgang Denk } 448a6826fbcSWolfgang Denk 449a6826fbcSWolfgang Denk 450a6826fbcSWolfgang Denk /* 451a6826fbcSWolfgang Denk * hdelete() 452a6826fbcSWolfgang Denk */ 453a6826fbcSWolfgang Denk 454a6826fbcSWolfgang Denk /* 455a6826fbcSWolfgang Denk * The standard implementation of hsearch(3) does not provide any way 456a6826fbcSWolfgang Denk * to delete any entries from the hash table. We extend the code to 457a6826fbcSWolfgang Denk * do that. 458a6826fbcSWolfgang Denk */ 459a6826fbcSWolfgang Denk 4607afcf3a5SJoe Hershberger static void _hdelete(const char *key, struct hsearch_data *htab, ENTRY *ep, 4617afcf3a5SJoe Hershberger int idx) 4627afcf3a5SJoe Hershberger { 4637afcf3a5SJoe Hershberger /* free used ENTRY */ 4647afcf3a5SJoe Hershberger debug("hdelete: DELETING key \"%s\"\n", key); 4657afcf3a5SJoe Hershberger free((void *)ep->key); 4667afcf3a5SJoe Hershberger free(ep->data); 467*170ab110SJoe Hershberger ep->callback = NULL; 4687afcf3a5SJoe Hershberger htab->table[idx].used = -1; 4697afcf3a5SJoe Hershberger 4707afcf3a5SJoe Hershberger --htab->filled; 4717afcf3a5SJoe Hershberger } 4727afcf3a5SJoe Hershberger 473c4e0057fSJoe Hershberger int hdelete_r(const char *key, struct hsearch_data *htab, int flag) 474a6826fbcSWolfgang Denk { 475a6826fbcSWolfgang Denk ENTRY e, *ep; 476a6826fbcSWolfgang Denk int idx; 477a6826fbcSWolfgang Denk 478a6826fbcSWolfgang Denk debug("hdelete: DELETE key \"%s\"\n", key); 479a6826fbcSWolfgang Denk 480a6826fbcSWolfgang Denk e.key = (char *)key; 481a6826fbcSWolfgang Denk 482c4e0057fSJoe Hershberger idx = hsearch_r(e, FIND, &ep, htab, 0); 483c4e0057fSJoe Hershberger if (idx == 0) { 484a6826fbcSWolfgang Denk __set_errno(ESRCH); 485a6826fbcSWolfgang Denk return 0; /* not found */ 486a6826fbcSWolfgang Denk } 487a6826fbcSWolfgang Denk 488c4e0057fSJoe Hershberger /* Check for permission */ 4897afcf3a5SJoe Hershberger if (htab->change_ok != NULL && 4907afcf3a5SJoe Hershberger htab->change_ok(ep, NULL, env_op_delete, flag)) { 4917afcf3a5SJoe Hershberger debug("change_ok() rejected deleting variable " 4927afcf3a5SJoe Hershberger "%s, skipping it!\n", key); 493c4e0057fSJoe Hershberger __set_errno(EPERM); 494c4e0057fSJoe Hershberger return 0; 495c4e0057fSJoe Hershberger } 496c4e0057fSJoe Hershberger 497*170ab110SJoe Hershberger /* If there is a callback, call it */ 498*170ab110SJoe Hershberger if (htab->table[idx].entry.callback && 499*170ab110SJoe Hershberger htab->table[idx].entry.callback(key, NULL, env_op_delete, flag)) { 500*170ab110SJoe Hershberger debug("callback() rejected deleting variable " 501*170ab110SJoe Hershberger "%s, skipping it!\n", key); 502*170ab110SJoe Hershberger __set_errno(EINVAL); 503*170ab110SJoe Hershberger return 0; 504*170ab110SJoe Hershberger } 505*170ab110SJoe Hershberger 5067afcf3a5SJoe Hershberger _hdelete(key, htab, ep, idx); 507a6826fbcSWolfgang Denk 508a6826fbcSWolfgang Denk return 1; 509a6826fbcSWolfgang Denk } 510a6826fbcSWolfgang Denk 511a6826fbcSWolfgang Denk /* 512a6826fbcSWolfgang Denk * hexport() 513a6826fbcSWolfgang Denk */ 514a6826fbcSWolfgang Denk 5157ac2fe2dSIlya Yanok #ifndef CONFIG_SPL_BUILD 516a6826fbcSWolfgang Denk /* 517a6826fbcSWolfgang Denk * Export the data stored in the hash table in linearized form. 518a6826fbcSWolfgang Denk * 519a6826fbcSWolfgang Denk * Entries are exported as "name=value" strings, separated by an 520a6826fbcSWolfgang Denk * arbitrary (non-NUL, of course) separator character. This allows to 521a6826fbcSWolfgang Denk * use this function both when formatting the U-Boot environment for 522a6826fbcSWolfgang Denk * external storage (using '\0' as separator), but also when using it 523a6826fbcSWolfgang Denk * for the "printenv" command to print all variables, simply by using 524a6826fbcSWolfgang Denk * as '\n" as separator. This can also be used for new features like 525a6826fbcSWolfgang Denk * exporting the environment data as text file, including the option 526a6826fbcSWolfgang Denk * for later re-import. 527a6826fbcSWolfgang Denk * 528a6826fbcSWolfgang Denk * The entries in the result list will be sorted by ascending key 529a6826fbcSWolfgang Denk * values. 530a6826fbcSWolfgang Denk * 531a6826fbcSWolfgang Denk * If the separator character is different from NUL, then any 532a6826fbcSWolfgang Denk * separator characters and backslash characters in the values will 533a6826fbcSWolfgang Denk * be escaped by a preceeding backslash in output. This is needed for 534a6826fbcSWolfgang Denk * example to enable multi-line values, especially when the output 535a6826fbcSWolfgang Denk * shall later be parsed (for example, for re-import). 536a6826fbcSWolfgang Denk * 537a6826fbcSWolfgang Denk * There are several options how the result buffer is handled: 538a6826fbcSWolfgang Denk * 539a6826fbcSWolfgang Denk * *resp size 540a6826fbcSWolfgang Denk * ----------- 541a6826fbcSWolfgang Denk * NULL 0 A string of sufficient length will be allocated. 542a6826fbcSWolfgang Denk * NULL >0 A string of the size given will be 543a6826fbcSWolfgang Denk * allocated. An error will be returned if the size is 544a6826fbcSWolfgang Denk * not sufficient. Any unused bytes in the string will 545a6826fbcSWolfgang Denk * be '\0'-padded. 546a6826fbcSWolfgang Denk * !NULL 0 The user-supplied buffer will be used. No length 547a6826fbcSWolfgang Denk * checking will be performed, i. e. it is assumed that 548a6826fbcSWolfgang Denk * the buffer size will always be big enough. DANGEROUS. 549a6826fbcSWolfgang Denk * !NULL >0 The user-supplied buffer will be used. An error will 550a6826fbcSWolfgang Denk * be returned if the size is not sufficient. Any unused 551a6826fbcSWolfgang Denk * bytes in the string will be '\0'-padded. 552a6826fbcSWolfgang Denk */ 553a6826fbcSWolfgang Denk 554a6826fbcSWolfgang Denk static int cmpkey(const void *p1, const void *p2) 555a6826fbcSWolfgang Denk { 556a6826fbcSWolfgang Denk ENTRY *e1 = *(ENTRY **) p1; 557a6826fbcSWolfgang Denk ENTRY *e2 = *(ENTRY **) p2; 558a6826fbcSWolfgang Denk 559a6826fbcSWolfgang Denk return (strcmp(e1->key, e2->key)); 560a6826fbcSWolfgang Denk } 561a6826fbcSWolfgang Denk 562be11235aSJoe Hershberger ssize_t hexport_r(struct hsearch_data *htab, const char sep, int flag, 56337f2fe74SWolfgang Denk char **resp, size_t size, 56437f2fe74SWolfgang Denk int argc, char * const argv[]) 565a6826fbcSWolfgang Denk { 566a6826fbcSWolfgang Denk ENTRY *list[htab->size]; 567a6826fbcSWolfgang Denk char *res, *p; 568a6826fbcSWolfgang Denk size_t totlen; 569a6826fbcSWolfgang Denk int i, n; 570a6826fbcSWolfgang Denk 571a6826fbcSWolfgang Denk /* Test for correct arguments. */ 572a6826fbcSWolfgang Denk if ((resp == NULL) || (htab == NULL)) { 573a6826fbcSWolfgang Denk __set_errno(EINVAL); 574a6826fbcSWolfgang Denk return (-1); 575a6826fbcSWolfgang Denk } 576a6826fbcSWolfgang Denk 577ff856286SSimon Glass debug("EXPORT table = %p, htab.size = %d, htab.filled = %d, " 578ff856286SSimon Glass "size = %zu\n", htab, htab->size, htab->filled, size); 579a6826fbcSWolfgang Denk /* 580a6826fbcSWolfgang Denk * Pass 1: 581a6826fbcSWolfgang Denk * search used entries, 582a6826fbcSWolfgang Denk * save addresses and compute total length 583a6826fbcSWolfgang Denk */ 584a6826fbcSWolfgang Denk for (i = 1, n = 0, totlen = 0; i <= htab->size; ++i) { 585a6826fbcSWolfgang Denk 586c81c1222SPeter Barada if (htab->table[i].used > 0) { 587a6826fbcSWolfgang Denk ENTRY *ep = &htab->table[i].entry; 58837f2fe74SWolfgang Denk int arg, found = 0; 58937f2fe74SWolfgang Denk 59037f2fe74SWolfgang Denk for (arg = 0; arg < argc; ++arg) { 59137f2fe74SWolfgang Denk if (strcmp(argv[arg], ep->key) == 0) { 59237f2fe74SWolfgang Denk found = 1; 59337f2fe74SWolfgang Denk break; 59437f2fe74SWolfgang Denk } 59537f2fe74SWolfgang Denk } 59637f2fe74SWolfgang Denk if ((argc > 0) && (found == 0)) 59737f2fe74SWolfgang Denk continue; 598a6826fbcSWolfgang Denk 599be11235aSJoe Hershberger if ((flag & H_HIDE_DOT) && ep->key[0] == '.') 600be11235aSJoe Hershberger continue; 601be11235aSJoe Hershberger 602a6826fbcSWolfgang Denk list[n++] = ep; 603a6826fbcSWolfgang Denk 604a6826fbcSWolfgang Denk totlen += strlen(ep->key) + 2; 605a6826fbcSWolfgang Denk 606a6826fbcSWolfgang Denk if (sep == '\0') { 607a6826fbcSWolfgang Denk totlen += strlen(ep->data); 608a6826fbcSWolfgang Denk } else { /* check if escapes are needed */ 609a6826fbcSWolfgang Denk char *s = ep->data; 610a6826fbcSWolfgang Denk 611a6826fbcSWolfgang Denk while (*s) { 612a6826fbcSWolfgang Denk ++totlen; 613a6826fbcSWolfgang Denk /* add room for needed escape chars */ 614a6826fbcSWolfgang Denk if ((*s == sep) || (*s == '\\')) 615a6826fbcSWolfgang Denk ++totlen; 616a6826fbcSWolfgang Denk ++s; 617a6826fbcSWolfgang Denk } 618a6826fbcSWolfgang Denk } 619a6826fbcSWolfgang Denk totlen += 2; /* for '=' and 'sep' char */ 620a6826fbcSWolfgang Denk } 621a6826fbcSWolfgang Denk } 622a6826fbcSWolfgang Denk 623a6826fbcSWolfgang Denk #ifdef DEBUG 624a6826fbcSWolfgang Denk /* Pass 1a: print unsorted list */ 625a6826fbcSWolfgang Denk printf("Unsorted: n=%d\n", n); 626a6826fbcSWolfgang Denk for (i = 0; i < n; ++i) { 627a6826fbcSWolfgang Denk printf("\t%3d: %p ==> %-10s => %s\n", 628a6826fbcSWolfgang Denk i, list[i], list[i]->key, list[i]->data); 629a6826fbcSWolfgang Denk } 630a6826fbcSWolfgang Denk #endif 631a6826fbcSWolfgang Denk 632a6826fbcSWolfgang Denk /* Sort list by keys */ 633a6826fbcSWolfgang Denk qsort(list, n, sizeof(ENTRY *), cmpkey); 634a6826fbcSWolfgang Denk 635a6826fbcSWolfgang Denk /* Check if the user supplied buffer size is sufficient */ 636a6826fbcSWolfgang Denk if (size) { 637a6826fbcSWolfgang Denk if (size < totlen + 1) { /* provided buffer too small */ 638ff856286SSimon Glass printf("Env export buffer too small: %zu, " 639ff856286SSimon Glass "but need %zu\n", size, totlen + 1); 640a6826fbcSWolfgang Denk __set_errno(ENOMEM); 641a6826fbcSWolfgang Denk return (-1); 642a6826fbcSWolfgang Denk } 643a6826fbcSWolfgang Denk } else { 644a6826fbcSWolfgang Denk size = totlen + 1; 645a6826fbcSWolfgang Denk } 646a6826fbcSWolfgang Denk 647a6826fbcSWolfgang Denk /* Check if the user provided a buffer */ 648a6826fbcSWolfgang Denk if (*resp) { 649a6826fbcSWolfgang Denk /* yes; clear it */ 650a6826fbcSWolfgang Denk res = *resp; 651a6826fbcSWolfgang Denk memset(res, '\0', size); 652a6826fbcSWolfgang Denk } else { 653a6826fbcSWolfgang Denk /* no, allocate and clear one */ 654a6826fbcSWolfgang Denk *resp = res = calloc(1, size); 655a6826fbcSWolfgang Denk if (res == NULL) { 656a6826fbcSWolfgang Denk __set_errno(ENOMEM); 657a6826fbcSWolfgang Denk return (-1); 658a6826fbcSWolfgang Denk } 659a6826fbcSWolfgang Denk } 660a6826fbcSWolfgang Denk /* 661a6826fbcSWolfgang Denk * Pass 2: 662a6826fbcSWolfgang Denk * export sorted list of result data 663a6826fbcSWolfgang Denk */ 664a6826fbcSWolfgang Denk for (i = 0, p = res; i < n; ++i) { 66584b5e802SWolfgang Denk const char *s; 666a6826fbcSWolfgang Denk 667a6826fbcSWolfgang Denk s = list[i]->key; 668a6826fbcSWolfgang Denk while (*s) 669a6826fbcSWolfgang Denk *p++ = *s++; 670a6826fbcSWolfgang Denk *p++ = '='; 671a6826fbcSWolfgang Denk 672a6826fbcSWolfgang Denk s = list[i]->data; 673a6826fbcSWolfgang Denk 674a6826fbcSWolfgang Denk while (*s) { 675a6826fbcSWolfgang Denk if ((*s == sep) || (*s == '\\')) 676a6826fbcSWolfgang Denk *p++ = '\\'; /* escape */ 677a6826fbcSWolfgang Denk *p++ = *s++; 678a6826fbcSWolfgang Denk } 679a6826fbcSWolfgang Denk *p++ = sep; 680a6826fbcSWolfgang Denk } 681a6826fbcSWolfgang Denk *p = '\0'; /* terminate result */ 682a6826fbcSWolfgang Denk 683a6826fbcSWolfgang Denk return size; 684a6826fbcSWolfgang Denk } 6857ac2fe2dSIlya Yanok #endif 686a6826fbcSWolfgang Denk 687a6826fbcSWolfgang Denk 688a6826fbcSWolfgang Denk /* 689a6826fbcSWolfgang Denk * himport() 690a6826fbcSWolfgang Denk */ 691a6826fbcSWolfgang Denk 692d5370febSGerlando Falauto /* 693d5370febSGerlando Falauto * Check whether variable 'name' is amongst vars[], 694d5370febSGerlando Falauto * and remove all instances by setting the pointer to NULL 695d5370febSGerlando Falauto */ 696d5370febSGerlando Falauto static int drop_var_from_set(const char *name, int nvars, char * vars[]) 697348b1f1cSGerlando Falauto { 698348b1f1cSGerlando Falauto int i = 0; 699d5370febSGerlando Falauto int res = 0; 700348b1f1cSGerlando Falauto 701348b1f1cSGerlando Falauto /* No variables specified means process all of them */ 702348b1f1cSGerlando Falauto if (nvars == 0) 703348b1f1cSGerlando Falauto return 1; 704348b1f1cSGerlando Falauto 705348b1f1cSGerlando Falauto for (i = 0; i < nvars; i++) { 706d5370febSGerlando Falauto if (vars[i] == NULL) 707d5370febSGerlando Falauto continue; 708d5370febSGerlando Falauto /* If we found it, delete all of them */ 709d5370febSGerlando Falauto if (!strcmp(name, vars[i])) { 710d5370febSGerlando Falauto vars[i] = NULL; 711d5370febSGerlando Falauto res = 1; 712348b1f1cSGerlando Falauto } 713d5370febSGerlando Falauto } 714d5370febSGerlando Falauto if (!res) 715348b1f1cSGerlando Falauto debug("Skipping non-listed variable %s\n", name); 716348b1f1cSGerlando Falauto 717d5370febSGerlando Falauto return res; 718348b1f1cSGerlando Falauto } 719348b1f1cSGerlando Falauto 720a6826fbcSWolfgang Denk /* 721a6826fbcSWolfgang Denk * Import linearized data into hash table. 722a6826fbcSWolfgang Denk * 723a6826fbcSWolfgang Denk * This is the inverse function to hexport(): it takes a linear list 724a6826fbcSWolfgang Denk * of "name=value" pairs and creates hash table entries from it. 725a6826fbcSWolfgang Denk * 726a6826fbcSWolfgang Denk * Entries without "value", i. e. consisting of only "name" or 727a6826fbcSWolfgang Denk * "name=", will cause this entry to be deleted from the hash table. 728a6826fbcSWolfgang Denk * 729a6826fbcSWolfgang Denk * The "flag" argument can be used to control the behaviour: when the 730a6826fbcSWolfgang Denk * H_NOCLEAR bit is set, then an existing hash table will kept, i. e. 731a6826fbcSWolfgang Denk * new data will be added to an existing hash table; otherwise, old 732a6826fbcSWolfgang Denk * data will be discarded and a new hash table will be created. 733a6826fbcSWolfgang Denk * 734a6826fbcSWolfgang Denk * The separator character for the "name=value" pairs can be selected, 735a6826fbcSWolfgang Denk * so we both support importing from externally stored environment 736a6826fbcSWolfgang Denk * data (separated by NUL characters) and from plain text files 737a6826fbcSWolfgang Denk * (entries separated by newline characters). 738a6826fbcSWolfgang Denk * 739a6826fbcSWolfgang Denk * To allow for nicely formatted text input, leading white space 740a6826fbcSWolfgang Denk * (sequences of SPACE and TAB chars) is ignored, and entries starting 741a6826fbcSWolfgang Denk * (after removal of any leading white space) with a '#' character are 742a6826fbcSWolfgang Denk * considered comments and ignored. 743a6826fbcSWolfgang Denk * 744a6826fbcSWolfgang Denk * [NOTE: this means that a variable name cannot start with a '#' 745a6826fbcSWolfgang Denk * character.] 746a6826fbcSWolfgang Denk * 747a6826fbcSWolfgang Denk * When using a non-NUL separator character, backslash is used as 748a6826fbcSWolfgang Denk * escape character in the value part, allowing for example for 749a6826fbcSWolfgang Denk * multi-line values. 750a6826fbcSWolfgang Denk * 751a6826fbcSWolfgang Denk * In theory, arbitrary separator characters can be used, but only 752a6826fbcSWolfgang Denk * '\0' and '\n' have really been tested. 753a6826fbcSWolfgang Denk */ 754a6826fbcSWolfgang Denk 755a6826fbcSWolfgang Denk int himport_r(struct hsearch_data *htab, 756348b1f1cSGerlando Falauto const char *env, size_t size, const char sep, int flag, 757c4e0057fSJoe Hershberger int nvars, char * const vars[]) 758a6826fbcSWolfgang Denk { 759a6826fbcSWolfgang Denk char *data, *sp, *dp, *name, *value; 760d5370febSGerlando Falauto char *localvars[nvars]; 761d5370febSGerlando Falauto int i; 762a6826fbcSWolfgang Denk 763a6826fbcSWolfgang Denk /* Test for correct arguments. */ 764a6826fbcSWolfgang Denk if (htab == NULL) { 765a6826fbcSWolfgang Denk __set_errno(EINVAL); 766a6826fbcSWolfgang Denk return 0; 767a6826fbcSWolfgang Denk } 768a6826fbcSWolfgang Denk 769a6826fbcSWolfgang Denk /* we allocate new space to make sure we can write to the array */ 770a6826fbcSWolfgang Denk if ((data = malloc(size)) == NULL) { 771ff856286SSimon Glass debug("himport_r: can't malloc %zu bytes\n", size); 772a6826fbcSWolfgang Denk __set_errno(ENOMEM); 773a6826fbcSWolfgang Denk return 0; 774a6826fbcSWolfgang Denk } 775a6826fbcSWolfgang Denk memcpy(data, env, size); 776a6826fbcSWolfgang Denk dp = data; 777a6826fbcSWolfgang Denk 778d5370febSGerlando Falauto /* make a local copy of the list of variables */ 779d5370febSGerlando Falauto if (nvars) 780d5370febSGerlando Falauto memcpy(localvars, vars, sizeof(vars[0]) * nvars); 781d5370febSGerlando Falauto 782a6826fbcSWolfgang Denk if ((flag & H_NOCLEAR) == 0) { 783a6826fbcSWolfgang Denk /* Destroy old hash table if one exists */ 784a6826fbcSWolfgang Denk debug("Destroy Hash Table: %p table = %p\n", htab, 785a6826fbcSWolfgang Denk htab->table); 786a6826fbcSWolfgang Denk if (htab->table) 787c4e0057fSJoe Hershberger hdestroy_r(htab); 788a6826fbcSWolfgang Denk } 789a6826fbcSWolfgang Denk 790a6826fbcSWolfgang Denk /* 791a6826fbcSWolfgang Denk * Create new hash table (if needed). The computation of the hash 792a6826fbcSWolfgang Denk * table size is based on heuristics: in a sample of some 70+ 793a6826fbcSWolfgang Denk * existing systems we found an average size of 39+ bytes per entry 794a6826fbcSWolfgang Denk * in the environment (for the whole key=value pair). Assuming a 795ea882bafSWolfgang Denk * size of 8 per entry (= safety factor of ~5) should provide enough 796ea882bafSWolfgang Denk * safety margin for any existing environment definitions and still 797a6826fbcSWolfgang Denk * allow for more than enough dynamic additions. Note that the 798a6826fbcSWolfgang Denk * "size" argument is supposed to give the maximum enviroment size 799ea882bafSWolfgang Denk * (CONFIG_ENV_SIZE). This heuristics will result in 800ea882bafSWolfgang Denk * unreasonably large numbers (and thus memory footprint) for 801ea882bafSWolfgang Denk * big flash environments (>8,000 entries for 64 KB 802fc5fc76bSAndreas Bießmann * envrionment size), so we clip it to a reasonable value. 803fc5fc76bSAndreas Bießmann * On the other hand we need to add some more entries for free 804fc5fc76bSAndreas Bießmann * space when importing very small buffers. Both boundaries can 805fc5fc76bSAndreas Bießmann * be overwritten in the board config file if needed. 806a6826fbcSWolfgang Denk */ 807a6826fbcSWolfgang Denk 808a6826fbcSWolfgang Denk if (!htab->table) { 809fc5fc76bSAndreas Bießmann int nent = CONFIG_ENV_MIN_ENTRIES + size / 8; 810ea882bafSWolfgang Denk 811ea882bafSWolfgang Denk if (nent > CONFIG_ENV_MAX_ENTRIES) 812ea882bafSWolfgang Denk nent = CONFIG_ENV_MAX_ENTRIES; 813a6826fbcSWolfgang Denk 814a6826fbcSWolfgang Denk debug("Create Hash Table: N=%d\n", nent); 815a6826fbcSWolfgang Denk 816a6826fbcSWolfgang Denk if (hcreate_r(nent, htab) == 0) { 817a6826fbcSWolfgang Denk free(data); 818a6826fbcSWolfgang Denk return 0; 819a6826fbcSWolfgang Denk } 820a6826fbcSWolfgang Denk } 821a6826fbcSWolfgang Denk 822a6826fbcSWolfgang Denk /* Parse environment; allow for '\0' and 'sep' as separators */ 823a6826fbcSWolfgang Denk do { 824a6826fbcSWolfgang Denk ENTRY e, *rv; 825a6826fbcSWolfgang Denk 826a6826fbcSWolfgang Denk /* skip leading white space */ 8274d91a6ecSJason Hobbs while (isblank(*dp)) 828a6826fbcSWolfgang Denk ++dp; 829a6826fbcSWolfgang Denk 830a6826fbcSWolfgang Denk /* skip comment lines */ 831a6826fbcSWolfgang Denk if (*dp == '#') { 832a6826fbcSWolfgang Denk while (*dp && (*dp != sep)) 833a6826fbcSWolfgang Denk ++dp; 834a6826fbcSWolfgang Denk ++dp; 835a6826fbcSWolfgang Denk continue; 836a6826fbcSWolfgang Denk } 837a6826fbcSWolfgang Denk 838a6826fbcSWolfgang Denk /* parse name */ 839a6826fbcSWolfgang Denk for (name = dp; *dp != '=' && *dp && *dp != sep; ++dp) 840a6826fbcSWolfgang Denk ; 841a6826fbcSWolfgang Denk 842a6826fbcSWolfgang Denk /* deal with "name" and "name=" entries (delete var) */ 843a6826fbcSWolfgang Denk if (*dp == '\0' || *(dp + 1) == '\0' || 844a6826fbcSWolfgang Denk *dp == sep || *(dp + 1) == sep) { 845a6826fbcSWolfgang Denk if (*dp == '=') 846a6826fbcSWolfgang Denk *dp++ = '\0'; 847a6826fbcSWolfgang Denk *dp++ = '\0'; /* terminate name */ 848a6826fbcSWolfgang Denk 849a6826fbcSWolfgang Denk debug("DELETE CANDIDATE: \"%s\"\n", name); 850d5370febSGerlando Falauto if (!drop_var_from_set(name, nvars, localvars)) 851348b1f1cSGerlando Falauto continue; 852a6826fbcSWolfgang Denk 853c4e0057fSJoe Hershberger if (hdelete_r(name, htab, flag) == 0) 854a6826fbcSWolfgang Denk debug("DELETE ERROR ##############################\n"); 855a6826fbcSWolfgang Denk 856a6826fbcSWolfgang Denk continue; 857a6826fbcSWolfgang Denk } 858a6826fbcSWolfgang Denk *dp++ = '\0'; /* terminate name */ 859a6826fbcSWolfgang Denk 860a6826fbcSWolfgang Denk /* parse value; deal with escapes */ 861a6826fbcSWolfgang Denk for (value = sp = dp; *dp && (*dp != sep); ++dp) { 862a6826fbcSWolfgang Denk if ((*dp == '\\') && *(dp + 1)) 863a6826fbcSWolfgang Denk ++dp; 864a6826fbcSWolfgang Denk *sp++ = *dp; 865a6826fbcSWolfgang Denk } 866a6826fbcSWolfgang Denk *sp++ = '\0'; /* terminate value */ 867a6826fbcSWolfgang Denk ++dp; 868a6826fbcSWolfgang Denk 869348b1f1cSGerlando Falauto /* Skip variables which are not supposed to be processed */ 870d5370febSGerlando Falauto if (!drop_var_from_set(name, nvars, localvars)) 871348b1f1cSGerlando Falauto continue; 872348b1f1cSGerlando Falauto 873a6826fbcSWolfgang Denk /* enter into hash table */ 874a6826fbcSWolfgang Denk e.key = name; 875a6826fbcSWolfgang Denk e.data = value; 876a6826fbcSWolfgang Denk 877c4e0057fSJoe Hershberger hsearch_r(e, ENTER, &rv, htab, flag); 878*170ab110SJoe Hershberger if (rv == NULL) 879ea882bafSWolfgang Denk printf("himport_r: can't insert \"%s=%s\" into hash table\n", 880ea882bafSWolfgang Denk name, value); 881a6826fbcSWolfgang Denk 882ea882bafSWolfgang Denk debug("INSERT: table %p, filled %d/%d rv %p ==> name=\"%s\" value=\"%s\"\n", 883ea882bafSWolfgang Denk htab, htab->filled, htab->size, 884ea882bafSWolfgang Denk rv, name, value); 885a6826fbcSWolfgang Denk } while ((dp < data + size) && *dp); /* size check needed for text */ 886a6826fbcSWolfgang Denk /* without '\0' termination */ 887ea882bafSWolfgang Denk debug("INSERT: free(data = %p)\n", data); 888a6826fbcSWolfgang Denk free(data); 889a6826fbcSWolfgang Denk 890d5370febSGerlando Falauto /* process variables which were not considered */ 891d5370febSGerlando Falauto for (i = 0; i < nvars; i++) { 892d5370febSGerlando Falauto if (localvars[i] == NULL) 893d5370febSGerlando Falauto continue; 894d5370febSGerlando Falauto /* 895d5370febSGerlando Falauto * All variables which were not deleted from the variable list 896d5370febSGerlando Falauto * were not present in the imported env 897d5370febSGerlando Falauto * This could mean two things: 898d5370febSGerlando Falauto * a) if the variable was present in current env, we delete it 899d5370febSGerlando Falauto * b) if the variable was not present in current env, we notify 900d5370febSGerlando Falauto * it might be a typo 901d5370febSGerlando Falauto */ 902c4e0057fSJoe Hershberger if (hdelete_r(localvars[i], htab, flag) == 0) 903d5370febSGerlando Falauto printf("WARNING: '%s' neither in running nor in imported env!\n", localvars[i]); 904d5370febSGerlando Falauto else 905d5370febSGerlando Falauto printf("WARNING: '%s' not in imported env, deleting it!\n", localvars[i]); 906d5370febSGerlando Falauto } 907d5370febSGerlando Falauto 908ea882bafSWolfgang Denk debug("INSERT: done\n"); 909a6826fbcSWolfgang Denk return 1; /* everything OK */ 910a6826fbcSWolfgang Denk } 911*170ab110SJoe Hershberger 912*170ab110SJoe Hershberger /* 913*170ab110SJoe Hershberger * hwalk_r() 914*170ab110SJoe Hershberger */ 915*170ab110SJoe Hershberger 916*170ab110SJoe Hershberger /* 917*170ab110SJoe Hershberger * Walk all of the entries in the hash, calling the callback for each one. 918*170ab110SJoe Hershberger * this allows some generic operation to be performed on each element. 919*170ab110SJoe Hershberger */ 920*170ab110SJoe Hershberger int hwalk_r(struct hsearch_data *htab, int (*callback)(ENTRY *)) 921*170ab110SJoe Hershberger { 922*170ab110SJoe Hershberger int i; 923*170ab110SJoe Hershberger int retval; 924*170ab110SJoe Hershberger 925*170ab110SJoe Hershberger for (i = 1; i <= htab->size; ++i) { 926*170ab110SJoe Hershberger if (htab->table[i].used > 0) { 927*170ab110SJoe Hershberger retval = callback(&htab->table[i].entry); 928*170ab110SJoe Hershberger if (retval) 929*170ab110SJoe Hershberger return retval; 930*170ab110SJoe Hershberger } 931*170ab110SJoe Hershberger } 932*170ab110SJoe Hershberger 933*170ab110SJoe Hershberger return 0; 934*170ab110SJoe Hershberger } 935