mirror of
https://github.com/torvalds/linux.git
synced 2024-11-27 14:41:39 +00:00
samples/bpf: unit test for BPF_MAP_TYPE_PERCPU_ARRAY
A sanity test for BPF_MAP_TYPE_PERCPU_ARRAY Signed-off-by: Ming Lei <tom.leiming@gmail.com> Signed-off-by: Alexei Starovoitov <ast@kernel.org> Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
parent
e155967179
commit
df570f5772
@ -236,6 +236,94 @@ static void test_arraymap_sanity(int i, void *data)
|
|||||||
close(map_fd);
|
close(map_fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void test_percpu_arraymap_many_keys(void)
|
||||||
|
{
|
||||||
|
unsigned nr_cpus = sysconf(_SC_NPROCESSORS_CONF);
|
||||||
|
unsigned nr_keys = 20000;
|
||||||
|
long values[nr_cpus];
|
||||||
|
int key, map_fd, i;
|
||||||
|
|
||||||
|
map_fd = bpf_create_map(BPF_MAP_TYPE_PERCPU_ARRAY, sizeof(key),
|
||||||
|
sizeof(values[0]), nr_keys);
|
||||||
|
if (map_fd < 0) {
|
||||||
|
printf("failed to create per-cpu arraymap '%s'\n",
|
||||||
|
strerror(errno));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < nr_cpus; i++)
|
||||||
|
values[i] = i + 10;
|
||||||
|
|
||||||
|
for (key = 0; key < nr_keys; key++)
|
||||||
|
assert(bpf_update_elem(map_fd, &key, values, BPF_ANY) == 0);
|
||||||
|
|
||||||
|
for (key = 0; key < nr_keys; key++) {
|
||||||
|
for (i = 0; i < nr_cpus; i++)
|
||||||
|
values[i] = 0;
|
||||||
|
assert(bpf_lookup_elem(map_fd, &key, values) == 0);
|
||||||
|
for (i = 0; i < nr_cpus; i++)
|
||||||
|
assert(values[i] == i + 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
close(map_fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void test_percpu_arraymap_sanity(int i, void *data)
|
||||||
|
{
|
||||||
|
unsigned nr_cpus = sysconf(_SC_NPROCESSORS_CONF);
|
||||||
|
long values[nr_cpus];
|
||||||
|
int key, next_key, map_fd;
|
||||||
|
|
||||||
|
map_fd = bpf_create_map(BPF_MAP_TYPE_PERCPU_ARRAY, sizeof(key),
|
||||||
|
sizeof(values[0]), 2);
|
||||||
|
if (map_fd < 0) {
|
||||||
|
printf("failed to create arraymap '%s'\n", strerror(errno));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < nr_cpus; i++)
|
||||||
|
values[i] = i + 100;
|
||||||
|
|
||||||
|
key = 1;
|
||||||
|
/* insert key=1 element */
|
||||||
|
assert(bpf_update_elem(map_fd, &key, values, BPF_ANY) == 0);
|
||||||
|
|
||||||
|
values[0] = 0;
|
||||||
|
assert(bpf_update_elem(map_fd, &key, values, BPF_NOEXIST) == -1 &&
|
||||||
|
errno == EEXIST);
|
||||||
|
|
||||||
|
/* check that key=1 can be found */
|
||||||
|
assert(bpf_lookup_elem(map_fd, &key, values) == 0 && values[0] == 100);
|
||||||
|
|
||||||
|
key = 0;
|
||||||
|
/* check that key=0 is also found and zero initialized */
|
||||||
|
assert(bpf_lookup_elem(map_fd, &key, values) == 0 &&
|
||||||
|
values[0] == 0 && values[nr_cpus - 1] == 0);
|
||||||
|
|
||||||
|
|
||||||
|
/* check that key=2 cannot be inserted due to max_entries limit */
|
||||||
|
key = 2;
|
||||||
|
assert(bpf_update_elem(map_fd, &key, values, BPF_EXIST) == -1 &&
|
||||||
|
errno == E2BIG);
|
||||||
|
|
||||||
|
/* check that key = 2 doesn't exist */
|
||||||
|
assert(bpf_lookup_elem(map_fd, &key, values) == -1 && errno == ENOENT);
|
||||||
|
|
||||||
|
/* iterate over two elements */
|
||||||
|
assert(bpf_get_next_key(map_fd, &key, &next_key) == 0 &&
|
||||||
|
next_key == 0);
|
||||||
|
assert(bpf_get_next_key(map_fd, &next_key, &next_key) == 0 &&
|
||||||
|
next_key == 1);
|
||||||
|
assert(bpf_get_next_key(map_fd, &next_key, &next_key) == -1 &&
|
||||||
|
errno == ENOENT);
|
||||||
|
|
||||||
|
/* delete shouldn't succeed */
|
||||||
|
key = 1;
|
||||||
|
assert(bpf_delete_elem(map_fd, &key) == -1 && errno == EINVAL);
|
||||||
|
|
||||||
|
close(map_fd);
|
||||||
|
}
|
||||||
|
|
||||||
#define MAP_SIZE (32 * 1024)
|
#define MAP_SIZE (32 * 1024)
|
||||||
static void test_map_large(void)
|
static void test_map_large(void)
|
||||||
{
|
{
|
||||||
@ -305,6 +393,7 @@ static void test_map_stress(void)
|
|||||||
run_parallel(100, test_hashmap_sanity, NULL);
|
run_parallel(100, test_hashmap_sanity, NULL);
|
||||||
run_parallel(100, test_percpu_hashmap_sanity, NULL);
|
run_parallel(100, test_percpu_hashmap_sanity, NULL);
|
||||||
run_parallel(100, test_arraymap_sanity, NULL);
|
run_parallel(100, test_arraymap_sanity, NULL);
|
||||||
|
run_parallel(100, test_percpu_arraymap_sanity, NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
#define TASKS 1024
|
#define TASKS 1024
|
||||||
@ -379,6 +468,9 @@ int main(void)
|
|||||||
test_hashmap_sanity(0, NULL);
|
test_hashmap_sanity(0, NULL);
|
||||||
test_percpu_hashmap_sanity(0, NULL);
|
test_percpu_hashmap_sanity(0, NULL);
|
||||||
test_arraymap_sanity(0, NULL);
|
test_arraymap_sanity(0, NULL);
|
||||||
|
test_percpu_arraymap_sanity(0, NULL);
|
||||||
|
test_percpu_arraymap_many_keys();
|
||||||
|
|
||||||
test_map_large();
|
test_map_large();
|
||||||
test_map_parallel();
|
test_map_parallel();
|
||||||
test_map_stress();
|
test_map_stress();
|
||||||
|
Loading…
Reference in New Issue
Block a user