1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2020 Facebook */
3 #include "bpf_iter.h"
4 #include "bpf_tracing_net.h"
5 #include <bpf/bpf_helpers.h>
6 #include <bpf/bpf_tracing.h>
7 
8 char _license[] SEC("license") = "GPL";
9 
10 struct {
11 	__uint(type, BPF_MAP_TYPE_SK_STORAGE);
12 	__uint(map_flags, BPF_F_NO_PREALLOC);
13 	__type(key, int);
14 	__type(value, int);
15 } sk_stg_map SEC(".maps");
16 
17 __u32 val_sum = 0;
18 __u32 ipv6_sk_count = 0;
19 
20 SEC("iter/bpf_sk_storage_map")
21 int dump_bpf_sk_storage_map(struct bpf_iter__bpf_sk_storage_map *ctx)
22 {
23 	struct sock *sk = ctx->sk;
24 	__u32 *val = ctx->value;
25 
26 	if (sk == (void *)0 || val == (void *)0)
27 		return 0;
28 
29 	if (sk->sk_family == AF_INET6)
30 		ipv6_sk_count++;
31 
32 	val_sum += *val;
33 	return 0;
34 }
35