lhash.c 8.7 KB
Newer Older
R
Rich Salz 已提交
1
/*
2
 * Copyright 1995-2017 The OpenSSL Project Authors. All Rights Reserved.
3
 *
R
Rich Salz 已提交
4 5 6 7
 * Licensed under the OpenSSL license (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
8 9 10 11 12
 */

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
13 14
#include <openssl/crypto.h>
#include <openssl/lhash.h>
15 16
#include "lhash_lcl.h"

17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
/*
 * A hashing implementation that appears to be based on the linear hashing
 * alogrithm:
 * https://en.wikipedia.org/wiki/Linear_hashing
 *
 * Litwin, Witold (1980), "Linear hashing: A new tool for file and table
 * addressing", Proc. 6th Conference on Very Large Databases: 212–223
 * http://hackthology.com/pdfs/Litwin-1980-Linear_Hashing.pdf
 *
 * From the wikipedia article "Linear hashing is used in the BDB Berkeley
 * database system, which in turn is used by many software systems such as
 * OpenLDAP, using a C implementation derived from the CACM article and first
 * published on the Usenet in 1988 by Esmond Pitt."
 *
 * The CACM paper is available here:
 * https://pdfs.semanticscholar.org/ff4d/1c5deca6269cc316bfd952172284dbf610ee.pdf
 */
34

35 36 37 38
#undef MIN_NODES
#define MIN_NODES       16
#define UP_LOAD         (2*LH_LOAD_MULT) /* load times 256 (default 2) */
#define DOWN_LOAD       (LH_LOAD_MULT) /* load times 256 (default 1) */
39

M
mrpre 已提交
40
static int expand(OPENSSL_LHASH *lh);
41 42
static void contract(OPENSSL_LHASH *lh);
static OPENSSL_LH_NODE **getrn(OPENSSL_LHASH *lh, const void *data, unsigned long *rhash);
43

44
OPENSSL_LHASH *OPENSSL_LH_new(OPENSSL_LH_HASHFUNC h, OPENSSL_LH_COMPFUNC c)
45
{
46
    OPENSSL_LHASH *ret;
47

R
Rich Salz 已提交
48
    if ((ret = OPENSSL_zalloc(sizeof(*ret))) == NULL)
R
Rich Salz 已提交
49
        return NULL;
R
Rich Salz 已提交
50
    if ((ret->b = OPENSSL_zalloc(sizeof(*ret->b) * MIN_NODES)) == NULL)
R
Rich Salz 已提交
51
        goto err;
52 53
    ret->comp = ((c == NULL) ? (OPENSSL_LH_COMPFUNC)strcmp : c);
    ret->hash = ((h == NULL) ? (OPENSSL_LH_HASHFUNC)OPENSSL_LH_strhash : h);
54 55 56 57 58
    ret->num_nodes = MIN_NODES / 2;
    ret->num_alloc_nodes = MIN_NODES;
    ret->pmax = MIN_NODES / 2;
    ret->up_load = UP_LOAD;
    ret->down_load = DOWN_LOAD;
59
    return ret;
R
Rich Salz 已提交
60

R
Rich Salz 已提交
61 62
err:
    OPENSSL_free(ret->b);
63
    OPENSSL_free(ret);
R
Rich Salz 已提交
64
    return NULL;
65
}
66

67
void OPENSSL_LH_free(OPENSSL_LHASH *lh)
68 69
{
    unsigned int i;
70
    OPENSSL_LH_NODE *n, *nn;
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85

    if (lh == NULL)
        return;

    for (i = 0; i < lh->num_nodes; i++) {
        n = lh->b[i];
        while (n != NULL) {
            nn = n->next;
            OPENSSL_free(n);
            n = nn;
        }
    }
    OPENSSL_free(lh->b);
    OPENSSL_free(lh);
}
86

87
void *OPENSSL_LH_insert(OPENSSL_LHASH *lh, void *data)
88 89
{
    unsigned long hash;
90
    OPENSSL_LH_NODE *nn, **rn;
91
    void *ret;
M
mrpre 已提交
92

93
    lh->error = 0;
M
mrpre 已提交
94 95 96
    if ((lh->up_load <= (lh->num_items * LH_LOAD_MULT / lh->num_nodes)) && !expand(lh))
        return NULL;        /* 'lh->error++' already done in 'expand' */

97 98 99
    rn = getrn(lh, data, &hash);

    if (*rn == NULL) {
R
Rich Salz 已提交
100
        if ((nn = OPENSSL_malloc(sizeof(*nn))) == NULL) {
101
            lh->error++;
102
            return NULL;
103 104 105 106 107 108 109 110 111 112 113 114 115
        }
        nn->data = data;
        nn->next = NULL;
        nn->hash = hash;
        *rn = nn;
        ret = NULL;
        lh->num_insert++;
        lh->num_items++;
    } else {                    /* replace same key */
        ret = (*rn)->data;
        (*rn)->data = data;
        lh->num_replace++;
    }
116
    return ret;
117
}
118

119
void *OPENSSL_LH_delete(OPENSSL_LHASH *lh, const void *data)
120 121
{
    unsigned long hash;
122
    OPENSSL_LH_NODE *nn, **rn;
123 124 125 126 127 128 129
    void *ret;

    lh->error = 0;
    rn = getrn(lh, data, &hash);

    if (*rn == NULL) {
        lh->num_no_delete++;
130
        return NULL;
131 132 133 134 135 136 137 138 139 140 141 142 143
    } else {
        nn = *rn;
        *rn = nn->next;
        ret = nn->data;
        OPENSSL_free(nn);
        lh->num_delete++;
    }

    lh->num_items--;
    if ((lh->num_nodes > MIN_NODES) &&
        (lh->down_load >= (lh->num_items * LH_LOAD_MULT / lh->num_nodes)))
        contract(lh);

144
    return ret;
145
}
146

147
void *OPENSSL_LH_retrieve(OPENSSL_LHASH *lh, const void *data)
148 149
{
    unsigned long hash;
150
    OPENSSL_LH_NODE **rn;
151 152 153 154 155 156
    void *ret;

    lh->error = 0;
    rn = getrn(lh, data, &hash);

    if (*rn == NULL) {
157
        lh->num_retrieve_miss++;
R
Rich Salz 已提交
158
        return NULL;
159 160
    } else {
        ret = (*rn)->data;
161
        lh->num_retrieve++;
162
    }
R
Rich Salz 已提交
163
    return ret;
164
}
165

166 167 168
static void doall_util_fn(OPENSSL_LHASH *lh, int use_arg,
                          OPENSSL_LH_DOALL_FUNC func,
                          OPENSSL_LH_DOALL_FUNCARG func_arg, void *arg)
169 170
{
    int i;
171
    OPENSSL_LH_NODE *a, *n;
172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191

    if (lh == NULL)
        return;

    /*
     * reverse the order so we search from 'top to bottom' We were having
     * memory leaks otherwise
     */
    for (i = lh->num_nodes - 1; i >= 0; i--) {
        a = lh->b[i];
        while (a != NULL) {
            n = a->next;
            if (use_arg)
                func_arg(a->data, arg);
            else
                func(a->data);
            a = n;
        }
    }
}
192

193
void OPENSSL_LH_doall(OPENSSL_LHASH *lh, OPENSSL_LH_DOALL_FUNC func)
194
{
195
    doall_util_fn(lh, 0, func, (OPENSSL_LH_DOALL_FUNCARG)0, NULL);
196
}
197

198
void OPENSSL_LH_doall_arg(OPENSSL_LHASH *lh, OPENSSL_LH_DOALL_FUNCARG func, void *arg)
199
{
200
    doall_util_fn(lh, 1, (OPENSSL_LH_DOALL_FUNC)0, func, arg);
201
}
202

M
mrpre 已提交
203
static int expand(OPENSSL_LHASH *lh)
204
{
205
    OPENSSL_LH_NODE **n, **n1, **n2, *np;
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227
    unsigned int p, pmax, nni, j;
    unsigned long hash;

    nni = lh->num_alloc_nodes;
    p = lh->p;
    pmax = lh->pmax;
    if (p + 1 >= pmax) {
        j = nni * 2;
        n = OPENSSL_realloc(lh->b, sizeof(OPENSSL_LH_NODE *) * j);
        if (n == NULL) {
            lh->error++;
            return 0;
        }
        lh->b = n;
        memset(n + nni, 0, sizeof(*n) * (j - nni));
        lh->pmax = nni;
        lh->num_alloc_nodes = j;
        lh->num_expand_reallocs++;
        lh->p = 0;
    } else {
        lh->p++;
    }
228 229 230 231

    lh->num_nodes++;
    lh->num_expands++;
    n1 = &(lh->b[p]);
232
    n2 = &(lh->b[p + pmax]);
233
    *n2 = NULL;
234 235 236 237 238 239 240 241 242 243 244 245

    for (np = *n1; np != NULL;) {
        hash = np->hash;
        if ((hash % nni) != p) { /* move it */
            *n1 = (*n1)->next;
            np->next = *n2;
            *n2 = np;
        } else
            n1 = &((*n1)->next);
        np = *n1;
    }

M
mrpre 已提交
246
    return 1;
247
}
248

249
static void contract(OPENSSL_LHASH *lh)
250
{
251
    OPENSSL_LH_NODE **n, *n1, *np;
252 253 254 255

    np = lh->b[lh->p + lh->pmax - 1];
    lh->b[lh->p + lh->pmax - 1] = NULL; /* 24/07-92 - eay - weird but :-( */
    if (lh->p == 0) {
R
Rich Salz 已提交
256
        n = OPENSSL_realloc(lh->b,
257
                            (unsigned int)(sizeof(OPENSSL_LH_NODE *) * lh->pmax));
258
        if (n == NULL) {
R
Rich Salz 已提交
259
            /* fputs("realloc error in lhash",stderr); */
260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282
            lh->error++;
            return;
        }
        lh->num_contract_reallocs++;
        lh->num_alloc_nodes /= 2;
        lh->pmax /= 2;
        lh->p = lh->pmax - 1;
        lh->b = n;
    } else
        lh->p--;

    lh->num_nodes--;
    lh->num_contracts++;

    n1 = lh->b[(int)lh->p];
    if (n1 == NULL)
        lh->b[(int)lh->p] = np;
    else {
        while (n1->next != NULL)
            n1 = n1->next;
        n1->next = np;
    }
}
283

284 285
static OPENSSL_LH_NODE **getrn(OPENSSL_LHASH *lh,
                               const void *data, unsigned long *rhash)
286
{
287
    OPENSSL_LH_NODE **ret, *n1;
288
    unsigned long hash, nn;
289
    OPENSSL_LH_COMPFUNC cf;
290 291

    hash = (*(lh->hash)) (data);
292
    lh->num_hash_calls++;
293 294 295 296 297 298 299 300 301
    *rhash = hash;

    nn = hash % lh->pmax;
    if (nn < lh->p)
        nn = hash % lh->num_alloc_nodes;

    cf = lh->comp;
    ret = &(lh->b[(int)nn]);
    for (n1 = *ret; n1 != NULL; n1 = n1->next) {
302
        lh->num_hash_comps++;
303 304 305 306
        if (n1->hash != hash) {
            ret = &(n1->next);
            continue;
        }
307
        lh->num_comp_calls++;
308 309 310 311
        if (cf(n1->data, data) == 0)
            break;
        ret = &(n1->next);
    }
312
    return ret;
313 314 315 316 317 318
}

/*
 * The following hash seems to work very well on normal text strings no
 * collisions on /usr/dict/words and it distributes on %2^n quite well, not
 * as good as MD5, but still good.
319
 */
320
unsigned long OPENSSL_LH_strhash(const char *c)
321 322 323 324 325 326 327
{
    unsigned long ret = 0;
    long n;
    unsigned long v;
    int r;

    if ((c == NULL) || (*c == '\0'))
328
        return ret;
329

330 331 332 333 334 335 336 337 338 339
    n = 0x100;
    while (*c) {
        v = n | (*c);
        n += 0x100;
        r = (int)((v >> 2) ^ v) & 0x0f;
        ret = (ret << r) | (ret >> (32 - r));
        ret &= 0xFFFFFFFFL;
        ret ^= v * v;
        c++;
    }
340
    return (ret >> 16) ^ ret;
341
}
342

343
unsigned long OPENSSL_LH_num_items(const OPENSSL_LHASH *lh)
344 345 346
{
    return lh ? lh->num_items : 0;
}
D
Dr. Stephen Henson 已提交
347

348
unsigned long OPENSSL_LH_get_down_load(const OPENSSL_LHASH *lh)
D
Dr. Stephen Henson 已提交
349 350 351 352
{
    return lh->down_load;
}

353
void OPENSSL_LH_set_down_load(OPENSSL_LHASH *lh, unsigned long down_load)
D
Dr. Stephen Henson 已提交
354 355 356 357
{
    lh->down_load = down_load;
}

358
int OPENSSL_LH_error(OPENSSL_LHASH *lh)
D
Dr. Stephen Henson 已提交
359 360 361
{
    return lh->error;
}