bpf/prog_tests/hashmap.c

49 	struct hashmap *map;
51 	map = hashmap__new(hash_fn, equal_fn, NULL);
52 	if (!ASSERT_OK_PTR(map, "hashmap__new"))
59 		err = hashmap__update(map, k, v, &oldk, &oldv);
65 			err = hashmap__add(map, k, v);
67 			err = hashmap__set(map, k, v, &oldk, &oldv);
76 		if (CHECK(!hashmap__find(map, k, &oldv), "elem_find",
83 	if (CHECK(hashmap__size(map) != ELEM_CNT, "hashmap__size",
84 		  "invalid map size: %zu\n", hashmap__size(map)))
86 	if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)),
88 		  "unexpected map capacity: %zu\n", hashmap__capacity(map)))
92 	hashmap__for_each_entry(map, entry, bkt) {
109 		err = hashmap__add(map, k, v);
115 			err = hashmap__update(map, k, v, &oldk, &oldv);
117 			err = hashmap__set(map, k, v, &oldk, &oldv);
123 		if (CHECK(!hashmap__find(map, k, &oldv), "elem_find",
131 	if (CHECK(hashmap__size(map) != ELEM_CNT, "hashmap__size",
132 		  "invalid updated map size: %zu\n", hashmap__size(map)))
134 	if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)),
136 		  "unexpected map capacity: %zu\n", hashmap__capacity(map)))
140 	hashmap__for_each_entry_safe(map, entry, tmp, bkt) {
154 	hashmap__for_each_key_entry(map, entry, 0) {
163 	hashmap__for_each_key_entry_safe(map, entry, tmp, 0) {
173 		if (CHECK(!hashmap__delete(map, k, &oldk, &oldv), "elem_del",
180 		if (CHECK(hashmap__delete(map, k, &oldk, &oldv), "elem_del",
188 	if (CHECK(hashmap__size(map) != ELEM_CNT - found_cnt, "elem_cnt",
189 		  "invalid updated map size (already deleted: %d): %zu\n",
190 		  found_cnt, hashmap__size(map)))
192 	if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)),
194 		  "unexpected map capacity: %zu\n", hashmap__capacity(map)))
197 	hashmap__for_each_entry_safe(map, entry, tmp, bkt) {
207 		if (CHECK(!hashmap__delete(map, k, &oldk, &oldv), "elem_del",
214 		if (CHECK(hashmap__delete(map, k, &oldk, &oldv), "elem_del",
224 	if (CHECK(hashmap__size(map) != 0, "hashmap__size",
225 		  "invalid updated map size (already deleted: %d): %zu\n",
226 		  found_cnt, hashmap__size(map)))
230 	hashmap__for_each_entry(map, entry, bkt) {
232 		      "unexpected map entries left: %ld = %ld\n",
237 	hashmap__clear(map);
238 	hashmap__for_each_entry(map, entry, bkt) {
240 		      "unexpected map entries left: %ld = %ld\n",
246 	hashmap__free(map);
264 	struct hashmap *map;
267 	map = hashmap__new(str_hash_fn, str_equal_fn, NULL);
268 	if (CHECK(!map, "hashmap__new", "can't allocate hashmap\n"))
275 	err = hashmap__insert(map, "a", "apricot", HASHMAP_ADD, NULL, NULL);
279 	err = hashmap__insert(map, "a", "apple", HASHMAP_SET, &old_key, &old_value);
285 	err = hashmap__add(map, "b", "banana");
289 	err = hashmap__set(map, "b", "breadfruit", &old_key, &old_value);
295 	err = hashmap__update(map, "b", "blueberry", &old_key, &old_value);
301 	err = hashmap__append(map, "c", "cherry");
305 	if (CHECK(!hashmap__delete(map, "c", &old_key, &old_value),
311 	CHECK(!hashmap__find(map, "b", &value), "hashmap__find", "can't find value for 'b'\n");
314 	if (CHECK(!hashmap__delete(map, "b", NULL, NULL),
319 	hashmap__for_each_entry(map, cur, bkt) {
331 	hashmap__free(map);
343 	struct hashmap *map;
348 	map = hashmap__new(collision_hash_fn, equal_fn, NULL);
349 	if (!ASSERT_OK_PTR(map, "hashmap__new"))
356 	err = hashmap__append(map, k1, 1);
359 	err = hashmap__append(map, k1, 2);
362 	err = hashmap__append(map, k1, 4);
366 	err = hashmap__append(map, k2, 8);
369 	err = hashmap__append(map, k2, 16);
372 	err = hashmap__append(map, k2, 32);
376 	if (CHECK(hashmap__size(map) != 6, "hashmap_size",
377 		  "invalid map size: %zu\n", hashmap__size(map)))
382 	hashmap__for_each_entry(map, entry, bkt) {
391 	hashmap__for_each_key_entry(map, entry, k1) {
400 	hashmap__for_each_key_entry(map, entry, k2) {
408 	hashmap__free(map);
415 	struct hashmap *map;
419 	map = hashmap__new(hash_fn, equal_fn, NULL);
420 	if (!ASSERT_OK_PTR(map, "hashmap__new"))
423 	if (CHECK(hashmap__size(map) != 0, "hashmap__size",
424 		  "invalid map size: %zu\n", hashmap__size(map)))
426 	if (CHECK(hashmap__capacity(map) != 0, "hashmap__capacity",
427 		  "invalid map capacity: %zu\n", hashmap__capacity(map)))
429 	if (CHECK(hashmap__find(map, k, NULL), "elem_find",
432 	if (CHECK(hashmap__delete(map, k, NULL, NULL), "elem_del",
436 	hashmap__for_each_entry(map, entry, bkt) {
440 	hashmap__for_each_key_entry(map, entry, k) {
446 	hashmap__free(map);