Groups | Search | Server Info | Keyboard shortcuts | Login | Register [http] [https] [nntp] [nntps]
Groups > linux.kernel > #1306445
| From | Ming Lei <tom.leiming@gmail.com> |
|---|---|
| Newsgroups | linux.kernel |
| Subject | [PATCH 9/9] samples/bpf: test percpu array map |
| Date | 2016-01-11 17:00 +0100 |
| Message-ID | <qPN2W-4pP-33@gated-at.bofh.it> (permalink) |
| References | <qPN2V-4pP-5@gated-at.bofh.it> |
| Organization | linux.* mail to news gateway |
Signed-off-by: Ming Lei <tom.leiming@gmail.com>
---
samples/bpf/test_maps.c | 110 ++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 110 insertions(+)
diff --git a/samples/bpf/test_maps.c b/samples/bpf/test_maps.c
index 6299ee9..ff5d373 100644
--- a/samples/bpf/test_maps.c
+++ b/samples/bpf/test_maps.c
@@ -142,6 +142,106 @@ static void test_arraymap_sanity(int i, void *data)
close(map_fd);
}
+static int handle_one_cpu(unsigned cpu, void *val_cpu, void *val)
+{
+ unsigned long *cnt = val;
+
+ *cnt += *(long *)val_cpu;
+ return 0;
+}
+
+
+static void test_percpu_arraymap_all_cpu_sanity(int i)
+{
+ int key, map_fd;
+ unsigned long value = 0;
+ unsigned long val_cpu;
+ unsigned cpu;
+ unsigned nr_keys = 2;
+ unsigned nr_cpus = sysconf(_SC_NPROCESSORS_CONF);
+
+ map_fd = bpf_create_map(BPF_MAP_TYPE_ARRAY_PERCPU, sizeof(key),
+ sizeof(value), nr_keys);
+ if (map_fd < 0) {
+ printf("failed to create arraymap '%s'\n", strerror(errno));
+ exit(1);
+ }
+
+ for (key = 0; key < nr_keys; key++) {
+ for (cpu = 0; cpu < nr_cpus; cpu++) {
+ val_cpu = key;
+ assert(bpf_update_elem_percpu(map_fd, &key, &val_cpu,
+ BPF_ANY, cpu) == 0);
+ }
+ }
+
+ for (key = 0; key < nr_keys; key++) {
+ assert(bpf_lookup_elem_allcpu(map_fd, &key, &val_cpu,
+ &value, handle_one_cpu) == 0);
+ assert(value == nr_cpus * key);
+ }
+
+ close(map_fd);
+}
+
+static void test_percpu_arraymap_single_cpu_sanity(int i, void *data)
+{
+ int key, next_key, map_fd;
+ long long value;
+ unsigned cpu = *(unsigned *)data;
+
+ map_fd = bpf_create_map(BPF_MAP_TYPE_ARRAY_PERCPU, sizeof(key),
+ sizeof(value), 2);
+ if (map_fd < 0) {
+ printf("failed to create arraymap '%s'\n", strerror(errno));
+ exit(1);
+ }
+
+ key = 1;
+ value = 1234;
+ /* insert key=1 element */
+ assert(bpf_update_elem_percpu(map_fd, &key, &value, BPF_ANY, cpu) == 0);
+
+ value = 0;
+ assert(bpf_update_elem_percpu(map_fd, &key, &value, BPF_NOEXIST,
+ cpu) == -1 && errno == EEXIST);
+
+ /* check that key=1 can be found */
+ assert(bpf_lookup_elem_percpu(map_fd, &key, &value, cpu) == 0 &&
+ value == 1234);
+
+ key = 0;
+ /* check that key=0 is also found and zero initialized */
+ assert(bpf_lookup_elem_percpu(map_fd, &key, &value, cpu) == 0 &&
+ value == 0);
+
+
+ /* key=0 and key=1 were inserted, check that key=2 cannot be inserted
+ * due to max_entries limit
+ */
+ key = 2;
+ assert(bpf_update_elem_percpu(map_fd, &key, &value, BPF_EXIST,
+ cpu) == -1 && errno == E2BIG);
+
+ /* check that key = 2 doesn't exist */
+ assert(bpf_lookup_elem_percpu(map_fd, &key, &value, cpu) == -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)
static void test_map_large(void)
{
@@ -281,8 +381,18 @@ static void test_map_parallel(void)
int main(void)
{
+ int cpu;
+ unsigned data;
+
test_hashmap_sanity(0, NULL);
test_arraymap_sanity(0, NULL);
+
+ for (cpu = 0; cpu < sysconf(_SC_NPROCESSORS_CONF); cpu++) {
+ data = cpu;
+ test_percpu_arraymap_single_cpu_sanity(0, &data);
+ }
+ test_percpu_arraymap_all_cpu_sanity(0);
+
test_map_large();
test_map_parallel();
test_map_stress();
--
1.9.1
Back to linux.kernel | Previous | Next — Previous in thread | Next in thread | Find similar | Unroll thread
[PATCH 0/9] bpf: support percpu ARRAY map Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
[PATCH 3/9] bpf: introduce percpu verion of lookup/update in bpf_map_ops Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
[PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Alexei Starovoitov <alexei.starovoitov@gmail.com> - 2016-01-11 20:10 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-12 06:10 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Alexei Starovoitov <alexei.starovoitov@gmail.com> - 2016-01-12 06:50 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-12 12:10 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Martin KaFai Lau <kafai@fb.com> - 2016-01-12 20:20 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-13 01:40 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Martin KaFai Lau <kafai@fb.com> - 2016-01-13 03:30 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-13 04:20 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Alexei Starovoitov <alexei.starovoitov@gmail.com> - 2016-01-13 06:30 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-13 16:00 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Alexei Starovoitov <alexei.starovoitov@gmail.com> - 2016-01-14 02:30 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-14 03:50 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Alexei Starovoitov <alexei.starovoitov@gmail.com> - 2016-01-14 06:10 +0100
Re: [PATCH 5/9] bpf: syscall: add percpu version of lookup/update elem Ming Lei <tom.leiming@gmail.com> - 2016-01-14 08:20 +0100
[PATCH 4/9] bpf: add percpu version of lookup/update element helpers Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
[PATCH 6/9] bpf: arraymap: introduce BPF_MAP_TYPE_ARRAY_PERCPU Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
Re: [PATCH 6/9] bpf: arraymap: introduce BPF_MAP_TYPE_ARRAY_PERCPU Alexei Starovoitov <alexei.starovoitov@gmail.com> - 2016-01-11 20:20 +0100
[PATCH 7/9] sample/bpf: introduces helpers for percpu array example Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
[PATCH 9/9] samples/bpf: test percpu array map Ming Lei <tom.leiming@gmail.com> - 2016-01-11 17:00 +0100
RE: [PATCH 9/9] samples/bpf: test percpu array map David Laight <David.Laight@ACULAB.COM> - 2016-01-12 16:50 +0100
csiph-web