compat.bpf.h 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. /*
  3. * Copyright (c) 2024 Meta Platforms, Inc. and affiliates.
  4. * Copyright (c) 2024 Tejun Heo <tj@kernel.org>
  5. * Copyright (c) 2024 David Vernet <dvernet@meta.com>
  6. */
  7. #ifndef __SCX_COMPAT_BPF_H
  8. #define __SCX_COMPAT_BPF_H
  9. #define __COMPAT_ENUM_OR_ZERO(__type, __ent) \
  10. ({ \
  11. __type __ret = 0; \
  12. if (bpf_core_enum_value_exists(__type, __ent)) \
  13. __ret = __ent; \
  14. __ret; \
  15. })
  16. /* v6.12: 819513666966 ("sched_ext: Add cgroup support") */
  17. #define __COMPAT_scx_bpf_task_cgroup(p) \
  18. (bpf_ksym_exists(scx_bpf_task_cgroup) ? \
  19. scx_bpf_task_cgroup((p)) : NULL)
  20. /* v6.12: 4c30f5ce4f7a ("sched_ext: Implement scx_bpf_dispatch[_vtime]_from_dsq()") */
  21. #define __COMPAT_scx_bpf_dispatch_from_dsq_set_slice(it, slice) \
  22. (bpf_ksym_exists(scx_bpf_dispatch_from_dsq_set_slice) ? \
  23. scx_bpf_dispatch_from_dsq_set_slice((it), (slice)) : (void)0)
  24. #define __COMPAT_scx_bpf_dispatch_from_dsq_set_vtime(it, vtime) \
  25. (bpf_ksym_exists(scx_bpf_dispatch_from_dsq_set_vtime) ? \
  26. scx_bpf_dispatch_from_dsq_set_vtime((it), (vtime)) : (void)0)
  27. #define __COMPAT_scx_bpf_dispatch_from_dsq(it, p, dsq_id, enq_flags) \
  28. (bpf_ksym_exists(scx_bpf_dispatch_from_dsq) ? \
  29. scx_bpf_dispatch_from_dsq((it), (p), (dsq_id), (enq_flags)) : false)
  30. #define __COMPAT_scx_bpf_dispatch_vtime_from_dsq(it, p, dsq_id, enq_flags) \
  31. (bpf_ksym_exists(scx_bpf_dispatch_vtime_from_dsq) ? \
  32. scx_bpf_dispatch_vtime_from_dsq((it), (p), (dsq_id), (enq_flags)) : false)
  33. /*
  34. * Define sched_ext_ops. This may be expanded to define multiple variants for
  35. * backward compatibility. See compat.h::SCX_OPS_LOAD/ATTACH().
  36. */
  37. #define SCX_OPS_DEFINE(__name, ...) \
  38. SEC(".struct_ops.link") \
  39. struct sched_ext_ops __name = { \
  40. __VA_ARGS__, \
  41. };
  42. #endif /* __SCX_COMPAT_BPF_H */