59 lines
1.7 KiB
C
59 lines
1.7 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2022 Bytedance */
|
|
|
|
#include <test_progs.h>
|
|
#include "test_map_lookup_percpu_elem.skel.h"
|
|
|
|
void test_map_lookup_percpu_elem(void)
|
|
{
|
|
struct test_map_lookup_percpu_elem *skel;
|
|
__u64 key = 0, sum;
|
|
int ret, i, nr_cpus = libbpf_num_possible_cpus();
|
|
__u64 *buf;
|
|
|
|
buf = malloc(nr_cpus*sizeof(__u64));
|
|
if (!ASSERT_OK_PTR(buf, "malloc"))
|
|
return;
|
|
|
|
for (i = 0; i < nr_cpus; i++)
|
|
buf[i] = i;
|
|
sum = (nr_cpus - 1) * nr_cpus / 2;
|
|
|
|
skel = test_map_lookup_percpu_elem__open();
|
|
if (!ASSERT_OK_PTR(skel, "test_map_lookup_percpu_elem__open"))
|
|
goto exit;
|
|
|
|
skel->rodata->my_pid = getpid();
|
|
skel->rodata->nr_cpus = nr_cpus;
|
|
|
|
ret = test_map_lookup_percpu_elem__load(skel);
|
|
if (!ASSERT_OK(ret, "test_map_lookup_percpu_elem__load"))
|
|
goto cleanup;
|
|
|
|
ret = test_map_lookup_percpu_elem__attach(skel);
|
|
if (!ASSERT_OK(ret, "test_map_lookup_percpu_elem__attach"))
|
|
goto cleanup;
|
|
|
|
ret = bpf_map_update_elem(bpf_map__fd(skel->maps.percpu_array_map), &key, buf, 0);
|
|
ASSERT_OK(ret, "percpu_array_map update");
|
|
|
|
ret = bpf_map_update_elem(bpf_map__fd(skel->maps.percpu_hash_map), &key, buf, 0);
|
|
ASSERT_OK(ret, "percpu_hash_map update");
|
|
|
|
ret = bpf_map_update_elem(bpf_map__fd(skel->maps.percpu_lru_hash_map), &key, buf, 0);
|
|
ASSERT_OK(ret, "percpu_lru_hash_map update");
|
|
|
|
syscall(__NR_getuid);
|
|
|
|
test_map_lookup_percpu_elem__detach(skel);
|
|
|
|
ASSERT_EQ(skel->bss->percpu_array_elem_sum, sum, "percpu_array lookup percpu elem");
|
|
ASSERT_EQ(skel->bss->percpu_hash_elem_sum, sum, "percpu_hash lookup percpu elem");
|
|
ASSERT_EQ(skel->bss->percpu_lru_hash_elem_sum, sum, "percpu_lru_hash lookup percpu elem");
|
|
|
|
cleanup:
|
|
test_map_lookup_percpu_elem__destroy(skel);
|
|
exit:
|
|
free(buf);
|
|
}
|