Commit fd9c40c5 authored by Alexei Starovoitov's avatar Alexei Starovoitov

Merge branch 'test_progs-asan'

Andrii Nakryiko says:

====================
Add necessary infra to build selftests with ASAN (or any other sanitizer). Fix
a bunch of found memory leaks and other memory access issues.

v1->v2:
  - don't add ASAN flavor, but allow extra flags for build (Alexei);
  - fix few more found issues, which somehow were missed first time.
====================
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parents 3271e8f3 e4e8f4d0
...@@ -59,7 +59,14 @@ struct hashmap *hashmap__new(hashmap_hash_fn hash_fn, ...@@ -59,7 +59,14 @@ struct hashmap *hashmap__new(hashmap_hash_fn hash_fn,
void hashmap__clear(struct hashmap *map) void hashmap__clear(struct hashmap *map)
{ {
struct hashmap_entry *cur, *tmp;
int bkt;
hashmap__for_each_entry_safe(map, cur, tmp, bkt) {
free(cur);
}
free(map->buckets); free(map->buckets);
map->buckets = NULL;
map->cap = map->cap_bits = map->sz = 0; map->cap = map->cap_bits = map->sz = 0;
} }
......
...@@ -6934,6 +6934,7 @@ int libbpf_find_vmlinux_btf_id(const char *name, ...@@ -6934,6 +6934,7 @@ int libbpf_find_vmlinux_btf_id(const char *name,
enum bpf_attach_type attach_type) enum bpf_attach_type attach_type)
{ {
struct btf *btf; struct btf *btf;
int err;
btf = libbpf_find_kernel_btf(); btf = libbpf_find_kernel_btf();
if (IS_ERR(btf)) { if (IS_ERR(btf)) {
...@@ -6941,7 +6942,9 @@ int libbpf_find_vmlinux_btf_id(const char *name, ...@@ -6941,7 +6942,9 @@ int libbpf_find_vmlinux_btf_id(const char *name,
return -EINVAL; return -EINVAL;
} }
return __find_vmlinux_btf_id(btf, name, attach_type); err = __find_vmlinux_btf_id(btf, name, attach_type);
btf__free(btf);
return err;
} }
static int libbpf_find_prog_btf_id(const char *name, __u32 attach_prog_fd) static int libbpf_find_prog_btf_id(const char *name, __u32 attach_prog_fd)
......
...@@ -30,8 +30,6 @@ test_tcpnotify_user ...@@ -30,8 +30,6 @@ test_tcpnotify_user
test_libbpf test_libbpf
test_tcp_check_syncookie_user test_tcp_check_syncookie_user
test_sysctl test_sysctl
test_hashmap
test_btf_dump
test_current_pid_tgid_new_ns test_current_pid_tgid_new_ns
xdping xdping
test_cpp test_cpp
...@@ -39,4 +37,4 @@ test_cpp ...@@ -39,4 +37,4 @@ test_cpp
/no_alu32 /no_alu32
/bpf_gcc /bpf_gcc
/tools /tools
/runqslower
...@@ -20,9 +20,10 @@ CLANG ?= clang ...@@ -20,9 +20,10 @@ CLANG ?= clang
LLC ?= llc LLC ?= llc
LLVM_OBJCOPY ?= llvm-objcopy LLVM_OBJCOPY ?= llvm-objcopy
BPF_GCC ?= $(shell command -v bpf-gcc;) BPF_GCC ?= $(shell command -v bpf-gcc;)
CFLAGS += -g -rdynamic -Wall -O2 $(GENFLAGS) -I$(CURDIR) \ SAN_CFLAGS ?=
-I$(INCLUDE_DIR) -I$(GENDIR) -I$(LIBDIR) -I$(TOOLSINCDIR) \ CFLAGS += -g -rdynamic -Wall -O2 $(GENFLAGS) $(SAN_CFLAGS) \
-I$(APIDIR) \ -I$(CURDIR) -I$(INCLUDE_DIR) -I$(GENDIR) -I$(LIBDIR) \
-I$(TOOLSINCDIR) -I$(APIDIR) \
-Dbpf_prog_load=bpf_prog_test_load \ -Dbpf_prog_load=bpf_prog_test_load \
-Dbpf_load_program=bpf_test_load_program -Dbpf_load_program=bpf_test_load_program
LDLIBS += -lcap -lelf -lz -lrt -lpthread LDLIBS += -lcap -lelf -lz -lrt -lpthread
...@@ -32,7 +33,7 @@ TEST_GEN_PROGS = test_verifier test_tag test_maps test_lru_map test_lpm_map test ...@@ -32,7 +33,7 @@ TEST_GEN_PROGS = test_verifier test_tag test_maps test_lru_map test_lpm_map test
test_align test_verifier_log test_dev_cgroup test_tcpbpf_user \ test_align test_verifier_log test_dev_cgroup test_tcpbpf_user \
test_sock test_btf test_sockmap get_cgroup_id_user test_socket_cookie \ test_sock test_btf test_sockmap get_cgroup_id_user test_socket_cookie \
test_cgroup_storage \ test_cgroup_storage \
test_netcnt test_tcpnotify_user test_sock_fields test_sysctl test_hashmap \ test_netcnt test_tcpnotify_user test_sock_fields test_sysctl \
test_progs-no_alu32 \ test_progs-no_alu32 \
test_current_pid_tgid_new_ns test_current_pid_tgid_new_ns
...@@ -324,7 +325,7 @@ $(TRUNNER_TEST_OBJS): $(TRUNNER_OUTPUT)/%.test.o: \ ...@@ -324,7 +325,7 @@ $(TRUNNER_TEST_OBJS): $(TRUNNER_OUTPUT)/%.test.o: \
$(TRUNNER_BPF_SKELS) \ $(TRUNNER_BPF_SKELS) \
$$(BPFOBJ) | $(TRUNNER_OUTPUT) $$(BPFOBJ) | $(TRUNNER_OUTPUT)
$$(call msg,TEST-OBJ,$(TRUNNER_BINARY),$$@) $$(call msg,TEST-OBJ,$(TRUNNER_BINARY),$$@)
cd $$(@D) && $$(CC) $$(CFLAGS) -c $(CURDIR)/$$< $$(LDLIBS) -o $$(@F) cd $$(@D) && $$(CC) -I. $$(CFLAGS) -c $(CURDIR)/$$< $$(LDLIBS) -o $$(@F)
$(TRUNNER_EXTRA_OBJS): $(TRUNNER_OUTPUT)/%.o: \ $(TRUNNER_EXTRA_OBJS): $(TRUNNER_OUTPUT)/%.o: \
%.c \ %.c \
......
...@@ -392,7 +392,7 @@ static struct core_reloc_test_case test_cases[] = { ...@@ -392,7 +392,7 @@ static struct core_reloc_test_case test_cases[] = {
.input = STRUCT_TO_CHAR_PTR(core_reloc_existence___minimal) { .input = STRUCT_TO_CHAR_PTR(core_reloc_existence___minimal) {
.a = 42, .a = 42,
}, },
.input_len = sizeof(struct core_reloc_existence), .input_len = sizeof(struct core_reloc_existence___minimal),
.output = STRUCT_TO_CHAR_PTR(core_reloc_existence_output) { .output = STRUCT_TO_CHAR_PTR(core_reloc_existence_output) {
.a_exists = 1, .a_exists = 1,
.b_exists = 0, .b_exists = 0,
......
...@@ -5,26 +5,17 @@ ...@@ -5,26 +5,17 @@
* *
* Copyright (c) 2019 Facebook * Copyright (c) 2019 Facebook
*/ */
#include <stdio.h> #include "test_progs.h"
#include <errno.h>
#include <linux/err.h>
#include "bpf/hashmap.h" #include "bpf/hashmap.h"
#define CHECK(condition, format...) ({ \ static int duration = 0;
int __ret = !!(condition); \
if (__ret) { \
fprintf(stderr, "%s:%d:FAIL ", __func__, __LINE__); \
fprintf(stderr, format); \
} \
__ret; \
})
size_t hash_fn(const void *k, void *ctx) static size_t hash_fn(const void *k, void *ctx)
{ {
return (long)k; return (long)k;
} }
bool equal_fn(const void *a, const void *b, void *ctx) static bool equal_fn(const void *a, const void *b, void *ctx)
{ {
return (long)a == (long)b; return (long)a == (long)b;
} }
...@@ -49,53 +40,55 @@ static inline size_t exp_cap(size_t sz) ...@@ -49,53 +40,55 @@ static inline size_t exp_cap(size_t sz)
#define ELEM_CNT 62 #define ELEM_CNT 62
int test_hashmap_generic(void) static void test_hashmap_generic(void)
{ {
struct hashmap_entry *entry, *tmp; struct hashmap_entry *entry, *tmp;
int err, bkt, found_cnt, i; int err, bkt, found_cnt, i;
long long found_msk; long long found_msk;
struct hashmap *map; struct hashmap *map;
fprintf(stderr, "%s: ", __func__);
map = hashmap__new(hash_fn, equal_fn, NULL); map = hashmap__new(hash_fn, equal_fn, NULL);
if (CHECK(IS_ERR(map), "failed to create map: %ld\n", PTR_ERR(map))) if (CHECK(IS_ERR(map), "hashmap__new",
return 1; "failed to create map: %ld\n", PTR_ERR(map)))
return;
for (i = 0; i < ELEM_CNT; i++) { for (i = 0; i < ELEM_CNT; i++) {
const void *oldk, *k = (const void *)(long)i; const void *oldk, *k = (const void *)(long)i;
void *oldv, *v = (void *)(long)(1024 + i); void *oldv, *v = (void *)(long)(1024 + i);
err = hashmap__update(map, k, v, &oldk, &oldv); err = hashmap__update(map, k, v, &oldk, &oldv);
if (CHECK(err != -ENOENT, "unexpected result: %d\n", err)) if (CHECK(err != -ENOENT, "hashmap__update",
return 1; "unexpected result: %d\n", err))
goto cleanup;
if (i % 2) { if (i % 2) {
err = hashmap__add(map, k, v); err = hashmap__add(map, k, v);
} else { } else {
err = hashmap__set(map, k, v, &oldk, &oldv); err = hashmap__set(map, k, v, &oldk, &oldv);
if (CHECK(oldk != NULL || oldv != NULL, if (CHECK(oldk != NULL || oldv != NULL, "check_kv",
"unexpected k/v: %p=%p\n", oldk, oldv)) "unexpected k/v: %p=%p\n", oldk, oldv))
return 1; goto cleanup;
} }
if (CHECK(err, "failed to add k/v %ld = %ld: %d\n", if (CHECK(err, "elem_add", "failed to add k/v %ld = %ld: %d\n",
(long)k, (long)v, err)) (long)k, (long)v, err))
return 1; goto cleanup;
if (CHECK(!hashmap__find(map, k, &oldv), if (CHECK(!hashmap__find(map, k, &oldv), "elem_find",
"failed to find key %ld\n", (long)k)) "failed to find key %ld\n", (long)k))
return 1; goto cleanup;
if (CHECK(oldv != v, "found value is wrong: %ld\n", (long)oldv)) if (CHECK(oldv != v, "elem_val",
return 1; "found value is wrong: %ld\n", (long)oldv))
goto cleanup;
} }
if (CHECK(hashmap__size(map) != ELEM_CNT, if (CHECK(hashmap__size(map) != ELEM_CNT, "hashmap__size",
"invalid map size: %zu\n", hashmap__size(map))) "invalid map size: %zu\n", hashmap__size(map)))
return 1; goto cleanup;
if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)), if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)),
"hashmap_cap",
"unexpected map capacity: %zu\n", hashmap__capacity(map))) "unexpected map capacity: %zu\n", hashmap__capacity(map)))
return 1; goto cleanup;
found_msk = 0; found_msk = 0;
hashmap__for_each_entry(map, entry, bkt) { hashmap__for_each_entry(map, entry, bkt) {
...@@ -103,42 +96,47 @@ int test_hashmap_generic(void) ...@@ -103,42 +96,47 @@ int test_hashmap_generic(void)
long v = (long)entry->value; long v = (long)entry->value;
found_msk |= 1ULL << k; found_msk |= 1ULL << k;
if (CHECK(v - k != 1024, "invalid k/v pair: %ld = %ld\n", k, v)) if (CHECK(v - k != 1024, "check_kv",
return 1; "invalid k/v pair: %ld = %ld\n", k, v))
goto cleanup;
} }
if (CHECK(found_msk != (1ULL << ELEM_CNT) - 1, if (CHECK(found_msk != (1ULL << ELEM_CNT) - 1, "elem_cnt",
"not all keys iterated: %llx\n", found_msk)) "not all keys iterated: %llx\n", found_msk))
return 1; goto cleanup;
for (i = 0; i < ELEM_CNT; i++) { for (i = 0; i < ELEM_CNT; i++) {
const void *oldk, *k = (const void *)(long)i; const void *oldk, *k = (const void *)(long)i;
void *oldv, *v = (void *)(long)(256 + i); void *oldv, *v = (void *)(long)(256 + i);
err = hashmap__add(map, k, v); err = hashmap__add(map, k, v);
if (CHECK(err != -EEXIST, "unexpected add result: %d\n", err)) if (CHECK(err != -EEXIST, "hashmap__add",
return 1; "unexpected add result: %d\n", err))
goto cleanup;
if (i % 2) if (i % 2)
err = hashmap__update(map, k, v, &oldk, &oldv); err = hashmap__update(map, k, v, &oldk, &oldv);
else else
err = hashmap__set(map, k, v, &oldk, &oldv); err = hashmap__set(map, k, v, &oldk, &oldv);
if (CHECK(err, "failed to update k/v %ld = %ld: %d\n", if (CHECK(err, "elem_upd",
(long)k, (long)v, err)) "failed to update k/v %ld = %ld: %d\n",
return 1; (long)k, (long)v, err))
if (CHECK(!hashmap__find(map, k, &oldv), goto cleanup;
if (CHECK(!hashmap__find(map, k, &oldv), "elem_find",
"failed to find key %ld\n", (long)k)) "failed to find key %ld\n", (long)k))
return 1; goto cleanup;
if (CHECK(oldv != v, "found value is wrong: %ld\n", (long)oldv)) if (CHECK(oldv != v, "elem_val",
return 1; "found value is wrong: %ld\n", (long)oldv))
goto cleanup;
} }
if (CHECK(hashmap__size(map) != ELEM_CNT, if (CHECK(hashmap__size(map) != ELEM_CNT, "hashmap__size",
"invalid updated map size: %zu\n", hashmap__size(map))) "invalid updated map size: %zu\n", hashmap__size(map)))
return 1; goto cleanup;
if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)), if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)),
"hashmap__capacity",
"unexpected map capacity: %zu\n", hashmap__capacity(map))) "unexpected map capacity: %zu\n", hashmap__capacity(map)))
return 1; goto cleanup;
found_msk = 0; found_msk = 0;
hashmap__for_each_entry_safe(map, entry, tmp, bkt) { hashmap__for_each_entry_safe(map, entry, tmp, bkt) {
...@@ -146,20 +144,21 @@ int test_hashmap_generic(void) ...@@ -146,20 +144,21 @@ int test_hashmap_generic(void)
long v = (long)entry->value; long v = (long)entry->value;
found_msk |= 1ULL << k; found_msk |= 1ULL << k;
if (CHECK(v - k != 256, if (CHECK(v - k != 256, "elem_check",
"invalid updated k/v pair: %ld = %ld\n", k, v)) "invalid updated k/v pair: %ld = %ld\n", k, v))
return 1; goto cleanup;
} }
if (CHECK(found_msk != (1ULL << ELEM_CNT) - 1, if (CHECK(found_msk != (1ULL << ELEM_CNT) - 1, "elem_cnt",
"not all keys iterated after update: %llx\n", found_msk)) "not all keys iterated after update: %llx\n", found_msk))
return 1; goto cleanup;
found_cnt = 0; found_cnt = 0;
hashmap__for_each_key_entry(map, entry, (void *)0) { hashmap__for_each_key_entry(map, entry, (void *)0) {
found_cnt++; found_cnt++;
} }
if (CHECK(!found_cnt, "didn't find any entries for key 0\n")) if (CHECK(!found_cnt, "found_cnt",
return 1; "didn't find any entries for key 0\n"))
goto cleanup;
found_msk = 0; found_msk = 0;
found_cnt = 0; found_cnt = 0;
...@@ -173,30 +172,31 @@ int test_hashmap_generic(void) ...@@ -173,30 +172,31 @@ int test_hashmap_generic(void)
found_cnt++; found_cnt++;
found_msk |= 1ULL << (long)k; found_msk |= 1ULL << (long)k;
if (CHECK(!hashmap__delete(map, k, &oldk, &oldv), if (CHECK(!hashmap__delete(map, k, &oldk, &oldv), "elem_del",
"failed to delete k/v %ld = %ld\n", "failed to delete k/v %ld = %ld\n",
(long)k, (long)v)) (long)k, (long)v))
return 1; goto cleanup;
if (CHECK(oldk != k || oldv != v, if (CHECK(oldk != k || oldv != v, "check_old",
"invalid deleted k/v: expected %ld = %ld, got %ld = %ld\n", "invalid deleted k/v: expected %ld = %ld, got %ld = %ld\n",
(long)k, (long)v, (long)oldk, (long)oldv)) (long)k, (long)v, (long)oldk, (long)oldv))
return 1; goto cleanup;
if (CHECK(hashmap__delete(map, k, &oldk, &oldv), if (CHECK(hashmap__delete(map, k, &oldk, &oldv), "elem_del",
"unexpectedly deleted k/v %ld = %ld\n", "unexpectedly deleted k/v %ld = %ld\n",
(long)oldk, (long)oldv)) (long)oldk, (long)oldv))
return 1; goto cleanup;
} }
if (CHECK(!found_cnt || !found_msk, if (CHECK(!found_cnt || !found_msk, "found_entries",
"didn't delete any key entries\n")) "didn't delete any key entries\n"))
return 1; goto cleanup;
if (CHECK(hashmap__size(map) != ELEM_CNT - found_cnt, if (CHECK(hashmap__size(map) != ELEM_CNT - found_cnt, "elem_cnt",
"invalid updated map size (already deleted: %d): %zu\n", "invalid updated map size (already deleted: %d): %zu\n",
found_cnt, hashmap__size(map))) found_cnt, hashmap__size(map)))
return 1; goto cleanup;
if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)), if (CHECK(hashmap__capacity(map) != exp_cap(hashmap__size(map)),
"hashmap__capacity",
"unexpected map capacity: %zu\n", hashmap__capacity(map))) "unexpected map capacity: %zu\n", hashmap__capacity(map)))
return 1; goto cleanup;
hashmap__for_each_entry_safe(map, entry, tmp, bkt) { hashmap__for_each_entry_safe(map, entry, tmp, bkt) {
const void *oldk, *k; const void *oldk, *k;
...@@ -208,53 +208,56 @@ int test_hashmap_generic(void) ...@@ -208,53 +208,56 @@ int test_hashmap_generic(void)
found_cnt++; found_cnt++;
found_msk |= 1ULL << (long)k; found_msk |= 1ULL << (long)k;
if (CHECK(!hashmap__delete(map, k, &oldk, &oldv), if (CHECK(!hashmap__delete(map, k, &oldk, &oldv), "elem_del",
"failed to delete k/v %ld = %ld\n", "failed to delete k/v %ld = %ld\n",
(long)k, (long)v)) (long)k, (long)v))
return 1; goto cleanup;
if (CHECK(oldk != k || oldv != v, if (CHECK(oldk != k || oldv != v, "elem_check",
"invalid old k/v: expect %ld = %ld, got %ld = %ld\n", "invalid old k/v: expect %ld = %ld, got %ld = %ld\n",
(long)k, (long)v, (long)oldk, (long)oldv)) (long)k, (long)v, (long)oldk, (long)oldv))
return 1; goto cleanup;
if (CHECK(hashmap__delete(map, k, &oldk, &oldv), if (CHECK(hashmap__delete(map, k, &oldk, &oldv), "elem_del",
"unexpectedly deleted k/v %ld = %ld\n", "unexpectedly deleted k/v %ld = %ld\n",
(long)k, (long)v)) (long)k, (long)v))
return 1; goto cleanup;
} }
if (CHECK(found_cnt != ELEM_CNT || found_msk != (1ULL << ELEM_CNT) - 1, if (CHECK(found_cnt != ELEM_CNT || found_msk != (1ULL << ELEM_CNT) - 1,
"found_cnt",
"not all keys were deleted: found_cnt:%d, found_msk:%llx\n", "not all keys were deleted: found_cnt:%d, found_msk:%llx\n",
found_cnt, found_msk)) found_cnt, found_msk))
return 1; goto cleanup;
if (CHECK(hashmap__size(map) != 0, if (CHECK(hashmap__size(map) != 0, "hashmap__size",
"invalid updated map size (already deleted: %d): %zu\n", "invalid updated map size (already deleted: %d): %zu\n",
found_cnt, hashmap__size(map))) found_cnt, hashmap__size(map)))
return 1; goto cleanup;
found_cnt = 0; found_cnt = 0;
hashmap__for_each_entry(map, entry, bkt) { hashmap__for_each_entry(map, entry, bkt) {
CHECK(false, "unexpected map entries left: %ld = %ld\n", CHECK(false, "elem_exists",
(long)entry->key, (long)entry->value); "unexpected map entries left: %ld = %ld\n",
return 1; (long)entry->key, (long)entry->value);
goto cleanup;
} }
hashmap__free(map); hashmap__clear(map);
hashmap__for_each_entry(map, entry, bkt) { hashmap__for_each_entry(map, entry, bkt) {
CHECK(false, "unexpected map entries left: %ld = %ld\n", CHECK(false, "elem_exists",
(long)entry->key, (long)entry->value); "unexpected map entries left: %ld = %ld\n",
return 1; (long)entry->key, (long)entry->value);
goto cleanup;
} }
fprintf(stderr, "OK\n"); cleanup:
return 0; hashmap__free(map);
} }
size_t collision_hash_fn(const void *k, void *ctx) static size_t collision_hash_fn(const void *k, void *ctx)
{ {
return 0; return 0;
} }
int test_hashmap_multimap(void) static void test_hashmap_multimap(void)
{ {
void *k1 = (void *)0, *k2 = (void *)1; void *k1 = (void *)0, *k2 = (void *)1;
struct hashmap_entry *entry; struct hashmap_entry *entry;
...@@ -262,121 +265,116 @@ int test_hashmap_multimap(void) ...@@ -262,121 +265,116 @@ int test_hashmap_multimap(void)
long found_msk; long found_msk;
int err, bkt; int err, bkt;
fprintf(stderr, "%s: ", __func__);
/* force collisions */ /* force collisions */
map = hashmap__new(collision_hash_fn, equal_fn, NULL); map = hashmap__new(collision_hash_fn, equal_fn, NULL);
if (CHECK(IS_ERR(map), "failed to create map: %ld\n", PTR_ERR(map))) if (CHECK(IS_ERR(map), "hashmap__new",
return 1; "failed to create map: %ld\n", PTR_ERR(map)))
return;
/* set up multimap: /* set up multimap:
* [0] -> 1, 2, 4; * [0] -> 1, 2, 4;
* [1] -> 8, 16, 32; * [1] -> 8, 16, 32;
*/ */
err = hashmap__append(map, k1, (void *)1); err = hashmap__append(map, k1, (void *)1);
if (CHECK(err, "failed to add k/v: %d\n", err)) if (CHECK(err, "elem_add", "failed to add k/v: %d\n", err))
return 1; goto cleanup;
err = hashmap__append(map, k1, (void *)2); err = hashmap__append(map, k1, (void *)2);
if (CHECK(err, "failed to add k/v: %d\n", err)) if (CHECK(err, "elem_add", "failed to add k/v: %d\n", err))
return 1; goto cleanup;
err = hashmap__append(map, k1, (void *)4); err = hashmap__append(map, k1, (void *)4);
if (CHECK(err, "failed to add k/v: %d\n", err)) if (CHECK(err, "elem_add", "failed to add k/v: %d\n", err))
return 1; goto cleanup;
err = hashmap__append(map, k2, (void *)8); err = hashmap__append(map, k2, (void *)8);
if (CHECK(err, "failed to add k/v: %d\n", err)) if (CHECK(err, "elem_add", "failed to add k/v: %d\n", err))
return 1; goto cleanup;
err = hashmap__append(map, k2, (void *)16); err = hashmap__append(map, k2, (void *)16);
if (CHECK(err, "failed to add k/v: %d\n", err)) if (CHECK(err, "elem_add", "failed to add k/v: %d\n", err))
return 1; goto cleanup;
err = hashmap__append(map, k2, (void *)32); err = hashmap__append(map, k2, (void *)32);
if (CHECK(err, "failed to add k/v: %d\n", err)) if (CHECK(err, "elem_add", "failed to add k/v: %d\n", err))
return 1; goto cleanup;
if (CHECK(hashmap__size(map) != 6, if (CHECK(hashmap__size(map) != 6, "hashmap_size",
"invalid map size: %zu\n", hashmap__size(map))) "invalid map size: %zu\n", hashmap__size(map)))
return 1; goto cleanup;
/* verify global iteration still works and sees all values */ /* verify global iteration still works and sees all values */
found_msk = 0; found_msk = 0;
hashmap__for_each_entry(map, entry, bkt) { hashmap__for_each_entry(map, entry, bkt) {
found_msk |= (long)entry->value; found_msk |= (long)entry->value;
} }
if (CHECK(found_msk != (1 << 6) - 1, if (CHECK(found_msk != (1 << 6) - 1, "found_msk",
"not all keys iterated: %lx\n", found_msk)) "not all keys iterated: %lx\n", found_msk))
return 1; goto cleanup;
/* iterate values for key 1 */ /* iterate values for key 1 */
found_msk = 0; found_msk = 0;
hashmap__for_each_key_entry(map, entry, k1) { hashmap__for_each_key_entry(map, entry, k1) {
found_msk |= (long)entry->value; found_msk |= (long)entry->value;
} }
if (CHECK(found_msk != (1 | 2 | 4), if (CHECK(found_msk != (1 | 2 | 4), "found_msk",
"invalid k1 values: %lx\n", found_msk)) "invalid k1 values: %lx\n", found_msk))
return 1; goto cleanup;
/* iterate values for key 2 */ /* iterate values for key 2 */
found_msk = 0; found_msk = 0;
hashmap__for_each_key_entry(map, entry, k2) { hashmap__for_each_key_entry(map, entry, k2) {
found_msk |= (long)entry->value; found_msk |= (long)entry->value;
} }
if (CHECK(found_msk != (8 | 16 | 32), if (CHECK(found_msk != (8 | 16 | 32), "found_msk",
"invalid k2 values: %lx\n", found_msk)) "invalid k2 values: %lx\n", found_msk))
return 1; goto cleanup;
fprintf(stderr, "OK\n"); cleanup:
return 0; hashmap__free(map);
} }
int test_hashmap_empty() static void test_hashmap_empty()
{ {
struct hashmap_entry *entry; struct hashmap_entry *entry;
int bkt; int bkt;
struct hashmap *map; struct hashmap *map;
void *k = (void *)0; void *k = (void *)0;
fprintf(stderr, "%s: ", __func__);
/* force collisions */ /* force collisions */
map = hashmap__new(hash_fn, equal_fn, NULL); map = hashmap__new(hash_fn, equal_fn, NULL);
if (CHECK(IS_ERR(map), "failed to create map: %ld\n", PTR_ERR(map))) if (CHECK(IS_ERR(map), "hashmap__new",
return 1; "failed to create map: %ld\n", PTR_ERR(map)))
goto cleanup;
if (CHECK(hashmap__size(map) != 0, if (CHECK(hashmap__size(map) != 0, "hashmap__size",
"invalid map size: %zu\n", hashmap__size(map))) "invalid map size: %zu\n", hashmap__size(map)))
return 1; goto cleanup;
if (CHECK(hashmap__capacity(map) != 0, if (CHECK(hashmap__capacity(map) != 0, "hashmap__capacity",
"invalid map capacity: %zu\n", hashmap__capacity(map))) "invalid map capacity: %zu\n", hashmap__capacity(map)))
return 1; goto cleanup;
if (CHECK(hashmap__find(map, k, NULL), "unexpected find\n")) if (CHECK(hashmap__find(map, k, NULL), "elem_find",
return 1; "unexpected find\n"))
if (CHECK(hashmap__delete(map, k, NULL, NULL), "unexpected delete\n")) goto cleanup;
return 1; if (CHECK(hashmap__delete(map, k, NULL, NULL), "elem_del",
"unexpected delete\n"))
goto cleanup;
hashmap__for_each_entry(map, entry, bkt) { hashmap__for_each_entry(map, entry, bkt) {
CHECK(false, "unexpected iterated entry\n"); CHECK(false, "elem_found", "unexpected iterated entry\n");
return 1; goto cleanup;
} }
hashmap__for_each_key_entry(map, entry, k) { hashmap__for_each_key_entry(map, entry, k) {
CHECK(false, "unexpected key entry\n"); CHECK(false, "key_found", "unexpected key entry\n");
return 1; goto cleanup;
} }
fprintf(stderr, "OK\n"); cleanup:
return 0; hashmap__free(map);
} }
int main(int argc, char **argv) void test_hashmap()
{ {
bool failed = false; if (test__start_subtest("generic"))
test_hashmap_generic();
if (test_hashmap_generic()) if (test__start_subtest("multimap"))
failed = true; test_hashmap_multimap();
if (test_hashmap_multimap()) if (test__start_subtest("empty"))
failed = true; test_hashmap_empty();
if (test_hashmap_empty())
failed = true;
return failed;
} }
...@@ -80,9 +80,6 @@ void test_ns_current_pid_tgid(void) ...@@ -80,9 +80,6 @@ void test_ns_current_pid_tgid(void)
"User pid/tgid %llu BPF pid/tgid %llu\n", id, bss.pid_tgid)) "User pid/tgid %llu BPF pid/tgid %llu\n", id, bss.pid_tgid))
goto cleanup; goto cleanup;
cleanup: cleanup:
if (!link) { bpf_link__destroy(link);
bpf_link__destroy(link);
link = NULL;
}
bpf_object__close(obj); bpf_object__close(obj);
} }
...@@ -6,6 +6,11 @@ ...@@ -6,6 +6,11 @@
#include <test_progs.h> #include <test_progs.h>
#include "bpf/libbpf_internal.h" #include "bpf/libbpf_internal.h"
/* AddressSanitizer sometimes crashes due to data dereference below, due to
* this being mmap()'ed memory. Disable instrumentation with
* no_sanitize_address attribute
*/
__attribute__((no_sanitize_address))
static void on_sample(void *ctx, int cpu, void *data, __u32 size) static void on_sample(void *ctx, int cpu, void *data, __u32 size)
{ {
int cpu_data = *(int *)data, duration = 0; int cpu_data = *(int *)data, duration = 0;
......
...@@ -351,6 +351,7 @@ int extract_build_id(char *build_id, size_t size) ...@@ -351,6 +351,7 @@ int extract_build_id(char *build_id, size_t size)
len = size; len = size;
memcpy(build_id, line, len); memcpy(build_id, line, len);
build_id[len] = '\0'; build_id[len] = '\0';
free(line);
return 0; return 0;
err: err:
fclose(fp); fclose(fp);
...@@ -420,6 +421,18 @@ static int libbpf_print_fn(enum libbpf_print_level level, ...@@ -420,6 +421,18 @@ static int libbpf_print_fn(enum libbpf_print_level level,
return 0; return 0;
} }
static void free_str_set(const struct str_set *set)
{
int i;
if (!set)
return;
for (i = 0; i < set->cnt; i++)
free((void *)set->strs[i]);
free(set->strs);
}
static int parse_str_list(const char *s, struct str_set *set) static int parse_str_list(const char *s, struct str_set *set)
{ {
char *input, *state = NULL, *next, **tmp, **strs = NULL; char *input, *state = NULL, *next, **tmp, **strs = NULL;
...@@ -756,11 +769,11 @@ int main(int argc, char **argv) ...@@ -756,11 +769,11 @@ int main(int argc, char **argv)
fprintf(stdout, "Summary: %d/%d PASSED, %d SKIPPED, %d FAILED\n", fprintf(stdout, "Summary: %d/%d PASSED, %d SKIPPED, %d FAILED\n",
env.succ_cnt, env.sub_succ_cnt, env.skip_cnt, env.fail_cnt); env.succ_cnt, env.sub_succ_cnt, env.skip_cnt, env.fail_cnt);
free(env.test_selector.blacklist.strs); free_str_set(&env.test_selector.blacklist);
free(env.test_selector.whitelist.strs); free_str_set(&env.test_selector.whitelist);
free(env.test_selector.num_set); free(env.test_selector.num_set);
free(env.subtest_selector.blacklist.strs); free_str_set(&env.subtest_selector.blacklist);
free(env.subtest_selector.whitelist.strs); free_str_set(&env.subtest_selector.whitelist);
free(env.subtest_selector.num_set); free(env.subtest_selector.num_set);
return env.fail_cnt ? EXIT_FAILURE : EXIT_SUCCESS; return env.fail_cnt ? EXIT_FAILURE : EXIT_SUCCESS;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment