e5f2e0076d74153644ce7618a3ed465d7ef97c6d
[tinc] / src / hash.c
1 /*
2     hash.c -- hash table management
3     Copyright (C) 2012 Guus Sliepen <guus@tinc-vpn.org>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "system.h"
21
22 #include "hash.h"
23 #include "xalloc.h"
24
25 /* Generic hash function */
26
27 static uint32_t hash_function(const void *p, size_t len) {
28         const uint8_t *q = p;
29         uint32_t hash = 0;
30         while(len > 0) {
31                 for(int i = len > 4 ? 4 : len; --i;)
32                         hash += q[i] << (8 * i);
33                 hash *= 0x9e370001UL; // Golden ratio prime.
34                 len -= 4;
35         }
36         return hash;
37 }
38
39 /* Map 32 bits int onto 0..n-1, without throwing away too many bits if n is 2^8 or 2^16 */
40
41 static uint32_t modulo(uint32_t hash, size_t n) {
42         if(n == 0x100)
43                 return (hash >> 24) ^ ((hash >> 16) & 0xff) ^ ((hash >> 8) & 0xff) ^ (hash & 0xff);
44         else if(n == 0x10000)
45                 return (hash >> 16) ^ (hash & 0xffff);
46         else
47                 return hash % n;
48 }
49
50 /* (De)allocation */
51
52 hash_t *hash_alloc(size_t n, size_t size) {
53         hash_t *hash = xmalloc_and_zero(sizeof *hash);
54         hash->n = n;
55         hash->size = size;
56         hash->keys = xmalloc(hash->n * hash->size);
57         hash->values = xmalloc_and_zero(hash->n * sizeof *hash->values);
58         return hash;
59 }
60
61 void hash_free(hash_t *hash) {
62         free(hash->keys);
63         free(hash->values);
64         free(hash);
65 }
66
67 /* Searching and inserting */
68
69 void hash_insert(hash_t *hash, const void *key, const void *value) {
70         uint32_t i = modulo(hash_function(key, hash->size), hash->n);
71         memcpy(hash->keys + i * hash->size, key, hash->size);
72         hash->values[i] = value;
73 }
74
75 void *hash_search(const hash_t *hash, const void *key) {
76         uint32_t i = modulo(hash_function(key, hash->size), hash->n);
77         if(hash->values[i] && !memcmp(key, hash->keys + i * hash->size, hash->size)) {
78                 return (void *)hash->values[i];
79         }
80         return NULL;
81 }
82
83 void *hash_search_or_insert(hash_t *hash, const void *key, const void *value) {
84         uint32_t i = modulo(hash_function(key, hash->size), hash->n);
85         if(hash->values[i] && !memcmp(key, hash->keys + i * hash->size, hash->size))
86                 return (void *)hash->values[i];
87         memcpy(hash->keys + i * hash->size, key, hash->size);
88         hash->values[i] = value;
89         return NULL;
90 }
91
92 /* Utility functions */
93
94 void hash_clear(hash_t *hash) {
95         memset(hash->values, 0, hash->n * sizeof *hash->values);
96 }
97
98 void hash_resize(hash_t *hash, size_t n) {
99         hash->keys = xrealloc(hash->keys, n * hash->size);
100         hash->values = xrealloc(hash->values, n * sizeof *hash->values);
101         if(n > hash->n)
102                 memset(hash->values + hash->n, 0, (n - hash->n) * sizeof *hash->values);
103 }