126b367e3SAlexei Starovoitov // SPDX-License-Identifier: GPL-2.0
226b367e3SAlexei Starovoitov /* Copyright (c) 2021 Facebook */
326b367e3SAlexei Starovoitov #include "vmlinux.h"
426b367e3SAlexei Starovoitov
526b367e3SAlexei Starovoitov #include <bpf/bpf_helpers.h>
626b367e3SAlexei Starovoitov #include <bpf/bpf_tracing.h>
726b367e3SAlexei Starovoitov #include <bpf/bpf_core_read.h>
826b367e3SAlexei Starovoitov
926b367e3SAlexei Starovoitov #define ATTR __always_inline
1026b367e3SAlexei Starovoitov #include "test_jhash.h"
1126b367e3SAlexei Starovoitov
1226b367e3SAlexei Starovoitov struct {
1326b367e3SAlexei Starovoitov __uint(type, BPF_MAP_TYPE_ARRAY);
1426b367e3SAlexei Starovoitov __type(key, u32);
1526b367e3SAlexei Starovoitov __type(value, u32);
1626b367e3SAlexei Starovoitov __uint(max_entries, 256);
1726b367e3SAlexei Starovoitov } array1 SEC(".maps");
1826b367e3SAlexei Starovoitov
1926b367e3SAlexei Starovoitov struct {
2026b367e3SAlexei Starovoitov __uint(type, BPF_MAP_TYPE_ARRAY);
2126b367e3SAlexei Starovoitov __type(key, u32);
2226b367e3SAlexei Starovoitov __type(value, u32);
2326b367e3SAlexei Starovoitov __uint(max_entries, 256);
2426b367e3SAlexei Starovoitov } array2 SEC(".maps");
2526b367e3SAlexei Starovoitov
randmap(int v,const struct net_device * dev)2626b367e3SAlexei Starovoitov static __noinline int randmap(int v, const struct net_device *dev)
2726b367e3SAlexei Starovoitov {
2826b367e3SAlexei Starovoitov struct bpf_map *map = (struct bpf_map *)&array1;
2926b367e3SAlexei Starovoitov int key = bpf_get_prandom_u32() & 0xff;
3026b367e3SAlexei Starovoitov int *val;
3126b367e3SAlexei Starovoitov
3226b367e3SAlexei Starovoitov if (bpf_get_prandom_u32() & 1)
3326b367e3SAlexei Starovoitov map = (struct bpf_map *)&array2;
3426b367e3SAlexei Starovoitov
3526b367e3SAlexei Starovoitov val = bpf_map_lookup_elem(map, &key);
3626b367e3SAlexei Starovoitov if (val)
3726b367e3SAlexei Starovoitov *val = bpf_get_prandom_u32() + v + dev->mtu;
3826b367e3SAlexei Starovoitov
3926b367e3SAlexei Starovoitov return 0;
4026b367e3SAlexei Starovoitov }
4126b367e3SAlexei Starovoitov
4226b367e3SAlexei Starovoitov SEC("tp_btf/xdp_devmap_xmit")
BPF_PROG(tp_xdp_devmap_xmit_multi,const struct net_device * from_dev,const struct net_device * to_dev,int sent,int drops,int err)4326b367e3SAlexei Starovoitov int BPF_PROG(tp_xdp_devmap_xmit_multi, const struct net_device
4426b367e3SAlexei Starovoitov *from_dev, const struct net_device *to_dev, int sent, int drops,
4526b367e3SAlexei Starovoitov int err)
4626b367e3SAlexei Starovoitov {
4726b367e3SAlexei Starovoitov return randmap(from_dev->ifindex, from_dev);
4826b367e3SAlexei Starovoitov }
4926b367e3SAlexei Starovoitov
5026b367e3SAlexei Starovoitov SEC("fentry/eth_type_trans")
BPF_PROG(fentry_eth_type_trans,struct sk_buff * skb,struct net_device * dev,unsigned short protocol)5126b367e3SAlexei Starovoitov int BPF_PROG(fentry_eth_type_trans, struct sk_buff *skb,
5226b367e3SAlexei Starovoitov struct net_device *dev, unsigned short protocol)
5326b367e3SAlexei Starovoitov {
5426b367e3SAlexei Starovoitov return randmap(dev->ifindex + skb->len, dev);
5526b367e3SAlexei Starovoitov }
5626b367e3SAlexei Starovoitov
5726b367e3SAlexei Starovoitov SEC("fexit/eth_type_trans")
BPF_PROG(fexit_eth_type_trans,struct sk_buff * skb,struct net_device * dev,unsigned short protocol)5826b367e3SAlexei Starovoitov int BPF_PROG(fexit_eth_type_trans, struct sk_buff *skb,
5926b367e3SAlexei Starovoitov struct net_device *dev, unsigned short protocol)
6026b367e3SAlexei Starovoitov {
6126b367e3SAlexei Starovoitov return randmap(dev->ifindex + skb->len, dev);
6226b367e3SAlexei Starovoitov }
6326b367e3SAlexei Starovoitov
6426b367e3SAlexei Starovoitov volatile const int never;
6526b367e3SAlexei Starovoitov
6626b367e3SAlexei Starovoitov struct __sk_bUfF /* it will not exist in vmlinux */ {
6726b367e3SAlexei Starovoitov int len;
6826b367e3SAlexei Starovoitov } __attribute__((preserve_access_index));
6926b367e3SAlexei Starovoitov
7026b367e3SAlexei Starovoitov struct bpf_testmod_test_read_ctx /* it exists in bpf_testmod */ {
7126b367e3SAlexei Starovoitov size_t len;
7226b367e3SAlexei Starovoitov } __attribute__((preserve_access_index));
7326b367e3SAlexei Starovoitov
7426b367e3SAlexei Starovoitov SEC("tc")
balancer_ingress(struct __sk_buff * ctx)7526b367e3SAlexei Starovoitov int balancer_ingress(struct __sk_buff *ctx)
7626b367e3SAlexei Starovoitov {
7726b367e3SAlexei Starovoitov void *data_end = (void *)(long)ctx->data_end;
7826b367e3SAlexei Starovoitov void *data = (void *)(long)ctx->data;
7926b367e3SAlexei Starovoitov void *ptr;
80*c8ed6685SAndrii Nakryiko int nh_off, i = 0;
8126b367e3SAlexei Starovoitov
8226b367e3SAlexei Starovoitov nh_off = 14;
8326b367e3SAlexei Starovoitov
8426b367e3SAlexei Starovoitov /* pragma unroll doesn't work on large loops */
8526b367e3SAlexei Starovoitov #define C do { \
8626b367e3SAlexei Starovoitov ptr = data + i; \
8726b367e3SAlexei Starovoitov if (ptr + nh_off > data_end) \
8826b367e3SAlexei Starovoitov break; \
8926b367e3SAlexei Starovoitov ctx->tc_index = jhash(ptr, nh_off, ctx->cb[0] + i++); \
9026b367e3SAlexei Starovoitov if (never) { \
9126b367e3SAlexei Starovoitov /* below is a dead code with unresolvable CO-RE relo */ \
9226b367e3SAlexei Starovoitov i += ((struct __sk_bUfF *)ctx)->len; \
9326b367e3SAlexei Starovoitov /* this CO-RE relo may or may not resolve
9426b367e3SAlexei Starovoitov * depending on whether bpf_testmod is loaded.
9526b367e3SAlexei Starovoitov */ \
9626b367e3SAlexei Starovoitov i += ((struct bpf_testmod_test_read_ctx *)ctx)->len; \
9726b367e3SAlexei Starovoitov } \
9826b367e3SAlexei Starovoitov } while (0);
9926b367e3SAlexei Starovoitov #define C30 C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;C;
10026b367e3SAlexei Starovoitov C30;C30;C30; /* 90 calls */
10126b367e3SAlexei Starovoitov return 0;
10226b367e3SAlexei Starovoitov }
10326b367e3SAlexei Starovoitov
104976a38e0SMatteo Croce typedef int (*func_proto_typedef___match)(long);
105976a38e0SMatteo Croce typedef int (*func_proto_typedef___doesnt_match)(char *);
106976a38e0SMatteo Croce typedef int (*func_proto_typedef_nested1)(func_proto_typedef___match);
107976a38e0SMatteo Croce
108976a38e0SMatteo Croce int proto_out[3];
109976a38e0SMatteo Croce
110976a38e0SMatteo Croce SEC("raw_tracepoint/sys_enter")
core_relo_proto(void * ctx)111976a38e0SMatteo Croce int core_relo_proto(void *ctx)
112976a38e0SMatteo Croce {
113976a38e0SMatteo Croce proto_out[0] = bpf_core_type_exists(func_proto_typedef___match);
114976a38e0SMatteo Croce proto_out[1] = bpf_core_type_exists(func_proto_typedef___doesnt_match);
115976a38e0SMatteo Croce proto_out[2] = bpf_core_type_exists(func_proto_typedef_nested1);
116976a38e0SMatteo Croce
117976a38e0SMatteo Croce return 0;
118976a38e0SMatteo Croce }
119976a38e0SMatteo Croce
12026b367e3SAlexei Starovoitov char LICENSE[] SEC("license") = "GPL";
121