1122531Sume/*
2122531Sume * Copyright 2016 Jakub Klama <jceel@FreeBSD.org>
3122531Sume * All rights reserved
4122531Sume *
5122531Sume * Redistribution and use in source and binary forms, with or without
6122531Sume * modification, are permitted providing that the following conditions
7122531Sume * are met:
8122531Sume * 1. Redistributions of source code must retain the above copyright
9122531Sume *    notice, this list of conditions and the following disclaimer.
10122531Sume * 2. Redistributions in binary form must reproduce the above copyright
11122531Sume *    notice, this list of conditions and the following disclaimer in the
12122531Sume *    documentation and/or other materials provided with the distribution.
13122531Sume *
14122531Sume * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15122531Sume * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16122531Sume * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17122531Sume * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
18122531Sume * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19122531Sume * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20122531Sume * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21122531Sume * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
22122531Sume * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
23122531Sume * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24122531Sume * POSSIBILITY OF SUCH DAMAGE.
25122531Sume *
26122531Sume */
27122531Sume
28122531Sume#ifndef LIB9P_HASHTABLE_H
29122531Sume#define LIB9P_HASHTABLE_H
30122531Sume
31122531Sume#include <pthread.h>
32122531Sume#include <sys/queue.h>
33175360Ssobomax
34122531Sumestruct ht {
35175360Ssobomax	struct ht_entry * 	ht_entries;
36122531Sume	ssize_t 		ht_nentries;
37122531Sume	pthread_rwlock_t	ht_rwlock;
38122531Sume};
39122531Sume
40122531Sumestruct ht_entry {
41122531Sume	TAILQ_HEAD(, ht_item) hte_items;
42122531Sume};
43122531Sume
44122531Sumestruct ht_item {
45122531Sume	uint32_t		hti_hash;
46122531Sume	void *			hti_data;
47122531Sume	TAILQ_ENTRY(ht_item)	hti_link;
48122531Sume};
49122531Sume
50122531Sumestruct ht_iter {
51122531Sume	struct ht *		htit_parent;
52122531Sume	struct ht_item *	htit_curr;
53122531Sume	struct ht_item *	htit_next;
54122531Sume	ssize_t			htit_slot;
55122531Sume};
56122531Sume
57122531Sume#ifdef __clang__
58122531Sume#pragma clang diagnostic push
59122531Sume#pragma clang diagnostic ignored "-Wthread-safety-analysis"
60#endif
61
62/*
63 * Obtain read-lock on hash table.
64 */
65static inline int
66ht_rdlock(struct ht *h)
67{
68
69	return (pthread_rwlock_rdlock(&h->ht_rwlock));
70}
71
72/*
73 * Obtain write-lock on hash table.
74 */
75static inline int
76ht_wrlock(struct ht *h)
77{
78
79	return (pthread_rwlock_wrlock(&h->ht_rwlock));
80}
81
82/*
83 * Release lock on hash table.
84 */
85static inline int
86ht_unlock(struct ht *h)
87{
88
89	return (pthread_rwlock_unlock(&h->ht_rwlock));
90}
91
92#ifdef __clang__
93#pragma clang diagnostic pop
94#endif
95
96void ht_init(struct ht *h, ssize_t size);
97void ht_destroy(struct ht *h);
98void *ht_find(struct ht *h, uint32_t hash);
99void *ht_find_locked(struct ht *h, uint32_t hash);
100int ht_add(struct ht *h, uint32_t hash, void *value);
101int ht_remove(struct ht *h, uint32_t hash);
102int ht_remove_locked(struct ht *h, uint32_t hash);
103int ht_remove_at_iter(struct ht_iter *iter);
104void ht_iter(struct ht *h, struct ht_iter *iter);
105void *ht_next(struct ht_iter *iter);
106
107#endif  /* LIB9P_HASHTABLE_H */
108