162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */
362306a36Sopenharmony_ci
462306a36Sopenharmony_ci#include <linux/bpf.h>
562306a36Sopenharmony_ci#include <time.h>
662306a36Sopenharmony_ci#include <errno.h>
762306a36Sopenharmony_ci#include <bpf/bpf_helpers.h>
862306a36Sopenharmony_ci#include "bpf_misc.h"
962306a36Sopenharmony_ci#include "bpf_tcp_helpers.h"
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_cichar _license[] SEC("license") = "GPL";
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_cistruct elem {
1462306a36Sopenharmony_ci	struct bpf_timer t;
1562306a36Sopenharmony_ci};
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_cistruct {
1862306a36Sopenharmony_ci	__uint(type, BPF_MAP_TYPE_ARRAY);
1962306a36Sopenharmony_ci	__uint(max_entries, 1);
2062306a36Sopenharmony_ci	__type(key, int);
2162306a36Sopenharmony_ci	__type(value, struct elem);
2262306a36Sopenharmony_ci} timer_map SEC(".maps");
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic int timer_cb_ret1(void *map, int *key, struct bpf_timer *timer)
2562306a36Sopenharmony_ci{
2662306a36Sopenharmony_ci	if (bpf_get_smp_processor_id() % 2)
2762306a36Sopenharmony_ci		return 1;
2862306a36Sopenharmony_ci	else
2962306a36Sopenharmony_ci		return 0;
3062306a36Sopenharmony_ci}
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_ciSEC("fentry/bpf_fentry_test1")
3362306a36Sopenharmony_ci__failure __msg("should have been in (0x0; 0x0)")
3462306a36Sopenharmony_ciint BPF_PROG2(test_ret_1, int, a)
3562306a36Sopenharmony_ci{
3662306a36Sopenharmony_ci	int key = 0;
3762306a36Sopenharmony_ci	struct bpf_timer *timer;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	timer = bpf_map_lookup_elem(&timer_map, &key);
4062306a36Sopenharmony_ci	if (timer) {
4162306a36Sopenharmony_ci		bpf_timer_init(timer, &timer_map, CLOCK_BOOTTIME);
4262306a36Sopenharmony_ci		bpf_timer_set_callback(timer, timer_cb_ret1);
4362306a36Sopenharmony_ci		bpf_timer_start(timer, 1000, 0);
4462306a36Sopenharmony_ci	}
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_ci	return 0;
4762306a36Sopenharmony_ci}
48