1
2
3#include <stdint.h>
4#include <stdlib.h>
5#include <stdio.h>
6#include <errno.h>
7#include <linux/err.h>
8#include "hashmap.h"
9#include "libbpf_internal.h"
10#include "strset.h"
11
12struct strset {
13 void *strs_data;
14 size_t strs_data_len;
15 size_t strs_data_cap;
16 size_t strs_data_max_len;
17
18
19 struct hashmap *strs_hash;
20};
21
22static size_t strset_hash_fn(const void *key, void *ctx)
23{
24 const struct strset *s = ctx;
25 const char *str = s->strs_data + (long)key;
26
27 return str_hash(str);
28}
29
30static bool strset_equal_fn(const void *key1, const void *key2, void *ctx)
31{
32 const struct strset *s = ctx;
33 const char *str1 = s->strs_data + (long)key1;
34 const char *str2 = s->strs_data + (long)key2;
35
36 return strcmp(str1, str2) == 0;
37}
38
39struct strset *strset__new(size_t max_data_sz, const char *init_data, size_t init_data_sz)
40{
41 struct strset *set = calloc(1, sizeof(*set));
42 struct hashmap *hash;
43 int err = -ENOMEM;
44
45 if (!set)
46 return ERR_PTR(-ENOMEM);
47
48 hash = hashmap__new(strset_hash_fn, strset_equal_fn, set);
49 if (IS_ERR(hash))
50 goto err_out;
51
52 set->strs_data_max_len = max_data_sz;
53 set->strs_hash = hash;
54
55 if (init_data) {
56 long off;
57
58 set->strs_data = malloc(init_data_sz);
59 if (!set->strs_data)
60 goto err_out;
61
62 memcpy(set->strs_data, init_data, init_data_sz);
63 set->strs_data_len = init_data_sz;
64 set->strs_data_cap = init_data_sz;
65
66 for (off = 0; off < set->strs_data_len; off += strlen(set->strs_data + off) + 1) {
67
68
69
70 err = hashmap__add(hash, (void *)off, (void *)off);
71 if (err == -EEXIST)
72 continue;
73 if (err)
74 goto err_out;
75 }
76 }
77
78 return set;
79err_out:
80 strset__free(set);
81 return ERR_PTR(err);
82}
83
84void strset__free(struct strset *set)
85{
86 if (IS_ERR_OR_NULL(set))
87 return;
88
89 hashmap__free(set->strs_hash);
90 free(set->strs_data);
91 free(set);
92}
93
94size_t strset__data_size(const struct strset *set)
95{
96 return set->strs_data_len;
97}
98
99const char *strset__data(const struct strset *set)
100{
101 return set->strs_data;
102}
103
104static void *strset_add_str_mem(struct strset *set, size_t add_sz)
105{
106 return libbpf_add_mem(&set->strs_data, &set->strs_data_cap, 1,
107 set->strs_data_len, set->strs_data_max_len, add_sz);
108}
109
110
111
112
113
114
115
116int strset__find_str(struct strset *set, const char *s)
117{
118 long old_off, new_off, len;
119 void *p;
120
121
122 len = strlen(s) + 1;
123 p = strset_add_str_mem(set, len);
124 if (!p)
125 return -ENOMEM;
126
127 new_off = set->strs_data_len;
128 memcpy(p, s, len);
129
130 if (hashmap__find(set->strs_hash, (void *)new_off, (void **)&old_off))
131 return old_off;
132
133 return -ENOENT;
134}
135
136
137
138
139
140
141
142int strset__add_str(struct strset *set, const char *s)
143{
144 long old_off, new_off, len;
145 void *p;
146 int err;
147
148
149
150
151
152
153
154
155
156 len = strlen(s) + 1;
157 p = strset_add_str_mem(set, len);
158 if (!p)
159 return -ENOMEM;
160
161 new_off = set->strs_data_len;
162 memcpy(p, s, len);
163
164
165
166
167
168 err = hashmap__insert(set->strs_hash, (void *)new_off, (void *)new_off,
169 HASHMAP_ADD, (const void **)&old_off, NULL);
170 if (err == -EEXIST)
171 return old_off;
172 if (err)
173 return err;
174
175 set->strs_data_len += len;
176 return new_off;
177}
178