Something went wrong. Try again.
Linux kernel mirror (for testing) git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166// SPDX-License-Identifier: GPL-2.0
#include <stdio.h>#include <errno.h>#include <string.h>#include <unistd.h>
#include <bpf/bpf.h>#include <bpf/libbpf.h>
#include <test_maps.h>
static int nr_cpus;
static void map_batch_update(int map_fd, __u32 max_entries, int *keys, __s64 *values, bool is_pcpu){ int i, j, err; int cpu_offset = 0; DECLARE_LIBBPF_OPTS(bpf_map_batch_opts, opts, .elem_flags = 0, .flags = 0, );
for (i = 0; i < max_entries; i++) { keys[i] = i; if (is_pcpu) { cpu_offset = i * nr_cpus; for (j = 0; j < nr_cpus; j++) (values + cpu_offset)[j] = i + 1 + j; } else { values[i] = i + 1; } }
err = bpf_map_update_batch(map_fd, keys, values, &max_entries, &opts); CHECK(err, "bpf_map_update_batch()", "error:%s\n", strerror(errno));}
static void map_batch_verify(int *visited, __u32 max_entries, int *keys, __s64 *values, bool is_pcpu){ int i, j; int cpu_offset = 0;
memset(visited, 0, max_entries * sizeof(*visited)); for (i = 0; i < max_entries; i++) { if (is_pcpu) { cpu_offset = i * nr_cpus; for (j = 0; j < nr_cpus; j++) { __s64 value = (values + cpu_offset)[j]; CHECK(keys[i] + j + 1 != value, "key/value checking", "error: i %d j %d key %d value %lld\n", i, j, keys[i], value); } } else { CHECK(keys[i] + 1 != values[i], "key/value checking", "error: i %d key %d value %lld\n", i, keys[i], values[i]); } visited[i] = 1; } for (i = 0; i < max_entries; i++) { CHECK(visited[i] != 1, "visited checking", "error: keys array at index %d missing\n", i); }}
static void __test_map_lookup_and_update_batch(bool is_pcpu){ int map_fd, *keys, *visited; __u32 count, total, total_success; const __u32 max_entries = 10; __u64 batch = 0; int err, step, value_size; void *values; DECLARE_LIBBPF_OPTS(bpf_map_batch_opts, opts, .elem_flags = 0, .flags = 0, );
map_fd = bpf_map_create(is_pcpu ? BPF_MAP_TYPE_PERCPU_ARRAY : BPF_MAP_TYPE_ARRAY, "array_map", sizeof(int), sizeof(__s64), max_entries, NULL); CHECK(map_fd == -1, "bpf_map_create()", "error:%s\n", strerror(errno));
value_size = sizeof(__s64); if (is_pcpu) value_size *= nr_cpus;
keys = calloc(max_entries, sizeof(*keys)); values = calloc(max_entries, value_size); visited = calloc(max_entries, sizeof(*visited)); CHECK(!keys || !values || !visited, "malloc()", "error:%s\n", strerror(errno));
/* test 1: lookup in a loop with various steps. */ total_success = 0; for (step = 1; step < max_entries; step++) { map_batch_update(map_fd, max_entries, keys, values, is_pcpu); map_batch_verify(visited, max_entries, keys, values, is_pcpu); memset(keys, 0, max_entries * sizeof(*keys)); memset(values, 0, max_entries * value_size); batch = 0; total = 0; /* iteratively lookup/delete elements with 'step' * elements each. */ count = step; while (true) { err = bpf_map_lookup_batch(map_fd, total ? &batch : NULL, &batch, keys + total, values + total * value_size, &count, &opts);
CHECK((err && errno != ENOENT), "lookup with steps", "error: %s\n", strerror(errno));
total += count; if (err) break;
}
CHECK(total != max_entries, "lookup with steps", "total = %u, max_entries = %u\n", total, max_entries);
map_batch_verify(visited, max_entries, keys, values, is_pcpu);
total_success++; }
CHECK(total_success == 0, "check total_success", "unexpected failure\n");
free(keys); free(values); free(visited); close(map_fd);}
static void array_map_batch_ops(void){ __test_map_lookup_and_update_batch(false); printf("test_%s:PASS\n", __func__);}
static void array_percpu_map_batch_ops(void){ __test_map_lookup_and_update_batch(true); printf("test_%s:PASS\n", __func__);}
void test_array_map_batch_ops(void){ nr_cpus = libbpf_num_possible_cpus();
CHECK(nr_cpus < 0, "nr_cpus checking", "error: get possible cpus failed");
array_map_batch_ops(); array_percpu_map_batch_ops();}