hash.c
上传用户:xiaozhuqw
上传日期:2009-11-15
资源大小:1338k
文件大小:4k
源码类别:

网络

开发平台:

Unix_Linux

  1. /* Hash routine.
  2.  * Copyright (C) 1998 Kunihiro Ishiguro
  3.  *
  4.  * This file is part of GNU Zebra.
  5.  *
  6.  * GNU Zebra is free software; you can redistribute it and/or modify
  7.  * it under the terms of the GNU General Public License as published
  8.  * by the Free Software Foundation; either version 2, or (at your
  9.  * option) any later version.
  10.  *
  11.  * GNU Zebra is distributed in the hope that it will be useful, but
  12.  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU General Public License
  17.  * along with GNU Zebra; see the file COPYING.  If not, write to the
  18.  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19.  * Boston, MA 02111-1307, USA.
  20.  */
  21. #include <zebra.h>
  22. #include "hash.h"
  23. #include "memory.h"
  24. /* Allocate a new hash.  */
  25. struct hash *
  26. hash_create_size (unsigned int size, 
  27.   unsigned int (*hash_key) (), int (*hash_cmp) ())
  28. {
  29.   struct hash *hash;
  30.   hash = XMALLOC (MTYPE_HASH, sizeof (struct hash));
  31.   hash->index = XMALLOC (MTYPE_HASH_INDEX, 
  32.  sizeof (struct hash_backet *) * size);
  33.   memset (hash->index, 0, sizeof (struct hash_backet *) * size);
  34.   hash->size = size;
  35.   hash->hash_key = hash_key;
  36.   hash->hash_cmp = hash_cmp;
  37.   hash->count = 0;
  38.   return hash;
  39. }
  40. /* Allocate a new hash with default hash size.  */
  41. struct hash *
  42. hash_create (unsigned int (*hash_key) (), int (*hash_cmp) ())
  43. {
  44.   return hash_create_size (HASHTABSIZE, hash_key, hash_cmp);
  45. }
  46. /* Utility function for hash_get().  When this function is specified
  47.    as alloc_func, return arugment as it is.  This function is used for
  48.    intern already allocated value.  */
  49. void *
  50. hash_alloc_intern (void *arg)
  51. {
  52.   return arg;
  53. }
  54. /* Lookup and return hash backet in hash.  If there is no
  55.    corresponding hash backet and alloc_func is specified, create new
  56.    hash backet.  */
  57. void *
  58. hash_get (struct hash *hash, void *data, void * (*alloc_func) ())
  59. {
  60.   unsigned int key;
  61.   unsigned int index;
  62.   void *newdata;
  63.   struct hash_backet *backet;
  64.   key = (*hash->hash_key) (data);
  65.   index = key % hash->size;
  66.   for (backet = hash->index[index]; backet != NULL; backet = backet->next) 
  67.     if (backet->key == key && (*hash->hash_cmp) (backet->data, data))
  68.       return backet->data;
  69.   if (alloc_func)
  70.     {
  71.       newdata = (*alloc_func) (data);
  72.       if (newdata == NULL)
  73. return NULL;
  74.       backet = XMALLOC (MTYPE_HASH_BACKET, sizeof (struct hash_backet));
  75.       backet->data = newdata;
  76.       backet->key = key;
  77.       backet->next = hash->index[index];
  78.       hash->index[index] = backet;
  79.       hash->count++;
  80.       return backet->data;
  81.     }
  82.   return NULL;
  83. }
  84. /* Hash lookup.  */
  85. void *
  86. hash_lookup (struct hash *hash, void *data)
  87. {
  88.   return hash_get (hash, data, NULL);
  89. }
  90. /* This function release registered value from specified hash.  When
  91.    release is successfully finished, return the data pointer in the
  92.    hash backet.  */
  93. void *
  94. hash_release (struct hash *hash, void *data)
  95. {
  96.   void *ret;
  97.   unsigned int key;
  98.   unsigned int index;
  99.   struct hash_backet *backet;
  100.   struct hash_backet *pp;
  101.   key = (*hash->hash_key) (data);
  102.   index = key % hash->size;
  103.   for (backet = pp = hash->index[index]; backet; backet = backet->next)
  104.     {
  105.       if (backet->key == key && (*hash->hash_cmp) (backet->data, data)) 
  106. {
  107.   if (backet == pp) 
  108.     hash->index[index] = backet->next;
  109.   else 
  110.     pp->next = backet->next;
  111.   ret = backet->data;
  112.   XFREE (MTYPE_HASH_BACKET, backet);
  113.   hash->count--;
  114.   return ret;
  115. }
  116.       pp = backet;
  117.     }
  118.   return NULL;
  119. }
  120. /* Iterator function for hash.  */
  121. void
  122. hash_iterate (struct hash *hash, 
  123.       void (*func) (struct hash_backet *, void *), void *arg)
  124. {
  125.   int i;
  126.   struct hash_backet *hb;
  127.   for (i = 0; i < hash->size; i++)
  128.     for (hb = hash->index[i]; hb; hb = hb->next)
  129.       (*func) (hb, arg);
  130. }
  131. /* Clean up hash.  */
  132. void
  133. hash_clean (struct hash *hash, void (*free_func) (void *))
  134. {
  135.   int i;
  136.   struct hash_backet *hb;
  137.   struct hash_backet *next;
  138.   for (i = 0; i < hash->size; i++)
  139.     {
  140.       for (hb = hash->index[i]; hb; hb = next)
  141. {
  142.   next = hb->next;
  143.       
  144.   if (free_func)
  145.     (*free_func) (hb->data);
  146.   XFREE (MTYPE_HASH_BACKET, hb);
  147.   hash->count--;
  148. }
  149.       hash->index[i] = NULL;
  150.     }
  151. }
  152. /* Free hash memory.  You may call hash_clean before call this
  153.    function.  */
  154. void
  155. hash_free (struct hash *hash)
  156. {
  157.   XFREE (MTYPE_HASH_INDEX, hash->index);
  158.   XFREE (MTYPE_HASH, hash);
  159. }