linux/tools/testing/selftests/bpf/progs/free_timer.c
Hou Tao 0a5d2efa38 selftests/bpf: Add test case for the freeing of bpf_timer
The main purpose of the test is to demonstrate the lock problem for the
free of bpf_timer under PREEMPT_RT. When freeing a bpf_timer which is
running on other CPU in bpf_timer_cancel_and_free(), hrtimer_cancel()
will try to acquire a spin-lock (namely softirq_expiry_lock), however
the freeing procedure has already held a raw-spin-lock.

The test first creates two threads: one to start timers and the other to
free timers. The start-timers thread will start the timer and then wake
up the free-timers thread to free these timers when the starts complete.
After freeing, the free-timer thread will wake up the start-timer thread
to complete the current iteration. A loop of 10 iterations is used.

Signed-off-by: Hou Tao <houtao1@huawei.com>
Link: https://lore.kernel.org/r/20250117101816.2101857-6-houtao@huaweicloud.com
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
2025-01-20 09:09:02 -08:00

71 lines
1.4 KiB
C

// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2025. Huawei Technologies Co., Ltd */
#include <linux/bpf.h>
#include <time.h>
#include <bpf/bpf_tracing.h>
#include <bpf/bpf_helpers.h>
#define MAX_ENTRIES 8
struct map_value {
struct bpf_timer timer;
};
struct {
__uint(type, BPF_MAP_TYPE_HASH);
__type(key, int);
__type(value, struct map_value);
__uint(max_entries, MAX_ENTRIES);
} map SEC(".maps");
static int timer_cb(void *map, void *key, struct map_value *value)
{
volatile int sum = 0;
int i;
bpf_for(i, 0, 1024 * 1024) sum += i;
return 0;
}
static int start_cb(int key)
{
struct map_value *value;
value = bpf_map_lookup_elem(&map, (void *)&key);
if (!value)
return 0;
bpf_timer_init(&value->timer, &map, CLOCK_MONOTONIC);
bpf_timer_set_callback(&value->timer, timer_cb);
/* Hope 100us will be enough to wake-up and run the overwrite thread */
bpf_timer_start(&value->timer, 100000, BPF_F_TIMER_CPU_PIN);
return 0;
}
static int overwrite_cb(int key)
{
struct map_value zero = {};
/* Free the timer which may run on other CPU */
bpf_map_update_elem(&map, (void *)&key, &zero, BPF_ANY);
return 0;
}
SEC("syscall")
int BPF_PROG(start_timer)
{
bpf_loop(MAX_ENTRIES, start_cb, NULL, 0);
return 0;
}
SEC("syscall")
int BPF_PROG(overwrite_timer)
{
bpf_loop(MAX_ENTRIES, overwrite_cb, NULL, 0);
return 0;
}
char _license[] SEC("license") = "GPL";