Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1/* SPDX-License-Identifier: GPL-2.0 */
2/*
3 * Copyright (c) 2023 Meta Platforms, Inc. and affiliates.
4 * Copyright (c) 2023 Tejun Heo <tj@kernel.org>
5 * Copyright (c) 2023 David Vernet <dvernet@meta.com>
6 */
7#ifndef __SCHED_EXT_COMMON_H
8#define __SCHED_EXT_COMMON_H
9
10#ifdef __KERNEL__
11#error "Should not be included by BPF programs"
12#endif
13
14#include <stdarg.h>
15#include <stdio.h>
16#include <stdlib.h>
17#include <stdint.h>
18#include <errno.h>
19#include "enum_defs.autogen.h"
20
21typedef uint8_t u8;
22typedef uint16_t u16;
23typedef uint32_t u32;
24typedef uint64_t u64;
25typedef int8_t s8;
26typedef int16_t s16;
27typedef int32_t s32;
28typedef int64_t s64;
29
30#define SCX_BUG(__fmt, ...) \
31 do { \
32 fprintf(stderr, "[SCX_BUG] %s:%d", __FILE__, __LINE__); \
33 if (errno) \
34 fprintf(stderr, " (%s)\n", strerror(errno)); \
35 else \
36 fprintf(stderr, "\n"); \
37 fprintf(stderr, __fmt __VA_OPT__(,) __VA_ARGS__); \
38 fprintf(stderr, "\n"); \
39 \
40 exit(EXIT_FAILURE); \
41 } while (0)
42
43#define SCX_BUG_ON(__cond, __fmt, ...) \
44 do { \
45 if (__cond) \
46 SCX_BUG((__fmt) __VA_OPT__(,) __VA_ARGS__); \
47 } while (0)
48
49/**
50 * RESIZE_ARRAY - Convenience macro for resizing a BPF array
51 * @__skel: the skeleton containing the array
52 * @elfsec: the data section of the BPF program in which the array exists
53 * @arr: the name of the array
54 * @n: the desired array element count
55 *
56 * For BPF arrays declared with RESIZABLE_ARRAY(), this macro performs two
57 * operations. It resizes the map which corresponds to the custom data
58 * section that contains the target array. As a side effect, the BTF info for
59 * the array is adjusted so that the array length is sized to cover the new
60 * data section size. The second operation is reassigning the skeleton pointer
61 * for that custom data section so that it points to the newly memory mapped
62 * region.
63 */
64#define RESIZE_ARRAY(__skel, elfsec, arr, n) \
65 do { \
66 size_t __sz; \
67 bpf_map__set_value_size((__skel)->maps.elfsec##_##arr, \
68 sizeof((__skel)->elfsec##_##arr->arr[0]) * (n)); \
69 (__skel)->elfsec##_##arr = \
70 bpf_map__initial_value((__skel)->maps.elfsec##_##arr, &__sz); \
71 } while (0)
72
73#include "user_exit_info.h"
74#include "compat.h"
75#include "enums.h"
76
77/* not available when building kernel tools/sched_ext */
78#if __has_include(<lib/sdt_task_defs.h>)
79#include "bpf_arena_common.h"
80#include <lib/sdt_task_defs.h>
81#endif
82
83#endif /* __SCHED_EXT_COMMON_H */