77 lines
1.5 KiB
C
77 lines
1.5 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright 2026 Google LLC */
|
|
|
|
#include <vmlinux.h>
|
|
#include <bpf/bpf_helpers.h>
|
|
#include "bpf_misc.h"
|
|
|
|
struct bpf_ws_lock;
|
|
|
|
struct bpf_ws_lock *bpf_wakeup_sources_read_lock(void) __ksym;
|
|
void bpf_wakeup_sources_read_unlock(struct bpf_ws_lock *lock) __ksym;
|
|
void *bpf_wakeup_sources_get_head(void) __ksym;
|
|
|
|
SEC("syscall")
|
|
__failure __msg("BPF_EXIT instruction in main prog would lead to reference leak")
|
|
int wakeup_source_lock_no_unlock(void *ctx)
|
|
{
|
|
struct bpf_ws_lock *lock;
|
|
|
|
lock = bpf_wakeup_sources_read_lock();
|
|
if (!lock)
|
|
return 0;
|
|
|
|
return 0;
|
|
}
|
|
|
|
SEC("syscall")
|
|
__failure __msg("access beyond struct")
|
|
int wakeup_source_access_lock_fields(void *ctx)
|
|
{
|
|
struct bpf_ws_lock *lock;
|
|
int val;
|
|
|
|
lock = bpf_wakeup_sources_read_lock();
|
|
if (!lock)
|
|
return 0;
|
|
|
|
val = *(int *)lock;
|
|
|
|
bpf_wakeup_sources_read_unlock(lock);
|
|
return val;
|
|
}
|
|
|
|
SEC("syscall")
|
|
__failure __msg("release kfunc bpf_wakeup_sources_read_unlock expects referenced PTR_TO_BTF_ID passed to R1")
|
|
int wakeup_source_unlock_no_lock(void *ctx)
|
|
{
|
|
struct bpf_ws_lock *lock = (void *)0x1;
|
|
|
|
bpf_wakeup_sources_read_unlock(lock);
|
|
|
|
return 0;
|
|
}
|
|
|
|
SEC("syscall")
|
|
__failure __msg("Possibly NULL pointer passed to trusted")
|
|
int wakeup_source_unlock_null(void *ctx)
|
|
{
|
|
bpf_wakeup_sources_read_unlock(NULL);
|
|
|
|
return 0;
|
|
}
|
|
|
|
SEC("syscall")
|
|
__failure __msg("R0 invalid mem access 'scalar'")
|
|
int wakeup_source_unsafe_dereference(void *ctx)
|
|
{
|
|
struct list_head *head = bpf_wakeup_sources_get_head();
|
|
|
|
if (head->next)
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|