mirror of
https://github.com/sched-ext/scx.git
synced 2024-12-12 11:37:18 +00:00
99d1179866
* enable ide's etc. to work on the bpf.c files this makes it so that clangd and ide tools which use clangd can work on the bpf.c code. nothing should actually be changed outside of that ide/editor environment, all the changes are ifdef'ed on LSP which is set in the added .clangd file. * move intf include out of both sides of ifdef toggle
116 lines
3.5 KiB
C
116 lines
3.5 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
|
|
|
|
enum uei_sizes {
|
|
UEI_REASON_LEN = 128,
|
|
UEI_MSG_LEN = 1024,
|
|
UEI_DUMP_DFL_LEN = 32768,
|
|
};
|
|
|
|
struct user_exit_info {
|
|
int kind;
|
|
s64 exit_code;
|
|
char reason[UEI_REASON_LEN];
|
|
char msg[UEI_MSG_LEN];
|
|
};
|
|
|
|
#ifdef __bpf__
|
|
|
|
#ifdef LSP
|
|
#include "../vmlinux/vmlinux.h"
|
|
#else
|
|
#include "vmlinux.h"
|
|
#endif
|
|
#include <bpf/bpf_core_read.h>
|
|
|
|
#define UEI_DEFINE(__name) \
|
|
char RESIZABLE_ARRAY(data, __name##_dump); \
|
|
const volatile u32 __name##_dump_len; \
|
|
struct user_exit_info __name SEC(".data")
|
|
|
|
#define UEI_RECORD(__uei_name, __ei) ({ \
|
|
bpf_probe_read_kernel_str(__uei_name.reason, \
|
|
sizeof(__uei_name.reason), (__ei)->reason); \
|
|
bpf_probe_read_kernel_str(__uei_name.msg, \
|
|
sizeof(__uei_name.msg), (__ei)->msg); \
|
|
bpf_probe_read_kernel_str(__uei_name##_dump, \
|
|
__uei_name##_dump_len, (__ei)->dump); \
|
|
if (bpf_core_field_exists((__ei)->exit_code)) \
|
|
__uei_name.exit_code = (__ei)->exit_code; \
|
|
/* use __sync to force memory barrier */ \
|
|
__sync_val_compare_and_swap(&__uei_name.kind, __uei_name.kind, \
|
|
(__ei)->kind); \
|
|
})
|
|
|
|
#else /* !__bpf__ */
|
|
|
|
#include <stdio.h>
|
|
#include <stdbool.h>
|
|
|
|
/* no need to call the following explicitly if SCX_OPS_LOAD() is used */
|
|
#define UEI_SET_SIZE(__skel, __ops_name, __uei_name) ({ \
|
|
u32 __len = (__skel)->struct_ops.__ops_name->exit_dump_len ?: UEI_DUMP_DFL_LEN; \
|
|
(__skel)->rodata->__uei_name##_dump_len = __len; \
|
|
RESIZE_ARRAY((__skel), data, __uei_name##_dump, __len); \
|
|
})
|
|
|
|
#define UEI_EXITED(__skel, __uei_name) ({ \
|
|
/* use __sync to force memory barrier */ \
|
|
__sync_val_compare_and_swap(&(__skel)->data->__uei_name.kind, -1, -1); \
|
|
})
|
|
|
|
#define UEI_REPORT(__skel, __uei_name) ({ \
|
|
struct user_exit_info *__uei = &(__skel)->data->__uei_name; \
|
|
char *__uei_dump = (__skel)->data_##__uei_name##_dump->__uei_name##_dump; \
|
|
if (__uei_dump[0] != '\0') { \
|
|
fputs("\nDEBUG DUMP\n", stderr); \
|
|
fputs("================================================================================\n\n", stderr); \
|
|
fputs(__uei_dump, stderr); \
|
|
fputs("\n================================================================================\n\n", stderr); \
|
|
} \
|
|
fprintf(stderr, "EXIT: %s", __uei->reason); \
|
|
if (__uei->msg[0] != '\0') \
|
|
fprintf(stderr, " (%s)", __uei->msg); \
|
|
fputs("\n", stderr); \
|
|
__uei->exit_code; \
|
|
})
|
|
|
|
/*
|
|
* We can't import vmlinux.h while compiling user C code. Let's duplicate
|
|
* scx_exit_code definition.
|
|
*/
|
|
enum scx_exit_code {
|
|
/* Reasons */
|
|
SCX_ECODE_RSN_HOTPLUG = 1LLU << 32,
|
|
|
|
/* Actions */
|
|
SCX_ECODE_ACT_RESTART = 1LLU << 48,
|
|
};
|
|
|
|
enum uei_ecode_mask {
|
|
UEI_ECODE_USER_MASK = ((1LLU << 32) - 1),
|
|
UEI_ECODE_SYS_RSN_MASK = ((1LLU << 16) - 1) << 32,
|
|
UEI_ECODE_SYS_ACT_MASK = ((1LLU << 16) - 1) << 48,
|
|
};
|
|
|
|
/*
|
|
* These macro interpret the ecode returned from UEI_REPORT().
|
|
*/
|
|
#define UEI_ECODE_USER(__ecode) ((__ecode) & UEI_ECODE_USER_MASK)
|
|
#define UEI_ECODE_SYS_RSN(__ecode) ((__ecode) & UEI_ECODE_SYS_RSN_MASK)
|
|
#define UEI_ECODE_SYS_ACT(__ecode) ((__ecode) & UEI_ECODE_SYS_ACT_MASK)
|
|
|
|
#define UEI_ECODE_RESTART(__ecode) (UEI_ECODE_SYS_ACT((__ecode)) == SCX_ECODE_ACT_RESTART)
|
|
|
|
#endif /* __bpf__ */
|
|
#endif /* __USER_EXIT_INFO_H */
|