160dd49eaSYonghong Song // SPDX-License-Identifier: GPL-2.0
260dd49eaSYonghong Song /* Copyright (c) 2020 Facebook */
360dd49eaSYonghong Song #include "bpf_iter.h"
460dd49eaSYonghong Song #include <bpf/bpf_helpers.h>
560dd49eaSYonghong Song #include <bpf/bpf_tracing.h>
660dd49eaSYonghong Song 
760dd49eaSYonghong Song char _license[] SEC("license") = "GPL";
860dd49eaSYonghong Song 
960dd49eaSYonghong Song struct key_t {
1060dd49eaSYonghong Song 	int a;
1160dd49eaSYonghong Song 	int b;
1260dd49eaSYonghong Song 	int c;
1360dd49eaSYonghong Song };
1460dd49eaSYonghong Song 
1560dd49eaSYonghong Song struct {
1660dd49eaSYonghong Song 	__uint(type, BPF_MAP_TYPE_PERCPU_ARRAY);
1760dd49eaSYonghong Song 	__uint(max_entries, 3);
1860dd49eaSYonghong Song 	__type(key, __u32);
1960dd49eaSYonghong Song 	__type(value, __u32);
2060dd49eaSYonghong Song } arraymap1 SEC(".maps");
2160dd49eaSYonghong Song 
2260dd49eaSYonghong Song /* will set before prog run */
2360dd49eaSYonghong Song volatile const __u32 num_cpus = 0;
2460dd49eaSYonghong Song 
2560dd49eaSYonghong Song __u32 key_sum = 0, val_sum = 0;
2660dd49eaSYonghong Song 
2760dd49eaSYonghong Song SEC("iter/bpf_map_elem")
dump_bpf_percpu_array_map(struct bpf_iter__bpf_map_elem * ctx)2860dd49eaSYonghong Song int dump_bpf_percpu_array_map(struct bpf_iter__bpf_map_elem *ctx)
2960dd49eaSYonghong Song {
3060dd49eaSYonghong Song 	__u32 *key = ctx->key;
3160dd49eaSYonghong Song 	void *pptr = ctx->value;
3260dd49eaSYonghong Song 	__u32 step;
3360dd49eaSYonghong Song 	int i;
3460dd49eaSYonghong Song 
3560dd49eaSYonghong Song 	if (key == (void *)0 || pptr == (void *)0)
3660dd49eaSYonghong Song 		return 0;
3760dd49eaSYonghong Song 
3860dd49eaSYonghong Song 	key_sum += *key;
3960dd49eaSYonghong Song 
4060dd49eaSYonghong Song 	step = 8;
4160dd49eaSYonghong Song 	for (i = 0; i < num_cpus; i++) {
4260dd49eaSYonghong Song 		val_sum += *(__u32 *)pptr;
4360dd49eaSYonghong Song 		pptr += step;
4460dd49eaSYonghong Song 	}
4560dd49eaSYonghong Song 	return 0;
4660dd49eaSYonghong Song }
47