Annotation of embedaddon/sqlite3/src/hash.c, revision 1.1
1.1 ! misho 1: /*
! 2: ** 2001 September 22
! 3: **
! 4: ** The author disclaims copyright to this source code. In place of
! 5: ** a legal notice, here is a blessing:
! 6: **
! 7: ** May you do good and not evil.
! 8: ** May you find forgiveness for yourself and forgive others.
! 9: ** May you share freely, never taking more than you give.
! 10: **
! 11: *************************************************************************
! 12: ** This is the implementation of generic hash-tables
! 13: ** used in SQLite.
! 14: */
! 15: #include "sqliteInt.h"
! 16: #include <assert.h>
! 17:
! 18: /* Turn bulk memory into a hash table object by initializing the
! 19: ** fields of the Hash structure.
! 20: **
! 21: ** "pNew" is a pointer to the hash table that is to be initialized.
! 22: */
! 23: void sqlite3HashInit(Hash *pNew){
! 24: assert( pNew!=0 );
! 25: pNew->first = 0;
! 26: pNew->count = 0;
! 27: pNew->htsize = 0;
! 28: pNew->ht = 0;
! 29: }
! 30:
! 31: /* Remove all entries from a hash table. Reclaim all memory.
! 32: ** Call this routine to delete a hash table or to reset a hash table
! 33: ** to the empty state.
! 34: */
! 35: void sqlite3HashClear(Hash *pH){
! 36: HashElem *elem; /* For looping over all elements of the table */
! 37:
! 38: assert( pH!=0 );
! 39: elem = pH->first;
! 40: pH->first = 0;
! 41: sqlite3_free(pH->ht);
! 42: pH->ht = 0;
! 43: pH->htsize = 0;
! 44: while( elem ){
! 45: HashElem *next_elem = elem->next;
! 46: sqlite3_free(elem);
! 47: elem = next_elem;
! 48: }
! 49: pH->count = 0;
! 50: }
! 51:
! 52: /*
! 53: ** The hashing function.
! 54: */
! 55: static unsigned int strHash(const char *z, int nKey){
! 56: int h = 0;
! 57: assert( nKey>=0 );
! 58: while( nKey > 0 ){
! 59: h = (h<<3) ^ h ^ sqlite3UpperToLower[(unsigned char)*z++];
! 60: nKey--;
! 61: }
! 62: return h;
! 63: }
! 64:
! 65:
! 66: /* Link pNew element into the hash table pH. If pEntry!=0 then also
! 67: ** insert pNew into the pEntry hash bucket.
! 68: */
! 69: static void insertElement(
! 70: Hash *pH, /* The complete hash table */
! 71: struct _ht *pEntry, /* The entry into which pNew is inserted */
! 72: HashElem *pNew /* The element to be inserted */
! 73: ){
! 74: HashElem *pHead; /* First element already in pEntry */
! 75: if( pEntry ){
! 76: pHead = pEntry->count ? pEntry->chain : 0;
! 77: pEntry->count++;
! 78: pEntry->chain = pNew;
! 79: }else{
! 80: pHead = 0;
! 81: }
! 82: if( pHead ){
! 83: pNew->next = pHead;
! 84: pNew->prev = pHead->prev;
! 85: if( pHead->prev ){ pHead->prev->next = pNew; }
! 86: else { pH->first = pNew; }
! 87: pHead->prev = pNew;
! 88: }else{
! 89: pNew->next = pH->first;
! 90: if( pH->first ){ pH->first->prev = pNew; }
! 91: pNew->prev = 0;
! 92: pH->first = pNew;
! 93: }
! 94: }
! 95:
! 96:
! 97: /* Resize the hash table so that it cantains "new_size" buckets.
! 98: **
! 99: ** The hash table might fail to resize if sqlite3_malloc() fails or
! 100: ** if the new size is the same as the prior size.
! 101: ** Return TRUE if the resize occurs and false if not.
! 102: */
! 103: static int rehash(Hash *pH, unsigned int new_size){
! 104: struct _ht *new_ht; /* The new hash table */
! 105: HashElem *elem, *next_elem; /* For looping over existing elements */
! 106:
! 107: #if SQLITE_MALLOC_SOFT_LIMIT>0
! 108: if( new_size*sizeof(struct _ht)>SQLITE_MALLOC_SOFT_LIMIT ){
! 109: new_size = SQLITE_MALLOC_SOFT_LIMIT/sizeof(struct _ht);
! 110: }
! 111: if( new_size==pH->htsize ) return 0;
! 112: #endif
! 113:
! 114: /* The inability to allocates space for a larger hash table is
! 115: ** a performance hit but it is not a fatal error. So mark the
! 116: ** allocation as a benign.
! 117: */
! 118: sqlite3BeginBenignMalloc();
! 119: new_ht = (struct _ht *)sqlite3Malloc( new_size*sizeof(struct _ht) );
! 120: sqlite3EndBenignMalloc();
! 121:
! 122: if( new_ht==0 ) return 0;
! 123: sqlite3_free(pH->ht);
! 124: pH->ht = new_ht;
! 125: pH->htsize = new_size = sqlite3MallocSize(new_ht)/sizeof(struct _ht);
! 126: memset(new_ht, 0, new_size*sizeof(struct _ht));
! 127: for(elem=pH->first, pH->first=0; elem; elem = next_elem){
! 128: unsigned int h = strHash(elem->pKey, elem->nKey) % new_size;
! 129: next_elem = elem->next;
! 130: insertElement(pH, &new_ht[h], elem);
! 131: }
! 132: return 1;
! 133: }
! 134:
! 135: /* This function (for internal use only) locates an element in an
! 136: ** hash table that matches the given key. The hash for this key has
! 137: ** already been computed and is passed as the 4th parameter.
! 138: */
! 139: static HashElem *findElementGivenHash(
! 140: const Hash *pH, /* The pH to be searched */
! 141: const char *pKey, /* The key we are searching for */
! 142: int nKey, /* Bytes in key (not counting zero terminator) */
! 143: unsigned int h /* The hash for this key. */
! 144: ){
! 145: HashElem *elem; /* Used to loop thru the element list */
! 146: int count; /* Number of elements left to test */
! 147:
! 148: if( pH->ht ){
! 149: struct _ht *pEntry = &pH->ht[h];
! 150: elem = pEntry->chain;
! 151: count = pEntry->count;
! 152: }else{
! 153: elem = pH->first;
! 154: count = pH->count;
! 155: }
! 156: while( count-- && ALWAYS(elem) ){
! 157: if( elem->nKey==nKey && sqlite3StrNICmp(elem->pKey,pKey,nKey)==0 ){
! 158: return elem;
! 159: }
! 160: elem = elem->next;
! 161: }
! 162: return 0;
! 163: }
! 164:
! 165: /* Remove a single entry from the hash table given a pointer to that
! 166: ** element and a hash on the element's key.
! 167: */
! 168: static void removeElementGivenHash(
! 169: Hash *pH, /* The pH containing "elem" */
! 170: HashElem* elem, /* The element to be removed from the pH */
! 171: unsigned int h /* Hash value for the element */
! 172: ){
! 173: struct _ht *pEntry;
! 174: if( elem->prev ){
! 175: elem->prev->next = elem->next;
! 176: }else{
! 177: pH->first = elem->next;
! 178: }
! 179: if( elem->next ){
! 180: elem->next->prev = elem->prev;
! 181: }
! 182: if( pH->ht ){
! 183: pEntry = &pH->ht[h];
! 184: if( pEntry->chain==elem ){
! 185: pEntry->chain = elem->next;
! 186: }
! 187: pEntry->count--;
! 188: assert( pEntry->count>=0 );
! 189: }
! 190: sqlite3_free( elem );
! 191: pH->count--;
! 192: if( pH->count<=0 ){
! 193: assert( pH->first==0 );
! 194: assert( pH->count==0 );
! 195: sqlite3HashClear(pH);
! 196: }
! 197: }
! 198:
! 199: /* Attempt to locate an element of the hash table pH with a key
! 200: ** that matches pKey,nKey. Return the data for this element if it is
! 201: ** found, or NULL if there is no match.
! 202: */
! 203: void *sqlite3HashFind(const Hash *pH, const char *pKey, int nKey){
! 204: HashElem *elem; /* The element that matches key */
! 205: unsigned int h; /* A hash on key */
! 206:
! 207: assert( pH!=0 );
! 208: assert( pKey!=0 );
! 209: assert( nKey>=0 );
! 210: if( pH->ht ){
! 211: h = strHash(pKey, nKey) % pH->htsize;
! 212: }else{
! 213: h = 0;
! 214: }
! 215: elem = findElementGivenHash(pH, pKey, nKey, h);
! 216: return elem ? elem->data : 0;
! 217: }
! 218:
! 219: /* Insert an element into the hash table pH. The key is pKey,nKey
! 220: ** and the data is "data".
! 221: **
! 222: ** If no element exists with a matching key, then a new
! 223: ** element is created and NULL is returned.
! 224: **
! 225: ** If another element already exists with the same key, then the
! 226: ** new data replaces the old data and the old data is returned.
! 227: ** The key is not copied in this instance. If a malloc fails, then
! 228: ** the new data is returned and the hash table is unchanged.
! 229: **
! 230: ** If the "data" parameter to this function is NULL, then the
! 231: ** element corresponding to "key" is removed from the hash table.
! 232: */
! 233: void *sqlite3HashInsert(Hash *pH, const char *pKey, int nKey, void *data){
! 234: unsigned int h; /* the hash of the key modulo hash table size */
! 235: HashElem *elem; /* Used to loop thru the element list */
! 236: HashElem *new_elem; /* New element added to the pH */
! 237:
! 238: assert( pH!=0 );
! 239: assert( pKey!=0 );
! 240: assert( nKey>=0 );
! 241: if( pH->htsize ){
! 242: h = strHash(pKey, nKey) % pH->htsize;
! 243: }else{
! 244: h = 0;
! 245: }
! 246: elem = findElementGivenHash(pH,pKey,nKey,h);
! 247: if( elem ){
! 248: void *old_data = elem->data;
! 249: if( data==0 ){
! 250: removeElementGivenHash(pH,elem,h);
! 251: }else{
! 252: elem->data = data;
! 253: elem->pKey = pKey;
! 254: assert(nKey==elem->nKey);
! 255: }
! 256: return old_data;
! 257: }
! 258: if( data==0 ) return 0;
! 259: new_elem = (HashElem*)sqlite3Malloc( sizeof(HashElem) );
! 260: if( new_elem==0 ) return data;
! 261: new_elem->pKey = pKey;
! 262: new_elem->nKey = nKey;
! 263: new_elem->data = data;
! 264: pH->count++;
! 265: if( pH->count>=10 && pH->count > 2*pH->htsize ){
! 266: if( rehash(pH, pH->count*2) ){
! 267: assert( pH->htsize>0 );
! 268: h = strHash(pKey, nKey) % pH->htsize;
! 269: }
! 270: }
! 271: if( pH->ht ){
! 272: insertElement(pH, &pH->ht[h], new_elem);
! 273: }else{
! 274: insertElement(pH, 0, new_elem);
! 275: }
! 276: return 0;
! 277: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>