1 /*
2  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "compat.h"
18 
19 #include <ohash.h>
20 #include <stdlib.h>
21 
22 #include "kamid.h"
23 #include "utils.h"
24 
25 static void	*hash_alloc(size_t, void *);
26 static void	*hash_calloc(size_t, size_t, void *);
27 static void	 hash_free(void *, void *);
28 
29 static int	 table_static_open(struct table *);
30 static int	 table_static_add(struct table *, const char *, const char *);
31 static int	 table_static_lookup(struct table *, const char *, char **);
32 static void	 table_static_close(struct table *);
33 
34 struct table_backend table_static = {
35 	"static",
36 	table_static_open,
37 	table_static_add,
38 	table_static_lookup,
39 	table_static_close,
40 };
41 
42 struct kp {
43 	char	*key;
44 	char	*val;
45 };
46 
47 static void *
48 hash_alloc(size_t len, void *d)
49 {
50 	return xmalloc(len);
51 }
52 
53 static void *
54 hash_calloc(size_t nmemb, size_t size, void *d)
55 {
56 	return xcalloc(nmemb, size);
57 }
58 
59 static void
60 hash_free(void *ptr, void *d)
61 {
62 	free(ptr);
63 }
64 
65 static int
66 table_static_open(struct table *t)
67 {
68 	struct ohash_info info = {
69 		.key_offset = offsetof(struct kp, key),
70 		.calloc = hash_calloc,
71 		.free = hash_free,
72 		.alloc = hash_alloc,
73 	};
74 
75 	t->t_handle = xmalloc(sizeof(struct ohash));
76 	ohash_init(t->t_handle, 5, &info);
77 	return 0;
78 }
79 
80 int
81 table_static_add(struct table *t, const char *key, const char *val)
82 {
83 	struct kp	*kp;
84 	unsigned int	 slot;
85 
86 	kp = xcalloc(1, sizeof(*kp));
87 	kp->key = xstrdup(key);
88 	if (val != NULL)
89 		kp->val = xstrdup(val);
90 
91 	slot = ohash_qlookup(t->t_handle, kp->key);
92 	ohash_insert(t->t_handle, slot, kp);
93 
94 	return 0;
95 }
96 
97 int
98 table_static_lookup(struct table *t, const char *key, char **ret_val)
99 {
100 	struct kp	*kp;
101 	unsigned int	 slot;
102 
103 	slot = ohash_qlookup(t->t_handle, key);
104 	if ((kp = ohash_find(t->t_handle, slot)) == NULL)
105 		return -1;
106 
107 	*ret_val = xstrdup(kp->val);
108 	return 0;
109 }
110 
111 static void
112 table_static_close(struct table *t)
113 {
114 	struct kp	*kp;
115 	unsigned int	 i;
116 
117 	for (kp = ohash_first(t->t_handle, &i);
118 	     kp != NULL;
119 	     kp = ohash_next(t->t_handle, &i)) {
120 		ohash_remove(t->t_handle, i);
121 		free(kp->key);
122 		free(kp->val);
123 		free(kp);
124 	}
125 
126 	free(t->t_handle);
127 }