1*57ddeb86SDavid Vernet // SPDX-License-Identifier: GPL-2.0
2*57ddeb86SDavid Vernet /* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */
3*57ddeb86SDavid Vernet 
4*57ddeb86SDavid Vernet #include <linux/bpf.h>
5*57ddeb86SDavid Vernet #include <time.h>
6*57ddeb86SDavid Vernet #include <errno.h>
7*57ddeb86SDavid Vernet #include <bpf/bpf_helpers.h>
8*57ddeb86SDavid Vernet #include "bpf_misc.h"
9*57ddeb86SDavid Vernet #include "bpf_tcp_helpers.h"
10*57ddeb86SDavid Vernet 
11*57ddeb86SDavid Vernet char _license[] SEC("license") = "GPL";
12*57ddeb86SDavid Vernet 
13*57ddeb86SDavid Vernet struct elem {
14*57ddeb86SDavid Vernet 	struct bpf_timer t;
15*57ddeb86SDavid Vernet };
16*57ddeb86SDavid Vernet 
17*57ddeb86SDavid Vernet struct {
18*57ddeb86SDavid Vernet 	__uint(type, BPF_MAP_TYPE_ARRAY);
19*57ddeb86SDavid Vernet 	__uint(max_entries, 1);
20*57ddeb86SDavid Vernet 	__type(key, int);
21*57ddeb86SDavid Vernet 	__type(value, struct elem);
22*57ddeb86SDavid Vernet } timer_map SEC(".maps");
23*57ddeb86SDavid Vernet 
timer_cb_ret1(void * map,int * key,struct bpf_timer * timer)24*57ddeb86SDavid Vernet static int timer_cb_ret1(void *map, int *key, struct bpf_timer *timer)
25*57ddeb86SDavid Vernet {
26*57ddeb86SDavid Vernet 	if (bpf_get_smp_processor_id() % 2)
27*57ddeb86SDavid Vernet 		return 1;
28*57ddeb86SDavid Vernet 	else
29*57ddeb86SDavid Vernet 		return 0;
30*57ddeb86SDavid Vernet }
31*57ddeb86SDavid Vernet 
32*57ddeb86SDavid Vernet SEC("fentry/bpf_fentry_test1")
33*57ddeb86SDavid Vernet __failure __msg("should have been in (0x0; 0x0)")
BPF_PROG2(test_ret_1,int,a)34*57ddeb86SDavid Vernet int BPF_PROG2(test_ret_1, int, a)
35*57ddeb86SDavid Vernet {
36*57ddeb86SDavid Vernet 	int key = 0;
37*57ddeb86SDavid Vernet 	struct bpf_timer *timer;
38*57ddeb86SDavid Vernet 
39*57ddeb86SDavid Vernet 	timer = bpf_map_lookup_elem(&timer_map, &key);
40*57ddeb86SDavid Vernet 	if (timer) {
41*57ddeb86SDavid Vernet 		bpf_timer_init(timer, &timer_map, CLOCK_BOOTTIME);
42*57ddeb86SDavid Vernet 		bpf_timer_set_callback(timer, timer_cb_ret1);
43*57ddeb86SDavid Vernet 		bpf_timer_start(timer, 1000, 0);
44*57ddeb86SDavid Vernet 	}
45*57ddeb86SDavid Vernet 
46*57ddeb86SDavid Vernet 	return 0;
47*57ddeb86SDavid Vernet }
48