51 lines
1.3 KiB
C
51 lines
1.3 KiB
C
|
/* SPDX-License-Identifier: GPL-2.0 */
|
||
|
|
/*
|
||
|
|
* Define struct user_exit_info which is shared between BPF and userspace parts
|
||
|
|
* to communicate exit status and other information.
|
||
|
|
*
|
||
|
|
* Copyright (c) 2022 Meta Platforms, Inc. and affiliates.
|
||
|
|
* Copyright (c) 2022 Tejun Heo <tj@kernel.org>
|
||
|
|
* Copyright (c) 2022 David Vernet <dvernet@meta.com>
|
||
|
|
*/
|
||
|
|
#ifndef __USER_EXIT_INFO_H
|
||
|
|
#define __USER_EXIT_INFO_H
|
||
|
|
|
||
|
|
struct user_exit_info {
|
||
|
|
int kind;
|
||
|
|
char reason[128];
|
||
|
|
char msg[1024];
|
||
|
|
};
|
||
|
|
|
||
|
|
#ifdef __bpf__
|
||
|
|
|
||
|
|
#include "vmlinux.h"
|
||
|
|
#include <bpf/bpf_core_read.h>
|
||
|
|
|
||
|
|
static inline void uei_record(struct user_exit_info *uei,
|
||
|
|
const struct scx_exit_info *ei)
|
||
|
|
{
|
||
|
|
bpf_probe_read_kernel_str(uei->reason, sizeof(uei->reason), ei->reason);
|
||
|
|
bpf_probe_read_kernel_str(uei->msg, sizeof(uei->msg), ei->msg);
|
||
|
|
/* use __sync to force memory barrier */
|
||
|
|
__sync_val_compare_and_swap(&uei->kind, uei->kind, ei->kind);
|
||
|
|
}
|
||
|
|
|
||
|
|
#else /* !__bpf__ */
|
||
|
|
|
||
|
|
static inline bool uei_exited(struct user_exit_info *uei)
|
||
|
|
{
|
||
|
|
/* use __sync to force memory barrier */
|
||
|
|
return __sync_val_compare_and_swap(&uei->kind, -1, -1);
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline void uei_print(const struct user_exit_info *uei)
|
||
|
|
{
|
||
|
|
fprintf(stderr, "EXIT: %s", uei->reason);
|
||
|
|
if (uei->msg[0] != '\0')
|
||
|
|
fprintf(stderr, " (%s)", uei->msg);
|
||
|
|
fputs("\n", stderr);
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif /* __bpf__ */
|
||
|
|
#endif /* __USER_EXIT_INFO_H */
|